@lakeql/cli 0.2.3 → 0.3.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (2) hide show
  1. package/dist/cli.mjs +7 -7
  2. package/package.json +4 -4
package/dist/cli.mjs CHANGED
@@ -1,9 +1,9 @@
1
1
  #!/usr/bin/env node
2
- import{Command as e,Option as t}from"@commander-js/extra-typings";import{readPackage as n}from"read-pkg";import{existsSync as r,readFileSync as i}from"node:fs";import{mkdir as a,readFile as o,rm as s,writeFile as c}from"node:fs/promises";import l from"node:path";import{generateCode as u}from"@lakeql/file-generator";import{generateConfig as d}from"@lakeql/file-generator/config";import{generateInterface as f}from"@lakeql/file-generator/interface";import{generateQuerySchema as ee}from"@lakeql/file-generator/query-schema";import{generateModel as p}from"@lakeql/schema-generator/graphql-schema";import{generateJsonSchemaFromFields as m}from"@lakeql/schema-generator/json-schema";import{generateConfigReqistry as h}from"@lakeql/file-generator/config-registry";import{globby as g}from"globby";import{camelCase as _,upperFirst as v}from"lodash-es";import y from"typescript";import{endpointDefinitionSchema as b,findDuplicateFieldNames as x}from"@lakeql/schema-generator/endpoint-schema";import{TrinoClient as S}from"@lakeql/trino-client";import{multiselect as C,question as w,select as T,validators as E}from"@topcli/prompts";import{ClimtTable as D}from"climt";import{createEnv as O}from"@t3-oss/env-core";import k from"dotenv";import{enum as te,number as ne,string as A}from"zod/v4";import{parseColumns as j}from"@lakeql/column-parser";import{error as M}from"@lakeql/logger/console";import{convertTrinoResponse as re}from"@lakeql/response-transformer";import N from"kleur";function P(e,t=0){let n=[],r=` `.repeat(t+1);for(let i of e)if(i.type===`Object`)n.push(`${r}${i.name}: Object`),i.fields&&n.push(P(i.fields,t+1));else if(i.type===`Array`){let e=i.items?.type??`Unknown`;n.push(`${r}${i.name}: Array<${e}>`),i.items?.type===`Object`&&i.items.fields&&n.push(P(i.items.fields,t+1))}else n.push(`${r}${i.name}: ${i.type}`);return n.filter(e=>e.length>0).join(`
3
- `)}function F(){return process.env.INIT_CWD??process.cwd()}const I=`lakeql.config.json`,ie={sourcePath:`.`};function ae(){let e=l.join(F(),I);if(!r(e))return ie;let t=i(e,`utf-8`),n=JSON.parse(t);return{...ie,...n}}function L(e){let t=ae(),n=F(),r=e??t.sourcePath;return l.isAbsolute(r)?r:l.resolve(n,r)}const R=new t(`--catalog <catalog>`,`catalog to use`).env(`HIVE_CATALOG`),z=new t(`--schema <schema>`,`schema to use`),oe=new t(`--table <table>`,`table to use`),se=new t(`--type <type>`,`Show tables or views`),B=new t(`--source-path <path>`,"Base path for generated code (resolved from the command invocation directory). Files are created in `schemas/generated|custom` inside this path.").default(F(),`command invocation directory`),ce=new t(`--skip-registry`,`Skip config registry generation`).default(!1);function le(){return new e(`create-registry`).description("Generates the config registry to ensure the type-safety while using `createPermission`").addOption(B).action(async({sourcePath:e})=>{await V(e!==process.env.INIT_CWD&&e!==process.cwd()?e:void 0)})}async function V(e){let t=L(e),n=await u({fileName:`config-registry.ts`,nodes:h({configPaths:(await g(`schemas/**/config.ts`,{cwd:t,onlyFiles:!0})).map(e=>{let t=l.parse(e);return l.join(t.dir,t.name)})})});r(l.join(t,`config-registry.ts`))&&await s(l.join(t,`config-registry.ts`),{force:!0,recursive:!0}),await c(l.join(t,`config-registry.ts`),n.text)}function ue(e,t){return{baseClassName:`${v(_(e))}_${v(_(t))}`,queryName:`${_(e)}${v(_(t))}`,mutationName:`create${_(e)}${v(_(t))}`}}const{factory:H}=y;function U(e){return H.createIdentifier(e)}function W(e){return U(e)}function G(e){return H.createStringLiteral(e)}function K(e,t){return H.createPropertyAssignment(U(e),t)}function q(e,t=!0){return H.createObjectLiteralExpression(e,t)}function de(e){return e?H.createTrue():H.createFalse()}function fe(e){return H.createParameterDeclaration(void 0,void 0,U(e))}function pe(e,t,n){return H.createArrowFunction(n?.async?[H.createToken(y.SyntaxKind.AsyncKeyword)]:void 0,void 0,e.map(e=>typeof e==`string`?fe(e):e),void 0,H.createToken(y.SyntaxKind.EqualsGreaterThanToken),t)}function me(e){return H.createParenthesizedExpression(e)}function he(e,t=!0){return pe([`t`],me(q(e,t)))}function J(e,t){return H.createPropertyAccessExpression(typeof e==`string`?U(e):e,U(t))}function ge(e,t=[],n){return H.createCallExpression(e,n,t)}function Y(e,t,n=[],r){return ge(J(e,t),n,r)}function _e(e,t=[],n){return Y(`builder`,e,t,n)}function ve(e,t){return H.createVariableStatement(void 0,H.createVariableDeclarationList([H.createVariableDeclaration(U(e),void 0,void 0,t)],y.NodeFlags.Const))}function X(e,t,n){return H.createImportDeclaration(void 0,H.createImportClause(n?.typeOnly??!1,void 0,H.createNamedImports(t.map(e=>H.createImportSpecifier(!1,void 0,U(e))))),G(e))}function ye(e,t,n){return H.createImportDeclaration(void 0,H.createImportClause(n?.typeOnly??!1,U(t),void 0),G(e))}function be({models:e,mutationName:t,mutationConfig:n,hasValidations:r,fieldDefinitions:i}){if(!n)return[];let a=Object.values(e).find(e=>e.root);if(!a)return[];let o=xe(i),s=Object.values(e).filter(e=>!e.root);return[...Se(r),...s.map(t=>Ce(t,e,o)),Ce(a,e,o),De(t,a,n,r)]}function xe(e){let t=new Map;if(!e)return t;for(let n of e){let e=n.options?.required===!0;t.set(n.name,e)}return t}function Se(e){let t=[X(`@lakeql/adapters`,[`executeWritePipeline`]),X(`@lakeql/trino-client`,[`TrinoClient`]),X(`@lakeql/api/builder`,[`builder`]),X(`~/env`,[`env`]),X(`./config`,[`hiveConfig`]),ye(`./json-schema.json`,`jsonSchema`)];return e&&t.push(X(`./validations`,[`validationSchema`])),t}function Ce(e,t,n){let r=`${e.modelName}Input`,i=Object.values(e.fields).map(e=>we(e,t,n));return ve(r,_e(`inputType`,[G(r),q([K(`fields`,he(i))])]))}function we(e,t,n){let r=K(`required`,de(n.get(e.name)??!0));if(e.isArray){let t=e.graphqlType.replace(/^\[/u,``).replace(/\]$/u,``),n=Ee(t);if(n)return K(e.name,Y(`t`,n,[q([r],!1)]));let i=`${t}Input`;return K(e.name,Y(`t`,`field`,[q([K(`type`,y.factory.createArrayLiteralExpression([W(i)])),r],!1)]))}let i=Te(e.graphqlType);if(i)return K(e.name,Y(`t`,i,[q([r],!1)]));if(Object.keys(t).includes(e.graphqlType)){let t=`${e.graphqlType}Input`;return K(e.name,Y(`t`,`field`,[q([K(`type`,W(t)),r],!1)]))}return K(e.name,Y(`t`,`string`,[q([r],!1)]))}function Te(e){switch(e){case`String`:return`string`;case`Int`:return`int`;case`Float`:return`float`;case`Boolean`:return`boolean`;default:return}}function Ee(e){switch(e){case`String`:return`stringList`;case`Int`:return`intList`;case`Float`:return`floatList`;case`Boolean`:return`booleanList`;default:return}}function De(e,t,n,r){let i=`${t.modelName}Input`,a=[];r&&a.push(y.factory.createExpressionStatement(Y(`validationSchema`,`parse`,[W(`input`)]))),a.push(y.factory.createVariableStatement(void 0,y.factory.createVariableDeclarationList([y.factory.createVariableDeclaration(W(`trinoClient`),void 0,void 0,y.factory.createNewExpression(W(`TrinoClient`),void 0,[q([K(`host`,J(`env`,`HIVE_HOST`)),K(`port`,J(`env`,`HIVE_PORT`)),K(`auth`,q([K(`type`,G(`basic`)),K(`username`,J(`env`,`HIVE_USERNAME`)),K(`password`,J(`env`,`HIVE_PASSWORD`))],!1)),K(`catalog`,J(`env`,`HIVE_CATALOG`)),K(`source`,J(`env`,`HIVE_SOURCE`))])]))],y.NodeFlags.Const)));let o=y.factory.createAwaitExpression(ge(W(`executeWritePipeline`),[q([K(`records`,W(`input`)),K(`jsonSchema`,y.factory.createAsExpression(W(`jsonSchema`),y.factory.createKeywordTypeNode(y.SyntaxKind.AnyKeyword))),K(`config`,q([K(`loadStrategy`,G(n.loadStrategy)),K(`basePath`,G(n.basePath)),K(`s3`,q([K(`bucket`,J(`env`,`S3_BUCKET`)),K(`region`,J(`env`,`S3_REGION`)),K(`endpoint`,J(`env`,`S3_ENDPOINT`)),K(`credentials`,q([K(`accessKeyId`,J(`env`,`S3_ACCESS_KEY_ID`)),K(`secretAccessKey`,J(`env`,`S3_SECRET_ACCESS_KEY`))],!1))])),K(`table`,q([K(`catalog`,J(`hiveConfig`,`catalog`)),K(`schema`,J(`hiveConfig`,`schema`)),K(`tableName`,J(`hiveConfig`,`tableName`))],!1)),K(`trinoClient`,W(`trinoClient`))]))])]));a.push(y.factory.createExpressionStatement(o)),a.push(y.factory.createReturnStatement(y.factory.createTrue()));let s=y.factory.createBlock(a,!0),c=pe([fe(`_root`),y.factory.createParameterDeclaration(void 0,void 0,y.factory.createObjectBindingPattern([y.factory.createBindingElement(void 0,void 0,y.factory.createIdentifier(`input`))]))],s,{async:!0}),l=K(e,Y(`t`,`boolean`,[q([K(`args`,q([K(`input`,Y(`t`,`arg`,[q([K(`type`,W(i)),K(`required`,de(!0))],!1)]))],!1)),K(`resolve`,c)])]));return y.factory.createExpressionStatement(_e(`mutationFields`,[he([l])]))}function Z(e){if(typeof e!=`object`||!e)return e;if(Array.isArray(e))return e.map(Z);let t={},n=Object.keys(e).toSorted();for(let r of n)t[r]=Z(e[r]);return t}function Oe(e){let t=Z(e);return`${JSON.stringify(t,null,2).replaceAll(`\r
2
+ import{Command as e,Option as t}from"@commander-js/extra-typings";import{readPackage as n}from"read-pkg";import{existsSync as r,readFileSync as i}from"node:fs";import{mkdir as a,readFile as o,rm as s,writeFile as c}from"node:fs/promises";import l from"node:path";import{confirm as u,multiselect as d,question as f,select as p,validators as m}from"@topcli/prompts";import{generateCode as h}from"@lakeql/file-generator";import{generateConfig as g}from"@lakeql/file-generator/config";import{generateInterface as _}from"@lakeql/file-generator/interface";import{generateQuerySchema as ee}from"@lakeql/file-generator/query-schema";import{generateModel as te}from"@lakeql/schema-generator/graphql-schema";import{generateJsonSchemaFromFields as ne}from"@lakeql/schema-generator/json-schema";import{generateConfigReqistry as v}from"@lakeql/file-generator/config-registry";import{globby as y}from"globby";import{camelCase as b,upperFirst as x}from"lodash-es";import S from"typescript";import{endpointDefinitionSchema as C,findDuplicateFieldNames as w}from"@lakeql/schema-generator/endpoint-schema";import{TrinoClient as T}from"@lakeql/trino-client";import{ClimtTable as E}from"climt";import{createEnv as re}from"@t3-oss/env-core";import ie from"dotenv";import{enum as ae,number as oe,string as D}from"zod/v4";import{parseColumns as O}from"@lakeql/column-parser";import{error as k}from"@lakeql/logger/console";import{convertTrinoResponse as A}from"@lakeql/response-transformer";import j from"kleur";function M(e,t=0){let n=[],r=` `.repeat(t+1);for(let i of e)if(i.type===`Object`)n.push(`${r}${i.name}: Object`),i.fields&&n.push(M(i.fields,t+1));else if(i.type===`Array`){let e=i.items?.type??`Unknown`;n.push(`${r}${i.name}: Array<${e}>`),i.items?.type===`Object`&&i.items.fields&&n.push(M(i.items.fields,t+1))}else n.push(`${r}${i.name}: ${i.type}`);return n.filter(e=>e.length>0).join(`
3
+ `)}function N(){return process.env.INIT_CWD??process.cwd()}const P=`lakeql.config.json`,se={sourcePath:`.`};function ce(){let e=l.join(N(),P);if(!r(e))return se;let t=i(e,`utf-8`),n=JSON.parse(t);return{...se,...n}}function F(e){let t=ce(),n=N(),r=e??t.sourcePath;return l.isAbsolute(r)?r:l.resolve(n,r)}const I=new t(`--catalog <catalog>`,`catalog to use`).env(`HIVE_CATALOG`),L=new t(`--schema <schema>`,`schema to use`),le=new t(`--table <table>`,`table to use`),ue=new t(`--type <type>`,`Show tables or views`),R=new t(`--source-path <path>`,"Base path for generated code (resolved from the command invocation directory). Files are created in `schemas/generated|custom` inside this path.").default(N(),`command invocation directory`),z=new t(`--skip-registry`,`Skip config registry generation`).default(!1);function de(){return new e(`create-registry`).description("Generates the config registry to ensure the type-safety while using `createPermission`").addOption(R).action(async({sourcePath:e})=>{await B(e!==process.env.INIT_CWD&&e!==process.cwd()?e:void 0)})}async function B(e){let t=F(e),n=await h({fileName:`config-registry.ts`,nodes:v({configPaths:(await y(`schemas/**/config.ts`,{cwd:t,onlyFiles:!0})).map(e=>{let t=l.parse(e);return l.join(t.dir,t.name)})})});r(l.join(t,`config-registry.ts`))&&await s(l.join(t,`config-registry.ts`),{force:!0,recursive:!0}),await c(l.join(t,`config-registry.ts`),n.text)}function fe(e,t){return{baseClassName:`${x(b(e))}_${x(b(t))}`,queryName:`${b(e)}${x(b(t))}`,mutationName:`create${b(e)}${x(b(t))}`}}const{factory:V}=S;function H(e){return V.createIdentifier(e)}function U(e){return H(e)}function W(e){return V.createStringLiteral(e)}function G(e,t){return V.createPropertyAssignment(H(e),t)}function K(e,t=!0){return V.createObjectLiteralExpression(e,t)}function pe(e){return e?V.createTrue():V.createFalse()}function me(e){return V.createParameterDeclaration(void 0,void 0,H(e))}function he(e,t,n){return V.createArrowFunction(n?.async?[V.createToken(S.SyntaxKind.AsyncKeyword)]:void 0,void 0,e.map(e=>typeof e==`string`?me(e):e),void 0,V.createToken(S.SyntaxKind.EqualsGreaterThanToken),t)}function ge(e){return V.createParenthesizedExpression(e)}function _e(e,t=!0){return he([`t`],ge(K(e,t)))}function q(e,t){return V.createPropertyAccessExpression(typeof e==`string`?H(e):e,H(t))}function ve(e,t=[],n){return V.createCallExpression(e,n,t)}function J(e,t,n=[],r){return ve(q(e,t),n,r)}function ye(e,t=[],n){return J(`builder`,e,t,n)}function be(e,t){return V.createVariableStatement(void 0,V.createVariableDeclarationList([V.createVariableDeclaration(H(e),void 0,void 0,t)],S.NodeFlags.Const))}function Y(e,t,n){return V.createImportDeclaration(void 0,V.createImportClause(n?.typeOnly??!1,void 0,V.createNamedImports(t.map(e=>V.createImportSpecifier(!1,void 0,H(e))))),W(e))}function xe(e,t,n){return V.createImportDeclaration(void 0,V.createImportClause(n?.typeOnly??!1,H(t),void 0),W(e))}function Se({models:e,mutationName:t,mutationConfig:n,hasValidations:r,fieldDefinitions:i}){if(!n)return[];let a=Object.values(e).find(e=>e.root);if(!a)return[];let o=Ce(i),s=Object.values(e).filter(e=>!e.root);return[...we(r),...s.map(t=>Te(t,e,o)),Te(a,e,o),ke(t,a,r)]}function Ce(e){let t=new Map;if(!e)return t;for(let n of e){let e=n.options?.required===!0;t.set(n.name,e)}return t}function we(e){let t=[Y(`@lakeql/adapters`,[`executeWritePipeline`]),Y(`@lakeql/trino-client`,[`TrinoClient`]),Y(`@lakeql/api/builder`,[`builder`]),Y(`~/env`,[`env`]),Y(`./config`,[`hiveConfig`,`storageConfig`]),xe(`./json-schema.json`,`jsonSchema`)];return e&&t.push(Y(`./validations`,[`validationSchema`])),t}function Te(e,t,n){let r=`${e.modelName}Input`,i=Object.values(e.fields).map(e=>Ee(e,t,n));return be(r,ye(`inputType`,[W(r),K([G(`fields`,_e(i))])]))}function Ee(e,t,n){let r=G(`required`,pe(n.get(e.name)??!0));if(e.isArray){let t=e.graphqlType.replace(/^\[/u,``).replace(/\]$/u,``),n=Oe(t);if(n)return G(e.name,J(`t`,n,[K([r],!1)]));let i=`${t}Input`;return G(e.name,J(`t`,`field`,[K([G(`type`,S.factory.createArrayLiteralExpression([U(i)])),r],!1)]))}let i=De(e.graphqlType);if(i)return G(e.name,J(`t`,i,[K([r],!1)]));if(Object.keys(t).includes(e.graphqlType)){let t=`${e.graphqlType}Input`;return G(e.name,J(`t`,`field`,[K([G(`type`,U(t)),r],!1)]))}return G(e.name,J(`t`,`string`,[K([r],!1)]))}function De(e){switch(e){case`String`:return`string`;case`Int`:return`int`;case`Float`:return`float`;case`Boolean`:return`boolean`;default:return}}function Oe(e){switch(e){case`String`:return`stringList`;case`Int`:return`intList`;case`Float`:return`floatList`;case`Boolean`:return`booleanList`;default:return}}function ke(e,t,n){let r=`${t.modelName}Input`,i=[];n&&i.push(S.factory.createExpressionStatement(J(`validationSchema`,`parse`,[U(`input`)]))),i.push(S.factory.createVariableStatement(void 0,S.factory.createVariableDeclarationList([S.factory.createVariableDeclaration(U(`trinoClient`),void 0,void 0,S.factory.createNewExpression(U(`TrinoClient`),void 0,[K([G(`host`,q(`env`,`HIVE_HOST`)),G(`port`,q(`env`,`HIVE_PORT`)),G(`auth`,K([G(`type`,W(`basic`)),G(`username`,q(`env`,`HIVE_USERNAME`)),G(`password`,q(`env`,`HIVE_PASSWORD`))],!1)),G(`catalog`,q(`env`,`HIVE_CATALOG`)),G(`source`,q(`env`,`HIVE_SOURCE`))])]))],S.NodeFlags.Const)));let a=S.factory.createAwaitExpression(ve(U(`executeWritePipeline`),[K([G(`records`,U(`input`)),G(`jsonSchema`,S.factory.createAsExpression(U(`jsonSchema`),S.factory.createKeywordTypeNode(S.SyntaxKind.AnyKeyword))),G(`config`,K([G(`loadStrategy`,q(`storageConfig`,`loadStrategy`)),G(`bucket`,q(`storageConfig`,`bucket`)),G(`basePath`,q(`storageConfig`,`basePath`)),G(`table`,K([G(`catalog`,q(`hiveConfig`,`catalog`)),G(`schema`,q(`hiveConfig`,`schema`)),G(`tableName`,q(`hiveConfig`,`tableName`))],!1)),G(`trinoClient`,U(`trinoClient`))]))])]));i.push(S.factory.createExpressionStatement(a)),i.push(S.factory.createReturnStatement(S.factory.createTrue()));let o=S.factory.createBlock(i,!0),s=he([me(`_root`),S.factory.createParameterDeclaration(void 0,void 0,S.factory.createObjectBindingPattern([S.factory.createBindingElement(void 0,void 0,S.factory.createIdentifier(`input`))]))],o,{async:!0}),c=G(e,J(`t`,`boolean`,[K([G(`args`,K([G(`input`,J(`t`,`arg`,[K([G(`type`,U(r)),G(`required`,pe(!0))],!1)]))],!1)),G(`resolve`,s)])]));return S.factory.createExpressionStatement(ye(`mutationFields`,[_e([c])]))}function X(e){if(typeof e!=`object`||!e)return e;if(Array.isArray(e))return e.map(X);let t={},n=Object.keys(e).toSorted();for(let r of n)t[r]=X(e[r]);return t}function Ae(e){let t=X(e);return`${JSON.stringify(t,null,2).replaceAll(`\r
4
4
  `,`
5
- `)}\n`}async function ke(e){let{definition:t,outputDir:n,skipRegistry:i,sourcePathOverride:o}=e,{catalog:h,schema:g,tableName:_,fields:v}=t,{baseClassName:y,queryName:b,mutationName:x}=ue(g,_),S=m(v),C=p({isRoot:!0,models:{},name:y,source:S}),w=Object.values(C).find(e=>e.root===!0),T=w?.fields?Object.values(w.fields).filter(e=>e.filter===!0).map(e=>({name:e.name,type:e.graphqlType})):[],E=[...new Set(T.map(e=>e.type))],D=[...new Set(Object.values(C).filter(e=>e.transformFields.length>0).flatMap(e=>e.transformFields))],O=[...new Set(Object.values(C).filter(e=>e.dateTimeFields.length>0).flatMap(e=>e.dateTimeFields))],k=await u({fileName:`config.ts`,nodes:d({catalog:h,queryName:b,schema:g,tableName:_,mutationName:[x]})}),te=await u({fileName:`interface.ts`,nodes:f(C)}),ne=await u({fileName:`query-schema.ts`,nodes:ee({dateTimeFields:O,filterFields:T,filterTypes:E,models:C,queryName:b,transformFields:D})}),A=v.some(e=>e.options?.validations&&e.options.validations.length>0),j=be({models:C,mutationName:x,mutationConfig:t.mutation,hasValidations:A,fieldDefinitions:v}),M=``;j.length>0&&(M=(await u({fileName:`mutation-schema.ts`,nodes:j})).text);let re=Oe(S),N=Oe(t),P=[{fileName:`config.ts`,content:k.text},{fileName:`interface.ts`,content:te.text},{fileName:`query-schema.ts`,content:ne.text},{fileName:`json-schema.json`,content:re},{fileName:`endpoint.json`,content:N}];M&&P.push({fileName:`mutation-schema.ts`,content:M}),r(n)&&await s(n,{force:!0,recursive:!0}),await a(n,{recursive:!0});for(let e of P)await c(l.join(n,e.fileName),e.content);return i||await V(o),{files:P,outputDir:n}}function Ae(){return new e(`create-endpoint`).description(`Create a custom endpoint from a JSON definition file`).addOption(new t(`--from-file <path>`,`Path to a JSON definition file conforming to the Endpoint_Definition_Format`).makeOptionMandatory(!0)).addOption(B).addOption(ce).action(async e=>{let{fromFile:t,sourcePath:n,skipRegistry:i}=e,a=l.isAbsolute(t)?t:l.resolve(process.cwd(),t);r(a)||(console.error(`Error: File not found: ${a}`),process.exit(1));let s;try{s=await o(a,`utf-8`)}catch(e){let t=e instanceof Error?e.message:String(e);console.error(`Error: Cannot read file: ${a}: ${t}`),process.exit(1)}let c;try{c=JSON.parse(s)}catch(e){let t=e instanceof Error?e.message:String(e);console.error(`Error: Invalid JSON in ${a}: ${t}`),process.exit(1)}let u=b.safeParse(c);if(!u.success){let e=u.error.issues.map(e=>` - ${e.path.length>0?e.path.join(`.`):`root`}: ${e.message}`).join(`
6
- `);console.error(`Error: Validation failed for ${a}:\n${e}`),process.exit(1)}let d=u.data,f=x(d.fields);if(f.length>0){let e=f.map(e=>{let t=e.path.length>0?e.path.join(`.`):`root`;return` - Duplicate field "${e.name}" at level: ${t}`}).join(`
7
- `);console.error(`Error: Validation failed for ${a}:\n${e}`),process.exit(1)}console.log(`
8
- Loaded definition summary:`),console.log(` tableName: ${d.tableName}`),console.log(` catalog: ${d.catalog}`),console.log(` schema: ${d.schema}`);let ee=d.mutation?d.mutation.loadStrategy:`disabled`;console.log(` mutation: ${ee}`),console.log(`
9
- Fields:`),console.log(P(d.fields)),console.log(``);let p=L(n===process.cwd()?void 0:n),m=l.join(p,`schemas/custom`,d.catalog,d.schema,d.tableName);await ke({definition:d,outputDir:m,skipRegistry:i,sourcePathOverride:n===process.cwd()?void 0:n}),console.log(`Endpoint generated successfully at: ${m}`)})}let je;function Me(){return k.config({path:`../../.env`}),O({runtimeEnv:process.env,server:{HIVE_CATALOG:A().min(1),HIVE_HOST:A(),HIVE_PASSWORD:A().min(1),HIVE_PORT:A().transform(e=>Number.parseInt(e,10)).pipe(ne()),HIVE_SOURCE:A().optional(),HIVE_USERNAME:A().min(1),LOG_LEVEL:te([`info`,`warn`,`error`,`debug`]).default(`warn`)}})}function Q(){return je||=Me(),je}function Ne(){let t=new e(`list-columns`);return t.description(`Lists the columns for the specified table`).addOption(R).addOption(z.makeOptionMandatory()).addOption(oe.makeOptionMandatory()).action(async({catalog:e,schema:t,table:n})=>{let r=Q(),i=e??r.HIVE_CATALOG,a=new S({auth:{password:r.HIVE_PASSWORD,type:`basic`,username:r.HIVE_USERNAME},catalog:i,host:r.HIVE_HOST,port:r.HIVE_PORT}),o=t;if(o===void 0){let e=await a.schemas({catalog:i});o=await T(`Choose a schema from the ${i} catalog`,{autocomplete:!0,choices:e})}let s=n;if(s===void 0){let e=await a.tables({catalog:i,schema:o});s=await T(`Choose a table from "${i}.${o}"`,{autocomplete:!0,choices:e})}let c=(await a.columns({catalog:i,schema:o,table:s})).map(([e,t,n,r])=>({description:r,extra:n,name:e,type:t})),l=new D;l.column(`Column Name`,`name`),l.column(`Type`,`type`),l.column(`Extra`,`extra`),l.column(`Description`,`description`),l.render(c)}),t}function Pe(){let t=new e(`list-schemas`);return t.description(`Lists the available schemas for the configured catalog`).addOption(R).action(async({catalog:e})=>{let t=Q(),n=e??t.HIVE_CATALOG,r=await new S({auth:{password:t.HIVE_PASSWORD,type:`basic`,username:t.HIVE_USERNAME},catalog:n,host:t.HIVE_HOST,port:t.HIVE_PORT}).schemas({catalog:n}),i=new D;i.column(`Schema Name`,`s`),i.render(r.map(e=>({s:e})))}),t}function Fe(){let t=new e(`list-tables`);return t.description(`Lists the available tables for the configured catalog/schema`).addOption(R).addOption(z).action(async({catalog:e,schema:t})=>{let n=Q(),r=e??n.HIVE_CATALOG,i=new S({auth:{password:n.HIVE_PASSWORD,type:`basic`,username:n.HIVE_USERNAME},catalog:r,host:n.HIVE_HOST,port:n.HIVE_PORT}),a=t;if(a===void 0){let e=await i.schemas({catalog:r});a=await T(`Choose a schema from the ${r} catalog`,{autocomplete:!0,choices:e})}let o=await i.tables({catalog:r,schema:a}),s=new D;s.column(`Table Name`,`t`),s.render(o.map(e=>({t:e})))}),t}function Ie(){let t=new e(`list-views`);return t.description(`Lists the available views for the configured catalog/schema`).addOption(R).addOption(z).action(async({catalog:e,schema:t})=>{let n=Q(),r=e??n.HIVE_CATALOG,i=new S({auth:{password:n.HIVE_PASSWORD,type:`basic`,username:n.HIVE_USERNAME},catalog:r,host:n.HIVE_HOST,port:n.HIVE_PORT}),a=t;if(a===void 0){let e=await i.schemas({catalog:r});a=await T(`Choose a schema from the ${r} catalog`,{choices:e})}let o=await i.views({catalog:r,schema:a}),s=new D;s.column(`View Name`,`v`),s.render(o.map(e=>({v:e})))}),t}function Le(e){switch(e.replaceAll(/[^a-zA-Z]/gu,``)){case`varchar`:return`String`;case`decimal`:case`double`:case`float`:return`Float`;case`integer`:case`bigint`:return`Integer`;case`boolean`:return`Boolean`;case`timestamp`:return`DateTime`;case`date`:return`Date`;default:return`String`}}function Re(e,t){return typeof t==`string`?{name:e,type:Le(t)}:Array.isArray(t)?{name:e,type:`Array`,items:ze(t)}:typeof t==`object`&&t?{name:e,type:`Object`,fields:$(t)}:{name:e,type:`String`}}function ze(e){if(e.length===0)return{type:`String`};let[t]=e;return typeof t==`string`?{type:Le(t)}:typeof t==`object`&&t&&!Array.isArray(t)?{type:`Object`,fields:$(t)}:{type:`String`}}function $(e){let t=[];for(let[n,r]of Object.entries(e))t.push(Re(n,r));return t}function Be(e){let{tableName:t,catalog:n,schema:r,parsedColumns:i}=e;return{version:`1.0`,tableName:t,catalog:n,schema:r,fields:$(i),mutation:!1}}function Ve(){let t=new e(`pull`),n=t.description(`Interactive query endpoint generation based on a remote table`).addOption(R).addOption(se).addOption(z.makeOptionMandatory(!1)).addOption(oe.makeOptionMandatory(!1).default([]).argParser((e,t)=>[...t,e])).addOption(ce).addOption(B);return n.action(async e=>{let n=Q(),{skipRegistry:r,sourcePath:i}=e,a=e.catalog??n.HIVE_CATALOG,{schema:o,table:s,type:c}=e,u=i===(process.env.INIT_CWD??process.cwd())?void 0:i,d=L(u),f=new S({auth:{password:n.HIVE_PASSWORD,type:`basic`,username:n.HIVE_USERNAME},catalog:a,host:n.HIVE_HOST,port:n.HIVE_PORT});if(!o){let e=await f.schemas({catalog:a});o=await T(`Choose a schema from the ${a} catalog`,{autocomplete:!0,choices:e})}if(s.length===0&&(c??=await T(`What do you want to see from ${a}/${o}`,{choices:[{label:`Show tables`,value:`tables`},{label:`Show views`,value:`views`}]})),s.length===0){let e=[];e=c===`views`?await f.views({catalog:a,schema:o}):await f.tables({catalog:a,schema:o}),e.length===0&&t.error(M(`There are no tables in schema '${a}.${o}'.`),{exitCode:0}),s=await C(`Choose the tables to pull`,{autocomplete:!0,choices:e,validators:[E.required()]})}for(let e of s){let t=j((await f.columns({catalog:a,schema:o,table:e})).map(e=>re({keys:[`name`,`type`,`extra`,`description`],values:e})));await ke({definition:Be({tableName:e,catalog:a,schema:o,parsedColumns:t}),outputDir:l.join(d,`schemas/generated`,a,o,e),skipRegistry:r,sourcePathOverride:u})}}),n}function He(){return new e(`init`).description(`Initialize a lakeql.config.json configuration file`).action(async()=>{let e=F(),t=l.join(e,I);if(r(t)&&await T(`lakeql.config.json already exists. Overwrite?`,{choices:[{label:`Yes`,value:`yes`},{label:`No`,value:`no`}]})===`no`){console.log(N.yellow(`Aborted.`));return}let n=r(l.join(e,`src`)),i;if(n)i=`src`,console.log(N.cyan(`Detected src/ directory — generated code will be placed in src/`));else{let e=await T(`Where should generated code be placed?`,{choices:[{label:`./ (project root)`,value:`.`},{label:`Custom path`,value:`__custom__`}]});i=e,e===`__custom__`&&(i=await w(`Enter the source path (relative to project root):`,{defaultValue:`.`}))}await c(t,`${JSON.stringify({sourcePath:i},null,2)}\n`),console.log(N.green(`Created ${I} at ${t}`))})}async function Ue(){let t=await n(),r=new e;r.configureHelp({sortSubcommands:!0}),r.name(`lakeql-cli`).description(`LakeQL CLI`).version(t.version).addCommand(Pe()).addCommand(Fe()).addCommand(Ie()).addCommand(Ne()).addCommand(Ve()).addCommand(Ae()).addCommand(le()).addCommand(He()),await r.parseAsync()}Ue().catch(()=>{process.exit(1)});export{};
5
+ `)}\n`}async function je(e){let{definition:t,outputDir:n,skipRegistry:i,sourcePathOverride:o}=e,{catalog:u,schema:d,tableName:f,fields:p}=t,{baseClassName:m,queryName:v,mutationName:y}=fe(d,f),b=ne(p),x=te({isRoot:!0,models:{},name:m,source:b}),S=Object.values(x).find(e=>e.root===!0),C=S?.fields?Object.values(S.fields).filter(e=>e.filter===!0).map(e=>({name:e.name,type:e.graphqlType})):[],w=[...new Set(C.map(e=>e.type))],T=[...new Set(Object.values(x).filter(e=>e.transformFields.length>0).flatMap(e=>e.transformFields))],E=[...new Set(Object.values(x).filter(e=>e.dateTimeFields.length>0).flatMap(e=>e.dateTimeFields))],re=await h({fileName:`config.ts`,nodes:g({catalog:u,queryName:v,schema:d,tableName:f,mutationName:[y],storageConfig:t.mutation?{loadStrategy:t.mutation.loadStrategy,type:t.mutation.type,bucket:t.mutation.bucket,basePath:t.mutation.basePath,region:t.mutation.region,endpoint:t.mutation.endpoint}:void 0})}),ie=await h({fileName:`interface.ts`,nodes:_(x)}),ae=await h({fileName:`query-schema.ts`,nodes:ee({dateTimeFields:E,filterFields:C,filterTypes:w,models:x,queryName:v,transformFields:T})}),oe=p.some(e=>e.options?.validations&&e.options.validations.length>0),D=Se({models:x,mutationName:y,mutationConfig:t.mutation,hasValidations:oe,fieldDefinitions:p}),O=``;D.length>0&&(O=(await h({fileName:`mutation-schema.ts`,nodes:D})).text);let k=Ae(b),A=Ae(t),j=[{fileName:`config.ts`,content:re.text},{fileName:`interface.ts`,content:ie.text},{fileName:`query-schema.ts`,content:ae.text},{fileName:`json-schema.json`,content:k},{fileName:`endpoint.json`,content:A}];O&&j.push({fileName:`mutation-schema.ts`,content:O}),r(n)&&await s(n,{force:!0,recursive:!0}),await a(n,{recursive:!0});for(let e of j)await c(l.join(n,e.fileName),e.content);return i||await B(o),{files:j,outputDir:n}}function Me(){return new e(`create-endpoint`).description(`Create a custom endpoint from a JSON definition file`).addOption(new t(`--from-file <path>`,`Path to a JSON definition file conforming to the Endpoint_Definition_Format`).makeOptionMandatory(!0)).addOption(R).addOption(z).addOption(new t(`--force`,`Overwrite existing files without prompting`).default(!1)).action(async e=>{let{fromFile:t,sourcePath:n,skipRegistry:i,force:a}=e,s=l.isAbsolute(t)?t:l.resolve(process.cwd(),t);r(s)||(console.error(`Error: File not found: ${s}`),process.exit(1));let c;try{c=await o(s,`utf-8`)}catch(e){let t=e instanceof Error?e.message:String(e);console.error(`Error: Cannot read file: ${s}: ${t}`),process.exit(1)}let d;try{d=JSON.parse(c)}catch(e){let t=e instanceof Error?e.message:String(e);console.error(`Error: Invalid JSON in ${s}: ${t}`),process.exit(1)}let f=C.safeParse(d);if(!f.success){let e=f.error.issues.map(e=>` - ${e.path.length>0?e.path.join(`.`):`root`}: ${e.message}`).join(`
6
+ `);console.error(`Error: Validation failed for ${s}:\n${e}`),process.exit(1)}let p=f.data,m=w(p.fields);if(m.length>0){let e=m.map(e=>{let t=e.path.length>0?e.path.join(`.`):`root`;return` - Duplicate field "${e.name}" at level: ${t}`}).join(`
7
+ `);console.error(`Error: Validation failed for ${s}:\n${e}`),process.exit(1)}console.log(`
8
+ Loaded definition summary:`),console.log(` tableName: ${p.tableName}`),console.log(` catalog: ${p.catalog}`),console.log(` schema: ${p.schema}`);let h=p.mutation?p.mutation.loadStrategy:`disabled`;console.log(` mutation: ${h}`),console.log(`
9
+ Fields:`),console.log(M(p.fields)),console.log(``);let g=F(n===process.cwd()?void 0:n),_=l.join(g,`schemas/custom`,p.catalog,p.schema,p.tableName);r(_)&&!a&&(await u(`Directory "${_}" already exists. Overwrite?`)||(console.log(`Aborted.`),process.exit(0))),await je({definition:p,outputDir:_,skipRegistry:i,sourcePathOverride:n===process.cwd()?void 0:n}),console.log(`Endpoint generated successfully at: ${_}`)})}let Z;function Ne(){return ie.config({path:`../../.env`}),re({runtimeEnv:process.env,server:{HIVE_CATALOG:D().min(1),HIVE_HOST:D(),HIVE_PASSWORD:D().min(1),HIVE_PORT:D().transform(e=>Number.parseInt(e,10)).pipe(oe()),HIVE_SOURCE:D().optional(),HIVE_USERNAME:D().min(1),LOG_LEVEL:ae([`info`,`warn`,`error`,`debug`]).default(`warn`)}})}function Q(){return Z||=Ne(),Z}function Pe(){let t=new e(`list-columns`);return t.description(`Lists the columns for the specified table`).addOption(I).addOption(L.makeOptionMandatory()).addOption(le.makeOptionMandatory()).action(async({catalog:e,schema:t,table:n})=>{let r=Q(),i=e??r.HIVE_CATALOG,a=new T({auth:{password:r.HIVE_PASSWORD,type:`basic`,username:r.HIVE_USERNAME},catalog:i,host:r.HIVE_HOST,port:r.HIVE_PORT}),o=t;if(o===void 0){let e=await a.schemas({catalog:i});o=await p(`Choose a schema from the ${i} catalog`,{autocomplete:!0,choices:e})}let s=n;if(s===void 0){let e=await a.tables({catalog:i,schema:o});s=await p(`Choose a table from "${i}.${o}"`,{autocomplete:!0,choices:e})}let c=(await a.columns({catalog:i,schema:o,table:s})).map(([e,t,n,r])=>({description:r,extra:n,name:e,type:t})),l=new E;l.column(`Column Name`,`name`),l.column(`Type`,`type`),l.column(`Extra`,`extra`),l.column(`Description`,`description`),l.render(c)}),t}function Fe(){let t=new e(`list-schemas`);return t.description(`Lists the available schemas for the configured catalog`).addOption(I).action(async({catalog:e})=>{let t=Q(),n=e??t.HIVE_CATALOG,r=await new T({auth:{password:t.HIVE_PASSWORD,type:`basic`,username:t.HIVE_USERNAME},catalog:n,host:t.HIVE_HOST,port:t.HIVE_PORT}).schemas({catalog:n}),i=new E;i.column(`Schema Name`,`s`),i.render(r.map(e=>({s:e})))}),t}function Ie(){let t=new e(`list-tables`);return t.description(`Lists the available tables for the configured catalog/schema`).addOption(I).addOption(L).action(async({catalog:e,schema:t})=>{let n=Q(),r=e??n.HIVE_CATALOG,i=new T({auth:{password:n.HIVE_PASSWORD,type:`basic`,username:n.HIVE_USERNAME},catalog:r,host:n.HIVE_HOST,port:n.HIVE_PORT}),a=t;if(a===void 0){let e=await i.schemas({catalog:r});a=await p(`Choose a schema from the ${r} catalog`,{autocomplete:!0,choices:e})}let o=await i.tables({catalog:r,schema:a}),s=new E;s.column(`Table Name`,`t`),s.render(o.map(e=>({t:e})))}),t}function Le(){let t=new e(`list-views`);return t.description(`Lists the available views for the configured catalog/schema`).addOption(I).addOption(L).action(async({catalog:e,schema:t})=>{let n=Q(),r=e??n.HIVE_CATALOG,i=new T({auth:{password:n.HIVE_PASSWORD,type:`basic`,username:n.HIVE_USERNAME},catalog:r,host:n.HIVE_HOST,port:n.HIVE_PORT}),a=t;if(a===void 0){let e=await i.schemas({catalog:r});a=await p(`Choose a schema from the ${r} catalog`,{choices:e})}let o=await i.views({catalog:r,schema:a}),s=new E;s.column(`View Name`,`v`),s.render(o.map(e=>({v:e})))}),t}function Re(e){switch(e.replaceAll(/[^a-zA-Z]/gu,``)){case`varchar`:return`String`;case`decimal`:case`double`:case`float`:return`Float`;case`integer`:case`bigint`:return`Integer`;case`boolean`:return`Boolean`;case`timestamp`:return`DateTime`;case`date`:return`Date`;default:return`String`}}function ze(e,t){return typeof t==`string`?{name:e,type:Re(t)}:Array.isArray(t)?{name:e,type:`Array`,items:Be(t)}:typeof t==`object`&&t?{name:e,type:`Object`,fields:$(t)}:{name:e,type:`String`}}function Be(e){if(e.length===0)return{type:`String`};let[t]=e;return typeof t==`string`?{type:Re(t)}:typeof t==`object`&&t&&!Array.isArray(t)?{type:`Object`,fields:$(t)}:{type:`String`}}function $(e){let t=[];for(let[n,r]of Object.entries(e))t.push(ze(n,r));return t}function Ve(e){let{tableName:t,catalog:n,schema:r,parsedColumns:i}=e;return{version:`1.0`,tableName:t,catalog:n,schema:r,fields:$(i),mutation:!1}}function He(){let t=new e(`pull`),n=t.description(`Interactive query endpoint generation based on a remote table`).addOption(I).addOption(ue).addOption(L.makeOptionMandatory(!1)).addOption(le.makeOptionMandatory(!1).default([]).argParser((e,t)=>[...t,e])).addOption(z).addOption(R);return n.action(async e=>{let n=Q(),{skipRegistry:r,sourcePath:i}=e,a=e.catalog??n.HIVE_CATALOG,{schema:o,table:s,type:c}=e,u=i===(process.env.INIT_CWD??process.cwd())?void 0:i,f=F(u),h=new T({auth:{password:n.HIVE_PASSWORD,type:`basic`,username:n.HIVE_USERNAME},catalog:a,host:n.HIVE_HOST,port:n.HIVE_PORT});if(!o){let e=await h.schemas({catalog:a});o=await p(`Choose a schema from the ${a} catalog`,{autocomplete:!0,choices:e})}if(s.length===0&&(c??=await p(`What do you want to see from ${a}/${o}`,{choices:[{label:`Show tables`,value:`tables`},{label:`Show views`,value:`views`}]})),s.length===0){let e=[];e=c===`views`?await h.views({catalog:a,schema:o}):await h.tables({catalog:a,schema:o}),e.length===0&&t.error(k(`There are no tables in schema '${a}.${o}'.`),{exitCode:0}),s=await d(`Choose the tables to pull`,{autocomplete:!0,choices:e,validators:[m.required()]})}for(let e of s){let t=O((await h.columns({catalog:a,schema:o,table:e})).map(e=>A({keys:[`name`,`type`,`extra`,`description`],values:e})));await je({definition:Ve({tableName:e,catalog:a,schema:o,parsedColumns:t}),outputDir:l.join(f,`schemas/generated`,a,o,e),skipRegistry:r,sourcePathOverride:u})}}),n}function Ue(){return new e(`init`).description(`Initialize a lakeql.config.json configuration file`).action(async()=>{let e=N(),t=l.join(e,P);if(r(t)&&await p(`lakeql.config.json already exists. Overwrite?`,{choices:[{label:`Yes`,value:`yes`},{label:`No`,value:`no`}]})===`no`){console.log(j.yellow(`Aborted.`));return}let n=r(l.join(e,`src`)),i;if(n)i=`src`,console.log(j.cyan(`Detected src/ directory — generated code will be placed in src/`));else{let e=await p(`Where should generated code be placed?`,{choices:[{label:`./ (project root)`,value:`.`},{label:`Custom path`,value:`__custom__`}]});i=e,e===`__custom__`&&(i=await f(`Enter the source path (relative to project root):`,{defaultValue:`.`}))}await c(t,`${JSON.stringify({sourcePath:i},null,2)}\n`),console.log(j.green(`Created ${P} at ${t}`))})}async function We(){let t=await n(),r=new e;r.configureHelp({sortSubcommands:!0}),r.name(`lakeql-cli`).description(`LakeQL CLI`).version(t.version).addCommand(Fe()).addCommand(Ie()).addCommand(Le()).addCommand(Pe()).addCommand(He()).addCommand(Me()).addCommand(de()).addCommand(Ue()),await r.parseAsync()}We().catch(()=>{process.exit(1)});export{};
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "@lakeql/cli",
3
- "version": "0.2.3",
3
+ "version": "0.3.0",
4
4
  "description": "CLI for introspecting Trino schemas and generating type-safe GraphQL endpoints",
5
5
  "keywords": [
6
6
  "cli",
@@ -56,11 +56,11 @@
56
56
  "read-pkg": "10.1.0",
57
57
  "zod": "4.4.3",
58
58
  "@lakeql/column-parser": "0.1.2",
59
- "@lakeql/file-generator": "0.1.3",
60
- "@lakeql/helpers": "0.1.2",
59
+ "@lakeql/file-generator": "0.1.5",
61
60
  "@lakeql/logger": "0.1.2",
62
61
  "@lakeql/response-transformer": "0.1.2",
63
- "@lakeql/schema-generator": "0.2.0",
62
+ "@lakeql/helpers": "0.1.2",
63
+ "@lakeql/schema-generator": "0.3.0",
64
64
  "@lakeql/trino-client": "0.3.0"
65
65
  },
66
66
  "devDependencies": {