outline/server/queues/processors/MarkdownImportsProcessor.ts

import type { Transaction } from "sequelize";
import type { ImportTaskInput } from "@shared/schema";
import { ImportTaskPhase, IntegrationService } from "@shared/types";
import type { Import, ImportTask } from "@server/models";
import MarkdownAPIImportTask from "../tasks/MarkdownAPIImportTask";
import ImportsProcessor from "./ImportsProcessor";

export default class MarkdownImportsProcessor extends ImportsProcessor<IntegrationService.Markdown> {
  protected canProcess(
    importModel: Import<IntegrationService.Markdown>
  ): boolean {
    return importModel.service === IntegrationService.Markdown;
  }

  protected getInitialPhase(): ImportTaskPhase {
    return ImportTaskPhase.Bootstrap;
  }

  protected async buildTasksInput(
    importModel: Import<IntegrationService.Markdown>,
    _transaction: Transaction
  ): Promise<ImportTaskInput<IntegrationService.Markdown>> {
    if (!importModel.scratch?.storageKey) {
      throw new Error(
        "Markdown import is missing scratch.storageKey for the bootstrap phase"
      );
    }

    return [{ externalId: importModel.input[0].externalId }];
  }

  protected async scheduleTask(
    importTask: ImportTask<IntegrationService.Markdown>
  ): Promise<void> {
    await new MarkdownAPIImportTask().schedule({ importTaskId: importTask.id });
  }
}