@lakeql/cli 0.2.1 → 0.2.3
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.
- package/dist/cli.mjs +5 -5
- package/package.json +1 -1
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
|
|
3
|
-
`)}function
|
|
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
|
|
4
4
|
`,`
|
|
5
|
-
`)}\n`}async function
|
|
6
|
-
`);console.error(`Error: Validation failed for ${a}:\n${e}`),process.exit(1)}let d=u.data,f=
|
|
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
7
|
`);console.error(`Error: Validation failed for ${a}:\n${e}`),process.exit(1)}console.log(`
|
|
8
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(
|
|
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{};
|