@tyyyho/treg 0.1.2

package/scripts/init-project/mrm-rules/test-jest.mjs

@@ -0,0 +1,75 @@
+import { packageJson } from "../mrm-core.mjs"
+import { installPackages, withProjectCwd, writeFile } from "./shared.mjs"
+
+function getJestConfig(testEnvironment) {
+  return `/** @type {import("jest").Config} */
+const config = {
+  testEnvironment: "${testEnvironment}",
+  setupFilesAfterEnv: ["<rootDir>/jest.setup.js"],
+  testMatch: ["**/*.test.[jt]s?(x)", "**/*.test.mjs"],
+}
+
+module.exports = config
+`
+}
+
+function getSetupFile(frameworkId) {
+  if (frameworkId === "react") {
+    return `require("@testing-library/jest-dom")
+
+// Jest setup (add custom matchers or globals here)
+`
+  }
+  return `// Jest setup (add custom matchers or globals here)
+`
+}
+
+export async function runTestJestRule(context) {
+  const { framework, projectDir, pm, force, dryRun } = context
+  const deps = ["jest"]
+  if (framework.testEnvironment === "jsdom") {
+    deps.push("jest-environment-jsdom")
+  }
+  if (framework.id === "react") {
+    deps.push("@testing-library/jest-dom")
+  }
+  installPackages(projectDir, pm, deps, true, dryRun)
+
+  await writeFile(
+    projectDir,
+    "jest.config.js",
+    getJestConfig(framework.testEnvironment),
+    force,
+    dryRun
+  )
+  await writeFile(
+    projectDir,
+    "jest.setup.js",
+    getSetupFile(framework.id),
+    force,
+    dryRun
+  )
+
+  withProjectCwd(projectDir, () => {
+    if (dryRun) {
+      console.log(
+        "[dry-run] Would set package scripts: test, test:watch, test:coverage"
+      )
+      return
+    }
+    packageJson()
+      .setScript(
+        "test",
+        "NODE_OPTIONS=--experimental-vm-modules jest --passWithNoTests"
+      )
+      .setScript(
+        "test:watch",
+        "NODE_OPTIONS=--experimental-vm-modules jest --watch"
+      )
+      .setScript(
+        "test:coverage",
+        "NODE_OPTIONS=--experimental-vm-modules jest --coverage"
+      )
+      .save()
+  })
+}

package/scripts/init-project/mrm-rules/test-vitest.mjs

@@ -0,0 +1,64 @@
+import { packageJson } from "../mrm-core.mjs"
+import { installPackages, withProjectCwd, writeFile } from "./shared.mjs"
+
+function getVitestConfig(framework) {
+  return `import { defineConfig } from "vitest/config"
+
+export default defineConfig({
+  test: {
+    environment: "${framework.testEnvironment}",
+    setupFiles: ["./vitest.setup.js"],
+  },
+})
+`
+}
+
+function getVitestSetup(framework) {
+  if (framework.id === "react") {
+    return `import "@testing-library/jest-dom/vitest"
+`
+  }
+  return `// Vitest setup
+`
+}
+
+export async function runTestVitestRule(context) {
+  const { framework, projectDir, pm, force, dryRun } = context
+  const deps = ["vitest"]
+  if (framework.testEnvironment === "jsdom") {
+    deps.push("jsdom")
+  }
+  if (framework.id === "react") {
+    deps.push("@testing-library/jest-dom")
+  }
+  installPackages(projectDir, pm, deps, true, dryRun)
+
+  await writeFile(
+    projectDir,
+    "vitest.config.mjs",
+    getVitestConfig(framework),
+    force,
+    dryRun
+  )
+  await writeFile(
+    projectDir,
+    "vitest.setup.js",
+    getVitestSetup(framework),
+    force,
+    dryRun
+  )
+
+  withProjectCwd(projectDir, () => {
+    if (dryRun) {
+      console.log(
+        "[dry-run] Would set package scripts: test, test:watch, test:coverage"
+      )
+      return
+    }
+    packageJson()
+      .setScript("test", "vitest run")
+      .setScript("test:watch", "vitest")
+      .setScript("test:coverage", "vitest run --coverage")
+      .save()
+  })
+}

package/scripts/init-project/mrm-rules/typescript.mjs

@@ -0,0 +1,44 @@
+import { json, packageJson } from "../mrm-core.mjs"
+import { installPackages, withProjectCwd } from "./shared.mjs"
+
+const TS_REQUIRED_OPTIONS = {
+  noImplicitAny: true,
+  noImplicitThis: true,
+  exactOptionalPropertyTypes: true,
+  noUncheckedIndexedAccess: true,
+  noUnusedLocals: true,
+  noUnusedParameters: true,
+}
+
+export async function runTypescriptRule(context) {
+  const { framework, projectDir, pm, dryRun } = context
+  installPackages(projectDir, pm, ["typescript"], true, dryRun)
+
+  withProjectCwd(projectDir, () => {
+    if (dryRun) {
+      console.log("[dry-run] Would update tsconfig.json")
+      console.log("[dry-run] Would set package script: type-check")
+      return
+    }
+
+    const tsconfig = json("tsconfig.json", {
+      compilerOptions: {},
+      exclude: [],
+    })
+    const mergedCompilerOptions = {
+      ...(tsconfig.get("compilerOptions") ?? {}),
+      ...TS_REQUIRED_OPTIONS,
+    }
+    const exclude = new Set(tsconfig.get("exclude", []))
+    for (const entry of framework.tsRequiredExcludes) {
+      exclude.add(entry)
+    }
+
+    tsconfig
+      .set("compilerOptions", mergedCompilerOptions)
+      .set("exclude", Array.from(exclude))
+      .save()
+
+    packageJson().setScript("type-check", "tsc --noEmit").save()
+  })
+}

package/scripts/init-project/package-manager.mjs

@@ -0,0 +1,68 @@
+import { execSync } from "node:child_process"
+import { existsSync } from "node:fs"
+import path from "node:path"
+
+export function detectPackageManager(projectDir) {
+  if (existsSync(path.join(projectDir, "pnpm-lock.yaml"))) {
+    return "pnpm"
+  }
+  if (existsSync(path.join(projectDir, "yarn.lock"))) {
+    return "yarn"
+  }
+  if (existsSync(path.join(projectDir, "package-lock.json"))) {
+    return "npm"
+  }
+  return "npm"
+}
+
+export function getRunCommand(pm) {
+  if (pm === "pnpm") return "pnpm"
+  if (pm === "yarn") return "yarn"
+  return "npm run"
+}
+
+export function runCommand(command, cwd, dryRun = false) {
+  if (dryRun) {
+    console.log(`[dry-run] Would run: ${command}`)
+    return
+  }
+  execSync(command, { cwd, stdio: "inherit" })
+}
+
+export function runScript(pm, scriptName, cwd, dryRun = false) {
+  if (pm === "pnpm") {
+    runCommand(`pnpm ${scriptName}`, cwd, dryRun)
+    return
+  }
+  if (pm === "yarn") {
+    runCommand(`yarn ${scriptName}`, cwd, dryRun)
+    return
+  }
+  runCommand(`npm run ${scriptName}`, cwd, dryRun)
+}
+
+export function installPackages(
+  pm,
+  projectDir,
+  packages,
+  isDev,
+  dryRun = false
+) {
+  if (packages.length === 0) return
+
+  const list = packages.join(" ")
+  let command = ""
+
+  if (pm === "pnpm") {
+    command = `pnpm add ${isDev ? "-D " : ""}${list}`
+  } else if (pm === "yarn") {
+    command = `yarn add ${isDev ? "-D " : ""}${list}`
+  } else {
+    command = `npm install ${isDev ? "-D " : ""}${list}`
+  }
+
+  console.log(
+    `${dryRun ? "[dry-run] " : ""}Installing ${isDev ? "dev " : ""}dependencies: ${packages.join(", ")}`
+  )
+  runCommand(command, projectDir, dryRun)
+}

package/scripts/init-project/package-manager.test.mjs

@@ -0,0 +1,21 @@
+import { mkdtemp, writeFile } from "node:fs/promises"
+import os from "node:os"
+import path from "node:path"
+import { describe, expect, it } from "@jest/globals"
+import { detectPackageManager } from "./package-manager.mjs"
+
+describe("detectPackageManager", () => {
+  it("defaults to npm when no lockfile exists", async () => {
+    const baseDir = await mkdtemp(path.join(os.tmpdir(), "treg-pm-"))
+    expect(detectPackageManager(baseDir)).toBe("npm")
+  })
+
+  it("detects pnpm from lockfile", async () => {
+    const baseDir = await mkdtemp(path.join(os.tmpdir(), "treg-pm-"))
+    await writeFile(
+      path.join(baseDir, "pnpm-lock.yaml"),
+      "lockfileVersion: 9\n"
+    )
+    expect(detectPackageManager(baseDir)).toBe("pnpm")
+  })
+})

package/scripts/init-project/utils.mjs

@@ -0,0 +1,12 @@
+export function hasPackage(pkg, name) {
+  return Boolean(
+    pkg.dependencies?.[name] ||
+    pkg.devDependencies?.[name] ||
+    pkg.peerDependencies?.[name]
+  )
+}
+
+export function formatStep(step, total, message, dryRun) {
+  const suffix = dryRun ? " [dry-run]" : ""
+  return `[${step}/${total}] ${message}${suffix}`
+}

package/scripts/init-project/utils.test.mjs

@@ -0,0 +1,22 @@
+import { describe, expect, it } from "@jest/globals"
+import { formatStep, hasPackage } from "./utils.mjs"
+
+describe("hasPackage", () => {
+  it("checks dependencies and devDependencies", () => {
+    const pkg = {
+      dependencies: { react: "^19.0.0" },
+      devDependencies: { eslint: "^9.0.0" },
+    }
+    expect(hasPackage(pkg, "react")).toBe(true)
+    expect(hasPackage(pkg, "eslint")).toBe(true)
+    expect(hasPackage(pkg, "typescript")).toBe(false)
+  })
+})
+
+describe("formatStep", () => {
+  it("appends dry-run suffix when needed", () => {
+    expect(formatStep(2, 3, "Run mrm rules", true)).toBe(
+      "[2/3] Run mrm rules [dry-run]"
+    )
+  })
+})

package/scripts/init-project.mjs

@@ -0,0 +1,7 @@
+#!/usr/bin/env node
+import { main } from "./init-project/index.mjs"
+
+main().catch(error => {
+  console.error(error)
+  process.exitCode = 1
+})