From e31c9168627c040317e5cc8566724f88910439ae Mon Sep 17 00:00:00 2001 From: Mitch Riedstra Date: Sat, 9 Jan 2021 00:49:16 -0500 Subject: Add some download stats, and a status endpoint. --- steam/package.go | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) (limited to 'steam') diff --git a/steam/package.go b/steam/package.go index 76914dc..7287e10 100644 --- a/steam/package.go +++ b/steam/package.go @@ -9,7 +9,7 @@ import ( ) // Package writes the package, returning bytes written and an error if any -func (g *Game) Package(wr io.Writer) error { +func (g *Game) Package(wr io.WriteCloser) error { if err := os.Chdir(g.LibraryPath); err != nil { return err } @@ -26,7 +26,14 @@ func (g *Game) Package(wr io.Writer) error { } } - return twriter.Flush() + err = twriter.Flush() + if err != nil { + return err + } + + err = twriter.Close() + + return wr.Close() } func (l *Library) Extract(r io.Reader) error { -- cgit v1.2.3