debugger-assets/go/debugger-assets/main.go

package main

import (
	"flag"
	"html/template"
	"io/ioutil"
	"net/http"
	"path/filepath"
	"runtime"
	"strings"

	"github.com/fiorix/go-web/autogzip"
	"github.com/gorilla/mux"
	"go.skia.org/infra/go/common"
	"go.skia.org/infra/go/httputils"
	"go.skia.org/infra/go/sklog"
)

// flags
var (
	local        = flag.Bool("local", false, "Running locally if true. As opposed to in production.")
	port         = flag.String("port", ":8000", "HTTP service address (e.g., ':8000')")
	promPort     = flag.String("prom_port", ":20000", "Metrics service address (e.g., ':10110')")
	resourcesDir = flag.String("resources_dir", "/usr/local/share/debugger-assets", "The directory to find templates, JS, and CSS files. If blank the current directory will be used.")
	v2AtRoot     = flag.Bool("v2_at_root", false, "Serves /res/v2.html as / (the new dubugger)")
	versionFile  = flag.String("version_file", "/etc/skia-prod/VERSION", "The full path of the Skia VERSION file.")
)

var (
	//
	templates *template.Template

	// version is the version of Skia we are running.
	version string
)

func loadTemplates() {
	templates = template.Must(template.New("").ParseFiles(
		filepath.Join(*resourcesDir, "/res/v2.html"),
	))
}

func mainHandler(w http.ResponseWriter, r *http.Request) {
	// Serve new debugger at root when configured to do so.
	if *v2AtRoot {
		context := map[string]string{
			"Version":      version,
			"VersionShort": version[0:7],
		}
		if err := templates.ExecuteTemplate(w, "v2.html", context); err != nil {
			sklog.Errorf("Failed to expand template: %s", err)
		}
	} else {
		http.Redirect(w, r, "https://legacy-debugger.skia.org", http.StatusPermanentRedirect)
	}
}

func makeResourceHandler() func(http.ResponseWriter, *http.Request) {
	fileServer := http.FileServer(http.Dir(*resourcesDir))
	return func(w http.ResponseWriter, r *http.Request) {
		w.Header().Add("Cache-Control", "max-age=300")
		w.Header().Set("Access-Control-Allow-Origin", "*")
		fileServer.ServeHTTP(w, r)
	}
}

func Init() {
	if *resourcesDir == "" {
		_, filename, _, _ := runtime.Caller(0)
		*resourcesDir = filepath.Join(filepath.Dir(filename), "../..")
	}

	b, err := ioutil.ReadFile(*versionFile)
	if err != nil {
		// Note that version info will be unavailable when running locally
		sklog.Debugf("Did not find Skia version file: %s", err)
		b = []byte("unknown")
	}
	version = strings.TrimSpace(string(b))

	loadTemplates()
}

func main() {
	common.InitWithMust(
		"debugger-assets",
		common.PrometheusOpt(promPort),
		common.MetricsLoggingOpt(),
	)
	Init()

	router := mux.NewRouter()
	router.PathPrefix("/res/").HandlerFunc(autogzip.HandleFunc(makeResourceHandler()))
	router.HandleFunc("/", mainHandler)

	http.Handle("/", httputils.HealthzAndHTTPS(httputils.LoggingRequestResponse(router)))
	sklog.Info("Ready to serve.")
	sklog.Fatal(http.ListenAndServe(*port, nil))
}