feat: support wdio/wdi5 in typescript

src/index.ts

@@ -1,6 +1,6 @@
 import { gray, greenBright } from "colorette"
 import { execSync } from "child_process"
-import { copyFile } from "fs/promises"
+import { copyFile, cp } from "fs/promises"
 
 const DEV_DEPS = [
     "@wdio/cli",
@@ -12,7 +12,13 @@ const DEV_DEPS = [
     "chromedriver"
 ]
 
+const DEV_DEPS_TS = [...DEV_DEPS, "ts-node", "typescript"]
+
 export async function run() {
+    process.argv.find((arg) => arg.includes("ts")) ? await initTS() : await initJS()
+}
+
+async function initJS() {
     console.log(gray("≡> copying wdio.conf.js into place..."))
     await copyFile(`${__dirname}/../templates/wdio.conf.js`, `${process.cwd()}/wdio.conf.js`)
     console.log(greenBright("👍 done!"))
@@ -21,7 +27,21 @@ export async function run() {
     execSync(`npm i ${DEV_DEPS.join(" ")} --save-dev`, { stdio: "inherit" })
     console.log(greenBright("👍 done!"))
 
-    console.log(gray("≡> adding wdi5 start command to package.json..."))
+    console.log(gray('≡> adding wdi5 start command ("wdi5") to package.json...'))
     execSync(`npm set-script wdi5 "wdio run wdio.conf.js"`, { stdio: "inherit" })
     console.log(greenBright("👍 done!"))
 }
+
+async function initTS() {
+    console.log(gray('≡> copying tsconfig.json + wdio.conf.ts into "./test/"...'))
+    await cp(`${__dirname}/../templates/test`, `${process.cwd()}/test`, { recursive: true })
+    console.log(greenBright("👍 done!"))
+
+    console.log(gray("≡> installing wdio + wdi5 and adding them as dev dependencies..."))
+    execSync(`npm i ${DEV_DEPS_TS.join(" ")} --save-dev`, { stdio: "inherit" })
+    console.log(greenBright("👍 done!"))
+
+    console.log(gray("≡> adding wdi5 start command to package.json..."))
+    execSync(`npm set-script wdi5 "wdio run test/wdio.conf.ts"`, { stdio: "inherit" })
+    console.log(greenBright("👍 done!"))
+}