import { Edge, Node } from '@vue-flow/core';
import { RULE_NODE_KEY, RULE_NODE_LOCAL_CACHE_KEY } from '/@/enums/cacheEnum';
import { createLocalStorage } from '/@/utils/cache';

const ruleNodeStorage = createLocalStorage({ prefixKey: RULE_NODE_LOCAL_CACHE_KEY });

interface RuleNodeCacheType {
  nodes?: Node[];
  edges?: Edge[];
  originX?: number;
  originY?: number;
}

export const setRuleNodeCache = ({
  nodes = [],
  edges = [],
  originX,
  originY,
}: RuleNodeCacheType) => {
  ruleNodeStorage.set(RULE_NODE_KEY, {
    nodes,
    edges,
    originX,
    originY,
  });
};

export const getRuleNodeCache = (): RuleNodeCacheType => ruleNodeStorage.get(RULE_NODE_KEY);

export const checkHasCacheRuleNode = () => !!getRuleNodeCache();

function initRuleNodeStorage() {
  const value = ruleNodeStorage.get(RULE_NODE_KEY);
  value && ruleNodeStorage.set(RULE_NODE_KEY, value);
}

initRuleNodeStorage();