Merge pull request #14 from snyk/feat/to-legacy-tree

Convert to legacy format

Author: gjvis

src/legacy/index.ts

@@ -5,6 +5,7 @@ import { DepGraphBuilder } from '../core/builder';
 
 export {
   depTreeToGraph,
+  graphToDepTree,
   DepTree,
 };
 
@@ -17,6 +18,7 @@ interface DepTreeDep {
 }
 
 interface DepTree extends DepTreeDep {
+  packageFormatVersion?: string;
   targetOS?: {
     name: string;
     version: string;
@@ -152,6 +154,77 @@ async function shortenNodeIds(depGraph: types.DepGraphInternal): Promise<types.D
   return builder.build();
 }
 
+async function graphToDepTree(depGraphInterface: types.DepGraph, pkgType: string): Promise<DepTree> {
+  const depGraph = (depGraphInterface as types.DepGraphInternal);
+
+  // TODO: implement cycles support
+  if (depGraph.hasCycles()) {
+    throw new Error('Conversion to DepTree does not support cyclic graphs yet');
+  }
+
+  const depTree = await buildSubtree(depGraph, depGraph.rootNodeId);
+
+  depTree.packageFormatVersion = constructPackageFormatVersion(pkgType);
+
+  const targetOS = constructTargetOS(depGraph);
+  if (targetOS) {
+    depTree.targetOS = targetOS;
+  }
+
+  return depTree;
+}
+
+function constructPackageFormatVersion(pkgType: string): string {
+  if (pkgType === 'maven') {
+    pkgType = 'mvn';
+  }
+  return `${pkgType}:0.0.1`;
+}
+
+function constructTargetOS(depGraph: types.DepGraph): { name: string; version: string; } {
+  if (['apk', 'apt', 'deb', 'rpm'].indexOf(depGraph.pkgManager.name) === -1) {
+    // .targetOS is undefined unless its a linux pkgManager
+    return;
+  }
+
+  if (!depGraph.pkgManager.repositories
+    || !depGraph.pkgManager.repositories.length
+    || !depGraph.pkgManager.repositories[0].alias) {
+      throw new Error('Incomplete .pkgManager, could not create .targetOS');
+  }
+
+  const [name, version] = depGraph.pkgManager.repositories[0].alias.split(':');
+  return { name, version };
+}
+
+async function buildSubtree(depGraph: types.DepGraphInternal, nodeId: string): Promise<DepTree> {
+  const nodePkg = depGraph.getNodePkg(nodeId);
+  const depTree: DepTree = {};
+  depTree.name = nodePkg.name;
+  depTree.version = nodePkg.version;
+
+  const depInstanceIds = depGraph.getNodeDepsNodeIds(nodeId);
+  if (!depInstanceIds || depInstanceIds.length === 0) {
+    return depTree;
+  }
+
+  for (const depInstId of depInstanceIds) {
+    const subtree = await buildSubtree(depGraph, depInstId);
+    if (!subtree) {
+      continue;
+    }
+
+    if (!depTree.dependencies) {
+      depTree.dependencies = {};
+    }
+
+    depTree.dependencies[subtree.name] = subtree;
+  }
+
+  await spinTheEventLoop();
+  return depTree;
+}
+
 async function spinTheEventLoop() {
   return new Promise((resolve) => setImmediate(resolve));
 }

test/fixtures/cyclic-dep-graph.json

@@ -0,0 +1,77 @@
+{
+  "schemaVersion": "1.0.0",
+  "pkgManager": {
+    "name": "pip"
+  },
+  "pkgs": [
+    {
+      "id": "[email protected]",
+      "info": {
+        "name": "toor",
+        "version": "1.0.0"
+      }
+    },
+    {
+      "id": "foo@2",
+      "info": {
+        "name": "foo",
+        "version": "2"
+      }
+    },
+    {
+      "id": "bar@3",
+      "info": {
+        "name": "bar",
+        "version": "3"
+      }
+    },
+    {
+      "id": "baz@4",
+      "info": {
+        "name": "baz",
+        "version": "4"
+      }
+    }
+  ],
+  "graph": {
+    "rootNodeId": "toor",
+    "nodes": [
+      {
+        "nodeId": "toor",
+        "pkgId": "[email protected]",
+        "deps": [
+          {
+            "nodeId": "foo@2|x"
+          }
+        ]
+      },
+      {
+        "nodeId": "foo@2|x",
+        "pkgId": "foo@2",
+        "deps": [
+          {
+            "nodeId": "bar@3|x"
+          }
+        ]
+      },
+      {
+        "nodeId": "bar@3|x",
+        "pkgId": "bar@3",
+        "deps": [
+          {
+            "nodeId": "baz@4|x"
+          }
+        ]
+      },
+      {
+        "nodeId": "baz@4|x",
+        "pkgId": "baz@4",
+        "deps": [
+          {
+            "nodeId": "foo@2|x"
+          }
+        ]
+      }
+    ]
+  }
+}