package com.stylefeng.guns.modular.system.controller.system;

import com.baomidou.mybatisplus.mapper.EntityWrapper;
import com.stylefeng.guns.core.common.constant.JwtConstants;
import com.stylefeng.guns.core.shiro.ShiroKit;
import com.stylefeng.guns.core.shiro.ShiroUser;
import com.stylefeng.guns.core.util.JwtTokenUtil;
import com.stylefeng.guns.modular.system.model.MenuEntryConfig;
import com.stylefeng.guns.modular.system.model.User;
import com.stylefeng.guns.modular.system.service.IMenuEntryConfigService;
import com.stylefeng.guns.modular.system.service.IUserService;
import com.stylefeng.guns.modular.system.util.ResultUtil;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.ResponseBody;

import javax.servlet.http.HttpServletRequest;
import java.io.UnsupportedEncodingException;
import java.net.URLEncoder;
import java.util.List;

/**
 * 后台嵌入外部页面。
 */
@Controller
@RequestMapping("")
public class EmbeddedPageController {

    private static final String EMBEDDED_IFRAME_VIEW = "/system/embedded/iframe.html";

    @Value("${embedded.route-switch-url}")
    private String routeSwitchUrl;

    @Value("${embedded.statistical-table-url}")
    private String statisticalTableUrl;

    @Autowired
    private IMenuEntryConfigService menuEntryConfigService;

    @Autowired
    private IUserService userService;

    @RequestMapping("/embedded/route-switch")
    public String routeSwitch(Model model) throws UnsupportedEncodingException {
        model.addAttribute("title", "路由切换页");
        model.addAttribute("iframeSrc", buildTokenUrl(routeSwitchUrl));
        return EMBEDDED_IFRAME_VIEW;
    }

    @RequestMapping("/embedded/statistical-table")
    public String statisticalTable(Model model) throws UnsupportedEncodingException {
        model.addAttribute("title", "统计板");
        model.addAttribute("iframeSrc", buildTokenUrl(statisticalTableUrl));
        return EMBEDDED_IFRAME_VIEW;
    }

    @RequestMapping("/gunsApi/tCompany/entry_config_list")
    @ResponseBody
    public Object entryConfigList(HttpServletRequest request) {
        Integer companyId = getCurrentCompanyId(request);
        if (companyId == null) {
            return ResultUtil.tokenErr("登录信息无效");
        }
        return ResultUtil.success(getMenuEntryConfigList(companyId));
    }

    private List<MenuEntryConfig> getMenuEntryConfigList(Integer companyId) {
        return menuEntryConfigService.selectList(
                new EntityWrapper<MenuEntryConfig>().eq("openCityId", companyId).orderBy("id", true));
    }

    private Integer getCurrentCompanyId(HttpServletRequest request) {
        String token = getBearerToken(request);
        if (token == null) {
            return null;
        }
        String userId = JwtTokenUtil.getUsernameFromToken(token);
        User user = userService.selectById(userId);
        return user == null ? null : user.getObjectId();
    }

    private String getBearerToken(HttpServletRequest request) {
        String requestHeader = request.getHeader(JwtConstants.AUTH_HEADER);
        if (requestHeader != null && requestHeader.startsWith("Bearer ")) {
            return requestHeader.substring(7);
        }
        return null;
    }

    private String buildTokenUrl(String baseUrl) throws UnsupportedEncodingException {
        return baseUrl + generateCurrentUserToken();
    }

    private String generateCurrentUserToken() throws UnsupportedEncodingException {
        ShiroUser user = ShiroKit.getUser();
        if (user == null || user.getId() == null) {
            return "";
        }
        String token = JwtTokenUtil.generateToken(user.getId().toString());
        return URLEncoder.encode(token, "UTF-8");
    }
}