@aurios/mizzle 1.1.4 → 1.1.5

package/.turbo/turbo-build.log

@@ -1,5 +1,5 @@
 
-> @aurios/mizzle@1.1.4 build /home/runner/work/mizzle/mizzle/packages/mizzle
+> @aurios/mizzle@1.1.5 build /home/runner/work/mizzle/mizzle/packages/mizzle
 > tsup
 
 CLI Building entry: {"index":"src/index.ts","columns":"src/columns/index.ts","table":"src/core/table.ts","snapshot":"src/core/snapshot.ts","diff":"src/core/diff.ts","introspection":"src/core/introspection.ts","db":"src/db.ts"}
@@ -11,21 +11,21 @@
 ESM Build start
 ESM dist/index.js                2.32 KB
 ESM dist/columns.js              299.00 B
-ESM dist/snapshot.js             193.00 B
 ESM dist/table.js                144.00 B
-ESM dist/chunk-TOYV2M4M.js       866.00 B
+ESM dist/snapshot.js             193.00 B
 ESM dist/diff.js                 147.00 B
-ESM dist/introspection.js        1.47 KB
+ESM dist/chunk-TOYV2M4M.js       866.00 B
 ESM dist/chunk-AQVECMXP.js       2.38 KB
+ESM dist/introspection.js        1.47 KB
 ESM dist/db.js                   166.00 B
+ESM dist/chunk-WTOBEKFQ.js       9.74 KB
 ESM dist/transaction-RE7LXTGV.js 193.00 B
-ESM dist/chunk-NPPZW6VT.js       9.59 KB
 ESM dist/chunk-UM3YF5EC.js       15.21 KB
 ESM dist/chunk-GPYZK4WY.js       1.12 KB
 ESM dist/chunk-DU7UPWBW.js       5.17 KB
-ESM ⚡️ Build success in 45ms
+ESM ⚡️ Build success in 76ms
 DTS Build start
-DTS ⚡️ Build success in 3819ms
+DTS ⚡️ Build success in 6614ms
 DTS dist/index.d.ts              3.84 KB
 DTS dist/diff.d.ts               485.00 B
 DTS dist/introspection.d.ts      258.00 B
@@ -33,5 +33,5 @@
 DTS dist/db.d.ts                 169.00 B
 DTS dist/columns.d.ts            250.00 B
 DTS dist/table.d.ts              403.00 B
-DTS dist/db-zHIHBm1E.d.ts        31.72 KB
+DTS dist/db-ZX6HGN1x.d.ts        31.83 KB
 DTS dist/operators-BVreW0ky.d.ts 29.24 KB

package/dist/{chunk-NPPZW6VT.js → chunk-WTOBEKFQ.js}

@@ -1 +1 @@
-import{D as Q,E as U,I as L,J as w,K as H,L as $,N as J,P as X,Q as Z,S as ee,e as P,p as q,q as G,r as z}from"./chunk-UM3YF5EC.js";import{b as D}from"./chunk-GPYZK4WY.js";import{d as E,e as d,f as k,i as v}from"./chunk-DU7UPWBW.js";import{DynamoDBClient as ne}from"@aws-sdk/client-dynamodb";import{DynamoDBDocumentClient as le}from"@aws-sdk/lib-dynamodb";import{GetCommand as ie,QueryCommand as se,ScanCommand as oe}from"@aws-sdk/lib-dynamodb";var O=class{constructor(t,e){this.client=t;this.fields=e}from(t){return new I(t,this.client,this.fields)}},I=class extends w{constructor(e,i,n){super(e,i);this.fields=n}static[d.ENTITY_KIND]="SelectBase";_whereClause;_limitVal;_pageSizeVal;_consistentReadVal;_sortForward=!0;_forcedIndexName;where(e){return this._whereClause=e,this}limit(e){return this._limitVal=e,this}pageSize(e){return this._pageSizeVal=e,this}consistentRead(e=!0){return this._consistentReadVal=e,this}sort(e){return this._sortForward=e,this}index(e){return this._forcedIndexName=e,this}iterator(){return async function*(){let e,i=0;do{let n=await this.fetchPage(e);for(let s of n.items)if(yield s,i++,this._limitVal!==void 0&&i>=this._limitVal)return;e=n.lastEvaluatedKey}while(e)}.bind(this)()}async fetchPage(e){let i=this.resolveKeys(this._whereClause,void 0,this._forcedIndexName);return i.hasPartitionKey&&i.hasSortKey&&!i.indexName&&!e?{items:await this.executeGet(i.keys)}:i.hasPartitionKey||i.indexName?await this.executeQuery(i,e):await this.executeScan(e)}async execute(){let{items:e}=await this.fetchPage();return e}async executeGet(e){let i=new ie({TableName:this.tableName,Key:e,ConsistentRead:this._consistentReadVal}),n=await this.client.send(i);return n.Item?[this.mapToLogical(n.Item)]:[]}async executeQuery(e,i){let{expressionAttributeNames:n,expressionAttributeValues:s,addName:r,addValue:o}=this.createExpressionContext(),a=[],c=new Set;for(let[p,_]of Object.entries(e.keys))a.push(`${r(p)} = ${o(_)}`),c.add(p);let m=a.join(" AND "),u=this._whereClause?z(this._whereClause,r,o,c):void 0,f=new se({TableName:this.tableName,IndexName:e.indexName,KeyConditionExpression:m,FilterExpression:u||void 0,ExpressionAttributeNames:Object.keys(n).length>0?n:void 0,ExpressionAttributeValues:Object.keys(s).length>0?s:void 0,Limit:this._pageSizeVal??this._limitVal,ScanIndexForward:this._sortForward,ConsistentRead:e.indexName?void 0:this._consistentReadVal,ExclusiveStartKey:i}),T=await this.client.send(f);return{items:(T.Items||[]).map(p=>this.mapToLogical(p)),lastEvaluatedKey:T.LastEvaluatedKey}}async executeScan(e){let{expressionAttributeNames:i,expressionAttributeValues:n,addName:s,addValue:r}=this.createExpressionContext(),o=this._whereClause?z(this._whereClause,s,r):void 0,a=new oe({TableName:this.tableName,FilterExpression:o,ExpressionAttributeNames:Object.keys(i).length>0?i:void 0,ExpressionAttributeValues:Object.keys(n).length>0?n:void 0,Limit:this._pageSizeVal??this._limitVal,ConsistentRead:this._consistentReadVal,ExclusiveStartKey:e}),c=await this.client.send(a);return{items:(c.Items||[]).map(m=>this.mapToLogical(m)),lastEvaluatedKey:c.LastEvaluatedKey}}};var N=class{constructor(t){this.schema=t}parse(t,e,i={}){let n=this.schema.entities[e];if(!n)throw new Error(`Root entity ${e} not found in schema`);let s=[],r=[];for(let o of t)this.isEntity(o,n.entity)?s.push({...o}):r.push(o);for(let o of s)for(let[a,c]of Object.entries(i)){if(!c)continue;let m=n.relations[a];if(!m)continue;let u=m.config.to,f=r.filter(T=>this.isEntity(T,u));f.length>0&&(m.type==="many"?o[a]=f:m.type==="one"&&(o[a]=f[0]||null))}return s}isEntity(t,e){let i=e[d.ENTITY_STRATEGY],n=e[d.PHYSICAL_TABLE];if(!n||!n[E.PARTITION_KEY])return!1;let s=n[E.PARTITION_KEY].name,r=n[E.SORT_KEY]?.name,o=this.matchStrategy(t[s],i.pk),a=r?this.matchStrategy(t[r],i.sk):!0;return o&&a}matchStrategy(t,e){if(!e)return!0;if(t==null)return!1;let i=String(t);if(e.type==="static")return i===e.segments[0];if(e.type==="prefix"||e.type==="composite"){let n=e.segments[0];return typeof n=="string"&&i.startsWith(n)}return!1}};var b=class l{constructor(t,e,i,n){this.client=t;this.table=e;this.schema=i;this.entityName=n}async findFirst(t={}){return(await this.findMany({...t,limit:1}))[0]}async findMany(t={}){let e=new I(this.table,this.client,void 0);t.orderBy&&e.sort(t.orderBy==="asc");let i;if(t.where){let r=this.table._?.columns||this.table;typeof t.where=="function"?i=t.where(r,G):i=t.where}let n,s=L(this.table,i);if(this.schema&&this.entityName&&(t.with||t.include)&&s.hasPartitionKey&&!s.indexName){let r=this.table[d.PHYSICAL_TABLE];if(!r)throw new Error("Physical table not found for entity");let o=r[E.PARTITION_KEY].name,a=s.keys[o];e.where(P({name:o},a));let m=(await e).map(f=>v(this.table,f));n=new N(this.schema).parse(m,this.entityName,t.with||t.include),t.limit&&(n=n.slice(0,t.limit))}else t.limit&&e.limit(t.limit),i&&e.where(i),n=await e;if(this.schema&&this.entityName&&(t.with||t.include)){let r=t.with||t.include;await Promise.all(n.map(async o=>{for(let[a,c]of Object.entries(r)){let m=this.schema.entities[this.entityName];if(!m)continue;let u=m.relations[a];if(!u)continue;let f=o[a]!==void 0,T=typeof c=="object"&&(c.with||c.include);if(f&&!T)continue;let p=u.config.to,_=Object.entries(this.schema.entities).find(([R,y])=>y.entity===p)?.[0],V=v(this.table,o);if(u.config.fields&&u.config.references){let R={};u.config.fields.forEach((y,g)=>{let h=u.config.references[g];if(!h)return;let F=Object.entries(p[d.COLUMNS]).find(([K,C])=>C===h||C.name===h.name)?.[0];if(F){let K=Object.entries(this.table[d.COLUMNS]).find(([de,W])=>W===y||W.name===y.name)?.[0],C=K?o[K]??o[y.name]:o[y.name];C!==void 0&&(R[F]=C)}}),V=R}let j=L(p,void 0,V);if(j.hasPartitionKey){let R=new l(this.client,p,this.schema,_),y=[];for(let[g,h]of Object.entries(j.keys))y.push(P({name:g},h));if(y.length>0){let g=y.length===1?y[0]:q(...y),h=typeof c=="object"?c:{};u.type==="one"?o[a]=await R.findFirst({...h,where:g}):o[a]=await R.findMany({...h,where:g})}}}}))}return n}};import{BatchGetCommand as re}from"@aws-sdk/lib-dynamodb";var M=class{constructor(t){this.client=t}items(t,e){return new A(t,this.client,e)}},A=class extends w{constructor(e,i,n){super(e,i);this.keysData=n}static[d.ENTITY_KIND]="BatchGetBase";async execute(){let e=[],i=[],n=this.keysData.map(o=>this.resolveKeys(void 0,o).keys),s=0,r=5;for(;n.length>0&&s<r;){s++;let o=new re({RequestItems:{[this.tableName]:{Keys:n}}}),a=await this.client.send(o);a.Responses?.[this.tableName]&&e.push(...a.Responses[this.tableName]);let c=a.UnprocessedKeys?.[this.tableName]?.Keys;c&&c.length>0?n=c:n=[]}return n.length>0&&i.push(...n),{succeeded:e,failed:i}}};import{BatchWriteCommand as ae}from"@aws-sdk/lib-dynamodb";var B=class{constructor(t){this.client=t}operations(t,e){return new Y(t,this.client,e)}},Y=class extends w{constructor(e,i,n){super(e,i);this.ops=n}static[d.ENTITY_KIND]="BatchWriteBase";async execute(){let e=0,i=[],s=[...this.ops.map(a=>{if(a.type==="put"){let c=a.item;if(H(c)>400*1024)throw new $("Item in batch exceeds the 400KB limit.");return{PutRequest:{Item:c}}}else return{DeleteRequest:{Key:a.keys}}})],r=0,o=5;for(;s.length>0&&r<o;){r++;let a=new ae({RequestItems:{[this.tableName]:s}}),m=(await this.client.send(a)).UnprocessedItems?.[this.tableName]||[];e+=s.length-m.length,m.length>0?s=m:s=[]}return s.length>0&&(i=s),{succeededCount:e,failed:i}}};var x=class{constructor(t,e){this.type=t;this.config=e}};function st(l,t){if(l instanceof D){let i=t({one:(n,s)=>new x("one",{to:n,...s}),many:(n,s)=>new x("many",{to:n,...s})});return{entity:l,config:i,[k.RELATION_CONFIG]:!0}}else{let e=l,i=t,n={one:{},many:{}};for(let[r,o]of Object.entries(e))n.one[r]=a=>new x("one",{to:o,...a}),n.many[r]=a=>new x("many",{to:o,...a}),n[r]=o;let s=i(n);return{schema:e,definitions:s,[k.RELATION_CONFIG]:!0}}}function te(l){let t={entities:{}};for(let[e,i]of Object.entries(l))i instanceof D&&(t.entities[e]={entity:i,relations:{}});for(let[,e]of Object.entries(l))if(e&&typeof e=="object"&&e[k.RELATION_CONFIG]){let i=e;if(i.entity){let n=e,s=Object.entries(t.entities).find(([r,o])=>o.entity===n.entity);if(s){let[,r]=s;r.relations={...r.relations,...n.config}}}else if(i.definitions){let n=e;for(let[s,r]of Object.entries(n.definitions)){let o=n.schema[s],a=Object.entries(t.entities).find(([c,m])=>m.entity===o);if(a&&r){let[,c]=a;c.relations={...c.relations,...r}}}}}return t}var S=class{docClient;schema;retryConfig;query;constructor(t,e,i){this.retryConfig={maxAttempts:i?.maxAttempts??3,baseDelay:i?.baseDelay??100},this.docClient=new U(le.from(t),new Q(this.retryConfig)),e&&(this.schema=te(e)),this.query=new Proxy({},{get:(n,s)=>{if(typeof s!="string")return;if(!this.schema)throw new Error("No relations defined. Initialize mizzle with a relations object to use db.query.");let r=this.schema.entities[s];if(!r)throw new Error(`Entity ${s} not found in relations schema.`);return new b(this.docClient,r.entity,this.schema,s)}})}insert(t){return new J(t,this.docClient)}select(t){return new O(this.docClient,t)}batchGet(t,e){return new M(this.docClient).items(t,e)}batchWrite(t,e){return new B(this.docClient).operations(t,e)}_query(t){return new b(this.docClient,t)}update(t){return new X(t,this.docClient)}delete(t,e){return new Z(t,this.docClient,e)}async transaction(t,e){let i=new ee(this.docClient),n=await e(i);if(n.length===0)return;if(n.length>100)throw new Error("DynamoDB transactions are limited to 100 items.");let{TransactionExecutor:s}=await import("./transaction-RE7LXTGV.js");await new s(this.docClient).execute(t,n)}};function gt(l){return l instanceof ne?new S(l):"client"in l&&l.client instanceof ne?new S(l.client,l.relations,l.retry):"client"in l&&l.client?new S(l.client,l.relations,l.retry):new S(l)}export{O as a,I as b,M as c,A as d,B as e,Y as f,b as g,x as h,st as i,te as j,S as k,gt as l};
+import{D as Q,E as U,I as L,J as w,K as H,L as $,N as J,P as X,Q as Z,S as ee,e as P,p as q,q as G,r as z}from"./chunk-UM3YF5EC.js";import{b as D}from"./chunk-GPYZK4WY.js";import{d as E,e as d,f as C,i as v}from"./chunk-DU7UPWBW.js";import{DynamoDBClient as ne}from"@aws-sdk/client-dynamodb";import{DynamoDBDocumentClient as le}from"@aws-sdk/lib-dynamodb";import{GetCommand as ie,QueryCommand as se,ScanCommand as oe}from"@aws-sdk/lib-dynamodb";var O=class{constructor(t,e){this.client=t;this.fields=e}from(t){return new b(t,this.client,this.fields)}},b=class extends w{constructor(e,i,n){super(e,i);this.fields=n}static[d.ENTITY_KIND]="SelectBase";_whereClause;_limitVal;_pageSizeVal;_consistentReadVal;_sortForward=!0;_forcedIndexName;where(e){return this._whereClause=e,this}limit(e){return this._limitVal=e,this}pageSize(e){return this._pageSizeVal=e,this}consistentRead(e=!0){return this._consistentReadVal=e,this}sort(e){return this._sortForward=e,this}index(e){return this._forcedIndexName=e,this}iterator(){return async function*(){let e,i=0;do{let n=await this.fetchPage(e);for(let s of n.items)if(yield s,i++,this._limitVal!==void 0&&i>=this._limitVal)return;e=n.lastEvaluatedKey}while(e)}.bind(this)()}async fetchPage(e){let i=this.resolveKeys(this._whereClause,void 0,this._forcedIndexName);return i.hasPartitionKey&&i.hasSortKey&&!i.indexName&&!e?{items:await this.executeGet(i.keys)}:i.hasPartitionKey||i.indexName?await this.executeQuery(i,e):await this.executeScan(e)}async execute(){let{items:e}=await this.fetchPage();return e}async executeGet(e){let i=new ie({TableName:this.tableName,Key:e,ConsistentRead:this._consistentReadVal}),n=await this.client.send(i);return n.Item?[this.mapToLogical(n.Item)]:[]}async executeQuery(e,i){let{expressionAttributeNames:n,expressionAttributeValues:s,addName:r,addValue:o}=this.createExpressionContext(),l=[],c=new Set;for(let[p,_]of Object.entries(e.keys))l.push(`${r(p)} = ${o(_)}`),c.add(p);let m=l.join(" AND "),u=this._whereClause?z(this._whereClause,r,o,c):void 0,f=new se({TableName:this.tableName,IndexName:e.indexName,KeyConditionExpression:m,FilterExpression:u||void 0,ExpressionAttributeNames:Object.keys(n).length>0?n:void 0,ExpressionAttributeValues:Object.keys(s).length>0?s:void 0,Limit:this._pageSizeVal??this._limitVal,ScanIndexForward:this._sortForward,ConsistentRead:e.indexName?void 0:this._consistentReadVal,ExclusiveStartKey:i}),T=await this.client.send(f);return{items:(T.Items||[]).map(p=>this.mapToLogical(p)),lastEvaluatedKey:T.LastEvaluatedKey}}async executeScan(e){let{expressionAttributeNames:i,expressionAttributeValues:n,addName:s,addValue:r}=this.createExpressionContext(),o=this._whereClause?z(this._whereClause,s,r):void 0,l=new oe({TableName:this.tableName,FilterExpression:o,ExpressionAttributeNames:Object.keys(i).length>0?i:void 0,ExpressionAttributeValues:Object.keys(n).length>0?n:void 0,Limit:this._pageSizeVal??this._limitVal,ConsistentRead:this._consistentReadVal,ExclusiveStartKey:e}),c=await this.client.send(l);return{items:(c.Items||[]).map(m=>this.mapToLogical(m)),lastEvaluatedKey:c.LastEvaluatedKey}}};var M=class{constructor(t){this.schema=t}parse(t,e,i={}){let n=this.schema.entities[e];if(!n)throw new Error(`Root entity ${e} not found in schema`);let s=[],r=[];for(let o of t)this.isEntity(o,n.entity)?s.push({...o}):r.push(o);for(let o of s)for(let[l,c]of Object.entries(i)){if(!c)continue;let m=n.relations[l];if(!m)continue;let u=m.config.to,f=r.filter(T=>this.isEntity(T,u));f.length>0&&(m.type==="many"?o[l]=f:m.type==="one"&&(o[l]=f[0]||null))}return s}isEntity(t,e){let i=e[d.ENTITY_STRATEGY],n=e[d.PHYSICAL_TABLE];if(!n||!n[E.PARTITION_KEY])return!1;let s=n[E.PARTITION_KEY].name,r=n[E.SORT_KEY]?.name,o=this.matchStrategy(t[s],i.pk),l=r?this.matchStrategy(t[r],i.sk):!0;return o&&l}matchStrategy(t,e){if(!e)return!0;if(t==null)return!1;let i=String(t);if(e.type==="static")return i===e.segments[0];if(e.type==="prefix"||e.type==="composite"){let n=e.segments[0];return typeof n=="string"&&i.startsWith(n)}return!1}};var k=class a{constructor(t,e,i,n){this.client=t;this.table=e;this.schema=i;this.entityName=n}async findFirst(t={}){return(await this.findMany({...t,limit:1}))[0]}async findMany(t={}){let e=new b(this.table,this.client,void 0);t.orderBy&&e.sort(t.orderBy==="asc");let i;if(t.where){let r=this.table._?.columns||this.table;typeof t.where=="function"?i=t.where(r,G):i=t.where}let n,s=L(this.table,i);if(this.schema&&this.entityName&&(t.with||t.include)&&s.hasPartitionKey&&!s.indexName){let r=this.table[d.PHYSICAL_TABLE];if(!r)throw new Error("Physical table not found for entity");let o=r[E.PARTITION_KEY].name,l=s.keys[o];e.where(P({name:o},l));let m=(await e).map(f=>v(this.table,f));n=new M(this.schema).parse(m,this.entityName,t.with||t.include),t.limit&&(n=n.slice(0,t.limit))}else t.limit&&e.limit(t.limit),i&&e.where(i),n=await e;if(this.schema&&this.entityName&&(t.with||t.include)){let r=t.with||t.include;await Promise.all(n.map(async o=>{for(let[l,c]of Object.entries(r)){let m=this.schema.entities[this.entityName];if(!m)continue;let u=m.relations[l];if(!u)continue;let f=o[l]!==void 0,T=typeof c=="object"&&(c.with||c.include);if(f&&!T)continue;let p=u.config.to,_=Object.entries(this.schema.entities).find(([R,y])=>y.entity===p)?.[0],V=v(this.table,o);if(u.config.fields&&u.config.references){let R={};u.config.fields.forEach((y,g)=>{let h=u.config.references[g];if(!h)return;let F=Object.entries(p[d.COLUMNS]).find(([K,I])=>I===h||I.name===h.name)?.[0];if(F){let K=Object.entries(this.table[d.COLUMNS]).find(([de,W])=>W===y||W.name===y.name)?.[0],I=K?o[K]??o[y.name]:o[y.name];I!==void 0&&(R[F]=I)}}),V=R}let j=L(p,void 0,V);if(j.hasPartitionKey){let R=new a(this.client,p,this.schema,_),y=[];for(let[g,h]of Object.entries(j.keys))y.push(P({name:g},h));if(y.length>0){let g=y.length===1?y[0]:q(...y),h=typeof c=="object"?c:{};u.type==="one"?o[l]=await R.findFirst({...h,where:g}):o[l]=await R.findMany({...h,where:g})}}}}))}return n}};import{BatchGetCommand as re}from"@aws-sdk/lib-dynamodb";var N=class{constructor(t){this.client=t}items(t,e){return new A(t,this.client,e)}},A=class extends w{constructor(e,i,n){super(e,i);this.keysData=n}static[d.ENTITY_KIND]="BatchGetBase";async execute(){let e=[],i=[],n=this.keysData.map(o=>this.resolveKeys(void 0,o).keys),s=0,r=5;for(;n.length>0&&s<r;){s++;let o=new re({RequestItems:{[this.tableName]:{Keys:n}}}),l=await this.client.send(o);l.Responses?.[this.tableName]&&e.push(...l.Responses[this.tableName]);let c=l.UnprocessedKeys?.[this.tableName]?.Keys;c&&c.length>0?n=c:n=[]}return n.length>0&&i.push(...n),{succeeded:e,failed:i}}};import{BatchWriteCommand as ae}from"@aws-sdk/lib-dynamodb";var B=class{constructor(t){this.client=t}operations(t,e){return new Y(t,this.client,e)}},Y=class extends w{constructor(e,i,n){super(e,i);this.ops=n}static[d.ENTITY_KIND]="BatchWriteBase";async execute(){let e=0,i=[],s=[...this.ops.map(l=>{if(l.type==="put"){let c=l.item;if(H(c)>400*1024)throw new $("Item in batch exceeds the 400KB limit.");return{PutRequest:{Item:c}}}else return{DeleteRequest:{Key:l.keys}}})],r=0,o=5;for(;s.length>0&&r<o;){r++;let l=new ae({RequestItems:{[this.tableName]:s}}),m=(await this.client.send(l)).UnprocessedItems?.[this.tableName]||[];e+=s.length-m.length,m.length>0?s=m:s=[]}return s.length>0&&(i=s),{succeededCount:e,failed:i}}};var x=class{constructor(t,e){this.type=t;this.config=e}};function st(a,t){if(a instanceof D){let i=t({one:(n,s)=>new x("one",{to:n,...s}),many:(n,s)=>new x("many",{to:n,...s})});return{entity:a,config:i,[C.RELATION_CONFIG]:!0}}else{let e=a,i=t,n={one:{},many:{}};for(let[r,o]of Object.entries(e))n.one[r]=l=>new x("one",{to:o,...l}),n.many[r]=l=>new x("many",{to:o,...l}),n[r]=o;let s=i(n);return{schema:e,definitions:s,[C.RELATION_CONFIG]:!0}}}function te(a){let t={entities:{}};if(a&&a[C.RELATION_CONFIG]&&a.definitions){let e=a;for(let[i,n]of Object.entries(e.schema))t.entities[i]={entity:n,relations:e.definitions[i]||{}};return t}for(let[e,i]of Object.entries(a))i instanceof D&&(t.entities[e]={entity:i,relations:{}});for(let[,e]of Object.entries(a))if(e&&typeof e=="object"&&e[C.RELATION_CONFIG]){let i=e;if(i.entity){let n=e,s=Object.entries(t.entities).find(([r,o])=>o.entity===n.entity);if(s){let[,r]=s;r.relations={...r.relations,...n.config}}}else if(i.definitions){let n=e;for(let[s,r]of Object.entries(n.definitions)){let o=n.schema[s],l=Object.entries(t.entities).find(([c,m])=>m.entity===o);if(l&&r){let[,c]=l;c.relations={...c.relations,...r}}}}}return t}var S=class{docClient;schema;retryConfig;query;constructor(t,e,i){this.retryConfig={maxAttempts:i?.maxAttempts??3,baseDelay:i?.baseDelay??100},this.docClient=new U(le.from(t),new Q(this.retryConfig)),e&&(this.schema=te(e)),this.query=new Proxy({},{get:(n,s)=>{if(typeof s!="string")return;if(!this.schema)throw new Error("No relations defined. Initialize mizzle with a relations object to use db.query.");let r=this.schema.entities[s];if(!r)throw new Error(`Entity ${s} not found in relations schema.`);return new k(this.docClient,r.entity,this.schema,s)}})}insert(t){return new J(t,this.docClient)}select(t){return new O(this.docClient,t)}batchGet(t,e){return new N(this.docClient).items(t,e)}batchWrite(t,e){return new B(this.docClient).operations(t,e)}_query(t){return new k(this.docClient,t)}update(t){return new X(t,this.docClient)}delete(t,e){return new Z(t,this.docClient,e)}async transaction(t,e){let i=new ee(this.docClient),n=await e(i);if(n.length===0)return;if(n.length>100)throw new Error("DynamoDB transactions are limited to 100 items.");let{TransactionExecutor:s}=await import("./transaction-RE7LXTGV.js");await new s(this.docClient).execute(t,n)}};function gt(a){return a instanceof ne?new S(a):"client"in a&&a.client instanceof ne?new S(a.client,a.relations,a.retry):"client"in a&&a.client?new S(a.client,a.relations,a.retry):new S(a)}export{O as a,b,N as c,A as d,B as e,Y as f,k as g,x as h,st as i,te as j,S as k,gt as l};

package/dist/{db-zHIHBm1E.d.ts → db-ZX6HGN1x.d.ts}

@@ -654,7 +654,9 @@ declare class RelationnalQueryBuilder<T extends Entity> {
     findMany<TOptions extends RelationalQueryOptions<T>>(options?: TOptions): Promise<InferRelationalModel<T, TOptions>[]>;
 }
 
-type QuerySchema<TSchema extends Record<string, any>> = {
+type QuerySchema<TSchema extends Record<string, any>> = TSchema extends MultiRelationsDefinition<infer T> ? {
+    [K in keyof T]: RelationnalQueryBuilder<T[K]>;
+} : {
     [K in keyof TSchema as TSchema[K] extends Entity ? K : never]: RelationnalQueryBuilder<TSchema[K] extends Entity ? TSchema[K] : never>;
 };
 /**

package/dist/db.d.ts

@@ -1,3 +1,3 @@
 import './operators-BVreW0ky.js';
-export { i as DynamoDB, M as MizzleConfig, Q as QuerySchema, X as mizzle } from './db-zHIHBm1E.js';
+export { i as DynamoDB, M as MizzleConfig, Q as QuerySchema, X as mizzle } from './db-ZX6HGN1x.js';
 import '@aws-sdk/client-dynamodb';

package/dist/db.js

@@ -1 +1 @@
-import{k as a,l as b}from"./chunk-NPPZW6VT.js";import"./chunk-UM3YF5EC.js";import"./chunk-GPYZK4WY.js";import"./chunk-DU7UPWBW.js";export{a as DynamoDB,b as mizzle};
+import{k as a,l as b}from"./chunk-WTOBEKFQ.js";import"./chunk-UM3YF5EC.js";import"./chunk-GPYZK4WY.js";import"./chunk-DU7UPWBW.js";export{a as DynamoDB,b as mizzle};

package/dist/index.d.ts

@@ -1,6 +1,6 @@
 import { E as Expression } from './operators-BVreW0ky.js';
 export { A as AnyTable, a as Assume, b as AtomicValues, B as BinaryExpression, C as ColumnType, c as ColumnsBuilder, d as Condition, e as ENTITY_SYMBOLS, f as Entity, g as EntityConfig, h as EntityWithColumns, F as FunctionExpression, I as INFER_MODE, i as IndexBuilder, j as IndexColumnConfig, k as InferInsertModel, l as InferModelFromColumns, m as InferSelectModel, n as InferSelectedModel, K as KeyStrategy, o as KeyStrategyType, L as LogicalExpression, M as MapColumnName, N as NULLS, O as ORDER, p as Operators, P as PhysicalTable, q as PhysicalTableConfig, R as RELATION_SYMBOLS, S as Simplify, r as StrategyCallback, s as StrategyResolution, T as TABLE_SYMBOLS, t as TableDefinition, U as Update, u as UpdateTableConfig, v as and, w as attributeExists, x as beginsWith, y as between, z as binary, D as binarySet, G as boolean, H as compositeKey, J as contains, Q as date, V as dynamoEntity, W as dynamoTable, X as eq, Y as getColumnBuilders, Z as getEntityColumns, _ as gsi, $ as gt, a0 as gte, a1 as inList, a2 as json, a3 as list, a4 as lsi, a5 as lt, a6 as lte, a7 as map, a8 as mapToLogical, a9 as number, aa as numberSet, ab as operators, ac as or, ad as prefixKey, ae as resolveStrategies, af as resolveTableName, ag as staticKey, ah as string, ai as stringSet, aj as uuid } from './operators-BVreW0ky.js';
-export { A as ActionType, a as AddAction, B as BatchGetBase, b as BatchGetBuilder, c as BatchGetResult, d as BatchWriteBase, e as BatchWriteBuilder, f as BatchWriteOperation, g as BatchWriteResult, C as ConditionCheckBuilder, D as DeleteAction, h as DeleteBuilder, i as DynamoDB, E as EntityMetadata, I as IncludeOptions, j as InferRelationalModel, k as InsertBase, l as InsertBuilder, m as InternalRelationalSchema, M as MizzleConfig, n as MultiRelationsCallback, o as MultiRelationsDefinition, Q as QuerySchema, R as Relation, p as RelationConfig, q as RelationType, r as RelationalQueryOptions, s as RelationnalQueryBuilder, t as RelationsCallback, u as RelationsDefinition, v as RelationsHelpers, w as RemoveAction, x as RetryConfig, y as RetryHandler, S as SelectBase, z as SelectBuilder, F as SelectedFields, G as SetAction, T as TransactionExecutor, H as TransactionProxy, U as UpdateAction, J as UpdateBuilder, K as add, L as addToSet, N as append, O as defineRelations, P as deleteFromSet, V as extractMetadata, W as ifNotExists, X as mizzle, Y as remove } from './db-zHIHBm1E.js';
+export { A as ActionType, a as AddAction, B as BatchGetBase, b as BatchGetBuilder, c as BatchGetResult, d as BatchWriteBase, e as BatchWriteBuilder, f as BatchWriteOperation, g as BatchWriteResult, C as ConditionCheckBuilder, D as DeleteAction, h as DeleteBuilder, i as DynamoDB, E as EntityMetadata, I as IncludeOptions, j as InferRelationalModel, k as InsertBase, l as InsertBuilder, m as InternalRelationalSchema, M as MizzleConfig, n as MultiRelationsCallback, o as MultiRelationsDefinition, Q as QuerySchema, R as Relation, p as RelationConfig, q as RelationType, r as RelationalQueryOptions, s as RelationnalQueryBuilder, t as RelationsCallback, u as RelationsDefinition, v as RelationsHelpers, w as RemoveAction, x as RetryConfig, y as RetryHandler, S as SelectBase, z as SelectBuilder, F as SelectedFields, G as SetAction, T as TransactionExecutor, H as TransactionProxy, U as UpdateAction, J as UpdateBuilder, K as add, L as addToSet, N as append, O as defineRelations, P as deleteFromSet, V as extractMetadata, W as ifNotExists, X as mizzle, Y as remove } from './db-ZX6HGN1x.js';
 export { SchemaChange, compareSchema } from './diff.js';
 import '@aws-sdk/client-dynamodb';
 import './snapshot.js';

package/dist/index.js

@@ -1 +1 @@
-import{a as Dr}from"./chunk-TOYV2M4M.js";import"./chunk-AQVECMXP.js";import{a as lr,b as kr,c as Ir,d as dr,e as hr,f as jr,g as qr,h as vr,i as zr,j as Ar,k as Er,l as Fr}from"./chunk-NPPZW6VT.js";import{A as O,B as P,C as Q,D as fr,F as xr,G as mr,H as nr,I as ir,L as sr,M as gr,N as cr,O as ur,P as Cr,Q as ar,R as br,S as wr,T as yr,a as u,b as l,c as k,d as C,e as a,f as b,g as w,h as y,i as I,j as d,k as h,l as j,m as q,n as v,o as z,p as A,q as D,r as E,s as F,t as G,u as H,v as J,w as K,x as L,y as M,z as N}from"./chunk-UM3YF5EC.js";import{a as or,b as tr,c as er,d as pr}from"./chunk-GPYZK4WY.js";import{a as p,b as f,c as x,d as m,e as n,f as i,g as s,h as g,i as c,k as R,l as S,m as T,n as U,o as V,p as W,q as X,r as Y,s as Z,t as _,u as $,v as B,w as rr}from"./chunk-DU7UPWBW.js";var o=class{constructor(t,e){this.type=t;this.config=e}};function Gr(r,t){return new o("gsi",{pk:r,sk:t})}function Hr(r){return new o("lsi",{sk:r})}export{H as AddAction,dr as BatchGetBase,Ir as BatchGetBuilder,jr as BatchWriteBase,hr as BatchWriteBuilder,l as BinaryExpression,br as ConditionCheckBuilder,K as DeleteAction,ar as DeleteBuilder,Er as DynamoDB,n as ENTITY_SYMBOLS,tr as Entity,u as Expression,C as FunctionExpression,p as INFER_MODE,o as IndexBuilder,ur as InsertBase,cr as InsertBuilder,sr as ItemSizeExceededError,k as LogicalExpression,x as NULLS,f as ORDER,or as PhysicalTable,i as RELATION_SYMBOLS,vr as Relation,qr as RelationnalQueryBuilder,J as RemoveAction,fr as RetryHandler,kr as SelectBase,lr as SelectBuilder,G as SetAction,m as TABLE_SYMBOLS,yr as TransactionExecutor,gr as TransactionFailedError,wr as TransactionProxy,F as UpdateAction,Cr as UpdateBuilder,L as add,P as addToSet,A as and,M as append,v as attributeExists,j as beginsWith,d as between,R as binary,S as binarySet,T as boolean,E as buildExpression,Dr as compareSchema,nr as compositeKey,q as contains,U as date,zr as defineRelations,Q as deleteFromSet,er as dynamoEntity,pr as dynamoTable,a as eq,Ar as extractMetadata,rr as getColumnBuilders,s as getEntityColumns,Gr as gsi,b as gt,w as gte,N as ifNotExists,h as inList,B as json,V as list,Hr as lsi,y as lt,I as lte,W as map,c as mapToLogical,Fr as mizzle,X as number,Y as numberSet,D as operators,z as or,mr as prefixKey,O as remove,ir as resolveStrategies,g as resolveTableName,xr as staticKey,Z as string,_ as stringSet,$ as uuid};
+import{a as Dr}from"./chunk-TOYV2M4M.js";import"./chunk-AQVECMXP.js";import{a as lr,b as kr,c as Ir,d as dr,e as hr,f as jr,g as qr,h as vr,i as zr,j as Ar,k as Er,l as Fr}from"./chunk-WTOBEKFQ.js";import{A as O,B as P,C as Q,D as fr,F as xr,G as mr,H as nr,I as ir,L as sr,M as gr,N as cr,O as ur,P as Cr,Q as ar,R as br,S as wr,T as yr,a as u,b as l,c as k,d as C,e as a,f as b,g as w,h as y,i as I,j as d,k as h,l as j,m as q,n as v,o as z,p as A,q as D,r as E,s as F,t as G,u as H,v as J,w as K,x as L,y as M,z as N}from"./chunk-UM3YF5EC.js";import{a as or,b as tr,c as er,d as pr}from"./chunk-GPYZK4WY.js";import{a as p,b as f,c as x,d as m,e as n,f as i,g as s,h as g,i as c,k as R,l as S,m as T,n as U,o as V,p as W,q as X,r as Y,s as Z,t as _,u as $,v as B,w as rr}from"./chunk-DU7UPWBW.js";var o=class{constructor(t,e){this.type=t;this.config=e}};function Gr(r,t){return new o("gsi",{pk:r,sk:t})}function Hr(r){return new o("lsi",{sk:r})}export{H as AddAction,dr as BatchGetBase,Ir as BatchGetBuilder,jr as BatchWriteBase,hr as BatchWriteBuilder,l as BinaryExpression,br as ConditionCheckBuilder,K as DeleteAction,ar as DeleteBuilder,Er as DynamoDB,n as ENTITY_SYMBOLS,tr as Entity,u as Expression,C as FunctionExpression,p as INFER_MODE,o as IndexBuilder,ur as InsertBase,cr as InsertBuilder,sr as ItemSizeExceededError,k as LogicalExpression,x as NULLS,f as ORDER,or as PhysicalTable,i as RELATION_SYMBOLS,vr as Relation,qr as RelationnalQueryBuilder,J as RemoveAction,fr as RetryHandler,kr as SelectBase,lr as SelectBuilder,G as SetAction,m as TABLE_SYMBOLS,yr as TransactionExecutor,gr as TransactionFailedError,wr as TransactionProxy,F as UpdateAction,Cr as UpdateBuilder,L as add,P as addToSet,A as and,M as append,v as attributeExists,j as beginsWith,d as between,R as binary,S as binarySet,T as boolean,E as buildExpression,Dr as compareSchema,nr as compositeKey,q as contains,U as date,zr as defineRelations,Q as deleteFromSet,er as dynamoEntity,pr as dynamoTable,a as eq,Ar as extractMetadata,rr as getColumnBuilders,s as getEntityColumns,Gr as gsi,b as gt,w as gte,N as ifNotExists,h as inList,B as json,V as list,Hr as lsi,y as lt,I as lte,W as map,c as mapToLogical,Fr as mizzle,X as number,Y as numberSet,D as operators,z as or,mr as prefixKey,O as remove,ir as resolveStrategies,g as resolveTableName,xr as staticKey,Z as string,_ as stringSet,$ as uuid};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aurios/mizzle",
-  "version": "1.1.4",
+  "version": "1.1.5",
   "description": "a drizzle-like orm for dynamoDB",
   "publishConfig": {
     "access": "public",
@@ -79,8 +79,8 @@
   "devDependencies": {
     "tsup": "^8.5.1",
     "@repo/shared": "0.0.3",
-    "@repo/typescript-config": "0.0.0",
-    "@repo/vitest-config": "0.0.0"
+    "@repo/vitest-config": "0.0.0",
+    "@repo/typescript-config": "0.0.0"
   },
   "scripts": {
     "check": "tsc --noEmit",

package/src/core/relations.ts

@@ -238,6 +238,18 @@ export function extractMetadata(schema: Record<string, unknown>): InternalRelati
     entities: {},
   };
 
+  // If the schema itself is a MultiRelationsDefinition, unwrap it
+  if (schema && (schema as any)[RELATION_SYMBOLS.RELATION_CONFIG] && (schema as any).definitions) {
+    const multiDef = schema as unknown as MultiRelationsDefinition;
+    for (const [key, entity] of Object.entries(multiDef.schema)) {
+      metadata.entities[key] = {
+        entity,
+        relations: multiDef.definitions[key] || {},
+      };
+    }
+    return metadata;
+  }
+
   // First pass: identify entities
   for (const [key, value] of Object.entries(schema)) {
     if (value instanceof Entity) {

package/src/db.ts

@@ -10,15 +10,21 @@ import { DeleteBuilder } from "./builders/delete";
 import { BatchGetBuilder } from "./builders/batch-get";
 import { BatchWriteBuilder, type BatchWriteOperation } from "./builders/batch-write";
 import { TransactionProxy } from "./builders/transaction";
-import { extractMetadata, type InternalRelationalSchema } from "./core/relations";
+import { extractMetadata, type InternalRelationalSchema, type MultiRelationsDefinition } from "./core/relations";
 import { RetryHandler, type RetryConfig } from "./core/retry";
 import { MizzleClient, type IMizzleClient } from "./core/client";
 
-export type QuerySchema<TSchema extends Record<string, any>> = {
-  [K in keyof TSchema as TSchema[K] extends Entity ? K : never]: RelationnalQueryBuilder<
-    TSchema[K] extends Entity ? TSchema[K] : never
-  >;
-};
+export type QuerySchema<TSchema extends Record<string, any>> = TSchema extends MultiRelationsDefinition<
+  infer T
+>
+  ? {
+      [K in keyof T]: RelationnalQueryBuilder<T[K]>;
+    }
+  : {
+      [K in keyof TSchema as TSchema[K] extends Entity ? K : never]: RelationnalQueryBuilder<
+        TSchema[K] extends Entity ? TSchema[K] : never
+      >;
+    };
 
 /**
  * DynamoDB database instance.

package/test/repro-user-issue.test.ts

@@ -0,0 +1,33 @@
+import { describe, it, expect } from "vitest";
+import { DynamoDBClient } from "@aws-sdk/client-dynamodb";
+import { dynamoTable, dynamoEntity } from "../src/core/table";
+import { defineRelations } from "../src/core/relations";
+import { string, uuid } from "../src/columns";
+import { mizzle } from "../src/db";
+
+describe("User Issue Reproduction: MultiRelationsDefinition as schema", () => {
+  const table = dynamoTable("mizzle-test", {
+    pk: string("pk"),
+    sk: string("sk"),
+  });
+
+  const shoppingLists = dynamoEntity(table, "shoppingLists", {
+    id: uuid("id"),
+    name: string("name"),
+  });
+
+  const client = new DynamoDBClient({ region: "us-east-1" });
+
+  it("should allow using MultiRelationsDefinition directly in mizzle config", () => {
+    const relations = defineRelations({ shoppingLists }, () => ({}));
+    
+    const db = mizzle({ 
+      client, 
+      relations 
+    });
+
+    // This is the part that fails in TypeScript for the user
+    // We are testing runtime behavior here, but the types should also be checked
+    expect(db.query.shoppingLists).toBeDefined();
+  });
+});