import '../core/models.dart';
import '../generators/documentation_generator.dart';
import '../generators/endpoint_code_generator.dart';
import '../generators/model_code_generator.dart';
import '../generators/retrofit_api_generator.dart';
import '../parsers/swagger_data_parser.dart';
import '../utils/file_utils.dart';
import 'base_command.dart';

/// Generate命令
/// 用于生成各种代码文件
class GenerateCommand extends BaseCommand {
  @override
  String get name => 'generate';

  @override
  String get description => '生成API代码文件（模型、端点、文档等）';

  @override
  String get usage => 'dart swagger_cli.dart generate [options]';

  @override
  List<CommandOption> get options => [
        const CommandOption(
          name: 'endpoints',
          shortName: 'e',
          description: '生成API端点常量',
          type: OptionType.flag,
        ),
        const CommandOption(
          name: 'models',
          shortName: 'm',
          description: '生成数据模型',
          type: OptionType.flag,
        ),
        const CommandOption(
          name: 'docs',
          shortName: 'd',
          description: '生成API文档',
          type: OptionType.flag,
        ),
        const CommandOption(
          name: 'api',
          shortName: 'r',
          description: '生成Retrofit风格API接口',
          type: OptionType.flag,
        ),
        const CommandOption(
          name: 'split-by-tags',
          shortName: 't',
          description: '按tags分组生成多个API文件',
          type: OptionType.flag,
        ),
        const CommandOption(
          name: 'all',
          shortName: 'a',
          description: '生成所有文件（默认）',
          type: OptionType.flag,
        ),
        const CommandOption(
          name: 'simple',
          shortName: 's',
          description: '使用简洁版模型生成',
          type: OptionType.flag,
        ),
        const CommandOption(
          name: 'output-dir',
          shortName: 'o',
          description: '输出目录',
          type: OptionType.string,
          defaultValue: 'generator',
        ),
      ];

  @override
  Future<int> execute(List<String> args) async {
    try {
      final parsedArgs = parseArguments(args);
      validateArguments(parsedArgs);

      await prepare(parsedArgs);

      // 解析Swagger文档
      progress('正在解析Swagger文档...');
      final parser = SwaggerDataParser();
      final document = await parser.fetchAndParseSwaggerDocument();

      // 解析生成选项
      final options = _parseGenerateOptions(parsedArgs);
      final outputDir =
          parsedArgs.getOption<String>('output-dir') ?? 'generator';
      final fullOutputDir = FileUtils.getProjectRootGeneratorDir();

      progress('输出目录: $fullOutputDir');

      // 确保输出目录存在
      await FileUtils.ensureDirectoryExists(fullOutputDir);

      int generatedFiles = 0;

      // 生成端点代码
      if (options.generateEndpoints) {
        progress('正在生成API端点常量...');
        final generator = EndpointCodeGenerator(document);
        final code = generator.generate();

        final filePath = '$fullOutputDir/api_paths.dart';
        await FileUtils.writeFile(filePath, code);
        success('API端点常量已保存到: $filePath');
        generatedFiles++;
      }

      // 生成模型代码
      if (options.generateModels) {
        progress('正在生成数据模型...');
        final generator = ModelCodeGenerator(
          document,
          useSimpleModels: options.useSimpleModels,
        );

        final modelsDir = '$fullOutputDir/api_models';
        await FileUtils.ensureDirectoryExists(modelsDir);

        final modelFiles = generator.generateSeparateModelFiles();

        for (final entry in modelFiles.entries) {
          final filePath = '$modelsDir/${entry.key}';
          await FileUtils.writeFile(filePath, entry.value);
          success('模型文件已保存到: $filePath');
          generatedFiles++;
        }
      }

      // 生成 Retrofit 风格的 API 接口
      if (options.generateApi) {
        if (options.splitByTags) {
          progress('正在按tags分组生成Retrofit风格API接口...');
          final generator = RetrofitApiGenerator(
            document,
            className: 'ApiClient',
            useRetrofit: true,
            useDio: true,
            splitByTags: true,
          );

          // 确保参数实体类已生成
          generator.ensureParameterEntitiesGenerated();

          // 生成按 tags 分组的多个 API 文件
          final tagApiFiles = generator.generateApiFilesByTags();

          final apiDir = '$fullOutputDir/api';
          await FileUtils.ensureDirectoryExists(apiDir);

          for (final entry in tagApiFiles.entries) {
            final fileName = entry.key;
            final code = entry.value;
            final filePath = '$apiDir/$fileName';
            await FileUtils.writeFile(filePath, code);
            success('API接口文件已保存到: $filePath');
            generatedFiles++;
          }

          // 生成主 API 文件
          final mainCode = generator.generateMainApiFile();
          final mainFilePath = '$apiDir/api_client.dart';
          await FileUtils.writeFile(mainFilePath, mainCode);
          success('主API接口文件已保存到: $mainFilePath');
          generatedFiles++;

          // 生成参数实体类文件
          final parameterEntityFiles = generator.generateParameterEntityFiles();
          if (parameterEntityFiles.isNotEmpty) {
            final modelsDir = '$fullOutputDir/api_models';
            await FileUtils.ensureDirectoryExists(modelsDir);

            for (final entry in parameterEntityFiles.entries) {
              final filePath = '$modelsDir/${entry.key}';
              await FileUtils.writeFile(filePath, entry.value);
              success('参数实体类文件已保存到: $filePath');
              generatedFiles++;
            }
          }
        } else {
          progress('正在生成Retrofit风格API接口...');
          final generator = RetrofitApiGenerator(
            document,
            className: 'ApiClient',
            useRetrofit: true,
            useDio: true,
            splitByTags: false,
          );

          // 确保参数实体类已生成
          generator.ensureParameterEntitiesGenerated();

          final code = generator.generate();

          final apiDir = '$fullOutputDir/api';
          await FileUtils.ensureDirectoryExists(apiDir);

          final filePath = '$apiDir/api.dart';
          await FileUtils.writeFile(filePath, code);
          success('Retrofit API接口已保存到: $filePath');
          generatedFiles++;

          // 生成参数实体类文件
          final parameterEntityFiles = generator.generateParameterEntityFiles();
          if (parameterEntityFiles.isNotEmpty) {
            final modelsDir = '$fullOutputDir/api_models';
            await FileUtils.ensureDirectoryExists(modelsDir);

            for (final entry in parameterEntityFiles.entries) {
              final filePath = '$modelsDir/${entry.key}';
              await FileUtils.writeFile(filePath, entry.value);
              success('参数实体类文件已保存到: $filePath');
              generatedFiles++;
            }
          }
        }
      }

      // 生成文档
      if (options.generateDocs) {
        progress('正在生成API文档...');
        final generator = DocumentationGenerator(document);
        final code = generator.generate();

        final filePath = '$fullOutputDir/api_documentation.md';
        await FileUtils.writeFile(filePath, code);
        success('API文档已保存到: $filePath');
        generatedFiles++;
      }

      // 生成摘要
      _generateSummary(document, fullOutputDir);

      success('代码生成完成！共生成 $generatedFiles 个文件');
      return 0;
    } catch (e) {
      print('❌ 生成失败: $e');
      return 1;
    }
  }

  /// 解析生成选项
  GenerateOptions _parseGenerateOptions(ParsedArguments args) {
    final hasAnyFlag = args.hasOption('endpoints') ||
        args.hasOption('models') ||
        args.hasOption('docs') ||
        args.hasOption('api');

    return GenerateOptions(
      generateEndpoints: hasAnyFlag
          ? (args.getOption<bool>('endpoints') ?? false)
          : (args.getOption<bool>('all') ?? true),
      generateModels: hasAnyFlag
          ? (args.getOption<bool>('models') ?? false)
          : (args.getOption<bool>('all') ?? true),
      generateDocs: hasAnyFlag
          ? (args.getOption<bool>('docs') ?? false)
          : (args.getOption<bool>('all') ?? true),
      generateApi: hasAnyFlag
          ? (args.getOption<bool>('api') ?? false)
          : (args.getOption<bool>('all') ?? true),
      useSimpleModels: args.getOption<bool>('simple') ?? false,
      splitByTags: args.getOption<bool>('split-by-tags') ?? false,
    );
  }

  /// 生成摘要信息
  void _generateSummary(SwaggerDocument document, String outputDir) {
    final summary = StringBuffer();
    summary.writeln('# 代码生成摘要');
    summary.writeln('');
    summary.writeln('**API标题**: ${document.title}');
    summary.writeln('**API版本**: ${document.version}');
    summary.writeln('**生成时间**: ${DateTime.now().toIso8601String()}');
    summary.writeln('');
    summary.writeln('## 统计信息');
    summary.writeln('- 控制器数量: ${document.controllers.length}');
    summary.writeln('- API路径数量: ${document.paths.length}');
    summary.writeln('- 数据模型数量: ${document.models.length}');
    summary.writeln('');
    summary.writeln('## 控制器列表');
    document.controllers.forEach((name, controller) {
      summary.writeln(
          '- **$name**: ${controller.description} (${controller.paths.length} 个路径)');
    });

    FileUtils.writeFile('$outputDir/SUMMARY.md', summary.toString());
  }
}

/// 生成选项
class GenerateOptions {
  final bool generateEndpoints;
  final bool generateModels;
  final bool generateDocs;
  final bool generateApi;
  final bool useSimpleModels;
  final bool splitByTags;

  const GenerateOptions({
    required this.generateEndpoints,
    required this.generateModels,
    required this.generateDocs,
    required this.generateApi,
    required this.useSimpleModels,
    required this.splitByTags,
  });
}