feat(cli): Improve bunfig.toml handling and test (#13)

Author: ShiboSoftwareDev

cli.ts

@@ -1,6 +1,6 @@
 #!/usr/bin/env bun
 
-import { mkdir, writeFile, access } from "node:fs/promises"
+import { mkdir, writeFile, access, readFile, appendFile } from "node:fs/promises"
 import { spawn } from "node:child_process"
 
 const EXAMPLE_TEST = `import { expect, test } from "bun:test"                                                             
@@ -21,7 +21,7 @@ const PRELOAD_FILE = `import "bun-match-svg"`
 const BUNFIG = `[test]
 preload = ["./tests/fixtures/preload.ts"]`
 
-async function installDependency() {
+export async function installDependency() {
   return new Promise((resolve, reject) => {
     const install = spawn('bun', ['add', '-d', 'bun-match-svg'], {
       stdio: 'inherit'
@@ -37,7 +37,7 @@ async function installDependency() {
   })
 }
 
-async function fileExists(path: string): Promise<boolean> {
+export async function fileExists(path: string): Promise<boolean> {
   try {
     await access(path)
     return true
@@ -46,7 +46,7 @@ async function fileExists(path: string): Promise<boolean> {
   }
 }
 
-async function init() {
+export async function init() {
   try {
     console.log("📦 Installing bun-match-svg...")
     await installDependency()
@@ -71,7 +71,20 @@ async function init() {
       await writeFile("bunfig.toml", BUNFIG)
       console.log("✅ Created bunfig.toml")
     } else {
-      console.log("🔒 bunfig.toml already exists, skipping.")
+      const bunfigContent = await readFile("bunfig.toml", "utf-8")
+      if (bunfigContent.includes('preload = ["./tests/fixtures/preload.ts"]')) {
+        console.log("🔒 bunfig.toml already has preload configuration, skipping.")
+      } else if (bunfigContent.match(/^\s*\[test\]\s*$/m)) {
+        const updatedContent = bunfigContent.replace(
+          /(^\s*\[test\]\s*$)/m,
+          `$1\npreload = ["./tests/fixtures/preload.ts"]`,
+        )
+        await writeFile("bunfig.toml", updatedContent)
+        console.log("✅ Updated bunfig.toml with preload configuration.")
+      } else {
+        await appendFile("bunfig.toml", `\n\n${BUNFIG}`)
+        console.log("✅ Added preload configuration to bunfig.toml.")
+      }
     }
 
     console.log("\n🎉 You can now run: bun test")

tests/cli.test.ts

@@ -0,0 +1,122 @@
+import {
+  afterAll,
+  afterEach,
+  beforeAll,
+  beforeEach,
+  describe,
+  expect,
+  spyOn,
+  test,
+  type Mock,
+} from "bun:test"
+import * as fs from "node:fs/promises"
+import * as path from "node:path"
+import * as cli from "../cli"
+
+const tempDir = path.join(__dirname, "cli-test-temp")
+const originalCwd = process.cwd()
+
+describe("cli init", () => {
+  let installSpy: Mock<typeof cli.installDependency>
+  let consoleSpy: Mock<typeof console.log>
+
+  beforeAll(async () => {
+    await fs.mkdir(tempDir, { recursive: true })
+  })
+
+  afterAll(async () => {
+    await fs.rm(tempDir, { recursive: true, force: true })
+  })
+
+  beforeEach(async () => {
+    process.chdir(tempDir)
+    installSpy = spyOn(cli, "installDependency").mockImplementation(async () => {})
+    consoleSpy = spyOn(console, "log").mockImplementation(() => {})
+  })
+
+  afterEach(async () => {
+    process.chdir(originalCwd)
+    const files = await fs.readdir(tempDir)
+    for (const file of files) {
+      await fs.rm(path.join(tempDir, file), { recursive: true, force: true })
+    }
+    installSpy.mockRestore()
+    consoleSpy.mockRestore()
+  })
+
+  test("should create files and config on a fresh project", async () => {
+    await cli.init()
+
+    expect(installSpy).toHaveBeenCalledTimes(1)
+
+    // Check files
+    const testFileExists = await cli.fileExists("tests/svg.test.ts")
+    expect(testFileExists).toBe(true)
+    const preloadFileExists = await cli.fileExists("tests/fixtures/preload.ts")
+    expect(preloadFileExists).toBe(true)
+    const bunfigExists = await cli.fileExists("bunfig.toml")
+    expect(bunfigExists).toBe(true)
+
+    // Check bunfig.toml content
+    const bunfigContent = await fs.readFile("bunfig.toml", "utf-8")
+    expect(bunfigContent).toContain(`[test]`)
+    expect(bunfigContent).toContain(`preload = ["./tests/fixtures/preload.ts"]`)
+
+    // Check console logs
+    const logs = consoleSpy.mock.calls.flat().join("\n")
+    expect(logs).toContain("✅ Created example test in tests/svg.test.ts")
+    expect(logs).toContain("✅ Created preload file in tests/fixtures/preload.ts")
+    expect(logs).toContain("✅ Created bunfig.toml")
+  })
+
+  test("should skip creating files that already exist", async () => {
+    // Pre-create files
+    await fs.mkdir("tests/fixtures", { recursive: true })
+    await fs.writeFile("tests/svg.test.ts", "test")
+    await fs.writeFile("tests/fixtures/preload.ts", "preload")
+    await fs.writeFile(
+      "bunfig.toml",
+      `[test]\npreload = ["./tests/fixtures/preload.ts"]`,
+    )
+
+    await cli.init()
+
+    const logs = consoleSpy.mock.calls.flat().join("\n")
+    expect(logs).toContain("🔒 tests/svg.test.ts already exists, skipping.")
+    expect(logs).toContain(
+      "🔒 tests/fixtures/preload.ts already exists, skipping.",
+    )
+    expect(logs).toContain(
+      "🔒 bunfig.toml already has preload configuration, skipping.",
+    )
+  })
+
+  test("should update bunfig.toml if [test] section exists", async () => {
+    await fs.writeFile("bunfig.toml", "[test]")
+
+    await cli.init()
+
+    const bunfigContent = await fs.readFile("bunfig.toml", "utf-8")
+    expect(bunfigContent).toBe(
+      `[test]\npreload = ["./tests/fixtures/preload.ts"]`,
+    )
+
+    const logs = consoleSpy.mock.calls.flat().join("\n")
+    expect(logs).toContain("✅ Updated bunfig.toml with preload configuration.")
+  })
+
+  test("should append to bunfig.toml if it exists but is unrelated", async () => {
+    await fs.writeFile("bunfig.toml", "[some-other-section]")
+
+    await cli.init()
+
+    const bunfigContent = await fs.readFile("bunfig.toml", "utf-8")
+    expect(bunfigContent).toContain("[some-other-section]")
+    expect(bunfigContent).toContain(
+      `\n\n[test]\npreload = ["./tests/fixtures/preload.ts"]`,
+    )
+
+    const logs = consoleSpy.mock.calls.flat().join("\n")
+    expect(logs).toContain("✅ Added preload configuration to bunfig.toml.")
+  })
+})