import type { FileItem } from '../../FileItem.ts';
import type { Options } from '../../Options.ts';
import { defaultOptions } from '../../Options.ts';
import { shouldAddItem } from '../shouldAddItem.ts';
import { isZip } from '../zip.ts';

import { expandAndFilter } from './expandAndFilter.ts';
import { fileItemsFromZip } from './fileItemsFromZip.ts';

/**
 * Some files in the fileItems may actually be zip. This method will unzip those files.
 * The method will actually not really unzip the files but only add them in the fileItems.
 * Unzipping will only take place when you want to actually retrieve the data.
 * @param fileItem - The file item to check and potentially unzip. if not zip file, return an array with the fileItem.
 * @param options - Options to filter the files and control the unzipping behavior.
 * @returns An array of file items, which may include unzipped files if the input was a zip file.
 */
export async function fileItemUnzip(
  fileItem: FileItem,
  options: Options = {},
): Promise<FileItem[]> {
  const { unzip = {}, filter = {}, logger } = options;
  let { zipExtensions = defaultOptions.unzip.zipExtensions } = unzip;
  const { recursive = defaultOptions.unzip.recursive } = unzip;
  zipExtensions = zipExtensions.map((extension) => extension.toLowerCase());
  const extension = fileItem.name.replace(/^.*\./, '').toLowerCase();

  if (!fileItem.sourceUUID) {
    throw new Error('fileItem.sourceUUID is not defined');
  }

  if (!zipExtensions.includes(extension)) {
    return [fileItem];
  }

  const buffer = await fileItem.arrayBuffer();

  if (!isZip(buffer)) {
    if (logger) {
      logger.info(
        `Could not unzip the following file: ${fileItem.relativePath}`,
      );
    }
    return [fileItem];
  }

  const zipFileItems = fileItemsFromZip(buffer, fileItem.sourceUUID, options);
  const fileItems: FileItem[] = [];
  for await (const zipEntry of zipFileItems) {
    zipEntry.parent = fileItem;
    zipEntry.relativePath = `${fileItem.relativePath}/${zipEntry.relativePath}`;
    zipEntry.sourceUUID = fileItem.sourceUUID;
    if (recursive) {
      const recursiveFiles = await expandAndFilter(zipEntry, options);
      fileItems.push(...recursiveFiles);
    } else if (shouldAddItem(zipEntry.relativePath, filter)) {
      fileItems.push(zipEntry);
    }
  }

  return fileItems;
}