package main import ( "bytes" "fmt" "io" "net/http" "strings" ) // RunAPI provides a basic REST API func RunAPI(cfg *config) { http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) { log.Debug(fmt.Sprintf("Web: %s", r.URL.Path)) if r.URL.Path != "/" { log.Debug("Web 404") http.NotFound(w, r) return } pageTemplate, err := Asset("assets/index.html") if err != nil { log.Fatal(fmt.Sprintf("Error with HTTP server template asset: %s", err.Error())) } fmt.Fprintf(w, string(pageTemplate)) }) http.HandleFunc("/assets/", webStaticHandler) if cfg.API.TLS { log.Notice(fmt.Sprintf("Starting HTTPS server on %s", cfg.API.Listen)) err := http.ListenAndServeTLS(cfg.API.Listen, cfg.API.Cert, cfg.API.Key, nil) if err != nil { log.Fatal(fmt.Sprintf("Error starting HTTPS server: %s", err.Error())) } } else { log.Notice(fmt.Sprintf("Starting HTTP server on %s", cfg.API.Listen)) err := http.ListenAndServe(cfg.API.Listen, nil) if err != nil { log.Fatal(fmt.Sprintf("Error starting HTTP server: %s", err.Error())) } } } // webStaticHandler serves embedded static web files (js&css) func webStaticHandler(w http.ResponseWriter, r *http.Request) { assetPath := r.URL.Path[1:] staticAsset, err := Asset(assetPath) if err != nil { log.Error(err.Error()) http.NotFound(w, r) return } headers := w.Header() if strings.HasSuffix(assetPath, ".js") { headers["Content-Type"] = []string{"application/javascript"} } else if strings.HasSuffix(assetPath, ".css") { headers["Content-Type"] = []string{"text/css"} } io.Copy(w, bytes.NewReader(staticAsset)) } type stringer string func (s stringer) String() string { return string(s) }