From 11e9dc447b7644495d532e59537f30e36b2a0b56 Mon Sep 17 00:00:00 2001
From: Alexander Plavin <alexander@plav.in>
Date: Thu, 5 Dec 2024 07:38:19 -0500
Subject: [PATCH] add AnnotatedChar support

---
 src/functionlenses.jl       |  8 +++++++-
 test/test_functionlenses.jl | 21 +++++++++++++++++++--
 2 files changed, 26 insertions(+), 3 deletions(-)

diff --git a/src/functionlenses.jl b/src/functionlenses.jl
index d35ff54d..58ea5a00 100644
--- a/src/functionlenses.jl
+++ b/src/functionlenses.jl
@@ -207,9 +207,15 @@ end
 
 if isdefined(Base, :AnnotatedString)
     # 1.11+
-    using Base: AnnotatedString, annotations
+    using Base: AnnotatedString, AnnotatedChar, annotations
+
     set(s::AbstractString, ::typeof(annotations), anns) = AnnotatedString(s, anns)
     set(s::AnnotatedString, ::typeof(annotations), anns) = AnnotatedString(s.string, anns)
     delete(s::AnnotatedString, ::typeof(annotations)) = s.string
     insert(s::AbstractString, ::typeof(annotations), anns) = AnnotatedString(s, anns)
+
+    set(s::AbstractChar, ::typeof(annotations), anns) = AnnotatedChar(s, anns)
+    set(s::AnnotatedChar, ::typeof(annotations), anns) = AnnotatedChar(s.char, anns)
+    delete(s::AnnotatedChar, ::typeof(annotations)) = s.char
+    insert(s::AbstractChar, ::typeof(annotations), anns) = AnnotatedChar(s, anns)
 end
diff --git a/test/test_functionlenses.jl b/test/test_functionlenses.jl
index cefc9bc5..f63733cb 100644
--- a/test/test_functionlenses.jl
+++ b/test/test_functionlenses.jl
@@ -399,7 +399,7 @@ end
 VERSION ≥ v"1.11" && @testset "AnnotatedStrings" begin
     using Base: AnnotatedChar, AnnotatedString, annotations
 
-    s = Base.AnnotatedString("good bad", [(region=1:4, label=:sentiment, value=+1), (region=6:8, label=:sentiment, value=-1)])
+    s = AnnotatedString("good bad", [(region=1:4, label=:sentiment, value=+1), (region=6:8, label=:sentiment, value=-1)])
     
     @test (@delete annotations(s))::String == "good bad"
     
@@ -407,7 +407,7 @@ VERSION ≥ v"1.11" && @testset "AnnotatedStrings" begin
     @test String(snew) == "good bad"
     @test annotations(snew) == [(region=1:4, label=:sentiment, value=+1)]
 
-    snew = (@set Base.annotations(s)[1].region = 2:6)
+    snew = (@set annotations(s)[1].region = 2:6)
     @test String(snew) == "good bad"
     @test annotations(snew) == [(region=2:6, label=:sentiment, value=+1), (region=6:8, label=:sentiment, value=-1)]
 
@@ -415,6 +415,23 @@ VERSION ≥ v"1.11" && @testset "AnnotatedStrings" begin
     test_getset_laws((@o annotations(_)[2].label), s, :abc, :def)
     test_getset_laws((@o annotations(_)[2].value), s, "sad", +2)
     test_insertdelete_laws((@o annotations(_)[2]), s, (region=2:2, label=:mylabel, value=+1))
+
+    
+    c = AnnotatedChar('x', [(label=:level, value="warning"), (label=:alphabet, value=1)])
+    
+    @test (@delete annotations(c))::Char == 'x'
+    
+    cnew = @delete annotations(c)[2]
+    @test Char(cnew) == 'x'
+    @test annotations(cnew) == [(label=:level, value="warning")]
+
+    cnew = (@set annotations(c)[1].label = :severity)
+    @test Char(cnew) == 'x'
+    @test annotations(cnew) == [(label=:severity, value="warning"), (label=:alphabet, value=1)]
+
+    test_getset_laws((@o annotations(_)[2].label), c, :abc, :def)
+    test_getset_laws((@o annotations(_)[1].value), c, 5, "bad")
+    test_insertdelete_laws((@o annotations(_)[2]), c, (label=:mylabel, value=+1))
 end
 
 @testset "custom binary function" begin