import { Router } from 'express';
import { z } from 'zod';
import { getDb } from '../db/index.js';
import { getAllPenalties, getCustomWeights, getRoutingScores, getRoutingStrategy, setCustomWeights, setRoutingStrategy } from '../services/router.js';
import { BANDIT_PRESETS } from '../services/scoring.js';
import { parseBudget } from '../lib/budget.js';
export const fallbackRouter = Router();
// ── Bandit routing strategy ─────────────────────────────────────────────────
// GET  /routing → active strategy, preset weights, the saved custom weights,
//                 and the per-model score breakdown (reliability / speed /
//                 intelligence + guardrails).
fallbackRouter.get('/routing', (_req, res) => {
    res.json({ ...getRoutingScores(), customWeights: getCustomWeights() });
});
const routingSchema = z.object({
    strategy: z.enum(['priority', 'balanced', 'smartest', 'fastest', 'reliable', 'custom']),
    // Only meaningful with strategy 'custom'. Any non-negative vector with a
    // positive sum is accepted; the server normalizes it to sum to 1.
    weights: z.object({
        reliability: z.number().min(0).max(1),
        speed: z.number().min(0).max(1),
        intelligence: z.number().min(0).max(1),
    }).refine(w => w.reliability + w.speed + w.intelligence > 0, {
        message: 'weights must not all be zero',
    }).optional(),
});
// PUT /routing → switch strategy. Presets are just weight vectors over the three
// axes; 'custom' uses the user-saved vector; 'priority' falls back to the legacy
// manual chain order.
fallbackRouter.put('/routing', (req, res) => {
    const parsed = routingSchema.safeParse(req.body);
    if (!parsed.success) {
        res.status(400).json({ error: { message: parsed.error.errors.map(e => e.message).join(', ') } });
        return;
    }
    if (parsed.data.strategy === 'custom' && parsed.data.weights) {
        setCustomWeights(parsed.data.weights);
    }
    setRoutingStrategy(parsed.data.strategy);
    res.json({ strategy: getRoutingStrategy(), presets: BANDIT_PRESETS, customWeights: getCustomWeights() });
});
// Get fallback chain (with dynamic penalties)
fallbackRouter.get('/', (_req, res) => {
    const db = getDb();
    const rows = db.prepare(`
    SELECT fc.model_db_id, fc.priority, fc.enabled,
           m.platform, m.model_id, m.display_name, m.intelligence_rank,
           m.speed_rank, m.size_label, m.rpm_limit, m.rpd_limit,
           m.monthly_token_budget, m.supports_vision, m.supports_tools
    FROM fallback_config fc
    JOIN models m ON m.id = fc.model_db_id
    ORDER BY fc.priority ASC
  `).all();
    // Count enabled keys per platform
    const keyCounts = db.prepare(`
    SELECT platform, COUNT(*) as count
    FROM api_keys WHERE enabled = 1
    GROUP BY platform
  `).all();
    const keyCountMap = new Map(keyCounts.map(k => [k.platform, k.count]));
    // Get current dynamic penalties
    const penalties = getAllPenalties();
    const penaltyMap = new Map(penalties.map(p => [p.modelDbId, p]));
    res.json(rows.map(r => {
        const penalty = penaltyMap.get(r.model_db_id);
        return {
            modelDbId: r.model_db_id,
            priority: r.priority,
            effectivePriority: r.priority + (penalty?.penalty ?? 0),
            penalty: penalty?.penalty ?? 0,
            rateLimitHits: penalty?.count ?? 0,
            enabled: r.enabled === 1,
            platform: r.platform,
            modelId: r.model_id,
            displayName: r.display_name,
            intelligenceRank: r.intelligence_rank,
            speedRank: r.speed_rank,
            sizeLabel: r.size_label,
            rpmLimit: r.rpm_limit,
            rpdLimit: r.rpd_limit,
            monthlyTokenBudget: r.monthly_token_budget,
            supportsVision: r.supports_vision === 1,
            supportsTools: r.supports_tools === 1,
            keyCount: keyCountMap.get(r.platform) ?? 0,
        };
    }));
});
const updateSchema = z.array(z.object({
    modelDbId: z.number(),
    priority: z.number(),
    enabled: z.boolean(),
}));
// Update fallback chain (full replace)
fallbackRouter.put('/', (req, res) => {
    const parsed = updateSchema.safeParse(req.body);
    if (!parsed.success) {
        res.status(400).json({ error: { message: parsed.error.errors.map(e => e.message).join(', ') } });
        return;
    }
    const db = getDb();
    const update = db.prepare(`
    UPDATE fallback_config SET priority = ?, enabled = ? WHERE model_db_id = ?
  `);
    const updateAll = db.transaction(() => {
        for (const entry of parsed.data) {
            update.run(entry.priority, entry.enabled ? 1 : 0, entry.modelDbId);
        }
    });
    updateAll();
    res.json({ success: true });
});
// `intelligence_rank` is scoped to each provider's own catalog — a provider's
// #1 model is not globally #1 (see issue #135: MiniMax's top model outranking
// Gemini Pro because both read "Intel #1"). `size_label` IS a cross-provider
// capability tier, so normalize on it first and use intelligence_rank only as
// an in-tier tiebreaker. Unknown labels sort last.
const INTELLIGENCE_TIER = "CASE m.size_label WHEN 'Frontier' THEN 1 WHEN 'Large' THEN 2 WHEN 'Medium' THEN 3 WHEN 'Small' THEN 4 ELSE 5 END";
// Sort presets — `orderBy` is selected from a fixed whitelist, never from
// user input directly, so the interpolation below is safe.
const SORT_PRESETS = {
    intelligence: `${INTELLIGENCE_TIER} ASC, m.intelligence_rank ASC`,
    speed: 'm.speed_rank ASC',
    budget: "CASE m.monthly_token_budget WHEN '~120M' THEN 1 WHEN '~50-100M' THEN 2 WHEN '~30M' THEN 3 WHEN '~18-45M' THEN 4 WHEN '~18M' THEN 5 WHEN '~15M' THEN 6 WHEN '~12M' THEN 7 WHEN '~6M' THEN 8 WHEN '~5-10M' THEN 9 WHEN '~4M' THEN 10 ELSE 11 END ASC",
};
fallbackRouter.post('/sort/:preset', (req, res) => {
    const preset = String(req.params.preset);
    const orderBy = SORT_PRESETS[preset];
    if (!orderBy) {
        res.status(400).json({ error: { message: `Unknown preset: ${preset}. Use: intelligence, speed, budget` } });
        return;
    }
    const db = getDb();
    const models = db.prepare(`SELECT m.id FROM models m ORDER BY ${orderBy}`).all();
    const update = db.prepare('UPDATE fallback_config SET priority = ? WHERE model_db_id = ?');
    const reorder = db.transaction(() => {
        for (let i = 0; i < models.length; i++) {
            update.run(i + 1, models[i].id);
        }
    });
    reorder();
    res.json({ success: true, preset });
});
// Token usage per model for the stacked bar
fallbackRouter.get('/token-usage', (_req, res) => {
    const db = getDb();
    // Get platforms that have enabled keys
    const platforms = db.prepare(`
    SELECT DISTINCT ak.platform
    FROM api_keys ak
    WHERE ak.enabled = 1
  `).all();
    const platformSet = new Set(platforms.map(p => p.platform));
    // Get monthly budget per model, ordered by fallback priority
    const models = db.prepare(`
    SELECT m.platform, m.model_id, m.display_name, m.monthly_token_budget,
           fc.priority
    FROM models m
    JOIN fallback_config fc ON fc.model_db_id = m.id
    WHERE m.enabled = 1
    ORDER BY fc.priority ASC
  `).all();
    // Build per-model breakdown (only platforms with keys)
    const modelBudgets = models
        .filter(m => platformSet.has(m.platform))
        .map(m => ({
        displayName: m.display_name,
        platform: m.platform,
        budget: parseBudget(m.monthly_token_budget),
    }));
    const totalBudget = modelBudgets.reduce((s, m) => s + m.budget, 0);
    // Tokens used this month
    const usage = db.prepare(`
    SELECT
      COALESCE(SUM(input_tokens + output_tokens), 0) as total_used
    FROM requests
    WHERE created_at >= datetime('now', 'start of month')
      AND request_type = 'chat'
  `).get();
    res.json({
        totalBudget,
        totalUsed: usage.total_used,
        models: modelBudgets,
    });
});
//# sourceMappingURL=fallback.js.map