import os
import time
import configparser
from pynput import mouse
import win32clipboard
import win32con
from html2text import html2text
import keyboard
import threading
import sys
import logging

class TextCaptureApp:
    def __init__(self):
        self.load_config()
        self.running = True
        self.last_right_click_time = 0
        self.setup_logging()

    def load_config(self):
        config = configparser.ConfigParser()
        config.read('config.ini')
        self.double_click_threshold = config.getfloat('Settings', 'double_click_threshold', fallback=0.3)
        self.save_location = config.get('Settings', 'save_location', fallback=os.path.join(os.path.expanduser('~'), 'Desktop'))
        self.copy_delay = config.getfloat('Settings', 'copy_delay', fallback=0.1)

    def setup_logging(self):
        logging.basicConfig(
            filename='text_capture.log',
            level=logging.INFO,
            format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
        )
        self.logger = logging.getLogger(__name__)

    def save_to_markdown(self, content):
        try:
            timestamp = time.strftime("%Y%m%d_%H%M%S")
            file_path = os.path.join(self.save_location, f'captured_text_{timestamp}.md')

            with open(file_path, 'w', encoding='utf-8') as f:
                f.write(content)
            self.logger.info(f"Content saved to: {file_path}")
        except Exception as e:
            self.logger.error(f"Error saving file: {e}")

    def get_clipboard_content(self):
        try:
            win32clipboard.OpenClipboard()

            try:
                content = win32clipboard.GetClipboardData(win32con.CF_HTML)
                self.logger.info("Clipboard HTML content captured")
                return html2text(content)
            except:
                pass

            try:
                content = win32clipboard.GetClipboardData(win32con.CF_TEXT)
                self.logger.info("Clipboard TEXT content captured")
                return content.decode('gbk')
            except:
                pass

            try:
                content = win32clipboard.GetClipboardData(win32con.CF_UNICODETEXT)
                self.logger.info("Clipboard UNICODE TEXT content captured")
                return content
            except:
                return None

        except Exception as e:
            self.logger.error(f"Error getting clipboard content: {e}")
            return None
        finally:
            try:
                win32clipboard.CloseClipboard()
            except Exception as e:
                self.logger.error(f"Error closing clipboard: {e}")

    def simulate_copy(self):
        keyboard.press('ctrl')
        keyboard.press('c')
        keyboard.release('c')
        keyboard.release('ctrl')
        time.sleep(self.copy_delay)  # 等待复制操作完成

    def on_click(self, x, y, button, pressed):
        if not pressed:
            return

        if button == mouse.Button.right:
            current_time = time.time()
            if (current_time - self.last_right_click_time) < self.double_click_threshold:
                self.simulate_copy()
                content = self.get_clipboard_content()
                if content:
                    self.save_to_markdown(content)
                else:
                    self.logger.warning("No content captured from clipboard")
            self.last_right_click_time = current_time

    def on_keyboard_event(self, event):
        # 监听Esc键退出程序
        if event.name == 'esc':
            self.running = False
            return False

    def run(self):
        self.logger.info("Text Capture App started")
        print("文本捕获程序已启动...")
        print("使用说明：")
        print("1. 选中要保存的文本")
        print("2. 快速双击鼠标右键来保存选中的文本")
        print("3. 按ESC键退出程序")

        mouse_listener = mouse.Listener(on_click=self.on_click)
        mouse_listener.start()

        # 启动键盘监听
        keyboard.on_press(self.on_keyboard_event)

        # Wait until the running flag becomes False
        while self.running:
            time.sleep(0.1)

        print("\n程序已退出")
        mouse_listener.stop()
        mouse_listener.join()
        keyboard_listener.stop()
        keyboard_listener.join()
        self.logger.info("Text Capture App stopped")

if __name__ == "__main__":
    try:
        app = TextCaptureApp()
        app.run()
    except Exception as e:
        logging.error(f"程序发生错误: {e}")
        input("按Enter键退出...")