Merge pull request #10 from mrobinsn/add-size-to-status

Add Size to GetStatus
mrobinsn · May 5, 2019 · 9500743 · 9500743
2 parents 12696af + 7dbd6cd
commit 9500743
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/rtorrent/rtorrent.go b/rtorrent/rtorrent.go
@@ -32,6 +32,7 @@ type Status struct {
 	DownRate       int
 	UpRate         int
 	Ratio          float64
+	Size           int
 }
 
 // File represents a file in rTorrent
@@ -286,5 +287,11 @@ func (r *RTorrent) GetStatus(t Torrent) (Status, error) {
 		return s, errors.Wrap(err, "d.ratio XMLRPC call failed")
 	}
 	s.Ratio = float64(results.([]interface{})[0].(int)) / float64(1000)
+	// Size
+	results, err = r.xmlrpcClient.Call("d.size_bytes", t.Hash)
+	if err != nil {
+		return s, errors.Wrap(err, "d.size_bytes XMLRPC call failed")
+	}
+	s.Size = results.([]interface{})[0].(int)
 	return s, nil
 }
diff --git a/rtorrent/rtorrent_test.go b/rtorrent/rtorrent_test.go
@@ -145,6 +145,7 @@ func TestRTorrent(t *testing.T) {
 					require.False(t, status.Completed)
 					require.NotZero(t, status.CompletedBytes)
 					require.NotZero(t, status.DownRate)
+					require.NotZero(t, status.Size)
 					// require.NotZero(t, status.UpRate)
 					//require.NotZero(t, status.Ratio)
 				})