cosmos-explorer/src/Explorer/Tree/UserDefinedFunction.ts

import * as ko from "knockout";
import * as ViewModels from "../../Contracts/ViewModels";
import * as Constants from "../../Common/Constants";
import * as DataModels from "../../Contracts/DataModels";
import { Action, ActionModifiers } from "../../Shared/Telemetry/TelemetryConstants";
import UserDefinedFunctionTab from "../Tabs/UserDefinedFunctionTab";
import ContextMenu from "../Menus/ContextMenu";
import TelemetryProcessor from "../../Shared/Telemetry/TelemetryProcessor";
import { ContextMenuButtonFactory } from "../ContextMenuButtonFactory";
import Collection from "./Collection";

export default class UserDefinedFunction implements ViewModels.UserDefinedFunction {
  public nodeKind: string;
  public container: ViewModels.Explorer;
  public collection: ViewModels.Collection;
  public self: string;
  public rid: string;
  public id: ko.Observable<string>;
  public body: ko.Observable<string>;
  public contextMenu: ViewModels.ContextMenu;

  constructor(container: ViewModels.Explorer, collection: ViewModels.Collection, data: DataModels.UserDefinedFunction) {
    this.nodeKind = "UserDefinedFunction";
    this.container = container;

    this.collection = collection;
    this.self = data._self;
    this.rid = data._rid;
    this.id = ko.observable(data.id);
    this.body = ko.observable(data.body);

    this.contextMenu = new ContextMenu(this.container, this.rid);
    this.contextMenu.options(ContextMenuButtonFactory.createUserDefinedFunctionContextMenuButton(container));
  }

  public static create(source: ViewModels.Collection, event: MouseEvent) {
    const id =
      source.container
        .openedTabs()
        .filter((tab: ViewModels.Tab) => tab.tabKind === ViewModels.CollectionTabKind.UserDefinedFunctions).length + 1;
    const userDefinedFunction = <DataModels.UserDefinedFunction>{
      id: "",
      body: "function userDefinedFunction(){}"
    };
    let userDefinedFunctionTab: ViewModels.Tab = new UserDefinedFunctionTab({
      resource: userDefinedFunction,
      isNew: true,
      tabKind: ViewModels.CollectionTabKind.UserDefinedFunctions,
      title: `New UDF ${id}`,
      tabPath: "",
      documentClientUtility: source.container.documentClientUtility,
      collection: source,
      node: source,
      hashLocation: `${Constants.HashRoutePrefixes.collectionsWithIds(source.databaseId, source.id())}/udf`,
      selfLink: "",
      isActive: ko.observable(false),
      onUpdateTabsButtons: source.container.onUpdateTabsButtons,
      openedTabs: source.container.openedTabs()
    });
    source.container.openedTabs.push(userDefinedFunctionTab);

    // Activate
    userDefinedFunctionTab.onTabClick();

    // Hide Context Menu (if necessary)
    source.contextMenu.hide(source, event);
  }

  public open = () => {
    this.select();

    let userDefinedFunctionTab: ViewModels.Tab = this.container
      .openedTabs()
      .filter(tab => tab.node && tab.node.rid === this.rid)[0];
    if (!userDefinedFunctionTab) {
      const userDefinedFunctionData = <DataModels.UserDefinedFunction>{
        _rid: this.rid,
        _self: this.self,
        id: this.id(),
        body: this.body()
      };

      userDefinedFunctionTab = new UserDefinedFunctionTab({
        resource: userDefinedFunctionData,
        isNew: false,
        tabKind: ViewModels.CollectionTabKind.UserDefinedFunctions,
        title: userDefinedFunctionData.id,
        tabPath: "",
        documentClientUtility: this.container.documentClientUtility,
        collection: this.collection,
        node: this,
        hashLocation: `${Constants.HashRoutePrefixes.collectionsWithIds(
          this.collection.databaseId,
          this.collection.id()
        )}/udfs/${this.id()}`,
        selfLink: "",
        isActive: ko.observable(false),
        onUpdateTabsButtons: this.container.onUpdateTabsButtons,
        openedTabs: this.container.openedTabs()
      });
      this.container.openedTabs.push(userDefinedFunctionTab);
    }

    // Activate
    userDefinedFunctionTab.onTabClick();
  };

  public select() {
    this.container.selectedNode(this);
    TelemetryProcessor.trace(Action.SelectItem, ActionModifiers.Mark, {
      description: "UDF item node",
      databaseAccountName: this.container.databaseAccount().name,
      defaultExperience: this.container.defaultExperience(),
      dataExplorerArea: Constants.Areas.ResourceTree
    });
  }

  public delete(source: Collection, event: MouseEvent | KeyboardEvent) {
    // Hide Context Menu (if necessary)
    this.contextMenu.hide(source, event);

    if (!window.confirm("Are you sure you want to delete the user defined function?")) {
      return;
    }

    const userDefinedFunctionData = <DataModels.UserDefinedFunction>{
      _rid: this.rid,
      _self: this.self,
      id: this.id(),
      body: this.body()
    };
    this.container.documentClientUtility.deleteUserDefinedFunction(this.collection, userDefinedFunctionData).then(
      () => {
        this.container.openedTabs.remove((tab: ViewModels.Tab) => tab.node && tab.node.rid === this.rid);
        this.collection.children.remove(this);
      },
      reason => {}
    );
  }

  public onKeyDown = (source: any, event: KeyboardEvent): boolean => {
    if (event.key === "Delete") {
      this.delete(source, event);
      return false;
    }

    return true;
  };

  public onMenuKeyDown = (source: any, event: KeyboardEvent): boolean => {
    if (event.key === "Escape") {
      this.contextMenu.hide(source, event);
      return false;
    }

    return true;
  };

  public onKeyPress = (source: any, event: KeyboardEvent): boolean => {
    if (event.key === " " || event.key === "Enter") {
      this.open();
      return false;
    }

    return true;
  };
}