cosmos-explorer/src/Terminal/index.ts

import { ServerConnection } from "@jupyterlab/services";
import "@jupyterlab/terminal/style/index.css";
import { HttpHeaders, TerminalQueryParams } from "../Common/Constants";
import { Action } from "../Shared/Telemetry/TelemetryConstants";
import * as TelemetryProcessor from "../Shared/Telemetry/TelemetryProcessor";
import { updateUserContext } from "../UserContext";
import "./index.css";
import { JupyterLabAppFactory } from "./JupyterLabAppFactory";

const getUrlVars = (): { [key: string]: string } => {
  const vars: { [key: string]: string } = {};
  window.location.href.replace(/[?&]+([^=&]+)=([^&]*)/gi, (_m, key, value): string => {
    vars[key] = decodeURIComponent(value);
    return value;
  });
  return vars;
};

const createServerSettings = (urlVars: { [key: string]: string }): ServerConnection.ISettings => {
  let body: BodyInit | undefined;
  let headers: HeadersInit | undefined;
  if (urlVars.hasOwnProperty(TerminalQueryParams.TerminalEndpoint)) {
    body = JSON.stringify({
      endpoint: urlVars[TerminalQueryParams.TerminalEndpoint],
    });
    headers = {
      [HttpHeaders.contentType]: "application/json",
    };
  }

  const server = urlVars[TerminalQueryParams.Server];
  let options: Partial<ServerConnection.ISettings> = {
    baseUrl: server,
    init: { body, headers },
    fetch: window.parent.fetch,
  };
  if (urlVars.hasOwnProperty(TerminalQueryParams.Token)) {
    options = {
      baseUrl: server,
      token: urlVars[TerminalQueryParams.Token],
      appendToken: true,
      init: { body, headers },
      fetch: window.parent.fetch,
    };
  }

  return ServerConnection.makeSettings(options);
};

const main = async (): Promise<void> => {
  const urlVars = getUrlVars();

  // Initialize userContext. Currently only subscriptionId is required by TelemetryProcessor
  updateUserContext({
    subscriptionId: urlVars[TerminalQueryParams.SubscriptionId],
  });

  const serverSettings = createServerSettings(urlVars);

  const data = { baseUrl: serverSettings.baseUrl };
  const startTime = TelemetryProcessor.traceStart(Action.OpenTerminal, data);

  try {
    if (urlVars.hasOwnProperty(TerminalQueryParams.Terminal)) {
      await JupyterLabAppFactory.createTerminalApp(serverSettings);
    } else {
      throw new Error("Only terminal is supported");
    }

    TelemetryProcessor.traceSuccess(Action.OpenTerminal, data, startTime);
  } catch (error) {
    TelemetryProcessor.traceFailure(Action.OpenTerminal, data, startTime);
  }
};

window.addEventListener("load", main);