import test from 'ava'

import { DiffbotClient } from '@/services'

import { isCI, ky } from '../_utils'

test('Diffbot.extractAnalyze - transitivebullsh.it', async (t) => {
  if (!process.env.DIFFBOT_API_KEY || isCI) {
    return t.pass()
  }

  t.timeout(2 * 60 * 1000)
  const client = new DiffbotClient({ ky })

  const result = await client.extractAnalyze({
    url: 'https://transitivebullsh.it'
  })
  // console.log(JSON.stringify(result, null, 2))
  t.is(result.type, 'list')
  t.is(result.objects?.length, 1)
})

// TODO
// test.only('Diffbot.extractAnalyze - theguardian.com/world/ukraine', async (t) => {
//   if (!process.env.DIFFBOT_API_KEY || isCI) {
//     return t.pass()
//   }

//   t.timeout(2 * 60 * 1000)
//   const client = new DiffbotClient({ ky })

//   const result = await client.extractAnalyze({
//     url: 'https://www.theguardian.com/world/ukraine'
//   })
//   console.log(JSON.stringify(result, null, 2))
//   t.is(result.type, 'list')
//   t.is(result.objects?.length, 1)
// })

test('Diffbot.extractArticle', async (t) => {
  if (!process.env.DIFFBOT_API_KEY || isCI) {
    return t.pass()
  }

  t.timeout(2 * 60 * 1000)
  const client = new DiffbotClient({ ky })

  const result = await client.extractArticle({
    url: 'https://www.nytimes.com/2023/05/31/magazine/ai-start-up-accelerator-san-francisco.html'
    // fields: ['meta']
  })
  // console.log(JSON.stringify(result, null, 2))
  t.is(result.objects[0].type, 'article')
})

test('Diffbot.knowledgeGraphSearch', async (t) => {
  if (!process.env.DIFFBOT_API_KEY || isCI) {
    return t.pass()
  }

  t.timeout(2 * 60 * 1000)
  const client = new DiffbotClient({ ky })

  const result = await client.knowledgeGraphSearch({
    type: 'query',
    query: 'Brown University',
    size: 10
  })
  // console.log(JSON.stringify(result, null, 2))
  t.true(Array.isArray(result.data))
  t.is(result.data.length, 10)
})

test('Diffbot.knowledgeGraphEnhance', async (t) => {
  if (!process.env.DIFFBOT_API_KEY || isCI) {
    return t.pass()
  }

  t.timeout(2 * 60 * 1000)
  const client = new DiffbotClient({ ky })

  const result = await client.knowledgeGraphEnhance({
    type: 'Person',
    name: 'Travis Fischer',
    url: 'https://transitivebullsh.it'
  })
  // console.log(JSON.stringify(result, null, 2))
  t.is(result.data[0]?.entity?.githubUri, 'github.com/transitive-bullshit')
})