cosmos-explorer/src/Explorer/Notebook/NotebookComponent/reducers.ts

import { actions, CoreRecord, reducers as nteractReducers } from "@nteract/core";
import { Action } from "redux";
import * as cdbActions from "./actions";
import { CdbRecord } from "./types";

export const coreReducer = (state: CoreRecord, action: Action) => {
  let typedAction;
  switch (action.type) {
    case cdbActions.CLOSE_NOTEBOOK: {
      typedAction = action as cdbActions.CloseNotebookAction;
      return state.setIn(
        ["entities", "contents", "byRef"],
        state.entities.contents.byRef.delete(typedAction.payload.contentRef),
      );
    }
    case actions.CHANGE_KERNEL_BY_NAME: {
      // Update content metadata
      typedAction = action as actions.ChangeKernelByName;
      const kernelSpecName = typedAction.payload.kernelSpecName;

      if (!state.currentKernelspecsRef) {
        return state;
      }

      const currentKernelspecs = state.entities.kernelspecs.byRef.get(state.currentKernelspecsRef);
      if (!currentKernelspecs) {
        return state;
      }

      // Find kernelspecs by name
      const kernelspecs = currentKernelspecs.byName.get(kernelSpecName);
      if (!kernelspecs) {
        return state;
      }

      const path = [
        "entities",
        "contents",
        "byRef",
        typedAction.payload.contentRef,
        "model",
        "notebook",
        "metadata",
        "kernelspec",
      ];
      // Update metadata
      return state
        .setIn(path.concat("name"), kernelspecs.name)
        .setIn(path.concat("displayName"), kernelspecs.displayName)
        .setIn(path.concat("language"), kernelspecs.language);
    }
    default:
      return nteractReducers.core(state as any, action as any);
  }
};

export const cdbReducer = (state: CdbRecord, action: Action) => {
  if (!state) {
    return null;
  }

  switch (action.type) {
    case cdbActions.UPDATE_KERNEL_RESTART_DELAY: {
      const typedAction = action as cdbActions.UpdateKernelRestartDelayAction;
      return state.set("kernelRestartDelayMs", typedAction.payload.delayMs);
    }

    case cdbActions.SET_HOVERED_CELL: {
      const typedAction = action as cdbActions.SetHoveredCellAction;
      return state.set("hoveredCellId", typedAction.payload.cellId);
    }

    case cdbActions.STORE_CELL_OUTPUT_SNAPSHOT: {
      const typedAction = action as cdbActions.StoreCellOutputSnapshotAction;
      state.cellOutputSnapshots.set(typedAction.payload.cellId, typedAction.payload.snapshot);
      // TODO Simpler datastructure to instantiate new Map?
      return state.set("cellOutputSnapshots", new Map(state.cellOutputSnapshots));
    }

    case cdbActions.STORE_NOTEBOOK_SNAPSHOT: {
      const typedAction = action as cdbActions.StoreNotebookSnapshotAction;
      // Clear pending request
      return state.set("notebookSnapshot", typedAction.payload).set("pendingSnapshotRequest", undefined);
    }

    case cdbActions.TAKE_NOTEBOOK_SNAPSHOT: {
      const typedAction = action as cdbActions.TakeNotebookSnapshotAction;
      // Clear previous snapshots
      return state
        .set("cellOutputSnapshots", new Map())
        .set("notebookSnapshot", undefined)
        .set("notebookSnapshotError", undefined)
        .set("pendingSnapshotRequest", typedAction.payload);
    }

    case cdbActions.NOTEBOOK_SNAPSHOT_ERROR: {
      const typedAction = action as cdbActions.NotebookSnapshotErrorAction;
      return state.set("notebookSnapshotError", typedAction.payload.error);
    }
  }
  return state;
};