import { equal } from 'node:assert/strict'
import { once } from 'node:events'
import { createServer } from 'node:http'

import { Client, RetryHandler } from 'undici'

let requestCount = 0
const chunks = []

const server = createServer({ joinDuplicateHeaders: true }, (req, res) => {
  requestCount++

  if (requestCount === 1) {
    equal(req.headers.range, 'bytes=0-3')
    res.setHeader('etag', '"same"')
    res.write('abc')
    setTimeout(() => res.destroy(), 50)
    return
  }

  equal(req.headers.range, 'bytes=3-')
  res.writeHead(206, {
    etag: '"same"',
    'content-range': 'bytes 3-5/*'
  })
  res.end('def')
})

server.listen(0)
await once(server, 'listening')

const client = new Client(`http://localhost:${server.address().port}`)

try {
  const { promise, resolve, reject } = Promise.withResolvers();
  const handler = new RetryHandler({
    method: 'GET',
    path: '/',
    headers: {},
    retryOptions: {
      minTimeout: 1,
      retry: (err, _ctx, done) => {
        if (err.message === 'Content-Range mismatch') {
          done(err)
          return
        }

        done(null)
      }
    }
  }, {
    dispatch: client.dispatch.bind(client),
    handler: {
      onResponseStart () {
        return true
      },
      onResponseData (_controller, chunk) {
        chunks.push(chunk)
        return true
      },
      onResponseEnd () {
        resolve()
      },
      onResponseError (_controller, err) {
        reject(err)
      }
    }
  })

  client.dispatch({
    method: 'GET',
    path: '/',
    headers: {
      range: 'bytes=0-3'
    }
  }, handler)

  await promise
  equal(Buffer.concat(chunks).toString(), 'abcdef')
} finally {
  await client.close()
  server.close()
  await once(server, 'close')
}