Exported missing pongoSchema

Author: oskardudycz

src/package-lock.json

@@ -1,6 +1,6 @@
 {
   "name": "@event-driven-io/pongo-core",
-  "version": "0.14.1",
+  "version": "0.14.2",
   "description": "Pongo - Mongo with strong consistency on top of Postgres",
   "type": "module",
   "engines": {

src/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@event-driven-io/pongo",
-  "version": "0.14.1",
+  "version": "0.14.2",
   "description": "Pongo - Mongo with strong consistency on top of Postgres",
   "type": "module",
   "scripts": {

src/packages/pongo/package.json

@@ -1,10 +1,11 @@
 import { Command } from 'commander';
 import fs from 'node:fs';
-import { objectEntries, type PongoSchemaConfig } from '../core';
 import {
+  objectEntries,
   toDbSchemaMetadata,
   type PongoDbSchemaMetadata,
-} from '../core/typing/schema';
+  type PongoSchemaConfig,
+} from '../core';
 
 const formatTypeName = (input: string): string => {
   if (input.length === 0) {

src/packages/pongo/src/commandLine/configFile.ts

@@ -7,12 +7,12 @@ import pg from 'pg';
 import type { PostgresDbClientOptions } from '../postgres';
 import { getPongoDb, type AllowedDbClientOptions } from './pongoDb';
 import { pongoSession } from './pongoSession';
-import type { PongoClient, PongoDb, PongoSession } from './typing/operations';
 import {
   proxyClientWithSchema,
   type PongoClientSchema,
   type PongoClientWithSchema,
-} from './typing/schema';
+} from './schema';
+import type { PongoClient, PongoDb, PongoSession } from './typing';
 
 export type PooledPongoClientOptions =
   | {

src/packages/pongo/src/core/pongoClient.ts

@@ -1,4 +1,191 @@
-import type { PongoClientSchema } from '../typing/schema';
+import {
+  type Document,
+  type PongoClient,
+  type PongoCollection,
+  type PongoDb,
+  type PongoDocument,
+  objectEntries,
+} from '../typing';
+
+export interface PongoCollectionSchema<
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
+  T extends PongoDocument = PongoDocument,
+> {
+  name: string;
+}
+
+// Database schema interface
+export interface PongoDbSchema<
+  T extends Record<string, PongoCollectionSchema> = Record<
+    string,
+    PongoCollectionSchema
+  >,
+> {
+  name?: string;
+  collections: T;
+}
+
+export interface PongoClientSchema<
+  T extends Record<string, PongoDbSchema> = Record<string, PongoDbSchema>,
+> {
+  dbs: T;
+}
+
+export type CollectionsMap<T extends Record<string, PongoCollectionSchema>> = {
+  [K in keyof T]: PongoCollection<
+    T[K] extends PongoCollectionSchema<infer U> ? U : PongoDocument
+  >;
+};
+
+export type PongoDbWithSchema<
+  T extends Record<string, PongoCollectionSchema>,
+  ConnectorType extends string = string,
+> = CollectionsMap<T> & PongoDb<ConnectorType>;
+
+export type DBsMap<T extends Record<string, PongoDbSchema>> = {
+  [K in keyof T]: CollectionsMap<T[K]['collections']>;
+};
+
+export type PongoClientWithSchema<T extends PongoClientSchema> = DBsMap<
+  T['dbs']
+> &
+  PongoClient;
+
+const pongoCollectionSchema = <T extends PongoDocument>(
+  name: string,
+): PongoCollectionSchema<T> => ({
+  name,
+});
+
+function pongoDbSchema<T extends Record<string, PongoCollectionSchema>>(
+  collections: T,
+): PongoDbSchema<T>;
+function pongoDbSchema<T extends Record<string, PongoCollectionSchema>>(
+  name: string,
+  collections: T,
+): PongoDbSchema<T>;
+function pongoDbSchema<T extends Record<string, PongoCollectionSchema>>(
+  nameOrCollections: string | T,
+  collections?: T | undefined,
+): PongoDbSchema<T> {
+  if (collections === undefined) {
+    if (typeof nameOrCollections === 'string') {
+      throw new Error('You need to provide colleciton definition');
+    }
+    return {
+      collections: nameOrCollections,
+    };
+  }
+
+  return nameOrCollections && typeof nameOrCollections === 'string'
+    ? {
+        name: nameOrCollections,
+        collections,
+      }
+    : { collections: collections };
+}
+
+const pongoClientSchema = <T extends Record<string, PongoDbSchema>>(
+  dbs: T,
+): PongoClientSchema<T> => ({
+  dbs,
+});
+
+export const pongoSchema = {
+  client: pongoClientSchema,
+  db: pongoDbSchema,
+  collection: pongoCollectionSchema,
+};
+
+// Factory function to create DB instances
+export const proxyPongoDbWithSchema = <
+  T extends Record<string, PongoCollectionSchema>,
+  ConnectorType extends string = string,
+>(
+  pongoDb: PongoDb<ConnectorType>,
+  dbSchema: PongoDbSchema<T>,
+  collections: Map<string, PongoCollection<Document>>,
+): PongoDbWithSchema<T, ConnectorType> => {
+  const collectionNames = Object.keys(dbSchema.collections);
+
+  for (const collectionName of collectionNames) {
+    collections.set(collectionName, pongoDb.collection(collectionName));
+  }
+
+  return new Proxy(
+    pongoDb as PongoDb<ConnectorType> & {
+      [key: string]: unknown;
+    },
+    {
+      get(target, prop: string) {
+        return collections.get(prop) ?? target[prop];
+      },
+    },
+  ) as PongoDbWithSchema<T, ConnectorType>;
+};
+
+// Factory function to create Client instances
+export const proxyClientWithSchema = <
+  TypedClientSchema extends PongoClientSchema,
+>(
+  client: PongoClient,
+  schema: TypedClientSchema | undefined,
+): PongoClientWithSchema<TypedClientSchema> => {
+  if (!schema) return client as PongoClientWithSchema<TypedClientSchema>;
+
+  const dbNames = Object.keys(schema.dbs);
+
+  return new Proxy(
+    client as PongoClient & {
+      [key: string]: unknown;
+    },
+    {
+      get(target, prop: string) {
+        if (dbNames.includes(prop)) return client.db(schema.dbs[prop]?.name);
+
+        return target[prop];
+      },
+    },
+  ) as PongoClientWithSchema<TypedClientSchema>;
+};
+
+export type PongoCollectionSchemaMetadata = {
+  name: string;
+};
+
+export type PongoDbSchemaMetadata = {
+  name?: string | undefined;
+  collections: PongoCollectionSchemaMetadata[];
+};
+
+export type PongoClientSchemaMetadata = {
+  databases: PongoDbSchemaMetadata[];
+  database: (name?: string) => PongoDbSchemaMetadata | undefined;
+};
+
+export const toDbSchemaMetadata = <TypedDbSchema extends PongoDbSchema>(
+  schema: TypedDbSchema,
+): PongoDbSchemaMetadata => ({
+  name: schema.name,
+  collections: objectEntries(schema.collections).map((c) => ({
+    name: c[1].name,
+  })),
+});
+
+export const toClientSchemaMetadata = <
+  TypedClientSchema extends PongoClientSchema,
+>(
+  schema: TypedClientSchema,
+): PongoClientSchemaMetadata => {
+  const databases = objectEntries(schema.dbs).map((e) =>
+    toDbSchemaMetadata(e[1]),
+  );
+
+  return {
+    databases,
+    database: (name) => databases.find((db) => db.name === name),
+  };
+};
 
 export interface PongoSchemaConfig<
   TypedClientSchema extends PongoClientSchema = PongoClientSchema,