This repository has been archived by the owner on Feb 9, 2022. It is now read-only.
/
root.go
85 lines (71 loc) · 2.52 KB
/
root.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
// Copyright 2017 Istio Authors
//
// 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.
// This is a simple command that is used to output the auto-generated collateral
// files for the various mixer CLI commands. More specifically, this outputs
// markdown files and man pages that describe the CLI commands, along with
// bash completion files.
package cmd
import (
"fmt"
"github.com/spf13/cobra"
"github.com/spf13/cobra/doc"
mixc "istio.io/mixer/cmd/client/cmd"
mixs "istio.io/mixer/cmd/server/cmd"
"istio.io/mixer/cmd/shared"
)
// GetRootCmd returns the root of the cobra command-tree.
func GetRootCmd(printf, fatalf shared.FormatFn) *cobra.Command {
outputDir := ""
rootCmd := &cobra.Command{
Use: "mixcol",
Short: "Generate collateral for mixer CLI commands",
PersistentPreRunE: func(cmd *cobra.Command, args []string) error {
if len(args) > 0 {
return fmt.Errorf("'%s' is an invalid argument", args[0])
}
return nil
},
Run: func(cmd *cobra.Command, args []string) {
work(printf, fatalf, outputDir)
},
}
rootCmd.Flags().StringVarP(&outputDir, "outputDir", "o", ".", "Directory where to generate the CLI collateral files")
return rootCmd
}
func work(printf, fatalf shared.FormatFn, outputDir string) {
roots := []*cobra.Command{
mixc.GetRootCmd(nil, nil, nil),
mixs.GetRootCmd(nil, nil, nil, nil, nil, nil),
}
printf("Outputting Mixer CLI collateral files to %s", outputDir)
for _, r := range roots {
hdr := doc.GenManHeader{
Title: "Istio Mixer",
Section: "Mixer CLI",
Manual: "Istio Mixer",
}
if err := doc.GenManTree(r, &hdr, outputDir); err != nil {
fatalf("Unable to output manpage tree: %v", err)
}
if err := doc.GenMarkdownTree(r, outputDir); err != nil {
fatalf("Unable to output markdown tree: %v", err)
}
if err := doc.GenYamlTree(r, outputDir); err != nil {
fatalf("Unable to output YAML tree: %v", err)
}
if err := r.GenBashCompletionFile(outputDir + "/" + r.Name() + ".bash"); err != nil {
fatalf("Unable to output bash completion file: %v", err)
}
}
}