add missing deps, load disk usage of dl dir

Author: jpillora

engine/engine.go

@@ -27,6 +27,10 @@ func New() *Engine {
 	return &Engine{ts: map[string]*Torrent{}}
 }
 
+func (e *Engine) Config() Config {
+	return e.config
+}
+
 func (e *Engine) Configure(c Config) error {
 	//recieve config
 	if e.client != nil {

server/server.go

@@ -61,8 +61,8 @@ type Server struct {
 	}
 }
 
+// Run the server
 func (s *Server) Run(version string) error {
-
 	isTLS := s.CertPath != "" || s.KeyPath != "" //poor man's XOR
 	if isTLS && (s.CertPath == "" || s.KeyPath == "") {
 		return fmt.Errorf("You must provide both key and cert paths")
@@ -125,9 +125,14 @@ func (s *Server) Run(version string) error {
 			time.Sleep(1 * time.Second)
 		}
 	}()
-
 	//start collecting stats
-	go s.state.Stats.System.loadStatsEvery(5 * time.Second)
+	go func() {
+		for {
+			c := s.engine.Config()
+			s.state.Stats.System.loadStats(c.DownloadDirectory)
+			time.Sleep(5 * time.Second)
+		}
+	}()
 
 	host := s.Host
 	if host == "" {

server/server_stats.go

@@ -2,7 +2,6 @@ package server
 
 import (
 	"runtime"
-	"time"
 
 	velox "github.com/jpillora/velox/go"
 	"github.com/shirou/gopsutil/cpu"
@@ -24,14 +23,7 @@ type stats struct {
 	pusher      velox.Pusher
 }
 
-func (s *stats) loadStatsEvery(interval time.Duration) {
-	for {
-		s.loadStats()
-		time.Sleep(interval)
-	}
-}
-
-func (s *stats) loadStats() {
+func (s *stats) loadStats(diskDir string) {
 	//count cpu cycles between last count
 	if stats, err := cpu.CPUTimes(false); err == nil {
 		stat := stats[0]
@@ -51,7 +43,7 @@ func (s *stats) loadStats() {
 		s.lastCPUStat = &stat
 	}
 	//count disk usage
-	if stat, err := disk.DiskUsage("/"); err == nil {
+	if stat, err := disk.DiskUsage(diskDir); err == nil {
 		s.DiskUsed = int64(stat.Used)
 		s.DiskTotal = int64(stat.Total)
 	}