import { ZipFile } from "yazl";
import type { NavigationNode } from "@shared/types";
import { FileOperationFormat } from "@shared/types";
import type { Collection, FileOperation } from "@server/models";
import type { Document } from "@server/models";
import ExportDocumentTreeTask from "./ExportDocumentTreeTask";

export default class ExportHTMLZipTask extends ExportDocumentTreeTask {
  public async exportCollections(
    collections: Collection[],
    fileOperation: FileOperation
  ) {
    const zip = new ZipFile();

    return await this.addCollectionsToArchive(
      zip,
      collections,
      FileOperationFormat.HTMLZip,
      fileOperation.options?.includeAttachments ?? true
    );
  }

  public async exportDocument(
    document: Document,
    documentStructure: NavigationNode[]
  ): Promise<string> {
    const zip = new ZipFile();

    return await this.addDocumentToArchive({
      document,
      documentStructure,
      format: FileOperationFormat.HTMLZip,
      zip,
    });
  }
}