import tkinter as tk
import time
import os
import logging
import datetime
import comtypes.client
from pycaw.pycaw import AudioUtilities, IAudioEndpointVolume
import psutil
import threading



# 全局標(biāo)志位，用于控制 monitor_and_terminate 線程的終止
stop_monitor_thread = threading.Event()


# 倒計(jì)時(shí)完成驗(yàn)證窗口
class Yanzheng:  # 休息窗口
    def __init__(self, shi, fen, miao):
        # 調(diào)用函數(shù)，將 config.txt 作為參數(shù)傳入
        self.modify_second_line('config.txt')
        toggle_mute()
        # 創(chuàng)建主窗口
        self.root = tk.Tk()
        self.root.title("Countdown Timer")
        # 使窗口最大化
        self.root.state('zoomed')
        # 設(shè)置窗口背景為黑色
        self.root.configure(bg='black')
        self.root.protocol("WM_DELETE_WINDOW", self.on_close)  # 為關(guān)閉窗口協(xié)議指定處理函數(shù) on_close
        # 隱藏最小化、最大化和關(guān)閉按鈕
        self.root.overrideredirect(True)
        # 讓窗口可以拖動(dòng)
        self.root.bind('<B1-Motion>', self.move_window)
        self.root.attributes('-topmost', True)
        # 初始倒計(jì)時(shí)時(shí)間為 2 小時(shí)（秒）
        self.shi = shi * 60 * 60
        self.fen = fen * 60
        self.miao = miao
        self.remaining_time = shi * 60 * 60 + fen * 60 + miao
        self.password_attempts = 5  # 記錄密碼輸入嘗試次數(shù)，初始為 5 次
        # 創(chuàng)建一個(gè)標(biāo)簽用于顯示倒計(jì)時(shí)
        self.label = tk.Label(self.root, text="休息中...", bg='black', fg='white', font=("Arial", 34))
        self.label.pack(expand=True)
        # 創(chuàng)建一個(gè)標(biāo)簽用于顯示剩余嘗試次數(shù)
        self.attempts_label = tk.Label(self.root, text=f"輸入管理員密碼解除限制，剩余嘗試次數(shù): {self.password_attempts}", bg='black', fg='white', font=("Arial", 18))
        self.attempts_label.pack()
        self.update_countdown()
        # 創(chuàng)建密碼輸入框
        self.password_entry = tk.Entry(self.root, show="*")
        self.password_entry.pack()
        self.password_entry.bind("<Return>", self.check_password)
        # 進(jìn)入主事件循環(huán)
        self.root.mainloop()

    def modify_second_line(self, filename):
        try:
            with open(filename, 'r') as file:
                lines = file.readlines()
            if len(lines) >= 2:
                lines[1] = '000001\n'
            with open(filename, 'w') as file:
                file.writelines(lines)
        except FileNotFoundError:
            print(f"文件 {filename} 未找到")
        except Exception as e:
            print(f"發(fā)生錯(cuò)誤: {e}")

    def on_close(self):
        return  # 此函數(shù)直接返回，不執(zhí)行任何關(guān)閉操作，從而阻止窗口關(guān)閉

    def update_countdown(self):
        if self.remaining_time > 0:
            # 將剩余時(shí)間轉(zhuǎn)換為 hh:mm:ss 格式
            countdown_text = time.strftime('%H:%M:%S', time.gmtime(self.remaining_time))
            self.label.config(text=countdown_text)
            self.remaining_time -= 1
            self.root.after(1000, self.update_countdown)  # 每秒更新一次
        else:
            self.shutdown()

    def move_window(self, event):
        self.root.geometry(f'+{event.x_root}+{event.y_root}')

    def check_password(self, event):
        password = self.password_entry.get()
        correct_password = mima  # 在此處設(shè)置正確的密碼
        if password == correct_password:
            toggle_mute()
            self.root.destroy()  # 關(guān)閉窗口，結(jié)束程序
            stop_monitor_thread.set()  # 設(shè)置標(biāo)志位，通知 monitor_and_terminate 線程終止
            thread1.join()  # 等待 monitor_and_terminate 線程終止
        else:
            self.password_attempts -= 1
            self.attempts_label.config(text=f"剩余嘗試次數(shù): {self.password_attempts}")  # 更新剩余嘗試次數(shù)的顯示
            if self.password_attempts <= 0:
                self.shutdown()
            else:
                self.password_entry.delete(0, tk.END)  # 清空密碼輸入框

    def shutdown(self):
        toggle_mute()
        os.system("shutdown /s /t 0")


class Daojishi:  # 倒計(jì)時(shí)窗口
    def __init__(self, shi, fen, miao):
        # 創(chuàng)建主窗口
        self.root = tk.Tk()
        self.root.title("倒計(jì)時(shí)")
        self.root.geometry("240x100")  # 設(shè)置窗口大小為 240x100
        # 設(shè)置窗口背景為黑色
        self.root.configure(bg='black')
        self.root.protocol("WM_DELETE_WINDOW", self.on_close)  # 為關(guān)閉窗口協(xié)議指定處理函數(shù) on_close
        # 隱藏最小化、最大化和關(guān)閉按鈕
        self.root.overrideredirect(True)
        # 讓窗口可以拖動(dòng)
        self.root.bind('<B1-Motion>', self.move_window)
        # 獲取屏幕的寬度和高度
        screen_width = self.root.winfo_screenwidth()
        screen_height = self.root.winfo_screenheight()
        # 計(jì)算窗口在屏幕中心的位置
        x = (screen_width - 240) // 2
        y = (screen_height - 100) // 2
        # 設(shè)置窗口的位置
        self.root.geometry(f"+{x}+{y}")
        self.root.attributes('-topmost', True)
        # 初始倒計(jì)時(shí)時(shí)間為 2 小時(shí)（秒）
        self.shi = shi * 60 * 60
        self.fen = fen * 60
        self.miao = miao
        self.remaining_time = shi * 60 * 60 + fen * 60 + miao
        # 創(chuàng)建一個(gè)標(biāo)簽用于顯示倒計(jì)時(shí)
        self.label = tk.Label(self.root, text="", bg='black', fg='white', font=('Helvetica', 28))
        self.label.pack(expand=True)
        # 開(kāi)始倒計(jì)時(shí)
        self.start_time = time.time()  # 記錄開(kāi)始時(shí)間
        self.update_countdown()
        # 進(jìn)入主事件循環(huán)
        self.root.mainloop()

    def on_close(self):
        self.root.overrideredirect(False)
        self.root.iconify()
        self.root.overrideredirect(True)
        # 此函數(shù)直接返回，不執(zhí)行任何關(guān)閉操作，從而阻止窗口關(guān)閉
    def update_countdown(self):
        if self.remaining_time > 0:
            # 將剩余時(shí)間轉(zhuǎn)換為 hh:mm:ss 格式
            countdown_text = time.strftime('%H:%M:%S', time.gmtime(self.remaining_time))
            self.label.config(text=countdown_text)
            self.remaining_time -= 1
            elapsed_time = time.time() - self.start_time  # 計(jì)算經(jīng)過(guò)的時(shí)間
            if elapsed_time >= 5:  # 每過(guò) 5 秒
                self.write_remaining_time_to_file()
                self.start_time = time.time()  # 重置開(kāi)始時(shí)間
            self.root.after(1000, self.update_countdown)  # 每秒更新一次
        else:
            self.root.destroy()
            d = Yanzheng(0, 0, yanzhengshijian)

    def move_window(self, event):
        self.root.geometry(f'+{event.x_root}+{event.y_root}')

    def write_remaining_time_to_file(self):
        try:
            with open('config.txt', 'r+') as config_file:
                lines = config_file.readlines()
                # 將剩余時(shí)間轉(zhuǎn)換為 hhmmss 格式
                remaining_time_hhmmss = time.strftime('%H%M%S', time.gmtime(self.remaining_time))
                lines[1] = remaining_time_hhmmss + '\n'
                config_file.seek(0)
                config_file.writelines(lines)
        except FileNotFoundError:
            logging.error("config.txt 文件未找到，請(qǐng)檢查文件路徑。")
            print("config.txt 文件未找到，請(qǐng)檢查文件路徑。")


# 文件讀取部分
try:
    with open('config.txt', 'r',encoding='utf-8') as config_file:  # 以只讀模式打開(kāi) config.txt 文件
        morenshijian = config_file.readline().strip()  # 讀取文件的第一行，默認(rèn)時(shí)間，并去除換行符
        shengyushijian = config_file.readline().strip()  # 讀取文件的第二行，剩余時(shí)間，并去除換行符
        shi = int(shengyushijian[0:2])
        fen = int(shengyushijian[2:4])
        miao = int(shengyushijian[4:6])
        riqi = config_file.readline().strip()  # 讀取文件的第三行，日期，并去除換行符
        yue = int(riqi[0:2])
        ri = int(riqi[2:4])
        yanzhengshijian = int(config_file.readline().strip())  # 讀取文件的第四行，驗(yàn)證時(shí)間，并去除換行符
        mima = config_file.readline().strip()  # 讀取文件的第五行，密碼，并去除換行符
except FileNotFoundError:
    logging.error("config.txt 文件未找到，請(qǐng)檢查文件路徑。")
    print("config.txt 文件未找到，請(qǐng)檢查文件路徑。")


def xiugai():
    try:
        with open('config.txt', 'r') as config_file:  # 以只讀模式打開(kāi) config.txt 文件
            # 跳過(guò)第一行，如果不需要使用第一行的信息
            config_file.readline()
            gengxinshengyushijian = config_file.readline().strip()  # 讀取文件的第二行，剩余時(shí)間，并去除換行符
            global gengxinshi, gengxinfen, gengxinmiao
            gengxinshi = int(gengxinshengyushijian[0:2])
            gengxinfen = int(gengxinshengyushijian[2:4])
            gengxinmiao = int(gengxinshengyushijian[4:6])
    except FileNotFoundError:
        print("config.txt 文件未找到")
        return


# 靜音部分
def toggle_mute():
    # 獲取系統(tǒng)默認(rèn)音頻設(shè)備
    devices = AudioUtilities.GetSpeakers()
    interface = devices.Activate(IAudioEndpointVolume._iid_, comtypes.CLSCTX_ALL, None)
    volume = comtypes.cast(interface, comtypes.POINTER(IAudioEndpointVolume))
    # 檢查當(dāng)前是否靜音
    is_muted = volume.GetMute()
    if is_muted:
        # 如果當(dāng)前是靜音，取消靜音
        volume.SetMute(0, None)
    else:
        # 如果當(dāng)前不是靜音，設(shè)置為靜音
        volume.SetMute(1, None)


# 任務(wù)管理器監(jiān)測(cè)部分
def monitor_and_terminate():
    target_processes = ["Taskmgr.exe", "perfmon.exe"]
    while not stop_monitor_thread.is_set():  # 檢查標(biāo)志位是否被設(shè)置
        # 遍歷系統(tǒng)中所有正在運(yùn)行的進(jìn)程
        for proc in psutil.process_iter():
            try:
                # 獲取進(jìn)程名稱
                proc_name = proc.name()
                # 如果進(jìn)程名稱在目標(biāo)進(jìn)程列表中
                if proc_name in target_processes:
                    # 終止進(jìn)程
                    proc.terminate()
                    print(f"Terminated process: {proc_name}")
            except (psutil.NoSuchProcess, psutil.AccessDenied, psutil.ZombieProcess):
                # 處理可能出現(xiàn)的異常
                pass
        # 休眠 1 秒，避免過(guò)度占用 CPU 資源
        time.sleep(1)


# 主程序
def main():
    print("主線程啟動(dòng)")
    now = datetime.datetime.now()
    # 獲取當(dāng)前月份
    month = now.month
    # 獲取當(dāng)前日期
    day = now.day
    huoqvriqi = str(month).zfill(2) + str(day).zfill(2)
    if month == yue and day == ri:
        print("未重置數(shù)據(jù)")  # 不更新日期
        q = Daojishi(shi, fen, miao)
    else:
        # 更新日期，更新剩余時(shí)間
        # 調(diào)用函數(shù)，將 'config.txt' 文件中的第二行替換為變量morenshijian
        # 讀取文件內(nèi)容
        with open('config.txt', 'r') as file:
            lines = file.readlines()
        # 檢查文件是否有至少三行
        if len(lines) >= 3:
            # 修改第二行和第三行
            lines[1] = morenshijian + '\n'  # 替換第二行內(nèi)容，添加換行符保持格式
            lines[2] = huoqvriqi + '\n'  # 替換第三行內(nèi)容，添加換行符保持格式
        # 將修改后的內(nèi)容寫回文件
        with open('config.txt', 'w') as file:
            file.writelines(lines)
        print("已重置數(shù)據(jù)")
        xiugai()
        q = Daojishi(gengxinshi, gengxinfen, gengxinmiao)


if __name__ == '__main__':
    # 創(chuàng)建線程
    thread1 = threading.Thread(target=monitor_and_terminate)
    thread3 = threading.Thread(target=main)

    # 啟動(dòng)線程
    thread1.start()
    thread3.start()

    # 等待所有線程執(zhí)行完畢

    thread1.join()
    thread3.join()