// Copyright (c) 2026 Cloudflare, Inc.
// Licensed under the Apache 2.0 license found in the LICENSE file or at:
//     https://opensource.org/licenses/Apache-2.0

import { owner_symbol, type Zlib } from 'node-internal:internal_zlib_base';

type InternalCompressCallback = (res: Error | ArrayBuffer) => void;

export function crc32(data: ArrayBufferView | string, value: number): number;

export function zlibSync(
  data: ArrayBufferView | string,
  options: ZlibOptions,
  mode: number
): ArrayBuffer;
export function zlib(
  data: ArrayBufferView | string,
  options: ZlibOptions,
  mode: number,
  cb: InternalCompressCallback
): void;

export function brotliDecompressSync(
  data: ArrayBufferView | string,
  options: BrotliOptions
): ArrayBuffer;
export function brotliDecompress(
  data: ArrayBufferView | string,
  options: BrotliOptions,
  cb: InternalCompressCallback
): void;

export function brotliCompressSync(
  data: ArrayBufferView | string,
  options: BrotliOptions
): ArrayBuffer;
export function brotliCompress(
  data: ArrayBufferView | string,
  options: BrotliOptions,
  cb: InternalCompressCallback
): void;

export function zstdDecompressSync(
  data: ArrayBufferView | string,
  options: ZstdOptions
): ArrayBuffer;
export function zstdDecompress(
  data: ArrayBufferView | string,
  options: ZstdOptions,
  cb: InternalCompressCallback
): void;

export function zstdCompressSync(
  data: ArrayBufferView | string,
  options: ZstdOptions
): ArrayBuffer;
export function zstdCompress(
  data: ArrayBufferView | string,
  options: ZstdOptions,
  cb: InternalCompressCallback
): void;

// zlib.constants (part of the API contract for node:zlib)
export const CONST_Z_NO_FLUSH: number;
export const CONST_Z_PARTIAL_FLUSH: number;
export const CONST_Z_SYNC_FLUSH: number;
export const CONST_Z_FULL_FLUSH: number;
export const CONST_Z_FINISH: number;
export const CONST_Z_BLOCK: number;

export const CONST_Z_OK: number;
export const CONST_Z_STREAM_END: number;
export const CONST_Z_NEED_DICT: number;
export const CONST_Z_ERRNO: number;
export const CONST_Z_STREAM_ERROR: number;
export const CONST_Z_DATA_ERROR: number;
export const CONST_Z_MEM_ERROR: number;
export const CONST_Z_BUF_ERROR: number;
export const CONST_Z_VERSION_ERROR: number;

export const CONST_Z_NO_COMPRESSION: number;
export const CONST_Z_BEST_SPEED: number;
export const CONST_Z_BEST_COMPRESSION: number;
export const CONST_Z_DEFAULT_COMPRESSION: number;
export const CONST_Z_FILTERED: number;
export const CONST_Z_HUFFMAN_ONLY: number;
export const CONST_Z_RLE: number;
export const CONST_Z_FIXED: number;
export const CONST_Z_DEFAULT_STRATEGY: number;
export const CONST_ZLIB_VERNUM: number;

export const CONST_DEFLATE: number;
export const CONST_INFLATE: number;
export const CONST_GZIP: number;
export const CONST_GUNZIP: number;
export const CONST_DEFLATERAW: number;
export const CONST_INFLATERAW: number;
export const CONST_UNZIP: number;
export const CONST_BROTLI_DECODE: number;
export const CONST_BROTLI_ENCODE: number;
export const CONST_ZSTD_ENCODE: number;
export const CONST_ZSTD_DECODE: number;

export const CONST_Z_MIN_WINDOWBITS: number;
export const CONST_Z_MAX_WINDOWBITS: number;
export const CONST_Z_DEFAULT_WINDOWBITS: number;
export const CONST_Z_MIN_CHUNK: number;
export const CONST_Z_MAX_CHUNK: number;
export const CONST_Z_DEFAULT_CHUNK: number;
export const CONST_Z_MIN_MEMLEVEL: number;
export const CONST_Z_MAX_MEMLEVEL: number;
export const CONST_Z_DEFAULT_MEMLEVEL: number;
export const CONST_Z_MIN_LEVEL: number;
export const CONST_Z_MAX_LEVEL: number;
export const CONST_Z_DEFAULT_LEVEL: number;

export const CONST_BROTLI_OPERATION_PROCESS: number;
export const CONST_BROTLI_OPERATION_FLUSH: number;
export const CONST_BROTLI_OPERATION_FINISH: number;
export const CONST_BROTLI_OPERATION_EMIT_METADATA: number;
export const CONST_BROTLI_PARAM_MODE: number;
export const CONST_BROTLI_MODE_GENERIC: number;
export const CONST_BROTLI_MODE_TEXT: number;
export const CONST_BROTLI_MODE_FONT: number;
export const CONST_BROTLI_DEFAULT_MODE: number;
export const CONST_BROTLI_PARAM_QUALITY: number;
export const CONST_BROTLI_MIN_QUALITY: number;
export const CONST_BROTLI_MAX_QUALITY: number;
export const CONST_BROTLI_DEFAULT_QUALITY: number;
export const CONST_BROTLI_PARAM_LGWIN: number;
export const CONST_BROTLI_MIN_WINDOW_BITS: number;
export const CONST_BROTLI_MAX_WINDOW_BITS: number;
export const CONST_BROTLI_LARGE_MAX_WINDOW_BITS: number;
export const CONST_BROTLI_DEFAULT_WINDOW: number;
export const CONST_BROTLI_PARAM_LGBLOCK: number;
export const CONST_BROTLI_MIN_INPUT_BLOCK_BITS: number;
export const CONST_BROTLI_MAX_INPUT_BLOCK_BITS: number;
export const CONST_BROTLI_PARAM_DISABLE_LITERAL_CONTEXT_MODELING: number;
export const CONST_BROTLI_PARAM_SIZE_HINT: number;
export const CONST_BROTLI_PARAM_LARGE_WINDOW: number;
export const CONST_BROTLI_PARAM_NPOSTFIX: number;
export const CONST_BROTLI_PARAM_NDIRECT: number;
export const CONST_BROTLI_DECODER_RESULT_ERROR: number;
export const CONST_BROTLI_DECODER_RESULT_SUCCESS: number;
export const CONST_BROTLI_DECODER_RESULT_NEEDS_MORE_INPUT: number;
export const CONST_BROTLI_DECODER_RESULT_NEEDS_MORE_OUTPUT: number;
export const CONST_BROTLI_DECODER_PARAM_DISABLE_RING_BUFFER_REALLOCATION: number;
export const CONST_BROTLI_DECODER_PARAM_LARGE_WINDOW: number;
export const CONST_BROTLI_DECODER_NO_ERROR: number;
export const CONST_BROTLI_DECODER_SUCCESS: number;
export const CONST_BROTLI_DECODER_NEEDS_MORE_INPUT: number;
export const CONST_BROTLI_DECODER_NEEDS_MORE_OUTPUT: number;
export const CONST_BROTLI_DECODER_ERROR_FORMAT_EXUBERANT_NIBBLE: number;
export const CONST_BROTLI_DECODER_ERROR_FORMAT_RESERVED: number;
export const CONST_BROTLI_DECODER_ERROR_FORMAT_EXUBERANT_META_NIBBLE: number;
export const CONST_BROTLI_DECODER_ERROR_FORMAT_SIMPLE_HUFFMAN_ALPHABET: number;
export const CONST_BROTLI_DECODER_ERROR_FORMAT_SIMPLE_HUFFMAN_SAME: number;
export const CONST_BROTLI_DECODER_ERROR_FORMAT_CL_SPACE: number;
export const CONST_BROTLI_DECODER_ERROR_FORMAT_HUFFMAN_SPACE: number;
export const CONST_BROTLI_DECODER_ERROR_FORMAT_CONTEXT_MAP_REPEAT: number;
export const CONST_BROTLI_DECODER_ERROR_FORMAT_BLOCK_LENGTH_1: number;
export const CONST_BROTLI_DECODER_ERROR_FORMAT_BLOCK_LENGTH_2: number;
export const CONST_BROTLI_DECODER_ERROR_FORMAT_TRANSFORM: number;
export const CONST_BROTLI_DECODER_ERROR_FORMAT_DICTIONARY: number;
export const CONST_BROTLI_DECODER_ERROR_FORMAT_WINDOW_BITS: number;
export const CONST_BROTLI_DECODER_ERROR_FORMAT_PADDING_1: number;
export const CONST_BROTLI_DECODER_ERROR_FORMAT_PADDING_2: number;
export const CONST_BROTLI_DECODER_ERROR_FORMAT_DISTANCE: number;
export const CONST_BROTLI_DECODER_ERROR_DICTIONARY_NOT_SET: number;
export const CONST_BROTLI_DECODER_ERROR_INVALID_ARGUMENTS: number;
export const CONST_BROTLI_DECODER_ERROR_ALLOC_CONTEXT_MODES: number;
export const CONST_BROTLI_DECODER_ERROR_ALLOC_TREE_GROUPS: number;
export const CONST_BROTLI_DECODER_ERROR_ALLOC_CONTEXT_MAP: number;
export const CONST_BROTLI_DECODER_ERROR_ALLOC_RING_BUFFER_1: number;
export const CONST_BROTLI_DECODER_ERROR_ALLOC_RING_BUFFER_2: number;
export const CONST_BROTLI_DECODER_ERROR_ALLOC_BLOCK_TYPE_TREES: number;
export const CONST_BROTLI_DECODER_ERROR_UNREACHABLE: number;

// Zstd mode aliases (Node.js naming)
export const CONST_ZSTD_COMPRESS: number;
export const CONST_ZSTD_DECOMPRESS: number;

// Zstd flush directives
export const CONST_ZSTD_e_continue: number;
export const CONST_ZSTD_e_flush: number;
export const CONST_ZSTD_e_end: number;

// Zstd compression parameters
export const CONST_ZSTD_c_compressionLevel: number;
export const CONST_ZSTD_c_windowLog: number;
export const CONST_ZSTD_c_hashLog: number;
export const CONST_ZSTD_c_chainLog: number;
export const CONST_ZSTD_c_searchLog: number;
export const CONST_ZSTD_c_minMatch: number;
export const CONST_ZSTD_c_targetLength: number;
export const CONST_ZSTD_c_strategy: number;
export const CONST_ZSTD_c_enableLongDistanceMatching: number;
export const CONST_ZSTD_c_ldmHashLog: number;
export const CONST_ZSTD_c_ldmMinMatch: number;
export const CONST_ZSTD_c_ldmBucketSizeLog: number;
export const CONST_ZSTD_c_ldmHashRateLog: number;
export const CONST_ZSTD_c_contentSizeFlag: number;
export const CONST_ZSTD_c_checksumFlag: number;
export const CONST_ZSTD_c_dictIDFlag: number;
export const CONST_ZSTD_c_nbWorkers: number;
export const CONST_ZSTD_c_jobSize: number;
export const CONST_ZSTD_c_overlapLog: number;

// Zstd decompression parameters
export const CONST_ZSTD_d_windowLogMax: number;

// Zstd strategy constants
export const CONST_ZSTD_fast: number;
export const CONST_ZSTD_dfast: number;
export const CONST_ZSTD_greedy: number;
export const CONST_ZSTD_lazy: number;
export const CONST_ZSTD_lazy2: number;
export const CONST_ZSTD_btlazy2: number;
export const CONST_ZSTD_btopt: number;
export const CONST_ZSTD_btultra: number;
export const CONST_ZSTD_btultra2: number;

// Zstd default compression level
export const CONST_ZSTD_CLEVEL_DEFAULT: number;

// Zstd error codes
export const CONST_ZSTD_error_no_error: number;
export const CONST_ZSTD_error_GENERIC: number;
export const CONST_ZSTD_error_prefix_unknown: number;
export const CONST_ZSTD_error_version_unsupported: number;
export const CONST_ZSTD_error_frameParameter_unsupported: number;
export const CONST_ZSTD_error_frameParameter_windowTooLarge: number;
export const CONST_ZSTD_error_corruption_detected: number;
export const CONST_ZSTD_error_checksum_wrong: number;
export const CONST_ZSTD_error_literals_headerWrong: number;
export const CONST_ZSTD_error_dictionary_corrupted: number;
export const CONST_ZSTD_error_dictionary_wrong: number;
export const CONST_ZSTD_error_dictionaryCreation_failed: number;
export const CONST_ZSTD_error_parameter_unsupported: number;
export const CONST_ZSTD_error_parameter_combination_unsupported: number;
export const CONST_ZSTD_error_parameter_outOfBound: number;
export const CONST_ZSTD_error_tableLog_tooLarge: number;
export const CONST_ZSTD_error_maxSymbolValue_tooLarge: number;
export const CONST_ZSTD_error_maxSymbolValue_tooSmall: number;
export const CONST_ZSTD_error_stabilityCondition_notRespected: number;
export const CONST_ZSTD_error_stage_wrong: number;
export const CONST_ZSTD_error_init_missing: number;
export const CONST_ZSTD_error_memory_allocation: number;
export const CONST_ZSTD_error_workSpace_tooSmall: number;
export const CONST_ZSTD_error_dstSize_tooSmall: number;
export const CONST_ZSTD_error_srcSize_wrong: number;
export const CONST_ZSTD_error_dstBuffer_null: number;
export const CONST_ZSTD_error_noForwardProgress_destFull: number;
export const CONST_ZSTD_error_noForwardProgress_inputEmpty: number;

export interface ZlibOptions {
  flush?: number | undefined;
  finishFlush?: number | undefined;
  chunkSize?: number | undefined;
  windowBits?: number | undefined;
  level?: number | undefined; // compression only
  memLevel?: number | undefined; // compression only
  strategy?: number | undefined; // compression only
  dictionary?: ArrayBufferView | undefined; // deflate/inflate only, empty dictionary by default
  info?: boolean | undefined;
  maxOutputLength?: number | undefined;
}

export interface BrotliOptions {
  flush?: number | undefined;
  finishFlush?: number | undefined;
  chunkSize?: number | undefined;
  params?:
    | {
        [key: number]: boolean | number;
      }
    | undefined;
  maxOutputLength?: number | undefined;
  // Not specified in NodeJS docs but the tests expect it
  info?: boolean | undefined;
}

export interface ZstdOptions {
  flush?: number | undefined;
  finishFlush?: number | undefined;
  chunkSize?: number | undefined;
  params?:
    | {
        [key: number]: boolean | number;
      }
    | undefined;
  maxOutputLength?: number | undefined;
  pledgedSrcSize?: number | undefined;
  // Not specified in NodeJS docs but the tests expect it
  info?: boolean | undefined;
}

type ErrorHandler = (errno: number, code: string, message: string) => void;
type ProcessHandler = () => void;

export abstract class CompressionStream {
  [owner_symbol]: Zlib;
  // Not used by C++ implementation but required to be Node.js compatible.
  inOff: number;
  buffer: NodeJS.TypedArray | null;
  cb: () => void;
  availOutBefore: number;
  availInBefore: number;
  flushFlag: number;

  constructor(mode: number);
  close(): void;
  write(
    flushFlag: number,
    inputBuffer: NodeJS.TypedArray,
    inputOffset: number,
    inputLength: number,
    outputBuffer: NodeJS.TypedArray,
    outputOffset: number,
    outputLength: number
  ): void;
  writeSync(
    flushFlag: number,
    inputBuffer: NodeJS.TypedArray,
    inputOffset: number,
    inputLength: number,
    outputBuffer: NodeJS.TypedArray,
    outputOffset: number,
    outputLength: number
  ): void;
  reset(): void;

  // Workerd specific functions
  setErrorHandler(cb: ErrorHandler): void;
}

export class ZlibStream extends CompressionStream {
  initialize(
    windowBits: number,
    level: number,
    memLevel: number,
    strategy: number,
    writeState: NodeJS.TypedArray,
    processCallback: ProcessHandler,
    dictionary: ZlibOptions['dictionary']
  ): void;
  params(level: number, strategy: number): void;
}

export class BrotliDecoder extends CompressionStream {
  initialize(
    params: Uint32Array,
    writeResult: Uint32Array,
    writeCallback: () => void
  ): boolean;
  params(): void;
}

export class BrotliEncoder extends CompressionStream {
  initialize(
    params: Uint32Array,
    writeResult: Uint32Array,
    writeCallback: () => void
  ): boolean;
  params(): void;
}

export class ZstdDecoder extends CompressionStream {
  initialize(
    params: Int32Array,
    writeResult: Uint32Array,
    writeCallback: () => void,
    pledgedSrcSize?: number
  ): boolean;
  params(): void;
}

export class ZstdEncoder extends CompressionStream {
  initialize(
    params: Int32Array,
    writeResult: Uint32Array,
    writeCallback: () => void,
    pledgedSrcSize?: number
  ): boolean;
  params(): void;
}