Loosen filename type constraints

Project.toml

@@ -1,6 +1,6 @@
 name = "FileIO"
 uuid = "5789e2e9-d7fb-5bc7-8068-2c6fae9b9549"
-version = "1.2.4"
+version = "1.3.0"
 
 [deps]
 Pkg = "44cfe95a-1eb2-52ea-b672-e2afdf69b78f"
@@ -9,8 +9,9 @@ Pkg = "44cfe95a-1eb2-52ea-b672-e2afdf69b78f"
 julia = "0.7, 1"
 
 [extras]
+FilePathsBase = "48062228-2e41-5def-b9a4-89aafe57970f"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 
 [targets]
-test = ["Test", "Random"]
+test = ["FilePathsBase", "Test", "Random"]

src/loadsave.jl

@@ -115,20 +115,17 @@ savestreaming
 # if a bare filename or IO stream are given, query for the format and dispatch
 # to the formatted handlers below
 for fn in (:load, :loadstreaming, :save, :savestreaming, :metadata)
-    @eval $fn(s::Union{AbstractString,IO}, args...; options...) =
-        $fn(query(s), args...; options...)
+    @eval $fn(file, args...; options...) = $fn(query(file), args...; options...)
 end
 
 # return a save function, so you can do `thing_to_save |> save("filename.ext")`
-function save(s::Union{AbstractString,IO}; options...)
-    data -> save(s, data; options...)
-end
+save(file; options...) = data -> save(file, data; options...)
 
 # Allow format to be overridden with first argument
-function save(df::Type{DataFormat{sym}}, f::AbstractString, data...; options...) where sym
+function save(df::Type{DataFormat{sym}}, filename, data...; options...) where sym
     libraries = applicable_savers(df)
     checked_import(libraries[1])
-    return Base.invokelatest(save, File(DataFormat{sym}, f), data...; options...)
+    return Base.invokelatest(save, File(DataFormat{sym}, filename), data...; options...)
 end
 
 function savestreaming(df::Type{DataFormat{sym}}, s::IO, data...; options...) where sym
@@ -143,10 +140,10 @@ function save(df::Type{DataFormat{sym}}, s::IO, data...; options...) where sym
     return Base.invokelatest(save, Stream(DataFormat{sym}, s), data...; options...)
 end
 
-function savestreaming(df::Type{DataFormat{sym}}, f::AbstractString, data...; options...) where sym
+function savestreaming(df::Type{DataFormat{sym}}, filename, data...; options...) where sym
     libraries = applicable_savers(df)
     checked_import(libraries[1])
-    return Base.invokelatest(savestreaming, File(DataFormat{sym}, f), data...; options...)
+    return Base.invokelatest(savestreaming, File(DataFormat{sym}, filename), data...; options...)
 end
 
 # do-syntax for streaming IO

src/query.jl

@@ -65,7 +65,7 @@ end
 `query(filename)` returns a `File` object with information about the
 format inferred from the file's extension and/or magic bytes.
 """
-function query(filename::AbstractString)
+function query(filename)
     _, ext = splitext(filename)
     if haskey(ext2sym, ext)
         sym = ext2sym[ext]
@@ -102,9 +102,7 @@ hasfunction(s::Tuple) = false #has magic
 `query(io, [filename])` returns a `Stream` object with information about the
 format inferred from the magic bytes.
 """
-query(io::IO, filename) = query(io, String(filename))
-
-function query(io::IO, filename::Union{Nothing, String} = nothing)
+function query(io::IO, filename = nothing)
     magic = Vector{UInt8}()
     pos = position(io)
     for p in magic_list

src/types.jl

@@ -26,7 +26,7 @@ DataFormat `fmt`.  For example, `File{fmtpng}(filename)` would indicate a PNG
 file.
 """
 struct File{F<:DataFormat} <: Formatted{F}
-    filename::String
+    filename
 end
 File(fmt::Type{DataFormat{sym}}, filename) where {sym} = File{fmt}(filename)
 
@@ -53,7 +53,7 @@ be used to improve error messages, etc.
 """
 struct Stream{F <: DataFormat, IOtype <: IO} <: Formatted{F}
     io::IOtype
-    filename::Union{String, Nothing}
+    filename
 end
 
 Stream(::Type{F}, io::IO) where {F<:DataFormat} = Stream{F,typeof(io)}(io, nothing)

test/loadsave.jl

@@ -22,30 +22,32 @@ try
     empty!(FileIO.sym2loader)
     empty!(FileIO.sym2saver)
     file_dir = joinpath(dirname(@__FILE__), "files")
-    @testset "Load" begin
+    file_path = Path(file_dir)
+
+    @testset "Load $(typeof(fp))" for fp in (file_dir, file_path)
 
         add_loader(format"PBMText", :TestLoadSave)
         add_loader(format"PBMBinary", :TestLoadSave)
         add_loader(format"HDF5", :TestLoadSave2)
         add_loader(format"JLD", :TestLoadSave)
         add_loader(format"GZIP", :TestLoadSave)
 
-        @test load(joinpath(file_dir,"file1.pbm")) == "PBMText"
-        @test load(joinpath(file_dir,"file2.pbm")) == "PBMBinary"
+        @test load(joinpath(fp,"file1.pbm")) == "PBMText"
+        @test load(joinpath(fp,"file2.pbm")) == "PBMBinary"
 
         # Regular HDF5 file with magic bytes starting at position 0
-        @test load(joinpath(file_dir,"file1.h5")) == "HDF5"
+        @test load(joinpath(fp,"file1.h5")) == "HDF5"
         # This one is actually a JLD file saved with an .h5 extension,
         # and the JLD magic bytes edited to prevent it from being recognized
         # as JLD.
         # JLD files are also HDF5 files, so this should be recognized as
         # HDF5. However, what makes this more interesting is that the
         # magic bytes start at position 512.
-        @test load(joinpath(file_dir,"file2.h5")) == "HDF5"
+        @test load(joinpath(fp,"file2.h5")) == "HDF5"
         # JLD file saved with .jld extension
-        @test load(joinpath(file_dir,"file.jld")) == "JLD"
+        @test load(joinpath(fp,"file.jld")) == "JLD"
         # GZIP file saved with .gz extension
-        @test load(joinpath(file_dir,"file.csv.gz")) == "GZIP"
+        @test load(joinpath(fp,"file.csv.gz")) == "GZIP"
         @test_throws Exception load("missing.fmt")
     end
 finally

test/query.jl

@@ -266,7 +266,10 @@ finally
     merge!(FileIO.sym2info, sym2info)
 end
 
-file_dir = joinpath(dirname(@__FILE__), "files")
+file_dir = joinpath(@__DIR__, "files")
+file_path = Path(file_dir)
+
+@testset "Querying with $(typeof(fp))" for fp in (file_dir, file_path)
     @testset "bedGraph" begin
         q = query(joinpath(file_dir, "file.bedgraph"))
         @test typeof(q) == File{format"bedGraph"}
@@ -342,6 +345,8 @@ end
             @test position(io)  == 0
         end
     end
+end
+
 @testset "Format with function for magic bytes" begin
     add_format(format"FUNCTION_FOR_MAGIC_BYTES", x -> 0x00, ".wav", [:WAV])
     del_format(format"FUNCTION_FOR_MAGIC_BYTES")

test/runtests.jl

@@ -1,6 +1,10 @@
 using FileIO
+using FilePathsBase
 using Test
 
+# Both FileIO and FilePathsBase export filename, but we only want the FileIO definition.
+using FileIO: filename
+
 struct MimeSaveTestType
 end
 

test/test_mimesave.jl

@@ -30,8 +30,8 @@ end
 
 data = MimeSaveTestType()
 
-output_filename = tempname()
-
+# Test with string and paths
+for output_filename in (tempname(), tmpname())
     for filetype in [".svg", ".pdf", ".eps", ".png", ".html"]
 
         try
@@ -46,5 +46,6 @@ for filetype in [".svg", ".pdf", ".eps", ".png", ".html"]
         end
 
     end
+end
 
 end