Borg/cmd/serve.go

package cmd

import (
	"fmt"
	"net/http"
	"os"
	"strings"

	"forge.lthn.ai/Snider/Borg/pkg/compress"
	"forge.lthn.ai/Snider/Borg/pkg/datanode"
	"forge.lthn.ai/Snider/Borg/pkg/tarfs"

	"github.com/spf13/cobra"
)

// serveCmd represents the serve command
var serveCmd = NewServeCmd()

func NewServeCmd() *cobra.Command {
	serveCmd := &cobra.Command{
		Use:   "serve [file]",
		Short: "Serve a packaged PWA file",
		Long:  `Serves the contents of a packaged PWA file using a static file server.`,
		Args:  cobra.ExactArgs(1),
		RunE: func(cmd *cobra.Command, args []string) error {
			dataFile := args[0]
			port, _ := cmd.Flags().GetString("port")

			rawData, err := os.ReadFile(dataFile)
			if err != nil {
				return fmt.Errorf("Error reading data file: %w", err)
			}

			data, err := compress.Decompress(rawData)
			if err != nil {
				return fmt.Errorf("Error decompressing data: %w", err)
			}

			var fs http.FileSystem
			if strings.HasSuffix(dataFile, ".matrix") {
				fs, err = tarfs.New(data)
				if err != nil {
					return fmt.Errorf("Error creating TarFS from matrix tarball: %w", err)
				}
			} else {
				dn, err := datanode.FromTar(data)
				if err != nil {
					return fmt.Errorf("Error creating DataNode from tarball: %w", err)
				}
				fs = http.FS(dn)
			}

			http.Handle("/", http.FileServer(fs))

			fmt.Printf("Serving PWA on http://localhost:%s\n", port)
			err = http.ListenAndServe(":"+port, nil)
			if err != nil {
				return fmt.Errorf("Error starting server: %w", err)
			}
			return nil
		},
	}
	serveCmd.PersistentFlags().String("port", "8080", "Port to serve the PWA on")
	return serveCmd
}

func GetServeCmd() *cobra.Command {
	return serveCmd
}

func init() {
	RootCmd.AddCommand(GetServeCmd())
}