Conversion of VarName to/from string

Author: penelopeysm

src/AbstractPPL.jl

@@ -8,6 +8,7 @@ export VarName,
     subsumes,
     subsumedby,
     varname,
+    varname_from_str,
     vsym,
     @varname,
     @vsym

src/varname.jl

@@ -663,6 +663,8 @@ function drop_escape(expr::Expr)
     return Expr(expr.head, map(x -> drop_escape(x), expr.args)...)
 end
 
+varname_from_str(str::AbstractString) = eval(drop_escape(varname(Meta.parse(str))))
+
 function _parse_obj_optic(ex)
     obj, optics = _parse_obj_optics(ex)
     optic = Expr(:call, Accessors.opticcompose, optics...)

test/varname.jl

@@ -137,4 +137,34 @@ end
         @inferred get(c, @varname(b.a[1]))
         @inferred Accessors.set(c, @varname(b.a[1]), 10)
     end
+
+    @testset "roundtrip conversion to/from string" begin
+        # Static optics
+        vns = [
+            @varname(x),
+            @varname(x.a),
+            @varname(x[1]),
+            @varname(x[1:10]),
+            @varname(x[1, 2]),
+            @varname(x[1, 2:5]),
+            @varname(x[:]),
+            @varname(x.a[1]),
+            @varname(x.a[1:10]),
+            @varname(x[1].a),
+        ]
+        for vn in vns
+            @test varname_from_str(repr(vn)) == vn
+        end
+
+        # Post-concretisation
+        x = ones(10)
+        vn = @varname(x[begin:end])
+        @test varname_from_str(repr(vn)) == vn
+
+        # When forcing concretisation
+        vn = @varname(x[:], true)
+        @test_broken varname_from_str(repr(vn)) == vn
+        dump(varname_from_str(repr(vn)))
+        dump(vn)
+    end
 end