Use weakdeps on Julia v1.9

Author: oschulz

Project.toml

@@ -9,6 +9,18 @@ GPUArraysCore = "46192b85-c4d5-4398-a991-12ede77f4527"
 StaticArraysCore = "1e83bf80-4336-4d27-bf5d-d5a4f845583c"
 Tables = "bd369af6-aec1-5ad0-b16a-f7cc5008161c"
 
+[weakdeps]
+DataAPI = "9a962f9c-6df0-11e9-0e5d-c546b8b5ee8a"
+GPUArraysCore = "46192b85-c4d5-4398-a991-12ede77f4527"
+StaticArraysCore = "1e83bf80-4336-4d27-bf5d-d5a4f845583c"
+Tables = "bd369af6-aec1-5ad0-b16a-f7cc5008161c"
+
+[extensions]
+StructArraysDataAPIExt = "DataAPI"
+StructArraysGPUArraysCoreExt = "GPUArraysCore"
+StructArraysStaticArraysCoreExt = "StaticArraysCore"
+StructArraysTablesExt = "Tables"
+
 [compat]
 Adapt = "1, 2, 3"
 DataAPI = "1"

ext/StructArraysDataAPIExt.jl

@@ -0,0 +1,16 @@
+module StructArraysDataAPIExt
+
+using StructArrays
+using StructArrays: Tup
+
+# Implement refarray and refvalue to deal with pooled arrays and weakrefstrings effectively
+import DataAPI: refarray, refvalue
+using DataAPI: defaultarray
+
+refarray(s::StructArray) = StructArray(map(refarray, components(s)))
+
+function refvalue(s::StructArray{T}, v::Tup) where {T}
+    createinstance(T, map(refvalue, components(s), v)...)
+end
+
+end # module

ext/StructArraysGPUArraysCoreExt.jl

@@ -0,0 +1,17 @@
+module StructArraysGPUArraysCoreExt
+
+using StructArrays
+import GPUArraysCore
+
+# for GPU broadcast
+import GPUArraysCore
+function GPUArraysCore.backend(::Type{T}) where {T<:StructArray}
+    backends = map_params(GPUArraysCore.backend, array_types(T))
+    backend, others = backends[1], tail(backends)
+    isconsistent = mapfoldl(isequal(backend), &, others; init=true)
+    isconsistent || throw(ArgumentError("all component arrays must have the same GPU backend"))
+    return backend
+end
+always_struct_broadcast(::GPUArraysCore.AbstractGPUArrayStyle) = true
+
+end # module

src/staticarrays_support.jl renamed to ext/StructArraysStaticArraysCoreExt.jl

@@ -1,3 +1,10 @@
+module StructArraysStaticArraysCoreExt
+
+using StructArrays
+using StructArrays: StructArrayStyle
+
+using Base.Broadcast: Broadcasted
+
 using StaticArraysCore: StaticArray, FieldArray, tuple_prod
 
 """
@@ -66,3 +73,5 @@ end
         return map(Base.Fix2(getfield, i), x)
     end
 end
+
+end # module

src/tables.jl renamed to ext/StructArraysTablesExt.jl

@@ -1,3 +1,6 @@
+module StructArraysTablesExt
+
+using StructArrays
 import Tables
 
 Tables.isrowtable(::Type{<:StructArray}) = true
@@ -38,3 +41,5 @@ for (f, g) in zip((:append!, :prepend!), (:push!, :pushfirst!))
         end
     end
 end
+
+end # module

src/StructArrays.jl

@@ -12,32 +12,16 @@ include("utils.jl")
 include("collect.jl")
 include("sort.jl")
 include("lazy.jl")
-include("tables.jl")
-include("staticarrays_support.jl")
 
-# Implement refarray and refvalue to deal with pooled arrays and weakrefstrings effectively
-import DataAPI: refarray, refvalue
-using DataAPI: defaultarray
-
-refarray(s::StructArray) = StructArray(map(refarray, components(s)))
-
-function refvalue(s::StructArray{T}, v::Tup) where {T}
-    createinstance(T, map(refvalue, components(s), v)...)
+@static if !isdefined(Base, :get_extension)
+    include("../ext/StructArraysDataAPIExt.jl")
+    include("../ext/StructArraysGPUArraysCoreExt.jl")
+    include("../ext/StructArraysTablesExt.jl")
+    include("../ext/StructArraysStaticArraysCoreExt.jl")
 end
 
 # Use Adapt allows for automatic conversion of CPU to GPU StructArrays
 import Adapt
 Adapt.adapt_structure(to, s::StructArray) = replace_storage(x->Adapt.adapt(to, x), s)
 
-# for GPU broadcast
-import GPUArraysCore
-function GPUArraysCore.backend(::Type{T}) where {T<:StructArray}
-    backends = map_params(GPUArraysCore.backend, array_types(T))
-    backend, others = backends[1], tail(backends)
-    isconsistent = mapfoldl(isequal(backend), &, others; init=true)
-    isconsistent || throw(ArgumentError("all component arrays must have the same GPU backend"))
-    return backend
-end
-always_struct_broadcast(::GPUArraysCore.AbstractGPUArrayStyle) = true
-
 end # module