root.ts

import { Command } from "@cliffy/command";
import { CompletionsCommand } from "@cliffy/command/completions";
import { UpgradeCommand } from "@cliffy/command/upgrade";
import { JsrProvider } from "@cliffy/command/upgrade/provider/jsr";
import manifest from "./deno.json" with { type: "json" };
import open from "open";
import { toText } from "@std/streams";
import { Table } from "@cliffy/table";
import { valCmd } from "./val.ts";
import { fetchValTown, printJson, printYaml } from "./lib.ts";
import { blobCmd } from "./blob.ts";
import { tableCmd } from "./table.ts";

const cmd: Command = new Command()
    .name("vt")
    .help({ colors: Deno.stdout.isTerminal() })
    .version(manifest.version)
    .action(
        () => {
            cmd.showHelp();
        },
    );

cmd.command("val", valCmd);
cmd.command("blob", blobCmd);
cmd.command("table", tableCmd);

cmd
    .command("api")
    .description("Make an API request.")
    .example("Get your user info", "vt api /v1/me")
    .arguments("<url-or-path:string>")
    .option("-X, --method <method:string>", "HTTP method.", { default: "GET" })
    .option("-d, --data <data:string>", "Request Body")
    .option("-H, --header <header:string>", "Request Header", { collect: true })
    .action(async ({ method, data, header }, url) => {
        const headers: Record<string, string> = {};
        for (const h of header || []) {
            const [key, value] = h.split(":", 2);
            headers[key.trim()] = value.trim();
        }

        let body: string | undefined;
        if (data == "@-") {
            body = await toText(Deno.stdin.readable);
        } else if (data) {
            body = data;
        }

        const resp = await fetchValTown(url, {
            method,
            headers,
            body,
        });

        if (!resp.ok) {
            console.error(await resp.text());
            Deno.exit(1);
        }

        if (resp.headers.get("Content-Type")?.includes("application/json")) {
            return printJson(await resp.json());
        }

        console.log(await resp.text());
    });

cmd
    .command("openapi")
    .description("View openapi specs")
    .hidden()
    .option("--web, -w", "Open in browser")
    .action(async ({ web }) => {
        if (web) {
            open("https://www.val.town/docs/openapi.html");
            Deno.exit(0);
        }

        const resp = await fetch("https://www.val.town/docs/openapi.yaml");
        if (resp.status != 200) {
            console.error(resp.statusText);
            Deno.exit(1);
        }

        if (!resp.ok) {
            console.error(await resp.text());
            Deno.exit(1);
        }

        printYaml(await resp.text());
    });

cmd.command("completions", new CompletionsCommand());

cmd
    .command("email")
    .description("Send an email.")
    .option("-t, --to <to:string>", "To")
    .option("-s, --subject <subject:string>", "Subject")
    .option("-b, --body <body:string>", "Body")
    .action(async (options) => {
        const resp = await fetchValTown("/v1/email", {
            method: "POST",
            body: JSON.stringify({
                from: "pomdtr.vt@valtown.email",
                to: options.to,
                subject: options.subject,
                text: options.body,
            }),
        });

        if (!resp.ok) {
            console.error(await resp.text());
            Deno.exit(1);
        }

        console.log("Email sent.");
    });

cmd
    .command("query")
    .description("Execute a query.")
    .arguments("<query:string>")
    .action(async (_, query) => {
        const resp = await fetchValTown("/v1/sqlite/execute", {
            method: "POST",
            body: JSON.stringify({ statement: query }),
        });

        if (!resp.ok) {
            console.error(await resp.text());
            Deno.exit(1);
        }

        const res = await resp.json();
        if (!Deno.stdout.isTerminal()) {
            console.log(JSON.stringify(res));
            return;
        }

        const table = new Table(...res.rows).header(res.columns);
        table.render();
    });

cmd.command(
    "upgrade",
    new UpgradeCommand({
        args: ["--allow-all"],
        provider: new JsrProvider({
            package: "@pomdtr/vt",
        }),
    }),
);

export { cmd };