const test = require('tape')
const path = require('path')
const os = require('os')
const rimraf = require('rimraf')
const SecretStack = require('secret-stack')
const caps = require('ssb-caps')
const FeedV1 = require('ppppp-db/feed-v1')
const p = require('util').promisify
const Algorithm = require('../lib/algorithm')
const { generateKeypair } = require('./util')

const createPeer = SecretStack({ appKey: caps.shs })
  .use(require('ppppp-db'))
  .use(require('ssb-box'))
  .use(require('../'))

const ALICE_DIR = path.join(os.tmpdir(), 'dagsync-alice')
const BOB_DIR = path.join(os.tmpdir(), 'dagsync-bob')
const aliceKeys = generateKeypair('alice')
const bobKeys = generateKeypair('bob')

test('sync a feed with goal=all', async (t) => {
  rimraf.sync(ALICE_DIR)
  rimraf.sync(BOB_DIR)

  const alice = createPeer({
    keys: aliceKeys,
    path: ALICE_DIR,
  })

  const bob = createPeer({
    keys: bobKeys,
    path: BOB_DIR,
  })

  await alice.db.loaded()
  await bob.db.loaded()

  const carolKeys = generateKeypair('carol')
  const carolMsgs = []
  const carolID = carolKeys.id
  const carolID_b58 = FeedV1.stripAuthor(carolID)
  for (let i = 1; i <= 10; i++) {
    const rec = await p(alice.db.create)({
      type: 'post',
      content: { text: 'm' + i },
      keys: carolKeys,
    })
    carolMsgs.push(rec.msg)
  }
  t.pass('alice has msgs 1..10 from carol')

  const carolRootHash = alice.db.getFeedRoot(carolID, 'post')
  const carolRootMsg = alice.db.get(carolRootHash)

  await p(bob.db.add)(carolRootMsg, carolRootHash)
  for (let i = 0; i < 7; i++) {
    await p(bob.db.add)(carolMsgs[i], carolRootHash)
  }

  {
    const arr = [...bob.db.msgs()]
      .filter((msg) => msg.metadata.who === carolID_b58 && msg.content)
      .map((msg) => msg.content.text)
    t.deepEquals(
      arr,
      ['m1', 'm2', 'm3', 'm4', 'm5', 'm6', 'm7'],
      'bob has msgs 1..7 from carol'
    )
  }

  bob.tangleSync.setGoal(carolRootHash, 'all')
  alice.tangleSync.setGoal(carolRootHash, 'all')

  const remoteAlice = await p(bob.connect)(alice.getAddress())
  t.pass('bob connected to alice')

  bob.tangleSync.initiate()
  await p(setTimeout)(1000)
  t.pass('tangleSync!')

  {
    const arr = [...bob.db.msgs()]
      .filter((msg) => msg.metadata.who === carolID_b58 && msg.content)
      .map((msg) => msg.content.text)
    t.deepEquals(
      arr,
      ['m1', 'm2', 'm3', 'm4', 'm5', 'm6', 'm7', 'm8', 'm9', 'm10'],
      'bob has msgs 1..10 from carol'
    )
  }

  await p(remoteAlice.close)(true)
  await p(alice.close)(true)
  await p(bob.close)(true)
})

test('sync a feed with goal=newest', async (t) => {
  rimraf.sync(ALICE_DIR)
  rimraf.sync(BOB_DIR)

  const alice = createPeer({
    keys: aliceKeys,
    path: ALICE_DIR,
  })

  const bob = createPeer({
    keys: bobKeys,
    path: BOB_DIR,
  })

  await alice.db.loaded()
  await bob.db.loaded()

  const carolKeys = generateKeypair('carol')
  const carolMsgs = []
  const carolID = carolKeys.id
  const carolID_b58 = FeedV1.stripAuthor(carolID)
  for (let i = 1; i <= 10; i++) {
    const rec = await p(alice.db.create)({
      type: 'post',
      content: { text: 'm' + i },
      keys: carolKeys,
    })
    carolMsgs.push(rec.msg)
  }
  t.pass('alice has msgs 1..10 from carol')

  const carolRootHash = alice.db.getFeedRoot(carolID, 'post')
  const carolRootMsg = alice.db.get(carolRootHash)

  await p(bob.db.add)(carolRootMsg, carolRootHash)
  for (let i = 0; i < 7; i++) {
    await p(bob.db.add)(carolMsgs[i], carolRootHash)
  }

  {
    const arr = [...bob.db.msgs()]
      .filter((msg) => msg.metadata.who === carolID_b58 && msg.content)
      .map((msg) => msg.content.text)
    t.deepEquals(
      arr,
      ['m1', 'm2', 'm3', 'm4', 'm5', 'm6', 'm7'],
      'bob has msgs 1..7 from carol'
    )
  }

  bob.tangleSync.setGoal(carolRootHash, 'newest-5')
  alice.tangleSync.setGoal(carolRootHash, 'all')

  const remoteAlice = await p(bob.connect)(alice.getAddress())
  t.pass('bob connected to alice')

  bob.tangleSync.initiate()
  await p(setTimeout)(1000)
  t.pass('tangleSync!')

  {
    const arr = [...bob.db.msgs()]
      .filter((msg) => msg.metadata.who === carolID_b58 && msg.content)
      .map((msg) => msg.content.text)
    t.deepEquals(
      arr,
      ['m6', 'm7', 'm8', 'm9', 'm10'],
      'bob has msgs 6..10 from carol'
    )
  }

  await p(remoteAlice.close)(true)
  await p(alice.close)(true)
  await p(bob.close)(true)
})

test('sync a feed with goal=newest but too far behind', async (t) => {
  rimraf.sync(ALICE_DIR)
  rimraf.sync(BOB_DIR)

  const alice = createPeer({
    keys: aliceKeys,
    path: ALICE_DIR,
  })

  const bob = createPeer({
    keys: bobKeys,
    path: BOB_DIR,
  })

  await alice.db.loaded()
  await bob.db.loaded()

  const carolKeys = generateKeypair('carol')
  const carolMsgs = []
  const carolID = carolKeys.id
  const carolID_b58 = FeedV1.stripAuthor(carolID)
  for (let i = 1; i <= 10; i++) {
    const rec = await p(alice.db.create)({
      type: 'post',
      content: { text: 'm' + i },
      keys: carolKeys,
    })
    carolMsgs.push(rec.msg)
  }

  const carolRootHash = alice.db.getFeedRoot(carolID, 'post')
  const carolRootMsg = alice.db.get(carolRootHash)

  const algo = new Algorithm(alice)
  await algo.pruneNewest(carolRootHash, 5)
  {
    const arr = [...alice.db.msgs()]
      .filter((msg) => msg.metadata.who === carolID_b58 && msg.content)
      .map((msg) => msg.content.text)
    t.deepEquals(
      arr,
      ['m6', 'm7', 'm8', 'm9', 'm10'],
      'alice has msgs 6..10 from carol'
    )
  }

  await p(bob.db.add)(carolRootMsg, carolRootHash)
  for (let i = 0; i < 2; i++) {
    await p(bob.db.add)(carolMsgs[i], carolRootHash)
  }

  {
    const arr = [...bob.db.msgs()]
      .filter((msg) => msg.metadata.who === carolID_b58 && msg.content)
      .map((msg) => msg.content.text)
    t.deepEquals(arr, ['m1', 'm2'], 'bob has msgs 1..2 from carol')
  }

  alice.tangleSync.setFeedGoal(carolID, 'post', 'newest-5')
  bob.tangleSync.setFeedGoal(carolID, 'post', 'newest-5')

  const remoteAlice = await p(bob.connect)(alice.getAddress())
  t.pass('bob connected to alice')

  bob.tangleSync.initiate()
  await p(setTimeout)(1000)
  t.pass('tangleSync!')

  {
    const arr = [...bob.db.msgs()]
      .filter((msg) => msg.metadata.who === carolID_b58 && msg.content)
      .map((msg) => msg.content.text)
    t.deepEquals(
      arr,
      ['m6', 'm7', 'm8', 'm9', 'm10'],
      'bob has msgs 6..10 from carol'
    )
  }

  await p(remoteAlice.close)(true)
  await p(alice.close)(true)
  await p(bob.close)(true)
})