This repository has been archived by the owner on Sep 14, 2022. It is now read-only.
/
node_list.go
100 lines (86 loc) · 2.4 KB
/
node_list.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
98
99
100
/*
Copyright © 2020 CAST AI
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package cmd
import (
"io"
"github.com/jedib0t/go-pretty/v6/table"
"github.com/sirupsen/logrus"
"github.com/spf13/cobra"
"github.com/castai/cli/pkg/client"
"github.com/castai/cli/pkg/client/sdk"
"github.com/castai/cli/pkg/command"
"github.com/castai/cli/pkg/prettytime"
)
func newNodeListCmd(log logrus.FieldLogger, api client.Interface) *cobra.Command {
cmd := &cobra.Command{
Use: "list",
Short: "List clusters nodes",
Run: func(cmd *cobra.Command, args []string) {
if err := handleListNodes(cmd, api); err != nil {
log.Fatal(err)
}
},
}
cmd.PersistentFlags().StringP(flagCluster, "c", "", "cluster name or ID")
command.AddJSONOutput(cmd)
return cmd
}
func handleListNodes(cmd *cobra.Command, api client.Interface) error {
cluster, err := getClusterFromFlag(cmd, api)
if err != nil {
return err
}
res, err := api.ListClusterNodes(cmd.Context(), sdk.ClusterId(cluster.Id))
if err != nil {
return err
}
if command.OutputJSON() {
command.PrintOutput(res)
return nil
}
printNodesListTable(cmd.OutOrStdout(), res)
return nil
}
func printNodesListTable(out io.Writer, items []sdk.Node) {
t := table.NewWriter()
t.SetStyle(command.DefaultTableStyle)
t.SetOutputMirror(out)
t.AppendHeader(table.Row{"ID", "Name", "Cloud", "Role", "Shape", "Status", "Age", "Public_IP", "Private_IP"})
for _, item := range items {
if item.Network == nil {
item.Network = &sdk.NodeNetwork{}
}
if item.State == nil {
var empty string
item.State = &sdk.NodeState{Phase: &empty}
}
t.AppendRow(table.Row{
nodeValueString(item.Id),
nodeValueString(item.Name),
item.Cloud,
item.Role,
item.Shape,
nodeValueString(item.State.Phase),
prettytime.Format(*item.CreatedAt),
item.Network.PublicIp,
item.Network.PrivateIp,
})
}
t.Render()
}
func nodeValueString(v *string) string {
if v == nil {
return ""
}
return *v
}