import { NextRequest, NextResponse } from 'next/server';

import { generateText } from 'ai';

import { glmModel } from '@/lib/glm';

import { runSSH } from '@/lib/workers/ssh';

import { setCap } from '@/lib/db';

export const runtime = 'nodejs';

export const dynamic = 'force-dynamic';

export const maxDuration = 120;

const OLLAMA = process.env.ZPM_OLLAMA_URL || 'http://REDACTED-IP:11434';

async function pingModel(model: string): Promise<{ ok: boolean; latencyMs: number; detail: string }> {

  const start = Date.now();

  try {

    if (model === 'glm') {

      const r = await generateText({ model: glmModel, prompt: 'Reply with exactly: OK' });

      return { ok: /ok/i.test(r.text), latencyMs: Date.now() - start, detail: r.text.trim().slice(0, 40) };

    }

    if (model === 'claude') {

      const r = await runSSH('REDACTED-IP', 'zion', 'echo ok', 15_000);

      return { ok: r.stdout.trim() === 'ok', latencyMs: Date.now() - start, detail: 'CT215 reachable' };

    }

    if (model === 'codex') {

      const r = await runSSH('REDACTED-IP', 'zion', 'echo ok', 15_000);

      return { ok: r.stdout.trim() === 'ok', latencyMs: Date.now() - start, detail: 'CT241 reachable' };

    }

    if (model === 'ollama') {

      const ctrl = new AbortController();

      const t = setTimeout(() => ctrl.abort(), 10_000);

      const res = await fetch(`${OLLAMA}/api/tags`, { signal: ctrl.signal });

      clearTimeout(t);

      return { ok: res.ok, latencyMs: Date.now() - start, detail: res.ok ? '3060 ollama up' : `http ${res.status}` };

    }

    return { ok: false, latencyMs: 0, detail: 'unknown model' };

  } catch (e) {

    return { ok: false, latencyMs: Date.now() - start, detail: (e as Error).message.slice(0, 80) };

  }

}

export async function POST(req: NextRequest) {

  const { model } = await req.json();

  if (!model) return NextResponse.json({ error: 'model required' }, { status: 400 });

  const res = await pingModel(model);

  if (model !== 'glm') setCap(model, res.ok ? 1 : 0, res.ok ? 'ping ok' : `ping fail: ${res.detail}`);

  return NextResponse.json({ model, ...res });

}