#pragma comment(linker, "/subsystem:\"Windows\" /entry:\"mainCRTStartup\"")
#include<windows.h>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<fstream>
#include<sstream>
#include<iomanip>
#include<string>
#include<thread>
#include<vector>
#include<ctime>
#include<map>
using namespace std;

ofstream logFile;                             // 日志文件
vector<string> blackList;                     // 黑名单
int blackListTimeCount = 0;                   // 黑名单时间计数
const int MAX_BLACKLIST_TIME_COUNT = 1800;    // 黑名单最大时间
const int TOLENT_MORE_TIME = 1200;            // 超出黑名单时间后，多长时间禁止摸鱼
const int PUNISH_CLICKS = 20;                 // 连续点击次数超过多少次后允许继续
const int PUNISH_CLICKS_RND = 10;             // 连续点击次数向上飘逸上限

// 从字符串中读取黑名单
void getBlackList(string str){
    if(str.substr(0, 10)!="BLACK_LIST") return;
    str = str.substr(str.find("=")+1);
    stringstream ss(str); string item;
    while(ss>>quoted(item)){
        if(item==",") continue;
        blackList.push_back(item);
    }
}
// 保存黑名单时间
void saveData(){
    // 保存黑名单时间到save.txt
    FILE *fp = fopen("save.txt", "w");
    if(fp!=NULL){
        time_t timeNow = time(0);
        tm *curTime = localtime(&timeNow);
        if(curTime->tm_hour<4){
            timeNow -= 86400;
            curTime = localtime(&timeNow);
        }
        fprintf(fp, "%d %d %d %d", blackListTimeCount,curTime->tm_year,curTime->tm_mon,curTime->tm_mday);
        fclose(fp);
    }
}
// 加载黑名单时间
void loadData(){
    FILE *fp = fopen("save.txt", "r");
    if(fp!=NULL){
        int curYear,curMonth,curDay;
        fscanf(fp, "%d%d%d%d", &blackListTimeCount,&curYear,&curMonth,&curDay);
        time_t timeNow = time(0);
        tm *curTime = localtime(&timeNow);
        if(curTime->tm_hour<4){
            timeNow -= 86400;
            curTime = localtime(&timeNow);
        }
        if(curTime->tm_year!=curYear||curTime->tm_mon!=curMonth||curTime->tm_mday!=curDay){
            blackListTimeCount = 0;
        }
        fclose(fp);
    }
}
// 获取当前时间字符串
string getTimeStr(){
    time_t now = time(0);
    tm *ltm = localtime(&now);
    char timeStr[1024];
    strftime(timeStr, sizeof(timeStr), "[%Y-%m-%d %H:%M:%S]", ltm);
    return (string)timeStr;
}
// 初始化程序
void initMonitor(){
    ShowWindow(GetConsoleWindow(), SW_HIDE);   // 隐藏控制台窗口
    system("chcp 936 >>nul");                  // 设置编码为GBK
    FreeConsole();                             // 隐藏控制台窗口    
    srand((unsigned)time(0));                  // 初始化随机数种子
    logFile.open("selfMonitor.log", ios::out); // 打开日志文件
    ifstream ifs(".env");
    if(ifs.is_open()){
        string line;
        while(getline(ifs, line))
            getBlackList(line);
        ifs.close();
    }
    loadData();
    logFile<<getTimeStr()<<"- INFO: 自监控程序启动，冰澄向你问好！"<<endl;
    logFile<<getTimeStr()<<"- INFO: 黑名单列表："<<endl;
    for(int i=0;i<blackList.size();i++)
        logFile<<'\"'<<blackList[i]<<'\"'<<endl;
    logFile<<getTimeStr()<<"- INFO: 当前黑名单时间计数："<<blackListTimeCount<<endl;
}

bool beginWithMuteFlag=false;
// TODO: 设置 muteAll 和 unmuteAll 函数，实现暂停播放功能
void muteAll(){ // 目前效果为：按压空格键（已取消）
    // keybd_event(VK_SPACE, 0, 0, 0); Sleep(1);
    // keybd_event(VK_SPACE, 0, KEYEVENTF_KEYUP, 0);
}
void unmuteAll(){muteAll();}
// 判断是否是早上四点
inline bool judgeEarlyMorning(){
    time_t now = time(0);
    tm *ltm = localtime(&now);
    if(ltm->tm_hour==4&&ltm->tm_min<5)
        return true;
    return false;
}
// 获取当前窗口标题
string getForegroundTitle(HWND hwnd){
    char title[1024];
    if (hwnd == NULL)   return "";
    if(GetWindowTextA(hwnd, title, 1024) == 0){
        logFile<<getTimeStr()<<"- WARN: GetWindowTextW failed at hwnd: "<<hwnd<<endl;
        logFile<<getTimeStr()<<"- WARN: Error content: "<<GetLastError()<<endl;
        return "";
    }
    return (string)title;
}
// 判断是否在黑名单中
bool inBlackList(string title){
    if(title=="") return false;
    bool isInBlackList = false;
    for(string blackTitle:blackList)
        if(title.find(blackTitle)!=string::npos)
            return true;
    return false;
}
// 关闭过程，通过复杂的操作迫使用户思考是否应该继续摸鱼
bool closingGame(HWND hwnd){
    bool isContinue = MessageBoxA(hwnd, "半小时到了！你确定你需要继续看这个？", "半个小时提醒", MB_YESNO) == IDYES;
    if(!isContinue) return false;
    MessageBoxA(hwnd, "如果你真的确定的话，按照我的指令连续点正确选项：", "唉，拿你没办法呢", MB_OK);
    for(int i=PUNISH_CLICKS+rand()%(PUNISH_CLICKS_RND+1);i>0;i--){
        if(rand()&1) isContinue = MessageBoxA(hwnd, "先思考一下，你现在真的有用电脑的必要吗？\n你现在应该做什么？\n想好了按确认", "请思考请思考", MB_YESNO) == IDYES;
        else         isContinue = MessageBoxA(hwnd, "先思考一下，你现在真的有用电脑的必要吗？\n你现在应该做什么？\n想好了按取消", "请思考请思考", MB_YESNO) == IDNO;
        if(!isContinue){
            MessageBoxA(hwnd, "看来你并没有真正下定决心呢……", "叹气……", MB_OK);
            return false;
        }
    }
    return true;
}
map<HWND,thread> warningSet;
// 启动超出时间限制禁言程序
void operateTimeExceeded(HWND hwnd){
    logFile<<getTimeStr()<<"- WARN: 黑名单时间超过"<<MAX_BLACKLIST_TIME_COUNT<<"秒，开始禁言程序"<<endl;
    if(warningSet.find(hwnd)!=warningSet.end()) return;
    warningSet[hwnd] = thread([hwnd](){
        muteAll();
        if(closingGame(hwnd)){
            MessageBoxA(hwnd, "好吧，看来你确实有急事，去忙吧，你再用二十分钟后我还会回来的！", "你先忙吧", MB_OK);
            blackListTimeCount = MAX_BLACKLIST_TIME_COUNT - TOLENT_MORE_TIME;
        }else{
            MessageBoxA(hwnd, "根据反摸鱼条例，你今天不准再摸鱼了！\n你有五秒钟保存重要文件，我数到五就强制关掉！","True Ending",MB_OK);
            Sleep(5*1000);
            PostMessageA(hwnd, WM_CLOSE, 0, 0);
        }
        unmuteAll();
        warningSet.erase(hwnd);
    });
    warningSet[hwnd].detach();
}
// 启动提醒程序
void operateTimeWarning(HWND hwnd,string warningMsg){
    logFile<<getTimeStr()<<"- INFO: 当前黑名单时间到达"<<blackListTimeCount<<"秒，开始提醒程序"<<endl;
    if(warningSet.find(hwnd)!=warningSet.end()) return;
    warningSet[hwnd] = thread([hwnd,warningMsg](){
        muteAll();
        MessageBoxA(hwnd, warningMsg.c_str(), "摸鱼提醒", MB_OK);
        unmuteAll();
        warningSet.erase(hwnd);
    });
    warningSet[hwnd].detach();
}
int main(){
    initMonitor();
    while(true){
        if(judgeEarlyMorning())
            blackListTimeCount = 0;
        HWND hwnd = GetForegroundWindow();       // 获取当前窗口句柄
        string title=getForegroundTitle(hwnd);   // 获取窗口标题
        if(inBlackList(title)){                  // 判断是否在黑名单中
            blackListTimeCount++;
            if(blackListTimeCount>=MAX_BLACKLIST_TIME_COUNT)
                operateTimeExceeded(hwnd);
            else if (blackListTimeCount%600==0)
                operateTimeWarning(hwnd,"你摸鱼已经十分钟了，冰澄命令你停下！");
            else if (blackListTimeCount%300==0)
                operateTimeWarning(hwnd,"你在摸鱼吗，又摸了五分钟了，冰澄提醒你正确休息");
        }
        Sleep(1000);                             // 休眠1秒钟
        static int clockCount = 0;               // 存档计时器
        if((++clockCount)%60==0)                 // 每分钟存档一次
            saveData();
    }
    return 0;
}