This commit introduces a new `DataNode` package, which provides an in-memory, `fs.FS`-compatible filesystem with a `debme`-like interface. The `DataNode` can be serialized to and from a TAR archive, making it suitable for storing downloaded assets. The `pwa` and `serve` commands have been refactored to use the `DataNode`. The `pwa` command now packages downloaded PWA assets into a `DataNode` and saves it as a `.dat` file. The `serve` command loads a `.dat` file into a `DataNode` and serves its contents.
49 lines
1.1 KiB
Go
49 lines
1.1 KiB
Go
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"os"
|
|
|
|
"borg-data-collector/pkg/datanode"
|
|
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
// serveCmd represents the serve command
|
|
var 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),
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
pwaFile := args[0]
|
|
port, _ := cmd.Flags().GetString("port")
|
|
|
|
pwaData, err := os.ReadFile(pwaFile)
|
|
if err != nil {
|
|
fmt.Printf("Error reading PWA file: %v\n", err)
|
|
return
|
|
}
|
|
|
|
dn, err := datanode.FromTar(pwaData)
|
|
if err != nil {
|
|
fmt.Printf("Error creating DataNode from tarball: %v\n", err)
|
|
return
|
|
}
|
|
|
|
http.Handle("/", http.FileServer(http.FS(dn)))
|
|
|
|
fmt.Printf("Serving PWA on http://localhost:%s\n", port)
|
|
err = http.ListenAndServe(":"+port, nil)
|
|
if err != nil {
|
|
fmt.Printf("Error starting server: %v\n", err)
|
|
return
|
|
}
|
|
},
|
|
}
|
|
|
|
func init() {
|
|
rootCmd.AddCommand(serveCmd)
|
|
serveCmd.PersistentFlags().String("port", "8080", "Port to serve the PWA on")
|
|
}
|