codemod

var CodemodCmd = &cobra.Command{
	Use:   "codemod",
	Short: "Helpers for codemods and project automation",
	Long:  `Run codemods or setup helpers for various project types.`,
}

var LintSetupCmd = &cobra.Command{
	Use:   "lint-setup",
	Short: "Setup linting and formatting for a Node.js project",
	Long:  `Install and configure linting, formatting, and pre-commit hooks for a Node.js project (eslint, prettier, husky, etc).`,
	Run: func(cmd *cobra.Command, args []string) {
		if _, err := os.Stat("package.json"); errors.Is(err, os.ErrNotExist) {
			log.Error("package.json not found in current directory")
			return
		}

		log.Info("Installing lint/format dependencies via npm...")
		installArgs := []string{"install", "--save-dev",
			"eslint", "@eslint/js",
			"@typescript-eslint/eslint-plugin", "@typescript-eslint/parser",
			"eslint-config-prettier", "eslint-plugin-prettier",
			"@eng618/prettier-config", "globals",
			"echo-eslint-config", "husky", "lint-staged", "prettier",
		}
		installCmd := execCommand("npm", installArgs...)
		installCmd.Stdout = log.Writer()

		stderrPipe, err := installCmd.StderrPipe()
		if err != nil {
			log.Error("Failed to get stderr pipe: %v", err)
			return
		}
		if err := installCmd.Start(); err != nil {
			log.Error("npm install failed to start: %v", err)
			return
		}

		stderrBytes, _ := io.ReadAll(stderrPipe)
		err = installCmd.Wait()
		if err != nil {
			stderrStr := string(stderrBytes)
			if strings.Contains(stderrStr, "--legacy-peer-deps") || strings.Contains(stderrStr, "could not resolve dependency") {
				log.Info("npm install failed due to peer deps, retrying with --legacy-peer-deps...")
				installArgs = append(installArgs, "--legacy-peer-deps")
				installCmd2 := execCommand("npm", installArgs...)
				installCmd2.Stdout = log.Writer()
				installCmd2.Stderr = log.ErrorWriter()
				if err2 := installCmd2.Run(); err2 != nil {
					log.Error("npm install with --legacy-peer-deps failed: %v", err2)
					return
				}
			} else {
				log.Error("npm install failed: %v", err)
				return
			}
		}

		log.Info("Writing eslint.config.mjs...")
		eslintConfig := `import echoConfig, { echoGlobalsOverride, echoJestGlobalsOverride } from 'echo-eslint-config';

export default [
  ...echoConfig,
  echoGlobalsOverride,
  echoJestGlobalsOverride,
];
`
		if err := os.WriteFile("eslint.config.mjs", []byte(eslintConfig), 0644); err != nil {
			log.Error("Failed to write eslint.config.mjs: %v", err)
			return
		}

		log.Info("Adding scripts and config to package.json...")
		pkgData, err := os.ReadFile("package.json")
		if err != nil {
			log.Error("Failed to read package.json: %v", err)
			return
		}
		var pkg map[string]interface{}
		if err := json.Unmarshal(pkgData, &pkg); err != nil {
			log.Error("Failed to parse package.json: %v", err)
			return
		}

		scripts, ok := pkg["scripts"].(map[string]interface{})
		if !ok {
			scripts = make(map[string]interface{})
		}
		scripts["format"] = "prettier --write ."
		scripts["format:ci"] = "prettier --check ."
		scripts["lint"] = "eslint . --ext .js,.jsx,.ts,.tsx,.mjs,.cjs"
		scripts["lint:fix"] = "eslint . --fix --ext .js,.jsx,.ts,.tsx,.mjs,.cjs"
		scripts["prepare"] = "husky || echo 'Husky not installed, probably in ci'"
		pkg["scripts"] = scripts

		pkg["lint-staged"] = map[string]interface{}{
			"*.{js,ts,md,jsx,tsx,mdx}": []string{"prettier --write", "eslint --cache --fix"},
		}

		pkg["prettier"] = "@eng618/prettier-config"

		newPkgData, err := json.MarshalIndent(pkg, "", "  ")
		if err != nil {
			log.Error("Failed to marshal package.json: %v", err)
			return
		}
		if err := os.WriteFile("package.json", newPkgData, 0644); err != nil {
			log.Error("Failed to write package.json: %v", err)
			return
		}

		log.Info("Setting up Husky and pre-commit hook...")
		huskyInstall := execCommand("npx", "husky", "install")
		huskyInstall.Stdout = log.Writer()
		huskyInstall.Stderr = log.ErrorWriter()
		if err := huskyInstall.Run(); err != nil {
			log.Error("Failed to run 'npx husky install': %v", err)
			return
		}

		huskyAdd := execCommand("npx", "husky", "add", ".husky/pre-commit", "npx lint-staged")
		huskyAdd.Stdout = log.Writer()
		huskyAdd.Stderr = log.ErrorWriter()
		if err := huskyAdd.Run(); err != nil {
			log.Error("Failed to add pre-commit hook: %v", err)
			return
		}

		log.Success("Linting, formatting, and pre-commit hooks are set up!")
	},
}