feat(cli): ✨ Add token generation command to CLI

cli/commands/user/token.ts

@@ -0,0 +1,111 @@
+import { randomString } from "@/math";
+import { Flags } from "@oclif/core";
+import chalk from "chalk";
+import ora from "ora";
+import { Token } from "~/classes/database/token";
+import { UserFinderCommand } from "~/cli/classes";
+import { formatArray } from "~/cli/utils/format";
+
+export default class UserToken extends UserFinderCommand<typeof UserToken> {
+    public static override description = "Generates access tokens for users";
+
+    public static override examples = [
+        "<%= config.bin %> <%= command.id %> johngastron --type username",
+        "<%= config.bin %> <%= command.id %> 018ec11c-c6cb-7a67-bd20-a4c81bf42912",
+        '<%= config.bin %> <%= command.id %> "*badword*" --pattern --type username',
+    ];
+
+    public static override flags = {
+        format: Flags.string({
+            char: "f",
+            description: "Output format",
+            options: ["json", "csv"],
+            exclusive: ["pretty-dates"],
+        }),
+        limit: Flags.integer({
+            char: "n",
+            description: "Limit the number of users",
+            default: 200,
+        }),
+    };
+
+    public static override args = {
+        identifier: UserFinderCommand.baseArgs.identifier,
+    };
+
+    public async run(): Promise<void> {
+        const { flags } = await this.parse(UserToken);
+
+        const foundUsers = await this.findUsers();
+        const users = foundUsers.filter((u) => u.isLocal());
+
+        if (!users || users.length === 0) {
+            this.log(chalk.bold(`${chalk.red("✗")} No users found`));
+
+            if (foundUsers.length > 0) {
+                this.log(
+                    chalk.bold(
+                        `${chalk.yellow("✗")} Found ${chalk.yellow(
+                            foundUsers.length,
+                        )} user(s) but none are local`,
+                    ),
+                );
+            }
+
+            this.exit(1);
+        }
+
+        // Display user
+        flags.print &&
+            this.log(
+                chalk.bold(
+                    `${chalk.green("✓")} Found ${chalk.green(
+                        users.length,
+                    )} user(s)`,
+                ),
+            );
+
+        flags.print &&
+            this.log(
+                formatArray(
+                    users.map((u) => u.data),
+                    [
+                        "id",
+                        "username",
+                        "displayName",
+                        "createdAt",
+                        "updatedAt",
+                        "isAdmin",
+                    ],
+                ),
+            );
+
+        const spinner = ora("Generating access tokens").start();
+        const tokens = await Promise.all(
+            users.map(
+                async (u) =>
+                    await Token.insert({
+                        accessToken: randomString(64, "base64url"),
+                        code: null,
+                        scope: "read write follow",
+                        tokenType: "Bearer",
+                        userId: u.id,
+                    }),
+            ),
+        );
+
+        spinner.succeed();
+
+        this.log(chalk.bold(`${chalk.green("✓")} Tokens generated`));
+
+        this.log(
+            formatArray(
+                tokens.map((t) => t.data),
+                ["accessToken", "userId"],
+                flags.format as "json" | "csv" | undefined,
+            ),
+        );
+
+        this.exit(0);
+    }
+}

cli/index.ts

@@ -11,6 +11,7 @@ export const commands = {
     "user:create": (await import("./commands/user/create.ts")).default,
     "user:reset": (await import("./commands/user/reset.ts")).default,
     "user:refetch": (await import("./commands/user/refetch.ts")).default,
+    "user:token": (await import("./commands/user/token.ts")).default,
     "emoji:add": (await import("./commands/emoji/add.ts")).default,
     "emoji:delete": (await import("./commands/emoji/delete.ts")).default,
     "emoji:list": (await import("./commands/emoji/list.ts")).default,