Program.cs

using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Logging;
using PartLibraryExtractor.Data;
using PartLibraryExtractor.Services;
using Npgsql;
using Pgvector.EntityFrameworkCore;

namespace PartLibraryExtractor
{
    class Program
    {
        static async Task Main(string[] args)
        {
            // 配置服务
            var services = ConfigureServices();

            using var serviceProvider = services.BuildServiceProvider();
            var logger = serviceProvider.GetRequiredService<ILogger<Program>>();

            try
            {
                logger.LogInformation("零部件库提取器启动");

                // 确保数据库连接正常
                using var scope = serviceProvider.CreateScope();
                var context = scope.ServiceProvider.GetRequiredService<PartLibraryContext>();
                
                try
                {
                    await context.Database.EnsureCreatedAsync();
                    logger.LogInformation("数据库连接正常");
                }
                catch (Exception ex)
                {
                    logger.LogError(ex, "数据库连接失败，请检查连接字符串和数据库状态");
                    return;
                }

                // 解析命令行参数
                if (args.Length == 0)
                {
                    ShowUsage();
                    return;
                }

                var command = args[0].ToLowerInvariant();
                var fileProcessor = scope.ServiceProvider.GetRequiredService<FileProcessor>();

                switch (command)
                {
                    case "process":
                        await ProcessFiles(args, fileProcessor, logger);
                        break;
                    case "convert":
                        await ConvertStepFile(args, scope.ServiceProvider, logger);
                        break;
                    case "search":
                        await SearchParts(args, fileProcessor, logger);
                        break;
                    case "stats":
                        await ShowStatistics(fileProcessor, logger);
                        break;
                    case "similar":
                        await FindSimilarParts(args, fileProcessor, logger);
                        break;
                    default:
                        ShowUsage();
                        break;
                }
            }
            catch (Exception ex)
            {
                logger.LogError(ex, "程序执行时发生错误");
            }
        }

        static IServiceCollection ConfigureServices()
        {
            var configuration = new ConfigurationBuilder()
                .SetBasePath(Directory.GetCurrentDirectory())
                .AddJsonFile("appsettings.json", optional: false)
                .Build();

            var services = new ServiceCollection();

            // 配置日志
            services.AddLogging(builder =>
            {
                builder.AddConsole();
                builder.AddConfiguration(configuration.GetSection("Logging"));
            });

            // 配置数据库
            services.AddDbContext<PartLibraryContext>(options =>
            {
                var connectionString = configuration.GetConnectionString("DefaultConnection");
                options.UseNpgsql(connectionString, npgsqlOptions =>
                {
                    npgsqlOptions.UseNetTopologySuite();
                    npgsqlOptions.UseVector();
                });
            });

            // 注册服务
            services.AddScoped<GeometryProcessor>();
            services.AddScoped<FileProcessor>();
            services.AddScoped<StepProcessor>();
            services.AddSingleton<IConfiguration>(configuration);

            return services;
        }

        static async Task ProcessFiles(string[] args, FileProcessor fileProcessor, ILogger logger)
        {
            if (args.Length < 2)
            {
                logger.LogError("请指定要处理的文件或目录路径");
                return;
            }

            var path = args[1];
            string? materialCode = null;
            decimal? thicknessMm = null;

            // 解析可选参数
            for (int i = 2; i < args.Length; i++)
            {
                if (args[i].StartsWith("--material="))
                {
                    materialCode = args[i].Substring("--material=".Length);
                }
                else if (args[i].StartsWith("--thickness="))
                {
                    if (decimal.TryParse(args[i].Substring("--thickness=".Length), out var thickness))
                    {
                        thicknessMm = thickness;
                    }
                }
            }

            logger.LogInformation("材料代码: {MaterialCode}", materialCode ?? "未指定");
            logger.LogInformation("厚度: {ThicknessMm}mm", thicknessMm?.ToString() ?? "未指定");

            // 检查是文件还是目录
            if (File.Exists(path))
            {
                // 处理单个文件
                logger.LogInformation("开始处理文件: {FilePath}", path);
                await fileProcessor.ProcessFileAsync(path, materialCode, thicknessMm);
            }
            else if (Directory.Exists(path))
            {
                // 处理目录
                logger.LogInformation("开始处理目录: {DirectoryPath}", path);
                await fileProcessor.ProcessDirectoryAsync(path, materialCode, thicknessMm);
            }
            else
            {
                logger.LogError("指定的文件或目录不存在: {Path}", path);
            }
        }

        static async Task SearchParts(string[] args, FileProcessor fileProcessor, ILogger logger)
        {
            var searchParams = new Dictionary<string, object>();

            for (int i = 1; i < args.Length; i++)
            {
                if (args[i].StartsWith("--material="))
                {
                    searchParams["materialCode"] = args[i].Substring("--material=".Length);
                }
                else if (args[i].StartsWith("--min-thickness="))
                {
                    if (decimal.TryParse(args[i].Substring("--min-thickness=".Length), out var thickness))
                    {
                        searchParams["minThickness"] = thickness;
                    }
                }
                else if (args[i].StartsWith("--max-thickness="))
                {
                    if (decimal.TryParse(args[i].Substring("--max-thickness=".Length), out var thickness))
                    {
                        searchParams["maxThickness"] = thickness;
                    }
                }
                else if (args[i].StartsWith("--min-area="))
                {
                    if (double.TryParse(args[i].Substring("--min-area=".Length), out var area))
                    {
                        searchParams["minArea"] = area;
                    }
                }
                else if (args[i].StartsWith("--max-area="))
                {
                    if (double.TryParse(args[i].Substring("--max-area=".Length), out var area))
                    {
                        searchParams["maxArea"] = area;
                    }
                }
            }

            var parts = await fileProcessor.SearchPartsAsync(
                materialCode: searchParams.GetValueOrDefault("materialCode") as string,
                minThickness: searchParams.GetValueOrDefault("minThickness") as decimal?,
                maxThickness: searchParams.GetValueOrDefault("maxThickness") as decimal?,
                minArea: searchParams.GetValueOrDefault("minArea") as double?,
                maxArea: searchParams.GetValueOrDefault("maxArea") as double?
            );

            logger.LogInformation("找到 {Count} 个零部件", parts.Count);

            foreach (var part in parts.Take(10))
            {
                logger.LogInformation("=== 零部件详情 ===");
                logger.LogInformation("ID: {PartId}", part.PartId);
                logger.LogInformation("文件名: {Filename}", part.Filename);
                logger.LogInformation("材料: {Material}", part.MaterialCode ?? "未指定");
                logger.LogInformation("厚度: {Thickness}mm", part.ThicknessMm?.ToString() ?? "未指定");
                logger.LogInformation("面积: {Area:F2}", part.FlatArea ?? 0);
                logger.LogInformation("长度: {Length:F2}", part.BboxL ?? 0);
                logger.LogInformation("宽度: {Width:F2}", part.BboxW ?? 0);
                logger.LogInformation("NFP哈希: {Hash}", part.NfpHash ?? "无");
                logger.LogInformation("上传批次: {Batch}", part.UploadBatch ?? "无");
                logger.LogInformation("创建时间: {CreatedAt}", part.CreatedAt);
                logger.LogInformation("几何数据: {Geometry}", part.ContourWkt?.ToString() ?? "无");
                logger.LogInformation("形状向量: {Vector}", part.ShapeVec != null ? "已生成" : "无");
                logger.LogInformation("");
            }
        }

        static async Task ShowStatistics(FileProcessor fileProcessor, ILogger logger)
        {
            var stats = await fileProcessor.GetStatisticsAsync();

            logger.LogInformation("=== 零部件库统计信息 ===");
            logger.LogInformation("总零部件数量: {TotalParts}", stats["TotalParts"]);
            logger.LogInformation("唯一材料种类: {UniqueMaterials}", stats["UniqueMaterials"]);
            logger.LogInformation("平均面积: {AverageArea:F2}", stats["AverageArea"]);
            logger.LogInformation("总面积: {TotalArea:F2}", stats["TotalArea"]);

            if (stats["FileTypes"] is Dictionary<string, int> fileTypes)
            {
                logger.LogInformation("文件类型分布:");
                foreach (var kvp in fileTypes)
                {
                    logger.LogInformation("  {Extension}: {Count}", kvp.Key, kvp.Value);
                }
            }
        }

        static async Task FindSimilarParts(string[] args, FileProcessor fileProcessor, ILogger logger)
        {
            if (args.Length < 2)
            {
                logger.LogError("请指定零部件ID");
                return;
            }

            if (!Guid.TryParse(args[1], out var partId))
            {
                logger.LogError("无效的零部件ID格式");
                return;
            }

            var limit = 5;
            if (args.Length > 2 && int.TryParse(args[2], out var parsedLimit))
            {
                limit = parsedLimit;
            }

            var similarParts = await fileProcessor.FindSimilarPartsAsync(partId, limit);

            logger.LogInformation("找到 {Count} 个相似零部件", similarParts.Count);

            foreach (var part in similarParts)
            {
                logger.LogInformation("ID: {PartId}, 文件名: {Filename}, 面积: {Area:F2}", 
                    part.PartId, part.Filename, part.FlatArea);
            }
        }

        static async Task ConvertStepFile(string[] args, IServiceProvider serviceProvider, ILogger logger)
        {
            if (args.Length < 3)
            {
                logger.LogError("请指定STEP文件路径和输出DXF文件路径");
                return;
            }

            var stepFilePath = args[1];
            var dxfOutputPath = args[2];

            if (!File.Exists(stepFilePath))
            {
                logger.LogError("STEP文件不存在: {StepFilePath}", stepFilePath);
                return;
            }

            var stepProcessor = serviceProvider.GetRequiredService<StepProcessor>();
            stepProcessor.ConvertStepToDxfWithBestMethod(stepFilePath, dxfOutputPath);
            
            logger.LogInformation("STEP文件转换完成: {DxfOutputPath}", dxfOutputPath);
        }

        static void ShowUsage()
        {
            Console.WriteLine("零部件库提取器使用说明:");
            Console.WriteLine();
            Console.WriteLine("命令:");
            Console.WriteLine("  convert <STEP文件路径> <DXF输出路径>");
            Console.WriteLine("    将STEP文件转换为DXF格式");
            Console.WriteLine();
            Console.WriteLine("  process <目录路径> [--material=材料代码] [--thickness=厚度]");
            Console.WriteLine("    处理指定目录中的DXF和SVG文件");
            Console.WriteLine();
            Console.WriteLine("  search [--material=材料代码] [--min-thickness=最小厚度] [--max-thickness=最大厚度]");
            Console.WriteLine("    [--min-area=最小面积] [--max-area=最大面积]");
            Console.WriteLine("    搜索零部件");
            Console.WriteLine();
            Console.WriteLine("  stats");
            Console.WriteLine("    显示统计信息");
            Console.WriteLine();
            Console.WriteLine("  similar <零部件ID> [数量限制]");
            Console.WriteLine("    查找相似零部件");
            Console.WriteLine();
            Console.WriteLine("示例:");
            Console.WriteLine("  PartLibraryExtractor convert ./initData/PCSWP2.step ./input/PCSWP2.dxf");
            Console.WriteLine("  PartLibraryExtractor process ./input --material=Q235 --thickness=2.0");
            Console.WriteLine("  PartLibraryExtractor search --material=Q235 --min-area=100");
            Console.WriteLine("  PartLibraryExtractor similar 12345678-1234-1234-1234-123456789012");
        }
    }
}