From 66949765d19505ee713b37790d9707cc9e7bf802 Mon Sep 17 00:00:00 2001 From: Odyssium Date: Sun, 2 Apr 2023 16:38:17 +0200 Subject: [PATCH] Add port flag to serve Signed-off-by: Odyssium --- cmd/gothub/serve.go | 9 ++++++++- serve/serve.go | 5 ++++- 2 files changed, 12 insertions(+), 2 deletions(-) diff --git a/cmd/gothub/serve.go b/cmd/gothub/serve.go index 0477e06..18e5d43 100644 --- a/cmd/gothub/serve.go +++ b/cmd/gothub/serve.go @@ -6,15 +6,22 @@ import ( "codeberg.org/gothub/gothub/serve" ) +var port string = "3000" + var serveCmd = &cobra.Command{ Use: "serve", Short: "Start the web server.", Long: `Start the web server.`, Run: func(cmd *cobra.Command, args []string) { - serve.Serve() + serve.Serve(port) }, } func init() { rootCmd.AddCommand(serveCmd) + + serveCmd.Flags().StringVarP(&port, "port", "p", "", "The port GotHub will listen to. Defaults to 3000, and overrides the GOTHUB_PORT environment variable.") + + // set port variable to the value of the port flag + port = serveCmd.Flag("port").Value.String() } diff --git a/serve/serve.go b/serve/serve.go index 0923b38..3c9cc18 100644 --- a/serve/serve.go +++ b/serve/serve.go @@ -21,7 +21,7 @@ import ( _ "github.com/joho/godotenv/autoload" ) -func Serve() { +func Serve(port string) { engine := html.New("./views", ".html") engine.AddFunc( @@ -135,5 +135,8 @@ func Serve() { if !ok { val = "3000" } + if port != "" { + val = port + } log.Fatal(app.Listen(":" + val)) }