fix colnames for FunctionSource, eg from QuackIO

Author: aplavin

Project.toml

@@ -1,7 +1,7 @@
 name = "SQLCollections"
 uuid = "3934435e-e6e6-4fea-a5c0-3a716143d619"
 authors = ["Alexander Plavin <alexander@plav.in>"]
-version = "0.1.4"
+version = "0.1.5"
 
 [deps]
 AccessorsExtra = "33016aad-b69d-45be-9359-82a41f556fd4"

src/SQLCollections.jl

@@ -30,7 +30,12 @@ SQLCollection(conn, tbl::Union{Symbol,AbstractString}) = SQLCollection(conn, Fro
 colnames(dbc::SQLCollection) = colnames(dbc.query; dbc.conn.catalog)
 colnames(q; catalog=nothing) = keys(q.label_map)
 colnames(q::FunSQL.SQLNode; kwargs...) = colnames(q[]; kwargs...)
-colnames(q::FunSQL.FromNode; catalog) = catalog[q.source::Symbol].columns |> keys
+colnames(q::FunSQL.FromNode; catalog) =
+	if q.source isa Symbol
+		catalog[q.source::Symbol].columns |> keys
+	elseif q.source isa FunSQL.FunctionSource
+		q.source.columns
+	end
 
 exists(dbc::SQLCollection) = 
     try