Base.setindex for NamedProperties

JuliaCollections · Sep 24, 2022 · d127a79 · d127a79
1 parent ee52a21
commit d127a79
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/src/PropertyDicts.jl b/src/PropertyDicts.jl
@@ -79,6 +79,16 @@ end
 
 const NamedProperties{syms,T<:Tuple,V} = PropertyDict{Symbol,V,NamedTuple{syms,T}}
 
+@inline function Base.setindex(npd::NamedProperties{syms}, v, key::Symbol) where {syms}
+    nt = getfield(npd, :d)
+    idx = Base.fieldindex(typeof(nt), key, false)
+    if idx === 0
+        return PropertyDict(NamedTuple{(syms..., key)}((values(nt)..., v)))
+    else
+        return PropertyDict(NamedTuple{syms}(ntuple(i -> idx === i ? v : getfield(nt, i), Val{nfields(syms)}())))
+    end
+end
+
 Base.IteratorSize(@nospecialize T::Type{<:PropertyDict}) = Base.IteratorSize(fieldtype(T, :d))
 Base.IteratorEltype(@nospecialize T::Type{<:PropertyDict}) = Base.IteratorEltype(eltype(T))
 

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -148,3 +148,9 @@ end
     @test @inferred(mergewith(combiner, a, b, c, PropertyDict())) ==
         PropertyDict((a = "1 and 1", b = "2 and 4 and 2", c = 3, d = 5))
 end
+
+@testset "setindex" begin
+    npd = Base.setindex(Base.setindex(PropertyDict(), 1, :x), 2, :y)
+    @test values(npd) == (1, 2)
+    @test keys(npd) == (:x, :y)
+end