Support color schemes

Author: tmr232

scripts/render-function.ts

@@ -5,8 +5,10 @@ import type Parser from "web-tree-sitter";
 import type { SyntaxNode } from "web-tree-sitter";
 import { type Language, supportedLanguages } from "../src/control-flow/cfg.ts";
 import {
-  deserializeColorList, getDarkColorList, getLightColorList,
-  listToScheme
+  deserializeColorList,
+  getDarkColorList,
+  getLightColorList,
+  listToScheme,
 } from "../src/control-flow/colors.ts";
 import { graphToDot } from "../src/control-flow/render.ts";
 import { getLanguage, iterFunctions } from "../src/file-parsing/bun.ts";
@@ -36,14 +38,12 @@ function writeError(message: string): void {
   Bun.write(Bun.stderr, `${message}\n`);
 }
 
-async function getColorScheme(
-  colors?:string
-) {
+export async function getColorScheme(colors?: string) {
   if (!colors || colors === "dark") {
-    return listToScheme(getDarkColorList())
+    return listToScheme(getDarkColorList());
   }
   if (colors === "light") {
-    return listToScheme(getLightColorList())
+    return listToScheme(getLightColorList());
   }
   return colors
     ? listToScheme(deserializeColorList(await Bun.file(colors).text()))
@@ -135,7 +135,7 @@ async function main() {
   const graphviz = await Graphviz.load();
   const cfg = buildCFG(func, language);
 
-  const colorScheme = await getColorScheme(values.colors)
+  const colorScheme = await getColorScheme(values.colors);
 
   const svg = graphviz.dot(graphToDot(cfg, false, undefined, colorScheme));
 

scripts/render-graph.ts

@@ -7,14 +7,21 @@ import type {
   GraphNode,
 } from "../src/control-flow/cfg-defs.ts";
 import { graphToDot } from "../src/control-flow/render.ts";
+import { getColorScheme } from "./render-function.ts";
 
 async function main() {
   const {
+    values,
     positionals: [_runtime, _this, gist_url],
   } = parseArgs({
     args: Bun.argv,
     strict: true,
     allowPositionals: true,
+    options: {
+      colors: {
+        type: "string",
+      },
+    },
   });
 
   if (!gist_url) {
@@ -39,9 +46,10 @@ async function main() {
     throw new Error("No entry found");
   }
   const cfg: CFG = { graph, entry, offsetToNode: [] };
-  const dot = graphToDot(cfg);
+  const colorScheme = await getColorScheme(values.colors);
+
   const graphviz = await Graphviz.load();
-  const svg = graphviz.dot(dot);
+  const svg = graphviz.dot(graphToDot(cfg, false, undefined, colorScheme));
   console.log(svg);
   // console.log(dot);
 }

scripts/scan-codebase.ts

@@ -43,7 +43,7 @@ async function* iterFunctionInfo(
   filenames: IterableIterator<string>,
 ): AsyncIterableIterator<{
   node_count: number;
-  start_position: { row:number, column:number };
+  start_position: { row: number; column: number };
   funcdef: string;
   filename: string;
 }> {