depot/web/fup/cmd/serve.go

// SPDX-FileCopyrightText: 2021 Luke Granger-Brown <depot@lukegb.com>
//
// SPDX-License-Identifier: Apache-2.0

package cmd

import (
	"context"
	"fmt"
	"log"
	"net/http"
	"os/exec"
	"strings"

	"github.com/google/safehtml"
	"github.com/spf13/cobra"
	"hg.lukegb.com/lukegb/depot/web/fup/fuphttp"
	"hg.lukegb.com/lukegb/depot/web/fup/fupstatic"
	"hg.lukegb.com/lukegb/depot/web/fup/minicheddar"
)

func init() {
	rootCmd.AddCommand(serveCmd)

	serveCmd.Flags().StringVar(&serveRoot, "root", "http://localhost:8191/", "Application root address.")
	serveCmd.Flags().StringVar(&serveStaticRoot, "static-root", "/static/", "Root address from which static assets should be referenced.")
	serveCmd.Flags().StringVarP(&serveBind, "listen", "l", ":8191", "Bind address for HTTP server.")
	serveCmd.Flags().BoolVar(&serveDirectOnly, "direct-only", false, "If set, all file serving will be proxied, even if the backend supports signed URLs.")

	serveCmd.Flags().StringVar(&serveCheddarPath, "cheddar-path", "cheddar", "Path to 'cheddar' binary to use for syntax highlighting. If it cannot be found, syntax highlighting and markdown rendering will be disabled.")
	serveCmd.Flags().StringVar(&serveCheddarAddr, "cheddar-address", "", "If non-empty, will be used instead of attempting to spawn a copy of cheddar.")
}

var (
	serveBind        string
	serveRoot        string
	serveStaticRoot  string
	serveDirectOnly  bool
	serveCheddarPath string
	serveCheddarAddr string

	serveCmd = &cobra.Command{
		Use:   "serve",
		Short: "Serve HTTP",
		RunE: func(cmd *cobra.Command, args []string) error {
			if !strings.HasSuffix(serveRoot, "/") {
				return fmt.Errorf("--root flag should end in / (value is %q)", serveRoot)
			}
			if !strings.HasSuffix(serveStaticRoot, "/") {
				return fmt.Errorf("--static-root flag should end in / (value is %q)", serveStaticRoot)
			}

			ctx := context.Background()
			highlighter, err := serveCheddar(ctx)
			if err != nil {
				return fmt.Errorf("spawning cheddar syntax highlighter: %v", err)
			}

			cfg := &fuphttp.Config{
				Templates:           fupstatic.Templates,
				Static:              fupstatic.Static,
				StaticRoot:          safehtml.TrustedResourceURLFromFlag(cmd.Flag("static-root").Value),
				AppRoot:             serveRoot,
				StorageURL:          bucketURL,
				RedirectToBlobstore: !serveDirectOnly,
				Highlighter:         highlighter,
			}
			a, err := fuphttp.New(ctx, cfg)
			if err != nil {
				return fmt.Errorf("constructing application: %w", err)
			}
			http.Handle("/", a.Handler())
			http.Handle("/static/", http.StripPrefix("/static/", http.FileServer(http.FS(fupstatic.Static))))
			log.Printf("Serving on %s", serveBind)
			return http.ListenAndServe(serveBind, nil)
		},
	}
)

func serveCheddar(ctx context.Context) (*minicheddar.Cheddar, error) {
	if serveCheddarAddr != "" {
		return minicheddar.Remote(serveCheddarAddr), nil
	}

	cpath, err := exec.LookPath(serveCheddarPath)
	if err != nil {
		log.Printf("couldn't find cheddar at %q; disabling syntax highlighting", serveCheddarPath)
		return nil, nil
	}

	return minicheddar.Spawn(ctx, cpath)
}