#   <AUTO_MAA:A MAA Multi Account Management and Automation Tool>
#   Copyright © <2024> <DLmaster361>

#   This file is part of AUTO_MAA.

#   AUTO_MAA is free software: you can redistribute it and/or modify
#   it under the terms of the GNU General Public License as published
#   by the Free Software Foundation, either version 3 of the License,
#   or (at your option) any later version.

#   AUTO_MAA is distributed in the hope that it will be useful,
#   but WITHOUT ANY WARRANTY; without even the implied warranty
#   of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See
#   the GNU General Public License for more details.

#   You should have received a copy of the GNU General Public License
#   along with AUTO_MAA. If not, see <https://www.gnu.org/licenses/>.

#   DLmaster_361@163.com

import os
import sys
import subprocess
import atexit
import sqlite3
import datetime
import time
import json
from termcolor import colored

#配置MAA运行参数
def setmaa(s,tel,game):
    with open(setpath,"r",encoding="utf-8") as f:
        data=json.load(f)
    if s==0:
        data["Configurations"]["Default"]["MainFunction.ActionAfterCompleted"]="ExitEmulatorAndSelf"        #完成后退出MAA和模拟器
        data["Configurations"]["Default"]["Start.RunDirectly"]="True"       #启动MAA后直接运行
        data["Configurations"]["Default"]["Start.StartEmulator"]="True"     #启动MAA后自动开启模拟器
    elif s==1:
        data["Configurations"]["Default"]["Start.AccountName"]=tel[:3]+"****"+tel[7:]       #账号切换
        data["Configurations"]["Default"]["TaskQueue.WakeUp.IsChecked"]="True"        #开始唤醒
        data["Configurations"]["Default"]["TaskQueue.Recruiting.IsChecked"]="False"        #自动公招
        data["Configurations"]["Default"]["TaskQueue.Base.IsChecked"]="False"      #基建换班
        data["Configurations"]["Default"]["TaskQueue.Combat.IsChecked"]="True"        #刷理智
        data["Configurations"]["Default"]["TaskQueue.Mission.IsChecked"]="False"       #领取奖励
        data["Configurations"]["Default"]["TaskQueue.Mall.IsChecked"]="False"       #获取信用及购物
        data["Configurations"]["Default"]["MainFunction.Stage1"]="Annihilation"     #主关卡
        data["Configurations"]["Default"]["MainFunction.Stage2"]=""     #备选关卡1
        data["Configurations"]["Default"]["MainFunction.Stage3"]=""     #备选关卡2
        data["Configurations"]["Default"]["Fight.RemainingSanityStage"]=""      #剩余理智关卡
        data["Configurations"]["Default"]["MainFunction.Series.Quantity"]="1"        #连战次数
        data["Configurations"]["Default"]["Penguin.IsDrGrandet"]="False"     #博朗台模式
        data["Configurations"]["Default"]["GUI.CustomStageCode"]="True"        #手动输入关卡名
        data["Configurations"]["Default"]["GUI.UseAlternateStage"]="False"        #使用备选关卡
        data["Configurations"]["Default"]["Fight.UseRemainingSanityStage"]="False"        #使用剩余理智
        data["Configurations"]["Default"]["Fight.UseExpiringMedicine"]="True"        #无限吃48小时内过期的理智药
    elif s==2:
        data["Configurations"]["Default"]["Start.AccountName"]=tel[:3]+"****"+tel[7:]       #账号切换
        data["Configurations"]["Default"]["TaskQueue.WakeUp.IsChecked"]="True"        #开始唤醒
        data["Configurations"]["Default"]["TaskQueue.Recruiting.IsChecked"]="True"        #自动公招
        data["Configurations"]["Default"]["TaskQueue.Base.IsChecked"]="True"      #基建换班
        data["Configurations"]["Default"]["TaskQueue.Combat.IsChecked"]="True"        #刷理智
        data["Configurations"]["Default"]["TaskQueue.Mission.IsChecked"]="True"       #领取奖励
        data["Configurations"]["Default"]["TaskQueue.Mall.IsChecked"]="True"       #获取信用及购物
        data["Configurations"]["Default"]["MainFunction.Stage1"]=game    #主关卡
        data["Configurations"]["Default"]["MainFunction.Stage2"]=""     #备选关卡1
        data["Configurations"]["Default"]["MainFunction.Stage3"]=""     #备选关卡2
        data["Configurations"]["Default"]["Fight.RemainingSanityStage"]=""      #剩余理智关卡
        if game=="1-7":
            data["Configurations"]["Default"]["MainFunction.Series.Quantity"]="6"        #连战次数
        else:
            data["Configurations"]["Default"]["MainFunction.Series.Quantity"]="1"        #连战次数
        data["Configurations"]["Default"]["Penguin.IsDrGrandet"]="False"     #博朗台模式
        data["Configurations"]["Default"]["GUI.CustomStageCode"]="True"        #手动输入关卡名
        data["Configurations"]["Default"]["GUI.UseAlternateStage"]="False"        #使用备选关卡
        data["Configurations"]["Default"]["Fight.UseRemainingSanityStage"]="False"        #使用剩余理智
        data["Configurations"]["Default"]["Fight.UseExpiringMedicine"]="True"        #无限吃48小时内过期的理智药
    with open(setpath,"w",encoding="utf-8") as f:
        json.dump(data,f)
    return True

#执行MAA任务
def runmaa(id,tel,game,num=3):
    #开始运行
    for i in range(num):
        global idnew,idold,idfail,idall,logx,logi
        runbook=[False for i in range(2)]
        for j in range(2):
            #配置MAA
            setmaa(j+1,tel,game)
            #创建MAA任务
            print(colored("等待中~",'yellow'))
            time.sleep(10)
            maa=subprocess.Popen([maapath])
            maapid=maa.pid
            #等待MAA启动
            idsuccess=idnew+idold
            idwait=[idx for idx in idall if not idx in idsuccess+idfail+[id]]
            os.system('cls')
            if i==0:
                if j==0:
                    print(colored("正在代理：",'white')+colored(id+"-剿灭",'blue'))
                elif j==1:
                    print(colored("正在代理：",'white')+colored(id+"-日常",'blue'))
            else:
                if j==0:
                    print(colored("正在代理：",'white')+colored(id+"-剿灭",'light_blue'))
                elif j==1:
                    print(colored("正在代理：",'white')+colored(id+"-日常",'light_blue'))
            print(colored("等待代理：",'white')+colored('，'.join(idwait),'yellow'))
            print(colored("代理成功：",'white')+colored('，'.join(idsuccess),'green'))
            print(colored("代理失败：",'white')+colored('，'.join(idfail),'red'))
            print(colored("运行日志：",'white')+colored("等待MAA初始化",'light_green'))
            time.sleep(60)
            #监测MAA运行状态
            while True:
                #打印基本信息
                os.system('cls')
                if i==0:
                    if j==0:
                        print(colored("正在代理：",'white')+colored(id+"-剿灭",'blue'))
                    elif j==1:
                        print(colored("正在代理：",'white')+colored(id+"-日常",'blue'))
                else:
                    if j==0:
                        print(colored("正在代理：",'white')+colored(id+"-剿灭",'light_blue'))
                    elif j==1:
                        print(colored("正在代理：",'white')+colored(id+"-日常",'light_blue'))
                print(colored("等待代理：",'white')+colored('，'.join(idwait),'yellow'))
                print(colored("代理成功：",'white')+colored('，'.join(idsuccess),'green'))
                print(colored("代理失败：",'white')+colored('，'.join(idfail),'red'))
                print(colored("运行日志：",'white'))
                #读取并保存MAA日志
                with open(logpath,'r',encoding='utf-8') as f:
                    logs=f.readlines()[-1:-11:-1]
                    print(colored(''.join(logs[::-1]),'light_green'),end='')
                    log=''.join(logs)
                    logx[logi]=log
                    logi=(logi+1) % len(logx)
                #判断MAA程序运行状态
                if ("任务已全部完成！" in log):
                    runbook[j]=True
                    break
                elif ("请检查连接设置或尝试重启模拟器与 ADB 或重启电脑" in log) or ("已停止" in log) or ("MaaAssistantArknights GUI exited" in log) or timeout():
                    os.system('taskkill /F /T /PID '+str(maapid))
                    break
                if j==0:
                    time.sleep(40)
                elif j==1:
                    time.sleep(10)
        if runbook[0] and runbook[1]:
            return True
    return False

#检查是否超时
def timeout():
    global logx
    log0=logx[0]
    for i in range(len(logx)):
        if logx[i]!=log0:
            return False
    return True

#更新已完成用户的数据
def updata(id):
    db=sqlite3.connect(DATABASE)
    cur=db.cursor()
    cur.execute("SELECT * FROM adminx WHERE admin=?",(id,))
    info=cur.fetchall()
    cur.execute("UPDATE adminx SET day=? WHERE admin=?",(info[0][2]-1,id))
    db.commit()
    cur.execute("UPDATE adminx SET last=? WHERE admin=?",(curdate,id))
    db.commit()
    cur.close()
    db.close()
    return 0

#资源回收
def cleanup():
    if os.path.exists("state/RUNNING"):
        os.remove("state/RUNNING")

#读取运行情况
if os.path.exists("state/RUNNING"):
    os._exit(1)
#标记当前正在运行
with open("state/RUNNING","w",encoding="utf-8") as f:
    print("RUNNING",file=f)
#设置回调函数
atexit.register(cleanup)
#获取PATH与用户数据
DATABASE="data/data.db"
db=sqlite3.connect(DATABASE)
cur=db.cursor()
cur.execute("SELECT * FROM pathset WHERE True")
path=cur.fetchall()
path=str(path[0][0])
setpath=path+"/config/gui.json"
logpath=path+"/debug/gui.log"
maapath=path+"/MAA.exe"
cur.execute("SELECT * FROM adminx WHERE True")
data=cur.fetchall()
data=[list(row) for row in data]
cur.close()
db.close()
#开始执行
curdate=datetime.date.today().strftime('%Y-%m-%d')
begintime=datetime.datetime.now().strftime("%H:%M")
idnew=[]
idold=[]
idfail=[]
idall=[data[i][0] for i in range(len(data))]
LOGXLEN=60
logx=['' for i in range(LOGXLEN)]
logi=0
#MAA预配置
setmaa(0,0,0)
#优先代理今日未完成的用户
for i in range(len(data)):
    if data[i][3]=='y' and data[i][4]!=curdate and data[i][2]>0:
        book=runmaa(data[i][0],data[i][1],data[i][5])
        if book:
            updata(data[i][0])
            idnew.append(data[i][0])
        else:
            idfail.append(data[i][0])
#次优先重复代理
for i in range(len(data)):
    if data[i][3]=='y' and data[i][4]==curdate and data[i][2]>0:
        book=runmaa(data[i][0],data[i][1],data[i][5])
        if book:
            idold.append(data[i][0])
        else:
            idall.remove(data[i][0])
endtime=datetime.datetime.now().strftime("%H:%M")
with open("log.txt","w",encoding="utf-8") as f:
    print("任务开始时间："+begintime+"，结束时间："+endtime,file=f)
    print("已完成数："+str(len(idnew))+"，未完成数："+str(len(idfail))+"，重复执行数："+str(len(idold)),file=f)
    if len(idfail)!=0:
        print("代理未完成的用户：",file=f)
    for i in range(len(idfail)):
        print(idfail[i],file=f)
if os.path.exists("state/BEGIN"):
    with open("state/END","w",encoding="utf-8") as f:
        print("END",file=f)
sys.exit(0)