feat: kill server after n downloads

cmd/lash/main.go

@@ -14,6 +14,7 @@ import (
 var versionFlag = flag.Bool("version", false, "Print out version and exit.")
 
 var port = flag.Int("p", 1337, "Set the port for LASH exchanges.")
+var numberOfShares = flag.Int("n", -1, "Number of shares to allow before killing the server.")
 
 func main() {
 	flag.Parse()
@@ -23,14 +24,22 @@ func main() {
 		return
 	}
 
-	srv := app.New()
+	ctx := lash.LashContext{
+		N: *numberOfShares,
+	}
+
+	srv := app.New(&ctx)
 
 	// TODO(vxc): Make this more portable
 	errx.Log("Your share link is %s", share.GenerateShareLink(*port))
 	errx.Log("Your token is \033[1;92m%s\033[0m", lash.Token)
 
 	errx.Log("starting server at http://0.0.0.0:%d", *port)
-	if err := http.ListenAndServe(fmt.Sprintf(":%d", *port), srv); err != nil {
+	server := http.Server{
+		Addr:    fmt.Sprintf(":%d", *port),
+		Handler: srv,
+	}
+	if err := server.ListenAndServe(); err != nil {
 		errx.FatalPerror(err)
 	}
 }