@wirechunk/cli 0.0.6 → 0.0.8

package/src/commands/ext-dev/get-db-url.ts

@@ -1,36 +0,0 @@
-import { existsSync } from 'node:fs';
-import { readFile, writeFile } from 'node:fs/promises';
-import { resolve } from 'node:path';
-import type { Env } from '../../env.ts';
-import type { WithGlobalOptions } from '../../global-options.js';
-import { getExtensionDbConnectInfo, replaceEnvVar } from '../../util.ts';
-
-type GetDbUrlOptions = {
-  extensionId?: string;
-  dbName?: string;
-};
-
-export const getDbUrl = async (
-  opts: WithGlobalOptions<GetDbUrlOptions>,
-  env: Env,
-): Promise<void> => {
-  const connInfo = getExtensionDbConnectInfo(opts, env);
-  const url = connInfo.url.toString();
-  const cwd = process.cwd();
-  const localEnvFilePath = opts.envMode
-    ? resolve(cwd, `./.env.${opts.envMode}.local`)
-    : resolve(cwd, './.env.local');
-  if (opts.verbose) {
-    console.log(`Loading file ${localEnvFilePath} to update environment variables`);
-  }
-  if (existsSync(localEnvFilePath)) {
-    const localEnv = await readFile(localEnvFilePath, 'utf8');
-    const lines = replaceEnvVar(localEnv.split('\n'), 'DATABASE_URL', url);
-    await writeFile(localEnvFilePath, lines.join('\n'));
-  } else {
-    if (opts.verbose) {
-      console.log(`Creating file ${localEnvFilePath}`);
-    }
-    await writeFile(localEnvFilePath, `DATABASE_URL=${url}\n`);
-  }
-};

package/src/commands/ext-dev/init-db.ts

@@ -1,228 +0,0 @@
-import type { DatabasePool } from 'slonik';
-import { createPool, sql } from 'slonik';
-import { z } from 'zod';
-import type { Env } from '../../env.js';
-import { requireCoreDbUrl } from '../../env.ts';
-import { isDuplicateDatabaseError } from '../../errors.ts';
-import type { WithGlobalOptions } from '../../global-options.js';
-import {
-  dbPoolOptions,
-  getExtensionDbConnectInfo,
-  requireExtensionIdOptionOrEnvVar,
-} from '../../util.ts';
-import { getDbUrl } from './get-db-url.ts';
-
-const initExtDb = async ({
-  // A connection to the extension database by a superuser role.
-  db,
-  extensionRoleName,
-  coreDbUrl,
-}: {
-  db: DatabasePool;
-  extensionRoleName: string;
-  coreDbUrl: URL;
-}) => {
-  await db.query(sql.unsafe`
-    create extension if not exists postgres_fdw
-  `);
-  await db.query(sql.unsafe`
-    create server if not exists wirechunk
-    foreign data wrapper postgres_fdw
-    options (
-      host ${sql.literalValue(coreDbUrl.hostname)},
-      port ${sql.literalValue(coreDbUrl.port)},
-      dbname ${sql.literalValue(coreDbUrl.pathname.slice(1))}
-    )
-  `);
-
-  const extRoleIdent = sql.identifier([extensionRoleName]);
-
-  await db.query(sql.unsafe`
-    create user mapping if not exists for ${extRoleIdent} server wirechunk
-    options (user ${sql.literalValue(extensionRoleName)}, password_required 'false')
-  `);
-  await db.query(sql.unsafe`
-    create foreign table if not exists "Users" (
-      "id" text not null,
-      "firstName" text not null,
-      "lastName" text not null,
-      "email" text not null,
-      "emailVerified" boolean not null,
-      "orgId" text,
-      "role" text,
-      "status" text not null,
-      "expiresAt" timestamptz,
-      "createdAt" timestamptz not null
-    ) server wirechunk options (schema_name 'public', table_name 'Users')
-  `);
-  await db.query(sql.unsafe`
-    create foreign table if not exists "Orgs" (
-      "id" text not null,
-      "name" text,
-      "primaryUserId" text,
-      "createdAt" timestamptz not null
-    ) server wirechunk options (schema_name 'public', table_name 'Orgs')
-  `);
-  await db.query(sql.unsafe`
-    create foreign table if not exists "Sites" (
-      "id" text not null,
-      "domain" text not null,
-      "orgId" text,
-      "name" text not null,
-      "createdAt" timestamptz not null
-    ) server wirechunk options (schema_name 'public', table_name 'Sites')
-  `);
-
-  // These are not the permissions granted to actual extension roles in production.
-  // Here we just want to simplify the development experience.
-  // At any moment, a developer may drop an extension's database and reinitialize it.
-  await db.query(sql.unsafe`
-    grant all on schema public to ${extRoleIdent}
-  `);
-  await db.query(sql.unsafe`
-    grant all on table "Orgs" to ${extRoleIdent}
-  `);
-  await db.query(sql.unsafe`
-    grant all on table "Sites" to ${extRoleIdent}
-  `);
-  await db.query(sql.unsafe`
-    grant all on table "Users" to ${extRoleIdent}
-  `);
-  await db.query(sql.unsafe`
-    alter default privileges grant all on schemas to ${extRoleIdent}
-  `);
-  await db.query(sql.unsafe`
-    alter default privileges grant all on types to ${extRoleIdent}
-  `);
-  await db.query(sql.unsafe`
-    alter default privileges grant all on tables to ${extRoleIdent}
-  `);
-  await db.query(sql.unsafe`
-    alter default privileges grant all on sequences to ${extRoleIdent}
-  `);
-  await db.query(sql.unsafe`
-    alter default privileges grant all on functions to ${extRoleIdent}
-  `);
-};
-
-const extensionSelectSchema = z.object({
-  id: z.string(),
-  platformId: z.string(),
-});
-
-const applyExtensionPolicy = async ({
-  extRole,
-  table,
-  platformId,
-  db,
-}: {
-  extRole: string;
-  table: string;
-  platformId: string;
-  db: DatabasePool;
-}) => {
-  const policyNameIdent = sql.identifier([`${table}_${extRole}_select`]);
-  const extRoleIdent = sql.identifier([extRole]);
-  const tableIdent = sql.identifier([table]);
-  await db.query(sql.unsafe`
-    drop policy if exists ${policyNameIdent} on ${tableIdent}
-  `);
-  await db.query(sql.unsafe`
-    create policy ${policyNameIdent} on ${tableIdent} as permissive
-    for select to ${extRoleIdent}
-    using ("platformId" = ${sql.literalValue(platformId)})
-  `);
-  await db.query(sql.unsafe`
-    grant select on table ${tableIdent} to ${extRoleIdent}
-  `);
-};
-
-type InitExtDbOptions = {
-  extensionId?: string;
-  dbName?: string;
-};
-
-export const initDb = async (
-  opts: WithGlobalOptions<InitExtDbOptions>,
-  env: Env,
-): Promise<void> => {
-  const coreDbUrl = requireCoreDbUrl(env);
-  const extensionId = requireExtensionIdOptionOrEnvVar(opts, env);
-
-  // This writes the DATABASE_URL environment variable to a .env.local (or .env.<env-mode>.local) file.
-  await getDbUrl(opts, env);
-
-  const db = await createPool(coreDbUrl, dbPoolOptions(opts));
-
-  const extension = await db.maybeOne(sql.type(extensionSelectSchema)`
-    select "id", "platformId"
-    from "Extensions"
-    where "id" = ${extensionId}
-  `);
-  if (!extension) {
-    console.error(`Extension with ID ${extensionId} not found`);
-    process.exit(1);
-  }
-
-  const coreDbUrlObject = new URL(coreDbUrl);
-  const extDb = getExtensionDbConnectInfo(opts, env);
-
-  try {
-    await db.query(sql.unsafe`
-      create database ${sql.identifier([extDb.dbName])}
-    `);
-  } catch (err) {
-    if (!isDuplicateDatabaseError(err)) {
-      console.error('Failed to create a database:', err);
-      process.exit(1);
-    }
-  }
-
-  const extRole = `ext_${extension.id}`;
-
-  const extRoleIdent = sql.identifier([extRole]);
-  await db.query(sql.unsafe`
-    do $$
-    begin
-      if not exists (
-        select 1 from pg_roles where rolname = ${sql.literalValue(extRole)}
-      ) then
-        create role ${extRoleIdent} login noinherit createdb;
-      end if;
-    end; $$
-  `);
-  await db.query(sql.unsafe`
-    grant ${extRoleIdent} to ${sql.identifier([coreDbUrlObject.username])}
-  `);
-
-  await db.query(sql.unsafe`
-    grant connect, create, temporary on database ${sql.identifier([extDb.dbName])} to ${extRoleIdent}
-  `);
-  await applyExtensionPolicy({
-    extRole,
-    table: 'Orgs',
-    platformId: extension.platformId,
-    db,
-  });
-  await applyExtensionPolicy({
-    extRole,
-    table: 'Sites',
-    platformId: extension.platformId,
-    db,
-  });
-  await applyExtensionPolicy({
-    extRole,
-    table: 'Users',
-    platformId: extension.platformId,
-    db,
-  });
-
-  const extDbSuper = new URL(extDb.url);
-  extDbSuper.username = coreDbUrlObject.username;
-
-  await initExtDb({
-    db: await createPool(extDbSuper.toString(), dbPoolOptions(opts)),
-    extensionRoleName: extRole,
-    coreDbUrl: coreDbUrlObject,
-  });
-};