Firebase Data Source Driver

1import {
2  getDataValue,
3  queryLocalRows,
4  type DataActionInput,
5  type DataActionResult,
6  type DataFilter,
7  type DataQuery,
8  type DataQueryResult,
9  type DataRow,
10  type DataSourceContext,
11  type DataSourceDriver,
12  type ResourceConfig,
13  type ResourceRuntime,
14} from "@/lib/fable-ui/core"
15import type {
16  FirebaseAuthLike,
17  FirebaseDriverConfig,
18  FirestorePathParam,
19  FirestoreResourceSource,
20  FirestoreSdk,
21} from "./firebase-driver.types"
22 
23async function loadFirestoreSdk(): Promise<FirestoreSdk> {
24  const moduleName = "firebase/firestore"
25  return (await import(moduleName)) as FirestoreSdk
26}
27 
28function mapFirestoreError(error: unknown) {
29  const message = error instanceof Error ? error.message : String(error || "")
30  const code =
31    typeof error === "object" && error && "code" in error
32      ? String((error as { code?: unknown }).code)
33      : ""
34 
35  if (/permission-denied|permission|forbidden/i.test(`${code} ${message}`)) {
36    return new Error("Firestore permission denied. Check authentication and security rules.")
37  }
38 
39  if (/failed-precondition|index/i.test(`${code} ${message}`)) {
40    return new Error("Firestore query requires an index. Check the Firebase console index suggestion.")
41  }
42 
43  if (/invalid-argument|invalid/i.test(`${code} ${message}`)) {
44    return new Error(`Firestore query is invalid: ${message}`)
45  }
46 
47  return error instanceof Error ? error : new Error("Firestore request failed.")
48}
49 
50function assertSafePathParam(key: string, value: string) {
51  if (!value || value.includes("/") || value.includes("..")) {
52    throw new Error(`Invalid Firestore path parameter "${key}".`)
53  }
54 
55  return value
56}
57 
58async function resolvePathParam(key: string, param: FirestorePathParam, ctx: DataSourceContext) {
59  const value = typeof param === "function" ? await param(ctx) : param
60  return assertSafePathParam(key, String(value))
61}
62 
63async function resolveCollectionPath(source: FirestoreResourceSource, ctx: DataSourceContext) {
64  const params: Record<string, string> = {}
65 
66  if (ctx.orgId) {
67    params.orgId = assertSafePathParam("orgId", String(ctx.orgId))
68  }
69 
70  if (ctx.tenantId) {
71    params.tenantId = assertSafePathParam("tenantId", String(ctx.tenantId))
72  }
73 
74  for (const [key, param] of Object.entries(source.pathParams ?? {})) {
75    params[key] = await resolvePathParam(key, param, ctx)
76  }
77 
78  return source.collection.replace(/\{([^}]+)\}/g, (_match, key: string) => {
79    const value = params[key]
80 
81    if (!value) {
82// … truncated