mirror of
https://github.com/psviderski/uncloud.git
synced 2026-08-26 19:13:34 +00:00
92 lines
2.7 KiB
Go
92 lines
2.7 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/netip"
|
|
"strings"
|
|
|
|
"github.com/psviderski/uncloud/cmd/uncloud/caddy"
|
|
cmdcontext "github.com/psviderski/uncloud/cmd/uncloud/context"
|
|
"github.com/psviderski/uncloud/cmd/uncloud/dns"
|
|
"github.com/psviderski/uncloud/cmd/uncloud/machine"
|
|
"github.com/psviderski/uncloud/cmd/uncloud/service"
|
|
"github.com/psviderski/uncloud/cmd/uncloud/volume"
|
|
"github.com/psviderski/uncloud/internal/cli"
|
|
"github.com/psviderski/uncloud/internal/cli/config"
|
|
"github.com/psviderski/uncloud/internal/fs"
|
|
"github.com/psviderski/uncloud/internal/version"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
type globalOptions struct {
|
|
configPath string
|
|
connect string
|
|
}
|
|
|
|
func main() {
|
|
opts := globalOptions{}
|
|
cmd := &cobra.Command{
|
|
Use: "uncloud",
|
|
Short: "A CLI tool for managing Uncloud resources such as clusters, machines, and services.",
|
|
Version: version.String(),
|
|
SilenceUsage: true,
|
|
SilenceErrors: true,
|
|
PersistentPreRunE: func(cmd *cobra.Command, args []string) error {
|
|
var conn *config.MachineConnection
|
|
if opts.connect != "" {
|
|
if strings.HasPrefix(opts.connect, "tcp://") {
|
|
addrPort, err := netip.ParseAddrPort(opts.connect[len("tcp://"):])
|
|
if err != nil {
|
|
return fmt.Errorf("parse TCP address: %w", err)
|
|
}
|
|
conn = &config.MachineConnection{
|
|
TCP: &addrPort,
|
|
}
|
|
} else {
|
|
dest := opts.connect
|
|
if strings.HasPrefix(dest, "ssh://") {
|
|
dest = dest[len("ssh://"):]
|
|
}
|
|
conn = &config.MachineConnection{
|
|
SSH: config.SSHDestination(dest),
|
|
}
|
|
}
|
|
}
|
|
|
|
configPath := fs.ExpandHomeDir(opts.configPath)
|
|
uncli, err := cli.New(configPath, conn)
|
|
if err != nil {
|
|
return fmt.Errorf("initialise CLI: %w", err)
|
|
}
|
|
cmd.SetContext(context.WithValue(cmd.Context(), "cli", uncli))
|
|
return nil
|
|
},
|
|
}
|
|
|
|
cmd.PersistentFlags().StringVar(&opts.connect, "connect", "",
|
|
"Connect to a remote cluster machine without using the Uncloud configuration file.\n"+
|
|
"Format: [ssh://]user@host[:port] or tcp://host:port")
|
|
// TODO: allow to override using UNCLOUD_CONFIG env var.
|
|
cmd.PersistentFlags().StringVar(&opts.configPath, "uncloud-config", "~/.config/uncloud/config.yaml",
|
|
"Path to the Uncloud configuration file.")
|
|
_ = cmd.MarkPersistentFlagFilename("uncloud-config", "yaml", "yml")
|
|
// TODO: make --context a global flag and pass it as a value of the command context.
|
|
|
|
cmd.AddCommand(
|
|
NewDeployCommand(),
|
|
caddy.NewRootCommand(),
|
|
cmdcontext.NewRootCommand(),
|
|
dns.NewRootCommand(),
|
|
machine.NewRootCommand(),
|
|
service.NewRootCommand(),
|
|
service.NewInspectCommand(),
|
|
service.NewListCommand(),
|
|
service.NewRmCommand(),
|
|
service.NewRunCommand(),
|
|
service.NewScaleCommand(),
|
|
volume.NewRootCommand(),
|
|
)
|
|
cobra.CheckErr(cmd.Execute())
|
|
}
|