import CramCodec, { Cursors } from './_base.ts'
import { ByteArrayLengthEncoding, CramEncoding } from '../encoding.ts'
import { CramFileBlock } from '../file.ts'
import { DataType } from './dataSeriesTypes.ts'
import CramSlice from '../slice/index.ts'
import { tinyMemoize } from '../util.ts'

type CramCodecFactory = <TData extends DataType = DataType>(
  encodingData: CramEncoding,
  dataType: TData | 'ignore',
) => CramCodec<TData>

export default class ByteArrayStopCodec extends CramCodec<
  'byteArray',
  ByteArrayLengthEncoding['parameters']
> {
  private instantiateCodec: CramCodecFactory

  constructor(
    parameters: ByteArrayLengthEncoding['parameters'],
    dataType: 'byteArray',
    instantiateCodec: CramCodecFactory,
  ) {
    super(parameters, dataType)
    this.instantiateCodec = instantiateCodec
  }

  decode(
    slice: CramSlice,
    coreDataBlock: CramFileBlock,
    blocksByContentId: Record<number, CramFileBlock>,
    cursors: Cursors,
  ) {
    const lengthCodec = this._getLengthCodec()
    const arrayLength =
      lengthCodec.decode(slice, coreDataBlock, blocksByContentId, cursors) || 0

    if (arrayLength > 0) {
      const dataCodec = this._getDataCodec()
      const subarray = dataCodec.getBytesSubarray(
        blocksByContentId,
        cursors,
        arrayLength,
      )
      if (subarray) {
        return subarray
      } else {
        const data = new Uint8Array(arrayLength)
        for (let i = 0; i < arrayLength; i += 1) {
          data[i] =
            dataCodec.decode(
              slice,
              coreDataBlock,
              blocksByContentId,
              cursors,
            ) || 0
        }
        return data
      }
    } else {
      return new Uint8Array(0)
    }
  }

  // memoize
  _getLengthCodec() {
    const encodingParams = this.parameters.lengthsEncoding
    return this.instantiateCodec(encodingParams, 'int')
  }

  // memoize
  _getDataCodec() {
    const encodingParams = this.parameters.valuesEncoding
    return this.instantiateCodec(encodingParams, 'byte')
  }
}

'_getLengthCodec _getDataCodec'.split(' ').forEach(method => {
  tinyMemoize(ByteArrayStopCodec, method)
})