@vrplatform/kysely 1.3.1 → 1.3.3

package/src/pganalyze-comment-plugin/index.ts

@@ -0,0 +1,27 @@
+import type {
+  KyselyPlugin,
+  PluginTransformQueryArgs,
+  PluginTransformResultArgs,
+  QueryResult,
+  RootOperationNode,
+  UnknownRow,
+} from 'kysely';
+import { CommentTransformer } from './comment-transformer';
+
+export class KyselyPganalyzeCommentPlugin implements KyselyPlugin {
+  readonly #transformer: CommentTransformer;
+
+  constructor(options: { repository: string }) {
+    this.#transformer = new CommentTransformer(options.repository);
+  }
+
+  transformQuery(args: PluginTransformQueryArgs): RootOperationNode {
+    return this.#transformer.transformRootNode(args.node);
+  }
+
+  async transformResult(
+    args: PluginTransformResultArgs
+  ): Promise<QueryResult<UnknownRow>> {
+    return args.result;
+  }
+}

package/src/test/index.ts

@@ -0,0 +1,63 @@
+import { afterAll } from 'bun:test';
+import type { Kysely } from 'kysely';
+import { type UseLocalKyselyOptions, useLocalKysely } from '../local';
+import type { DB } from '../v1.generated';
+
+/**
+ * @deprecated Use `useLocalKyselyResource` instead.
+ */
+export async function useTestKysely(
+  cleanupStrategy: 'afterAll',
+  args?: UseLocalKyselyOptions
+) {
+  const kysely = await useLocalKysely(args);
+
+  if (cleanupStrategy === 'afterAll')
+    afterAll(async () => {
+      await kysely.destroy();
+    });
+
+  return {
+    kysely,
+  };
+}
+
+export async function useTestAutoCleanupKysely(args?: UseLocalKyselyOptions) {
+  const kysely = await useLocalKysely(args);
+
+  afterAll(async () => {
+    await kysely.destroy();
+  });
+
+  return kysely;
+}
+
+type ReturnType = Kysely<DB> & {
+  [Symbol.dispose]: () => Promise<void>;
+};
+
+export async function useLocalKyselyResource(
+  args?: UseLocalKyselyOptions
+): Promise<ReturnType> {
+  const kysely = await useLocalKysely(args);
+
+  return new Proxy(
+    {},
+    {
+      get(_target, prop) {
+        if (prop === Symbol.dispose) {
+          return async () => {
+            await kysely.destroy();
+          };
+        }
+        if (prop === 'kysely') {
+          return kysely;
+        }
+        if (typeof kysely[prop] === 'function') {
+          return kysely[prop].bind(kysely);
+        }
+        return kysely[prop];
+      },
+    }
+  ) as ReturnType;
+}

package/src/v1.generated.ts

@@ -1342,6 +1342,7 @@ export interface PublicReservation {
   lineMap: Json | null;
   listingConnectionId: string | null;
   listingId: string | null;
+  matchPaymentCustomerRef: string | null;
   metadata: Json | null;
   nights: number | null;
   otaReservationId: string | null;

package/src/test.ts

@@ -1,19 +0,0 @@
-import { afterAll } from 'bun:test';
-import { useLocalKysely } from './local';
-
-export async function useTestKysely(
-  cleanupStrategy: 'afterAll' | 'using' | 'manual'
-) {
-  const kysely = await useLocalKysely();
-
-  if (cleanupStrategy === 'afterAll')
-    afterAll(async () => {
-      await kysely.destroy();
-    });
-  return {
-    kysely,
-    [Symbol.dispose]: async () => {
-      await kysely.destroy();
-    },
-  };
-}