feat: add command-line flag for customizable port

Author: ksamirdev

README.md

@@ -26,10 +26,10 @@ Head to [Releases](https://github.com/ksamirdev/schedy/releases) and grab the la
 ### 2. Run
 
 ```bash
-./schedy
+./schedy --port 8081
 ```
 
-Schedy will listen on `:8080` by default.
+Schedy will listen on the port you specify with `--port` (default: `8080`).
 
 ---
 

cmd/schedy/main.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"context"
+	"flag"
 	"log"
 	"net/http"
 	"os"
@@ -15,6 +16,9 @@ import (
 )
 
 func main() {
+	port := flag.String("port", "8080", "port to listen on")
+	flag.Parse()
+
 	store, err := scheduler.NewBadgerStore("data")
 	if err != nil {
 		log.Fatal(err)
@@ -26,15 +30,16 @@ func main() {
 	mux := http.NewServeMux()
 	mux.HandleFunc("/tasks", handler.CreateTask)
 
-	srv := &http.Server{Addr: ":8080", Handler: mux}
+	addr := ":" + *port
+	srv := &http.Server{Addr: addr, Handler: mux}
 
 	ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt)
 	defer stop()
 
 	go r.Start(ctx)
 
 	go func() {
-		log.Println("Listening on :8080")
+		log.Printf("Listening on %s", addr)
 		if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
 			log.Fatal(err)
 		}