safing-portmaster/cmds/updatemgr/main.go

package main

import (
	"fmt"
	"os"
	"path/filepath"

	"github.com/spf13/cobra"

	"github.com/safing/portmaster/base/updater"
	"github.com/safing/portmaster/base/utils"
)

var (
	registry *updater.ResourceRegistry
	distDir  string
)

var rootCmd = &cobra.Command{
	Use:   "updatemgr",
	Short: "A simple tool to assist in the update and release process",
	PersistentPreRunE: func(cmd *cobra.Command, args []string) error {
		// Check if the distribution directory exists.
		absDistPath, err := filepath.Abs(distDir)
		if err != nil {
			return fmt.Errorf("failed to get absolute path of distribution directory: %w", err)
		}
		_, err = os.Stat(absDistPath)
		if err != nil {
			return fmt.Errorf("failed to access distribution directory: %w", err)
		}

		registry = &updater.ResourceRegistry{}
		err = registry.Initialize(utils.NewDirStructure(absDistPath, utils.PublicReadPermission))
		if err != nil {
			return err
		}

		err = registry.ScanStorage("")
		if err != nil {
			return err
		}

		return nil
	},
	SilenceUsage: true,
}

func init() {
	flags := rootCmd.PersistentFlags()
	flags.StringVar(&distDir, "dist-dir", "dist", "Set the distribution directory. Falls back to ./dist if available.")
}

func main() {
	if err := rootCmd.Execute(); err != nil {
		os.Exit(1)
	}
}