joyent-portal/packages/ui-toolkit/src/topology/functions.js

import Constants from './constants';

const getAngleFromPoints = (source, target) => {
  const lineAngle = Math.atan2(target.y - source.y, target.x - source.x);
  const lineAngleDeg = lineAngle * 180 / Math.PI;
  const zeroToThreeSixty = lineAngleDeg < 0 ? 360 + lineAngleDeg : lineAngleDeg;

  return zeroToThreeSixty;
};

const getPosition = (angle, positions, position, noCorners = false) => {
  const positionIndex = noCorners
    ? Math.round(angle / 90) * 2
    : Math.round(angle / 45);

  const offsetPosition = positions[positionIndex];

  return {
    id: offsetPosition.id,
    x: position.x + offsetPosition.x,
    y: position.y + offsetPosition.y
  };
};

const getPositions = (rect, halfCorner = 0) => [
  {
    id: 'r',
    x: rect.right,
    y: 0
  },
  {
    id: 'br',
    x: rect.right - halfCorner,
    y: rect.bottom - halfCorner
  },
  {
    id: 'b',
    x: 0,
    y: rect.bottom
  },
  {
    id: 'bl',
    x: rect.left + halfCorner,
    y: rect.bottom - halfCorner
  },
  {
    id: 'l',
    x: rect.left,
    y: 0
  },
  {
    id: 'tl',
    x: rect.left + halfCorner,
    y: rect.top + halfCorner
  },
  {
    id: 't',
    x: 0,
    y: rect.top
  },
  {
    id: 'tr',
    x: rect.right - halfCorner,
    y: rect.top + halfCorner
  },
  {
    id: 'r',
    x: rect.right,
    y: 0
  }
];

/* const getRect = data =>
  data.children ? Constants.nodeRectWithChildren : Constants.nodeRect; */

const calculateLineLayout = ({ source, target }) => {
  // Actually, this will need to be got dynamically, in case them things are different sizes
  // yeah right, now you'll get to do exactly that

  const halfCorner = 2;

  const sourcePositions = getPositions(source.nodeRect, halfCorner);
  const sourceAngle = getAngleFromPoints(source, target);
  const sourcePosition = getPosition(sourceAngle, sourcePositions, source);

  const targetPositions = getPositions(target.nodeRect, halfCorner);
  const targetAngle = getAngleFromPoints(target, sourcePosition);
  const targetPosition = getPosition(targetAngle, targetPositions, target); // , true);

  const arrowAngle = getAngleFromPoints(sourcePosition, targetPosition);

  return {
    source,
    target,
    sourcePosition,
    targetPosition,
    arrowAngle
  };
};

const getStatusesLength = data =>
  data.transitionalStatus ? 1 : data.instanceStatuses.length;

const getStatusesHeight = data => {
  const statuses = data.children
    ? data.children.reduce(
        (statuses, child) => statuses + getStatusesLength(child),
        0
      )
    : getStatusesLength(data);

  return statuses ? Constants.statusHeight * statuses + 6 : 0;
};

const getContentRect = (data, isChild = false) => {
  const contentSize = isChild
    ? Constants.childContentSize
    : Constants.contentSize;

  const { height } = contentSize;
  const contentHeight = height + getStatusesHeight(data);

  return {
    ...Constants.contentPosition,
    width: contentSize.width,
    height: contentHeight
  };
};

const getNodeRect = data => {
  const nodeSize = data.children
    ? Constants.nodeSizeWithChildren
    : Constants.nodeSize;

  const { width, height } = nodeSize;
  const nodeHeight = height + getStatusesHeight(data);

  return {
    left: -width / 2,
    right: width / 2,
    top: -height / 2,
    bottom: nodeHeight - height / 2,
    width,
    height: nodeHeight
  };
};

export { getContentRect, getNodeRect, calculateLineLayout };