import chalk from "chalk";
import fs from "fs-extra";
import path from "path";
import { readTsConfigFile } from "../utils/read-config-file.js";

export type Framework = "nextjs" | "nestjs" | "express" | "fastify";

export function detectFramework(projectRoot: string): Framework {
  const packageJson = JSON.parse(
    fs.readFileSync(path.join(projectRoot, "package.json"), "utf-8")
  );

  // Check for NestJS
  if (
    packageJson.dependencies?.["@nestjs/core"] ||
    packageJson.devDependencies?.["@nestjs/core"]
  ) {
    return "nestjs";
  }

  // Check for Next.js
  if (packageJson.dependencies?.next || packageJson.devDependencies?.next) {
    return "nextjs";
  }

  // Check for Fastify
  if (
    packageJson.dependencies?.fastify ||
    packageJson.devDependencies?.fastify
  ) {
    return "fastify";
  }

  // Default to express
  return "express";
}

export function detectTypeScript(projectRoot: string): boolean {
  // check if project has been initialized with typescript or default to javascript
  const packageJson = JSON.parse(
    fs.readFileSync(path.join(projectRoot, "package.json"), "utf-8")
  );

  if (
    !packageJson.devDependencies?.typescript &&
    !packageJson.dependencies?.typescript
  ) {
    console.log(
      chalk.yellow(
        "Please initialize xmcp within a TypeScript project. Missing dependency: typescript"
      )
    );
    return false;
  }

  // if typescript is in the dependencies, check for tsconfig.json and validate it
  if (fs.existsSync(path.join(projectRoot, "tsconfig.json"))) {
    try {
      const tsconfigPath = path.join(projectRoot, "tsconfig.json");
      const tsconfigContent = readTsConfigFile(tsconfigPath);
      return typeof tsconfigContent === "object" && tsconfigContent !== null;
    } catch (error) {
      console.log(
        chalk.yellow(
          `Please initialize xmcp within a TypeScript project. ${error instanceof Error ? error.message : "Invalid tsconfig.json file"}`
        )
      );
      return false;
    }
  }

  console.log(
    chalk.yellow(
      "Please initialize xmcp within a TypeScript project. Missing file: tsconfig.json"
    )
  );

  return false;
}