<?php

namespace core\plugin\traits;

use Illuminate\Database\Schema\Builder;
use support\Db;

/**
 * 迁移操作 Trait
 */
trait MigrationTrait
{
    /**
     * 插件临时目录路径
     */
    const RUNTIME_PLUGIN_PATH = 'install/plugin';

    /**
     * 运行迁移
     */
    public function runMigrations(): void
    {
        $migrationsDir = $this->findMigrationsDir();

        if (!$migrationsDir) {
            $this->output("⚠️ No migrations directory found");
            return;
        }

        $this->output("📁 Migrations dir: {$migrationsDir}");

        // 获取已执行的迁移
        $executed = $this->getExecutedMigrations();

        // 获取所有迁移文件
        $files = glob($migrationsDir . '/*.php');

        $pending = [];
        foreach ($files as $file) {
            $basename = basename($file, '.php');
            if (!in_array($basename, $executed)) {
                $pending[] = [$file, $basename];
            }
        }

        sort($pending);

        if (empty($pending)) {
            $this->output("✅ No pending migrations");
            return;
        }

        $this->output("📋 Pending migrations: " . count($pending));

        $batch = $this->getNextBatch();
        $schema = $this->getSchemaBuilder();

        // 直接执行迁移
        foreach ($pending as $migration) {
            $this->runMigration($migration[0], $migration[1], $schema);
        }

        // 记录批次
        $this->recordBatch($batch, $pending);

        $this->output("✅ Migrations completed (batch: {$batch})");
    }
    
    /**
     * 运行单个迁移
     */
    protected function runMigration(string $file, string $name, Builder $schema): void
    {
        $this->output("  📝 Migrating: {$name}");

        try {
            $migration = require $file;
            $migration->up($schema);
            $this->output("  ✅ Migrated: {$name}");
        } catch (\Throwable $e) {
            $this->output("  ❌ Error: {$e->getMessage()}");
            throw $e;
        }
    }
    
    /**
     * 回滚迁移
     * @param bool $dropTables 是否删除表
     */
    public function rollbackMigrations(bool $dropTables = false): void
    {
        if ($dropTables) {
            $this->dropAllTables();
            $this->saveMigrationRows([]);
            $this->output("✅ All plugin tables dropped");
            return;
        }

        // 不删除表，只清理迁移日志
        $this->saveMigrationRows([]);
        $this->output("📝 Migration records cleared (tables preserved)");
    }

    /**
     * 删除插件所有表
     */
    protected function dropAllTables(): void
    {
        $tables = $this->getPluginTables();
        if (empty($tables)) {
            $this->output("  ⚠️ No tables to drop");
            return;
        }

        $schema = $this->getSchemaBuilder();
        foreach ($tables as $table) {
            $this->output("  🗑️ Dropping table: {$table}");
            $schema->dropIfExists($table);
        }
    }

    /**
     * 获取插件相关的所有表
     */
    protected function getPluginTables(): array
    {
        $tables = [];
        $migrationsDir = $this->findMigrationsDir();

        // 从迁移文件解析表名
        if (is_dir($migrationsDir)) {
            $files = glob($migrationsDir . '/*.php');
            foreach ($files as $file) {
                $content = file_get_contents($file);
                if (preg_match_all('/->create\([\'"](\w+)[\'"]/', $content, $matches)) {
                    $tables = array_merge($tables, $matches[1]);
                }
            }
        }
        
        // 从数据库查询以插件名前缀的表
        $prefix = $this->pluginName . '_';
        try {
            $connection = $this->connection ?? null;
            $db = Db::connection($connection)->getDoctrineConnection();
            $tablesList = $db->getSchemaManager()->listTableNames();
            
            foreach ($tablesList as $tableName) {
                if (strpos($tableName, $prefix) === 0) {
                    $tables[] = $tableName;
                }
            }
        } catch (\Throwable $e) {
            // 忽略查询错误
        }

        return array_unique($tables);
    }
    
    /**
     * 回滚单个迁移
     */
    protected function rollbackMigration(string $name, Builder $schema): void
    {
        $this->output("  📝 Rolling back: {$name}");

        $migrationsDir = $this->findMigrationsDir();
        $file = $migrationsDir . '/' . $name . '.php';

        if (!file_exists($file)) {
            $this->output("  ⚠️ Migration file not found: {$file}");
            return;
        }

        try {
            $migration = require $file;
            $migration->down($schema);
            $this->output("  ✅ Rolled back: {$name}");
        } catch (\Throwable $e) {
            $this->output("  ❌ Error: {$e->getMessage()}");
        }
    }

    /**
     * 查找迁移目录
     */
    protected function findMigrationsDir(): ?string
    {
        $resourceDir = $this->getConfig('resource.migration', 'database/migrations');
        
        $dirs = [
            $this->pluginPath . '/resource/database/migrations',
            $this->pluginPath . '/' . $resourceDir,
            $this->pluginPath . '/install/migrations',
            base_path("resource/database/migrations/plugin/{$this->pluginName}"),
        ];
        
        foreach ($dirs as $dir) {
            if (is_dir($dir)) {
                return $dir;
            }
        }
        
        return null;
    }

    /**
     * 获取 Schema Builder
     */
    protected function getSchemaBuilder(): Builder
    {
        return Db::connection($this->connection ?? null)->getSchemaBuilder();
    }

    /**
     * 获取迁移日志文件
     */
    protected function getMigrationLogFile(): string
    {
        return runtime_path(self::RUNTIME_PLUGIN_PATH . '/' . $this->pluginName . '-migrations.log');
    }

    /**
     * 获取已执行的迁移
     */
    protected function getExecutedMigrations(): array
    {
        $logFile = $this->getMigrationLogFile();
        
        if (!file_exists($logFile)) {
            return [];
        }
        
        $content = trim(file_get_contents($logFile));
        
        if (empty($content)) {
            return [];
        }
        
        return array_filter(explode(PHP_EOL, $content));
    }

    /**
     * 获取迁移记录行
     */
    protected function getMigrationRows(): array
    {
        $logFile = $this->getMigrationLogFile();
        
        if (!file_exists($logFile)) {
            return [];
        }
        
        $content = trim(file_get_contents($logFile));
        
        if (empty($content)) {
            return [];
        }
        
        $rows = [];
        foreach (explode(PHP_EOL, $content) as $line) {
            $parts = explode(',', $line, 2);
            if (count($parts) === 2) {
                $rows[] = $parts;
            }
        }
        
        return $rows;
    }

    /**
     * 保存迁移记录行
     */
    protected function saveMigrationRows(array $rows): void
    {
        $logFile = $this->getMigrationLogFile();
        
        $logDir = dirname($logFile);
        if (!is_dir($logDir)) {
            mkdir($logDir, 0755, true);
        }
        
        if (empty($rows)) {
            @unlink($logFile);
            return;
        }
        
        $lines = array_map(fn($r) => "{$r[0]},{$r[1]}", $rows);
        file_put_contents($logFile, implode(PHP_EOL, $lines) . PHP_EOL, LOCK_EX);
    }

    /**
     * 获取下一批次号
     */
    protected function getNextBatch(): int
    {
        $rows = $this->getMigrationRows();
        
        if (empty($rows)) {
            return 1;
        }
        
        $batches = array_map(fn($row) => (int)$row[0], $rows);
        
        return max($batches) + 1;
    }

    /**
     * 记录批次
     */
    protected function recordBatch(int $batch, array $migrations): void
    {
        $logFile = $this->getMigrationLogFile();
        
        $logDir = dirname($logFile);
        if (!is_dir($logDir)) {
            mkdir($logDir, 0755, true);
        }
        
        $lines = array_map(fn($m) => "{$batch},{$m[1]}", $migrations);
        file_put_contents($logFile, PHP_EOL . implode(PHP_EOL, $lines), FILE_APPEND | LOCK_EX);
    }

    /**
     * 清理日志
     */
    protected function clearLogs(): void
    {
        @unlink($this->getMigrationLogFile());
        @unlink($this->getSeedLogFile());
    }

    /**
     * 导出 SQL（从迁移文件生成）
     * 
     * @param bool $withData 是否包含数据
     * @param string|null $outputFile 输出文件路径（null 则输出到插件目录）
     * @return bool 是否成功
     */
    public function exportSql(bool $withData = false, ?string $outputFile = null): bool
    {
        $migrationsDir = $this->findMigrationsDir();

        if (!$migrationsDir || !is_dir($migrationsDir)) {
            $this->output("⚠️ No migrations directory found");
            return false;
        }

        $files = glob($migrationsDir . '/*.php');

        if (empty($files)) {
            $this->output("⚠️ No migration files found");
            return false;
        }

        $this->output("📤 Exporting SQL from " . count($files) . " migration file(s)...");

        // 从迁移文件生成 SQL
        $sql = $this->generateSqlFromMigrations($files, $withData);

        // 默认输出到插件的 resource/database 目录
        if (!$outputFile) {
            $outputDir = $this->pluginPath . '/resource/database';
            if (!is_dir($outputDir)) {
                mkdir($outputDir, 0755, true);
            }
            $filename = $withData ? 'install.sql' : 'structure.sql';
            $outputFile = $outputDir . '/' . $filename;
        }

        $dir = dirname($outputFile);
        if (!is_dir($dir)) {
            mkdir($dir, 0755, true);
        }
        file_put_contents($outputFile, $sql);

        $this->output("✅ SQL exported to: {$outputFile}");

        return true;
    }
    
    /**
     * 从迁移文件生成 SQL
     */
    protected function generateSqlFromMigrations(array $files, bool $withData): string
    {
        $sql = "-- Plugin: {$this->pluginName}\n";
        $sql .= "-- Exported at: " . date('Y-m-d H:i:s') . "\n";
        $sql .= "-- Generated from migration files\n\n";
        
        $sql .= "SET FOREIGN_KEY_CHECKS = 0;\n\n";
        
        // 按文件名排序
        sort($files);
        
        foreach ($files as $file) {
            $basename = basename($file, '.php');
            $this->output("  📤 Processing: {$basename}");

            $sql .= $this->generateSqlFromMigrationFile($file, $withData);
            $sql .= "\n";
        }
        
        $sql .= "SET FOREIGN_KEY_CHECKS = 1;\n";
        
        return $sql;
    }
    
    /**
     * 从单个迁移文件生成 SQL
     */
    protected function generateSqlFromMigrationFile(string $file, bool $withData): string
    {
        $sql = "";
        $content = file_get_contents($file);
        
        // 提取所有表名
        preg_match_all('/->create\([\'"](\w+)[\'"]\s*,/', $content, $tableMatches);
        
        if (empty($tableMatches[1])) {
            return $sql;
        }
        
        foreach ($tableMatches[1] as $tableName) {
            $sql .= "-- ----------------------------\n";
            $sql .= "-- Table structure for {$tableName}\n";
            $sql .= "-- ----------------------------\n";
            $sql .= "DROP TABLE IF EXISTS `{$tableName}`;\n";
            
            // 从迁移代码提取 CREATE TABLE 部分
            // 匹配 ->create('table_name', function (Blueprint $table) { ... });
            $pattern = '/->create\([\'"]' . preg_quote($tableName, '/') . '[\'"]\s*,\s*function\s*\([^)]*\)\s*\{([\s\S]*?)\}\s*\)/';
            
            if (preg_match($pattern, $content, $matches)) {
                $body = $matches[1];
                $sql .= $this->generateCreateTableFromBlueprint($tableName, $body);
            }
            
            $sql .= ";\n\n";
            
            // 如果需要数据，尝试从数据库获取（如果表存在）
            if ($withData) {
                $sql .= $this->generateInsertDataSql($tableName);
                $sql .= "\n";
            }
        }
        
        return $sql;
    }
    
    /**
     * 从 Blueprint 代码生成 CREATE TABLE
     */
    protected function generateCreateTableFromBlueprint(string $tableName, string $body): string
    {
        $columns = [];
        $indexes = [];
        
        // 匹配 $table->列方法
        // $table->id();
        // $table->string('name', 100);
        // $table->text('content')->nullable();
        // $table->foreignId('user_id')->references('id')->on('users');
        
        // 匹配各种列定义
        $lines = explode("\n", $body);
        
        foreach ($lines as $line) {
            $line = trim($line);
            if (empty($line) || $line === '$table->') {
                continue;
            }
            
            // 移除 $table-> 前缀和结尾分号
            $line = ltrim($line, '$table->');
            $line = rtrim($line, ';');
            
            // 匹配方法调用
            if (preg_match('/^(\w+)\s*\(([^)]*)\)(.*)/', $line, $matches)) {
                $method = $matches[1];
                $args = $matches[2];
                $modifiers = $matches[3] ?? '';
                
                $colDef = $this->parseBlueprintMethod($method, $args, $modifiers);
                
                if ($colDef) {
                    $columns[] = $colDef;
                }
            }
        }
        
        if (empty($columns)) {
            return "CREATE TABLE `{$tableName}` ()";
        }
        
        return "CREATE TABLE `{$tableName}` (\n  " . implode(",\n  ", $columns) . "\n) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci";
    }
    
    /**
     * 解析 Blueprint 方法
     */
    protected function parseBlueprintMethod(string $method, string $args, string $modifiers): ?string
    {
        // 解析参数
        $argList = [];
        if ($args) {
            // 简单解析逗号分隔的参数
            $depth = 0;
            $current = '';
            for ($i = 0; $i < strlen($args); $i++) {
                $char = $args[$i];
                if ($char === '(') {
                    $depth++;
                    $current .= $char;
                } elseif ($char === ')') {
                    $depth--;
                    $current .= $char;
                } elseif ($char === ',' && $depth === 0) {
                    $argList[] = trim($current);
                    $current = '';
                } else {
                    $current .= $char;
                }
            }
            if ($current) {
                $argList[] = trim($current);
            }
        }
        
        // 处理修饰符
        $nullable = strpos($modifiers, '->nullable()') !== false;
        $default = null;
        if (preg_match('/->default\(([^)]+)\)/', $modifiers, $dm)) {
            $default = $dm[1];
        }
        $autoIncrement = strpos($modifiers, '->autoIncrement()') !== false 
                      || strpos($modifiers, '->autoIncrements()') !== false;
        $primary = strpos($modifiers, '->primary()') !== false;
        
        // 根据方法名生成列定义
        return match ($method) {
            'id' => $this->formatColumn('id', 'bigint', $nullable, $default, $autoIncrement, $primary),
            'bigIncrements' => $this->formatColumn(null, 'bigint', false, null, true, true),
            'bigInteger' => $this->formatColumn($argList[0] ?? null, 'bigint', $nullable, $default, $autoIncrement, $primary),
            'integer', 'increments', 'mediumIncrements', 'smallIncrements', 'tinyIncrements' => 
                $this->formatColumn($argList[0] ?? null, 'int', $nullable, $default, $autoIncrement, $primary),
            'bigInteger' => $this->formatColumn($argList[0] ?? null, 'bigint', $nullable, $default, $autoIncrement, $primary),
            'string' => $this->formatColumn($argList[0] ?? null, 'varchar(' . ($argList[1] ?? 255) . ')', $nullable, $default, false, false),
            'text' => $this->formatColumn($argList[0] ?? null, 'text', $nullable, $default, false, false),
            'longText' => $this->formatColumn($argList[0] ?? null, 'longtext', $nullable, $default, false, false),
            'mediumText' => $this->formatColumn($argList[0] ?? null, 'mediumtext', $nullable, $default, false, false),
            'json' => $this->formatColumn($argList[0] ?? null, 'json', $nullable, $default, false, false),
            'boolean' => $this->formatColumn($argList[0] ?? null, 'tinyint(1)', $nullable, $default, false, false),
            'date' => $this->formatColumn($argList[0] ?? null, 'date', $nullable, $default, false, false),
            'datetime' => $this->formatColumn($argList[0] ?? null, 'datetime', $nullable, $default, false, false),
            'timestamp' => $this->formatColumn($argList[0] ?? null, 'timestamp', $nullable, $default, false, false),
            'time' => $this->formatColumn($argList[0] ?? null, 'time', $nullable, $default, false, false),
            'float' => $this->formatColumn($argList[0] ?? null, 'float', $nullable, $default, false, false),
            'double' => $this->formatColumn($argList[0] ?? null, 'double', $nullable, $default, false, false),
            'decimal' => $this->formatColumn($argList[0] ?? null, 'decimal(' . ($argList[1] ?? '10,2') . ')', $nullable, $default, false, false),
            'enum' => $this->formatColumn($argList[0] ?? null, 'enum(' . ($argList[1] ?? '') . ')', $nullable, $default, false, false),
            'uuid' => $this->formatColumn($argList[0] ?? null, 'char(36)', $nullable, $default, false, false),
            'foreignId', 'foreignIdFor' => $this->formatColumn($argList[0] ?? null, 'bigint', $nullable, $default, false, false),
            'foreignUuid' => $this->formatColumn($argList[0] ?? null, 'char(36)', $nullable, $default, false, false),
            'rememberToken' => $this->formatColumn('remember_token', 'varchar(100)', true, null, false, false),
            'timestamps' => "`created_at` datetime NULL,\n  `updated_at` datetime NULL",
            'timestampable' => "`created_at` datetime NULL,\n  `updated_at` datetime NULL",
            'softDeletes' => "`deleted_at` datetime NULL",
            'softDeletesTz' => "`deleted_at` datetime NULL",
            default => null,
        };
    }
    
    /**
     * 格式化列定义
     */
    protected function formatColumn(?string $name, string $type, bool $nullable, ?string $default, bool $autoIncrement, bool $primary): ?string
    {
        if (!$name) {
            return null;
        }
        
        $col = "`{$name}` {$type}";
        
        if ($nullable) {
            $col .= " NULL";
        } else {
            $col .= " NOT NULL";
        }
        
        if ($autoIncrement) {
            $col .= " AUTO_INCREMENT";
        }
        
        if ($default !== null) {
            $col .= " DEFAULT " . $default;
        }
        
        if ($primary) {
            $col .= ", PRIMARY KEY (`{$name}`)";
        }
        
        return $col;
    }
    
    /**
     * 生成 INSERT DATA SQL（从数据库）
     */
    protected function generateInsertDataSql(string $table): string
    {
        $sql = "-- ----------------------------\n";
        $sql .= "-- Data for {$table}\n";
        $sql .= "-- ----------------------------\n";
        
        try {
            $connection = $this->connection ?? null;
            $pdo = Db::connection($connection)->getPdo();
            
            // 检查表是否存在
            $stmt = $pdo->query("SHOW TABLES LIKE '{$table}'");
            if ($stmt->rowCount() === 0) {
                $sql .= "-- Table does not exist, skipping data\n";
                return $sql;
            }
            
            $stmt = $pdo->query("SELECT * FROM `{$table}`");
            $rows = $stmt->fetchAll(\PDO::FETCH_ASSOC);
            
            if (empty($rows)) {
                $sql .= "-- No data\n";
                return $sql;
            }
            
            foreach ($rows as $row) {
                $values = array_map([$this, 'quoteValue'], array_values($row));
                $sql .= "INSERT INTO `{$table}` (`" . implode('`, `', array_keys($row)) . "`) VALUES (" . implode(', ', $values) . ");\n";
            }
            
        } catch (\Throwable $e) {
            $sql .= "-- Error: {$e->getMessage()}\n";
        }
        
        return $sql;
    }
    
    /**
     * 转义值
     */
    protected function quoteValue(mixed $value): string
    {
        if ($value === null) {
            return 'NULL';
        }
        
        if (is_bool($value)) {
            return $value ? '1' : '0';
        }
        
        if (is_int($value) || is_float($value)) {
            return (string)$value;
        }
        
        // 使用 PDO 的 quote
        $connection = $this->connection ?? null;
        $pdo = Db::connection($connection)->getPdo();
        return $pdo->quote($value);
    }
}