devpod/cmd/pro/pro.go at 6d4691a479a40049887177e0abb273ea5cd1a5a8 · skevetter/devpod · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
package pro

import (
	"context"
	"fmt"
	"os"

	"github.com/sirupsen/logrus"
	"github.com/skevetter/devpod/cmd/flags"
	"github.com/skevetter/devpod/cmd/pro/add"
	"github.com/skevetter/devpod/cmd/pro/daemon"
	proflags "github.com/skevetter/devpod/cmd/pro/flags"
	"github.com/skevetter/devpod/cmd/pro/provider"
	"github.com/skevetter/devpod/cmd/pro/reset"
	"github.com/skevetter/devpod/pkg/config"
	providerpkg "github.com/skevetter/devpod/pkg/provider"
	"github.com/skevetter/devpod/pkg/workspace"
	"github.com/skevetter/log"
	"github.com/spf13/cobra"
)

// NewProCmd returns a new command.
func NewProCmd(flags *flags.GlobalFlags, streamLogger *log.StreamLogger) *cobra.Command {
	globalFlags := &proflags.GlobalFlags{GlobalFlags: flags}
	proCmd := &cobra.Command{
		Use:           "pro",
		Short:         "DevPod Pro commands",
		SilenceUsage:  true,
		SilenceErrors: true,
		Args:          cobra.NoArgs,
		PersistentPreRunE: func(c *cobra.Command, _ []string) error {
			globalFlags = proflags.SetGlobalFlags(c.PersistentFlags())
			if flags.Silent {
				streamLogger.SetLevel(logrus.FatalLevel)
			}
			if flags.Debug {
				streamLogger.SetLevel(logrus.DebugLevel)
			}

			if os.Getenv(config.EnvDebug) == config.BoolTrue {
				log.Default.SetLevel(logrus.DebugLevel)
			}
			if flags.LogOutput == "json" {
				log.Default.SetFormat(log.JSONFormat)
			}

			return nil
		},
	}

	proCmd.AddCommand(NewLoginCmd(globalFlags))
	proCmd.AddCommand(NewListCmd(globalFlags))
	proCmd.AddCommand(NewDeleteCmd(globalFlags))
	proCmd.AddCommand(NewImportCmd(globalFlags))
	proCmd.AddCommand(NewStartCmd(globalFlags))
	proCmd.AddCommand(NewRebuildCmd(globalFlags))
	proCmd.AddCommand(NewSleepCmd(globalFlags))
	proCmd.AddCommand(NewWakeupCmd(globalFlags))
	proCmd.AddCommand(reset.NewResetCmd(globalFlags))
	proCmd.AddCommand(provider.NewProProviderCmd(globalFlags))
	proCmd.AddCommand(daemon.NewCmd(globalFlags))
	proCmd.AddCommand(add.NewAddCmd(globalFlags))
	proCmd.AddCommand(NewWatchWorkspacesCmd(globalFlags))
	proCmd.AddCommand(NewSelfCmd(globalFlags))
	proCmd.AddCommand(NewVersionCmd(globalFlags))
	proCmd.AddCommand(NewListProjectsCmd(globalFlags))
	proCmd.AddCommand(NewListWorkspacesCmd(globalFlags))
	proCmd.AddCommand(NewListTemplatesCmd(globalFlags))
	proCmd.AddCommand(NewListClustersCmd(globalFlags))
	proCmd.AddCommand(NewCreateWorkspaceCmd(globalFlags))
	proCmd.AddCommand(NewUpdateWorkspaceCmd(globalFlags))
	proCmd.AddCommand(NewCheckHealthCmd(globalFlags))
	proCmd.AddCommand(NewCheckUpdateCmd(globalFlags))
	proCmd.AddCommand(NewUpdateProviderCmd(globalFlags))
	return proCmd
}

func findProProvider(
	ctx context.Context,
	context, provider, host string,
	log log.Logger,
) (*config.Config, *providerpkg.ProviderConfig, error) {
	devPodConfig, err := config.LoadConfig(context, provider)
	if err != nil {
		return nil, nil, err
	}

	pCfg, err := workspace.ProviderFromHost(ctx, devPodConfig, host, log)
	if err != nil {
		return devPodConfig, nil, fmt.Errorf("load provider: %w", err)
	}

	return devPodConfig, pCfg, nil
}