# wrapper_embed.py – 由 PyInstaller 打包，运行时提取并解密原始 EXE
import os, sys, json, hashlib, tempfile, subprocess, struct
from datetime import datetime
import mysql.connector
import tkinter as tk
from tkinter import messagebox
import platform

MAGIC = b'ENC_MAGIC'
HEAD = 512
KEY = b'EXEWrapper#2024'


def xor(data):
    return bytes(d ^ KEY[i % len(KEY)] for i, d in enumerate(data))


def get_machine_code():
    import uuid
    return hashlib.md5(str(uuid.getnode()).encode()).hexdigest()[:16].upper()


def validate(key, machine):
    try:
        conn = mysql.connector.connect(
            host='taiyiagi.xyz', user='taiyi', password='taiyi1224',
            database='filesend', connection_timeout=10, autocommit=True
        )
        cur = conn.cursor(dictionary=True)
        cur.execute('SELECT * FROM license_keys WHERE key_code=%s', (key,))
        print("验证激活码")
        rec = cur.fetchone()
        if not rec: return False, '激活码不存在'
        if rec['status'] == 'banned': return False, '激活码被封禁'
        if rec['end_time'] < datetime.now(): return False, '激活码已过期'
        if rec['status'] == 'active':
            return (True, '验证成功') if rec['machine_code'] == machine else (False, '激活码已绑定其他设备')
        if rec['status'] == 'unused':
            cur.execute('UPDATE license_keys SET status=%s, machine_code=%s, start_time=NOW() WHERE key_code=%s',
                        ('active', machine, key))
            return cur.rowcount > 0, '激活成功' if cur.rowcount else '激活失败'
        return False, '激活码状态异常'
    except Exception as e:
        return False, str(e)


def get_bundled_data_path():
    """获取打包数据文件路径"""
    if getattr(sys, 'frozen', False):
        # 如果是打包后的exe，数据文件在临时目录中
        bundle_dir = sys._MEIPASS
    else:
        # 如果是脚本运行，数据文件在当前目录
        bundle_dir = os.path.dirname(os.path.abspath(__file__))

    # 查找 .dat 文件
    for file in os.listdir(bundle_dir):
        if file.endswith('.dat'):
            return os.path.join(bundle_dir, file)
    return None


def extract_exe():
    """从打包的数据文件中提取并解密原始EXE"""
    try:
        # 首先尝试从打包的数据文件中读取
        data_file = get_bundled_data_path()
        if data_file and os.path.exists(data_file):
            with open(data_file, 'rb') as f:
                data = f.read()
        else:
            # 如果没有找到数据文件，尝试从当前exe文件末尾读取
            if getattr(sys, 'frozen', False):
                exe = sys.executable
                with open(exe, 'rb') as f:
                    data = f.read()
            else:
                print("未找到加密数据文件")
                return None

        # 查找魔数位置
        pos = data.find(MAGIC)
        if pos == -1:
            print("未找到加密数据标识")
            return None

        pos += len(MAGIC)

        # 读取头部信息
        if pos + HEAD > len(data):
            print("数据文件损坏：头部信息不完整")
            return None

        header_data = data[pos:pos + HEAD].rstrip(b'\0')
        try:
            hdr = json.loads(header_data.decode())
        except json.JSONDecodeError as e:
            print(f"头部信息解析失败: {e}")
            return None

        pos += HEAD

        # 解密数据
        if pos >= len(data):
            print("数据文件损坏：没有加密数据")
            return None

        encrypted_data = data[pos:]
        if len(encrypted_data) == 0:
            print("加密数据为空")
            return None

        dec = xor(encrypted_data)

        # 验证文件完整性
        if hashlib.sha256(dec).hexdigest() != hdr['hash']:
            print("文件完整性校验失败")
            return None

        if len(dec) != hdr['size']:
            print(f"文件大小不匹配，期望: {hdr['size']}, 实际: {len(dec)}")
            return None

        # 写入临时文件
        temp_dir = tempfile.gettempdir()
        filename = hdr['filename']
        tmp = os.path.join(temp_dir, f"temp_{os.getpid()}_{filename}")

        try:
            with open(tmp, 'wb') as f:
                f.write(dec)

            # 设置可执行权限
            if hasattr(os, 'chmod'):
                os.chmod(tmp, 0o755)

            print(f"成功提取文件到: {tmp}")
            return tmp

        except Exception as e:
            print(f"写入临时文件失败: {e}")
            return None

    except Exception as e:
        print(f"提取EXE文件时发生错误: {e}")
        return None


def run(path):
    """运行解密后的程序"""
    try:
        if not os.path.exists(path):
            print(f"文件不存在: {path}")
            return False

        print(f"正在启动程序: {path}")

        if os.name == 'nt':  # Windows
            # 使用 CREATE_NEW_CONSOLE 标志在新窗口中启动
            import subprocess
            subprocess.Popen([path],
                             creationflags=subprocess.CREATE_NEW_CONSOLE,
                             close_fds=False)
        else:  # Linux/macOS
            subprocess.Popen([path], close_fds=True)

        # 给程序一些启动时间
        import time
        time.sleep(1)

        return True

    except Exception as e:
        print(f"启动程序失败: {e}")
        return False


def gui():
    def ok():
        k = e.get().strip()
        if not k:
            return messagebox.showerror('错误', '请输入激活码')

        try:
            m = get_machine_code()
            success, msg = validate(k, m)

            if success:
                messagebox.showinfo('成功', '激活成功！正在启动程序...')
                root.destroy()

                # 提取并运行程序
                exe = extract_exe()
                if exe:
                    if run(exe):
                        print("程序启动成功")
                        # 延迟退出，给目标程序启动时间
                        import time
                        time.sleep(2)
                        sys.exit(0)
                    else:
                        messagebox.showerror('错误', '程序启动失败')
                else:
                    messagebox.showerror('错误', '无法提取程序文件，请检查文件完整性')
            else:
                messagebox.showerror('失败', msg)

        except Exception as e:
            messagebox.showerror('错误', f'验证过程出错: {str(e)}')

    root = tk.Tk()
    root.title('软件激活')
    root.geometry('400x200')
    root.resizable(False, False)

    # 居中显示窗口
    root.eval('tk::PlaceWindow . center')

    # 界面元素
    tk.Label(root, text='请输入激活码', font=('', 13)).pack(pady=15)
    e = tk.Entry(root, width=25)
    e.pack()
    e.focus()

    tk.Button(root, text='验证并启动', command=ok).pack(pady=15)

    # 显示机器码（用于调试）
    machine_code = get_machine_code()
    tk.Label(root, text=f'机器码: {machine_code}', font=('', 8), fg='gray').pack(pady=(10, 0))

    root.bind('<Return>', lambda x: ok())
    root.protocol('WM_DELETE_WINDOW', sys.exit)
    root.mainloop()


if __name__ == '__main__':
    # 添加调试信息
    print("程序启动...")
    print(f"Python版本: {sys.version}")
    print(f"当前工作目录: {os.getcwd()}")
    print(f"是否为打包环境: {getattr(sys, 'frozen', False)}")

    if getattr(sys, 'frozen', False):
        print(f"打包临时目录: {sys._MEIPASS}")
        print(f"打包临时目录内容: {os.listdir(sys._MEIPASS)}")

    gui()