package com.ks.tool.bkz.controller;

import com.alibaba.fastjson.JSONObject;
import com.google.gson.Gson;
import com.ks.tool.bkz.entity.user.SDLJShareOpenHistory;
import com.ks.tool.bkz.entity.user.UserInfo;
import com.ks.tool.bkz.exception.CardPwdException;
import com.ks.tool.bkz.exception.SDLJShareOpenHistoryException;
import com.ks.tool.bkz.exception.UserException;
import com.ks.tool.bkz.service.manager.RedisManager;
import com.ks.tool.bkz.service.user.SDLJShareOpenHistoryService;
import com.ks.tool.bkz.service.user.UserService;
import com.ks.tool.bkz.service.user.UserUpgradeService;
import com.ks.tool.bkz.util.*;
import com.ks.tool.bkz.util.email.MailSenderUtil;
import com.ks.tool.bkz.vo.user.UserInfoVO;
import io.netty.util.Constant;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.ResponseBody;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;
import java.io.IOException;

@Controller
@RequestMapping("user")
public class UserController {

    private final static String RANDKEY = "LoginImageCode";

    @Resource
    private RedisManager redisManager;

    @Resource
    private UserService userService;

    @Resource
    private SDLJShareOpenHistoryService sdljShareOpenHistoryService;

    @Resource
    private UserUpgradeService userUpgradeService;


    private String getVCodeKey(String account) {

        return "login-vcode-" + StringUtil.Md5(account);
    }

    /**
     * 登录
     *
     * @param account
     * @param vcode
     * @return
     */
    @RequestMapping("login")
    @ResponseBody
    public String login(String account, String vcode, HttpServletRequest request) {
        if (StringUtil.isNullOrEmpty(account))
            return JsonUtil.loadFalseResult(1, "账号不能为空");
        if (StringUtil.isNullOrEmpty(vcode))
            return JsonUtil.loadFalseResult(1, "验证码不能为空");
        String key = getVCodeKey(account);
        String oldVcode = redisManager.get(key);
//        if (oldVcode == null || !oldVcode.equalsIgnoreCase(vcode))
//            return JsonUtil.loadFalseResult(1, "验证码错误");
        redisManager.delete(key);
        account = account.trim();
        UserInfo user = userService.selectByAccount(account);
        if (user == null) {
            try {
                user = new UserInfo();
                user.setAccount(account);
                user.setState(UserInfo.STATE_NORMAL);
                userService.addUser(user);
                userService.login(account, request.getRemoteHost() + ":" + request.getRemotePort());

            } catch (UserException e) {
                return JsonUtil.loadFalseResult(e.getCode(), e.getMsg());
            } catch (Exception e) {
                return JsonUtil.loadFalseResult(1000, "服务器内部出错，请稍后再试！");
            }
        } else {
            userService.login(account, request.getRemoteHost() + ":" + request.getRemotePort());
        }

        SDLJShareOpenHistory history = sdljShareOpenHistoryService.selectLatestHistory(user.getId());

        String token = UserUtil.getToken(account, System.currentTimeMillis());
        return JsonUtil.loadTrueResult(JsonUtil.getSimpleGson().toJson(new UserInfoVO(user.getAccount(), token, history == null ? null : history.getExpireTime())));
    }

    @RequestMapping(value = "getUserInfo", method = RequestMethod.POST)
    @ResponseBody
    public String getUserInfo(HttpServletRequest request) {
        String token = request.getHeader("token");
        String account = UserUtil.getAccountFromToken(token);
        UserInfo user = userService.selectByAccount(account);
        if (user != null) {
            SDLJShareOpenHistory history = sdljShareOpenHistoryService.selectLatestHistory(user.getId());
            return JsonUtil.loadTrueResult(JsonUtil.getSimpleGson().toJson(new UserInfoVO(user.getAccount(), "", history == null ? null : history.getExpireTime())));
        }
        return JsonUtil.loadFalseResult(1, "用户不存在");
    }


    /**
     * 分享版续费
     *
     * @param request
     * @return
     */
    @RequestMapping(value = "renewShare", method = RequestMethod.POST)
    @ResponseBody
    public String renewShare(String card, String pwd, HttpServletRequest request) {
        if (StringUtil.isNullOrEmpty(card) || StringUtil.isNullOrEmpty(pwd))
            return JsonUtil.loadFalseResult(1, "请输入卡号和密码");
        String token = request.getHeader("token");
        String account = UserUtil.getAccountFromToken(token);
        UserInfo user = userService.selectByAccount(account);
        if (user != null) {
            try {
                userUpgradeService.upgradeSDLJShare(user.getId(), card, pwd);
                SDLJShareOpenHistory history = sdljShareOpenHistoryService.selectLatestHistory(user.getId());
                if (history == null) {
                    try {
                        Thread.sleep(100);
                    } catch (Exception e) {
                    }
                    history = sdljShareOpenHistoryService.selectLatestHistory(user.getId());
                }
                return JsonUtil.loadTrueResult(JsonUtil.getSimpleGson().toJson(new UserInfoVO(user.getAccount(), "", history.getExpireTime())));
            } catch (UserException e) {
                return JsonUtil.loadFalseResult(2, e.getMsg());
            } catch (CardPwdException e) {
                return JsonUtil.loadFalseResult(3, e.getMsg());
            } catch (SDLJShareOpenHistoryException e) {
                return JsonUtil.loadFalseResult(4, e.getMsg());
            }

        } else {
            return JsonUtil.loadFalseResult(1, "用户不存在");
        }
    }


    /**
     * 获取邮箱验证码
     *
     * @param email
     * @param request
     * @return
     */
    @RequestMapping("getVerifyCode")
    @ResponseBody
    public String getVerifyCode(String email, String vcode, HttpServletRequest request) {
        if (!StringUtil.isEmail(email)) {
            return JsonUtil.loadFalseResult(1, "邮箱格式不正确");
        }

        HttpSession session = request.getSession(true);
        if (session.getAttribute(RANDKEY) == null || !session.getAttribute(RANDKEY).toString().equalsIgnoreCase(vcode))
            return JsonUtil.loadFalseResult(1, "图片验证码错误");
        session.removeAttribute(RANDKEY);
        String source = "0123456789";
        String code = "";
        for (int i = 0; i < 6; i++) {
            code += source.charAt((int) (Math.random() * source.length()));
        }
        MailSenderUtil.sendEmail(email, "爆款猪登录验证", "登录验证码为：" + code);
        redisManager.save(getVCodeKey(email), code, 5 * 60);
        return JsonUtil.loadTrueResult("");
    }

    /**
     * 图片验证码
     *
     * @param request
     * @param response
     */
    @RequestMapping(value = "/imgcode", method = RequestMethod.GET)
    public void imgCode(HttpServletRequest request, HttpServletResponse response) {
        response.setHeader("Pragma", "No-cache");
        response.setHeader("Cache-Control", "no-cache");
        response.setDateHeader("Expires", 0);
        response.setContentType("image/jpeg");

        // 生成随机字串
        String verifyCode = VerifyCodeUtil.generateVerifyCode(4);
        // 存入会话session
        HttpSession session = request.getSession(true);
        session.setAttribute(RANDKEY, verifyCode.toLowerCase());
        // 生成图片
        int w = 200, h = 80;
        try {
            VerifyCodeUtil.outputImage(w, h, response.getOutputStream(), verifyCode);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

}