import { CliLeaf, CliFlagInput, CliTerseError } from '@alwaysai/alwayscli';
import { TargetJsonFile } from '../../core/app';
import { VENV_BIN_ACTIVATE } from '../../paths';
import { ALWAYSAI_CLI_EXECUTABLE_NAME } from '../../constants';

export const appShellCliLeaf = CliLeaf({
  name: 'shell',
  description: 'Launch a bash shell in the target directory',
  namedInputs: {
    'no-container': CliFlagInput({
      description:
        'Open the shell on the target host, not in a container on the target host'
    }),
    superuser: CliFlagInput({
      description: 'Open the shell as superuser "root"'
    })
  },
  async action(_, opts) {
    const targetJsonFile = TargetJsonFile();
    const targetJson = targetJsonFile.read();
    const tty = process.stdin.isTTY;
    switch (targetJson.targetProtocol) {
      case 'docker:': {
        targetJsonFile.readContainerSpawner().runForegroundSync({
          exe: '/bin/bash',
          args: ['--rcfile', VENV_BIN_ACTIVATE],
          tty,
          cwd: '.',
          expose5000: true,
          superuser: opts.superuser
        });
        break;
      }

      case 'ssh+docker:': {
        if (opts['no-container']) {
          if (opts.superuser) {
            throw new CliTerseError(
              '--superuser is not yet supported with --no-container'
            );
          }
          targetJsonFile.readHostSpawner().runForegroundSync({
            exe: '/bin/bash',
            tty,
            cwd: '.',
            expose5000: true
          });
        } else {
          targetJsonFile.readContainerSpawner().runForegroundSync({
            exe: '/bin/bash',
            args: ['--rcfile', VENV_BIN_ACTIVATE],
            tty,
            cwd: '.',
            expose5000: true,
            superuser: opts.superuser
          });
        }
        break;
      }

      default: {
        throw new CliTerseError(
          `${ALWAYSAI_CLI_EXECUTABLE_NAME} app shell only supports the following protocols: "docker:", "ssh+docker:"`
        );
      }
    }
  }
});