import { Query as GherkinQuery } from '@cucumber/gherkin-utils'
import { Readable, Writable } from 'stream'

import GherkinQueryStream from './GherkinQueryStream'
import makeTestCase from './makeTestCase'
import makeTestPlan from './makeTestPlan'
import SupportCode from './SupportCode'
import { MakeTestPlan, RunOptions } from './types'

const DEFAULT_OPTIONS: RunOptions = {
  allowedRetries: 0,
}

export default async function runCucumber(
  supportCode: SupportCode,
  gherkinEnvelopeStream: Readable,
  gherkinQuery: GherkinQuery,
  envelopeOutputStream: Writable,
  runOptions: RunOptions = DEFAULT_OPTIONS,
  makeTestPlanFn: MakeTestPlan<SupportCode> = makeTestPlan
) {
  const gherkinQueryStream = new GherkinQueryStream(gherkinQuery)
  gherkinEnvelopeStream
    .pipe(gherkinQueryStream)
    .pipe(envelopeOutputStream, { end: false })

  await new Promise((resolve, reject) => {
    gherkinQueryStream.on('end', resolve)
    gherkinQueryStream.on('error', reject)
    gherkinEnvelopeStream.on('error', reject)
  })

  const testPlan = makeTestPlanFn(
    gherkinQuery,
    supportCode,
    runOptions,
    makeTestCase
  )
  await testPlan.execute((envelope) => {
    envelopeOutputStream.write(envelope)
    if (envelope.testRunFinished) {
      envelopeOutputStream.end()
    }
  })
}