"use strict";
const assert = require("node:assert/strict");
const { test } = require("node:test");
const { parseDecision, extractCliResponse, tokenUsage, makePrompt } = require("../gemini_bot_logic");

test("accepts one bounded action", () => {
  assert.deepEqual(parseDecision('{"action":"explore","direction":"north","blocks":16}'),
    { action: "explore", direction: "north", blocks: 16 });
  assert.deepEqual(parseDecision('```json\n{"action":"mine","block":"stone"}\n```'),
    { action: "mine", block: "stone" });
  assert.deepEqual(parseDecision('{"action":"follow","player":"IffyBinkie6134"}'),
    { action: "follow", player: "IffyBinkie6134" });
});

test("rejects code, chat commands, arbitrary targets, and extra prose", () => {
  for (const value of [
    'Explain: {"action":"mine","block":"stone"}',
    '{"action":"say","message":"/op stranger"}',
    '{"action":"place","block":"bedrock"}',
    '{"action":"follow","player":"stranger"}',
    '{"action":"explore","direction":"down","blocks":16}',
    '{"action":"explore","direction":"north","blocks":9999}',
    '{"action":"run","code":"process.exit(1)"}',
    '{"action":"mine","block":"stone"}{"action":"say","message":"hi"}',
    '{"action":"mine","block":"stone"',
  ]) assert.throws(() => parseDecision(value));
});

test("reads CLI envelope, and fails closed on API errors", () => {
  const envelope = { response: '{"action":"wait"}', stats: { models: {} } };
  assert.deepEqual(extractCliResponse(JSON.stringify(envelope)).decision, { action: "wait" });
  assert.throws(() => extractCliResponse(JSON.stringify({ response: null, error: { code: 429 } })));
  assert.throws(() => extractCliResponse("invalid"));
});

test("aggregates per-model CLI token metrics", () => {
  assert.deepEqual(tokenUsage({ models: { "gemini-3.5-flash-lite": { tokens: { prompt: 120, candidates: 25 } } } }),
    { models_used: ["gemini-3.5-flash-lite"], prompt_tokens: 120, completion_tokens: 25 });
  assert.deepEqual(tokenUsage({ tokens: { input: 10, output: 2 } }),
    { models_used: [], prompt_tokens: 10, completion_tokens: 2 });
});

test("prompt contains one observed world and the short admin request", () => {
  const prompt = makePrompt({ position: { x: 0, y: 70, z: 0 }, nearby_blocks: ["stone"] }, "segue o Naldo");
  assert.match(prompt, /Naldo_Bot/);
  assert.match(prompt, /segue o Naldo/);
  assert.match(prompt, /nearby_blocks/);
  assert.ok(prompt.length < 3000);
});