instead of iteratively assigning colors to categories, make it consis…

…tent

.gitignore

@@ -23,3 +23,4 @@ src/sw.mjs
 gen/*
 # ignore elm.json for it's generated at build time from elm.json.base via generate.js
 elm.json
+.idea

package.json

@@ -1,7 +1,7 @@
 {
   "name": "GraphSense Pathfinder",
   "author": "Iknaio Cryptoasset Analytics",
-  "description": "Web application for the GraphSense Cryptoasset Analyitcs Platform",
+  "description": "Web application for the GraphSense Cryptoasset Analytics Platform",
   "version": "1.1.2",
   "license": "MIT",
   "scripts": {

src/Config/Update.elm

@@ -1,13 +1,15 @@
 module Config.Update exposing (Config)
 
 import Color exposing (Color)
+import Dict exposing (Dict)
 import Model.Graph.Coords exposing (BBox)
 import Model.Locale as Locale
 
 
 type alias Config =
     { defaultColor : Color
     , colorScheme : List Color
+    , categoryToColorIndex : Dict String Int
     , highlightsColorScheme : List Color
     , locale : Locale.Model
     , size : Maybe BBox -- position and size of the main pane

src/Main.elm

@@ -37,6 +37,7 @@ main =
         uc =
             { defaultColor = config.theme.graph.defaultColor
             , colorScheme = config.theme.graph.colorScheme
+            , categoryToColorIndex = config.theme.graph.categoryToColorIndex
             , highlightsColorScheme = config.theme.graph.highlightsColorScheme
             , locale = Locale.init default |> first
             , size = Nothing

src/Theme/Graph.elm

@@ -2,6 +2,7 @@ module Theme.Graph exposing (Graph, default)
 
 import Color exposing (Color)
 import Css exposing (Style)
+import Dict exposing (Dict)
 import Model.Graph exposing (NodeType)
 import Model.Graph.Tool as Tool
 
@@ -41,6 +42,7 @@ type alias Graph =
     , navbarRight : List Style
     , tool : Bool -> Tool.Status -> List Style
     , colorScheme : List Color
+    , categoryToColorIndex : Dict String Int
     , highlightsColorScheme : List Color
     , lightnessFactor :
         Bool
@@ -114,6 +116,7 @@ default =
     , navbarRight = []
     , tool = \_ _ -> []
     , colorScheme = []
+    , categoryToColorIndex = Dict.empty
     , highlightsColorScheme = []
     , lightnessFactor =
         \_ ->

src/Update/Graph/Color.elm

@@ -23,3 +23,25 @@ update { defaultColor, colorScheme } colors category =
                     colors
             )
         |> Maybe.withDefault colors
+
+
+
+-- Additionally, this one receives a categoryToColorIndex dictionary
+-- this way, we can update the colors of the graph based on the category
+-- instead of iterating over all the colors, also it doesnt need the colors dict
+
+
+updateWithIndex : Update.Config -> Dict String Color -> Maybe String -> Dict String Color
+updateWithIndex { defaultColor, colorScheme, categoryToColorIndex } colors category =
+    case category of
+        Just cat ->
+            let
+                color =
+                    Dict.get cat categoryToColorIndex
+                        |> Maybe.andThen (\index -> List.Extra.getAt index colorScheme)
+                        |> Maybe.withDefault defaultColor
+            in
+            Dict.insert cat color colors
+
+        Nothing ->
+            colors

src/Update/Graph/Entity.elm

@@ -75,7 +75,7 @@ addAddressToEntity plugins uc colors address entity =
                 Address.init plugins entity address
 
             newColors =
-                Color.update uc colors newAddress.category
+                Color.updateWithIndex uc colors newAddress.category
         in
         { updatedEntity =
             { entity

src/Update/Graph/Layer.elm

@@ -339,7 +339,7 @@ addEntityHere plugins uc position entity { layer, colors, new, repositioned } =
         }
     , new = Set.insert newEntity.id new
     , repositioned = Set.union newRepositioned repositioned
-    , colors = Color.update uc colors newEntity.category
+    , colors = Color.updateWithIndex uc colors newEntity.category
     }
 
 

tests/Graph/Update/TestLayer.elm

@@ -414,6 +414,7 @@ config : Update.Config
 config =
     { defaultColor = Color.grey
     , colorScheme = [ Color.red, Color.blue, Color.green ]
+    , categoryToColorIndex = Dict.fromList [("exchange", 0)]
     }
 
 

tests/Setup.elm

@@ -27,6 +27,7 @@ start locale =
         uc =
             { defaultColor = Theme.default.graph.defaultColor
             , colorScheme = Theme.default.graph.colorScheme
+            , categoryToColorIndex = Theme.default.graph.categoryToColorIndex
             }
 
         plugins =

themes/Iknaio.elm

@@ -1,5 +1,6 @@
 module Iknaio exposing (theme)
 
+import Dict exposing (Dict)
 import Color exposing (rgb255)
 import Css exposing (..)
 import Css.Transitions
@@ -667,6 +668,35 @@ theme =
                     , rgb255 184 176 172
                     , rgb255 87 120 164
                     ]
+                |> s_categoryToColorIndex
+                    (Dict.fromList
+                        [ ( "exchange", 0 )
+                        , ( "coinjoin", 3 )
+                        , ( "perpetrator", 1)
+                        , ( "defi", 2)
+                        , ( "miner", 4)
+                        , ( "payment_processor", 0)
+                        , ( "user", 5)
+                        , ( "gambling", 6)
+                        , ( "defi_lending", 7)
+                        , ( "market", 8)
+                        , ( "mixing_service", 9)
+                        , ( "defi_dex", 9)
+                        , ( "donation", 9)
+                        , ( "service", 9)
+                        , ( "wallet_service", 9)
+                        , ( "hosting", 9)
+                        , ( "shop", 9)
+                        , ( "entity", 9)
+                        , ( "organization", 9)
+                        , ( "vpn", 9)
+                        , ( "faucet", 9)
+                        , ( "defi_bridge", 9)
+                        , ( "ico_wallet", 9)
+                        , ( "atm", 9)
+                        , ( "mining_service", 9)
+                        ]
+                    )
                 |> (\graph ->
                         s_highlightsColorScheme
                             (graph.colorScheme