'use strict'

const Benchmark = require('benchmark')
const pull = require('pull-stream')
const pair = require('pull-pair/duplex')
const PeerId = require('peer-id')
const crypto = require('libp2p-crypto')

const secio = require('../src')

const suite = new Benchmark.Suite('secio')
const ids = []

suite.add('createKey', function (d) {
  crypto.keys.generateKeyPair('RSA', 2048, (err, key) => {
    if (err) { throw err }
    PeerId.createFromPrivKey(key.bytes, (err, id) => {
      if (err) { throw err }

      ids.push(id)
      d.resolve()
    })
  })
}, {
  defer: true
})
.add('send', function (deferred) {
  const p = pair()

  createSession(p[0], (err, local) => {
    if (err) { throw err }
    createSession(p[1], (err, remote) => {
      if (err) { throw err }

      sendMessages(local, remote)
    })
  })

  function sendMessages (local, remote) {
    pull(
      pull.infinite(),
      pull.take(100),
      pull.map((val) => Buffer.from(val.toString())),
      local
    )

    pull(
      remote,
      pull.take(100),
      pull.collect((err, chunks) => {
        if (err) throw err
        if (chunks.length !== 100) throw new Error('Did not receive enough chunks')
        deferred.resolve()
      })
    )
  }
}, {
  defer: true
})
.on('cycle', (event) => {
  console.log(String(event.target))
})
// run async
.run({
  async: true
})

function createSession (insecure, cb) {
  crypto.keys.generateKeyPair('RSA', 2048, (err, key) => {
    if (err) { return cb(err) }

    PeerId.createFromPrivKey(key.bytes, (err, id) => {
      if (err) { return cb(err) }

      cb(null, secio.encrypt(id, key, insecure))
    })
  })
}