// ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
// ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========

import { describe, expect, it } from 'vitest'
import { RAMIndexCacheStore } from '../../../cache/index/ram.ts'
import { materialize } from '../../../io/types.ts'
import { PathSpec } from '../../../types.ts'
import { FakeDiscordTransport, makeFakeResource, seedChannel, seedGuild } from './_test_util.ts'
import { DISCORD_RG } from './rg.ts'

const DEC = new TextDecoder()

async function runRg(
  paths: PathSpec[],
  texts: string[],
  flags: Record<string, string | boolean>,
  options: { index?: RAMIndexCacheStore; transport?: FakeDiscordTransport } = {},
): Promise<{ stdout: string; exitCode: number }> {
  const cmd = DISCORD_RG[0]
  if (cmd === undefined) throw new Error('rg not registered')
  const transport = options.transport ?? new FakeDiscordTransport()
  const resource = makeFakeResource(transport)
  const result = await cmd.fn(resource.accessor, paths, texts, {
    stdin: null,
    flags,
    filetypeFns: null,
    cwd: '/',
    resource,
    ...(options.index !== undefined ? { index: options.index } : {}),
  })
  if (result === null) return { stdout: '', exitCode: 0 }
  const [out, io] = result
  const buf =
    out === null
      ? new Uint8Array()
      : out instanceof Uint8Array
        ? out
        : await materialize(out as AsyncIterable<Uint8Array>)
  return { stdout: DEC.decode(buf), exitCode: io.exitCode }
}

describe('discord rg', () => {
  it('uses native search for a channel directory (useNative scope)', async () => {
    const idx = new RAMIndexCacheStore()
    await seedGuild(idx, '/mnt/discord', 'My_Server__G1', 'G1')
    await seedChannel(idx, '/mnt/discord', 'My_Server__G1', 'general__C1', 'C1')
    const transport = new FakeDiscordTransport((_method, endpoint) => {
      if (endpoint === '/guilds/G1/messages/search') {
        return {
          total_results: 1,
          messages: [
            [
              {
                id: '175928847299117056',
                content: 'hello world',
                channel_id: 'C1',
                timestamp: '2016-04-30T12:00:00.000+00:00',
                author: { username: 'alice' },
              },
            ],
          ],
        }
      }
      return null
    })
    const out = await runRg(
      [
        new PathSpec({
          original: '/mnt/discord/My_Server__G1/channels/general__C1',
          directory: '/mnt/discord/My_Server__G1/channels/general__C1',
          resolved: false,
          prefix: '/mnt/discord',
        }),
      ],
      ['hello'],
      {},
      { index: idx, transport },
    )
    expect(transport.calls[0]?.endpoint).toBe('/guilds/G1/messages/search')
    expect(transport.calls[0]?.params?.content).toBe('hello')
    const lines = out.stdout.split('\n').filter((l) => l !== '')
    expect(lines.length).toBe(1)
    expect(lines[0]).toContain('hello world')
  })

  it('returns exit 1 when native search has no matches', async () => {
    const transport = new FakeDiscordTransport(() => ({ total_results: 0, messages: [] }))
    const out = await runRg(
      [
        new PathSpec({
          original: '/mnt/discord/My_Server__G1/channels/general__C1',
          directory: '/mnt/discord/My_Server__G1/channels/general__C1',
          resolved: false,
          prefix: '/mnt/discord',
        }),
      ],
      ['hello'],
      {},
      { transport },
    )
    expect(out.exitCode).toBe(1)
  })
})
