From 88043fd0424be0ac139d5151f39f217384728a9f Mon Sep 17 00:00:00 2001
From: Trang Doan <doantranghp2000@gmail.com>
Date: Mon, 21 Jul 2025 15:02:25 -0400
Subject: [PATCH 1/2] add color setting

---
 .../src/components/DiscourseContextView.tsx   |  8 +++-
 .../obsidian/src/components/NodeTypeModal.tsx |  9 ++++-
 .../src/components/NodeTypeSettings.tsx       | 40 +++++++++++++++++--
 3 files changed, 52 insertions(+), 5 deletions(-)

diff --git a/apps/obsidian/src/components/DiscourseContextView.tsx b/apps/obsidian/src/components/DiscourseContextView.tsx
index faa497dcd..9407f79ca 100644
--- a/apps/obsidian/src/components/DiscourseContextView.tsx
+++ b/apps/obsidian/src/components/DiscourseContextView.tsx
@@ -49,7 +49,13 @@ const DiscourseContext = ({ activeFile }: DiscourseContextProps) => {
     return (
       <>
         <div className="mb-6">
-          <div className="text-md mb-2 font-bold">
+          <div className="text-md mb-2 flex items-center gap-2 font-bold">
+            {nodeType.color && (
+              <div
+                className="h-4 w-4 rounded-full"
+                style={{ backgroundColor: nodeType.color }}
+              />
+            )}
             {nodeType.name || "Unnamed Node Type"}
           </div>
 
diff --git a/apps/obsidian/src/components/NodeTypeModal.tsx b/apps/obsidian/src/components/NodeTypeModal.tsx
index 217167bff..ccea86d53 100644
--- a/apps/obsidian/src/components/NodeTypeModal.tsx
+++ b/apps/obsidian/src/components/NodeTypeModal.tsx
@@ -24,7 +24,14 @@ export class NodeTypeModal extends SuggestModal<DiscourseNode> {
   }
 
   renderSuggestion(nodeType: DiscourseNode, el: HTMLElement) {
-    el.createEl("div", { text: nodeType.name });
+    const container = el.createDiv({ cls: "flex items-center gap-2" });
+    if (nodeType.color) {
+      container.createDiv({
+        cls: "h-4 w-4 rounded-full",
+        attr: { style: `background-color: ${nodeType.color};` },
+      });
+    }
+    container.createDiv({ text: nodeType.name });
   }
 
   async onChooseSuggestion(nodeType: DiscourseNode) {
diff --git a/apps/obsidian/src/components/NodeTypeSettings.tsx b/apps/obsidian/src/components/NodeTypeSettings.tsx
index 65a3e5f75..75d147781 100644
--- a/apps/obsidian/src/components/NodeTypeSettings.tsx
+++ b/apps/obsidian/src/components/NodeTypeSettings.tsx
@@ -7,14 +7,14 @@ import { DiscourseNode } from "~/types";
 import { ConfirmationModal } from "./ConfirmationModal";
 import { getTemplateFiles, getTemplatePluginInfo } from "~/utils/templates";
 
-type EditableFieldKey = keyof Omit<DiscourseNode, "id" | "shortcut" | "color">;
+type EditableFieldKey = keyof Omit<DiscourseNode, "id" | "shortcut">;
 
 type BaseFieldConfig = {
   key: EditableFieldKey;
   label: string;
   description: string;
   required?: boolean;
-  type: "text" | "select";
+  type: "text" | "select" | "color";
   placeholder?: string;
   validate?: (
     value: string,
@@ -68,6 +68,13 @@ const FIELD_CONFIGS: Record<EditableFieldKey, BaseFieldConfig> = {
     type: "select",
     required: false,
   },
+  color: {
+    key: "color",
+    label: "Color",
+    description: "The color to use for this node type",
+    type: "color",
+    required: false,
+  },
 };
 
 const FIELD_CONFIG_ARRAY = Object.values(FIELD_CONFIGS);
@@ -92,6 +99,23 @@ const TextField = ({
   />
 );
 
+const ColorField = ({
+  value,
+  error,
+  onChange,
+}: {
+  value: string;
+  error?: string;
+  onChange: (value: string) => void;
+}) => (
+  <input
+    type="color"
+    value={value || "#000000"}
+    onChange={(e) => onChange(e.target.value)}
+    className={`h-8 w-20 ${error ? "input-error" : ""}`}
+  />
+);
+
 const TemplateField = ({
   value,
   error,
@@ -367,6 +391,8 @@ const NodeTypeSettings = () => {
             templateConfig={templateConfig}
             templateFiles={templateFiles}
           />
+        ) : fieldConfig.type === "color" ? (
+          <ColorField value={value} error={error} onChange={handleChange} />
         ) : (
           <TextField
             fieldConfig={fieldConfig}
@@ -391,7 +417,15 @@ const NodeTypeSettings = () => {
           onClick={() => startEditing(index)}
         >
           <div className="flex items-center justify-between">
-            <span className="font-medium">{nodeType.name}</span>
+            <div className="flex items-center gap-2">
+            {nodeType.color && (
+              <div
+                className="h-4 w-4 rounded-full"
+                style={{ backgroundColor: nodeType.color }}
+              />
+            )}
+            <span>{nodeType.name}</span>
+          </div>
             <div className="flex gap-2">
               <button
                 className="icon-button"

From 1612427e42b050ee00d7ec685e3f71dbb611000f Mon Sep 17 00:00:00 2001
From: Trang Doan <doantranghp2000@gmail.com>
Date: Tue, 22 Jul 2025 15:07:55 -0400
Subject: [PATCH 2/2] add default color

---
 apps/obsidian/src/constants.ts | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/apps/obsidian/src/constants.ts b/apps/obsidian/src/constants.ts
index 811470ee0..73258322e 100644
--- a/apps/obsidian/src/constants.ts
+++ b/apps/obsidian/src/constants.ts
@@ -6,16 +6,19 @@ export const DEFAULT_NODE_TYPES: Record<string, DiscourseNode> = {
     id: generateUid("node"),
     name: "Question",
     format: "QUE - {content}",
+    color: "#99890e",
   },
   Claim: {
     id: generateUid("node"),
     name: "Claim",
     format: "CLM - {content}",
+    color: "#7DA13E",
   },
   Evidence: {
     id: generateUid("node"),
     name: "Evidence",
     format: "EVD - {content}",
+    color: "#DB134A",
   },
 };
 export const DEFAULT_RELATION_TYPES: Record<string, DiscourseRelationType> = {