Merge pull request #7 from Guz013/main

cmd/root.go

@@ -15,9 +15,10 @@ var rootCmd = &cobra.Command{
 		dark, _ := cmd.Flags().GetBool("dark")
 		browser, _ := cmd.Flags().GetBool("browser")
 		openReadme, _ := cmd.Flags().GetBool("readme")
+		host, _ := cmd.Flags().GetString("host")
 		port, _ := cmd.Flags().GetInt("port")
 
-		client := pkg.Client{Dark: dark, OpenBrowser: browser, Port: port, OpenReadme: openReadme}
+		client := pkg.Client{Dark: dark, OpenBrowser: browser, Host: host, Port: port, OpenReadme: openReadme}
 
 		var file string
 		if len(args) == 1 {
@@ -39,5 +40,6 @@ func init() {
 	rootCmd.Flags().BoolP("dark", "d", false, "Activate darkmode")
 	rootCmd.Flags().BoolP("browser", "b", true, "Open new browser tab")
 	rootCmd.Flags().BoolP("readme", "r", true, "Open readme if no file provided")
+	rootCmd.Flags().StringP("host", "H", "localhost", "Host to use")
 	rootCmd.Flags().IntP("port", "p", 6419, "Port to use")
 }

pkg/client.go

@@ -3,6 +3,7 @@ package pkg
 type Client struct {
 	Dark        bool
 	OpenBrowser bool
+	Host        string
 	Port        int
 	OpenReadme  bool
 }

pkg/webserver.go

@@ -62,7 +62,7 @@ func (client *Client) Serve(file string) error {
 		}
 	})
 
-	addr := fmt.Sprintf("http://localhost:%d/", client.Port)
+	addr := fmt.Sprintf("http://%s:%d/", client.Host, client.Port)
 	if file == "" {
 		// If README.md exists then open README.md at beginning
 		readme := "README.md"