alwaysai/src/components/device/target-hostname-check-component.ts

The alwaysAI command-line interface (CLI)

import { CLI_TERSE_ERROR } from '@alwaysai/alwayscli';
import {
  PROCESS_EXITED_WITH_NON_ZERO_STATUS_CODE,
  echo,
  promptForInput,
  connectBySsh,
  TIMED_OUT_CONNECTING_TO
} from '../../util';
import { connectBySshComponent } from '../general';
import { setUpPasswordlessSshComponent } from '../general/set-up-passwordless-ssh-component';

export async function targetHostnameCheckComponent(props: {
  targetHostname?: string;
  prompt?: boolean;
}) {
  let targetHostname: string = props.targetHostname ?? '';
  let connected = false;
  while (!connected) {
    if (props.prompt) {
      targetHostname = await promptForTargetHostname();
    }
    try {
      if (props.prompt) {
        await connectBySshComponent({
          targetHostname,
          warnOrFail: 'warn'
        });
      } else {
        await connectBySsh({ targetHostname });
      }
      connected = true;
    } catch (exception) {
      if (
        exception &&
        exception.code === CLI_TERSE_ERROR &&
        typeof exception.message === 'string' &&
        (exception.message.includes(PROCESS_EXITED_WITH_NON_ZERO_STATUS_CODE) ||
          exception.message.includes(TIMED_OUT_CONNECTING_TO))
      ) {
        if (exception.message.includes('Permission denied')) {
          await setUpPasswordlessSshComponent({ targetHostname });
          connected = true;
        } else {
          echo(exception.message);
          echo(
            'Cannot connect to your device. Please check the address and try again.'
          );
        }
      } else {
        throw exception;
      }
    }
  }

  return targetHostname;

  async function promptForTargetHostname() {
    const answers = await promptForInput({
      purpose: 'for the target hostname',
      questions: [
        {
          type: 'text',
          name: 'hostname',
          message:
            'Please enter the hostname (with optional user name) to connect to your device via ssh (e.g. "pi@1.2.3.4"):',
          initial: targetHostname,
          validate: (value) => (!value ? 'Value is required' : true)
        }
      ]
    });
    return answers.hostname as string;
  }
}