src/cli.ts

Functions

Imports

It imports src/cloudflare.ts, src/drill.ts, src/node.ts. No module of this repo imports it.

Source

1#!/usr/bin/env node
2import { cloudflareScratch } from "./cloudflare.js";
3import { runRestoreDrill } from "./drill.js";
4import { directoryBucket, pickBackend, sqliteScratch } from "./node.js";
5
6const USAGE = `usage: d1-dump drill --dir <downloaded dump root> [--backend sqlite|d1] [--prefix <backups/json/run id/>] [--max-age-hours <n|off>]
7  sqlite  restores into a local SQLite file and needs no credential (the default when no Cloudflare credentials are set)
8  d1      restores into a scratch D1 database over the REST API; needs CLOUDFLARE_ACCOUNT_ID and CLOUDFLARE_API_TOKEN
9          (create and delete D1 databases) in the environment`;
10
11function arg(args: string[], name: string): string | undefined {
12  const i = args.indexOf(name);
13  return i === -1 ? undefined : args[i + 1];
14}
15
16async function main(args: string[]): Promise<number> {
17  const dir = arg(args, "--dir");
18  const accountId = process.env.CLOUDFLARE_ACCOUNT_ID;
19  const token = process.env.CLOUDFLARE_API_TOKEN;
20  const asked = arg(args, "--backend");
21  const picked = pickBackend(asked, Boolean(accountId && token));
22  if (args[0] !== "drill" || !dir || "error" in picked) {
23    console.error(`${"error" in picked ? `d1-dump: ${picked.error}\n` : ""}${USAGE}`);
24    return 2;
25  }
26  const { backend, why } = picked;
27  console.log(`backend: ${backend} (${why})`);
28  const age = arg(args, "--max-age-hours");
29  const result = await runRestoreDrill(directoryBucket(dir), {
30    createScratch: backend === "d1" ? cloudflareScratch(accountId!, token!) : sqliteScratch(),
31    prefix: arg(args, "--prefix"),
32    maxAgeHours: age === undefined ? undefined : age === "off" ? null : Number(age),
33  });
34  console.log(`restore drill ${result.ok ? "PASSED" : "FAILED"}: ${result.run ?? "no dump"} into ${result.scratch} (${backend})`);
35  for (const c of result.checks) console.log(`  ${c.ok ? "ok  " : "FAIL"} ${c.name}: ${c.detail}`);
36  return result.ok ? 0 : 1;
37}
38
39main(process.argv.slice(2)).then(
40  (code) => process.exit(code),
41  (err: unknown) => {
42    console.error(`d1-dump: ${err instanceof Error ? err.message : String(err)}`);
43    process.exit(1);
44  }
45);