setup.ts

import * as cryptolib from "crypto";
import dotenv from "dotenv";
import fs from "fs";
import inquirer from "inquirer";
import path from "path";
import type { ExecException } from "child_process";
import { exec, spawn, execSync } from "child_process";
import { MongoClient } from "mongodb";
import { MAXIMUM_IMAGE_SIZE_LIMIT_KB } from "@constants";
import {
  askForMongoDBUrl,
  checkConnection,
  checkExistingMongoDB,
} from "@setup/MongoDB";
import { askToKeepValues } from "@setup/askToKeepValues";
import { getNodeEnvironment } from "@setup/getNodeEnvironment";
import { isValidEmail } from "@setup/isValidEmail";
import { validateRecaptcha } from "@setup/reCaptcha";
import {
  askForRedisUrl,
  checkExistingRedis,
  checkRedisConnection,
} from "@setup/redisConfiguration";
import {
  setImageUploadSize,
  validateImageFileSize,
} from "@setup/setImageUploadSize";
import { askForSuperAdminEmail } from "@setup/superAdmin";
import { updateEnvVariable } from "@setup/updateEnvVariable";
import { verifySmtpConnection } from "@setup/verifySmtpConnection";
import { loadDefaultOrganiation } from "@utilities/loadDefaultOrg";
import { isMinioInstalled } from "@setup/isMinioInstalled";
import { installMinio } from "@setup/installMinio";

dotenv.config();

// Check if all the fields in .env.sample are present in .env
/**
 * The function `checkEnvFile` checks if any fields are missing in the .env file compared to the .env.sample file, and
 * if so, it copies the missing fields from .env.sampale to .env.
 */
export function checkEnvFile(): void {
  if (process.env.NODE_ENV === "test") {
    const env = dotenv.parse(fs.readFileSync(".env_test"));
    const envSample = dotenv.parse(fs.readFileSync(".env.sample"));
    const misplaced = Object.keys(envSample).filter((key) => !(key in env));
    if (misplaced.length > 0) {
      // copy the missing fields from .env.sample to .env
      for (const key of misplaced) {
        fs.appendFileSync(".env_test", `${key}=${envSample[key]}\n`);
      }
    }
  } else {
    const env = dotenv.parse(fs.readFileSync(".env"));
    const envSample = dotenv.parse(fs.readFileSync(".env.sample"));
    const misplaced = Object.keys(envSample).filter((key) => !(key in env));
    if (misplaced.length > 0) {
      // copy the missing fields from .env.sample to .env
      for (const key of misplaced) {
        fs.appendFileSync(".env", `${key}=${envSample[key]}\n`);
      }
    }
  }
}

/**
 * The function `setNodeEnvironment` sets the Node environment by reading the value from a file, updating the process
 * environment variable, and updating a configuration file.
 */
export async function setNodeEnvironment(): Promise<void> {
  if (process.env.NODE_ENV === "test") {
    try {
      const nodeEnv = await getNodeEnvironment();
      const config = dotenv.parse(fs.readFileSync(".env_test"));
      config.NODE_ENV = nodeEnv;
      updateEnvVariable(config);
    } catch (err) {
      console.error(err);
      abort();
    }
  } else {
    try {
      const nodeEnv = await getNodeEnvironment();
      process.env.NODE_ENV = nodeEnv;

      const config = dotenv.parse(fs.readFileSync(".env"));
      config.NODE_ENV = nodeEnv;
      updateEnvVariable(config);
    } catch (err) {
      console.error(err);
      abort();
    }
  }
}

// Generate and update the access and refresh token secrets in .env
/**
 * The function `accessAndRefreshTokens` generates and updates access and refresh tokens if they are
 * null.
 * @param accessTokenSecret - A string representing the access token secret. It is
 * initially set to `null` and will be generated if it is `null`.
 * @param refreshTokenSecret - The `refreshTokenSecret` parameter is a string that
 * represents the secret key used to generate and verify refresh tokens. Refresh tokens are typically
 * used in authentication systems to obtain new access tokens without requiring the user to
 * re-authenticate.
 */
export async function accessAndRefreshTokens(
  accessTokenSecret: string | null,
  refreshTokenSecret: string | null,
): Promise<void> {
  if (process.env.NODE_ENV === "test") {
    const config = dotenv.parse(fs.readFileSync(".env_test"));

    if (accessTokenSecret === null) {
      accessTokenSecret = cryptolib.randomBytes(32).toString("hex");
      config.ACCESS_TOKEN_SECRET = accessTokenSecret;
      updateEnvVariable(config);
    }

    if (refreshTokenSecret === null) {
      refreshTokenSecret = cryptolib.randomBytes(32).toString("hex");
      config.REFRESH_TOKEN_SECRET = refreshTokenSecret;
      updateEnvVariable(config);
    }
  } else {
    const config = dotenv.parse(fs.readFileSync(".env"));

    if (accessTokenSecret === null) {
      accessTokenSecret = cryptolib.randomBytes(32).toString("hex");
      config.ACCESS_TOKEN_SECRET = accessTokenSecret;
      updateEnvVariable(config);
    }

    if (refreshTokenSecret === null) {
      refreshTokenSecret = cryptolib.randomBytes(32).toString("hex");
      config.REFRESH_TOKEN_SECRET = refreshTokenSecret;
      updateEnvVariable(config);
    }
  }
}

function transactionLogPath(logPath: string | null): void {
  const config = dotenv.parse(fs.readFileSync(".env"));
  config.LOG = "true";
  if (!logPath) {
    // Check if the logs/transaction.log file exists, if not, create it
    const defaultLogPath = path.resolve(__dirname, "logs");
    const defaultLogFile = path.join(defaultLogPath, "transaction.log");
    if (!fs.existsSync(defaultLogPath)) {
      console.log("Creating logs/transaction.log file...");
      fs.mkdirSync(defaultLogPath);
    }

    config.LOG_PATH = defaultLogFile;
  } else {
    // Remove the logs files, if exists
    const logsDirPath = path.resolve(__dirname, "logs");
    if (fs.existsSync(logsDirPath)) {
      fs.readdirSync(logsDirPath).forEach((file: string) => {
        if (file !== "README.md") {
          const curPath = path.join(logsDirPath, file);
          fs.unlinkSync(curPath);
        }
      });
    }
    config.LOG_PATH = logPath;
  }
}

async function askForTransactionLogPath(): Promise<string> {
  let logPath: string | null = null;
  let isValidPath = false;

  while (!isValidPath) {
    const response = await inquirer.prompt([
      {
        type: "input",
        name: "logPath",
        message: "Enter absolute path of log file:",
        default: null,
      },
    ]);
    logPath = response.logPath;

    if (logPath && fs.existsSync(logPath)) {
      try {
        fs.accessSync(logPath, fs.constants.R_OK | fs.constants.W_OK);
        isValidPath = true;
      } catch {
        console.error(
          "The file is not readable/writable. Please enter a valid file path.",
        );
      }
    } else {
      console.error(
        "Invalid path or file does not exist. Please enter a valid file path.",
      );
    }
  }

  return logPath as string;
}

//Wipes the existing data in the database
/**
 * The function `wipeExistingData` deletes all existing data in a MongoDB database if the database URL is provided.
 * @param url - A string representing the URL of the MongoDB database.
 * @returns The function returns a Promise that resolves to `void`.
 */
export async function wipeExistingData(url: string): Promise<void> {
  const client = new MongoClient(`${url}`);
  try {
    await client.connect();
    const db = client.db();
    const collections = await db.listCollections().toArray();

    if (collections.length > 0) {
      for (const collection of collections) {
        await db.collection(collection.name).deleteMany({});
      }
      console.log("All existing data has been deleted.");
    }
  } catch {
    console.error("Could not connect to database to check for data");
  }
  client.close();
  // return shouldImport;
}

//Check if the database is empty
/**
 * The function `checkDb` checks if a MongoDB database is empty by connecting to the database and checking if any
 * collections exist.
 * @param url - A string representing the URL of the MongoDB database.
 * @returns The function returns a Promise that resolves to a boolean value.
 */
export async function checkDb(url: string): Promise<boolean> {
  let dbEmpty = false;
  const client = new MongoClient(`${url}`);
  try {
    await client.connect();
    const db = client.db();
    const collections = await db.listCollections().toArray();
    if (collections.length > 0) {
      console.log("Existing data found in the database");
      dbEmpty = false;
    } else {
      dbEmpty = true;
    }
  } catch {
    console.error("Could not connect to database to check for data");
  }
  client.close();
  return dbEmpty;
}
//Import sample data
/**
 * The function `importData` imports sample data into a MongoDB database if the database URL is provided and if it
 * is determined that existing data should be wiped.
 * @returns The function returns a Promise that resolves to `void`.
 */
export async function importData(): Promise<void> {
  if (!process.env.MONGO_DB_URL) {
    console.log("Couldn't find mongodb url");
    return;
  }

  console.log("Importing sample data...");
  if (process.env.NODE_ENV !== "test") {
    await exec(
      "npm run import:sample-data",
      (error: ExecException | null, stdout: string, stderr: string) => {
        if (error) {
          console.error(`Error: ${error.message}`);
          abort();
        }
        if (stderr) {
          console.error(`Error: ${stderr}`);
          abort();
        }
        console.log(`Output: ${stdout}`);
      },
    );
  }
}

//Import Default data
/**
 * The function `importDefaultData` imports default data into a MongoDB database if the database URL is provided.
 * @returns The function returns a Promise that resolves to `void`.
 */
export async function importDefaultData(): Promise<void> {
  if (!process.env.MONGO_DB_URL) {
    console.log("Couldn't find mongodb url");
    return;
  }
  console.log("Importing default data...");
  if (process.env.NODE_ENV !== "test") {
    await loadDefaultOrganiation();
  }
}
// get the redis url
/**
 * The `redisConfiguration` function updates the Redis configuration by prompting the user for the
 * Redis URL, checking the connection, and updating the environment variables and .env file
 * accordingly.
 */
export async function redisConfiguration(): Promise<void> {
  try {
    let host!: string;
    let port!: number;
    let password!: string;
    let url = await checkExistingRedis();
    let isConnected = url !== null;

    if (isConnected) {
      console.log("Redis URL detected: " + url);
      const { keepValues } = await inquirer.prompt({
        type: "confirm",
        name: "keepValues",
        message: `Do you want to connect to the detected Redis URL?`,
        default: true,
      });

      if (keepValues) {
        console.log("Keeping existing Redis URL: " + url);
        host = "localhost";
        port = 6379;
        password = "";
      } else {
        isConnected = false;
      }
    }
    url = "";

    while (!isConnected) {
      const result = await askForRedisUrl();
      host = result.host;
      port = result.port;
      password = result.password;
      url = `redis://${password ? password + "@" : ""}${host}:${port}`;
      isConnected = await checkRedisConnection(url);
    }

    if (isConnected) {
      console.log("\nConnection to Redis successful! 🎉");
    }

    // Set the Redis parameters in .env_test
    if (process.env.NODE_ENV === "test") {
      // Update the .env_test file
      const config = dotenv.parse(fs.readFileSync(".env_test"));
      config.REDIS_HOST = host;
      config.REDIS_PORT = port.toString();
      config.REDIS_PASSWORD = password;
      updateEnvVariable(config);
    } else {
      // Set the Redis parameters in process.env
      process.env.REDIS_HOST = host;
      process.env.REDIS_PORT = port.toString();
      process.env.REDIS_PASSWORD = password;

      // Update the .env file
      const config = dotenv.parse(fs.readFileSync(".env"));
      config.REDIS_HOST = host;
      config.REDIS_PORT = port.toString();
      config.REDIS_PASSWORD = password;
      updateEnvVariable(config);
    }
  } catch (err) {
    console.error(err);
    abort();
  }
}

// Get the super admin email
/**
 * The function `superAdmin` prompts the user for a super admin email, updates a configuration file
 * with the email, and handles any errors that occur.
 */
export async function superAdmin(): Promise<void> {
  try {
    const email = await askForSuperAdminEmail();
    if (process.env.NODE_ENV === "test") {
      const config = dotenv.parse(fs.readFileSync(".env_test"));
      config.LAST_RESORT_SUPERADMIN_EMAIL = email;
      updateEnvVariable(config);
    } else {
      const config = dotenv.parse(fs.readFileSync(".env"));
      config.LAST_RESORT_SUPERADMIN_EMAIL = email;
      updateEnvVariable(config);
    }
  } catch (err) {
    console.log(err);
    abort();
  }
}

// Get the mongodb url
/**
 * The `mongoDB` function manages the connection to a MongoDB database.
 * It performs the following steps:
 * 1. Checks if there is an existing MongoDB URL in the environment variables.
 * 2. If an existing URL is found, it attempts to establish a connection to the URL.
 *    - If the connection is successful, it returns the URL.
 *    - If the connection fails, it returns null.
 * 3. If a URL is returned from the previous step (i.e., a successful connection was made), it prompts the user to either keep the existing URL or change it.
 * 4. If null is returned from the previous step (i.e., no successful connection was made), it prompts the user to enter a new MongoDB URL.
 * 5. It then enters a loop where it continues to prompt the user for a MongoDB URL until a successful connection can be made.
 * 6. Once a successful connection is made, it saves the MongoDB URL to the environment variables.
 * This function is part of the initial setup process and is designed to ensure a valid MongoDB connection before proceeding.
 */
export async function mongoDB(): Promise<void> {
  let DB_URL = process.env.MONGO_DB_URL;

  try {
    let url = await checkExistingMongoDB();
    let isConnected = url !== null;

    if (isConnected) {
      console.log("MongoDB URL detected: " + url);
      const { keepValues } = await inquirer.prompt({
        type: "confirm",
        name: "keepValues",
        message: `Do you want to connect to the detected MongoDB URL?`,
        default: true,
      });

      if (keepValues) {
        console.log("Keeping existing MongoDB URL: " + url);
      } else {
        isConnected = false;
      }
    }

    while (!isConnected) {
      url = await askForMongoDBUrl();
      isConnected = await checkConnection(url);
    }

    if (isConnected) {
      console.log("\nConnection to MongoDB successful! 🎉");
    }
    if (process.env.NODE_ENV === "test") {
      DB_URL = `${url?.endsWith("/talawa-api") ? url : `${url}/talawa-api`}`;
      const config = dotenv.parse(fs.readFileSync(".env_test"));
      // Not updating actual environmental variable when in testing environment.
      config.MONGO_DB_URL = DB_URL;
      updateEnvVariable(config);
    } else {
      DB_URL = `${url?.endsWith("/talawa-api") ? url : `${url}/talawa-api`}`;
      const config = dotenv.parse(fs.readFileSync(".env"));
      config.MONGO_DB_URL = DB_URL;
      // Modifying the environment variable to be able to access the URL in importData function.
      process.env.MONGO_DB_URL = DB_URL;
      updateEnvVariable(config);
    }
  } catch (err) {
    console.error(err);
    abort();
  }
}

/*
  For Docker setup
*/

function getDockerComposeCommand(): { command: string; args: string[] } {
  let dockerComposeCmd = "docker-compose"; // Default to v1
  let args = ["-f", "docker-compose.dev.yaml", "up", "--build", "-d"];

  try {
    // Test if 'docker compose' works (v2)
    execSync("docker compose version", { stdio: "ignore" });
    dockerComposeCmd = "docker";
    args = ["compose", ...args]; // Prefix 'compose' for v2
  } catch (error) {
    console.log(error);
    dockerComposeCmd =
      process.platform === "win32" ? "docker-compose.exe" : "docker-compose";
  }

  return { command: dockerComposeCmd, args };
}

const DOCKER_COMPOSE_TIMEOUT_MS = 300000;
async function runDockerComposeWithLogs(): Promise<void> {
  // Check if Docker daemon is running
  try {
    await new Promise((resolve, reject) => {
      const dockerCheck = spawn(
        process.platform === "win32" ? "docker.exe" : "docker",
        ["info"],
        { stdio: "ignore" },
      );
      dockerCheck.on("error", reject);
      dockerCheck.on("close", (code) =>
        code === 0
          ? resolve(null)
          : reject(
              new Error(
                "Docker daemon not running. Please ensure Docker Desktop is running and try again.",
              ),
            ),
      );
    });
  } catch (error: unknown) {
    const errorMessage = error instanceof Error ? error.message : String(error);
    throw new Error(
      `Docker daemon check failed. Please ensure:
       1. Docker Desktop is installed and running
       2. You have necessary permissions
       3. Docker service is healthy
       Details: ${errorMessage}`,
    );
  }

  return new Promise((resolve, reject) => {
    const { command, args } = getDockerComposeCommand();
    let isCompleted = false;

    const dockerCompose = spawn(command, args, { stdio: "inherit" });

    const timeout = setTimeout(() => {
      if (isCompleted) return;
      dockerCompose.kill();
      reject(new Error("Docker compose operation timed out after 5 minutes"));
    }, DOCKER_COMPOSE_TIMEOUT_MS);

    dockerCompose.on("error", (error) => {
      if (isCompleted) return;
      isCompleted = true;
      clearTimeout(timeout);
      console.error("Error running docker-compose:", error);
      reject(error);
    });

    dockerCompose.on("close", (code) => {
      if (isCompleted) return;
      isCompleted = true;
      clearTimeout(timeout);
      if (code === 0) {
        console.log("Docker Compose completed successfully.");
        resolve();
      } else {
        reject(new Error(`Docker Compose exited with code ${code}`));
      }
    });
  });
}

//Get recaptcha details
/**
 * The function `recaptcha` prompts the user to enter a reCAPTCHA secret key, validates the input, and
 * allows the user to choose whether to keep the entered value or try again.
 */
export async function recaptcha(): Promise<void> {
  const { recaptchaSecretKey } = await inquirer.prompt([
    {
      type: "input",
      name: "recaptchaSecretKey",
      message: "Enter your reCAPTCHA secret key:",
      validate: async (input: string): Promise<boolean | string> => {
        if (validateRecaptcha(input)) {
          return true;
        }
        return "Invalid reCAPTCHA secret key. Please try again.";
      },
    },
  ]);

  const shouldKeepDetails = await askToKeepValues();

  if (process.env.NODE_ENV === "test") {
    if (shouldKeepDetails) {
      const config = dotenv.parse(fs.readFileSync(".env_test"));
      config.RECAPTCHA_SECRET_KEY = recaptchaSecretKey;
      updateEnvVariable(config);
    }
  } else {
    if (shouldKeepDetails) {
      const config = dotenv.parse(fs.readFileSync(".env"));
      config.RECAPTCHA_SECRET_KEY = recaptchaSecretKey;
      updateEnvVariable(config);
    } else {
      await recaptcha();
    }
  }
}

/**
 * The function `recaptchaSiteKey` prompts the user to enter a reCAPTCHA site key, validates the input,
 * and updates the environment variable if the user chooses to keep the entered value.
 */
export async function recaptchaSiteKey(): Promise<void> {
  if (process.env.RECAPTCHA_SITE_KEY) {
    console.log(
      `\nreCAPTCHA site key already exists with the value ${process.env.RECAPTCHA_SITE_KEY}`,
    );
  }

  const { recaptchaSiteKeyInp } = await inquirer.prompt([
    {
      type: "input",
      name: "recaptchaSiteKeyInp",
      message: "Enter your reCAPTCHA site key:",
      validate: async (input: string): Promise<boolean | string> => {
        if (validateRecaptcha(input)) {
          return true;
        }
        return "Invalid reCAPTCHA site key. Please try again.";
      },
    },
  ]);

  const shouldKeepDetails = await askToKeepValues();

  if (process.env.NODE_ENV === "test") {
    if (shouldKeepDetails) {
      const config = dotenv.parse(fs.readFileSync(".env_test"));
      config.RECAPTCHA_SITE_KEY = recaptchaSiteKeyInp;
      updateEnvVariable(config);
    }
  } else {
    if (shouldKeepDetails) {
      const config = dotenv.parse(fs.readFileSync(".env"));
      config.RECAPTCHA_SITE_KEY = recaptchaSiteKeyInp;
      updateEnvVariable(config);
    } else {
      await recaptchaSiteKey();
    }
  }
}

/**
 * The `abort` function logs a message and exits the process.
 */
export function abort(): void {
  console.log("\nSetup process aborted. 🫠");
  process.exit(1);
}

//Get mail username and password
/**
 * The function `twoFactorAuth` prompts the user to set up Two-Factor Authentication Google Account and
 * then collects their email and generated password to update environment variables.
 */
export async function twoFactorAuth(): Promise<void> {
  console.log("\nIMPORTANT");
  console.log(
    "\nEnsure that you have Two-Factor Authentication set up on your Google Account.",
  );
  console.log("\nVisit Url: https://myaccount.google.com");
  console.log(
    "\nSelect Security and under Signing in to Google section select App Passwords.",
  );
  console.log(
    "\nClick on Select app section and choose Other(Custom name), enter talawa as the custom name and press Generate button.",
  );

  const { email, password } = await inquirer.prompt([
    {
      type: "input",
      name: "email",
      message: "Enter your email:",
      validate: (input: string): string | boolean =>
        isValidEmail(input) || "Invalid email. Please try again.",
    },
    {
      type: "password",
      name: "password",
      message: "Enter the generated password:",
    },
  ]);
  if (process.env.NODE_ENV === "test") {
    const config = dotenv.parse(fs.readFileSync(".env_test"));

    config.MAIL_USERNAME = email;
    config.MAIL_PASSWORD = password;
    updateEnvVariable(config);
  } else {
    const config = dotenv.parse(fs.readFileSync(".env"));

    config.MAIL_USERNAME = email;
    config.MAIL_PASSWORD = password;
    updateEnvVariable(config);
  }
}

/**
 * The function `configureSmtp` prompts the user to configure SMTP settings for sending emails through
 * Talawa and saves the configuration in a .env file.
 * @returns a Promise that resolves to void.
 */
export async function configureSmtp(): Promise<void> {
  const smtpConfig = await inquirer.prompt([
    {
      type: "input",
      name: "SMTP_HOST",
      message: "Enter SMTP host:",
    },
    {
      type: "input",
      name: "SMTP_PORT",
      message: "Enter SMTP port:",
    },
    {
      type: "input",
      name: "SMTP_USERNAME",
      message: "Enter SMTP username:",
    },
    {
      type: "password",
      name: "SMTP_PASSWORD",
      message: "Enter SMTP password:",
    },
    {
      type: "confirm",
      name: "SMTP_SSL_TLS",
      message: "Use SSL/TLS for SMTP?",
      default: false,
    },
  ]);

  const isValidSmtpConfig =
    smtpConfig.SMTP_HOST &&
    smtpConfig.SMTP_PORT &&
    smtpConfig.SMTP_USERNAME &&
    smtpConfig.SMTP_PASSWORD;

  if (!isValidSmtpConfig) {
    console.error(
      "Invalid SMTP configuration. Please provide all required parameters.",
    );
    return;
  }

  const { success, error } = await verifySmtpConnection(smtpConfig);

  if (!success) {
    console.error(
      "SMTP configuration verification failed. Please check your SMTP settings.",
    );
    if (error instanceof Error) {
      console.log(error.message);
    }
    return;
  }

  if (process.env.NODE_ENV === "test") {
    const config = dotenv.parse(fs.readFileSync(".env_test"));
    config.IS_SMTP = "true";
    Object.assign(config, smtpConfig);
    updateEnvVariable(config);
  } else {
    const config = dotenv.parse(fs.readFileSync(".env"));
    config.IS_SMTP = "true";
    Object.assign(config, smtpConfig);
    updateEnvVariable(config);
  }
  console.log("SMTP configuration saved successfully.");
}

/**
 * Configures MinIO settings, including installation check, data directory, and credentials.
 *
 * This function performs the following steps:
 * 1. Checks if MinIO is installed (for non-Docker installations)
 * 2. Prompts for MinIO installation if not found
 * 3. Checks for existing MinIO data directory configuration
 * 4. Allows user to change the data directory if desired
 * 5. Prompts for MinIO root user, password, and bucket name
 * 6. Updates the environment variables with the new configuration
 *
 * @param isDockerInstallation - A boolean indicating whether the setup is for a Docker installation.
 * @throws Will throw an error if there are issues with file operations or user input validation.
 * @returns A Promise that resolves when the configuration is complete.
 */
export async function configureMinio(
  isDockerInstallation: boolean,
): Promise<void> {
  if (!isDockerInstallation) {
    console.log("Checking MinIO installation...");
    if (isMinioInstalled()) {
      console.log("MinIO is already installed.");
    } else {
      console.log("MinIO is not installed on your system.");
      const { installMinioNow } = await inquirer.prompt([
        {
          type: "confirm",
          name: "installMinioNow",
          message: "Would you like to install MinIO now?",
          default: true,
        },
      ]);
      if (installMinioNow) {
        console.log("Installing MinIO...");
        try {
          await installMinio();
          console.log("Successfully installed MinIO on your system.");
        } catch (err) {
          console.error(err);
          return;
        }
      } else {
        console.log(
          "MinIO installation skipped. Please install MinIO manually before proceeding.",
        );
        return;
      }
    }
  }

  const envFile = process.env.NODE_ENV === "test" ? ".env_test" : ".env";
  const config = dotenv.parse(fs.readFileSync(envFile));

  const currentDataDir = config.MINIO_DATA_DIR || process.env.MINIO_DATA_DIR;
  let changeDataDir = false;

  if (currentDataDir) {
    console.log(
      `[MINIO] Existing MinIO data directory found: ${currentDataDir}`,
    );
    const { confirmChange } = await inquirer.prompt([
      {
        type: "confirm",
        name: "confirmChange",
        message:
          "Do you want to change the MinIO data directory? (Warning: All existing data will be lost)",
        default: false,
      },
    ]);
    changeDataDir = confirmChange;
  }

  if (!currentDataDir || changeDataDir) {
    const { MINIO_DATA_DIR } = await inquirer.prompt([
      {
        type: "input",
        name: "MINIO_DATA_DIR",
        message: "Enter MinIO data directory (press Enter for default):",
        default: "./data",
        validate: (input: string): boolean | string =>
          input.trim() !== "" ? true : "MinIO data directory is required.",
      },
    ]);

    if (changeDataDir && currentDataDir) {
      try {
        fs.rmSync(currentDataDir, { recursive: true, force: true });
        console.log(
          `[MINIO] Removed existing data directory: ${currentDataDir}`,
        );
      } catch (err) {
        console.error(`[MINIO] Error removing existing data directory: ${err}`);
      }
    }

    config.MINIO_DATA_DIR = MINIO_DATA_DIR;
    console.log(`[MINIO] MinIO data directory set to: ${MINIO_DATA_DIR}`);

    let fullPath = MINIO_DATA_DIR;
    if (!path.isAbsolute(MINIO_DATA_DIR)) {
      fullPath = path.join(process.cwd(), MINIO_DATA_DIR);
    }
    if (!fs.existsSync(fullPath)) {
      fs.mkdirSync(fullPath, { recursive: true });
    }
  }

  const minioConfig = await inquirer.prompt([
    {
      type: "input",
      name: "MINIO_ROOT_USER",
      message: "Enter MinIO root user:",
      default:
        config.MINIO_ROOT_USER || process.env.MINIO_ROOT_USER || "talawa",
      validate: (input: string): boolean | string =>
        input.trim() !== "" ? true : "MinIO root user is required.",
    },
    {
      type: "password",
      name: "MINIO_ROOT_PASSWORD",
      message: "Enter MinIO root password:",
      default:
        config.MINIO_ROOT_PASSWORD ||
        process.env.MINIO_ROOT_PASSWORD ||
        "talawa1234",
      validate: (input: string): boolean | string =>
        input.trim() !== "" ? true : "MinIO root password is required.",
    },
    {
      type: "input",
      name: "MINIO_BUCKET",
      message: "Enter MinIO bucket name:",
      default: config.MINIO_BUCKET || process.env.MINIO_BUCKET || "talawa",
      validate: (input: string): boolean | string =>
        input.trim() !== "" ? true : "MinIO bucket name is required.",
    },
  ]);

  const minioEndpoint = isDockerInstallation
    ? "http://minio:9000"
    : "http://localhost:9000";

  config.MINIO_ENDPOINT = minioEndpoint;
  config.MINIO_ROOT_USER = minioConfig.MINIO_ROOT_USER;
  config.MINIO_ROOT_PASSWORD = minioConfig.MINIO_ROOT_PASSWORD;
  config.MINIO_BUCKET = minioConfig.MINIO_BUCKET;

  updateEnvVariable(config);
  console.log("[MINIO] MinIO configuration added successfully.\n");
}

/**
 * The main function sets up the Talawa API by prompting the user to configure various environment
 * variables and import sample data if desired.
 */
async function main(): Promise<void> {
  console.log("Welcome to the Talawa API setup! 🚀");

  if (!fs.existsSync(".env")) {
    fs.copyFileSync(".env.sample", ".env");
  } else {
    checkEnvFile();
  }

  if (!fs.existsSync(".env_test")) {
    fs.copyFileSync(".env.sample", ".env_test");
  } else {
    checkEnvFile();
  }

  if (process.env.NODE_ENV) {
    console.log(`\nNode environment is already set to ${process.env.NODE_ENV}`);
  }
  await setNodeEnvironment();

  let accessToken: string | null = "",
    refreshToken: string | null = "";
  if (process.env.ACCESS_TOKEN_SECRET) {
    console.log(
      `\nAccess token secret already exists with the value:\n${process.env.ACCESS_TOKEN_SECRET}`,
    );
  }
  const { shouldGenerateAccessToken } = await inquirer.prompt({
    type: "confirm",
    name: "shouldGenerateAccessToken",
    message: "Would you like us to auto-generate a new access token secret?",
    default: process.env.ACCESS_TOKEN_SECRET ? false : true,
  });

  if (shouldGenerateAccessToken) {
    accessToken = null;
  }

  if (process.env.REFRESH_TOKEN_SECRET) {
    console.log(
      `\nRefresh token secret already exists with the value:\n${process.env.REFRESH_TOKEN_SECRET}`,
    );
  }
  const { shouldGenerateRefreshToken } = await inquirer.prompt({
    type: "confirm",
    name: "shouldGenerateRefreshToken",
    message:
      "Would you like to us to auto-generate a new refresh token secret?",
    default: process.env.REFRESH_TOKEN_SECRET ? false : true,
  });

  if (shouldGenerateRefreshToken) {
    refreshToken = null;
  }

  accessAndRefreshTokens(accessToken, refreshToken);

  const { shouldLog } = await inquirer.prompt({
    type: "confirm",
    name: "shouldLog",
    message: "Would you like to enable logging for the database transactions?",
    default: true,
  });

  if (shouldLog) {
    if (process.env.LOG_PATH) {
      console.log(
        `\n Log path already exists with the value:\n${process.env.LOG_PATH}`,
      );
    }
    let logPath: string | null = null;
    const { shouldUseCustomLogPath } = await inquirer.prompt({
      type: "confirm",
      name: "shouldUseCustomLogPath",
      message: "Would you like to provide a custom path for storing logs?",
      default: false,
    });

    if (shouldUseCustomLogPath) {
      logPath = await askForTransactionLogPath();
    }
    transactionLogPath(logPath);
  }

  const { isDockerInstallation } = await inquirer.prompt({
    type: "confirm",
    name: "isDockerInstallation",
    message: "Are you setting up this project using Docker?",
    default: process.env.MONGO ? false : true,
  });

  if (isDockerInstallation) {
    const DB_URL = "mongodb://localhost:27017/talawa-api";
    const REDIS_HOST = "localhost";
    const REDIS_PORT = "6379"; // default Redis port
    const REDIS_PASSWORD = "";
    const MINIO_ENDPOINT = "http://minio:9000";

    const { pwdVariable } = await inquirer.prompt({
      type: "input",
      name: "pwdVariable",
      message:
        "Please enter the value for PWD (working directory for Docker setup):",
      default: ".",
    });

    const config = dotenv.parse(fs.readFileSync(".env"));

    config.MONGO_DB_URL = DB_URL;
    config.REDIS_HOST = REDIS_HOST;
    config.REDIS_PORT = REDIS_PORT;
    config.REDIS_PASSWORD = REDIS_PASSWORD;
    config.MINIO_ENDPOINT = MINIO_ENDPOINT;
    config.PWD = pwdVariable;

    process.env.MONGO_DB_URL = DB_URL;
    process.env.REDIS_HOST = REDIS_HOST;
    process.env.REDIS_PORT = REDIS_PORT;
    process.env.REDIS_PASSWORD = REDIS_PASSWORD;
    process.env.MINIO_ENDPOINT = MINIO_ENDPOINT;
    process.env.PWD = pwdVariable;

    updateEnvVariable(config);
    console.log(`Your MongoDB URL is:\n${process.env.MONGO_DB_URL}`);
    console.log(`Your Redis host is:\n${process.env.REDIS_HOST}`);
    console.log(`Your Redis port is:\n${process.env.REDIS_PORT}`);
    console.log(`Your MinIO endpoint is:\n${process.env.MINIO_ENDPOINT}`);
    console.log(`Your PWD value is:\n${process.env.PWD}`);
  }

  if (!isDockerInstallation) {
    // Redis configuration
    if (process.env.REDIS_HOST && process.env.REDIS_PORT) {
      const redisPasswordStr = process.env.REDIS_PASSWORD
        ? "X".repeat(process.env.REDIS_PASSWORD.length)
        : "";

      const url = `redis://${
        process.env.REDIS_PASSWORD ? redisPasswordStr + "@" : ""
      }${process.env.REDIS_HOST}:${process.env.REDIS_PORT}`;

      console.log(`\nRedis URL already exists with the value:\n${url}`);

      const { shouldSetupRedis } = await inquirer.prompt({
        type: "confirm",
        name: "shouldSetupRedis",
        message: "Would you like to change the existing Redis URL?",
        default:
          process.env.REDIS_HOST && process.env.REDIS_PORT ? false : true,
      });

      if (shouldSetupRedis) {
        await redisConfiguration();
      }
    } else {
      await redisConfiguration();
    }

    // MongoDB configuration
    if (process.env.MONGO_DB_URL) {
      console.log(
        `\nMongoDB URL already exists with the value:\n${process.env.MONGO_DB_URL}`,
      );

      const { shouldSetupMongo } = await inquirer.prompt({
        type: "confirm",
        name: "shouldSetupMongo",
        message: "Would you like to change the existing Mongo DB URL?",
        default: false,
      });

      if (shouldSetupMongo) {
        await mongoDB();
      }
    } else {
      await mongoDB();
    }
  }
  if (process.env.RECAPTCHA_SECRET_KEY) {
    console.log(
      `\nreCAPTCHA secret key already exists with the value ${process.env.RECAPTCHA_SECRET_KEY}`,
    );
  }
  const { shouldSetRecaptcha } = await inquirer.prompt({
    type: "confirm",
    name: "shouldSetRecaptcha",
    message: "Would you like to set up a reCAPTCHA secret key?",
    default: process.env.RECAPTCHA_SECRET_KEY ? false : true,
  });

  if (shouldSetRecaptcha) {
    await recaptcha();
    await recaptchaSiteKey();
  }

  const { serverPort } = await inquirer.prompt([
    {
      type: "input",
      name: "serverPort",
      message: "Enter the Talawa-API server port:",
      default: process.env.SERVER_PORT || 4000,
    },
  ]);
  if (process.env.NODE_ENV === "development") {
    const config = dotenv.parse(fs.readFileSync(".env"));
    config.SERVER_PORT = serverPort;
    updateEnvVariable(config);
  }
  console.log(
    "\n You can configure either SMTP or Mail for sending emails through Talawa.\n",
  );

  if (process.env.MAIL_USERNAME) {
    console.log(
      `Mail username already exists with the value ${process.env.MAIL_USERNAME}`,
    );
  }
  const { shouldSetMail } = await inquirer.prompt([
    {
      type: "confirm",
      name: "shouldSetMail",
      message: "Would you like to setup the mail username and password?",
      default: process.env.MAIL_USERNAME ? false : true,
    },
  ]);
  if (shouldSetMail) {
    await twoFactorAuth();
  } else {
    console.log("Mail configuration skipped.\n");

    const { shouldConfigureSmtp } = await inquirer.prompt({
      type: "confirm",
      name: "shouldConfigureSmtp",
      message: "Would you like to configure SMTP for Talawa to send emails?",
      default: true,
    });

    if (shouldConfigureSmtp) {
      await configureSmtp();
    } else {
      console.log("SMTP configuration skipped.\n");
    }
  }

  console.log(
    `\nConfiguring MinIO storage...\n` +
      `${
        isDockerInstallation
          ? `Since you are using Docker, MinIO will be configured with the Docker-specific endpoint: http://minio:9000.\n`
          : `Since you are not using Docker, MinIO will be configured with the local endpoint: http://localhost:9000.\n`
      }`,
  );

  await configureMinio(isDockerInstallation);

  if (process.env.LAST_RESORT_SUPERADMIN_EMAIL) {
    console.log(
      `\nSuper Admin of last resort already exists with the value ${process.env.LAST_RESORT_SUPERADMIN_EMAIL}`,
    );
  }

  const { shouldSetSuperUserEmail } = await inquirer.prompt([
    {
      type: "confirm",
      name: "shouldSetSuperUserEmail",
      message: "Would you like to setup a Super Admin email of last resort?",
      default: process.env.LAST_RESORT_SUPERADMIN_EMAIL ? false : true,
    },
  ]);
  if (shouldSetSuperUserEmail) {
    await superAdmin();
  }
  // check if mail_username is set, if not, set it to mail_username's value
  else if (
    !shouldSetSuperUserEmail &&
    !process.env.LAST_RESORT_SUPERADMIN_EMAIL
    // process.env.MAIL_USERNAME
  ) {
    if (process.env.MAIL_USERNAME) {
      console.log(
        "No super admin email configured, setting it to mail username's value.",
      );
    }
    const config = dotenv.parse(fs.readFileSync(".env"));
    config.LAST_RESORT_SUPERADMIN_EMAIL = config.MAIL_USERNAME;
    updateEnvVariable(config);
  }

  const { imageSizeLimit } = await inquirer.prompt([
    {
      type: "input",
      name: "imageSizeLimit",
      message: `Enter the maximum size limit of Images uploaded (in MB) max: ${
        MAXIMUM_IMAGE_SIZE_LIMIT_KB / 1000
      }`,
      default: 3,
      validate: (input: number): boolean | string =>
        validateImageFileSize(input) ||
        `Enter a valid number between 0 and ${
          MAXIMUM_IMAGE_SIZE_LIMIT_KB / 1000
        }`,
    },
  ]);

  await setImageUploadSize(imageSizeLimit * 1000);

  if (!isDockerInstallation) {
    if (!process.env.MONGO_DB_URL) {
      console.log("Couldn't find mongodb url");
      return;
    }
    const isDbEmpty = await checkDb(process.env.MONGO_DB_URL);
    if (!isDbEmpty) {
      const { shouldOverwriteData } = await inquirer.prompt({
        type: "confirm",
        name: "shouldOverwriteData",
        message: "Do you want to overwrite the existing data?",
        default: false,
      });
      if (shouldOverwriteData) {
        await wipeExistingData(process.env.MONGO_DB_URL);
        const { overwriteDefaultData } = await inquirer.prompt({
          type: "confirm",
          name: "overwriteDefaultData",
          message:
            "Do you want to import the required default data to start using Talawa in a production environment?",
          default: false,
        });
        if (overwriteDefaultData) {
          await importDefaultData();
        } else {
          const { overwriteSampleData } = await inquirer.prompt({
            type: "confirm",
            name: "overwriteSampleData",
            message:
              "Do you want to import Talawa sample data for testing and evaluation purposes?",
            default: false,
          });
          if (overwriteSampleData) {
            await importData();
          }
        }
      }
    } else {
      const { shouldImportSampleData } = await inquirer.prompt({
        type: "confirm",
        name: "shouldImportSampleData",
        message:
          "Do you want to import Talawa sample data for testing and evaluation purposes?",
        default: false,
      });
      await wipeExistingData(process.env.MONGO_DB_URL);
      if (shouldImportSampleData) {
        await importData();
      } else {
        await importDefaultData();
      }
    }
  }

  console.log(
    "\nCongratulations! Talawa API has been successfully setup! 🥂🎉",
  );

  const { shouldStartDockerContainers } = await inquirer.prompt({
    type: "confirm",
    name: "shouldStartDockerContainers",
    message: "Do you want to start the Docker containers now?",
    default: true,
  });

  const { shouldImportSampleData } = await inquirer.prompt({
    type: "confirm",
    name: "shouldImportSampleData",
    message:
      "Do you want to import Talawa sample data for testing and evaluation purposes?",
    default: true,
  });

  if (isDockerInstallation) {
    if (shouldStartDockerContainers) {
      console.log("Starting docker container...");
      try {
        await runDockerComposeWithLogs();
        console.log("Docker containers have been built successfully!");
        // Wait for mongoDB to be ready
        console.log("Waiting for mongoDB to be ready...");
        let isConnected = false;
        const maxRetries = 30; // 30 seconds timeout
        let retryCount = 0;
        while (!isConnected) {
          if (retryCount >= maxRetries) {
            throw new Error(
              "Timed out waiting for MongoDB to be ready after 30 seconds",
            );
          }
          try {
            const client = new MongoClient(process.env.MONGO_DB_URL as string);
            await client.connect();
            await client.db().command({ ping: 1 });
            client.close();
            isConnected = true;
            console.log("MongoDB is ready!");
          } catch (err) {
            const error = err instanceof Error ? err.message : String(err);
            console.log(
              `Waiting for MongoDB to be ready... Retry ${retryCount + 1}/${maxRetries}. Details: ${error}`,
            );
            await new Promise((resolve) => setTimeout(resolve, 1000));
            retryCount++;
          }
        }

        if (shouldImportSampleData) {
          await importData();
        }
      } catch (err) {
        console.log("Some error occurred: " + err);
      }
    }
  }
}

main();