@tstdl/base 0.92.79 → 0.92.80

package/orm/decorators.d.ts

@@ -1,12 +1,19 @@
+import type { SQL } from 'drizzle-orm';
 import type { LiteralUnion } from 'type-fest';
 import type { AbstractConstructor, TypedOmit } from '../types.js';
-import type { EntityType } from './entity.js';
+import type { Entity, EntityType } from './entity.js';
+import type { PgTableFromType } from './server/types.js';
 type IndexMethod = LiteralUnion<'hash' | 'btree' | 'gist' | 'spgist' | 'gin' | 'brin' | 'hnsw' | 'ivfflat', string>;
+export type CheckBuilder<T extends Entity = any> = (table: PgTableFromType<string, EntityType<T>>) => SQL;
 export type OrmTableReflectionData = {
     name?: string;
     schema?: string;
     unique?: UniqueReflectionData[];
     index?: IndexReflectionData[];
+    checks?: {
+        name: string;
+        builder: CheckBuilder;
+    }[];
 };
 export type OrmColumnReflectionData = {
     name?: string;
@@ -46,6 +53,7 @@ export declare function createTableAndColumnDecorator(data?: OrmColumnReflection
 export declare function Column(options: OrmColumnReflectionData): PropertyDecorator;
 export declare function PrimaryKey(): PropertyDecorator;
 export declare function References(type: () => EntityType): PropertyDecorator;
+export declare function Check<T extends Entity>(name: string, builder: CheckBuilder<T>): ClassDecorator;
 export declare function Encrypted(): PropertyDecorator;
 export declare function Embedded(type: AbstractConstructor, options?: TypedOmit<NonNullable<OrmColumnReflectionData['embedded']>, 'type'>): PropertyDecorator;
 type TableOptions = Partial<Pick<OrmTableReflectionData, 'name' | 'schema'>>;

package/orm/decorators.js

@@ -1,4 +1,4 @@
-import { createClassDecorator, createDecorator, createPropertyDecorator } from '../reflection/utils.js';
+import { createClassDecorator, createDecorator, createPropertyDecorator } from '../reflection/index.js';
 import { Property } from '../schema/index.js';
 import { assertNotArrayPass, isArray, isString } from '../utils/type-guards.js';
 export function createTableDecorator(data) {
@@ -19,6 +19,15 @@ export function PrimaryKey() {
 export function References(type) {
     return createColumnDecorator({ references: type });
 }
+export function Check(name, builder) {
+    return createClassDecorator({
+        handler: (_, metadata) => {
+            const checks = metadata.data.tryGet('orm')?.checks ?? [];
+            checks.push({ name, builder });
+            metadata.data.set('orm', { checks }, true);
+        }
+    });
+}
 export function Encrypted() {
     return createColumnDecorator({ encrypted: true });
 }

package/orm/server/drizzle/schema-converter.js

@@ -1,5 +1,5 @@
 import { toCamelCase, toSnakeCase } from 'drizzle-orm/casing';
-import { boolean, doublePrecision, index, integer, jsonb, pgSchema, primaryKey, text, unique, uniqueIndex, uuid } from 'drizzle-orm/pg-core';
+import { boolean, check, doublePrecision, index, integer, jsonb, pgSchema, primaryKey, text, unique, uniqueIndex, uuid } from 'drizzle-orm/pg-core';
 import { MultiKeyMap } from '../../../data-structures/multi-key-map.js';
 import { tryGetEnumName } from '../../../enumeration/enumeration.js';
 import { NotSupportedError } from '../../../errors/not-supported.error.js';
@@ -76,7 +76,8 @@ export function _getDrizzleTableFromType(type, schemaName) {
             }
             return constraint;
         }) ?? []),
-        ...(tableReflectionData?.index?.map((data) => buildIndex(table, data)) ?? [])
+        ...(tableReflectionData?.index?.map((data) => buildIndex(table, data)) ?? []),
+        ...(tableReflectionData?.checks?.map((data) => check(data.name, data.builder(table))) ?? [])
     ]);
     drizzleSchema[columnDefinitionsSymbol] = columnDefinitions;
     return drizzleSchema;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tstdl/base",
-  "version": "0.92.79",
+  "version": "0.92.80",
   "author": "Patrick Hein",
   "publishConfig": {
     "access": "public"
@@ -145,7 +145,7 @@
     "@types/node": "22",
     "@types/nodemailer": "6.4",
     "@types/pg": "8.11",
-    "@typescript-eslint/eslint-plugin": "8.24",
+    "@typescript-eslint/eslint-plugin": "8.25",
     "concurrently": "9.1",
     "drizzle-kit": "0.30",
     "eslint": "9.21",

package/reflection/utils.js

@@ -3,9 +3,7 @@ import { noop } from '../utils/noop.js';
 import { assert, isDefined, isFunction, isSymbol } from '../utils/type-guards.js';
 import { getDecoratorData } from './decorator-data.js';
 import { reflectionRegistry } from './registry.js';
-// eslint-disable-next-line max-lines-per-function
 export function createDecorator(options, handler = noop) {
-    // eslint-disable-next-line max-statements, max-lines-per-function
     function decoratorWrapper(...args) {
         const data = getDecoratorData(...args);
         const optionsType = (data.type == 'constructor-parameter') ? 'constructorParameter'