import Vorpal from "vorpal";
import { logError } from "../../../../utils/log";
import { getProjectNamespace } from "../../../../utils/projects";
import { envAutocompletion } from "../project/utils/autocompletions";
import ensureCluster from "../project/utils/ensureCluster";
import {
  getMongodbShell,
  getProjectMongodbAllPodsSortedWithLabel
} from "./utils";

export default (vorpal: Vorpal) =>
  vorpal
    .command(
      "project-mongo-get-shell <env>",
      "get a shell to a mongodb in the environment"
    )
    .autocomplete(envAutocompletion)
    .action(async function({ env }) {
      await ensureCluster.call(this);
      const namespace = await getProjectNamespace(env);
      const podNames = await getProjectMongodbAllPodsSortedWithLabel(env);
      if (podNames.length === 0) {
        logError(this, "sorry, no pods found");
        return;
      }
      let podName;
      if (podNames.length === 1) {
        podName = podNames[0];
      } else {
        podName = (
          await this.prompt({
            type: "list",
            name: "podName",
            choices: podNames,
            message: "Which pod? 🤔"
          })
        ).podName;
      }

      return getMongodbShell(namespace, podName);
    });