import { createClient } from '@clickhouse/client'
import { isProgressRow } from '@clickhouse/client-common'

type Data = { number: string }

void (async () => {
  const client = createClient()
  const rs = await client.query({
    query: 'SELECT number FROM system.numbers LIMIT 100',
    format: 'JSONEachRowWithProgress',
  })

  let totalRows = 0
  let totalProgressRows = 0

  const stream = rs.stream<Data>()
  for await (const rows of stream) {
    for (const row of rows) {
      const decodedRow = row.json()
      if (isProgressRow(decodedRow)) {
        console.log('Got progress:', decodedRow.progress)
        totalProgressRows++
      } else {
        totalRows++
        if (totalRows % 100 === 0) {
          // Bug is here! Types say "row" is always defined, but it can now be undefined
          console.log('Sample row:', decodedRow.row)
        }
      }
    }
  }

  console.log('Total rows:', totalRows)
  console.log('Total progress rows:', totalProgressRows)

  await client.close()
})()