349 lines
10 KiB
JavaScript
349 lines
10 KiB
JavaScript
const convert = require('xml-js');
|
|
const utils = require('./utils');
|
|
|
|
let resources = {
|
|
name: 'error',
|
|
height: 512,
|
|
index: [],
|
|
links: {
|
|
'arrow-head': {},
|
|
'arrow-head-reverse': {},
|
|
'line-start': {},
|
|
'line-end': {},
|
|
'dashed-step': {}
|
|
},
|
|
icons: {}
|
|
};
|
|
try {
|
|
resources = require('../resources.json');
|
|
} catch (err) {
|
|
console.error('fa-diagrams: SVG resources could not be loaded: ' + err);
|
|
}
|
|
|
|
/**
|
|
* @typedef Node2
|
|
* @property {string} name
|
|
* @property {number} x
|
|
* @property {number} y
|
|
* @property {string} icon
|
|
*/
|
|
|
|
/**
|
|
* @typedef Link2
|
|
* @property {string} from
|
|
* @property {string} to
|
|
* @property {string|undefined} type
|
|
*/
|
|
|
|
const SUB_DEF = {
|
|
'_': 'string',
|
|
'text': 'string',
|
|
'icon': 'string',
|
|
'color': 'string',
|
|
'font': 'string',
|
|
'font-size': 'number',
|
|
'font-style': 'string',
|
|
'scale': 'number',
|
|
};
|
|
|
|
const NODE_DEF = {
|
|
'name': '!string',
|
|
'icon': '!string',
|
|
'x': '!number',
|
|
'y': '!number',
|
|
'color': 'string',
|
|
'scale': 'number',
|
|
'top': SUB_DEF,
|
|
'bottom': SUB_DEF,
|
|
'left': SUB_DEF,
|
|
'right': SUB_DEF
|
|
};
|
|
|
|
const LINK_DEF = {
|
|
'from': '!string',
|
|
'to': '!string',
|
|
'type': 'string',
|
|
'color': 'string',
|
|
'scale': 'number',
|
|
'size': 'number',
|
|
'top': SUB_DEF,
|
|
'bottom': SUB_DEF,
|
|
};
|
|
|
|
const DEFAULT_OPTIONS = {
|
|
'beautify': false,
|
|
'scale': 128,
|
|
'h-spacing': 1.3,
|
|
'color': 'black',
|
|
'icons': {
|
|
'scale': 1,
|
|
'color': ''
|
|
},
|
|
'links': {
|
|
'scale': 1,
|
|
'color': '',
|
|
'size': 0
|
|
},
|
|
'texts': {
|
|
'font': 'sans-serif',
|
|
'font-size': '20',
|
|
'font-style': 'normal',
|
|
'color': ''
|
|
}
|
|
};
|
|
|
|
const SUBSTITUTES = {
|
|
'font-awesome': {
|
|
'fas': 'solid',
|
|
'far': 'regular',
|
|
'fab': 'brands'
|
|
}
|
|
};
|
|
|
|
const DEFAULT_SCALE = 0.4;
|
|
const LINK_MARGIN = (1 - DEFAULT_SCALE) / 2;
|
|
|
|
module.exports = (options) => {
|
|
|
|
options = utils.merge(DEFAULT_OPTIONS, options);
|
|
|
|
const self = {
|
|
|
|
/**
|
|
* Find icon data from given name
|
|
* @param {string} name
|
|
* @returns {null|{path: string, width: number}}
|
|
*/
|
|
getIcon: (name) => {
|
|
if (!name || !name.trim())
|
|
return null;
|
|
|
|
let search = utils.ezClone(resources.index);
|
|
const spl = name.trim().split(' ').map(t => t.indexOf('fa-') === 0 ? t.substr(3) : t);
|
|
|
|
for (let i = 0; i < spl.length; i++) {
|
|
//replace fas by regular for example
|
|
if (Object.keys(SUBSTITUTES[resources.name] || {}).includes(spl[i])) {
|
|
spl[i] = SUBSTITUTES[resources.name][spl[i]];
|
|
}
|
|
if (resources.index.includes(spl[i])) {
|
|
search = [spl.splice(i, 1)];
|
|
}
|
|
}
|
|
|
|
name = spl[0];
|
|
|
|
for (let i = 0; i < search.length; i++) {
|
|
if (resources.icons[search[i]] && resources.icons[search[i]][name]) {
|
|
return resources.icons[search[i]][name];
|
|
}
|
|
}
|
|
|
|
return null;
|
|
},
|
|
|
|
/**
|
|
* Create the correct path from the type and width
|
|
* @param {string} type
|
|
* @param {number} width
|
|
* @return {string|null}
|
|
*/
|
|
getLinkPath: (type, width) => {
|
|
const arrowHead = 'v46.059c0 21.382 25.851 32.09 40.971 16.971 l86.059 -86.059c9.373-9.373 9.373-24.569 0-33.941l-86.059-86.059c-15.119-15.119-40.971-4.411-40.971 16.971v46.059'; //134.059
|
|
const arrowHeadR = 'v-46.059c0-21.382-25.851-32.09-40.971-16.971l-86.059 86.059c-9.373 9.373-9.373 24.568 0 33.941l86.059 86.059c15.119 15.119 40.971 4.411 40.971-16.971v-46.059'; //134.059
|
|
const lineEnd = 'c6.627 0 12 -5.373 12 -12v-56c0 -6.627 -5.373 -12 -12 -12'; // 12
|
|
const lineStart = 'c-6.627 0-12 5.373-12 12v56c0 6.627 5.373 12 12 12'; // 12
|
|
const dashedStep = `${lineStart}h56${lineEnd}`; // 80
|
|
switch (type) {
|
|
case 'none':
|
|
return null;
|
|
default:
|
|
return `M12 216${resources.links['arrow-head'].path}h${width * 512 - 146.059}${arrowHead}z`;
|
|
case 'line':
|
|
return `M12 216${lineStart}h${width * 512 - 24}${lineEnd}z`;
|
|
case 'double':
|
|
return `M134.059 216${arrowHeadR}h${width * 512 - 268.06}${arrowHead}z`;
|
|
case 'split-double':
|
|
return `M12 126${lineStart}h${width * 512 - 146.059}${arrowHead}M134.059 306${arrowHeadR}h${width * 512 - 146.059}${lineEnd}z`;
|
|
case 'dashed':
|
|
const n1 = Math.floor((width * 512 - 134.059) / (80 * 2.1));
|
|
const space1 = ((width * 512 - 134.059) - n1 * 56 - 24) / (n1);
|
|
return `M12 216${new Array(n1).fill(`${dashedStep}h${space1}`).join('')}v80${arrowHead}z`;
|
|
case 'dashed-line':
|
|
const n2 = Math.floor(width * 512 / (80 * 2.1));
|
|
const space2 = (width * 512 - n2 * 56 - 24) / (n2 - 1);
|
|
return `M12 216${new Array(n2).fill(dashedStep).join(`h${space2}`)}z`;
|
|
case 'dashed-double':
|
|
const n3 = Math.floor((width * 512 - 134.059 * 2) / (80 * 2.1));
|
|
const space3 = ((width * 512 - 134.059 * 2) - n3 * 56 - 24) / (n3 + 1);
|
|
return `M134.059 216${arrowHeadR}m${space3}-80${new Array(n3).fill(`${dashedStep}h${space3}`).join('')}v80${arrowHead}z`;
|
|
case 'dashed-split-double':
|
|
const n4 = Math.floor((width * 512 - 134.059) / (80 * 2.1));
|
|
const space4 = ((width * 512 - 134.059) - n4 * 56 - 24) / (n4);
|
|
return `M12 126${new Array(n4).fill(`${dashedStep}h${space4}`).join('')}v80${arrowHead}M134.059 306${arrowHeadR}m${space4} -80${new Array(n4).fill(`${dashedStep}`).join(`h${space4}`)}z`;
|
|
}
|
|
},
|
|
|
|
/**
|
|
* Get the width and height of the graph of nodes
|
|
* @param {Object<string,Node2>} nodes
|
|
* @returns {{w: number, h: number}}
|
|
*/
|
|
getBounds: (nodes) => {
|
|
const list = Object.values(nodes);
|
|
if (list.length === 0)
|
|
return {w: 0, h: 0}; //empty
|
|
let maxX = 0;
|
|
let maxY = 0;
|
|
list.forEach(n => {
|
|
maxX = Math.max(n.x, maxX);
|
|
maxY = Math.max(n.y, maxY);
|
|
});
|
|
return {w: maxX + 1, h: maxY + 1};
|
|
},
|
|
|
|
/**
|
|
* @param {{g:Object[]}} data
|
|
* @param {Object<string,Node2>} nodes
|
|
*/
|
|
renderNodes: (data, nodes) => {
|
|
Object.values(nodes).forEach(node => {
|
|
const icon = self.getIcon(node.icon);
|
|
if (icon) {
|
|
const scale = (node['scale'] || options['icons']['scale']) * DEFAULT_SCALE;
|
|
const group = {
|
|
'_attributes': {
|
|
'transform': `translate(${(node.x + 0.5) * options['h-spacing']} ${node.y + 0.5})`,
|
|
},
|
|
'g': {
|
|
'_attributes': {
|
|
'transform': `scale(${scale / 512} ${scale / 512}) translate(${-icon.width / 2} ${-256})`,
|
|
'stroke': (node['color'] || options['icons']['color'] || undefined),
|
|
'fill': (node['color'] || options['icons']['color'] || undefined)
|
|
},
|
|
'path': {
|
|
'_attributes': {
|
|
'd': icon.path,
|
|
}
|
|
}
|
|
}
|
|
};
|
|
data['g'].push(group);
|
|
}
|
|
});
|
|
},
|
|
|
|
/**
|
|
* @param {{g:Object[]}} data
|
|
* @param {Object<string,Node2>} nodes
|
|
* @param {Link2[]} links
|
|
*/
|
|
renderLinks: (data, nodes, links) => {
|
|
links.forEach(link => {
|
|
const src = nodes[link.from];
|
|
const dst = nodes[link.to];
|
|
|
|
const posX = ((src.x + dst.x) / 2 + 0.5) * options['h-spacing'];
|
|
const posY = (src.y + dst.y) / 2 + 0.5;
|
|
|
|
const angle = Math.atan2(dst.y - src.y, (dst.x - src.x) * options['h-spacing']) * 180 / Math.PI;
|
|
const scale = (link['scale'] || options['links']['scale']) * DEFAULT_SCALE;
|
|
|
|
let size = link['size'] || options['links']['size'];
|
|
|
|
if (!size) {
|
|
let dx = Math.abs(dst.x - src.x) * options['h-spacing'];
|
|
if (dx > 0)
|
|
dx -= LINK_MARGIN * 2;
|
|
let dy = Math.abs(dst.y - src.y);
|
|
if (dy > 0)
|
|
dy -= LINK_MARGIN * 2;
|
|
|
|
size = Math.sqrt(Math.pow(dx, 2) + Math.pow(dy, 2)) / scale;
|
|
}
|
|
|
|
const path = self.getLinkPath(link.type, size);
|
|
|
|
if (!path)
|
|
return;
|
|
|
|
const group = {
|
|
'_attributes': {
|
|
'transform': `translate(${posX} ${posY}) rotate(${angle})`
|
|
},
|
|
'g': {
|
|
'_attributes': {
|
|
'transform': `scale(${scale / 512} ${scale / 512}) translate(${(-256 * size)} ${-256})`,
|
|
'stroke': (link['color'] || options['links']['color'] || undefined),
|
|
'fill': (link['color'] || options['links']['color'] || undefined)
|
|
},
|
|
'path': {
|
|
'_attributes': {
|
|
'd': path
|
|
}
|
|
}
|
|
}
|
|
};
|
|
data['g'].push(group);
|
|
});
|
|
},
|
|
|
|
/**
|
|
* Convert xml-js data into correct svg xml string
|
|
* @param {Object} data
|
|
* @param {{w:number, h:number}} bounds
|
|
* @returns {string}
|
|
*/
|
|
toXML: (data, bounds) => {
|
|
const xml = {
|
|
'svg': {
|
|
'_attributes': {
|
|
'xmlns': 'http://www.w3.org/2000/svg',
|
|
'viewBox': `0 0 ${bounds.w * options['h-spacing']} ${bounds.h}`,
|
|
'width': bounds.w * options['h-spacing'] * options['scale'] / DEFAULT_SCALE,
|
|
'height': bounds.h * options['scale'] / DEFAULT_SCALE,
|
|
'stroke': options['color'],
|
|
'fill': options['color']
|
|
}
|
|
}
|
|
};
|
|
Object.keys(data).forEach(key => {
|
|
xml['svg'][key] = data[key];
|
|
});
|
|
return convert.js2xml(xml, {
|
|
compact: true,
|
|
spaces: options['beautify'] ? '\t' : 0
|
|
});
|
|
},
|
|
|
|
/**
|
|
* @param {Object<string,Node2>} nodes
|
|
* @param {Link2[]} links
|
|
*/
|
|
compute: (nodes, links) => {
|
|
|
|
Object.keys(nodes).forEach(key => {
|
|
const res = utils.isValid(nodes[key], NODE_DEF);
|
|
if (res)
|
|
throw `node '${key}' is invalid at key ${res}`;
|
|
});
|
|
|
|
links.forEach((link, i) => {
|
|
const res = utils.isValid(link, LINK_DEF);
|
|
if (res)
|
|
throw `link ${i} (${link.from}->${link.to}) is invalid at key ${res}`;
|
|
});
|
|
|
|
const bounds = self.getBounds(nodes);
|
|
|
|
const data = {'g': []};
|
|
|
|
self.renderNodes(data, nodes);
|
|
self.renderLinks(data, nodes, links);
|
|
|
|
return self.toXML(data, bounds);
|
|
}
|
|
};
|
|
|
|
return self;
|
|
}; |