@vrplatform/kysely 1.3.17 → 1.3.21

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

@@ -1,117 +0,0 @@
-import {
-  type DeleteQueryNode,
-  type InsertQueryNode,
-  OperationNodeTransformer,
-  type RootOperationNode,
-  SelectModifierNode,
-  type SelectQueryNode,
-  sql,
-  type UpdateQueryNode,
-} from 'kysely';
-
-function createCommentNode(repositoryName: string): SelectModifierNode {
-  return SelectModifierNode.createWithExpression(
-    sql.raw(`/* repository: ${repositoryName} */`).toOperationNode()
-  );
-}
-/**
- * Transformer that adds a SQL comment to queries for pganalyze tracking.
- * The comment format is: repository: REPOSITORY_NAME
- *
- * This works by appending the comment to queries using the endModifiers property,
- * similar to how modifyEnd works.
- */
-export class CommentTransformer extends OperationNodeTransformer {
-  constructor(readonly _repositoryName: string) {
-    super();
-  }
-
-  transformRootNode(node: RootOperationNode): RootOperationNode {
-    if (node.kind === 'SelectQueryNode') {
-      return this.transformSelectQuery(node);
-    }
-    if (node.kind === 'InsertQueryNode') {
-      return this.transformInsertQuery(node);
-    }
-    if (node.kind === 'UpdateQueryNode') {
-      return this.transformUpdateQuery(node);
-    }
-    if (node.kind === 'DeleteQueryNode') {
-      return this.transformDeleteQuery(node);
-    }
-    return node;
-  }
-
-  protected override transformSelectQuery(
-    node: SelectQueryNode
-  ): SelectQueryNode {
-    const transformed = super.transformSelectQuery(node);
-    return this.addCommentToSelectQuery(transformed);
-  }
-
-  protected override transformInsertQuery(
-    node: InsertQueryNode
-  ): InsertQueryNode {
-    const transformed = super.transformInsertQuery(node);
-    return this.addCommentToInsertQuery(transformed);
-  }
-
-  protected override transformUpdateQuery(
-    node: UpdateQueryNode
-  ): UpdateQueryNode {
-    const transformed = super.transformUpdateQuery(node);
-    return this.addCommentToUpdateQuery(transformed);
-  }
-
-  protected override transformDeleteQuery(
-    node: DeleteQueryNode
-  ): DeleteQueryNode {
-    const transformed = super.transformDeleteQuery(node);
-    return this.addCommentToDeleteQuery(transformed);
-  }
-
-  private addCommentToSelectQuery(node: SelectQueryNode): SelectQueryNode {
-    // Append comment using endModifiers property, similar to modifyEnd
-    const commentNode = createCommentNode(this._repositoryName);
-
-    return {
-      ...node,
-      endModifiers: node.endModifiers
-        ? [...node.endModifiers, commentNode]
-        : [commentNode],
-    };
-  }
-
-  private addCommentToInsertQuery(node: InsertQueryNode): InsertQueryNode {
-    const commentNode = createCommentNode(this._repositoryName);
-
-    return {
-      ...node,
-      endModifiers: node.endModifiers
-        ? [...node.endModifiers, commentNode]
-        : [commentNode],
-    };
-  }
-
-  private addCommentToUpdateQuery(node: UpdateQueryNode): UpdateQueryNode {
-    const commentNode = createCommentNode(this._repositoryName);
-
-    return {
-      ...node,
-      endModifiers: node.endModifiers
-        ? [...node.endModifiers, commentNode]
-        : [commentNode],
-    };
-  }
-
-  private addCommentToDeleteQuery(node: DeleteQueryNode): DeleteQueryNode {
-    const commentNode = createCommentNode(this._repositoryName);
-
-    return {
-      ...node,
-      endModifiers: node.endModifiers
-        ? [...node.endModifiers, commentNode]
-        : [commentNode],
-    };
-  }
-}

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

@@ -1,27 +0,0 @@
-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/plugins.ts

@@ -1,30 +0,0 @@
-import {
-  type Dialect,
-  Kysely as K,
-  type KyselyPlugin,
-  type LogConfig,
-} from 'kysely';
-import { CustomCasePlugin } from './camel-case';
-import { KyselyPganalyzeCommentPlugin } from './pganalyze-comment-plugin';
-import type { DB } from './v1.generated';
-
-export function getKysely(
-  dialect: Dialect,
-  options: { repositoryName?: string; log?: true | LogConfig }
-) {
-  const plugins: KyselyPlugin[] = [new CustomCasePlugin()];
-
-  if (options?.repositoryName) {
-    plugins.push(
-      new KyselyPganalyzeCommentPlugin({
-        repository: options.repositoryName,
-      })
-    );
-  }
-  const kysely = new K<DB>({
-    dialect,
-    plugins,
-    log: options?.log === true ? ['query', 'error'] : options?.log,
-  });
-  return kysely;
-}

package/src/test/index.ts

@@ -1,64 +0,0 @@
-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) {
-        // somehow not working if using Symbol.dispose
-        if (prop === Symbol.asyncDispose) {
-          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;
-}