Add some download stats, and a status endpoint.

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 {