fs-go/main.go

package main

import (
	"flag"
	"fmt"
	"github.com/jianboy/fs-go/util"
	"log"
	"net/http"
	"strconv"
)

var port int
var dir string

func main() {
	banner()
	// Handle the flags that are provided.
	flag.IntVar(&port, "port", 8080, "The port that the server will listen to.")
	flag.StringVar(&dir, "dir", ".", "The root directory that will be served.")
	// Parse the flags.
	flag.Parse()

	log.Printf("server is start %v: %v", port, dir)
	//check port
	// for i := 0; i < 5; i++ {
	// 	if util.checkAvailablePort(port) {
	// 		break
	// 	} else {
	// 		// log.Println(strconv.Itoa(port) + " is not available.")
	// 		log.Printf("server error on port %v: %v", port, err)
	// 		port++
	// 	}
	// }
	// util.KillProcessByName("fs-go.exe")  // 执行会把自己给kill掉
	go util.OpenBrowser("http://localhost:" + strconv.Itoa(port)) // Open browser in a goroutine

	err := http.ListenAndServe(":"+fmt.Sprintf("%d", port), http.HandlerFunc(handler))
	if err != nil {
		log.Fatal("server error on port %v: %v", port, err)

	}
	log.Printf("Server started on port %d\n", port)
}

func banner() {
	fmt.Println(`
                   _______  _______         _______  _______ 
                  (  ____ \(  ____ \       (  ____ \(  ___  )
                  | (    \/| (    \/       | (    \/| (   ) |
                  | (__    | (_____  _____ | |      | |   | |
                  |  __)   (_____  )(_____)| | ____ | |   | |
                  | (            ) |       | | \_  )| |   | |
                  | )      /\____) |       | (___) || (___) |
                  |/       \_______)       (_______)(_______)
			author: liuyuqi.gov@msn.cn
	`)
}

// remove header cache
func handler(w http.ResponseWriter, r *http.Request) {
	w.Header().Set("Cache-Control", "no-cache, no-store, must-revalidate") // HTTP 1.1.
	w.Header().Set("Pragma", "no-cache")                                   // HTTP 1.0.
	w.Header().Set("Expires", "0")                                         // Proxies.

	if r.URL.Path == "/favicon.ico" {
		return
	}
	http.StripPrefix("/", http.FileServer(http.Dir(dir))).ServeHTTP(w, r)
}