@rebasepro/server-postgres 0.9.1-canary.16c42e9 → 0.9.1-canary.26fe4b2

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (35) hide show
  1. package/README.md +21 -0
  2. package/dist/PostgresBackendDriver.d.ts +18 -0
  3. package/dist/PostgresBootstrapper.d.ts +7 -1
  4. package/dist/auth/services.d.ts +53 -53
  5. package/dist/index.es.js +730 -194
  6. package/dist/index.es.js.map +1 -1
  7. package/dist/schema/auth-bootstrap-sql.d.ts +1 -1
  8. package/dist/schema/auth-schema.d.ts +24 -24
  9. package/dist/schema/doctor.d.ts +1 -1
  10. package/dist/schema/introspect-db-logic.d.ts +0 -5
  11. package/dist/schema/introspect-db-naming.d.ts +10 -0
  12. package/dist/security/policy-drift.d.ts +30 -0
  13. package/dist/security/rls-enforcement.d.ts +2 -2
  14. package/dist/services/channel-history.d.ts +118 -0
  15. package/dist/services/realtimeService.d.ts +69 -2
  16. package/package.json +8 -31
  17. package/src/PostgresBackendDriver.ts +56 -5
  18. package/src/PostgresBootstrapper.ts +18 -1
  19. package/src/auth/ensure-tables.ts +97 -17
  20. package/src/auth/services.ts +134 -133
  21. package/src/cli.ts +60 -0
  22. package/src/schema/auth-bootstrap-sql.ts +7 -1
  23. package/src/schema/auth-schema.ts +13 -13
  24. package/src/schema/doctor.ts +45 -20
  25. package/src/schema/introspect-db-inference.ts +1 -1
  26. package/src/schema/introspect-db-logic.ts +1 -10
  27. package/src/schema/introspect-db-naming.ts +15 -0
  28. package/src/schema/introspect-runtime.ts +1 -1
  29. package/src/security/policy-drift.test.ts +106 -1
  30. package/src/security/policy-drift.ts +56 -0
  31. package/src/security/rls-enforcement.ts +11 -5
  32. package/src/services/channel-history.ts +343 -0
  33. package/src/services/realtimeService.ts +198 -10
  34. package/src/services/row-pipeline.ts +27 -3
  35. package/src/websocket.ts +30 -11
package/src/cli.ts CHANGED
@@ -176,6 +176,7 @@ async function dbCommand(subcommand: string, rawArgs: string[]): Promise<void> {
176
176
 
177
177
  if (databaseUrl) {
178
178
  await applyPolicies(databaseUrl);
179
+ await reconcilePolicies(databaseUrl, collectionsPath);
179
180
  await ensureRlsUserRole(databaseUrl);
180
181
  } else {
181
182
  logger.warn(chalk.yellow(" ⚠️ DATABASE_URL not found in environment, skipping RLS policies application."));
@@ -269,6 +270,65 @@ async function applyPolicies(databaseUrl: string): Promise<void> {
269
270
  }
270
271
  }
271
272
 
273
+ /**
274
+ * Remove the policies an earlier push superseded but never dropped.
275
+ *
276
+ * `policies.sql` only DROPs the names it is about to CREATE, and a rule's
277
+ * generated name contains a hash of its own semantics — so editing a rule
278
+ * writes a *new* policy and abandons the old one. Postgres ORs PERMISSIVE
279
+ * policies together, which makes an abandoned grant outrank every tightening
280
+ * that replaced it, and push reported success the whole time.
281
+ *
282
+ * Runs after `applyPolicies` so the current policies are already in place: the
283
+ * drift check then sees exactly the set that should survive, and anything else
284
+ * on a managed table is by definition left over.
285
+ */
286
+ async function reconcilePolicies(databaseUrl: string, collectionsPath: string): Promise<void> {
287
+ try {
288
+ const { checkPolicyDrift, dropOrphanedPolicies, formatPolicyDrift, hasDrift } =
289
+ await import("./security/policy-drift");
290
+ const { loadCollections } = await import("./schema/doctor");
291
+
292
+ const collections = await loadCollections(path.resolve(process.cwd(), collectionsPath));
293
+ const { Client } = await import("pg");
294
+ const client = new Client({ connectionString: databaseUrl });
295
+ await client.connect();
296
+ try {
297
+ const drift = await checkPolicyDrift(client as never, collections);
298
+ const { dropped, kept } = await dropOrphanedPolicies(client as never, drift, collections);
299
+
300
+ for (const p of dropped) {
301
+ logger.info(chalk.gray(` ✓ Dropped superseded policy "${p.name}" on ${p.schema}.${p.table}`));
302
+ }
303
+ if (dropped.length > 0) {
304
+ logger.info(chalk.green(` ✓ Removed ${dropped.length} superseded RLS ${dropped.length === 1 ? "policy" : "policies"}.`));
305
+ }
306
+
307
+ // Custom-named orphans are indistinguishable from policies someone
308
+ // wrote in SQL deliberately, so they are reported, never dropped.
309
+ if (kept.length > 0) {
310
+ logger.warn(chalk.yellow(" ⚠️ Policies in the database that no collection describes:"));
311
+ for (const p of kept) {
312
+ logger.warn(chalk.yellow(` • ${p.schema}.${p.table} → "${p.name}" (${p.command} TO ${p.roles.join(", ")})`));
313
+ }
314
+ logger.warn(chalk.yellow(" These still grant access. Drop them by hand if they are stale."));
315
+ }
316
+
317
+ // Missing/diverged are not push's to fix, but staying silent about
318
+ // them is how a database ends up not matching its config.
319
+ const remaining = { ...drift, orphaned: kept };
320
+ if (hasDrift(remaining) && (remaining.missing.length > 0 || remaining.diverged.length > 0)) {
321
+ logger.warn(chalk.yellow(" ⚠️ RLS policies do not match your collections:"));
322
+ logger.warn(formatPolicyDrift({ ...remaining, orphaned: [] }));
323
+ }
324
+ } finally {
325
+ await client.end();
326
+ }
327
+ } catch (err) {
328
+ logger.warn(chalk.yellow(` ⚠️ Could not reconcile RLS policies: ${err instanceof Error ? err.message : String(err)}`));
329
+ }
330
+ }
331
+
272
332
  async function branchCommand(rawArgs: string[]): Promise<void> {
273
333
  const branchAction = rawArgs[2]; // create, list, delete, info
274
334
 
@@ -24,8 +24,14 @@
24
24
  export const AUTH_BOOTSTRAP_SQL = `-- Auth schema + RLS helper functions (required by the policies below)
25
25
  CREATE SCHEMA IF NOT EXISTS auth;
26
26
 
27
+ -- Falls back to the pre-rename \`app.user_id\` so a database that has taken the
28
+ -- new schema but is still served by an older backend keeps resolving the
29
+ -- principal. Drop the COALESCE once no such deployment remains.
27
30
  CREATE OR REPLACE FUNCTION auth.uid() RETURNS text AS $$
28
- SELECT NULLIF(current_setting('app.user_id', true), '');
31
+ SELECT COALESCE(
32
+ NULLIF(current_setting('app.uid', true), ''),
33
+ NULLIF(current_setting('app.user_id', true), '')
34
+ );
29
35
  $$ LANGUAGE sql STABLE;
30
36
 
31
37
  CREATE OR REPLACE FUNCTION auth.jwt() RETURNS jsonb AS $$
@@ -35,14 +35,14 @@ export function createAuthSchema(usersSchemaName = "rebase") {
35
35
  */
36
36
  const refreshTokens = tableCreator("refresh_tokens", {
37
37
  id: uuid("id").defaultRandom().primaryKey(),
38
- userId: uuid("user_id").notNull().references(() => users.id, { onDelete: "cascade" }),
38
+ uid: uuid("uid").notNull().references(() => users.id, { onDelete: "cascade" }),
39
39
  tokenHash: varchar("token_hash", { length: 255 }).notNull().unique(),
40
40
  expiresAt: timestamp("expires_at").notNull(),
41
41
  userAgent: varchar("user_agent", { length: 500 }),
42
42
  ipAddress: varchar("ip_address", { length: 45 }),
43
43
  createdAt: timestamp("created_at").defaultNow().notNull()
44
44
  }, (table) => ({
45
- uniqueDeviceSession: unique("unique_device_session").on(table.userId, table.userAgent, table.ipAddress)
45
+ uniqueDeviceSession: unique("unique_device_session").on(table.uid, table.userAgent, table.ipAddress)
46
46
  }));
47
47
 
48
48
  /**
@@ -50,7 +50,7 @@ export function createAuthSchema(usersSchemaName = "rebase") {
50
50
  */
51
51
  const passwordResetTokens = tableCreator("password_reset_tokens", {
52
52
  id: uuid("id").defaultRandom().primaryKey(),
53
- userId: uuid("user_id").notNull().references(() => users.id, { onDelete: "cascade" }),
53
+ uid: uuid("uid").notNull().references(() => users.id, { onDelete: "cascade" }),
54
54
  tokenHash: varchar("token_hash", { length: 255 }).notNull().unique(),
55
55
  expiresAt: timestamp("expires_at").notNull(),
56
56
  usedAt: timestamp("used_at"),
@@ -71,7 +71,7 @@ export function createAuthSchema(usersSchemaName = "rebase") {
71
71
  */
72
72
  const userIdentities = tableCreator("user_identities", {
73
73
  id: uuid("id").defaultRandom().primaryKey(),
74
- userId: uuid("user_id").notNull().references(() => users.id, { onDelete: "cascade" }),
74
+ uid: uuid("uid").notNull().references(() => users.id, { onDelete: "cascade" }),
75
75
  provider: varchar("provider", { length: 50 }).notNull(), // e.g. 'google', 'linkedin'
76
76
  providerId: varchar("provider_id", { length: 255 }).notNull(),
77
77
  profileData: jsonb("profile_data"),
@@ -86,7 +86,7 @@ export function createAuthSchema(usersSchemaName = "rebase") {
86
86
  */
87
87
  const mfaFactors = tableCreator("mfa_factors", {
88
88
  id: uuid("id").defaultRandom().primaryKey(),
89
- userId: uuid("user_id").notNull().references(() => users.id, { onDelete: "cascade" }),
89
+ uid: uuid("uid").notNull().references(() => users.id, { onDelete: "cascade" }),
90
90
  factorType: varchar("factor_type", { length: 20 }).notNull(), // 'totp'
91
91
  secretEncrypted: varchar("secret_encrypted", { length: 500 }).notNull(),
92
92
  friendlyName: varchar("friendly_name", { length: 255 }),
@@ -112,7 +112,7 @@ export function createAuthSchema(usersSchemaName = "rebase") {
112
112
  */
113
113
  const recoveryCodes = tableCreator("recovery_codes", {
114
114
  id: uuid("id").defaultRandom().primaryKey(),
115
- userId: uuid("user_id").notNull().references(() => users.id, { onDelete: "cascade" }),
115
+ uid: uuid("uid").notNull().references(() => users.id, { onDelete: "cascade" }),
116
116
  codeHash: varchar("code_hash", { length: 255 }).notNull(),
117
117
  usedAt: timestamp("used_at"),
118
118
  createdAt: timestamp("created_at").defaultNow().notNull()
@@ -123,7 +123,7 @@ export function createAuthSchema(usersSchemaName = "rebase") {
123
123
  */
124
124
  const magicLinkTokens = tableCreator("magic_link_tokens", {
125
125
  id: uuid("id").defaultRandom().primaryKey(),
126
- userId: uuid("user_id").notNull().references(() => users.id, { onDelete: "cascade" }),
126
+ uid: uuid("uid").notNull().references(() => users.id, { onDelete: "cascade" }),
127
127
  tokenHash: varchar("token_hash", { length: 255 }).notNull().unique(),
128
128
  expiresAt: timestamp("expires_at").notNull(),
129
129
  usedAt: timestamp("used_at"),
@@ -171,28 +171,28 @@ export const usersRelations = relations(users, ({ many }) => ({
171
171
 
172
172
  export const refreshTokensRelations = relations(refreshTokens, ({ one }) => ({
173
173
  user: one(users, {
174
- fields: [refreshTokens.userId],
174
+ fields: [refreshTokens.uid],
175
175
  references: [users.id]
176
176
  })
177
177
  }));
178
178
 
179
179
  export const passwordResetTokensRelations = relations(passwordResetTokens, ({ one }) => ({
180
180
  user: one(users, {
181
- fields: [passwordResetTokens.userId],
181
+ fields: [passwordResetTokens.uid],
182
182
  references: [users.id]
183
183
  })
184
184
  }));
185
185
 
186
186
  export const userIdentitiesRelations = relations(userIdentities, ({ one }) => ({
187
187
  user: one(users, {
188
- fields: [userIdentities.userId],
188
+ fields: [userIdentities.uid],
189
189
  references: [users.id]
190
190
  })
191
191
  }));
192
192
 
193
193
  export const mfaFactorsRelations = relations(mfaFactors, ({ one, many }) => ({
194
194
  user: one(users, {
195
- fields: [mfaFactors.userId],
195
+ fields: [mfaFactors.uid],
196
196
  references: [users.id]
197
197
  }),
198
198
  challenges: many(mfaChallenges)
@@ -207,14 +207,14 @@ export const mfaChallengesRelations = relations(mfaChallenges, ({ one }) => ({
207
207
 
208
208
  export const recoveryCodesRelations = relations(recoveryCodes, ({ one }) => ({
209
209
  user: one(users, {
210
- fields: [recoveryCodes.userId],
210
+ fields: [recoveryCodes.uid],
211
211
  references: [users.id]
212
212
  })
213
213
  }));
214
214
 
215
215
  export const magicLinkTokensRelations = relations(magicLinkTokens, ({ one }) => ({
216
216
  user: one(users, {
217
- fields: [magicLinkTokens.userId],
217
+ fields: [magicLinkTokens.uid],
218
218
  references: [users.id]
219
219
  })
220
220
  }));
@@ -37,7 +37,7 @@ export type IssueSeverity = "error" | "warning" | "info";
37
37
 
38
38
  export interface DoctorIssue {
39
39
  severity: IssueSeverity;
40
- category: "missing_table" | "missing_column" | "type_mismatch" | "missing_constraint" | "schema_stale" | "missing_enum" | "enum_value_mismatch" | "missing_foreign_key" | "sdk_stale";
40
+ category: "missing_table" | "missing_column" | "type_mismatch" | "missing_constraint" | "schema_stale" | "missing_enum" | "enum_value_mismatch" | "missing_foreign_key" | "sdk_stale" | "sdk_not_generated";
41
41
  table?: string;
42
42
  column?: string;
43
43
  expected?: string;
@@ -61,19 +61,29 @@ export function getExpectedColumnType(prop: Property): string | null {
61
61
  const sp = prop as StringProperty;
62
62
  if (sp.enum) return "USER-DEFINED"; // pgEnum → USER-DEFINED in information_schema
63
63
  if ("isId" in sp && sp.isId === "uuid") return "uuid";
64
- if (sp.columnType === "text") return "text";
64
+ if (sp.columnType === "uuid") return "uuid";
65
+ // A markdown/multiline string compiles to `text`, not varchar — the
66
+ // generator treats those UI hints as column-type signals, so the
67
+ // expectation here must too or every such column reads as drift.
68
+ if (sp.columnType === "text" || sp.ui?.markdown || sp.ui?.multiline) return "text";
65
69
  if (sp.columnType === "char") return "character";
66
70
  return "character varying";
67
71
  }
68
72
  case "number": {
69
73
  const np = prop as NumberProperty;
70
- if (np.columnType === "double precision") return "double precision";
71
- if (np.columnType === "real") return "real";
72
- if (np.columnType === "bigint") return "bigint";
73
- if (np.columnType === "serial") return "integer"; // serial is integer under the hood
74
- if (np.columnType === "bigserial") return "bigint";
75
- if (np.columnType === "integer") return "integer";
76
- if (np.columnType === "numeric") return "numeric";
74
+ if (np.columnType) {
75
+ // The generator passes any columnType straight through to drizzle,
76
+ // so mirror that rather than enumerating a subset (which reported
77
+ // drift for anything unlisted, e.g. smallint). Serial types are
78
+ // integers with a sequence default; information_schema reports the
79
+ // underlying width.
80
+ const serialWidths: Record<string, string> = {
81
+ serial: "integer",
82
+ bigserial: "bigint",
83
+ smallserial: "smallint"
84
+ };
85
+ return serialWidths[np.columnType] ?? np.columnType;
86
+ }
77
87
  if (np.validation?.integer || ("isId" in np && np.isId)) return "integer";
78
88
  return "numeric";
79
89
  }
@@ -201,15 +211,18 @@ export async function checkCollectionsVsSdk(
201
211
  ): Promise<{ passed: boolean; issues: DoctorIssue[] }> {
202
212
  const issues: DoctorIssue[] = [];
203
213
 
204
- // Check if SDK file exists
214
+ // The typed SDK is opt-in — nothing in a scaffolded project imports it until
215
+ // you choose to. A project that never generated one isn't drifting, so report
216
+ // it as information rather than a warning; otherwise `doctor` can never come
217
+ // back clean on a fresh project and users learn to ignore its output.
205
218
  if (!fs.existsSync(sdkFilePath)) {
206
219
  issues.push({
207
- severity: "warning",
208
- category: "sdk_stale",
209
- message: `Generated SDK typedefs file does not exist at "${sdkFilePath}".`,
210
- fix: "Run `rebase generate-sdk`"
220
+ severity: "info",
221
+ category: "sdk_not_generated",
222
+ message: "Typed SDK not generated (optional).",
223
+ fix: "Run `rebase generate-sdk` if you want typed collection access"
211
224
  });
212
- return { passed: false,
225
+ return { passed: true,
213
226
  issues };
214
227
  }
215
228
 
@@ -631,11 +644,15 @@ export function renderReport(report: DoctorReport): void {
631
644
  }
632
645
 
633
646
  function renderPhase(label: string, passed: boolean, issues: DoctorIssue[]): void {
634
- if (passed) {
647
+ const errorCount = issues.filter((i) => i.severity === "error").length;
648
+ const warnCount = issues.filter((i) => i.severity === "warning").length;
649
+ const infoIssues = issues.filter((i) => i.severity === "info");
650
+
651
+ // Informational notes don't make a phase unhealthy, so key the header off
652
+ // real problems rather than `passed` alone.
653
+ if (errorCount === 0 && warnCount === 0) {
635
654
  logger.info(` ${chalk.green("✅")} ${label}: ${chalk.green("In sync")}`);
636
655
  } else {
637
- const errorCount = issues.filter((i) => i.severity === "error").length;
638
- const warnCount = issues.filter((i) => i.severity === "warning").length;
639
656
  const parts: string[] = [];
640
657
  if (errorCount > 0) parts.push(`${errorCount} error${errorCount > 1 ? "s" : ""}`);
641
658
  if (warnCount > 0) parts.push(`${warnCount} warning${warnCount > 1 ? "s" : ""}`);
@@ -643,7 +660,14 @@ function renderPhase(label: string, passed: boolean, issues: DoctorIssue[]): voi
643
660
  }
644
661
  logger.info("");
645
662
 
646
- for (const issue of issues) {
663
+ // Notes render as a quiet one-liner, not a full drift box.
664
+ for (const issue of infoIssues) {
665
+ const fixPart = issue.fix ? chalk.gray(` — ${issue.fix}`) : "";
666
+ logger.info(` ${chalk.gray("ℹ")} ${chalk.gray(issue.message)}${fixPart}`);
667
+ logger.info("");
668
+ }
669
+
670
+ for (const issue of issues.filter((i) => i.severity !== "info")) {
647
671
  const severityIcon = issue.severity === "error" ? chalk.red("✗") : chalk.yellow("⚠");
648
672
  const categoryLabel = formatCategory(issue.category);
649
673
  logger.info(` ${chalk.gray("┌─")} ${severityIcon} ${chalk.bold(categoryLabel)} ${chalk.gray("─".repeat(Math.max(0, 42 - categoryLabel.length)))}`);
@@ -674,7 +698,8 @@ function formatCategory(cat: DoctorIssue["category"]): string {
674
698
  missing_enum: "Missing Enum",
675
699
  enum_value_mismatch: "Enum Value Mismatch",
676
700
  missing_foreign_key: "Missing Foreign Key",
677
- sdk_stale: "Stale SDK Types"
701
+ sdk_stale: "Stale SDK Types",
702
+ sdk_not_generated: "SDK Types Not Generated"
678
703
  };
679
704
  return labels[cat];
680
705
  }
@@ -1,4 +1,4 @@
1
- import { humanize } from "./introspect-db-logic";
1
+ import { humanize } from "./introspect-db-naming";
2
2
 
3
3
  export interface InferenceResult {
4
4
  propType?: string; // If the inference changes the base type
@@ -7,6 +7,7 @@
7
7
  * and consumed directly by tests.
8
8
  */
9
9
  import { inferPropertyFromData } from "./introspect-db-inference";
10
+ import { humanize } from "./introspect-db-naming";
10
11
 
11
12
  // ── Typed interfaces for SQL query results ────────────────────────────
12
13
 
@@ -117,16 +118,6 @@ export function singularize(word: string): string {
117
118
  return word;
118
119
  }
119
120
 
120
- /**
121
- * Convert a snake_case name to a human-readable Title Case label.
122
- * e.g. "created_at" -> "Created At", "customer_id" -> "Customer Id"
123
- */
124
- export function humanize(snakeName: string): string {
125
- return snakeName
126
- .replace(/_/g, " ")
127
- .replace(/\b\w/g, (c) => c.toUpperCase());
128
- }
129
-
130
121
  /**
131
122
  * Convert a snake_case table name to a camelCase + "Collection" variable name.
132
123
  * e.g. "company_token" -> "companyTokenCollection"
@@ -0,0 +1,15 @@
1
+ /**
2
+ * Naming helpers shared by the introspection modules. These live apart from
3
+ * `introspect-db-logic.ts` because the inference pass needs them too, and
4
+ * importing them from there would close a cycle back through this module.
5
+ */
6
+
7
+ /**
8
+ * Convert a snake_case name to a human-readable Title Case label.
9
+ * e.g. "created_at" -> "Created At", "customer_id" -> "Customer Id"
10
+ */
11
+ export function humanize(snakeName: string): string {
12
+ return snakeName
13
+ .replace(/_/g, " ")
14
+ .replace(/\b\w/g, (c) => c.toUpperCase());
15
+ }
@@ -23,11 +23,11 @@ import {
23
23
  buildTablesMap,
24
24
  buildEnumMap,
25
25
  identifyJoinTables,
26
- humanize,
27
26
  singularize,
28
27
  mapPgType,
29
28
  getIconForTable
30
29
  } from "./introspect-db-logic";
30
+ import { humanize } from "./introspect-db-naming";
31
31
 
32
32
  export interface IntrospectedSchema {
33
33
  tablesMap: Map<string, TableMeta>;
@@ -1,7 +1,7 @@
1
1
  import { describe, expect, it } from "@jest/globals";
2
2
  import type { CollectionConfig } from "@rebasepro/types";
3
3
 
4
- import { checkPolicyDrift, parseExpectedPolicies, formatPolicyDrift, hasDrift, type PolicyRef, type Queryable } from "./policy-drift";
4
+ import { checkPolicyDrift, parseExpectedPolicies, formatPolicyDrift, hasDrift, dropOrphanedPolicies, isGeneratedPolicyName, type PolicyRef, type Queryable } from "./policy-drift";
5
5
  import { generatePostgresPoliciesDdl } from "../schema/generate-postgres-ddl-logic";
6
6
 
7
7
  function collection(slug: string): CollectionConfig {
@@ -154,3 +154,108 @@ describe("checkPolicyDrift", () => {
154
154
  expect(drift.missing.length).toBeGreaterThan(0);
155
155
  });
156
156
  });
157
+
158
+ describe("isGeneratedPolicyName", () => {
159
+ it("recognises the generator's own shape", () => {
160
+ expect(isGeneratedPolicyName("documents_insert_a1b2c3d", "documents")).toBe(true);
161
+ // One rule spanning several operations appends the operation index.
162
+ expect(isGeneratedPolicyName("documents_update_a1b2c3d_1", "documents")).toBe(true);
163
+ });
164
+
165
+ it("does not claim names a human could have written", () => {
166
+ expect(isGeneratedPolicyName("owner_access", "documents")).toBe(false);
167
+ expect(isGeneratedPolicyName("documents_default_admin_read", "documents")).toBe(false);
168
+ // Right shape, wrong table — belongs to something else.
169
+ expect(isGeneratedPolicyName("teams_insert_a1b2c3d", "documents")).toBe(false);
170
+ // A digest is 7 lowercase hex characters, nothing else.
171
+ expect(isGeneratedPolicyName("documents_insert_notahex", "documents")).toBe(false);
172
+ expect(isGeneratedPolicyName("documents_grant_a1b2c3d", "documents")).toBe(false);
173
+ });
174
+ });
175
+
176
+ describe("dropOrphanedPolicies", () => {
177
+ /** Records the DDL issued so the test can assert on what was dropped. */
178
+ function recordingDb(rows: Record<string, unknown>[]) {
179
+ const executed: string[] = [];
180
+ const db: Queryable = {
181
+ query: async (text: string) => {
182
+ if (!/^SELECT/i.test(text)) executed.push(text);
183
+ return { rows: rows as never[] };
184
+ }
185
+ };
186
+ return { db, executed };
187
+ }
188
+
189
+ it("drops the policy a rule edit superseded", async () => {
190
+ // The reported failure: tightening a rule renames its policy, and the
191
+ // permissive original stays live and keeps ORing itself back in.
192
+ const cols = [collection("documents")];
193
+ const expected = parseExpectedPolicies(generatePostgresPoliciesDdl(cols));
194
+ const stale = {
195
+ schemaname: "public", tablename: "documents", policyname: "documents_insert_dead1ee",
196
+ roles: ["public"], cmd: "INSERT", qual: null, with_check: "true"
197
+ };
198
+ const live = [...expected.map((p) => liveRow(p)), stale];
199
+
200
+ const drift = await checkPolicyDrift(dbWith(live), cols);
201
+ const { db, executed } = recordingDb(live);
202
+ const { dropped, kept } = await dropOrphanedPolicies(db, drift, cols);
203
+
204
+ expect(dropped).toHaveLength(1);
205
+ expect(dropped[0].name).toBe("documents_insert_dead1ee");
206
+ expect(kept).toHaveLength(0);
207
+ expect(executed).toEqual([
208
+ 'DROP POLICY IF EXISTS "documents_insert_dead1ee" ON "public"."documents"'
209
+ ]);
210
+ });
211
+
212
+ it("leaves a hand-written policy alone and reports it instead", async () => {
213
+ const cols = [collection("documents")];
214
+ const expected = parseExpectedPolicies(generatePostgresPoliciesDdl(cols));
215
+ const live = [
216
+ ...expected.map((p) => liveRow(p)),
217
+ { schemaname: "public", tablename: "documents", policyname: "ops_break_glass", roles: ["public"], cmd: "ALL", qual: "true", with_check: null }
218
+ ];
219
+
220
+ const drift = await checkPolicyDrift(dbWith(live), cols);
221
+ const { db, executed } = recordingDb(live);
222
+ const { dropped, kept } = await dropOrphanedPolicies(db, drift, cols);
223
+
224
+ expect(dropped).toHaveLength(0);
225
+ expect(kept.map((p) => p.name)).toEqual(["ops_break_glass"]);
226
+ expect(executed).toEqual([]);
227
+ });
228
+
229
+ it("never touches a table the collections do not describe", async () => {
230
+ // Another application sharing the schema owns this table; a
231
+ // generator-shaped name there is coincidence, not our leftover.
232
+ const cols = [collection("documents")];
233
+ const expected = parseExpectedPolicies(generatePostgresPoliciesDdl(cols));
234
+ const live = [
235
+ ...expected.map((p) => liveRow(p)),
236
+ { schemaname: "public", tablename: "legacy", policyname: "legacy_select_a1b2c3d", roles: ["public"], cmd: "SELECT", qual: "true", with_check: null }
237
+ ];
238
+
239
+ const drift = await checkPolicyDrift(dbWith(live), cols);
240
+ const { db, executed } = recordingDb(live);
241
+ const { dropped, kept } = await dropOrphanedPolicies(db, drift, cols);
242
+
243
+ expect(dropped).toHaveLength(0);
244
+ expect(kept.map((p) => p.name)).toEqual(["legacy_select_a1b2c3d"]);
245
+ expect(executed).toEqual([]);
246
+ });
247
+
248
+ it("does nothing when the database already matches", async () => {
249
+ const cols = [collection("documents")];
250
+ const expected = parseExpectedPolicies(generatePostgresPoliciesDdl(cols));
251
+ const live = expected.map((p) => liveRow(p));
252
+
253
+ const drift = await checkPolicyDrift(dbWith(live), cols);
254
+ const { db, executed } = recordingDb(live);
255
+ const { dropped, kept } = await dropOrphanedPolicies(db, drift, cols);
256
+
257
+ expect(dropped).toHaveLength(0);
258
+ expect(kept).toHaveLength(0);
259
+ expect(executed).toEqual([]);
260
+ });
261
+ });
@@ -192,6 +192,62 @@ export async function checkPolicyDrift(
192
192
  return drift;
193
193
  }
194
194
 
195
+ /**
196
+ * Does this name look like one the generator produced for this table?
197
+ *
198
+ * Unnamed rules compile to `<table>_<op>_<sha1[0:7]>` (plus `_<idx>` when one
199
+ * rule spans several operations), and the hash covers the rule's semantics — so
200
+ * *editing* a rule renames its policy. The policy under the old name is left
201
+ * behind by `db push`, which only DROPs the names it is about to CREATE, and
202
+ * Postgres ORs PERMISSIVE policies together: a superseded `USING (true)` keeps
203
+ * granting everything no matter how tight its replacement is.
204
+ *
205
+ * Matching the shape is what makes dropping them safe. A hand-written policy
206
+ * would have to collide with a 7-hex digest to be mistaken for generated one;
207
+ * a policy named anything else is left alone and merely reported, because a
208
+ * custom name is indistinguishable from one someone wrote in SQL on purpose.
209
+ */
210
+ export function isGeneratedPolicyName(name: string, table: string): boolean {
211
+ return new RegExp(`^${table.replace(/[.*+?^${}()|[\]\\]/g, "\\$&")}_(select|insert|update|delete|all)_[0-9a-f]{7}(_\\d+)?$`)
212
+ .test(name);
213
+ }
214
+
215
+ export interface OrphanCleanup {
216
+ /** Superseded generated policies that were dropped. */
217
+ dropped: PolicyRef[];
218
+ /** Orphans left in place because their names are not generator-shaped. */
219
+ kept: PolicyRef[];
220
+ }
221
+
222
+ /**
223
+ * Drop the policies an earlier push superseded but never removed.
224
+ *
225
+ * Only touches tables the collections describe — a table with no expected
226
+ * policy is not ours to reconcile, and scanning by schema alone would sweep up
227
+ * policies belonging to something else sharing the database.
228
+ */
229
+ export async function dropOrphanedPolicies(
230
+ client: Queryable,
231
+ drift: PolicyDrift,
232
+ collections: CollectionConfig[]
233
+ ): Promise<OrphanCleanup> {
234
+ const expected = parseExpectedPolicies(generatePostgresPoliciesDdl(collections));
235
+ const managed = new Set(expected.map((p) => `${p.schema}.${p.table}`));
236
+
237
+ const cleanup: OrphanCleanup = { dropped: [], kept: [] };
238
+ for (const p of drift.orphaned) {
239
+ if (!managed.has(`${p.schema}.${p.table}`) || !isGeneratedPolicyName(p.name, p.table)) {
240
+ cleanup.kept.push(p);
241
+ continue;
242
+ }
243
+ // Identifiers are quoted, and the name came from pg_policies rather than
244
+ // from user input, so it is already a valid identifier.
245
+ await client.query(`DROP POLICY IF EXISTS "${p.name}" ON "${p.schema}"."${p.table}"`);
246
+ cleanup.dropped.push(p);
247
+ }
248
+ return cleanup;
249
+ }
250
+
195
251
  export const hasDrift = (d: PolicyDrift): boolean =>
196
252
  d.missing.length > 0 || d.orphaned.length > 0 || d.diverged.length > 0;
197
253
 
@@ -59,7 +59,7 @@ export interface ConnectionPosture {
59
59
  }
60
60
 
61
61
  export interface AuthContext {
62
- userId: string;
62
+ uid: string;
63
63
  /** Raw roles as carried on the user (strings or `{ id }` objects). */
64
64
  roles: unknown[];
65
65
  }
@@ -200,7 +200,7 @@ export async function ensureAppRole(run: RawSqlRunner, schemas: string[]): Promi
200
200
  * SECURITY: this function is only ever called on the **user** path (the server
201
201
  * context uses the base/owner driver and never calls it). The default policies
202
202
  * treat `auth.uid() IS NULL` as the trusted server context, and `auth.uid()`
203
- * is `NULLIF(current_setting('app.user_id'), '')` — so an EMPTY user id would
203
+ * is `NULLIF(current_setting('app.uid'), '')` — so an EMPTY user id would
204
204
  * be read as NULL and silently escalate a user request to server privileges.
205
205
  * Coerce empty/blank ids to `ANONYMOUS_USER_ID` here, at the single chokepoint,
206
206
  * rather than trusting every caller (e.g. realtime subscription auth) to do it.
@@ -208,15 +208,21 @@ export async function ensureAppRole(run: RawSqlRunner, schemas: string[]): Promi
208
208
  * semantics: it is why `auth.uid() IS NOT NULL` is true for anonymous requests.
209
209
  */
210
210
  export async function applyAuthContext(tx: SqlTx, auth: AuthContext, userRole?: string): Promise<void> {
211
- const userId = typeof auth.userId === "string" && auth.userId.trim() !== "" ? auth.userId : ANONYMOUS_USER_ID;
211
+ const uid = typeof auth.uid === "string" && auth.uid.trim() !== "" ? auth.uid : ANONYMOUS_USER_ID;
212
212
  const normalizedRoles = auth.roles.map((r: unknown) =>
213
213
  typeof r === "string" ? r : (r as Record<string, unknown>)?.id ?? String(r)
214
214
  );
215
+ // `app.user_id` is the pre-rename spelling, still written because policies
216
+ // are data: a database provisioned before the rename holds rules compiled
217
+ // to `current_setting('app.user_id')`, and those predicates would evaluate
218
+ // to NULL — failing open or locking out — if we stopped setting it. Drop
219
+ // the alias only once no live database carries a legacy policy.
215
220
  await tx.execute(drizzleSql`
216
221
  SELECT
217
- set_config('app.user_id', ${userId}, true),
222
+ set_config('app.uid', ${uid}, true),
223
+ set_config('app.user_id', ${uid}, true),
218
224
  set_config('app.user_roles', ${normalizedRoles.join(",")}, true),
219
- set_config('app.jwt', ${JSON.stringify({ sub: userId, roles: auth.roles })}, true)
225
+ set_config('app.jwt', ${JSON.stringify({ sub: uid, roles: auth.roles })}, true)
220
226
  `);
221
227
  if (userRole) {
222
228
  await tx.execute(drizzleSql.raw(`SET LOCAL ROLE ${quoteIdent(userRole)}`));