/
list_policy_commands.go
127 lines (108 loc) · 3.06 KB
/
list_policy_commands.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
118
119
120
121
122
123
124
125
126
127
// lhhelp.go: Part of the Antha language
// Copyright (C) 2015 The Antha authors. All rights reserved.
//
// This program is free software; you can redistribute it and/or
// modify it under the terms of the GNU General Public License
// as published by the Free Software Foundation; either version 2
// of the License, or (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software
// Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
//
// For more information relating to the software or licensing issues please
// contact license@antha-lang.org or write to the Antha team c/o
// Synthace Ltd. The London Bioscience Innovation Centre
// 2 Royal College St, London NW1 0NH UK
package cmd
import (
"encoding/csv"
"encoding/json"
"fmt"
"os"
"sort"
"strings"
"github.com/antha-lang/antha/antha/anthalib/wtype"
"github.com/ghodss/yaml"
"github.com/spf13/cobra"
"github.com/spf13/viper"
)
var listPolicyCommandsCmd = &cobra.Command{
Use: "policyCommands",
Short: "List available liquid handling policy commands",
RunE: listPolicyCommands,
}
type simplePolicyCommand struct {
Name string
Type string
Desc string
}
type simplePolicyCommands []simplePolicyCommand
func (a simplePolicyCommands) Less(i, j int) bool {
return a[i].Name < a[j].Name
}
func (a simplePolicyCommands) Swap(i, j int) {
a[i], a[j] = a[j], a[i]
}
func (a simplePolicyCommands) Len() int {
return len(a)
}
func listPolicyCommands(cmd *cobra.Command, args []string) error {
if err := viper.BindPFlags(cmd.Flags()); err != nil {
return err
}
var cs simplePolicyCommands
for _, c := range wtype.GetPolicyConsequents() {
cs = append(cs, simplePolicyCommand{
Name: c.Name,
Type: c.TypeName(),
Desc: c.Desc,
})
}
sort.Sort(cs)
output := viper.GetString("output")
switch output {
case jsonOutput:
bs, err := json.MarshalIndent(cs, "", " ")
if err != nil {
return err
}
_, err = fmt.Println(string(bs))
return err
case yamlOutput:
bs, err := yaml.Marshal(cs)
if err != nil {
return err
}
_, err = fmt.Print(string(bs))
return err
case csvOutput:
var lines [][]string
lines = append(lines, []string{"name", "type", "description"})
for _, c := range cs {
lines = append(lines, []string{c.Name, c.Type, c.Desc})
}
w := csv.NewWriter(os.Stdout)
err := w.WriteAll(lines)
return err
case textOutput:
var lines []string
lines = append(lines, "name,type,description")
for _, c := range cs {
lines = append(lines, fmt.Sprintf("%s, %s, %s", c.Name, c.Type, c.Desc))
}
_, err := fmt.Println(strings.Join(lines, "\n"))
return err
default:
return fmt.Errorf("unknown output format %q", output)
}
}
func init() {
c := listPolicyCommandsCmd
listCmd.AddCommand(c)
}