|
1 | | -import { generateJuliaStruct, generatePythonStruct, generateRustStruct, generateTypeScriptType } from './languages'; |
2 | | -import { Token, tokenize } from './tokenizer'; |
| 1 | +#!/usr/bin/env node |
3 | 2 |
|
4 | | -export * from './languages'; |
5 | | -export * from './tokenizer'; |
| 3 | +import fs from 'fs/promises'; |
6 | 4 |
|
7 | | -export type SupportedLanguage = 'typescript' | 'python' | 'julia' | 'rust'; |
| 5 | +import { Command, Option } from '@commander-js/extra-typings'; |
8 | 6 |
|
9 | | -export function convertToLanguage(language: SupportedLanguage, token: Token) { |
10 | | - switch (language) { |
11 | | - case 'typescript': |
12 | | - return generateTypeScriptType(token); |
| 7 | +import { convertToLanguage, SupportedLanguage } from './lib'; |
| 8 | +import { tokenize } from './tokenizer'; |
13 | 9 |
|
14 | | - case 'python': |
15 | | - return generatePythonStruct(token); |
| 10 | +export * from './lib'; |
16 | 11 |
|
17 | | - case 'julia': |
18 | | - return generateJuliaStruct(token); |
| 12 | +const program = new Command(); |
19 | 13 |
|
20 | | - case 'rust': |
21 | | - return generateRustStruct(token); |
| 14 | +program |
| 15 | + .name('json2struct') |
| 16 | + .description('Easily translate JSON into type definitions') |
| 17 | + .version('0.4.1') |
| 18 | + .configureOutput({ |
| 19 | + writeOut: (str) => process.stdout.write(`[OUT] ${str}`), |
| 20 | + writeErr: (str) => process.stdout.write(`[ERR] ${str}`), |
| 21 | + outputError: (str, write) => write(`\x1b[31m${str}\x1b[0m`), |
| 22 | + }); |
22 | 23 |
|
23 | | - default: |
24 | | - throw new Error(`${language} is not supported`); |
25 | | - } |
26 | | -} |
| 24 | +program |
| 25 | + .command('convert <input>', { isDefault: true }) |
| 26 | + .description('Convert JSON file to type file') |
| 27 | + .option('-o --output <output-file>') |
| 28 | + .option('--overwrite') |
| 29 | + .addOption( |
| 30 | + new Option('-l --language <output-language>') |
| 31 | + .choices<SupportedLanguage[]>(['typescript', 'python', 'julia', 'rust']) |
| 32 | + .default('typescript') |
| 33 | + ) |
| 34 | + .action(async (inputPath, args) => { |
| 35 | + console.info(`\u001b[32mjson2struct: Converting ${inputPath} to ${args.language}:\u001b[0m`); |
27 | 36 |
|
28 | | -/** |
29 | | - * |
30 | | - * @param language the language to translate to |
31 | | - * @param json unparsed json string |
32 | | - */ |
33 | | -export default function json2struct(language: string, json: string) { |
34 | | - const parsedJson = JSON.parse(json); |
| 37 | + if (!args?.output?.length && args?.overwrite) { |
| 38 | + program.error('--overwrite options requires an output path'); |
| 39 | + return; |
| 40 | + } |
35 | 41 |
|
36 | | - const tokens = tokenize(parsedJson); |
| 42 | + const fileContent = await fs.readFile(inputPath); |
37 | 43 |
|
38 | | - return convertToLanguage(language as SupportedLanguage, tokens); |
39 | | -} |
| 44 | + const json = JSON.parse(fileContent.toString()); |
| 45 | + |
| 46 | + const tokens = tokenize(json); |
| 47 | + |
| 48 | + const generatedStruct = convertToLanguage((args?.language as SupportedLanguage) ?? 'typescript', tokens); |
| 49 | + |
| 50 | + if (args.output?.length) { |
| 51 | + if (args?.overwrite) { |
| 52 | + await fs.writeFile(args.output, generatedStruct); |
| 53 | + } else { |
| 54 | + await fs.appendFile(args.output, generatedStruct); |
| 55 | + } |
| 56 | + } |
| 57 | + |
| 58 | + console.info(generatedStruct); |
| 59 | + }); |
| 60 | + |
| 61 | +program.addHelpCommand(); |
| 62 | + |
| 63 | +program.parse(); |
0 commit comments