cosmos-explorer/src/Explorer/Tables/DataTable/TableCommands.ts

import _ from "underscore";
import Q from "q";
import * as DataTableUtilities from "./DataTableUtilities";
import * as DataTableOperations from "./DataTableOperations";
import TableEntityListViewModel from "./TableEntityListViewModel";
import * as Entities from "../Entities";
import * as ViewModels from "../../../Contracts/ViewModels";
import * as TableColumnOptionsPane from "../../Panes/Tables/TableColumnOptionsPane";
import Explorer from "../../Explorer";

export default class TableCommands {
  // Command Ids
  public static editEntityCommand: string = "edit";
  public static deleteEntitiesCommand: string = "delete";
  public static reorderColumnsCommand: string = "reorder";
  public static resetColumnsCommand: string = "reset";
  public static customizeColumnsCommand: string = "customizeColumns";

  private _container: Explorer;

  constructor(container: Explorer) {
    this._container = container;
  }

  public isEnabled(commandName: string, selectedEntites: Entities.ITableEntity[]): boolean {
    var singleItemSelected: boolean = DataTableUtilities.containSingleItem(selectedEntites);
    var atLeastOneItemSelected: boolean = DataTableUtilities.containItems(selectedEntites);
    switch (commandName) {
      case TableCommands.editEntityCommand:
        return singleItemSelected;
      case TableCommands.deleteEntitiesCommand:
      case TableCommands.reorderColumnsCommand:
        return atLeastOneItemSelected;
      default:
        break;
    }

    return false;
  }

  public tryOpenEntityEditor(viewModel: TableEntityListViewModel): boolean {
    if (this.isEnabled(TableCommands.editEntityCommand, viewModel.selected())) {
      this.editEntityCommand(viewModel);
      return true;
    }
    return false;
  }

  /**
   * Edit entity
   */
  public editEntityCommand(viewModel: TableEntityListViewModel): Q.Promise<any> {
    if (!viewModel) {
      return null; // Error
    }

    if (!DataTableUtilities.containSingleItem(viewModel.selected())) {
      return null; // Erorr
    }

    var entityToUpdate: Entities.ITableEntity = viewModel.selected()[0];
    var originalNumberOfProperties = entityToUpdate ? 0 : Object.keys(entityToUpdate).length - 1; // .metadata is always a property for etag

    this._container.editTableEntityPane.originEntity = entityToUpdate;
    this._container.editTableEntityPane.tableViewModel = viewModel;
    this._container.editTableEntityPane.originalNumberOfProperties = originalNumberOfProperties;
    this._container.editTableEntityPane.open();
    return null;
  }

  public deleteEntitiesCommand(viewModel: TableEntityListViewModel): Q.Promise<any> {
    if (!viewModel) {
      return null; // Error
    }
    if (!DataTableUtilities.containItems(viewModel.selected())) {
      return null; // Error
    }
    var entitiesToDelete: Entities.ITableEntity[] = viewModel.selected();
    let deleteMessage: string = "Are you sure you want to delete the selected entities?";
    if (viewModel.queryTablesTab.container.isPreferredApiCassandra()) {
      deleteMessage = "Are you sure you want to delete the selected rows?";
    }
    if (window.confirm(deleteMessage)) {
      viewModel.queryTablesTab.container.tableDataClient
        .deleteDocuments(viewModel.queryTablesTab.collection, entitiesToDelete)
        .then((results: any) => {
          return viewModel.removeEntitiesFromCache(entitiesToDelete).then(() => {
            viewModel.redrawTableThrottled();
          });
        });
    }
    return null;
  }

  public customizeColumnsCommand(viewModel: TableEntityListViewModel): Q.Promise<any> {
    var table: DataTables.DataTable = viewModel.table;
    var displayedColumnNames: string[] = DataTableOperations.getDataTableHeaders(table);
    var columnsCount: number = displayedColumnNames.length;
    var currentOrder: number[] = DataTableOperations.getInitialOrder(columnsCount);
    //Debug.assert(!!table && !!currentOrder && displayedColumnNames.length === currentOrder.length);

    var currentSettings: boolean[];
    try {
      currentSettings = currentOrder.map((value: number, index: number) => {
        return table.column(index).visible();
      });
    } catch (err) {
      // Error
    }

    let parameters: TableColumnOptionsPane.IColumnSetting = <TableColumnOptionsPane.IColumnSetting>{
      columnNames: displayedColumnNames,
      order: currentOrder,
      visible: currentSettings
    };

    this._container.tableColumnOptionsPane.tableViewModel = viewModel;
    this._container.tableColumnOptionsPane.parameters = parameters;
    this._container.tableColumnOptionsPane.open();
    return null;
  }

  public reorderColumnsBasedOnSelectedEntities(viewModel: TableEntityListViewModel): Q.Promise<boolean> {
    var selected = viewModel.selected();
    if (!selected || !selected.length) {
      return null;
    }

    var table = viewModel.table;
    var currentColumnNames: string[] = DataTableOperations.getDataTableHeaders(table);
    var headersCount: number = currentColumnNames.length;

    var headersUnion: string[] = DataTableUtilities.getPropertyIntersectionFromTableEntities(
      selected,
      viewModel.queryTablesTab.container.isPreferredApiCassandra()
    );

    // An array with elements representing indexes of selected entities' header union out of initial headers.
    var orderOfLeftHeaders: number[] = headersUnion.map((item: string) => currentColumnNames.indexOf(item));

    // An array with elements representing initial order of the table.
    var initialOrder: number[] = DataTableOperations.getInitialOrder(headersCount);

    // An array with elements representing indexes of headers not present in selected entities' header union.
    var orderOfRightHeaders: number[] = _.difference(initialOrder, orderOfLeftHeaders);

    // This will be the target order, with headers in selected entities on the left while others on the right, both in the initial order, respectively.
    var targetOrder: number[] = orderOfLeftHeaders.concat(orderOfRightHeaders);

    return DataTableOperations.reorderColumns(table, targetOrder);
  }

  public resetColumns(viewModel: TableEntityListViewModel): void {
    viewModel.reloadTable();
  }
}