using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Configuration;
using Polly;
using Polly.Extensions.Http;
using Prism.Ioc;
using Prism.Modularity;
using System.Net.Http;
using System.Text;
using YY.Admin.Core.Services;
using YY.Admin.Core.Session;
using YY.Admin.EventBus;
using YY.Admin.Infrastructure.Hubs;
using YY.Admin.Infrastructure.Network;
using YY.Admin.Infrastructure.Storage;
using YY.Admin.Infrastructure.Sync;
using YY.Admin.Services.Service;
using YY.Admin.Services.Service.Category;
using YY.Admin.Services.Service.Customer;
using YY.Admin.Services.Service.Dict;
using YY.Admin.Services.Service.MixerMaterialTareStrategy;
using YY.Admin.Services.Service.MixerMaterial;
using YY.Admin.Services.Service.Supplier;
using YY.Admin.Services.Service.RawMaterialCard;
using YY.Admin.Services.Service.RawMaterialEntry;
using YY.Admin.Services.Service.Vehicle;
using YY.Admin.Services.Service.Warehouse;
using YY.Admin.Services.Service.WarehouseArea;
using YY.Admin.Services.Service.WeightRecord;
using YY.Admin.Services.Service.Print;

namespace YY.Admin.Module;

public class SyncModule : IModule
{
    public void RegisterTypes(IContainerRegistry containerRegistry)
    {
        containerRegistry.RegisterSingleton<TokenStore>();
        containerRegistry.RegisterSingleton<HttpSyncClient>();
        containerRegistry.RegisterSingleton<OutboxProcessor>();
        containerRegistry.RegisterSingleton<IJeecgUserMirrorPullOutbox, JeecgUserMirrorPullOutbox>();
        containerRegistry.RegisterSingleton<INetworkMonitor, NetworkMonitor>();
        containerRegistry.RegisterSingleton<ISignalRService, StompWebSocketService>();
        containerRegistry.RegisterSingleton<IUserSyncOutbox, UserSyncOutbox>();
        // 本地用户事件订阅器：将增删改操作入 Outbox 回传后端
        containerRegistry.RegisterSingleton<SysUserEventSubscriber>();

        // 车辆管理：免密 API 直连 + STOMP 实时通知
        containerRegistry.RegisterSingleton<IVehicleService, VehicleService>();
        containerRegistry.RegisterSingleton<VehicleSyncCoordinator>();
        // 客户管理：免密 API 直连 + STOMP 实时通知
        containerRegistry.RegisterSingleton<ICustomerService, CustomerService>();
        containerRegistry.RegisterSingleton<CustomerSyncCoordinator>();
        // 供应商管理：免密 API 直连 + STOMP 实时通知
        containerRegistry.RegisterSingleton<ISupplierService, SupplierService>();
        containerRegistry.RegisterSingleton<SupplierSyncCoordinator>();
        // 磅单管理：免密 API 直连 + STOMP 实时通知
        containerRegistry.RegisterSingleton<IWeightRecordService, WeightRecordService>();
        containerRegistry.RegisterSingleton<WeightRecordSyncCoordinator>();
        // 密炼物料信息：API直连 + STOMP实时通知
        containerRegistry.RegisterSingleton<IMixerMaterialService, MixerMaterialService>();
        containerRegistry.RegisterSingleton<MixerMaterialSyncCoordinator>();
        // 原料入场记录：免密 API 直连 + STOMP 实时通知
        containerRegistry.RegisterSingleton<IRawMaterialEntryService, RawMaterialEntryService>();
        containerRegistry.RegisterSingleton<RawMaterialEntrySyncCoordinator>();
        // 原材料卡片：免密 API 直连 + STOMP 实时通知
        containerRegistry.RegisterSingleton<IRawMaterialCardService, RawMaterialCardService>();
        containerRegistry.RegisterSingleton<RawMaterialCardSyncCoordinator>();
        // 仓库数据只读缓存（供库区等模块筛选使用，不含 CRUD 页面）
        containerRegistry.RegisterSingleton<IWarehouseService, WarehouseService>();
        // 库区管理：免密 API 直连 + STOMP 实时通知
        containerRegistry.RegisterSingleton<IWarehouseAreaService, WarehouseAreaService>();
        containerRegistry.RegisterSingleton<WarehouseAreaSyncCoordinator>();
        // 密炼物料皮重策略：免密 API 直连 + STOMP 实时通知
        containerRegistry.RegisterSingleton<IMixerMaterialTareStrategyService, MixerMaterialTareStrategyService>();
        containerRegistry.RegisterSingleton<MixerMaterialTareStrategySyncCoordinator>();
        // 分类字典：启动同步 + 断线重连补刷
        containerRegistry.RegisterSingleton<CategorySyncCoordinator>();
        // 数据字典：启动同步 + 断线重连补刷
        containerRegistry.RegisterSingleton<DictSyncCoordinator>();
        // 统一轮询管理器（修改 SyncPollManager.PollInterval 即可调整所有模块的轮询间隔）
        containerRegistry.RegisterSingleton<SyncPollManager>();
        // 打印服务：PrintDot 桥接器 + 打印模板（含 STOMP 实时同步 + 本地缓存）
        containerRegistry.RegisterSingleton<IPrintDotService, PrintDotService>();
        containerRegistry.RegisterSingleton<IPrintTemplateService, PrintTemplateService>();
        containerRegistry.RegisterSingleton<IPrintBizTemplateBindService, PrintBizTemplateBindService>();
        containerRegistry.RegisterSingleton<PrintTemplateSyncCoordinator>();
        containerRegistry.RegisterSingleton<PrintBizTemplateBindSyncCoordinator>();

        var serviceCollection = new ServiceCollection();
        serviceCollection.AddTransient<DisconnectGuardHandler>();
        serviceCollection.AddHttpClient("JeecgApi", (sp, client) =>
        {
            var config = containerRegistry.GetContainer().Resolve<IConfiguration>();
            var baseUrl = config.GetValue<string>("JeecgIntegration:BaseUrl")?.TrimEnd('/');
            if (!string.IsNullOrWhiteSpace(baseUrl))
            {
                client.BaseAddress = new Uri(baseUrl);
            }

            var tokenStore = containerRegistry.GetContainer().Resolve<TokenStore>();
            var token = tokenStore.GetTokenAsync(default).ConfigureAwait(false).GetAwaiter().GetResult();
            if (!string.IsNullOrWhiteSpace(token))
            {
                client.DefaultRequestHeaders.Authorization = new System.Net.Http.Headers.AuthenticationHeaderValue("Bearer", token);
            }
            client.DefaultRequestHeaders.TryAddWithoutValidation("X-Platform", "desktop");
            // .NET HttpClient 要求请求头值为 ASCII；主机名/中文姓名否则会报 Request headers must contain only ASCII characters
            client.DefaultRequestHeaders.TryAddWithoutValidation("X-Host-Name", ToAsciiHttpHeaderValue(Environment.MachineName));
            var currentUser = AppSession.CurrentUser;
            client.DefaultRequestHeaders.TryAddWithoutValidation("X-User-Name", ToAsciiHttpHeaderValue(currentUser?.Account ?? "unknown"));
            client.DefaultRequestHeaders.TryAddWithoutValidation("X-Real-Name", ToAsciiHttpHeaderValue(currentUser?.RealName ?? ""));
        }).AddPolicyHandler(GetRetryPolicy())
          .AddHttpMessageHandler<DisconnectGuardHandler>();

        var provider = serviceCollection.BuildServiceProvider();
        var httpClientFactory = provider.GetRequiredService<IHttpClientFactory>();
        containerRegistry.RegisterInstance(httpClientFactory);
    }

    public void OnInitialized(IContainerProvider containerProvider)
    {
        var networkMonitor = containerProvider.Resolve<INetworkMonitor>();
        var outboxProcessor = containerProvider.Resolve<OutboxProcessor>();
        var signalService = containerProvider.Resolve<ISignalRService>();

        _ = networkMonitor.StartAsync(CancellationToken.None);
        _ = outboxProcessor.StartConsumerAsync(CancellationToken.None);
        // 用户镜像 + 设备指令：统一 STOMP（/ws/device），免密与设备 Token 模式均启动
        _ = Task.Run(() => signalService.ConnectUnifiedDeviceChannelAsync(CancellationToken.None));
        // 强制实例化事件订阅器（单例，构造函数内完成订阅注册）
        _ = containerProvider.Resolve<SysUserEventSubscriber>();
        // 统一轮询管理器必须在各协调器之前实例化，确保 Register() 能正常接收注册
        _ = containerProvider.Resolve<SyncPollManager>();
        // 强制实例化车辆同步协调器（构造函数内订阅 STOMP 车辆变更事件）
        _ = containerProvider.Resolve<VehicleSyncCoordinator>();
        // 强制实例化客户同步协调器
        _ = containerProvider.Resolve<CustomerSyncCoordinator>();
        // 强制实例化供应商同步协调器
        _ = containerProvider.Resolve<SupplierSyncCoordinator>();
        // 强制实例化磅单同步协调器
        _ = containerProvider.Resolve<WeightRecordSyncCoordinator>();
        // 强制实例化密炼物料同步协调器
        _ = containerProvider.Resolve<MixerMaterialSyncCoordinator>();
        // 强制实例化原料入场记录同步协调器
        _ = containerProvider.Resolve<RawMaterialEntrySyncCoordinator>();
        // 强制实例化原材料卡片同步协调器
        _ = containerProvider.Resolve<RawMaterialCardSyncCoordinator>();
        // 强制实例化库区同步协调器
        _ = containerProvider.Resolve<WarehouseAreaSyncCoordinator>();
        // 强制实例化密炼物料皮重策略同步协调器
        _ = containerProvider.Resolve<MixerMaterialTareStrategySyncCoordinator>();
        // 强制实例化分类字典同步协调器
        _ = containerProvider.Resolve<CategorySyncCoordinator>();
        // 强制实例化数据字典同步协调器
        _ = containerProvider.Resolve<DictSyncCoordinator>();
        // 强制实例化打印模板同步协调器
        _ = containerProvider.Resolve<PrintTemplateSyncCoordinator>();
        _ = containerProvider.Resolve<PrintBizTemplateBindSyncCoordinator>();
    }

    private static IAsyncPolicy<HttpResponseMessage> GetRetryPolicy()
    {
        return HttpPolicyExtensions
            .HandleTransientHttpError()
            .WaitAndRetryAsync(3, retryAttempt => TimeSpan.FromSeconds(Math.Pow(2, retryAttempt)));
    }

    /// <summary>
    /// 将自定义请求头转为 ASCII（含非 ASCII 时整体做 UTF-8 Base64，前缀 B64. 便于后端按需解码）。
    /// </summary>
    private static string ToAsciiHttpHeaderValue(string? value)
    {
        if (string.IsNullOrEmpty(value))
        {
            return string.Empty;
        }

        foreach (var c in value)
        {
            if (c > 127)
            {
                return "B64." + Convert.ToBase64String(Encoding.UTF8.GetBytes(value));
            }
        }

        return value;
    }
}