diff options
Diffstat (limited to 'steam/package.go')
| -rw-r--r-- | steam/package.go | 11 |
1 files changed, 9 insertions, 2 deletions
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 { |