Desugaring of export/public

Author: c42f

JuliaLowering/src/desugaring.jl

@@ -3908,6 +3908,16 @@ function expand_import(ctx, ex)
     ]
 end
 
+# Expand `public` or `export`
+function expand_public(ctx, ex)
+    @ast ctx ex [K"call"
+        module_public::K"Value"
+        ctx.mod::K"Value"
+        (kind(ex) == K"export")::K"Bool"
+        (e.name_val::K"String" for e in children(ex))...
+    ]
+end
+
 #-------------------------------------------------------------------------------
 # Expand module definitions
 
@@ -4148,7 +4158,7 @@ function expand_forms_2(ctx::DesugaringContext, ex::SyntaxTree, docs=nothing)
     elseif k == K"import" || k == K"using"
         expand_import(ctx, ex)
     elseif k == K"export" || k == K"public"
-        TODO(ex)
+        expand_public(ctx, ex)
     elseif k == K"abstract" || k == K"primitive"
         expand_forms_2(ctx, expand_abstract_or_primitive_type(ctx, ex))
     elseif k == K"struct"

JuliaLowering/src/runtime.jl

@@ -193,6 +193,12 @@ function module_import(into_mod::Module, is_using::Bool,
     nothing
 end
 
+function module_public(mod::Module, is_exported::Bool, identifiers...)
+    for ident in identifiers
+        @ccall jl_module_public(mod::Module, Symbol(ident)::Symbol, is_exported::Cint)::Cvoid
+    end
+end
+
 # Return the current exception. In JuliaLowering we use this rather than the
 # special form `K"the_exception"` to reduces the number of special forms.
 Base.@assume_effects :removable :nothrow function current_exception()

JuliaLowering/test/import.jl

@@ -14,6 +14,17 @@ JuliaLowering.include_string(test_mod, """
 @test test_mod.st2 === JuliaLowering.SyntaxTree
 @test test_mod.parsestmt === JuliaSyntax.parsestmt
 
+JuliaLowering.include_string(test_mod, """
+x = 1
+y = 2
+export x
+public y
+""")
+@test Base.isexported(test_mod, :x)
+@test Base.ispublic(test_mod, :x)
+@test Base.ispublic(test_mod, :y)
+@test !Base.isexported(test_mod, :y)
+
 C = JuliaLowering.include_string(test_mod, """
 module C
     module D

JuliaLowering/test/import_ir.jl

@@ -45,3 +45,17 @@ function f()
 #   └─────────┘ ── this syntax is only allowed in top level code
 end
 
+########################################
+# Export
+export a, b, c
+#---------------------
+1   (call JuliaLowering.module_public TestMod true "a" "b" "c")
+2   (return %₁)
+
+########################################
+# Public
+public a, b, c
+#---------------------
+1   (call JuliaLowering.module_public TestMod false "a" "b" "c")
+2   (return %₁)
+