const fixtures = require('webtorrent-fixtures')
const fs = require('fs')
const http = require('http')
const test = require('tape')
const WebTorrent = require('../../')
const zlib = require('zlib')

function assertBlocked (t, torrent, addr) {
  torrent.once('blockedPeer', function (_addr) {
    t.equal(addr, _addr)
  })
  t.notOk(torrent.addPeer(addr))
}

function assertReachable (t, torrent, addr) {
  torrent.once('peer', function (_addr) {
    t.equal(addr, _addr)
  })
  t.ok(torrent.addPeer(addr))
}

test('blocklist (single IP)', function (t) {
  t.plan(9)

  const client = new WebTorrent({
    dht: false,
    tracker: false,
    blocklist: ['1.2.3.4']
  })
  client.on('error', function (err) { t.fail(err) })
  client.on('warning', function (err) { t.fail(err) })

  // blocklist isn't fully loaded until `ready` event
  client.on('ready', function () {
    client.add(fixtures.leaves.parsedTorrent, function (torrent) {
      assertBlocked(t, torrent, '1.2.3.4:1234')
      assertBlocked(t, torrent, '1.2.3.4:6969')
      assertReachable(t, torrent, '1.1.1.1:1234')
      assertReachable(t, torrent, '1.1.1.1:6969')

      client.destroy(function (err) {
        t.error(err, 'client destroyed')
      })
    })
  })
})

test('blocklist (array of IPs)', function (t) {
  t.plan(13)

  const client = new WebTorrent({
    dht: false,
    tracker: false,
    blocklist: ['1.2.3.4', '5.6.7.8']
  })
    .on('error', function (err) { t.fail(err) })
    .on('warning', function (err) { t.fail(err) })
    .on('ready', function () {
      client.add(fixtures.leaves.parsedTorrent, function (torrent) {
        assertBlocked(t, torrent, '1.2.3.4:1234')
        assertBlocked(t, torrent, '1.2.3.4:6969')
        assertBlocked(t, torrent, '5.6.7.8:1234')
        assertBlocked(t, torrent, '5.6.7.8:6969')
        assertReachable(t, torrent, '1.1.1.1:1234')
        assertReachable(t, torrent, '1.1.1.1:6969')

        client.destroy(function (err) {
          t.error(err, 'client destroyed')
        })
      })
    })
})

// 48 asserts
function assertList (t, torrent) {
  assertBlocked(t, torrent, '1.2.3.0:1234')
  assertBlocked(t, torrent, '1.2.3.0:6969')

  assertBlocked(t, torrent, '1.2.3.1:1234')
  assertBlocked(t, torrent, '1.2.3.1:6969')

  assertBlocked(t, torrent, '1.2.3.1:1234')
  assertBlocked(t, torrent, '1.2.3.1:6969')

  assertBlocked(t, torrent, '1.2.3.254:1234')
  assertBlocked(t, torrent, '1.2.3.254:6969')

  assertBlocked(t, torrent, '1.2.3.255:1234')
  assertBlocked(t, torrent, '1.2.3.255:6969')

  assertBlocked(t, torrent, '5.6.7.0:1234')
  assertBlocked(t, torrent, '5.6.7.0:6969')

  assertBlocked(t, torrent, '5.6.7.128:1234')
  assertBlocked(t, torrent, '5.6.7.128:6969')

  assertBlocked(t, torrent, '5.6.7.255:1234')
  assertBlocked(t, torrent, '5.6.7.255:6969')

  assertReachable(t, torrent, '1.1.1.1:1234')
  assertReachable(t, torrent, '1.1.1.1:6969')

  assertReachable(t, torrent, '2.2.2.2:1234')
  assertReachable(t, torrent, '2.2.2.2:6969')

  assertReachable(t, torrent, '1.2.4.0:1234')
  assertReachable(t, torrent, '1.2.4.0:6969')

  assertReachable(t, torrent, '1.2.2.0:1234')
  assertReachable(t, torrent, '1.2.2.0:6969')
}

test('blocklist (array of IP ranges)', function (t) {
  t.plan(49)
  const client = new WebTorrent({
    dht: false,
    tracker: false,
    blocklist: [
      { start: '1.2.3.0', end: '1.2.3.255' },
      { start: '5.6.7.0', end: '5.6.7.255' }
    ]
  })
    .on('error', function (err) { t.fail(err) })
    .on('warning', function (err) { t.fail(err) })
    .on('ready', function () {
      client.add(fixtures.leaves.parsedTorrent, function (torrent) {
        assertList(t, torrent)
        client.destroy(function (err) {
          t.error(err, 'client destroyed')
        })
      })
    })
})

test('blocklist (http url)', function (t) {
  t.plan(51)
  const server = http.createServer(function (req, res) {
    // Check that WebTorrent declares a user agent
    t.ok(req.headers['user-agent'].indexOf('WebTorrent') !== -1)

    fs.createReadStream(fixtures.blocklist.path).pipe(res)
  })

  server.listen(0, function () {
    const port = server.address().port
    const url = 'http://127.0.0.1:' + port
    const client = new WebTorrent({
      dht: false,
      tracker: false,
      blocklist: url
    })
      .on('error', function (err) { t.fail(err) })
      .on('warning', function (err) { t.fail(err) })
      .on('ready', function () {
        client.add(fixtures.leaves.parsedTorrent, function (torrent) {
          assertList(t, torrent)
          client.destroy(function (err) {
            t.error(err, 'client destroyed')
          })
          server.close(function () {
            t.pass('server closed')
          })
        })
      })
  })
})

test('blocklist (http url with gzip encoding)', function (t) {
  t.plan(51)
  const server = http.createServer(function (req, res) {
    // Check that WebTorrent declares a user agent
    t.ok(req.headers['user-agent'].indexOf('WebTorrent') !== -1)

    res.setHeader('content-encoding', 'gzip')
    fs.createReadStream(fixtures.blocklist.path)
      .pipe(zlib.createGzip())
      .pipe(res)
  })

  server.listen(0, function () {
    const port = server.address().port
    const url = 'http://127.0.0.1:' + port
    const client = new WebTorrent({
      dht: false,
      tracker: false,
      blocklist: url
    })
      .on('error', function (err) { t.fail(err) })
      .on('warning', function (err) { t.fail(err) })
      .on('ready', function () {
        client.add(fixtures.leaves.parsedTorrent, function (torrent) {
          assertList(t, torrent)
          client.destroy(function (err) {
            t.error(err, 'client destroyed')
          })
          server.close(function () {
            t.pass('server closed')
          })
        })
      })
  })
})

test('blocklist (http url with deflate encoding)', function (t) {
  t.plan(51)
  const server = http.createServer(function (req, res) {
    // Check that WebTorrent declares a user agent
    t.ok(req.headers['user-agent'].indexOf('WebTorrent') !== -1)

    res.setHeader('content-encoding', 'deflate')
    fs.createReadStream(fixtures.blocklist.path)
      .pipe(zlib.createDeflate())
      .pipe(res)
  })

  server.listen(0, function () {
    const port = server.address().port
    const url = 'http://127.0.0.1:' + port
    const client = new WebTorrent({
      dht: false,
      tracker: false,
      blocklist: url
    })
      .on('error', function (err) { t.fail(err) })
      .on('warning', function (err) { t.fail(err) })
      .on('ready', function () {
        client.add(fixtures.leaves.parsedTorrent, function (torrent) {
          assertList(t, torrent)
          client.destroy(function (err) {
            t.error(err, 'client destroyed')
          })
          server.close(function () {
            t.pass('server closed')
          })
        })
      })
  })
})

test('blocklist (fs path)', function (t) {
  t.plan(49)
  const client = new WebTorrent({
    dht: false,
    tracker: false,
    blocklist: fixtures.blocklist.path
  })
    .on('error', function (err) { t.fail(err) })
    .on('warning', function (err) { t.fail(err) })
    .on('ready', function () {
      client.add(fixtures.leaves.parsedTorrent, function (torrent) {
        assertList(t, torrent)
        client.destroy(function (err) {
          t.error(err, 'client destroyed')
        })
      })
    })
})

test('blocklist (fs path with gzip)', function (t) {
  t.plan(49)
  const client = new WebTorrent({
    dht: false,
    tracker: false,
    blocklist: fixtures.blocklist.gzipPath
  })
    .on('error', function (err) { t.fail(err) })
    .on('warning', function (err) { t.fail(err) })
    .on('ready', function () {
      client.add(fixtures.leaves.parsedTorrent, function (torrent) {
        assertList(t, torrent)
        client.destroy(function (err) {
          t.error(err, 'client destroyed')
        })
      })
    })
})