feat(ct): generate playwright/index.[tj]sx (#75)

Author: sand4rt

src/generator.ts

@@ -19,7 +19,7 @@ import fs from 'fs';
 import { prompt } from 'enquirer';
 import colors from 'ansi-colors';
 
-import { executeCommands, createFiles, executeTemplate, Command, languageToFileExtension } from './utils';
+import { executeCommands, createFiles, executeTemplate, Command, languageToFileExtension, getFileExtensionCT } from './utils';
 import { packageManager } from './packageManager';
 
 export type PromptOptions = {
@@ -192,7 +192,7 @@ export class Generator {
         command: packageManager.installDevDependency(`${ctPackageName}${packageTag}`),
       });
 
-      const extension = languageToFileExtension(answers.language);
+      const extension = getFileExtensionCT(answers.language, answers.framework);
       const htmlTemplate = executeTemplate(this._readAsset(path.join('playwright', 'index.html')), { extension }, new Map());
       files.set('playwright/index.html', htmlTemplate);
 

src/utils.ts

@@ -20,6 +20,7 @@ import path from 'path';
 
 import { prompt } from 'enquirer';
 import colors from 'ansi-colors';
+import { PromptOptions } from './generator';
 
 export type Command = {
   command: string;
@@ -80,7 +81,18 @@ export function executeTemplate(input: string, args: Record<string, string>, sec
   return result.join('\n');
 }
 
-export function languageToFileExtension(language: 'JavaScript' | 'TypeScript'): 'js' | 'ts' {
+export function getFileExtensionCT(language: PromptOptions['language'], framework: PromptOptions['framework']) {
+  const isJsxFramework = framework === 'solid' || framework === 'react';
+  if (isJsxFramework && language === 'JavaScript')
+    return 'jsx';
+  else if (isJsxFramework && language === 'TypeScript')
+    return 'tsx';
+  else if (language === 'TypeScript')
+    return 'ts';
+  return 'js';
+}
+
+export function languageToFileExtension(language: PromptOptions['language']): 'js' | 'ts' {
   return language === 'JavaScript' ? 'js' : 'ts';
 }