glaciers-in-archives/snowman

package cmd

import (
    "fmt"
    "log"
    "net/http"
    "os"
    "strconv"

    "github.com/spf13/cobra"
)

var port int
var serverInterface string

func loggingHandler(h http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        log.Println(r.URL.Path)
        h.ServeHTTP(w, r)
    })
}

// serverCmd represents the server command
var serverCmd = &cobra.Command{
    Use:   "server",
    Short: "Serves your site through an HTTP server.",
    Long:  `This command will serve your site through Snowman's built-in webserver. It's intended only for usage during development.`,
    Args:  cobra.RangeArgs(0, 2),
    RunE: func(cmd *cobra.Command, args []string) error {
        if _, err := os.Stat("site"); err != nil {
            fmt.Println("No site found. Did your run snowman build?")
        }

        fs := http.FileServer(http.Dir("site/"))
        address := serverInterface + ":" + strconv.Itoa(port)
        fmt.Println("Serving site at http://" + address + ". Hold ctrl+c to exit.")
        if err := http.ListenAndServe(address, loggingHandler(fs)); err != nil {
            // utils.ErrorExit() wont work here has
            log.Println(err) // #TODO shutdown gracefully
        }
        // #TODO shutdown gracefully

        return nil
    },
}

func init() {
    rootCmd.AddCommand(serverCmd)
    serverCmd.Flags().IntVarP(&port, "port", "p", 8000, "Port on which the server will listen.")
    serverCmd.Flags().StringVarP(&serverInterface, "address", "a", "127.0.0.1", "Address to which the server will bind.")
}