feat(cli-v2): implement db status and db stats commands

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

Author: sroussey

src/cli/groups/db.ts

@@ -1,6 +1,8 @@
 import type { Command } from "commander";
 import { setupAllDatabases } from "../../config/setupAllDatabases";
 import { runCommand } from "../runCommand";
+import { getDbStatus, getDbStats } from "../queries/DbStatus";
+import { renderTable } from "../output/TableRenderer";
 
 export function addDbCommands(program: Command): void {
   const db = program.command("db").description("Database management commands");
@@ -15,14 +17,45 @@ export function addDbCommands(program: Command): void {
 
   db.command("status")
     .description("Show database connection status")
-    .action(async () => {
-      console.log("not yet implemented");
+    .option("--format <format>", "Output format (table, json)", "table")
+    .action(async (options: Record<string, string>) => {
+      await runCommand(async () => {
+        const status = await getDbStatus();
+
+        if (options.format === "json") {
+          console.log(JSON.stringify(status, null, 2));
+          return;
+        }
+
+        const fmt = (n: number): string => n.toLocaleString();
+        console.log("Database Status\n");
+        console.log(`  Entities:              ${fmt(status.entityCount)}`);
+        console.log(`  Filings:               ${fmt(status.filingCount)}`);
+        console.log(`  Company Facts:         ${fmt(status.factsCount)}`);
+        console.log(`  Processed Submissions: ${fmt(status.processedSubmissions)}`);
+        console.log(`  Processed Facts:       ${fmt(status.processedFacts)}`);
+        console.log(`  Processed Filings:     ${fmt(status.processedFilings)}`);
+      });
     });
 
   db.command("stats")
     .description("Show row counts and database size")
-    .action(async () => {
-      console.log("not yet implemented");
+    .option("--format <format>", "Output format (table, json)", "table")
+    .action(async (options: Record<string, string>) => {
+      await runCommand(async () => {
+        const stats = await getDbStats();
+
+        const columns = [
+          { key: "table", header: "Table", width: 25 },
+          { key: "rows", header: "Rows", width: 12 },
+        ];
+
+        console.log(
+          renderTable(stats as unknown as Record<string, unknown>[], columns, {
+            format: (options.format as "table" | "json") ?? "table",
+          })
+        );
+      });
     });
 
   db.command("reset")

src/cli/queries/DbStatus.test.ts

@@ -0,0 +1,62 @@
+import { beforeEach, describe, expect, it } from "bun:test";
+import { resetDependencyInjectionsForTesting } from "../../config/TestingDI";
+import { globalServiceRegistry } from "@workglow/util";
+import { ENTITY_REPOSITORY_TOKEN } from "../../storage/entity/EntitySchema";
+import { getDbStatus, getDbStats } from "./DbStatus";
+
+describe("getDbStatus", () => {
+  beforeEach(() => {
+    resetDependencyInjectionsForTesting();
+  });
+
+  it("returns zero counts for empty db", async () => {
+    const result = await getDbStatus();
+    expect(result.entityCount).toBe(0);
+    expect(result.filingCount).toBe(0);
+    expect(result.factsCount).toBe(0);
+    expect(result.processedSubmissions).toBe(0);
+    expect(result.processedFacts).toBe(0);
+    expect(result.processedFilings).toBe(0);
+  });
+
+  it("counts entities after insertion", async () => {
+    const repo = globalServiceRegistry.get(ENTITY_REPOSITORY_TOKEN);
+    await repo.put({
+      cik: 1318605,
+      name: "Tesla, Inc.",
+      type: null,
+      sic: 3711,
+      ein: null,
+      description: null,
+      website: null,
+      investor_website: null,
+      category: null,
+      fiscal_year: null,
+      state_incorporation: "TX",
+      state_incorporation_desc: null,
+    });
+
+    const result = await getDbStatus();
+    expect(result.entityCount).toBe(1);
+  });
+});
+
+describe("getDbStats", () => {
+  beforeEach(() => {
+    resetDependencyInjectionsForTesting();
+  });
+
+  it("returns array of table stats", async () => {
+    const result = await getDbStats();
+    expect(Array.isArray(result)).toBe(true);
+    expect(result.length).toBeGreaterThanOrEqual(10);
+  });
+
+  it("each stat has table and rows properties", async () => {
+    const result = await getDbStats();
+    for (const stat of result) {
+      expect(typeof stat.table).toBe("string");
+      expect(typeof stat.rows).toBe("number");
+    }
+  });
+});

src/cli/queries/DbStatus.ts

@@ -0,0 +1,108 @@
+import { globalServiceRegistry } from "@workglow/util";
+import {
+  ENTITY_REPOSITORY_TOKEN,
+} from "../../storage/entity/EntitySchema";
+import {
+  FILING_REPOSITORY_TOKEN,
+} from "../../storage/filing/FilingSchema";
+import {
+  COMPANY_FACTS_REPOSITORY_TOKEN,
+} from "../../storage/facts/CompanyFactsSchema";
+import {
+  PROCESSED_SUBMISSIONS_REPOSITORY_TOKEN,
+} from "../../storage/processing/ProcessedSubmissionsSchema";
+import {
+  PROCESSED_FACTS_REPOSITORY_TOKEN,
+} from "../../storage/processing/ProcessedFactsSchema";
+import {
+  PROCESSED_FILINGS_REPOSITORY_TOKEN,
+} from "../../storage/processing/ProcessedFilingsSchema";
+import {
+  INVESTMENT_OFFERING_REPOSITORY_TOKEN,
+} from "../../storage/investment-offering/InvestmentOfferingSchema";
+import {
+  CROWDFUNDING_REPOSITORY_TOKEN,
+} from "../../storage/portal/CrowdfundingSchema";
+import {
+  PERSON_REPOSITORY_TOKEN,
+} from "../../storage/person/PersonSchema";
+import {
+  ADDRESS_REPOSITORY_TOKEN,
+} from "../../storage/address/AddressSchema";
+import {
+  PHONE_REPOSITORY_TOKEN,
+} from "../../storage/phone/PhoneSchema";
+import {
+  COMPANY_REPOSITORY_TOKEN,
+} from "../../storage/company/CompanySchema";
+import {
+  PORTAL_REPOSITORY_TOKEN,
+} from "../../storage/portal/PortalSchema";
+import type { ServiceToken } from "@workglow/util";
+
+export interface DbStatusResult {
+  readonly entityCount: number;
+  readonly filingCount: number;
+  readonly factsCount: number;
+  readonly processedSubmissions: number;
+  readonly processedFacts: number;
+  readonly processedFilings: number;
+}
+
+export interface TableStat {
+  readonly table: string;
+  readonly rows: number;
+}
+
+async function countRows(token: ServiceToken<{ getAll(): Promise<unknown[]> }>): Promise<number> {
+  const repo = globalServiceRegistry.get(token);
+  const all = (await repo.getAll()) ?? [];
+  return all.length;
+}
+
+export async function getDbStatus(): Promise<DbStatusResult> {
+  const [entityCount, filingCount, factsCount, processedSubmissions, processedFacts, processedFilings] =
+    await Promise.all([
+      countRows(ENTITY_REPOSITORY_TOKEN as any),
+      countRows(FILING_REPOSITORY_TOKEN as any),
+      countRows(COMPANY_FACTS_REPOSITORY_TOKEN as any),
+      countRows(PROCESSED_SUBMISSIONS_REPOSITORY_TOKEN as any),
+      countRows(PROCESSED_FACTS_REPOSITORY_TOKEN as any),
+      countRows(PROCESSED_FILINGS_REPOSITORY_TOKEN as any),
+    ]);
+
+  return {
+    entityCount,
+    filingCount,
+    factsCount,
+    processedSubmissions,
+    processedFacts,
+    processedFilings,
+  };
+}
+
+const TABLE_TOKENS: ReadonlyArray<{
+  readonly table: string;
+  readonly token: ServiceToken<{ getAll(): Promise<unknown[]> }>;
+}> = [
+  { table: "entity", token: ENTITY_REPOSITORY_TOKEN as any },
+  { table: "filing", token: FILING_REPOSITORY_TOKEN as any },
+  { table: "company_facts", token: COMPANY_FACTS_REPOSITORY_TOKEN as any },
+  { table: "investment_offering", token: INVESTMENT_OFFERING_REPOSITORY_TOKEN as any },
+  { table: "crowdfunding", token: CROWDFUNDING_REPOSITORY_TOKEN as any },
+  { table: "person", token: PERSON_REPOSITORY_TOKEN as any },
+  { table: "address", token: ADDRESS_REPOSITORY_TOKEN as any },
+  { table: "phone", token: PHONE_REPOSITORY_TOKEN as any },
+  { table: "company", token: COMPANY_REPOSITORY_TOKEN as any },
+  { table: "portal", token: PORTAL_REPOSITORY_TOKEN as any },
+];
+
+export async function getDbStats(): Promise<TableStat[]> {
+  const results = await Promise.all(
+    TABLE_TOKENS.map(async ({ table, token }) => {
+      const rows = await countRows(token);
+      return { table, rows };
+    })
+  );
+  return results;
+}