Termix/src/ui/sidebar/quick-connect-host.ts

import type { SSHHostData } from "@/types";
import type { Host } from "@/types/ui-types";
import type { HostProtocolDef } from "./host-protocols";

type QuickConnectInput = Pick<
  Host,
  "ip" | "port" | "username" | "authType" | "password" | "key" | "credentialId"
> & {
  /** A plugin protocol; SSH when omitted. */
  protocol?: HostProtocolDef;
  domain?: string;
};

export const QUICK_CONNECT_ID_PREFIX = "quick-connect-";

export function isQuickConnectHost(host: Pick<Host, "id">): boolean {
  return host.id.startsWith(QUICK_CONNECT_ID_PREFIX);
}

export function createQuickConnectHost(input: QuickConnectInput): Host {
  const protocol = input.protocol;
  if (protocol) {
    // Protocol logins are core host fields named after the protocol.
    const login = {
      [`${protocol.id}AuthType`]: "direct",
      [`${protocol.id}User`]: input.username,
      [`${protocol.id}Password`]: input.password,
    };
    return {
      ...createQuickConnectHost({ ...input, protocol: undefined, port: 22 }),
      ...login,
      port: input.port,
      enableTerminal: false,
      enableCommandHistory: false,
      enableFileManager: false,
      enableTunnel: false,
      enableTerminalToolbar: false,
      enableAiAssistant: false,
      enableSsh: false,
      domain: input.domain,
      pluginSettings: {
        [protocol.pluginId]: {
          [protocol.settingKey]: true,
          ...(protocol.portKey ? { [protocol.portKey]: input.port } : {}),
        },
      },
    };
  }
  return {
    id: `${QUICK_CONNECT_ID_PREFIX}${Date.now()}`,
    name: `${input.username}@${input.ip}`,
    ip: input.ip,
    port: input.port,
    username: input.username,
    authType: input.authType,
    password: input.authType === "password" ? input.password : undefined,
    key: input.authType === "key" ? input.key : undefined,
    credentialId:
      input.authType === "credential" ? input.credentialId : undefined,
    folder: "",
    online: false,
    cpu: null,
    ram: null,
    lastAccess: new Date().toISOString(),
    pin: false,
    defaultPath: "",
    serverTunnels: [],
    quickActions: [],
    enableTerminal: true,
    enableCommandHistory: true,
    enableFileManager: true,
    enableTunnel: true,
    enableProxmox: false,
    enableProxmoxStats: false,
    enableTmuxMonitor: false,
    enableTerminalToolbar: true,
    enableAiAssistant: false,
    enableSsh: true,
    sshPort: input.port,
  };
}

export function quickConnectHostToPayload(host: Host): SSHHostData {
  return {
    name: host.name,
    ip: host.ip,
    port: host.port,
    username: host.username,
    authType: host.authType,
    password: host.password,
    key: host.key,
    credentialId: host.credentialId
      ? Number.parseInt(host.credentialId, 10)
      : null,
    folder: host.folder,
    pin: host.pin,
    defaultPath: host.defaultPath,
    enableTerminal: true,
    enableSessionLogging: true,
    enableCommandHistory: host.enableCommandHistory,
    enableFileManager: host.enableFileManager,
    enableTunnel: host.enableTunnel,
    enableProxmox: host.enableProxmox,
    enableTmuxMonitor: host.enableTmuxMonitor,
    enableTerminalToolbar: host.enableTerminalToolbar,
    enableAiAssistant: host.enableAiAssistant,
    showTerminalInSidebar: true,
    showFileManagerInSidebar: true,
    showTunnelInSidebar: true,
    showDockerInSidebar: true,
    showServerStatsInSidebar: true,
    connectionType: "ssh",
    enableSsh: true,
    sshPort: host.sshPort,
  };
}