@mikro-orm/entity-generator 7.1.0-dev.15 → 7.1.0-dev.17

package/SourceFile.d.ts

@@ -1,4 +1,4 @@
-import { type Dictionary, type EmbeddableOptions, type EntityMetadata, type EntityOptions, type EntityProperty, type GenerateOptions, type IndexOptions, type NamingStrategy, type OneToOneOptions, type Platform, type UniqueOptions } from '@mikro-orm/core';
+import { type Dictionary, type EmbeddableOptions, type EntityMetadata, type EntityOptions, type EntityPartitionBy, type EntityProperty, type GenerateOptions, type IndexOptions, type NamingStrategy, type OneToOneOptions, type Platform, type UniqueOptions } from '@mikro-orm/core';
 /**
  * @see https://github.com/tc39/proposal-regexp-unicode-property-escapes#other-examples
  */
@@ -30,6 +30,7 @@ export declare class SourceFile {
     protected serializeObject(options: {}, wordwrap?: number, spaces?: number, level?: number): string;
     protected serializeValue(val: unknown, wordwrap?: number, spaces?: number, level?: number): unknown;
     protected getEntityDeclOptions(): EntityOptions<unknown>;
+    protected getPartitionByDecl(partitionBy: EntityPartitionBy): Dictionary;
     protected getEmbeddableDeclOptions(): EmbeddableOptions<unknown>;
     private getCollectionDecl;
     private getPropertyDecorator;

package/SourceFile.js

@@ -481,6 +481,9 @@ export class SourceFile {
         if (this.meta.comment) {
             options.comment = this.quote(this.meta.comment);
         }
+        if (this.meta.partitionBy) {
+            options.partitionBy = this.getPartitionByDecl(this.meta.partitionBy);
+        }
         if (this.meta.readonly && !this.meta.virtual) {
             options.readonly = this.meta.readonly;
         }
@@ -489,6 +492,40 @@ export class SourceFile {
         }
         return this.getCollectionDecl(options);
     }
+    getPartitionByDecl(partitionBy) {
+        const result = { type: this.quote(partitionBy.type) };
+        // Introspected metadata from `toEntityPartitionBy` always emits a string or string[];
+        // callback-form expressions only exist in hand-written entity metadata. Fail loud if a
+        // callback ever reaches the generator — stringifying `fn.toString()` would produce source
+        // that re-imports nothing and will not compile.
+        if (typeof partitionBy.expression === 'function') {
+            throw new Error(`Cannot emit entity source for ${this.meta.className}: partitionBy.expression is a callback. ` +
+                `Entity generator expects string or string[] expressions from catalog introspection.`);
+        }
+        const expression = partitionBy.expression;
+        if (Array.isArray(expression)) {
+            result.expression = expression.map(key => this.quote(String(key)));
+        }
+        else {
+            result.expression = this.quote(String(expression));
+        }
+        if (partitionBy.type === 'hash') {
+            result.partitions = Array.isArray(partitionBy.partitions)
+                ? partitionBy.partitions.map(name => this.quote(String(name)))
+                : partitionBy.partitions;
+        }
+        else {
+            result.partitions = partitionBy.partitions.map(partition => {
+                const entry = {};
+                if (partition.name) {
+                    entry.name = this.quote(partition.name);
+                }
+                entry.values = this.quote(partition.values);
+                return entry;
+            });
+        }
+        return result;
+    }
     getEmbeddableDeclOptions() {
         const options = {};
         const result = this.getCollectionDecl(options);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mikro-orm/entity-generator",
-  "version": "7.1.0-dev.15",
+  "version": "7.1.0-dev.17",
   "description": "TypeScript ORM for Node.js based on Data Mapper, Unit of Work and Identity Map patterns. Supports MongoDB, MySQL, PostgreSQL and SQLite databases as well as usage with vanilla JavaScript.",
   "keywords": [
     "data-mapper",
@@ -47,13 +47,13 @@
     "copy": "node ../../scripts/copy.mjs"
   },
   "dependencies": {
-    "@mikro-orm/sql": "7.1.0-dev.15"
+    "@mikro-orm/sql": "7.1.0-dev.17"
   },
   "devDependencies": {
     "@mikro-orm/core": "^7.0.11"
   },
   "peerDependencies": {
-    "@mikro-orm/core": "7.1.0-dev.15"
+    "@mikro-orm/core": "7.1.0-dev.17"
   },
   "engines": {
     "node": ">= 22.17.0"