foundation for self-upgrade logic

This commit is contained in:
Dane Everitt 2021-01-08 22:49:19 -08:00
parent 948d927eb9
commit 6fe2468a5a
No known key found for this signature in database
GPG Key ID: EEA66103B3D71F53
2 changed files with 63 additions and 0 deletions

View File

@ -75,6 +75,7 @@ func init() {
rootCommand.AddCommand(versionCommand)
rootCommand.AddCommand(configureCmd)
rootCommand.AddCommand(diagnosticsCmd)
rootCommand.AddCommand(newUpgradeCommand())
}
// Get the configuration path based on the arguments provided.

62
cmd/upgrade.go Normal file
View File

@ -0,0 +1,62 @@
package cmd
import (
"errors"
"fmt"
"github.com/apex/log"
"github.com/pterodactyl/wings/loggers/cli"
"github.com/spf13/cobra"
"runtime"
)
type upgrader struct{}
func newUpgradeCommand() *cobra.Command {
u := upgrader{}
command := &cobra.Command{
Use: "upgrade",
Short: "Performs a self-upgrade for Wings.",
Long: `Queries GitHub to find the latest Wings release and then downloads it, replacing
the existing system binary. This will use checksums and GPG signatures present on
the uploaded assets to validate that they have been released by the Pterodactyl team.
Once downloaded the Wings systemd process will be restarted if it is present on the
system, therefore this command MUST be executed as a root user.
This command can only be executed on ARM64/AMD64 Linux systems. All other systems will
report an error when executing this command.
`,
PersistentPreRun: func(cmd *cobra.Command, args []string) {
log.SetLevel(log.InfoLevel)
if debug {
log.SetLevel(log.DebugLevel)
}
log.SetHandler(cli.Default)
},
PreRunE: func(cmd *cobra.Command, args []string) error {
if runtime.GOOS != "linux" {
return errors.New(fmt.Sprintf("upgrade: os not supported: %s", runtime.GOOS))
}
if runtime.GOARCH != "arm64" && runtime.GOARCH != "amd64" {
return errors.New(fmt.Sprintf("upgrade: unexpected architecture: %s", runtime.GOARCH))
}
return nil
},
RunE: func(cmd *cobra.Command, args []string) error {
return u.execute()
},
}
command.PersistentFlags().String("version", "latest", "download a specific version of Wings")
command.PersistentFlags().String("repository", "pterodactyl/wings", "the repository to use when looking for updates -- if set, GPG verification is skipped")
command.PersistentFlags().String("auth-token", "", "a GitHub authentication token to use for private repositories")
command.PersistentFlags().Bool("download-only", false, "if set, do not restart wings after downloading")
return command
}
// Executes a self-upgrade of Wings by pulling down the latest version from GitHub
// (or the given flag version) and then restarting the Wings process.
func (u *upgrader) execute() error {
return nil
}