Add byte I/O support for FileBuffer.

Author: rofinn

Project.toml

@@ -1,7 +1,7 @@
 name = "FilePathsBase"
 uuid = "48062228-2e41-5def-b9a4-89aafe57970f"
 authors = ["Rory Finnegan"]
-version = "0.9.1"
+version = "0.9.2"
 
 [deps]
 Dates = "ade2ca70-3891-5945-98fb-dc099432e06a"

src/buffer.jl

@@ -61,10 +61,18 @@ function Base.read(buffer::FileBuffer, ::Type{String})
     read(buffer.io, String)
 end
 
+function Base.read(buffer::FileBuffer, ::Type{UInt8})
+    if buffer.io.size == 0
+        write(buffer.io, read(buffer.path))
+        seekstart(buffer)
+    end
+    read(buffer.io, UInt8)
+end
+
 #=
 NOTE: We need to define multiple methods because of ambiguity error with base IO methods.
 =#
-function Base.write(buffer::FileBuffer, x::Vector{UInt8})
+function Base.write(buffer::FileBuffer, x::Union{UInt8, Vector{UInt8}})
     iswritable(buffer) || throw(ArgumentError("write failed, FileBuffer is not writeable"))
     write(buffer.io, x)
 end

test/buffer.jl

@@ -14,6 +14,11 @@ using FilePathsBase: FileBuffer
             @test !eof(io)
             @test read(p, String) == read(io, String)
             @test eof(io)
+            seekstart(io)
+            for b in read(p)
+                @test read(io, UInt8) == b
+            end
+            @test eof(io)
         finally
             close(io)
         end