-
Notifications
You must be signed in to change notification settings - Fork 3
/
main.go
117 lines (98 loc) · 2.74 KB
/
main.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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
package main
import (
"flag"
"fmt"
"code.cloudfoundry.org/cli/plugin"
"github.com/aegershman/cf-report-usage-plugin/internal/presentation"
"github.com/aegershman/cf-report-usage-plugin/internal/report"
log "github.com/sirupsen/logrus"
)
type reportUsageCmd struct{}
type orgNamesFlag struct {
names []string
}
func (o *orgNamesFlag) String() string {
return fmt.Sprint(o.names)
}
func (o *orgNamesFlag) Set(value string) error {
o.names = append(o.names, value)
return nil
}
type formatFlag struct {
formats []string
}
func (f *formatFlag) String() string {
return fmt.Sprint(f.formats)
}
func (f *formatFlag) Set(value string) error {
f.formats = append(f.formats, value)
return nil
}
// reportUsageCommand is the "real" main entrypoint into program execution
func (cmd *reportUsageCmd) reportUsageCommand(cli plugin.CliConnection, args []string) {
var (
orgNamesFlag orgNamesFlag
formatFlag formatFlag
logLevelFlag string
)
flagss := flag.NewFlagSet(args[0], flag.ContinueOnError)
flagss.Var(&orgNamesFlag, "o", "")
flagss.Var(&formatFlag, "format", "")
flagss.StringVar(&logLevelFlag, "log-level", "info", "")
if err := flagss.Parse(args[1:]); err != nil {
log.Fatalln(err)
}
logLevel, err := log.ParseLevel(logLevelFlag)
if err != nil {
log.Fatalln(err)
}
log.SetLevel(logLevel)
reporter, err := report.NewClient(cli)
if err != nil {
log.Fatalln(err)
}
summaryReport, err := reporter.GetSummaryReportByOrgNames(orgNamesFlag.names...)
if err != nil {
log.Fatalln(err)
}
presenter := presentation.NewPresenter(*summaryReport, formatFlag.formats...) // todo hacky pointer
presenter.Render()
}
// Run -
func (cmd *reportUsageCmd) Run(cli plugin.CliConnection, args []string) {
switch args[0] {
case "report-usage":
cmd.reportUsageCommand(cli, args)
default:
log.Debugln("did you know plugin commands can still get ran when uninstalling a plugin? interesting, right?")
return
}
}
// GetMetadata -
func (cmd *reportUsageCmd) GetMetadata() plugin.PluginMetadata {
return plugin.PluginMetadata{
Name: "cf-report-usage-plugin",
Version: plugin.VersionType{
Major: 3,
Minor: 3,
Build: 2,
},
Commands: []plugin.Command{
{
Name: "report-usage",
HelpText: "View AIs, SIs and memory usage for orgs and spaces",
UsageDetails: plugin.Usage{
Usage: "cf report-usage [-o orgName...] [--format formatChoice...]",
Options: map[string]string{
"o": "organization(s) included in report. Flag can be provided multiple times.",
"format": "format to print as (options: string,table,json). Flag can be provided multiple times.",
"log-level": "(options: info,debug,trace) (default: info)",
},
},
},
},
}
}
func main() {
plugin.Start(new(reportUsageCmd))
}