boundary/internal/servers/controller/handler_ui.go

// +build ui

package controller

import (
	"net/http"
	"strings"

	"github.com/hashicorp/watchtower/internal/ui"
)

func init() {
	handleUi = handleUiWithAssets
}

func handleUiWithAssets(c *Controller) http.Handler {
	var nextHandler http.Handler
	if c.conf.RawConfig.PassthroughDirectory != "" {
		nextHandler = devPassthroughHandler(c.logger, c.conf.RawConfig.PassthroughDirectory)
	} else {
		nextHandler = http.FileServer(ui.AssetFile())
	}

	rootHandler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		switch r.URL.Path {
		case "/":
			irw := newIndexResponseWriter(c.conf.DefaultOrgId)
			nextHandler.ServeHTTP(irw, r)
			irw.writeToWriter(w)

		default:
			nextHandler.ServeHTTP(w, r)
		}
	})

	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		if r.Method != http.MethodGet {
			w.WriteHeader(http.StatusMethodNotAllowed)
			return
		}

		dotIndex := strings.LastIndex(r.URL.Path, ".")
		switch dotIndex {
		case -1:
			// For all paths without an extension serve /index.html
			r.URL.Path = "/"

		default:
			switch r.URL.Path {
			case "/", "/favicon.png", "/assets/styles.css":

			default:
				for i := dotIndex + 1; i < len(r.URL.Path); i++ {
					intVal := r.URL.Path[i]
					// Current guidance from FE is if it's only alphanum after
					// the last dot, treat it as an extension
					if intVal < '0' ||
						(intVal > '9' && intVal < 'A') ||
						(intVal > 'Z' && intVal < 'a') ||
						intVal > 'z' {
						// Not an extension. Serve the contents of index.html
						r.URL.Path = "/"
					}
				}
			}
		}

		// Fall through to the next handler
		rootHandler.ServeHTTP(w, r)
	})
}