package web;

import com.alibaba.druid.support.json.JSONUtils;
import common.util.TemplateUtil;
import common.util.WebUtil;
import model.authority.User;
import model.chart.ChartBuilderParams;
import model.myPanel.MyCharts;
import model.myPanel.MyPanel;
import org.apache.commons.codec.binary.Base64;
import org.apache.shiro.SecurityUtils;
import org.apache.shiro.authc.AuthenticationException;
import org.apache.shiro.authc.IncorrectCredentialsException;
import org.apache.shiro.authc.UnknownAccountException;
import org.apache.shiro.authc.UsernamePasswordToken;
import org.apache.shiro.subject.Subject;
import org.springframework.stereotype.Controller;
import org.springframework.ui.ModelMap;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.context.WebApplicationContext;
import org.springframework.web.context.support.WebApplicationContextUtils;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.servlet.ModelAndView;
import org.springframework.web.servlet.mvc.support.RedirectAttributes;
import service.chart.ChartOption;
import service.chart.TableData;
import service.chart.bar.echarts.Bar;
import service.chart.line.echarts.Line;
import service.chart.pie.echarts.Pie;
import service.myPanel.MyChartsService;
import service.myPanel.MyPanelService;

import javax.annotation.Resource;
import javax.imageio.ImageIO;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpSession;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * Created by ct on 2016/8/18.
 */
@Controller
public class DefaultController {

    @Resource
    private MyPanelService myPanelService;

    @Resource
    private MyChartsService myChartsService;

    @Resource
    private TableData tableData;

    @RequestMapping("/login")
    public String login(User user,  RedirectAttributes redirectAttributes){
        final Subject subject = SecurityUtils.getSubject();
        final UsernamePasswordToken usernamePasswordToken = new UsernamePasswordToken(user.getUserName(), user.getPassword());
        String redirect = "redirect:index.page";
        try {
            subject.login(usernamePasswordToken);
            redirect = "redirect:query.page";
        }catch (final UnknownAccountException uae){
            redirectAttributes.addFlashAttribute("error_message", "身份认证失败,请确认用户名和密码是否正确！");
        }catch (final IncorrectCredentialsException ice) {
            redirectAttributes.addFlashAttribute("error_message", "身份认证失败,请确认用户名和密码是否正确！");
        }catch (final AuthenticationException ae) {
            redirectAttributes.addFlashAttribute("error_message", "身份认证失败,请确认用户名和密码是否正确！");
            usernamePasswordToken.clear();
        }
        return redirect;
    }

    /**
     * 添加设计面板
     *
     * @param myPanel
     * @throws Exception
     */
    @RequestMapping("/addPanel")
    @ResponseBody
    public Object addPanel(MyPanel myPanel) throws Exception {
        return myPanelService.insert(myPanel);
    }

    /**
     * 显示设计面板
     *
     * @param exportId
     * @param map
     * @throws Exception
     */
    @RequestMapping("/showPanel.page")
    public Object showPanel(String exportId, ModelMap map) throws Exception {
        MyPanel myPanel = myPanelService.queryAsObject(exportId);
        map.addAttribute("exportId", myPanel.getExportId());
        map.addAttribute("htmlCode", myPanel.getHtmlCode());
        map.addAttribute("jsCode", myPanel.getJsCode());
        return "panel/designPanel";
    }

    /**
     * 保存当前设计面板并截取非实时预览图保存
     *
     * @param myPanel
     * @throws Exception
     */
    @RequestMapping("/export")
    @ResponseBody
    public Object export(MyPanel myPanel) throws Exception {
        myPanelService.update(myPanel);
        String snapshotImg = WebUtil.snapshotHtmlImageBase64Format(myPanel.getExtraMsg(), myPanel.getExportId());
        MyPanel panel = new MyPanel();
        panel.setExportId(myPanel.getExportId());
        panel.setImg(snapshotImg);
        return myPanelService.update(panel);
    }

    /**
     * 预览
     *
     * @param exportId
     * @param map
     * @throws Exception
     */
    @RequestMapping("/share.page")
    public Object share(String exportId, ModelMap map) throws Exception {
        MyPanel myPanel = myPanelService.queryAsObject(exportId);
        map.addAttribute("htmlCode", myPanel.getHtmlCode());
        map.addAttribute("exportId",exportId);
        return "export/exportTemplate";
    }

    /**
     * 获取预览图表配置
     *
     * @param cids
     * @throws Exception
     */
    @RequestMapping("/getShareOptions")
    @ResponseBody
    public Object getShareOptions(int[] cids,MyCharts myCharts) throws Exception{
        List<MyCharts> list = new ArrayList<>();
        for(int i=0;i<cids.length;i++){
            myCharts.setId(String.valueOf(cids[i]));
            list.add(myChartsService.selectOneChartInfo(myCharts));
        }
        return list;
    }

    /**
     * 跳转至主页面
     *
     * @throws Exception
     */
    @RequestMapping("/query.page")
    public Object query() throws Exception {
        return "panel/myPanel";
    }

    @RequestMapping("/dataAnalysis.page")
    public Object dataAnalysis() throws Exception {
        return "panel/dataAnalysis";
    }

    @RequestMapping("/sqlClient.page")
    public Object sqlClient() throws Exception {
        return "panel/sqlClient";
    }

    @RequestMapping("/index.page")
    public Object index() throws  Exception {
        return "panel/index";
    }

    @RequestMapping("/dashboard.page")
    public Object dashboard() throws  Exception {
        return "panel/dashboard";
    }

    /**
     * 分页查询panel
     *
     * @param myPanel
     * @throws Exception
     */
    @RequestMapping("/selectList")
    @ResponseBody
    public Object selectList(MyPanel myPanel) throws Exception {
        myPanel.setPageSize(myPanel.getPageSize());
        myPanel.setPage(myPanel.getPage());
        Map<String, Object> respMap = new HashMap<>();
        respMap.put("data", myPanelService.queryAsList(myPanel));
        respMap.put("totalPage", myPanel.getTotalPage());
        return respMap;
    }

    /**
     * 删除一块设计面板
     *
     * @param exportId
     * @throws Exception
     */
    @RequestMapping("/deleteOne")
    @ResponseBody
    public Object deleteOne(String exportId) throws Exception {
        return myPanelService.delete(exportId);
    }

    /**
     * 添加我的图表
     *
     * @param myCharts
     * @throws Exception
     */
    @RequestMapping("/addCharts")
    @ResponseBody
    public Object addCharts(MyCharts myCharts) throws  Exception{
        return myChartsService.insert(myCharts);
    }

    /**
     * 更新图表信息
     *
     * @param myCharts
     * @throws Exception
     */
    @RequestMapping("/updateChartInfo")
    @ResponseBody
    public Object updateChartInfo(@RequestHeader(required = false) String imgBase64, MyCharts myCharts) throws Exception {
        Map map = TemplateUtil.genObjFormJson(myCharts.getJsCode(), Map.class);
        map.put("image", imgBase64);
        myCharts.setJsCode(TemplateUtil.genJsonStr4Obj(map));
        return myChartsService.update(myCharts);
    }

    /**
     * 批量更新图表信息
     * @param list
     * @return
     * @throws Exception
     */
    @RequestMapping("/updateChartsInfo")
    @ResponseBody
    public Object updateChartsInfo(@RequestBody List<MyCharts> list) throws Exception {
        Map<String, Object> resMap = new HashMap<>();
        for(MyCharts myCharts : list){
            myChartsService.update(myCharts);
        }
        resMap.put("success", true);
        return resMap;
    }

    @RequestMapping("/selectOneChartInfo")
    @ResponseBody
    public Object selectOneChartInfo(MyCharts myCharts) throws Exception {
        myCharts =  myChartsService.selectOneChartInfo(myCharts);
        return myCharts;
    }

    /**
     * 查询所有图表信息
     *
     * @param myCharts
     * @throws Exception
     */
    @RequestMapping("/selectChartInfo")
    @ResponseBody
    public Object selectList(MyCharts myCharts) throws  Exception {
        myCharts.setPaging(false);
        Map<String, Object> respMap = new HashMap<>();
        respMap.put("data",myChartsService.selectChartInfo(myCharts));
        return respMap;
    }

    /**
     *
     * @param imgFile
     * @return
     * @throws Exception
     */
    @RequestMapping("/imgToBase64")
    @ResponseBody
    public Object imgToBase64(@RequestParam MultipartFile[] imgFile) throws Exception {
        Map<String, Object> respMap = new HashMap<>();
        String imgBase64 = "";
        if(imgFile.length > 0){
            if(ImageIO.read(imgFile[0].getInputStream()) != null){
                byte[] content = imgFile[0].getBytes();
                imgBase64 = Base64.encodeBase64String(content);
                respMap.put("imgBase64",imgBase64);
            }else{
                respMap.put("imgBase64","noImage");
            }
        }else{
            respMap.put("imgBase64","noFile");
        }
        return respMap;
    }

    /**
     * 动态渲染图形
     *
     * @param chartBuilderParams
     * @param request
     * @return
     * @throws Exception
     */
    @RequestMapping(value = "/render", method = RequestMethod.POST, produces = "application/json; charset=UTF-8")
    @ResponseBody
    public Object render(@RequestBody ChartBuilderParams chartBuilderParams, HttpServletRequest request) throws Exception {

        WebApplicationContext context = WebApplicationContextUtils.getWebApplicationContext(request.getServletContext());

        ChartOption chartOption = null;

        if (chartBuilderParams.getChartType() == ChartBuilderParams.ChartType.pie) {
            chartOption = context.getBean(Pie.class);
        } else if (chartBuilderParams.getChartType() == ChartBuilderParams.ChartType.line) {
            chartOption = context.getBean(Line.class);
        } else if (chartBuilderParams.getChartType() == ChartBuilderParams.ChartType.bar) {
            chartOption = context.getBean(Bar.class);
        } else if (chartBuilderParams.getChartType() == ChartBuilderParams.ChartType.table) {
               return tableData.getTableData(chartBuilderParams);
        }

        return TemplateUtil.genJsonStr4Obj(chartOption.transform(chartBuilderParams), true);
    }
}