pocketbase-import/csv.ts

// @deno-types="https://unpkg.com/pocketbase@0.12.0/dist/pocketbase.es.d.mts"
import PocketBase, {
  Collection,
  SchemaField,
} from "https://unpkg.com/pocketbase@0.12.0/dist/pocketbase.es.mjs";
import "https://deno.land/std@0.178.0/dotenv/load.ts";
import { parse } from "https://deno.land/std@0.175.0/flags/mod.ts";
import { parseCsv } from "./utils/csv.ts";
import { createSchema, parseData } from "./utils/pocketbase.ts";

/**
 * Structures and populates a new collection from a CSV file.
 * @returns
 */
async function importCsv() {
  // config data
  const pbUrl = Deno.env.get("POCKETBASE_URL") ?? "http://localhost:8090";
  const adminName = Deno.env.get("ADMIN_EMAIL") ?? "";
  const adminPass = Deno.env.get("ADMIN_PASSWORD") ?? "";

  // parse CLI args
  const options = parse(Deno.args, {
    string: ["input", "delimiter", "quote"],
    boolean: ["id", "lf"],
    default: {
      /**
       * Name of the CSV file to import.
       */
      input: null,
      /**
       * Value separator (defaults to `,`).
       */
      delimiter: ",",
      /**
       * Quote character (defaults to `'`).
       */
      quote: "'",
      /**
       * Flag to always set `_id` column type to Plain text (detected by default).
       */
      id: false,
      /**
       * Whether LF end-of-line should be used (defaults to CRLF).
       */
      lf: false,
    },
  });

  if (options.input === null) {
    console.error("%cOptionError: CSV file name not supplied", "color: red");
    return;
  }

  // parser options
  const csvOptions = {
    columnSeparator: options.delimiter,
    lineSeparator: options.lf ? "\n" : "\r\n",
    quote: options.quote,
  };

  // parses CSV
  const data = await parseCsv(options.input, csvOptions);

  // empty file
  if (data === null) {
    console.error(
      `%c[Import] No data to import from ${options.input}`,
      "color: red",
    );
    return;
  }

  // sanitize the file name for collection name
  const collectName = options.input.replace(".csv", "");

  // connect to pocketbase
  const pb = new PocketBase(pbUrl);

  // authenticate as super admin
  const _authResponse = await pb.admins.authWithPassword(adminName, adminPass);

  // collection schema object
  const schema: SchemaField[] = createSchema(data, options.id);

  const creationDate = new Date().toISOString();

  // the new collection
  const collection = new Collection({
    name: collectName,
    type: "base",
    system: false,
    schema,
    listRule: null,
    viewRule: null,
    createRule: null,
    updateRule: null,
    deleteRule: null,
    options: {},
    created: creationDate,
    updated: creationDate,
  });

  // show the submitted collection
  console.log(collection);

  // create the new collection
  // import will fail if a collection with the same name exists
  await pb.collections.import([collection]);

  console.log(
    `%c[Import] Collection '${collectName}' created!`,
    "color: green",
  );

  // rows to be sent via PocketBase API
  const rows = parseData(data, schema);

  // number of successfully inserted rows
  let insertCount = 0;

  for (insertCount; insertCount < rows.length; insertCount++) {
    try {
      await pb.collection(collectName).create(rows[insertCount], {
        "$autoCancel": false,
      });
    } catch (e) {
      // breaks on first error
      console.error(e);
      break;
    }
  }

  const color = insertCount === rows.length ? "green" : "orange";
  console.log(
    `%c[Import] Imported rows: ${insertCount}/${rows.length}`,
    `color: ${color}`,
  );
}

importCsv();