label-studio/src/core/Helpers.js

Data Labeling Tool that is backend agnostic and can be embedded into your applications

import { getType, getSnapshot } from "mobx-state-tree";
import { guidGenerator } from "../utils/unique";

/**
 * TODO: refactor
 */
export { guidGenerator };

/**
 * Helper function to detect HTX Component
 * @param {any} component
 * @param {string} name
 * @returns {boolean}
 */
export function isHtx(component, name) {
  return typeof component.type === "function" && component.type.name === "Htx" + name;
}

/**
 * Clone node with new ID
 * @param {*} node
 */
export function cloneNode(node) {
  const snapshot = getSnapshot(node);
  const snapshotRandomId = getType(node).create({
    ...snapshot,
    id: guidGenerator(),
  });
  return snapshotRandomId;
}

/**
 *
 * @param {*} fromModel
 */
export function restoreNewsnapshot(fromModel) {
  const snapshot = getSnapshot(fromModel);

  /**
   * Need to modify ID
   */
  const modifySnapshot = getType(fromModel).create({
    ...snapshot,
    id: guidGenerator(),
  });

  /**
   * Update state
   */
  return modifySnapshot;
}