import { Injectable, Logger, OnModuleInit } from '@nestjs/common';
import { SchedulerRegistry } from '@nestjs/schedule';
import { CronJob } from 'cron';
import * as path from 'path';
import { PrismaService } from '../prisma/prisma.service';
import {
  ColdImportService,
  SyncAlreadyRunningError,
} from '../modules/sync/cold-import/cold-import.service';
import { OrgTreeService } from '../modules/auth/org-tree';
import { PersonaService } from '../modules/persona/persona.service';
import { PlanEngineService } from '../modules/plan/engine/plan-engine.service';

/**
 * SyncIncrementalSchedulerService — DW 直连增量自动跑(**每宿主独立 cron**)
 *
 * 每天在各 host 自己的 DW 刷新窗口后触发:
 *   1. ColdImportService.importDirectory({ incremental: true })(游标增量 + stub 补齐)
 *   2. Persona 重算(本次 sync 写入的 distinct patient_id)
 *   3. org-tree 预热 + Plan 重算(per tenant)
 *
 * **多 host 调度**(启动时动态注册,取代单一全局 @Cron):
 *   - 发现 data/<host>/manifest.yaml 中 auto_sync=true 的 host;
 *   - 每个 host 用**自己 manifest 的 incremental_cron + timezone**注册独立 CronJob
 *     —— 上游 DW 刷新窗口因 host 而异,各自在就绪后跑,不再挤在一个全局时刻;
 *   - manifest 没配 cron → 回退全局 PAC_INCREMENTAL_CRON(按 host manifest 时区解释)。
 *   - PAC_INCREMENTAL_HOSTS env(逗号分隔)仍作显式 override 逃生口(用全局 cron)。
 *
 * 跑失败:cursor 不前进 → 下次自动 catchup;log ERROR 不抛。
 */
@Injectable()
export class SyncIncrementalSchedulerService implements OnModuleInit {
  private readonly logger = new Logger(SyncIncrementalSchedulerService.name);

  constructor(
    private readonly prisma: PrismaService,
    private readonly coldImport: ColdImportService,
    private readonly orgTree: OrgTreeService,
    private readonly persona: PersonaService,
    private readonly planEngine: PlanEngineService,
    private readonly scheduler: SchedulerRegistry,
  ) {}

  onModuleInit(): void {
    const globalCron = process.env.PAC_INCREMENTAL_CRON;
    const dataDir = this.dataDir();
    // env override(逗号分隔)优先;否则自动发现 auto_sync=true 的 host
    const envHosts = (process.env.PAC_INCREMENTAL_HOSTS ?? '')
      .split(',')
      .map((h) => h.trim())
      .filter(Boolean);
    const hosts = envHosts.length > 0 ? envHosts : this.coldImport.discoverAutoSyncHostDirs(dataDir);

    if (hosts.length === 0) {
      this.logger.log('sync-incremental: 无 auto_sync host,不注册 cron');
      return;
    }

    for (const host of hosts) {
      const ops = this.coldImport.getHostOpsConfig(host);
      // 每宿主 cron:manifest.incremental_cron 优先,否则全局 env;都没有 → 不注册(该 host 不自动同步)
      const cronExpr = ops.incrementalCron ?? globalCron;
      if (!cronExpr) {
        this.logger.warn(
          `sync-incremental: host=${host} 无 cron(manifest.incremental_cron 与 PAC_INCREMENTAL_CRON 都未设),跳过注册`,
        );
        continue;
      }
      const timeZone = ops.timezone ?? 'Asia/Shanghai';
      const jobName = `sync-incremental:${host}`;
      const job = new CronJob(
        cronExpr,
        () => {
          void this.runHostSafe(host);
        },
        null,
        false,
        timeZone,
      );
      this.scheduler.addCronJob(jobName, job as unknown as Parameters<SchedulerRegistry['addCronJob']>[1]);
      job.start();
      this.logger.log(
        `sync-incremental: 注册 host=${host} cron='${cronExpr}' tz=${timeZone}` +
          `${ops.incrementalCron ? '' : '(全局默认)'}`,
      );
    }
  }

  private dataDir(): string {
    return process.env.PAC_INCREMENTAL_DATA_DIR ?? path.resolve(__dirname, '../../data');
  }

  /// 单 host 跑一轮(cron 回调用,吞异常不影响该 host 下次 / 别的 host)
  private async runHostSafe(host: string): Promise<void> {
    try {
      await this.runOne(path.join(this.dataDir(), host));
    } catch (err) {
      if (err instanceof SyncAlreadyRunningError) {
        this.logger.warn(`sync-incremental: host=${host} skip(并发锁拦截):${err.message}`);
      } else {
        this.logger.error(`sync-incremental: host=${host} failed: ${(err as Error).message}`);
      }
    }
  }

  /// 跑单个 host 的完整链路(可手动调,debug 用)
  async runOne(dir: string): Promise<void> {
    const started = Date.now();
    const r = await this.coldImport.importDirectory(dir, { incremental: true });
    this.logger.log(
      `sync-incremental: cold-import done host=${r.hostName} ` +
        `patients=${r.totals.patientsUpserted} txns=${r.totals.transactionsWritten} ` +
        `facts(created=${r.totals.factsCreated} superseded=${r.totals.factsSuperseded})`,
    );

    // ─── Persona recompute(affected patient only)──────
    if (r.syncLogId) {
      const syncLog = await this.prisma.syncLog.findUnique({ where: { id: r.syncLogId } });
      if (syncLog) {
        const affected = await this.prisma.patientTransaction.findMany({
          where: { hostId: r.hostId, createdAt: { gte: syncLog.startedAt } },
          select: { patientId: true },
          distinct: ['patientId'],
        });
        const pids = affected.map((p) => p.patientId).filter((p): p is string => !!p);
        let ok = 0;
        let failed = 0;
        // 串行:persona 重算是 **Node-CPU-bound**(16 特征 JS 计算占主),实测 conc=8 不提速、
        //   还因竞争略慢(本地 300 患者 30s→32s)。异步并发只能重叠 I/O,重叠不了单线程 CPU。
        //   真要提速需 profile 特征计算或多进程/worker,不在此处做。
        for (const pid of pids) {
          try {
            await this.persona.recompute({
              patientId: pid,
              source: `incremental-sync:${r.runId}`,
            });
            ok++;
          } catch (err) {
            failed++;
            this.logger.warn(`persona patient=${pid}: ${(err as Error).message}`);
          }
        }
        this.logger.log(`persona recompute: success=${ok} failed=${failed} (${pids.length} affected)`);
      }
    }

    const tenants = await this.prisma.patient.findMany({
      where: { hostId: r.hostId },
      distinct: ['tenantId'],
      select: { tenantId: true },
    });

    // ─── org-tree 主动预热(消除"当天第一个非 admin 登录用户撞冷派生"的尾延迟)──────
    // 本次同步刚写完新数据,强制重派生一次(forceRefresh),后续 24h TTL 内任何用户登录
    // 都直接命中缓存 —— 冷派生的代价(生产 379 万行,实测约 5s)由本 cron(后台)吞掉,
    // 不再由真实用户请求承担。单个 tenant 预热失败不影响其余 tenant / 后续 sync 步骤。
    for (const t of tenants) {
      try {
        await this.orgTree.getTree(r.hostId, t.tenantId, true);
      } catch (err) {
        this.logger.warn(
          `org-tree 预热失败 host=${r.hostId} tenant=${t.tenantId}: ${(err as Error).message}`,
        );
      }
    }

    // ─── Plan recompute(per tenant)──────
    let plansCreated = 0;
    for (const t of tenants) {
      const r2 = await this.planEngine.runAllForHost({
        hostId: r.hostId,
        tenantId: t.tenantId,
        now: new Date(),
      });
      plansCreated += r2.plansCreated;
    }
    this.logger.log(
      `plan recompute: ${tenants.length} tenants, plansCreated=${plansCreated} ` +
        `(total elapsed=${Date.now() - started}ms)`,
    );
  }
}
