Extract persistence code into own class

main.go

@@ -7,13 +7,14 @@ import (
 	"os/signal"
 	"syscall"
 
+	"github.com/mdiluz/rove/pkg/persistence"
 	"github.com/mdiluz/rove/pkg/server"
 	"github.com/mdiluz/rove/pkg/version"
 )
 
 var ver = flag.Bool("version", false, "Display version number")
 var port = flag.Int("port", 8080, "The port to host on")
-var data = flag.String("data", "/tmp/", "Directory to store persistant data")
+var data = flag.String("data", os.TempDir(), "Directory to store persistant data")
 
 func main() {
 	flag.Parse()
@@ -23,9 +24,12 @@ func main() {
 		os.Exit(0)
 	}
 
+	// Set the persistence path
+	persistence.SetPath(*data)
+
 	s := server.NewServer(
 		server.OptionPort(*port),
-		server.OptionPersistentData(*data))
+		server.OptionPersistentData())
 
 	fmt.Println("Initialising...")
 	if err := s.Initialise(); err != nil {