Split default+modified faces into base/light/dark

Author: tecosaur

docs/src/internals.md

@@ -16,6 +16,7 @@ StyledStrings.Legacy.RENAMED_COLORS
 StyledStrings.Legacy.legacy_color
 StyledStrings.Legacy.load_env_colors!
 StyledStrings.ansi_4bit
+StyledStrings.setcolors!
 StyledStrings.face!
 StyledStrings.getface
 StyledStrings.load_customisations!

src/faces.jl

@@ -403,20 +403,23 @@ const FACES = let default = Dict{Symbol, Face}(
         :bright_magenta => (r = 0xbf, g = 0x60, b = 0xca),
         :bright_cyan    => (r = 0x26, g = 0xc6, b = 0xda),
         :bright_white   => (r = 0xf6, g = 0xf5, b = 0xf4))
-    (; default, basecolors,
+    (themes = (base = default, light = Dict{Symbol, Face}(), dark = Dict{Symbol, Face}()),
+     modifications = (base = Dict{Symbol, Face}(), light = Dict{Symbol, Face}(), dark = Dict{Symbol, Face}()),
      current = ScopedValue(copy(default)),
-     modifications = ScopedValue(Dict{Symbol, Face}()),
+     basecolors = basecolors,
      lock = ReentrantLock())
 end
 
 ## Adding and resetting faces ##
 
 """
-    addface!(name::Symbol => default::Face)
+    addface!(name::Symbol => default::Face, theme::Symbol = :base)
 
 Create a new face by the name `name`. So long as no face already exists by this
-name, `default` is added to both `FACES``.default` and (a copy of) to
-`FACES`.`current`, with the current value returned.
+name, `default` is added to both `FACES.themes[theme]` and (a copy of) to
+`FACES.current`, with the current value returned.
+
+The `theme` should be either `:base`, `:light`, or `:dark`.
 
 Should the face `name` already exist, `nothing` is returned.
 
@@ -429,11 +432,12 @@ Face (sample)
      underline: true
 ```
 """
-function addface!((name, default)::Pair{Symbol, Face})
-    @lock FACES.lock if !haskey(FACES.default, name)
-        FACES.default[name] = default
-        FACES.current[][name] = if haskey(FACES.current[], name)
-            merge(copy(default), FACES.current[][name])
+function addface!((name, default)::Pair{Symbol, Face}, theme::Symbol = :base)
+    current = FACES.current[]
+    @lock FACES.lock if !haskey(FACES.themes[theme], name)
+        FACES.themes[theme][name] = default
+        current[name] = if haskey(current, name)
+            merge(copy(default), current[name])
         else
             copy(default)
         end
@@ -449,10 +453,12 @@ function resetfaces!()
     @lock FACES.lock begin
         current = FACES.current[]
         empty!(current)
-        for (key, val) in FACES.default
+        for (key, val) in FACES.themes.base
             current[key] = val
         end
-        empty!(FACES.modifications[])
+        if current === FACES.current.default # Only when top-level
+            map(empty!, values(FACES.modifications))
+        end
         current
     end
 end
@@ -466,13 +472,15 @@ If the face `name` does not exist, nothing is done and `nothing` returned.
 In the unlikely event that the face `name` does not have a default value,
 it is deleted, a warning message is printed, and `nothing` returned.
 """
-function resetfaces!(name::Symbol)
-    @lock FACES.lock if !haskey(FACES.current[], name)
-    elseif haskey(FACES.default, name)
-        delete!(FACES.modifications[], name)
-        FACES.current[][name] = copy(FACES.default[name])
+function resetfaces!(name::Symbol, theme::Symbol = :base)
+    current = FACES.current[]
+    @lock FACES.lock if !haskey(current, name) # Nothing to reset
+    elseif haskey(FACES.themes[theme], name)
+        current === FACES.current.default &&
+            delete!(FACES.modifications[theme], name)
+        current[name] = copy(FACES.themes[theme][name])
     else # This shouldn't happen
-        delete!(FACES.current[], name)
+        delete!(current, name)
         @warn """The face $name was reset, but it had no default value, and so has been deleted instead!,
                  This should not have happened, perhaps the face was added without using `addface!`?"""
     end
@@ -658,18 +666,17 @@ Face (sample)
     foreground: #ff0000
 ```
 """
-function loadface!((name, update)::Pair{Symbol, Face})
+function loadface!((name, update)::Pair{Symbol, Face}, theme::Symbol = :base)
     @lock FACES.lock begin
-        mface = get(FACES.modifications[], name, nothing)
-        if !isnothing(mface)
-            update = merge(mface, update)
-        end
-        FACES.modifications[][name] = update
-        cface = get(FACES.current[], name, nothing)
-        if !isnothing(cface)
-            update = merge(cface, update)
+        current = FACES.current[]
+        if FACES.current.default === current # Only save top-level modifications
+            mface = get(FACES.modifications[theme], name, nothing)
+            isnothing(mface) || (update = merge(mface, update))
+            FACES.modifications[theme][name] = update
         end
-        FACES.current[][name] = update
+        cface = get(current, name, nothing)
+        isnothing(cface) || (update = merge(cface, update))
+        current[name] = update
     end
 end
 
@@ -684,7 +691,9 @@ end
 
 For each face specified in `Dict`, load it to `FACES``.current`.
 """
-function loaduserfaces!(faces::Dict{String, Any}, prefix::Union{String, Nothing}=nothing)
+function loaduserfaces!(faces::Dict{String, Any}, prefix::Union{String, Nothing}=nothing, theme::Symbol = :base)
+    theme == :base && prefix ∈ map(String, setdiff(keys(FACES.themes), (:base,))) &&
+        return loaduserfaces!(faces, nothing, Symbol(prefix))
     for (name, spec) in faces
         fullname = if isnothing(prefix)
             name
@@ -694,7 +703,7 @@ function loaduserfaces!(faces::Dict{String, Any}, prefix::Union{String, Nothing}
         fspec = filter((_, v)::Pair -> !(v isa Dict), spec)
         fnest = filter((_, v)::Pair -> v isa Dict, spec)
         !isempty(fspec) &&
-            loadface!(Symbol(fullname) => convert(Face, fspec))
+            loadface!(Symbol(fullname) => convert(Face, fspec), theme)
         !isempty(fnest) &&
             loaduserfaces!(fnest, fullname)
     end
@@ -783,23 +792,60 @@ function recolor(f::Function)
     nothing
 end
 
+"""
+    setcolors!(color::Vector{Pair{Symbol, RGBTuple}})
+
+Update the known base colors with those in `color`, and recalculate current faces.
+
+`color` should be a complete list of known colours. If `:foreground` and
+`:background` are both specified, the faces in the light/dark theme will be
+loaded. Otherwise, only the base theme will be applied.
+"""
 function setcolors!(color::Vector{Pair{Symbol, RGBTuple}})
-    @lock recolor_lock begin
+    lock(recolor_lock)
+    lock(FACES.lock)
+    try
+        # Apply colors
+        fg, bg = nothing, nothing
         for (name, rgb) in color
             FACES.basecolors[name] = rgb
+            if name === :foreground
+                fg = rgb
+            elseif name === :background
+                bg = rgb
+            end
+        end
+        newtheme = if !isnothing(fg) && !isnothing(bg)
+            :unknown
+        else
+            ifelse(sum(fg) > sum(bg), :light, :dark)
         end
+        # Reset all themes to defaults
         current = FACES.current[]
-        for (name, _) in FACES.modifications[]
-            default = get(FACES.default, name, nothing)
+        for theme in keys(FACES.themes), (name, _) in FACES.modifications[theme]
+            default = get(FACES.themes.base, name, nothing)
             isnothing(default) && continue
             current[name] = default
         end
+        if newtheme ∈ keys(FACES.themes)
+            for (name, face) in FACES.themes[newtheme]
+                current[name] = merge(current[name], face)
+            end
+        end
+        # Run recolor hooks
         for hook in recolor_hooks
             hook()
         end
-        for (name, face) in FACES.modifications[]
-            current[name] = merge(current[name], face)
+        # Layer on modifications
+        for theme in keys(FACES.themes)
+            theme ∈ (:base, newtheme) || continue
+            for (name, face) in FACES.modifications[theme]
+                current[name] = merge(current[name], face)
+            end
         end
+    finally
+        unlock(FACES.lock)
+        unlock(recolor_lock)
     end
 end
 

src/io.jl

@@ -243,7 +243,7 @@ function _ansi_writer(string_writer::F, io::IO, s::Union{<:AnnotatedString, SubS
     # We need to make sure that the customisations are loaded
     # before we start outputting any styled content.
     load_customisations!()
-    default = FACES.default[:default]
+    default = FACES.themes.base[:default]
     if get(io, :color, false)::Bool
         buf = IOBuffer() # Avoid the overhead in repeatedly printing to `stdout`
         lastface::Face = default

test/runtests.jl

@@ -254,35 +254,35 @@ end
          strikethrough: false
                inverse: false\
         """
-    @test sprint(show, MIME("text/plain"), FACES.default[:red], context = :color => true) |> choppkg ==
+    @test sprint(show, MIME("text/plain"), FACES.themes.base[:red], context = :color => true) |> choppkg ==
         """
         Face (\e[31msample\e[39m)
             foreground: \e[31m■\e[39m red\
         """
-    @test sprint(show, FACES.default[:red]) |> choppkg ==
+    @test sprint(show, FACES.themes.base[:red]) |> choppkg ==
         "Face(foreground=SimpleColor(:red))"
-    @test sprint(show, MIME("text/plain"), FACES.default[:red], context = :compact => true) |> choppkg ==
+    @test sprint(show, MIME("text/plain"), FACES.themes.base[:red], context = :compact => true) |> choppkg ==
         "Face(foreground=SimpleColor(:red))"
-    @test sprint(show, MIME("text/plain"), FACES.default[:red], context = (:compact => true, :color => true)) |> choppkg ==
+    @test sprint(show, MIME("text/plain"), FACES.themes.base[:red], context = (:compact => true, :color => true)) |> choppkg ==
         "Face(\e[31msample\e[39m)"
-    @test sprint(show, MIME("text/plain"), FACES.default[:highlight], context = :compact => true) |> choppkg ==
+    @test sprint(show, MIME("text/plain"), FACES.themes.base[:highlight], context = :compact => true) |> choppkg ==
         "Face(inverse=true, inherit=[:emphasis])"
     with_terminfo(vt100) do # Not truecolor capable
-        @test sprint(show, MIME("text/plain"), FACES.default[:region], context = :color => true) |> choppkg ==
+        @test sprint(show, MIME("text/plain"), FACES.themes.base[:region], context = :color => true) |> choppkg ==
             """
             Face (\e[48;5;237msample\e[49m)
                 background: \e[38;5;237m■\e[39m #3a3a3a\
             """
     end
     with_terminfo(fancy_term) do # Truecolor capable
-        @test sprint(show, MIME("text/plain"), FACES.default[:region], context = :color => true) |> choppkg ==
+        @test sprint(show, MIME("text/plain"), FACES.themes.base[:region], context = :color => true) |> choppkg ==
             """
             Face (\e[48;2;58;58;58msample\e[49m)
                 background: \e[38;2;58;58;58m■\e[39m #3a3a3a\
             """
     end
     with_terminfo(vt100) do # Ensure `enter_reverse_mode` exists
-        @test sprint(show, MIME("text/plain"), FACES.default[:highlight], context = :color => true) |> choppkg ==
+        @test sprint(show, MIME("text/plain"), FACES.themes.base[:highlight], context = :color => true) |> choppkg ==
             """
             Face (\e[34m\e[7msample\e[39m\e[27m)
                    inverse: true