|
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