hakurei/cmd/pkgserver/main.go

package main

import (
	"cmp"
	"context"
	"embed"
	"fmt"
	"log"
	"net/http"
	"os"
	"os/signal"
	"slices"
	"syscall"
	"unique"

	"hakurei.app/command"
	"hakurei.app/container/check"
	"hakurei.app/internal/pkg"
	"hakurei.app/internal/rosa"
	"hakurei.app/message"
)

//go:generate sh -c "sass ui/static/dark.scss ui/static/dark.css && sass ui/static/light.scss ui/static/light.css && tsc ui/static/index.ts"
//go:embed ui/*
var content embed.FS

func serveWebUI(w http.ResponseWriter, r *http.Request) {
	fmt.Printf("serveWebUI: %s\n", r.URL.Path)
	w.Header().Set("Cache-Control", "no-cache, no-store, must-revalidate")
	w.Header().Set("Pragma", "no-cache")
	w.Header().Set("Expires", "0")
	w.Header().Set("X-Content-Type-Options", "nosniff")
	w.Header().Set("X-XSS-Protection", "1")
	w.Header().Set("X-Frame-Options", "DENY")

	http.ServeFileFS(w, r, content, "ui/index.html")
}
func serveStaticContent(w http.ResponseWriter, r *http.Request) {
	fmt.Printf("serveStaticContent: %s\n", r.URL.Path)
	switch r.URL.Path {
	case "/static/style.css":
		darkTheme := r.CookiesNamed("dark_theme")
		if len(darkTheme) > 0 && darkTheme[0].Value == "true" {
			http.ServeFileFS(w, r, content, "ui/static/dark.css")
		} else {
			http.ServeFileFS(w, r, content, "ui/static/light.css")
		}
		break
	case "/favicon.ico":
		http.ServeFileFS(w, r, content, "ui/static/favicon.ico")
		break
	case "/static/index.js":
		http.ServeFileFS(w, r, content, "ui/static/index.js")
		break
	default:
		http.NotFound(w, r)

	}
}
func serveAPI(pi *PackageIndex) func(w http.ResponseWriter, r *http.Request) {
	return func(w http.ResponseWriter, r *http.Request) {}
}

type SortOrders int

const (
	DeclarationDescending SortOrders = iota
	DeclarationAscending
	NameAscending
	NameDescending
	limitSortOrders
)

type PackageIndex [limitSortOrders][rosa.PresetUnexportedStart]*PackageIndexEntry

type PackageIndexEntry struct {
	id          unique.Handle[pkg.ID]
	name        string
	description string
	website     string
	version     string
	status      []byte
}

func createPackageIndex(cache *pkg.Cache, report *rosa.Report) *PackageIndex {
	var index PackageIndex
	var work []PackageIndexEntry
	for p := range rosa.PresetUnexportedStart {
		m := rosa.GetMetadata(p)
		v := rosa.Std.Version(p)
		a := rosa.Std.Load(p)
		id := cache.Ident(a)
		status, n := report.ArtifactOf(id)
		work[p] = PackageIndexEntry{
			id:          id,
			name:        m.Name,
			description: m.Description,
			website:     m.Website,
			version:     v,
			status:      status[:n],
		}
	}
	for i, p := range work {
		index[DeclarationAscending][i] = &p
	}
	slices.Reverse(work)
	for i, p := range work {
		index[DeclarationDescending][i] = &p
	}
	slices.SortFunc(work, func(a PackageIndexEntry, b PackageIndexEntry) int {
		return cmp.Compare(a.name, b.name)
	})
	for i, p := range work {
		index[NameAscending][i] = &p
	}
	slices.Reverse(work)
	for i, p := range work {
		index[NameDescending][i] = &p
	}

	return &index
}
func main() {
	log.SetFlags(0)
	log.SetPrefix("pkgserver: ")

	var (
		flagBaseDir string
		flagPort    uint16
	)

	ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM, syscall.SIGHUP)
	defer stop()
	msg := message.New(log.Default())

	c := command.New(os.Stderr, log.Printf, "pkgserver", func(args []string) error {
		reportPath := args[0]
		baseDir, err := check.NewAbs(flagBaseDir)
		if err != nil {
			return err
		}
		cache, err := pkg.Open(ctx, msg, 0, baseDir)
		if err != nil {
			return err
		}
		report, err := rosa.OpenReport(reportPath)
		if err != nil {
			return err
		}
		defer report.HandleAccess(&err)()
		index := createPackageIndex(cache, report)
		http.HandleFunc("GET /{$}", serveWebUI)
		http.HandleFunc("GET /favicon.ico", serveStaticContent)
		http.HandleFunc("GET /static/", serveStaticContent)
		http.HandleFunc("GET /api/", serveAPI(index))
		err = http.ListenAndServe(fmt.Sprintf(":%d", flagPort), nil)
		if err != nil {
			return err
		}
		return nil
	}).Flag(
		&flagBaseDir,
		"b", command.StringFlag(""),
		"base directory for cache",
	).Flag(
		&flagPort,
		"p", command.IntFlag(8067),
		"http listen port",
	)
	c.MustParse(os.Args[1:], func(e error) {
		log.Fatal(e)
	})

}