@stoplight/graphite/graph/graph.js

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
const tslib_1 = require("tslib");
const mobx_1 = require("mobx");
const types_1 = require("../notifier/types");
const reporter_1 = require("../reporter");
const types_2 = require("../types");
const dom_1 = require("./dom");
const nodes_1 = require("./nodes");
const utils_1 = require("./utils");
const nanoid = require('nanoid');
function createGraph(props) {
    return new Graph(props);
}
exports.createGraph = createGraph;
class Graph {
    constructor(props) {
        this.dom = {
            nodes: {},
            nodesByUri: {},
            nodesByType: {},
        };
        this.getNodeById = (id) => {
            return this.dom.nodes[id];
        };
        this.getNodeByUri = (uri) => {
            return this.dom.nodesByUri[uri];
        };
        this.getNodesByType = (type) => {
            return this.dom.nodesByType[type] || [];
        };
        this.processIndexedTree = (tree, parentId) => {
            for (const item of tree) {
                const node = this.indexNode({
                    category: nodes_1.NodeCategory.Source,
                    type: item.type,
                    path: item.path,
                    parentId,
                });
                if (item.children && item.children.length) {
                    this.processIndexedTree(item.children, node.id);
                }
            }
        };
        this.id = props.id;
        this._idGenerator = node => generateNodeId(node, props.idGenerator || exports.defaultIdGenerator, this);
        this.notifier = props.notifier;
    }
    get nodeValues() {
        return Object.values(this.dom.nodes);
    }
    get sourceNodes() {
        return this.nodeValues.filter(utils_1.isSourceNode);
    }
    get virtualNodes() {
        return this.nodeValues.filter(utils_1.isVirtualNode);
    }
    get rootNodes() {
        return this.sourceNodes.filter(utils_1.isRootNode);
    }
    applyPatch(patch) {
        const patchWithTrace = Object.assign(Object.assign({}, patch), { trace: Object.assign({ instanceId: this.id }, patch.trace) });
        const eventGroup = this.notifier.createEmitGroup();
        eventGroup.emit(types_1.GraphiteEvent.DidPatch, patchWithTrace);
        let result;
        try {
            result = dom_1.applyPatch(this.dom, patch, eventGroup);
        }
        finally {
            eventGroup.flush();
        }
        return result;
    }
    addNode(node, trace, indexOnly = false) {
        const existingByUri = this.getNodeByPathWithParent(node.path, node.parentId);
        if (existingByUri)
            return existingByUri;
        const id = node.id || this._idGenerator(node);
        const existingById = this.dom.nodes[id];
        if (existingById) {
            reporter_1.reporter.warn(`Warning: addNode() node with id ${id} already exists. It has uri '${existingById.uri}'.`);
        }
        this.applyPatch({
            operations: [
                {
                    op: dom_1.GraphOp.AddNode,
                    node: Object.assign(Object.assign({}, node), { id }),
                    indexOnly,
                },
            ],
            trace: trace || {
                instanceId: this.id,
            },
        });
        return this.dom.nodes[id];
    }
    indexNode(node, trace) {
        return this.addNode(node, trace, true);
    }
    setSourceNodeDiagnostics(id, source, diagnostics, trace) {
        this.applyPatch({
            operations: [
                {
                    op: dom_1.GraphOp.SetSourceNodeDiagnostics,
                    id,
                    source,
                    diagnostics,
                },
            ],
            trace: trace || {
                instanceId: this.id,
            },
        });
    }
    setSourceNodeProp(id, prop, value, trace) {
        this.applyPatch({
            operations: [
                {
                    op: dom_1.GraphOp.SetSourceNodeProp,
                    id,
                    prop,
                    value,
                },
            ],
            trace: trace || {
                instanceId: this.id,
            },
        });
    }
    reportError(nodeId, error, trace) {
        const pluginError = {
            data: error,
            message: error.message,
            nodeId,
            code: types_2.GraphiteErrorCode.Plugin,
            trace: trace || {
                instanceId: this.id,
            },
        };
        reporter_1.reporter.reportError(pluginError);
    }
    patchSourceNodeProp(id, prop, value, trace) {
        this.applyPatch({
            operations: [
                {
                    op: dom_1.GraphOp.PatchSourceNodeProp,
                    id,
                    prop,
                    value,
                },
            ],
            trace: trace || {
                instanceId: this.id,
            },
        });
    }
    moveNode(id, newParentId, newPath, trace = {}) {
        this.applyPatch({
            operations: [
                {
                    op: dom_1.GraphOp.MoveNode,
                    id,
                    newParentId,
                    newPath,
                },
            ],
            trace,
        });
    }
    removeNode(id, trace) {
        this.applyPatch({
            operations: [
                {
                    op: dom_1.GraphOp.RemoveNode,
                    id,
                },
            ],
            trace: trace || {
                instanceId: this.id,
            },
        });
    }
    printTree() {
        return utils_1.printTree(this.rootNodes);
    }
    dehydrate() {
        return dehydrate(this.dom);
    }
    dispose() {
    }
    getNodeByPathWithParent(path, parentId) {
        const parent = parentId && this.getNodeById(parentId);
        const uri = utils_1.combinePathAndUri(path, parent ? parent.uri : '');
        return this.getNodeByUri(uri);
    }
}
tslib_1.__decorate([
    mobx_1.computed,
    tslib_1.__metadata("design:type", Object),
    tslib_1.__metadata("design:paramtypes", [])
], Graph.prototype, "nodeValues", null);
tslib_1.__decorate([
    mobx_1.computed,
    tslib_1.__metadata("design:type", Object),
    tslib_1.__metadata("design:paramtypes", [])
], Graph.prototype, "sourceNodes", null);
tslib_1.__decorate([
    mobx_1.computed,
    tslib_1.__metadata("design:type", Object),
    tslib_1.__metadata("design:paramtypes", [])
], Graph.prototype, "virtualNodes", null);
tslib_1.__decorate([
    mobx_1.computed,
    tslib_1.__metadata("design:type", Object),
    tslib_1.__metadata("design:paramtypes", [])
], Graph.prototype, "rootNodes", null);
tslib_1.__decorate([
    mobx_1.action.bound,
    tslib_1.__metadata("design:type", Function),
    tslib_1.__metadata("design:paramtypes", [Object]),
    tslib_1.__metadata("design:returntype", void 0)
], Graph.prototype, "applyPatch", null);
tslib_1.__decorate([
    mobx_1.action.bound,
    tslib_1.__metadata("design:type", Function),
    tslib_1.__metadata("design:paramtypes", [Object, Object, Boolean]),
    tslib_1.__metadata("design:returntype", void 0)
], Graph.prototype, "addNode", null);
tslib_1.__decorate([
    mobx_1.action.bound,
    tslib_1.__metadata("design:type", Function),
    tslib_1.__metadata("design:paramtypes", [String, String, Object, Object]),
    tslib_1.__metadata("design:returntype", void 0)
], Graph.prototype, "setSourceNodeProp", null);
tslib_1.__decorate([
    mobx_1.action.bound,
    tslib_1.__metadata("design:type", Function),
    tslib_1.__metadata("design:paramtypes", [String, String, Array, Object]),
    tslib_1.__metadata("design:returntype", void 0)
], Graph.prototype, "patchSourceNodeProp", null);
tslib_1.__decorate([
    mobx_1.action.bound,
    tslib_1.__metadata("design:type", Function),
    tslib_1.__metadata("design:paramtypes", [String, Object, String, Object]),
    tslib_1.__metadata("design:returntype", void 0)
], Graph.prototype, "moveNode", null);
tslib_1.__decorate([
    mobx_1.action.bound,
    tslib_1.__metadata("design:type", Function),
    tslib_1.__metadata("design:paramtypes", [String, Object]),
    tslib_1.__metadata("design:returntype", void 0)
], Graph.prototype, "removeNode", null);
exports.addNodeTree = mobx_1.action((helpers, tree, parentId) => {
    for (const node of tree) {
        const n = helpers.addNode(Object.assign(Object.assign({}, node), { parentId }));
        if (node.children && node.children.length) {
            exports.addNodeTree(helpers, node.children, n.id);
        }
    }
});
exports.defaultIdGenerator = () => nanoid();
function generateNodeId(node, idGenerator, nodeFinder) {
    const parent = node.parentId ? nodeFinder.getNodeById(node.parentId) : void 0;
    return idGenerator(node, utils_1.combinePathAndUri(node.path, parent ? parent.uri : void 0));
}
exports.generateNodeId = generateNodeId;
function dehydrate(data) {
    const dehydrated = {
        nodes: {},
    };
    for (const id in data.nodes) {
        if (!{}.hasOwnProperty.call(data.nodes, id))
            continue;
        dehydrated.nodes[id] = data.nodes[id].dehydrate();
    }
    return dehydrated;
}
//# sourceMappingURL=graph.js.map