import assert from 'node:assert/strict';
import test from 'node:test';
import {
  appendAgentVisibleText,
  formatPrefetchedSearchContext,
  prefetchLiveSearchResults,
  resolvePortalSearxngEndpoint,
  shouldPrefetchLiveSearch,
} from './mindsearch-prefetch.mjs';

test('resolvePortalSearxngEndpoint rewrites docker host for in-process Portal', () => {
  assert.equal(
    resolvePortalSearxngEndpoint({
      TKMIND_SEARCH_SEARXNG_URL: 'http://host.docker.internal:20080/search',
    }),
    'http://127.0.0.1:20080/search',
  );
  assert.equal(resolvePortalSearxngEndpoint({}), '');
});

test('shouldPrefetchLiveSearch only runs for web skills and can be disabled', () => {
  assert.equal(shouldPrefetchLiveSearch({ suggestedSkill: 'web' }), true);
  assert.equal(shouldPrefetchLiveSearch({ selectedSkill: 'search-enhanced' }), true);
  assert.equal(shouldPrefetchLiveSearch({ suggestedSkill: 'static-page-publish' }), false);
  assert.equal(shouldPrefetchLiveSearch({
    suggestedSkill: 'web',
    env: { MEMIND_LIVE_SEARCH_PREFETCH: '0' },
  }), false);
});

test('formatPrefetchedSearchContext asks the agent not to scrape hot-list pages first', () => {
  const text = formatPrefetchedSearchContext([
    { rank: 1, title: '今日热榜', url: 'https://example.com/hot', snippet: '话题' },
  ], { query: '抖音今天的热门话题是什么' });
  assert.match(text, /联网搜索预取/);
  assert.match(text, /抖音今天的热门话题是什么/);
  assert.match(text, /禁止先 fetch_url/);
  assert.match(text, /tkmind_search/);
  assert.match(text, /https:\/\/example.com\/hot/);
});

test('appendAgentVisibleText keeps displayText and appends to the first text part', () => {
  const next = appendAgentVisibleText({
    role: 'user',
    content: [{ type: 'text', text: '请使用 web 技能：问题' }],
    metadata: { displayText: '抖音今天的热门话题是什么' },
  }, '【联网搜索预取】result');
  assert.equal(next.metadata.displayText, '抖音今天的热门话题是什么');
  assert.match(next.content[0].text, /请使用 web 技能：问题/);
  assert.match(next.content[0].text, /联网搜索预取/);
});

test('prefetchLiveSearchResults calls SearXNG and fails open on errors', async () => {
  const hits = await prefetchLiveSearchResults({
    query: '抖音热门话题',
    suggestedSkill: 'web',
    env: { TKMIND_SEARCH_SEARXNG_URL: 'http://127.0.0.1:20080/search' },
    searchImpl: async (query, options) => {
      assert.equal(query, '抖音热门话题');
      assert.equal(options.endpoint, 'http://127.0.0.1:20080/search');
      return [{ rank: 1, title: '话题A', url: 'https://a.example', snippet: '摘要' }];
    },
  });
  assert.equal(hits.provider, 'searxng');
  assert.equal(hits.resultCount, 1);
  assert.match(hits.injectionText, /话题A/);

  const skipped = await prefetchLiveSearchResults({
    query: '抖音热门话题',
    suggestedSkill: 'web',
    env: {},
    searchImpl: async () => {
      throw new Error('should not run');
    },
  });
  assert.equal(skipped, null);

  const failed = await prefetchLiveSearchResults({
    query: '抖音热门话题',
    suggestedSkill: 'web',
    env: { TKMIND_SEARCH_SEARXNG_URL: 'http://127.0.0.1:20080/search' },
    searchImpl: async () => {
      throw new Error('searxng down');
    },
  });
  assert.equal(failed, null);
});