'use strict'

const { Buffer } = require('buffer')
const textEncoder = new TextEncoder()

exports.all = function (test, testCommon) {
  if (!testCommon.supports.encodings.buffer) return

  // NOTE: adapted from levelup
  test('test put() and get() with buffer value and buffer valueEncoding', async function (t) {
    const db = testCommon.factory()
    await db.open()
    await db.put('test', testBuffer(), { valueEncoding: 'buffer' })
    t.same(await db.get('test', { valueEncoding: 'buffer' }), testBuffer())
    return db.close()
  })

  // NOTE: adapted from levelup
  test('test put() and get() with buffer value and buffer valueEncoding in factory', async function (t) {
    const db = testCommon.factory({ valueEncoding: 'buffer' })
    await db.open()
    await db.put('test', testBuffer())
    t.same(await db.get('test'), testBuffer())
    return db.close()
  })

  // NOTE: adapted from levelup
  test('test put() and get() with buffer key and buffer keyEncoding', async function (t) {
    const db = testCommon.factory()
    await db.open()
    await db.put(testBuffer(), 'test', { keyEncoding: 'buffer' })
    t.same(await db.get(testBuffer(), { keyEncoding: 'buffer' }), 'test')
    return db.close()
  })

  // NOTE: adapted from levelup
  test('test put() and get() with buffer key and utf8 keyEncoding', async function (t) {
    const db = testCommon.factory()
    await db.open()
    await db.put(Buffer.from('foo🐄'), 'test', { keyEncoding: 'utf8' })
    t.same(await db.get(Buffer.from('foo🐄'), { keyEncoding: 'utf8' }), 'test')
    return db.close()
  })

  // NOTE: adapted from levelup
  test('test put() and get() with string value and buffer valueEncoding', async function (t) {
    const db = testCommon.factory()
    await db.open()
    await db.put('test', 'foo🐄', { valueEncoding: 'buffer' })
    t.same(await db.get('test', { valueEncoding: 'buffer' }), Buffer.from('foo🐄'))
    t.same(await db.get('test', { valueEncoding: 'utf8' }), 'foo🐄')
    return db.close()
  })

  // NOTE: adapted from memdown
  test('put() as string, get() as buffer and vice versa', async function (t) {
    const db = testCommon.factory()
    await db.open()
    const enc = { keyEncoding: 'buffer', valueEncoding: 'buffer' }
    const [a, b] = ['🐄', '🐄 says moo']

    const promise1 = db.put(a, a).then(async () => {
      const value = await db.get(Buffer.from(a), enc)
      t.same(value, Buffer.from(a), 'got buffer value')
    })

    const promise2 = db.put(Buffer.from(b), Buffer.from(b), enc).then(async () => {
      const value = await db.get(b)
      t.same(value, b, 'got string value')
    })

    await Promise.all([promise1, promise2])
    return db.close()
  })

  // NOTE: adapted from memdown
  test('put() stringifies input to buffer', async function (t) {
    const db = testCommon.factory()
    await db.open()
    await db.put(1, 2)

    const it = db.iterator({ keyEncoding: 'buffer', valueEncoding: 'buffer' })
    const entries = await it.all()

    t.same(entries[0][0], Buffer.from('1'), 'key was stringified')
    t.same(entries[0][1], Buffer.from('2'), 'value was stringified')

    return db.close()
  })

  // NOTE: adapted from memdown
  test('put() as string, iterate as buffer', async function (t) {
    const db = testCommon.factory({ keyEncoding: 'utf8', valueEncoding: 'utf8' })
    await db.open()
    await db.put('🐄', '🐄')

    const it = db.iterator({ keyEncoding: 'buffer', valueEncoding: 'buffer' })
    const entries = await it.all()

    t.same(entries, [[Buffer.from('🐄'), Buffer.from('🐄')]])
    return db.close()
  })

  // NOTE: adapted from memdown
  test('put() as buffer, iterate as string', async function (t) {
    const db = testCommon.factory({ keyEncoding: 'buffer', valueEncoding: 'buffer' })
    await db.open()
    await db.put(Buffer.from('🐄'), Buffer.from('🐄'))

    const it = db.iterator({ keyEncoding: 'utf8', valueEncoding: 'utf8' })
    const entries = await it.all()

    t.same(entries, [['🐄', '🐄']])
    return db.close()
  })

  test('put() as view, iterate as view', async function (t) {
    const db = testCommon.factory({ keyEncoding: 'view', valueEncoding: 'view' })
    const cow = textEncoder.encode('🐄')
    await db.open()
    await db.put(cow, cow)

    const it = db.iterator()
    const entries = await it.all()
    const key = Buffer.isBuffer(entries[0][0]) ? Buffer.from(cow) : cow // Valid, Buffer is a Uint8Array
    const value = Buffer.isBuffer(entries[0][1]) ? Buffer.from(cow) : cow

    t.same(entries, [[key, value]])
    return db.close()
  })

  test('put() as string, iterate as view', async function (t) {
    const db = testCommon.factory({ keyEncoding: 'utf8', valueEncoding: 'utf8' })
    const cow = textEncoder.encode('🐄')
    await db.open()
    await db.put('🐄', '🐄')

    const it = db.iterator({ keyEncoding: 'view', valueEncoding: 'view' })
    const entries = await it.all()
    const key = Buffer.isBuffer(entries[0][0]) ? Buffer.from(cow) : cow // Valid, Buffer is a Uint8Array
    const value = Buffer.isBuffer(entries[0][1]) ? Buffer.from(cow) : cow

    t.same(entries, [[key, value]])
    return db.close()
  })

  test('put() as view, iterate as string', async function (t) {
    const db = testCommon.factory({ keyEncoding: 'view', valueEncoding: 'view' })
    const cow = textEncoder.encode('🐄')
    await db.open()
    await db.put(cow, cow)

    const it = db.iterator({ keyEncoding: 'utf8', valueEncoding: 'utf8' })
    const entries = await it.all()

    t.same(entries, [['🐄', '🐄']])
    return db.close()
  })

  // NOTE: adapted from levelup
  test('batch() with multiple puts with buffer valueEncoding per batch', async function (t) {
    const db = testCommon.factory()
    await db.open()
    await db.batch([
      { type: 'put', key: 'foo', value: testBuffer() },
      { type: 'put', key: 'bar', value: testBuffer() },
      { type: 'put', key: 'baz', value: 'abazvalue' }
    ], { valueEncoding: 'buffer' })

    t.same(await db.get('foo', { valueEncoding: 'buffer' }), testBuffer())
    t.same(await db.get('bar', { valueEncoding: 'buffer' }), testBuffer())
    t.same(await db.get('baz', { valueEncoding: 'buffer' }), Buffer.from('abazvalue'))

    return db.close()
  })

  test('batch() with multiple puts with buffer valueEncoding per operation', async function (t) {
    const db = testCommon.factory()
    await db.open()
    await db.batch([
      { type: 'put', key: 'foo', value: testBuffer(), valueEncoding: 'buffer' },
      { type: 'put', key: 'bar', value: testBuffer(), valueEncoding: 'buffer' },
      { type: 'put', key: 'baz', value: 'abazvalue', valueEncoding: 'buffer' }
    ])

    t.same(await db.get('foo', { valueEncoding: 'buffer' }), testBuffer())
    t.same(await db.get('bar', { valueEncoding: 'buffer' }), testBuffer())
    t.same(await db.get('baz', { valueEncoding: 'buffer' }), Buffer.from('abazvalue'))

    return db.close()
  })

  // NOTE: adapted from encoding-down
  test('batch() with buffer encoding in factory', async function (t) {
    const operations = [{
      type: 'put',
      key: Buffer.from([1, 2, 3]),
      value: Buffer.from([4, 5, 6])
    }, {
      type: 'put',
      key: Buffer.from([7, 8, 9]),
      value: Buffer.from([10, 11, 12])
    }]

    const db = testCommon.factory({ keyEncoding: 'buffer', valueEncoding: 'buffer' })
    await db.open()
    await db.batch(operations)

    t.same(await db.get(operations[0].key), operations[0].value)
    t.same(await db.get(operations[1].key), operations[1].value)

    return db.close()
  })

  for (const keyEncoding of ['buffer', 'view']) {
    // NOTE: adapted from memdown
    test(`storage is byte-aware (${keyEncoding} encoding)`, function (t) {
      const db = testCommon.factory({ keyEncoding })

      db.open(function (err) {
        t.ifError(err, 'no open error')

        const one = Buffer.from('80', 'hex')
        const two = Buffer.from('c0', 'hex')

        t.ok(two.toString() === one.toString(), 'would be equal when not byte-aware')
        t.ok(two.compare(one) > 0, 'but greater when byte-aware')

        db.put(one, 'one', function (err) {
          t.ifError(err, 'no put() error')

          db.get(one, function (err, value) {
            t.ifError(err, 'no get() error')
            t.is(value, 'one', 'value one ok')

            db.put(two, 'two', function (err) {
              t.ifError(err, 'no put() error')

              db.get(one, function (err, value) {
                t.ifError(err, 'no get() error')
                t.is(value, 'one', 'value one did not change')

                db.close(t.end.bind(t))
              })
            })
          })
        })
      })
    })
  }
}

function testBuffer () {
  return Buffer.from('0080c0ff', 'hex')
}