-
Notifications
You must be signed in to change notification settings - Fork 508
/
cmd_version.go
97 lines (86 loc) · 2.9 KB
/
cmd_version.go
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
95
96
97
package cli
import (
"context"
"fmt"
"github.com/spf13/cobra"
"google.golang.org/grpc/codes"
"google.golang.org/grpc/status"
empty "google.golang.org/protobuf/types/known/emptypb"
"github.com/telepresenceio/telepresence/rpc/v2/common"
"github.com/telepresenceio/telepresence/rpc/v2/daemon"
"github.com/telepresenceio/telepresence/rpc/v2/manager"
"github.com/telepresenceio/telepresence/v2/pkg/client"
"github.com/telepresenceio/telepresence/v2/pkg/client/cli/ann"
"github.com/telepresenceio/telepresence/v2/pkg/client/cli/util"
)
func versionCommand() *cobra.Command {
return &cobra.Command{
Use: "version",
Args: cobra.NoArgs,
Short: "Show version",
PreRunE: util.ForcedUpdateCheck,
RunE: PrintVersion,
Annotations: map[string]string{
ann.RootDaemon: ann.Optional,
ann.UserDaemon: ann.Optional,
},
}
}
// PrintVersion requests version info from the daemon and prints both client and daemon version.
func PrintVersion(cmd *cobra.Command, _ []string) error {
if err := util.InitCommand(cmd); err != nil {
return err
}
fmt.Fprintf(cmd.OutOrStdout(), "Client: %s\n",
client.DisplayVersion())
ctx := cmd.Context()
version, err := daemonVersion(ctx)
switch {
case err == nil:
fmt.Fprintf(cmd.OutOrStdout(), "Root Daemon: %s (api v%d)\n", version.Version, version.ApiVersion)
case err == util.ErrNoRootDaemon:
fmt.Fprintln(cmd.OutOrStdout(), "Root Daemon: not running")
default:
fmt.Fprintf(cmd.OutOrStdout(), "Root Daemon: error: %v\n", err)
}
version, err = connectorVersion(ctx)
switch {
case err == nil:
fmt.Fprintf(cmd.OutOrStdout(), "User Daemon: %s (api v%d)\n", version.Version, version.ApiVersion)
var mgrVer *manager.VersionInfo2
mgrVer, err = managerVersion(ctx)
switch {
case err == nil:
fmt.Fprintf(cmd.OutOrStdout(), "Traffic Manager: %s\n", mgrVer.Version)
case status.Code(err) == codes.Unavailable:
fmt.Fprintln(cmd.OutOrStdout(), "Traffic Manager: not connected")
default:
fmt.Fprintf(cmd.OutOrStdout(), "Traffic Manager: error: %v\n", err)
}
case err == util.ErrNoUserDaemon:
fmt.Fprintln(cmd.OutOrStdout(), "User Daemon: not running")
default:
fmt.Fprintf(cmd.OutOrStdout(), "User Daemon: error: %v\n", err)
}
return nil
}
func daemonVersion(ctx context.Context) (*common.VersionInfo, error) {
if conn, err := client.DialSocket(ctx, client.DaemonSocketName); err == nil {
defer conn.Close()
return daemon.NewDaemonClient(conn).Version(ctx, &empty.Empty{})
}
return nil, util.ErrNoRootDaemon
}
func connectorVersion(ctx context.Context) (*common.VersionInfo, error) {
if userD := util.GetUserDaemon(ctx); userD != nil {
return userD.Version(ctx, &empty.Empty{})
}
return nil, util.ErrNoUserDaemon
}
func managerVersion(ctx context.Context) (*manager.VersionInfo2, error) {
userD := util.GetUserDaemon(ctx)
if userD == nil {
return nil, util.ErrNoUserDaemon
}
return manager.NewManagerClient(userD.Conn).Version(ctx, &empty.Empty{})
}