package com.terra.system.controller.all;

import com.terra.system.annotation.SysLog;
import com.terra.system.config.PropertiesConfig;
import com.terra.system.entity.all.ResponseMsg;
import com.terra.system.entity.all.SettingData;
import com.terra.system.entity.all.StaticData;
import com.terra.system.entity.sys.LoginEntity;
import com.terra.system.entity.sys.OperateEntity;
import com.terra.system.entity.sys.TokenEntity;
import com.terra.system.entity.sys.UserEntity;
import com.terra.system.helper.*;
import com.terra.system.service.all.ScheduleService;
import com.terra.system.service.all.SignService;
import com.terra.system.service.sys.LoginService;
import com.terra.system.service.sys.OperateService;
import com.terra.system.service.sys.TokenService;
import com.terra.system.service.sys.UserService;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.Parameter;
import io.swagger.v3.oas.annotations.Parameters;
import javax.annotation.Resource;

import io.swagger.v3.oas.annotations.tags.Tag;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.servlet.ModelAndView;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

/**
 * 签名控制器
 * @author WWW
 * @date   2022-09-21
 */
@Tag(name = "运维管理\\签名管理")
@RestController
@RequestMapping("/sign")
public class SignController extends BaseController {
    @Resource
    UserService userService;

    @Resource
    LoginService loginService;

    @Resource
    TokenService tokenService;

    @Resource
    SignService signService;

    @Resource
    private OperateService operateService;

    @Resource
    PropertiesConfig propertiesConfig;

    @Resource
    private ScheduleService scheduleService;

    @SysLog()
    @Operation(summary = "跳转首页")
    @GetMapping({"/", "/toIndex"})
    public ModelAndView toIndex(ModelAndView mv, HttpServletRequest req) {
        mv.setViewName("index");

        UserEntity ue = tokenService.getCurrentUser(req);
        if (ue != null) {
            mv.addObject("msg", "Hello，" + ue.getUname() + " ！");
        }

        return mv;
    }

    @SysLog()
    @Operation(summary = "跳转登录页")
    @GetMapping("/toLogin")
    public ModelAndView toLogin(ModelAndView mv) {
        mv.setViewName("login");

        return mv;
    }

    @SysLog()
    @Operation(summary = "数据库监控")
    @GetMapping(value = "/toDruid")
    public ModelAndView toDruid(HttpServletRequest req, HttpServletResponse res) {
        ModelAndView mv = new ModelAndView();
        mv.setViewName("druid");

        try {
            UserEntity ue = tokenService.getCurrentUser(req);
            if (ue != null) {
                String sessionId = WebHelper.getCookieByKey(StaticData.DRUID_COOKIE_KEY, req);
                if (StringHelper.isNull(sessionId)) {
                    signService.loginDruid(req, res);
                }
            }
        } catch (Exception ex) {
            log.error(ex.getMessage(), ex);
        }

        return mv;
    }

    @SysLog()
    @Operation(summary = "资源监控")
    @GetMapping(value = "/toMonitor")
    public ModelAndView toMonitor(ModelAndView mv, HttpServletRequest req, HttpServletResponse res) {
        try {
            mv.setViewName("redirect:/toLogin");

            UserEntity ue = tokenService.getCurrentUser(req);
            if (ue != null) {
                mv.setViewName("monitor");
            }
        } catch (Exception ex) {
            log.error(ex.getMessage(), ex);
        }

        return mv;
    }

    @SysLog()
    @Operation(summary = "登录")
    @Parameters({
            @Parameter(name = "user", description = "用戶名", example = "")
    })
    @PostMapping(value = "/login", produces = "application/json; charset=UTF-8")
    public ResponseMsg<TokenEntity> login(@RequestBody UserEntity user, HttpServletRequest req, HttpServletResponse res) {
        try {
            String str = userService.validateLoginPwd(user);
            if (str != null) {
                return fail(str, null);
            }

            UserEntity ue = userService.selectByUid(user.getUid());
            if (ue == null) {
                return fail("用户名不存在", null);
            }

            if (!Md5Helper.validatePassword(user.getPwd(), ue.getPwd())) {
                tokenService.setPwdErrCache(ue);
                return fail("密码不正确", null);
            }

            LoginEntity le = loginService.getNewLogin(ue.getId(), 1, 1, 1, req);
            Integer rows = loginService.insertLogin(le);
            if (rows == 0) {
                return fail("创建登录日志失败", null);
            }

            TokenEntity te = tokenService.getNewToken(ue, req);
            rows = tokenService.insertToken(te);
            if (rows == 0) {
                return fail("创建令牌失败", null);
            }

            te.setAutoLogOut(SettingData.AUTO_LOGOUT);
            tokenService.saveToken(ue, te, req, res);

            int onlineUser = scheduleService.countOnlineUsers();
            String msg = onlineUser >= SettingData.MAX_USER_LOGIN ? "警告：系统已经到达用户访问量的上限！" : "";

            return success(msg, te);
        } catch (Exception ex) {
            return fail(ex, null);
        }
    }

    @SysLog()
    @Operation(summary = "登出")
    @GetMapping(value = "/logout")
    public ResponseMsg<Boolean> logout(HttpServletRequest req, HttpServletResponse res) {
        try {
            String token = WebHelper.getToken(req);
            if (StringHelper.isEmpty(token)) {
                return fail("没有检测到令牌", false);
            }

            Boolean flag = tokenService.logout(token, req, res);

            return success(flag ? "登出成功" : "登出失败", flag);
        } catch (Exception ex) {
            return fail(ex, false);
        }
    }

    @SysLog()
    @Operation(summary = "检查是/否登录")
    @GetMapping("/check")
    public ResponseMsg<Boolean> check(HttpServletRequest req, HttpServletResponse res) {
        try {
            Boolean flag = tokenService.isLogin(req, res);
            if (flag) {
                // 写日志
                UserEntity ue = tokenService.getCurrentUser(req);
                LoginEntity le = loginService.getNewLogin(ue.getId(), 1, 2, 1, req);
                Integer rows = loginService.insertLogin(le);
            }

            return success(flag ? "用户已登录" : "用户未登录", flag);
        } catch (Exception ex) {
            return fail(ex, false);
        }
    }

    @SysLog()
    @Operation(summary = "获取当前用户")
    @GetMapping("/getCurrentUser")
    public ResponseMsg<UserEntity> getCurrentUser(HttpServletRequest req) {
        try {
            UserEntity ue = tokenService.getCurrentUser(req);
            if (ue == null) {
                return fail("没有找到", null);
            }

            return success(ue);
        } catch (Exception ex) {
            return fail(ex, null);
        }
    }

    @SysLog()
    @Operation(summary = "获取RSA加密公钥")
    @GetMapping("/getPublicKey")
    public ResponseMsg<String> getPublicKey() {
        try {
            String key = RsaHelper.getPublicKey();

            return success(key);
        } catch (Exception ex) {
            return fail(ex, null);
        }
    }

    @Operation(summary = "插入操作日志")
    @GetMapping("/insertOpLog")
    public ResponseMsg<Object> insertOpLog(String m1, String m2, HttpServletRequest req) {
        try {
            if (StringHelper.isEmpty(m1)) {
                return fail("一级模块必填", 0);
            }

            OperateEntity oe = new OperateEntity();
            oe.setIp(WebHelper.getIpAddress(req));
            oe.setModular1(m1);
            oe.setModular2(m2);
            oe.setType(0);

            UserEntity ue = tokenService.getCurrentUser(req);
            if (ue != null) {
                oe.setUserid(ue.getId());
            }

            int rows = operateService.insertOperate(oe);

            return success(rows);
        } catch (Exception ex) {
            return fail(ex, 0);
        }
    }
}