Added colors and function for writing colors. Currently the last color writeen overwrites all edges.

Author: natefrisch01

main.ts

@@ -1,4 +1,4 @@
-import { Plugin } from 'obsidian';
+import { Plugin, WorkspaceLeaf, Notice } from 'obsidian';
 import { getAPI } from 'obsidian-dataview';
 
 export default class UniqueMetadataKeysPlugin extends Plugin {
@@ -11,6 +11,11 @@ export default class UniqueMetadataKeysPlugin extends Plugin {
             name: 'Print Unique Metadata Keys',
             callback: () => this.printUniqueMetadataKeys()
         });
+        this.addCommand({
+            id: 'print-graph-leaf',
+            name: 'Print Graph Leaf',
+            callback: () => this.findGraphLeaf()
+        });
     }
 
     printUniqueMetadataKeys() {
@@ -28,10 +33,10 @@ export default class UniqueMetadataKeysPlugin extends Plugin {
 
         // Define a list of categorical colors as single integers
         const colors = [
-            9991159, 13948116, 16772608, 4280421, 16113345, 
-            9474192, 4617584, 15790320, 12320668, 16185078, 
-            32896, 15132410, 10040346, 16777128, 8388608, 
-            11184895, 8421376, 16761035, 255, 8421504
+            16343842, 7260435, 11226103, 8810003, 1997538, 
+            11796368, 9731429, 16177103, 15601550, 7601461, 
+            1066150, 6197085, 5122908, 1339852, 2975129, 
+            1364806, 3203221, 14122353, 7027020, 8280444
         ];
 
         // Map each unique key to a color
@@ -44,7 +49,10 @@ export default class UniqueMetadataKeysPlugin extends Plugin {
         this.keyColorMap.forEach((color, key) => console.log(`${key}: ${color}`));
     }
 
-    getColorForKey(key: string): number | undefined {
+    getColorForKey(key: string | null): number | undefined {
+        if (key === null) {
+            return 0;
+        }
         return this.keyColorMap.get(key);
     }
 
@@ -66,4 +74,34 @@ export default class UniqueMetadataKeysPlugin extends Plugin {
     isLink(value: any): boolean {
         return typeof value === 'object' && value.hasOwnProperty('path');
     }
+
+    findGraphLeaf(): WorkspaceLeaf | null{
+        let graphLeaves = this.app.workspace.getLeavesOfType('graph');
+		if (graphLeaves.length != 1) {
+			if (graphLeaves.length < 1) {
+				new Notice('No graph view open');
+			} else {
+				new Notice('More than one graph view open, please choose an active one');
+			}
+			return null;
+		}
+        const links = graphLeaves[0].view.renderer.links;
+        
+        for (let i = 0; i < 2; i++) {
+            const sourceId = links[i].source.id;
+            const targetId = links[i].target.id;
+            console.log(sourceId);
+            console.log(targetId);
+            const linkKey = this.getMetadataKeyForLink(sourceId, targetId);
+            const linkColor = this.getColorForKey(linkKey);
+            console.log(linkColor);
+            console.log(links[i].renderer);
+            links[i].line._tintRGB = linkColor;
+            links[i].renderer.colors.line.rgb = links[i].line._tintRGB;
+            
+        }
+
+        
+		return graphLeaves[0]
+    }
 }