Introduce recolouring hooks

Author: tecosaur

docs/src/index.md

@@ -342,4 +342,5 @@ StyledStrings.parse(::Type{StyledStrings.SimpleColor}, ::String)
 StyledStrings.tryparse(::Type{StyledStrings.SimpleColor}, ::String)
 StyledStrings.merge(::StyledStrings.Face, ::StyledStrings.Face)
 StyledStrings.blend(::StyledStrings.SimpleColor, ::StyledStrings.SimpleColor, ::Real)
+StyledStrings.recolor
 ```

src/StyledStrings.jl

@@ -9,7 +9,7 @@ using Base.ScopedValues: ScopedValue, with, @with
 export AnnotatedString, AnnotatedChar, AnnotatedIOBuffer, annotations, annotate!, annotatedstring
 
 export @styled_str
-public Face, addface!, withfaces, styled, SimpleColor, blend
+public Face, addface!, withfaces, styled, SimpleColor, blend, recolor
 
 include("faces.jl")
 include("io.jl")

src/faces.jl

@@ -763,6 +763,46 @@ function Base.convert(::Type{Face}, spec::Dict{String,Any})
          end)
 end
 
+## Recolouring ##
+
+const recolor_hooks = Function[]
+const recolor_lock = ReentrantLock()
+
+"""
+    recolor(f::Function)
+
+Register a hook function `f` to be called whenever the colors change.
+
+Usually hooks will be called once after terminal colors have been
+determined. These hooks enable dynamic retheming, but are specifically *not* run when faces
+are changed. They sit in between the default faces and modifications layered on
+top with `loadface!` and user customisations.
+"""
+function recolor(f::Function)
+    @lock recolor_lock push!(recolor_hooks, f)
+    nothing
+end
+
+function setcolors!(color::Vector{Pair{Symbol, RGBTuple}})
+    @lock recolor_lock begin
+        for (name, rgb) in color
+            FACES.basecolors[name] = rgb
+        end
+        current = FACES.current[]
+        for (name, _) in FACES.modifications[]
+            default = get(FACES.default, name, nothing)
+            isnothing(default) && continue
+            current[name] = default
+        end
+        for hook in recolor_hooks
+            hook()
+        end
+        for (name, face) in FACES.modifications[]
+            current[name] = merge(current[name], face)
+        end
+    end
+end
+
 ## Color utils ##
 
 """