rework import / using when using weak exts

JuliaPlots · Feb 5, 2023 · 016edc7 · 016edc7
1 parent adc63db
commit 016edc7
Show file tree

Hide file tree

Showing 4 changed files with 27 additions and 15 deletions.
diff --git a/ext/FreeTypeExt.jl b/ext/FreeTypeExt.jl
@@ -4,11 +4,8 @@
 module FreeTypeExt
 
 import UnicodePlots
-@static if isdefined(Base, :get_extension)
-    using FreeType, FileIO
-else
-    using ..FreeType, ..FileIO
-end
+UnicodePlots.@ext_imp_use :using FreeType
+UnicodePlots.@ext_imp_use :import FileIO
 using StaticArrays
 using ColorTypes
 

diff --git a/ext/ImageInTerminalExt.jl b/ext/ImageInTerminalExt.jl
@@ -1,11 +1,7 @@
 module ImageInTerminalExt
 
 import UnicodePlots
-@static if isdefined(Base, :get_extension)
-    import ImageInTerminal
-else
-    import ..ImageInTerminal
-end
+UnicodePlots.@ext_imp_use :import ImageInTerminal
 using ColorTypes
 
 UnicodePlots.sixel_encode(args...; kw...) = ImageInTerminal.sixel_encode(args...; kw...)  # COV_EXCL_LINE

diff --git a/ext/UnitfulExt.jl b/ext/UnitfulExt.jl
@@ -2,11 +2,7 @@ module UnitfulExt
 
 import UnicodePlots
 import UnicodePlots: KEYWORDS, Plot, Canvas
-@static if isdefined(Base, :get_extension)
-    import Unitful: Quantity, RealOrRealQuantity, ustrip, unit
-else
-    import ..Unitful: Quantity, RealOrRealQuantity, ustrip, unit
-end
+UnicodePlots.@ext_imp_use :import Unitful Quantity RealOrRealQuantity ustrip unit
 
 function unit_str(x, fancy)
     io = IOContext(PipeBuffer(), :fancy_exponent => fancy)

diff --git a/src/common.jl b/src/common.jl
@@ -198,6 +198,29 @@ const BORDER_COLOR = Ref(:dark_gray)
 
 ############################################################################################
 # misc
+
+"""
+    @ext_imp_use :import Unitful Quantity RealOrRealQuantity
+
+Equivalent to the following, for `Requires` or weak deps:
+```
+@static if isdefined(Base, :get_extension)
+    import Unitful: Quantity, RealOrRealQuantity
+else
+    import ..Unitful: Quantity, RealOrRealQuantity
+end
+```
+"""
+macro ext_imp_use(imp_use::QuoteNode, mod::Symbol, args...)
+    dots = ntuple(_ -> :., isdefined(Base, :get_extension) ? 1 : 3)
+    ex = if length(args) > 0
+        Expr(:(:), Expr(dots..., mod), Expr.(:., args)...)
+    else
+        Expr(dots..., mod)
+    end
+    Expr(imp_use.value, ex) |> esc
+end
+
 const FSCALES = (identity = identity, ln = log, log2 = log2, log10 = log10)  # forward
 const ISCALES = (identity = identity, ln = exp, log2 = exp2, log10 = exp10)  # inverse
 const BASES = (identity = nothing, ln = "ℯ", log2 = "2", log10 = "10")