Crimson client: created add-host and query-host

go/src/infra/crimson/cmd/cmdhelper/cmdhelper.go

 // Copyright 2016 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 package cmdhelper
 
 import (
         "bufio"
         "bytes"
         "encoding/json"
@@ skipping 324 matching lines @@
 func PrintIPRange(ipRanges []*crimson.IPRange, format FormatType) {
         lines, err := FormatIPRange(ipRanges, format)
         if err != nil {
                 fmt.Fprintf(os.Stderr, "ERROR: %s", err)
                 return
         }
         for _, s := range lines {
                 fmt.Println(s)
         }
 }
+
+// FormatHostList formats a list of hosts for pretty-printing.
+func FormatHostList(hostList *crimson.HostList, format FormatType) ([]string, error) {
+        var formatter Formatter
+
+        switch format {
+        case jsonFormat:
+                jsonBytes, err := json.Marshal(hostList.Hosts)

[Vadim Sh., 2016/06/30 23:58:12]

nit: consider using https://golang.org/pkg/encoding/json/#MarshalIndent instead
(with indent == "  "). Otherwise it may be too dense.

[pgervais, 2016/07/01 01:01:00]

Done.

+                if err != nil {
+                        return []string{}, err
+                }
+                return []string{string(jsonBytes)}, nil
+        case textFormat:
+                formatter = &TextFormatter{}
+        case csvFormat:
+                formatter = &CSVFormatter{}
+        }
+        rows := [][]string{{"mac", "ip", "site", "hostname", "class"}}
+        for _, host := range hostList.Hosts {
+                rows = append(rows, []string{
+                        fmt.Sprintf("%s", host.MacAddr),
+                        fmt.Sprintf("%s", host.Ip),
+                        fmt.Sprintf("%s", host.Site),
+                        fmt.Sprintf("%s", host.Hostname),
+                        fmt.Sprintf("%s", host.BootClass),
+                })
+        }
+        return formatter.FormatRows(rows), nil
+}
+
+// PrintHostList pretty-prints a list of hosts.
+func PrintHostList(hostList *crimson.HostList, format FormatType) {
+        lines, err := FormatHostList(hostList, format)
+        if err != nil {
+                fmt.Fprintf(os.Stderr, "ERROR: %s", err)
+                return
+        }
+        for _, s := range lines {
+                fmt.Println(s)
+        }
+}