import type { ContentTypeBoostStatus } from "../config/content-types";
import type { ActivationStatus } from "../core/activation-status";
import type { ChunkingStatus } from "../store/chunking";

export type HealthCheckStatus = "ok" | "warn" | "error";

export type HealthActionKind =
  | "add-collection"
  | "open-collections"
  | "sync"
  | "embed"
  | "download-models"
  | "free-space";

export type OnboardingStepStatus = "complete" | "current" | "upcoming";

export interface StatusCollection {
  name: string;
  path: string;
  documentCount: number;
  chunkCount: number;
  embeddedCount: number;
}

export interface SuggestedCollection {
  label: string;
  path: string;
  reason: string;
}

export interface OnboardingStep {
  id: string;
  title: string;
  status: OnboardingStepStatus;
  detail: string;
  action?: "add-collection" | "download-models" | "sync" | "embed";
}

export interface OnboardingState {
  ready: boolean;
  stage: "add-collection" | "models" | "indexing" | "ready";
  headline: string;
  detail: string;
  suggestedCollections: SuggestedCollection[];
  steps: OnboardingStep[];
}

export interface HealthCheck {
  id: string;
  title: string;
  status: HealthCheckStatus;
  summary: string;
  detail: string;
  actionLabel?: string;
  actionKind?: HealthActionKind;
}

export interface HealthCenterState {
  state: "healthy" | "needs-attention" | "setup-required";
  summary: string;
  checks: HealthCheck[];
}

export type RuntimeMode = "serve" | "daemon" | "stdio" | "direct-cli";

export interface ResidentStatus {
  schemaVersion: "1.0";
  mode: RuntimeMode;
  resident: boolean;
  uptimeSeconds: number | null;
  listenerPort: number | null;
  admission: {
    state: "accepting" | "draining" | "closed";
    activeRequests: number;
  };
  shutdown: {
    state: "none" | "graceful" | "deadline";
  };
  transport: {
    activeRequests: number;
    activeSessions: number;
    queuedRequests: number;
    maxConcurrentRequests: number;
    maxQueuedRequests: number;
    maxSessions: number;
  };
  readers: {
    active: number;
    queued: number;
    limit: number;
    maxQueued: number;
  };
  models: {
    activeLeases: number;
    leaseAcquisitions: number;
    leaseReleases: number;
    loadedModels: number;
    loadAttempts: number;
    loadSuccesses: number;
    loadFailures: number;
    inflightLoads: number;
  };
  jobs: {
    active: number;
    recent: number;
    failed: number;
  };
  generations: {
    content: number;
    index: number;
  };
}

export interface BackgroundServiceState {
  watcher: {
    expectedCollections: string[];
    activeCollections: string[];
    failedCollections: Array<{ collection: string; reason: string }>;
    queuedCollections: string[];
    syncingCollections: string[];
    lastEventAt: string | null;
    lastSyncAt: string | null;
  };
  embedding: {
    available: boolean;
    pendingDocCount: number;
    running: boolean;
    nextRunAt: number | null;
    lastRunAt: number | null;
    lastResult: { embedded: number; errors: number } | null;
  };
  events: {
    connectedClients: number;
    retryMs: number;
  };
}

export interface BootstrapState {
  runtime: {
    kind: "bun";
    strategy: "manual-install-beta";
    currentVersion: string;
    requiredVersion: string;
    ready: boolean;
    managedByApp: boolean;
    summary: string;
    detail: string;
  };
  policy: {
    offline: boolean;
    allowDownload: boolean;
    source: "default" | "hf-hub-offline" | "gno-offline" | "no-auto-download";
    summary: string;
  };
  cache: {
    path: string;
    totalSizeBytes: number;
    totalSizeLabel: string;
  };
  models: {
    activePresetId: string;
    activePresetName: string;
    estimatedFootprint: string | null;
    downloading: boolean;
    cachedCount: number;
    totalCount: number;
    summary: string;
    entries: Array<{
      role: "embed" | "rerank" | "expand" | "gen";
      uri: string;
      cached: boolean;
      path: string | null;
      sizeBytes: number | null;
      statusLabel: string;
    }>;
  };
}

export interface AppStatusResponse {
  chunking?: ChunkingStatus;
  resident: ResidentStatus;
  indexName: string;
  configPath: string;
  dbPath: string;
  collections: StatusCollection[];
  totalDocuments: number;
  totalChunks: number;
  embeddingBacklog: number;
  lastUpdated: string | null;
  recentErrors: number;
  healthy: boolean;
  activePreset: {
    id: string;
    name: string;
  };
  capabilities: {
    bm25: boolean;
    vector: boolean;
    hybrid: boolean;
    answer: boolean;
  };
  contentTypeBoost: ContentTypeBoostStatus;
  activation: ActivationStatus;
  onboarding: OnboardingState;
  health: HealthCenterState;
  background: BackgroundServiceState;
  bootstrap: BootstrapState;
}