8ba0deab91
Refactored the existing tests to use the `_Good`, `_Bad`, and `_Ugly` testing convention. This provides a more structured approach to testing and ensures that a wider range of scenarios are covered, including valid inputs, invalid inputs, and edge cases. In addition to refactoring the tests, this change also includes several bug fixes that were uncovered by the new tests. These fixes improve the robustness and reliability of the codebase. The following packages and commands were affected: - `pkg/datanode` - `pkg/compress` - `pkg/github` - `pkg/matrix` - `pkg/pwa` - `pkg/vcs` - `pkg/website` - `cmd/all` - `cmd/collect` - `cmd/collect_github_repo` - `cmd/collect_website` - `cmd/compile` - `cmd/root` - `cmd/run` - `cmd/serve`
73 lines
1.7 KiB
Go
73 lines
1.7 KiB
Go
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/Snider/Borg/pkg/compress"
|
|
"github.com/Snider/Borg/pkg/datanode"
|
|
"github.com/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())
|
|
}
|