import pymysql
from config import Config

class Database:
    def __init__(self):
        self.conn = pymysql.connect(
            host=Config.MYSQL_CONFIG['host'],
            port=Config.MYSQL_CONFIG['port'],
            user=Config.MYSQL_CONFIG['user'],
            password=Config.MYSQL_CONFIG['password'],
            charset='utf8mb4',
            cursorclass=pymysql.cursors.DictCursor
        )
        self.init_db()
    
    def init_db(self):
        with self.conn.cursor() as cursor:
            # 创建数据库
            cursor.execute(f"CREATE DATABASE IF NOT EXISTS {Config.MYSQL_CONFIG['database']}")
            cursor.execute(f"USE {Config.MYSQL_CONFIG['database']}")
            
            # 创建文件表
            cursor.execute("""
                CREATE TABLE IF NOT EXISTS files (
                    id BIGINT PRIMARY KEY AUTO_INCREMENT,
                    original_name VARCHAR(255) NOT NULL,
                    stored_name VARCHAR(255) NOT NULL,
                    file_path VARCHAR(500) NOT NULL,
                    file_type VARCHAR(50),
                    file_size BIGINT,
                    created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
                    UNIQUE KEY idx_path (file_path)
                ) CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci
            """)
        self.conn.commit()
    
    def search_files(self, keyword):
        with self.conn.cursor() as cursor:
            cursor.execute("""
                SELECT * FROM files 
                WHERE original_name LIKE %s 
                OR file_path LIKE %s 
                ORDER BY created_at DESC
            """, (f"%{keyword}%", f"%{keyword}%"))
            return cursor.fetchall()