refactor: pass a short lang to Generator.for()

Relate to #31
php-coder · Mar 9, 2024 · b090ac2 · b090ac2
1 parent 4c4bcf2
commit b090ac2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/cli.js b/src/cli.js
@@ -353,7 +353,8 @@ const main = async (argv) => {
         fs.mkdirSync(destDir, {recursive: true})
     }
 
-    const generator = Generator.for(argv.lang)
+    const lang = lang2extension(argv.lang)
+    const generator = Generator.for(lang)
 
     await createApp(destDir, argv)
     await createDb(destDir, argv)

diff --git a/src/generator/Generator.js b/src/generator/Generator.js
@@ -13,7 +13,7 @@ module.exports = class Generator {
                 return new TsGenerator()
             case 'go':
                 return new GoGenerator()
-            case 'python':
+            case 'py':
                 return new PyGenerator()
             default:
                 throw new Error(`Unsupported language: ${lang}`)