@m1212e/rumble 0.6.0 → 0.6.2

package/index.d.cts

@@ -1,12 +1,13 @@
 import * as drizzle_orm_pg_core from 'drizzle-orm/pg-core';
+import { PgEnum } from 'drizzle-orm/pg-core';
 import * as SchemaBuilder from '@pothos/core';
 import SchemaBuilder__default, { SchemaTypes, BasePlugin, PothosOutputFieldConfig } from '@pothos/core';
 import * as _pothos_plugin_drizzle from '@pothos/plugin-drizzle';
 import { DrizzleClient } from '@pothos/plugin-drizzle';
+import { GraphQLFieldResolver, GraphQLError } from 'graphql';
 import * as graphql_yoga from 'graphql-yoga';
 import { createPubSub, YogaServerOptions } from 'graphql-yoga';
 import * as drizzle_orm from 'drizzle-orm';
-import { GraphQLFieldResolver, GraphQLError } from 'graphql';
 
 declare const pluginName = "ManualFiltersPlugin";
 
@@ -83,6 +84,11 @@ type RumbleInput<UserContext extends Record<string, any>, DB extends GenericDriz
     defaultLimit?: number | undefined | null;
 };
 
+type EnumTypes = PgEnum<any>;
+type NonEnumFields<T> = {
+    [K in keyof T as T[K] extends EnumTypes ? never : K]: T[K];
+};
+
 declare const rumble: <UserContext extends Record<string, any>, DB extends GenericDrizzleDbTypeConstraints, RequestEvent extends Record<string, any>, PothosConfig extends CustomRumblePothosConfig, Action extends string = "read" | "update" | "delete">(rumbleInput: RumbleInput<UserContext, DB, RequestEvent, Action, PothosConfig>) => {
     /**
    * The ability builder. Use it to declare whats allowed for each entity in your DB.
@@ -196,7 +202,7 @@ declare const rumble: <UserContext extends Record<string, any>, DB extends Gener
     /**
      * A function for creating default objects for your schema
      */
-    object: <ExplicitTableName extends keyof NonNullable<DB["_"]["relations"]["schema"]>, RefName extends string>({ table, refName, readAction, extend, }: {
+    object: <ExplicitTableName extends keyof NonEnumFields<NonNullable<DB["_"]["relations"]["schema"]>>, RefName extends string>({ table, refName, readAction, extend, }: {
         table: ExplicitTableName;
         refName?: RefName | undefined;
         readAction?: Action | undefined;
@@ -287,7 +293,7 @@ declare const rumble: <UserContext extends Record<string, any>, DB extends Gener
     /**
      * A function for creating where args to filter entities
      */
-    arg: <ExplicitTableName extends keyof NonNullable<DB["_"]["relations"]["schema"]>, RefName_1 extends string>({ table, refName, dbName, }: Partial<{
+    arg: <ExplicitTableName extends keyof NonEnumFields<NonNullable<DB["_"]["relations"]["schema"]>>, RefName_1 extends string>({ table, refName, dbName, }: Partial<{
         table: ExplicitTableName;
         refName: RefName_1 | undefined;
         dbName: string;
@@ -343,7 +349,7 @@ declare const rumble: <UserContext extends Record<string, any>, DB extends Gener
      * A function for creating default READ queries.
      * Make sure the objects for the table you are creating the queries for are implemented
      */
-    query: <ExplicitTableName extends keyof NonNullable<DB["_"]["relations"]["schema"]>>({ table, readAction, listAction, }: {
+    query: <ExplicitTableName extends keyof NonEnumFields<NonNullable<DB["_"]["relations"]["schema"]>>>({ table, readAction, listAction, }: {
         table: ExplicitTableName;
         readAction?: Action | undefined;
         listAction?: Action | undefined;
@@ -351,7 +357,7 @@ declare const rumble: <UserContext extends Record<string, any>, DB extends Gener
     /**
      * A function for creating a pubsub instance for a table. Use this to publish or subscribe events
      */
-    pubsub: <ExplicitTableName extends keyof NonNullable<DB["_"]["relations"]["schema"]>>({ table, }: {
+    pubsub: <ExplicitTableName extends keyof NonEnumFields<NonNullable<DB["_"]["relations"]["schema"]>>>({ table, }: {
         table: ExplicitTableName;
     }) => {
         registerOnInstance({ instance, action, primaryKeyValue, }: {

package/index.d.ts

@@ -1,12 +1,13 @@
 import * as drizzle_orm_pg_core from 'drizzle-orm/pg-core';
+import { PgEnum } from 'drizzle-orm/pg-core';
 import * as SchemaBuilder from '@pothos/core';
 import SchemaBuilder__default, { SchemaTypes, BasePlugin, PothosOutputFieldConfig } from '@pothos/core';
 import * as _pothos_plugin_drizzle from '@pothos/plugin-drizzle';
 import { DrizzleClient } from '@pothos/plugin-drizzle';
+import { GraphQLFieldResolver, GraphQLError } from 'graphql';
 import * as graphql_yoga from 'graphql-yoga';
 import { createPubSub, YogaServerOptions } from 'graphql-yoga';
 import * as drizzle_orm from 'drizzle-orm';
-import { GraphQLFieldResolver, GraphQLError } from 'graphql';
 
 declare const pluginName = "ManualFiltersPlugin";
 
@@ -83,6 +84,11 @@ type RumbleInput<UserContext extends Record<string, any>, DB extends GenericDriz
     defaultLimit?: number | undefined | null;
 };
 
+type EnumTypes = PgEnum<any>;
+type NonEnumFields<T> = {
+    [K in keyof T as T[K] extends EnumTypes ? never : K]: T[K];
+};
+
 declare const rumble: <UserContext extends Record<string, any>, DB extends GenericDrizzleDbTypeConstraints, RequestEvent extends Record<string, any>, PothosConfig extends CustomRumblePothosConfig, Action extends string = "read" | "update" | "delete">(rumbleInput: RumbleInput<UserContext, DB, RequestEvent, Action, PothosConfig>) => {
     /**
    * The ability builder. Use it to declare whats allowed for each entity in your DB.
@@ -196,7 +202,7 @@ declare const rumble: <UserContext extends Record<string, any>, DB extends Gener
     /**
      * A function for creating default objects for your schema
      */
-    object: <ExplicitTableName extends keyof NonNullable<DB["_"]["relations"]["schema"]>, RefName extends string>({ table, refName, readAction, extend, }: {
+    object: <ExplicitTableName extends keyof NonEnumFields<NonNullable<DB["_"]["relations"]["schema"]>>, RefName extends string>({ table, refName, readAction, extend, }: {
         table: ExplicitTableName;
         refName?: RefName | undefined;
         readAction?: Action | undefined;
@@ -287,7 +293,7 @@ declare const rumble: <UserContext extends Record<string, any>, DB extends Gener
     /**
      * A function for creating where args to filter entities
      */
-    arg: <ExplicitTableName extends keyof NonNullable<DB["_"]["relations"]["schema"]>, RefName_1 extends string>({ table, refName, dbName, }: Partial<{
+    arg: <ExplicitTableName extends keyof NonEnumFields<NonNullable<DB["_"]["relations"]["schema"]>>, RefName_1 extends string>({ table, refName, dbName, }: Partial<{
         table: ExplicitTableName;
         refName: RefName_1 | undefined;
         dbName: string;
@@ -343,7 +349,7 @@ declare const rumble: <UserContext extends Record<string, any>, DB extends Gener
      * A function for creating default READ queries.
      * Make sure the objects for the table you are creating the queries for are implemented
      */
-    query: <ExplicitTableName extends keyof NonNullable<DB["_"]["relations"]["schema"]>>({ table, readAction, listAction, }: {
+    query: <ExplicitTableName extends keyof NonEnumFields<NonNullable<DB["_"]["relations"]["schema"]>>>({ table, readAction, listAction, }: {
         table: ExplicitTableName;
         readAction?: Action | undefined;
         listAction?: Action | undefined;
@@ -351,7 +357,7 @@ declare const rumble: <UserContext extends Record<string, any>, DB extends Gener
     /**
      * A function for creating a pubsub instance for a table. Use this to publish or subscribe events
      */
-    pubsub: <ExplicitTableName extends keyof NonNullable<DB["_"]["relations"]["schema"]>>({ table, }: {
+    pubsub: <ExplicitTableName extends keyof NonEnumFields<NonNullable<DB["_"]["relations"]["schema"]>>>({ table, }: {
         table: ExplicitTableName;
     }) => {
         registerOnInstance({ instance, action, primaryKeyValue, }: {

package/index.js

@@ -4,6 +4,6 @@ import {createYoga,createPubSub}from'graphql-yoga';import {One,relationsFilterTo
 
 `+$(m.loc));else if(this.source&&this.locations)for(let m of this.locations)t+=`
 
-`+k(this.source,m);return t}toJSON(){let t={message:this.message};return this.locations!=null&&(t.locations=this.locations),this.path!=null&&(t.path=this.path),this.extensions!=null&&Object.keys(this.extensions).length>0&&(t.extensions=this.extensions),t}};function H(e){return e===void 0||e.length===0?void 0:e}var C=class extends Error{constructor(t){super(t),this.name="RumbleError";}},F=class extends v{};var O=(e,t)=>new C(`RumbleError: Unknown SQL type '${e}'. Please open an issue (https://github.com/m1212e/rumble/issues) so it can be added. (${t})`);function J(e){if(["serial","int","integer","tinyint","smallint","mediumint"].includes(e))return {value1:1,value2:2};if(["real","decimal","double","float"].includes(e))return {value1:1.1,value2:2.2};if(["string","text","varchar","char","text(256)"].includes(e))return {value1:"a",value2:"b"};if(["uuid"].includes(e))return {value1:"fba31870-5528-42d7-b27e-2e5ee657aea5",value2:"fc65db81-c2d1-483d-8a25-a30e2cf6e02d"};if(["boolean"].includes(e))return {value1:true,value2:false};if(["timestamp","datetime"].includes(e))return {value1:new Date(2022,1,1),value2:new Date(2022,1,2)};if(["date"].includes(e))return {value1:new Date(2022,1,1),value2:new Date(2022,1,2)};if(["json"].includes(e))return {value1:{a:1},value2:{b:2}};throw O(e,"Distinct")}var G=Symbol.for("drizzle:Name"),W=Symbol.for("drizzle:Columns");function E({dbName:e,tsName:t,table:m,db:u}){let n=m;if(t&&(n=u._.relations.schema[t]),e&&(n=Object.values(u._.relations.schema).find(a=>a[G]===e)),!n)throw new C(`Could not find schema for ${JSON.stringify({tsName:t,dbName:e,table:m?.[G]}).toString()}`);return {tableSchema:n,columns:n[W],get primaryColumns(){return Object.entries(n[W]).filter(([a,i])=>i.primary).reduce((a,[i,l])=>(a[i]=l,a),{})},relations:u._.relations.config[t],dbName:n[G],get tsName(){return Object.entries(u._.relations.schema).find(([a,i])=>i===n).at(0)}}}function fe(e){return typeof e!="function"}function Te(e){return typeof e=="function"&&e.constructor.name!=="AsyncFunction"}var Y=({db:e,actions:t,defaultLimit:m})=>{let u={},n={},a={},i=l=>{for(let o of t)a[l]||(a[l]={}),a[l][o]||(a[l][o]=[]);return {allow:o=>{let p=n[l];p||(p={},n[l]=p);let r=Array.isArray(o)?o:[o];for(let f of r){let s=p[f];s||(s="unspecified",p[f]=s);}return {when:f=>{for(let s of r)p[s]==="unspecified"&&(p[s]=[]),p[s].push(f);}}},filter:o=>{let p=Array.isArray(o)?o:[o];return {by:r=>{for(let f of p)a[l][f].push(r);}}}}};for(let l of Object.keys(e.query))u[l]=i(l);return {...u,registeredQueryFilters:n,registeredFilters:a,buildWithUserContext:l=>{let o={},p=r=>({filter:(f,s)=>{let d=y=>{let B=N(()=>{if(!(!y?.where&&!s?.inject?.where))return s?.inject?.where?{AND:[y?.where,s?.inject?.where]}:y?.where}),A=N(()=>{let P=B();if(!P)return;let ue=E({tsName:r,db:e});return relationsFilterToSQL(ue.tableSchema,P)}),q=N(()=>{let P=y?.limit??m;return s?.inject?.limit&&(!P||P>s.inject.limit)&&(P=s.inject.limit),y?.limit&&(!P||y.limit>P)&&(P=y.limit),P??void 0}),w=N(()=>{if(!(!y?.columns&&!s?.inject?.columns))return {...y?.columns,...s?.inject?.columns}});return {query:{single:{get where(){return B()},get columns(){return w()}},many:{get where(){return B()},get columns(){return w()},get limit(){return q()}}},sql:{get where(){return A()},get columns(){return w()},get limit(){return q()}}}},c=()=>{let y=E({db:e,tsName:r});if(Object.keys(y.primaryColumns).length===0)throw new C(`No primary key found for entity ${r.toString()}`);let B=Object.values(y.primaryColumns)[0],A=J(B.getSQLType());return {where:{AND:[{[B.name]:A.value1},{[B.name]:A.value2}]}}},b=n?.[r]?.[f];if(b==="unspecified")return d();b||(b=[c()]);let x=b.filter(fe),D=b.filter(Te).map(y=>y(l)),h=D.some(y=>y==="allow"),T=[...x,...D].filter(y=>y!==void 0).filter(y=>y!=="allow");!h&&T.length===0&&(T=[c()]);let g;for(let y of T)y?.limit&&(g===void 0||y.limit>g)&&(g=y.limit);let S;for(let y of [...T,s?.inject])y?.columns&&(S===void 0?S=y.columns:S={...S,...y.columns});let R=h?[]:T.filter(y=>y?.where).map(y=>y.where),L=R.length>0?{OR:R}:void 0;return d({where:L,columns:S,limit:g})},explicitFilters:f=>a[r][f]});for(let r of Object.keys(e.query))o[r]=p(r);return o}}};var X=({context:e,abilityBuilder:t})=>async m=>{let u=e?await e(m):{};return {...u,abilities:t.buildWithUserContext(u)}};function I(e){return String(e).charAt(0).toUpperCase()+String(e).slice(1)}function U(e){return e instanceof PgEnumColumn}var Z=({db:e,schemaBuilder:t})=>{let m=new Map;return ({tsName:n,enumColumn:a,refName:i})=>{let l,o;if(n){let s=E({db:e,tsName:n});l=n.toString();let d=Object.values(e._.relations.schema).filter(c=>typeof c=="object").map(c=>Object.values(c[Symbol.for("drizzle:Columns")])).flat(2).find(c=>c.config?.enum===s);if(!d)throw new C(`Could not find applied enum column for ${n.toString()}.
-Please ensure that you use the enum at least once as a column of a table!`);o=d.enumValues;}else if(a){let s=Object.entries(e._.relations.schema).find(([d,c])=>c===a.config.enum);if(!s)throw new C(`Could not find enum in schema for ${a.name}!`);l=s[0],o=a.enumValues;}if(!l||!o)throw new C("Could not determine enum structure!");let p=i??`${I(toCamelCase(l))}Enum`,r=m.get(p);return r||(r=t.enumType(p,{values:o}),m.set(p,r),r)}};function Q(e){let t;if(["serial","int","integer","tinyint","smallint","mediumint"].includes(e)&&(t="Int"),["real","decimal","double","float"].includes(e)&&(t="Float"),["string","text","varchar","char","text(256)"].includes(e)&&(t="String"),["uuid"].includes(e)&&(t="ID"),["boolean"].includes(e)&&(t="Boolean"),["timestamp","datetime"].includes(e)&&(t="DateTime"),["date"].includes(e)&&(t="Date"),["json"].includes(e)&&(t="JSON"),t!==void 0)return t;throw O(e,"SQL to GQL")}var ee=({db:e,schemaBuilder:t,makePubSubInstance:m,argImplementer:u,enumImplementer:n,abilityBuilder:a})=>({table:i,refName:l,readAction:o="read",extend:p})=>{let r=E({db:e,tsName:i});Object.keys(r.primaryColumns).length===0&&console.warn(`Could not find primary key for ${i.toString()}. Cannot register subscriptions!`);let f=Object.values(r.primaryColumns)[0],{registerOnInstance:s}=m({table:i});return t.drizzleObject(i,{name:l??I(i.toString()),subscribe:(d,c,b)=>{if(!f)return;let x=c[f.name];if(!x){console.warn(`Could not find primary key value for ${JSON.stringify(c)}. Cannot register subscription!`);return}s({instance:d,action:"updated",primaryKeyValue:x});},applyFilters:a?.registeredFilters?.[i]?.[o],fields:d=>{let c=r.columns,b=(h,T,g)=>{let S=Q(h);switch(S){case "Int":return d.exposeInt(T,{nullable:g});case "String":return d.exposeString(T,{nullable:g});case "Boolean":return d.exposeBoolean(T,{nullable:g});case "Date":return d.field({type:"Date",resolve:R=>R[T],nullable:g});case "DateTime":return d.field({type:"DateTime",resolve:R=>R[T],nullable:g});case "Float":return d.exposeFloat(T,{nullable:g});case "ID":return d.exposeID(T,{nullable:g});case "JSON":return d.field({type:"JSON",resolve:R=>R[T],nullable:g});default:throw new C(`Unsupported object type ${S} for column ${T}`)}},x=Object.entries(c).reduce((h,[T,g])=>{if(U(g)){let S=n({enumColumn:g});h[T]=d.field({type:S,resolve:R=>R[T],nullable:!g.notNull});}else h[T]=b(g.getSQLType(),T,!g.notNull);return h},{}),D=Object.entries(r.relations).reduce((h,[T,g])=>{let S=E({db:e,table:g.targetTable}),R=u({dbName:S.dbName}),L=false,B="many";return g instanceof One&&(L=g.optional,B="single"),h[T]=d.relation(T,{args:{where:d.arg({type:R,required:false})},nullable:L,query:(A,q)=>q.abilities[S.tsName].filter(o,{inject:{where:A.where}}).query[B]}),h},{});return p?{...x,...D,...p(d)??{}}:{...x,...D}}})};var he="RUMBLE_SUBSCRIPTION_NOTIFICATION",Ce="REMOVED",De="UPDATED",Se="CREATED",ne=({subscriptions:e,db:t})=>{let m=e?createPubSub(...e):createPubSub();return {pubsub:m,makePubSubInstance:({table:n})=>{function a({action:i,tableName:l,primaryKeyValue:o}){let p;switch(i){case "created":p=Se;break;case "removed":p=Ce;break;case "updated":p=De;break;default:throw new Error(`Unknown action: ${i}`)}return `${he}/${l}${o?`/${o}`:""}/${p}`}return {registerOnInstance({instance:i,action:l,primaryKeyValue:o}){let p=a({tableName:n.toString(),action:l,primaryKeyValue:o});i.register(p);},created(){let i=a({tableName:n.toString(),action:"created"});return m.publish(i)},removed(i){let l=a({tableName:n.toString(),action:"removed"});return m.publish(l)},updated(i){let o=(Array.isArray(i)?i:[i]).map(r=>a({tableName:n.toString(),action:"updated",primaryKeyValue:r})),p=Array.from(new Set(o));for(let r of p)m.publish(r);}}}}};var j=e=>{if(!e)throw new F("Value not found but required (findFirst)");return e},Re=e=>{let t=e.at(0);if(!t)throw new F("Value not found but required (firstEntry)");return t},re=async({filters:e,entities:t,context:m})=>(await Promise.all(e.map(u=>u({context:m,entities:t})))).reduce((u,n)=>(u.push(...n),u),[]);var ie=({db:e,schemaBuilder:t,argImplementer:m,makePubSubInstance:u})=>({table:n,readAction:a="read",listAction:i="read"})=>{let l=m({table:n}),{registerOnInstance:o}=u({table:n});return t.queryFields(p=>({[`findMany${I(n.toString())}`]:p.drizzleField({type:[n],nullable:false,smartSubscription:true,subscribe:(r,f,s,d,c)=>{o({instance:r,action:"created"}),o({instance:r,action:"removed"});},args:{where:p.arg({type:l,required:false})},resolve:(r,f,s,d,c)=>{let b=d.abilities[n].filter(i,{inject:{where:s.where}}).query.many,x=r(b);return b.columns&&(x.columns=b.columns),e.query[n].findMany(x)}}),[`findFirst${I(n.toString())}`]:p.drizzleField({type:n,nullable:false,smartSubscription:true,args:{where:p.arg({type:l,required:false})},resolve:(r,f,s,d,c)=>{let b=d.abilities[n].filter(a,{inject:{where:s.where}}).query.single,x=r(b);return b.columns&&(x.columns=b.columns),e.query[n].findFirst(x).then(j)}})}))};var oe="ManualFiltersPlugin",se=oe,Pe="applyFilters",K=class extends BasePlugin{wrapResolve(t,m){return async(u,n,a,i)=>{let l=(m?.type).type?.ref.currentConfig.pothosOptions[Pe];if(!l||!Array.isArray(l)||l.length===0)return t(u,n,a,i);let o=await t(u,n,a,i),p=Array.isArray(o)?o:[o],r=Array.isArray(l)?l:[l],f=await re({filters:r,entities:p,context:a});return Array.isArray(o)?f:f[0]??null}}};Ee.registerPlugin(oe,K);var ae=({db:e,disableDefaultObjects:t,pubsub:m,pothosConfig:u})=>{let n=new Ee({plugins:[se,Ae,ve,...u?.plugins??[]],...u,drizzle:{client:e,relations:e._.relations,getTableConfig(a){return {columns:Object.values(a[Symbol.for("drizzle:Columns")]),primaryKeys:Object.values(a[Symbol.for("drizzle:Columns")]).filter(i=>i.primary)}}},smartSubscriptions:{...subscribeOptionsFromIterator((a,i)=>m.subscribe(a))}});return n.addScalarType("JSON",JSONResolver),n.addScalarType("Date",DateResolver),n.addScalarType("DateTime",DateTimeISOResolver),t?.query||n.queryType({}),t?.subscription||n.subscriptionType({}),t?.mutation||n.mutationType({}),{schemaBuilder:n}};var Oe=e=>`${I(toCamelCase(e.toString()))}WhereInputArgument`,le=({db:e,schemaBuilder:t,enumImplementer:m})=>{let u=new Map,n=({table:a,refName:i,dbName:l})=>{let o=E({db:e,dbName:l,tsName:a}),p=i??Oe(o.tsName),r=u.get(p);return r||(r=t.inputType(p,{fields:s=>{let d=x=>{let D=Q(x);switch(D){case "Int":return s.int({required:false});case "String":return s.string({required:false});case "Boolean":return s.boolean({required:false});case "Date":return s.field({type:"Date",required:false});case "DateTime":return s.field({type:"DateTime",required:false});case "Float":return s.float({required:false});case "ID":return s.id({required:false});case "JSON":return s.field({type:"JSON",required:false});default:throw new C(`Unsupported argument type ${D} for column ${x}`)}},c=Object.entries(o.columns).reduce((x,[D,h])=>{if(U(h)){let T=m({enumColumn:h});x[D]=s.field({type:T,required:false});}else x[D]=d(h.getSQLType());return x},{}),b=Object.entries(o.relations).reduce((x,[D,h])=>{let T=E({db:e,table:h.targetTable}),g=n({dbName:T.dbName});return x[D]=s.field({type:g,required:false}),x},{});return {...c,...b}}}),u.set(p,r),r)};return n};var Qe=e=>{e.actions||(e.actions=["read","update","delete"]),e.defaultLimit===void 0&&(e.defaultLimit=100);let t=Y(e),m=X({...e,abilityBuilder:t}),{makePubSubInstance:u,pubsub:n}=ne({...e}),{schemaBuilder:a}=ae({...e,pubsub:n}),i=Z({...e,schemaBuilder:a}),l=le({...e,schemaBuilder:a,enumImplementer:i}),o=ee({...e,schemaBuilder:a,makePubSubInstance:u,argImplementer:l,enumImplementer:i,abilityBuilder:t}),p=ie({...e,schemaBuilder:a,argImplementer:l,makePubSubInstance:u});return {abilityBuilder:t,schemaBuilder:a,createYoga:f=>createYoga({...f,schema:a.toSchema(),context:m}),object:o,arg:l,query:p,pubsub:u,enum_:i}};export{C as RumbleError,F as RumbleErrorSafe,j as assertFindFirstExists,Re as assertFirstEntryExists,Qe as rumble};//# sourceMappingURL=index.js.map
+`+k(this.source,m);return t}toJSON(){let t={message:this.message};return this.locations!=null&&(t.locations=this.locations),this.path!=null&&(t.path=this.path),this.extensions!=null&&Object.keys(this.extensions).length>0&&(t.extensions=this.extensions),t}};function H(e){return e===void 0||e.length===0?void 0:e}var C=class extends Error{constructor(t){super(t),this.name="RumbleError";}},F=class extends v{};var O=(e,t)=>new C(`RumbleError: Unknown SQL type '${e}'. Please open an issue (https://github.com/m1212e/rumble/issues) so it can be added. (${t})`);function J(e){if(["serial","int","integer","tinyint","smallint","mediumint"].includes(e))return {value1:1,value2:2};if(["real","decimal","double","float"].includes(e))return {value1:1.1,value2:2.2};if(["string","text","varchar","char","text(256)"].includes(e))return {value1:"a",value2:"b"};if(["uuid"].includes(e))return {value1:"fba31870-5528-42d7-b27e-2e5ee657aea5",value2:"fc65db81-c2d1-483d-8a25-a30e2cf6e02d"};if(["boolean"].includes(e))return {value1:true,value2:false};if(["timestamp","datetime"].includes(e))return {value1:new Date(2022,1,1),value2:new Date(2022,1,2)};if(["date"].includes(e))return {value1:new Date(2022,1,1),value2:new Date(2022,1,2)};if(["json"].includes(e))return {value1:{a:1},value2:{b:2}};throw O(e,"Distinct")}var G=Symbol.for("drizzle:Name"),W=Symbol.for("drizzle:Columns");function P({dbName:e,tsName:t,table:m,db:u}){let n=m;if(t&&(n=u._.relations.schema[t]),e&&(n=Object.values(u._.relations.schema).find(a=>a[G]===e)),!n)throw new C(`Could not find schema for ${JSON.stringify({tsName:t,dbName:e,table:m?.[G]}).toString()}`);return {tableSchema:n,columns:n[W],get primaryColumns(){return Object.entries(n[W]).filter(([a,i])=>i.primary).reduce((a,[i,l])=>(a[i]=l,a),{})},relations:u._.relations.config[t],dbName:n[G],get tsName(){return Object.entries(u._.relations.schema).find(([a,i])=>i===n).at(0)}}}function fe(e){return typeof e!="function"}function Te(e){return typeof e=="function"&&e.constructor.name!=="AsyncFunction"}var Y=({db:e,actions:t,defaultLimit:m})=>{let u={},n={},a={},i=l=>{for(let o of t)a[l]||(a[l]={}),a[l][o]||(a[l][o]=[]);return {allow:o=>{let p=n[l];p||(p={},n[l]=p);let r=Array.isArray(o)?o:[o];for(let f of r){let s=p[f];s||(s="unspecified",p[f]=s);}return {when:f=>{for(let s of r)p[s]==="unspecified"&&(p[s]=[]),p[s].push(f);}}},filter:o=>{let p=Array.isArray(o)?o:[o];return {by:r=>{for(let f of p)a[l][f].push(r);}}}}};for(let l of Object.keys(e.query))u[l]=i(l);return {...u,registeredQueryFilters:n,registeredFilters:a,buildWithUserContext:l=>{let o={},p=r=>({filter:(f,s)=>{let d=y=>{let E=N(()=>{if(!(!y?.where&&!s?.inject?.where))return s?.inject?.where?{AND:[y?.where,s?.inject?.where]}:y?.where}),A=N(()=>{let B=E();if(!B)return;let ue=P({tsName:r,db:e});return relationsFilterToSQL(ue.tableSchema,B)}),q=N(()=>{let B=y?.limit??m;return s?.inject?.limit&&(!B||B>s.inject.limit)&&(B=s.inject.limit),y?.limit&&(!B||y.limit>B)&&(B=y.limit),B??void 0}),w=N(()=>{if(!(!y?.columns&&!s?.inject?.columns))return {...y?.columns,...s?.inject?.columns}});return {query:{single:{get where(){return E()},get columns(){return w()}},many:{get where(){return E()},get columns(){return w()},get limit(){return q()}}},sql:{get where(){return A()},get columns(){return w()},get limit(){return q()}}}},c=()=>{let y=P({db:e,tsName:r});if(Object.keys(y.primaryColumns).length===0)throw new C(`No primary key found for entity ${r.toString()}`);let E=Object.values(y.primaryColumns)[0],A=J(E.getSQLType());return {where:{AND:[{[E.name]:A.value1},{[E.name]:A.value2}]}}},b=n?.[r]?.[f];if(b==="unspecified")return d();b||(b=[c()]);let x=b.filter(fe),D=b.filter(Te).map(y=>y(l)),h=D.some(y=>y==="allow"),T=[...x,...D].filter(y=>y!==void 0).filter(y=>y!=="allow");!h&&T.length===0&&(T=[c()]);let g;for(let y of T)y?.limit&&(g===void 0||y.limit>g)&&(g=y.limit);let S;for(let y of [...T,s?.inject])y?.columns&&(S===void 0?S=y.columns:S={...S,...y.columns});let R=h?[]:T.filter(y=>y?.where).map(y=>y.where),L=R.length>0?{OR:R}:void 0;return d({where:L,columns:S,limit:g})},explicitFilters:f=>a[r][f]});for(let r of Object.keys(e.query))o[r]=p(r);return o}}};var X=({context:e,abilityBuilder:t})=>async m=>{let u=e?await e(m):{};return {...u,abilities:t.buildWithUserContext(u)}};function I(e){return String(e).charAt(0).toUpperCase()+String(e).slice(1)}function U(e){return e instanceof PgEnumColumn}var Z=({db:e,schemaBuilder:t})=>{let m=new Map;return ({tsName:n,enumColumn:a,refName:i})=>{let l,o;if(n){let s=e._.relations.schema[n];l=n.toString();let d=Object.values(e._.relations.schema).filter(c=>typeof c=="object").map(c=>Object.values(c[Symbol.for("drizzle:Columns")])).flat(2).find(c=>c.config?.enum===s);if(!d)throw new C(`Could not find applied enum column for ${n.toString()}.
+Please ensure that you use the enum at least once as a column of a table!`);o=d.enumValues;}else if(a){let s=Object.entries(e._.relations.schema).find(([d,c])=>c===a.config.enum);if(!s)throw new C(`Could not find enum in schema for ${a.name}!`);l=s[0],o=a.enumValues;}if(!l||!o)throw new C("Could not determine enum structure!");let p=i??`${I(toCamelCase(l))}Enum`,r=m.get(p);return r||(r=t.enumType(p,{values:o}),m.set(p,r),r)}};function Q(e){let t;if(["serial","int","integer","tinyint","smallint","mediumint"].includes(e)&&(t="Int"),["real","decimal","double","float"].includes(e)&&(t="Float"),["string","text","varchar","char","text(256)"].includes(e)&&(t="String"),["uuid"].includes(e)&&(t="ID"),["boolean"].includes(e)&&(t="Boolean"),["timestamp","datetime"].includes(e)&&(t="DateTime"),["date"].includes(e)&&(t="Date"),["json"].includes(e)&&(t="JSON"),t!==void 0)return t;throw O(e,"SQL to GQL")}var ee=({db:e,schemaBuilder:t,makePubSubInstance:m,argImplementer:u,enumImplementer:n,abilityBuilder:a})=>({table:i,refName:l,readAction:o="read",extend:p})=>{let r=P({db:e,tsName:i});Object.keys(r.primaryColumns).length===0&&console.warn(`Could not find primary key for ${i.toString()}. Cannot register subscriptions!`);let f=Object.values(r.primaryColumns)[0],{registerOnInstance:s}=m({table:i});return t.drizzleObject(i,{name:l??I(i.toString()),subscribe:(d,c,b)=>{if(!f)return;let x=c[f.name];if(!x){console.warn(`Could not find primary key value for ${JSON.stringify(c)}. Cannot register subscription!`);return}s({instance:d,action:"updated",primaryKeyValue:x});},applyFilters:a?.registeredFilters?.[i]?.[o],fields:d=>{let c=r.columns,b=(h,T,g)=>{let S=Q(h);switch(S){case "Int":return d.exposeInt(T,{nullable:g});case "String":return d.exposeString(T,{nullable:g});case "Boolean":return d.exposeBoolean(T,{nullable:g});case "Date":return d.field({type:"Date",resolve:R=>R[T],nullable:g});case "DateTime":return d.field({type:"DateTime",resolve:R=>R[T],nullable:g});case "Float":return d.exposeFloat(T,{nullable:g});case "ID":return d.exposeID(T,{nullable:g});case "JSON":return d.field({type:"JSON",resolve:R=>R[T],nullable:g});default:throw new C(`Unsupported object type ${S} for column ${T}`)}},x=Object.entries(c).reduce((h,[T,g])=>{if(U(g)){let S=n({enumColumn:g});h[T]=d.field({type:S,resolve:R=>R[T],nullable:!g.notNull});}else h[T]=b(g.getSQLType(),T,!g.notNull);return h},{}),D=Object.entries(r.relations??{}).reduce((h,[T,g])=>{let S=P({db:e,table:g.targetTable}),R=u({dbName:S.dbName}),L=false,E="many";return g instanceof One&&(L=g.optional,E="single"),h[T]=d.relation(T,{args:{where:d.arg({type:R,required:false})},nullable:L,query:(A,q)=>q.abilities[S.tsName].filter(o,{inject:{where:A.where}}).query[E]}),h},{});return p?{...x,...D,...p(d)??{}}:{...x,...D}}})};var he="RUMBLE_SUBSCRIPTION_NOTIFICATION",Ce="REMOVED",De="UPDATED",Se="CREATED",ne=({subscriptions:e,db:t})=>{let m=e?createPubSub(...e):createPubSub();return {pubsub:m,makePubSubInstance:({table:n})=>{function a({action:i,tableName:l,primaryKeyValue:o}){let p;switch(i){case "created":p=Se;break;case "removed":p=Ce;break;case "updated":p=De;break;default:throw new Error(`Unknown action: ${i}`)}return `${he}/${l}${o?`/${o}`:""}/${p}`}return {registerOnInstance({instance:i,action:l,primaryKeyValue:o}){let p=a({tableName:n.toString(),action:l,primaryKeyValue:o});i.register(p);},created(){let i=a({tableName:n.toString(),action:"created"});return m.publish(i)},removed(i){let l=a({tableName:n.toString(),action:"removed"});return m.publish(l)},updated(i){let o=(Array.isArray(i)?i:[i]).map(r=>a({tableName:n.toString(),action:"updated",primaryKeyValue:r})),p=Array.from(new Set(o));for(let r of p)m.publish(r);}}}}};var j=e=>{if(!e)throw new F("Value not found but required (findFirst)");return e},Re=e=>{let t=e.at(0);if(!t)throw new F("Value not found but required (firstEntry)");return t},re=async({filters:e,entities:t,context:m})=>(await Promise.all(e.map(u=>u({context:m,entities:t})))).reduce((u,n)=>(u.push(...n),u),[]);var ie=({db:e,schemaBuilder:t,argImplementer:m,makePubSubInstance:u})=>({table:n,readAction:a="read",listAction:i="read"})=>{let l=m({table:n}),{registerOnInstance:o}=u({table:n});return t.queryFields(p=>({[`findMany${I(n.toString())}`]:p.drizzleField({type:[n],nullable:false,smartSubscription:true,subscribe:(r,f,s,d,c)=>{o({instance:r,action:"created"}),o({instance:r,action:"removed"});},args:{where:p.arg({type:l,required:false})},resolve:(r,f,s,d,c)=>{let b=d.abilities[n].filter(i,{inject:{where:s.where}}).query.many,x=r(b);return b.columns&&(x.columns=b.columns),e.query[n].findMany(x)}}),[`findFirst${I(n.toString())}`]:p.drizzleField({type:n,nullable:false,smartSubscription:true,args:{where:p.arg({type:l,required:false})},resolve:(r,f,s,d,c)=>{let b=d.abilities[n].filter(a,{inject:{where:s.where}}).query.single,x=r(b);return b.columns&&(x.columns=b.columns),e.query[n].findFirst(x).then(j)}})}))};var oe="ManualFiltersPlugin",se=oe,Pe="applyFilters",K=class extends BasePlugin{wrapResolve(t,m){return async(u,n,a,i)=>{let l=(m?.type).type?.ref.currentConfig.pothosOptions[Pe];if(!l||!Array.isArray(l)||l.length===0)return t(u,n,a,i);let o=await t(u,n,a,i),p=Array.isArray(o)?o:[o],r=Array.isArray(l)?l:[l],f=await re({filters:r,entities:p,context:a});return Array.isArray(o)?f:f[0]??null}}};Ee.registerPlugin(oe,K);var ae=({db:e,disableDefaultObjects:t,pubsub:m,pothosConfig:u})=>{let n=new Ee({plugins:[se,Ae,ve,...u?.plugins??[]],...u,drizzle:{client:e,relations:e._.relations,getTableConfig(a){return {columns:Object.values(a[Symbol.for("drizzle:Columns")]),primaryKeys:Object.values(a[Symbol.for("drizzle:Columns")]).filter(i=>i.primary)}}},smartSubscriptions:{...subscribeOptionsFromIterator((a,i)=>m.subscribe(a))}});return n.addScalarType("JSON",JSONResolver),n.addScalarType("Date",DateResolver),n.addScalarType("DateTime",DateTimeISOResolver),t?.query||n.queryType({}),t?.subscription||n.subscriptionType({}),t?.mutation||n.mutationType({}),{schemaBuilder:n}};var Oe=e=>`${I(toCamelCase(e.toString()))}WhereInputArgument`,le=({db:e,schemaBuilder:t,enumImplementer:m})=>{let u=new Map,n=({table:a,refName:i,dbName:l})=>{let o=P({db:e,dbName:l,tsName:a}),p=i??Oe(o.tsName),r=u.get(p);return r||(r=t.inputType(p,{fields:s=>{let d=x=>{let D=Q(x);switch(D){case "Int":return s.int({required:false});case "String":return s.string({required:false});case "Boolean":return s.boolean({required:false});case "Date":return s.field({type:"Date",required:false});case "DateTime":return s.field({type:"DateTime",required:false});case "Float":return s.float({required:false});case "ID":return s.id({required:false});case "JSON":return s.field({type:"JSON",required:false});default:throw new C(`Unsupported argument type ${D} for column ${x}`)}},c=Object.entries(o.columns).reduce((x,[D,h])=>{if(U(h)){let T=m({enumColumn:h});x[D]=s.field({type:T,required:false});}else x[D]=d(h.getSQLType());return x},{}),b=Object.entries(o.relations??{}).reduce((x,[D,h])=>{let T=P({db:e,table:h.targetTable}),g=n({dbName:T.dbName});return x[D]=s.field({type:g,required:false}),x},{});return {...c,...b}}}),u.set(p,r),r)};return n};var Qe=e=>{e.actions||(e.actions=["read","update","delete"]),e.defaultLimit===void 0&&(e.defaultLimit=100);let t=Y(e),m=X({...e,abilityBuilder:t}),{makePubSubInstance:u,pubsub:n}=ne({...e}),{schemaBuilder:a}=ae({...e,pubsub:n}),i=Z({...e,schemaBuilder:a}),l=le({...e,schemaBuilder:a,enumImplementer:i}),o=ee({...e,schemaBuilder:a,makePubSubInstance:u,argImplementer:l,enumImplementer:i,abilityBuilder:t}),p=ie({...e,schemaBuilder:a,argImplementer:l,makePubSubInstance:u});return {abilityBuilder:t,schemaBuilder:a,createYoga:f=>createYoga({...f,schema:a.toSchema(),context:m}),object:o,arg:l,query:p,pubsub:u,enum_:i}};export{C as RumbleError,F as RumbleErrorSafe,j as assertFindFirstExists,Re as assertFirstEntryExists,Qe as rumble};//# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map