支持 --open 浏览器自动打开网址

Author: hellojukay

main.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"flag"
+	"fmt"
 	"log"
 	"net/http"
 	"os"
@@ -10,13 +11,15 @@ import (
 )
 
 var dir string
-var port string
+var port int
 var version bool
+var open bool
 
 func init() {
-	flag.StringVar(&port, "p", "8100", "port to serve on")
+	flag.IntVar(&port, "p", 8100, "port to serve on")
 	flag.StringVar(&dir, "path", "./", "the directory of static file to host")
 	flag.BoolVar(&version, "version", false, "print program version")
+	flag.BoolVar(&open, "open", false, "open with default browser")
 	flag.Parse()
 
 	if version {
@@ -32,10 +35,17 @@ func init() {
 }
 func main() {
 
-	http.Handle("/", FileServer(Dir(abs(dir))))
+	if open {
+		go func() {
+			if err := Open(fmt.Sprintf("http://127.0.0.1:%d", port)); err != nil {
+				log.Printf("can not open default browser %s", err)
+			}
 
-	log.Printf("Serving %s on HTTP port: %s\n", abs(dir), port)
-	log.Fatal(http.ListenAndServe(":"+port, nil))
+		}()
+	}
+	http.Handle("/", FileServer(Dir(abs(dir))))
+	log.Printf("Serving %s on HTTP port: %d\n", abs(dir), port)
+	log.Fatal(http.ListenAndServe(fmt.Sprintf(":%d", port), nil))
 }
 
 func abs(dir string) string {

open.go

@@ -0,0 +1,17 @@
+//go:build !windows || !darwin
+// +build !windows !darwin
+
+package main
+
+import (
+	"fmt"
+	"os/exec"
+)
+
+func Open(url string) error {
+	bin, err := exec.LookPath("xdg-open")
+	if err != nil {
+		return fmt.Errorf("can not find xdg-open, %s", err.Error())
+	}
+	return exec.Command(bin, url).Run()
+}

open_darwin.go

@@ -0,0 +1,12 @@
+//go:build darwin
+// +build darwin
+
+package main
+
+func Open(url string) error {
+	bin, err := exec.LookPath("open")
+	if err != nil {
+		return fmt.Errorf("can not find open, %s", err.Error())
+	}
+	return exec.Command(bin, url).Run()
+}

open_windows.go

@@ -0,0 +1,24 @@
+//go:build windows
+// +build windows
+
+package main
+
+// start up cmd.exe then run start command to open default browser
+func Open(url string) error {
+	bin, err := exec.LookPath("cmd")
+	if err != nil {
+		return fmt.Errorf("can not find start, %s", err.Error())
+	}
+	c := exec.Cmd{
+		Path: bin,
+	}
+	w, err := c.StdinPipe()
+	if err != nil {
+		return err
+	}
+	c.Start()
+	w.Write([]byte(fmt.Sprintf("start %s\n", url)))
+	time.Sleep(time.Second)
+	w.Close()
+	return c.Wait()
+}