yq/yaml.go

211 lines
5.1 KiB
Go
Raw Normal View History

2015-09-26 22:15:49 +00:00
package main
import (
2015-09-28 02:00:38 +00:00
"fmt"
2015-10-10 01:57:11 +00:00
"github.com/mikefarah/yaml/Godeps/_workspace/src/github.com/spf13/cobra"
"github.com/mikefarah/yaml/Godeps/_workspace/src/gopkg.in/yaml.v2"
2015-09-28 02:00:38 +00:00
"io/ioutil"
"os"
"strconv"
"strings"
2015-09-26 22:15:49 +00:00
)
2015-10-06 05:07:09 +00:00
var trimOutput = true
var writeInplace = false
var writeScript = ""
2015-10-11 06:06:52 +00:00
var inputJSON = false
2015-10-10 23:00:22 +00:00
var outputToJSON = false
2015-10-06 05:07:09 +00:00
2015-09-26 22:15:49 +00:00
func main() {
2015-10-13 10:42:36 +00:00
var cmdRead = createReadCmd()
var cmdWrite = createWriteCmd()
var rootCmd = &cobra.Command{Use: "yaml"}
rootCmd.PersistentFlags().BoolVarP(&trimOutput, "trim", "t", true, "trim yaml output")
rootCmd.PersistentFlags().BoolVarP(&outputToJSON, "tojson", "j", false, "output as json")
rootCmd.PersistentFlags().BoolVarP(&inputJSON, "fromjson", "J", false, "input as json")
rootCmd.AddCommand(cmdRead, cmdWrite)
rootCmd.Execute()
}
func createReadCmd() *cobra.Command {
return &cobra.Command{
2015-10-06 05:07:09 +00:00
Use: "read [yaml_file] [path]",
Aliases: []string{"r"},
Short: "yaml r sample.yaml a.b.c",
Example: `
yaml read things.yaml a.b.c
yaml r - a.b.c (reads from stdin)
yaml r things.yaml a.*.c
yaml r things.yaml a.array[0].blah
yaml r things.yaml a.array[*].blah
2015-10-13 10:42:36 +00:00
`,
2015-10-06 05:07:09 +00:00
Long: "Outputs the value of the given path in the yaml file to STDOUT",
Run: readProperty,
2015-09-28 02:00:38 +00:00
}
2015-10-13 10:42:36 +00:00
}
2015-09-26 23:38:44 +00:00
2015-10-13 10:42:36 +00:00
func createWriteCmd() *cobra.Command {
2015-10-06 05:07:09 +00:00
var cmdWrite = &cobra.Command{
Use: "write [yaml_file] [path] [value]",
Aliases: []string{"w"},
Short: "yaml w [--inplace/-i] [--script/-s script_file] sample.yaml a.b.c newValueForC",
2015-10-06 05:07:09 +00:00
Example: `
yaml write things.yaml a.b.c cat
yaml write --inplace things.yaml a.b.c cat
yaml w -i things.yaml a.b.c cat
yaml w --script update_script.yaml things.yaml
yaml w -i -s update_script.yaml things.yaml
2015-10-13 10:42:36 +00:00
`,
2015-10-06 05:07:09 +00:00
Long: `Updates the yaml file w.r.t the given path and value.
Outputs to STDOUT unless the inplace flag is used, in which case the file is updated instead.
Update Scripts:
Note that you can give an update script to perform more sophisticated updated. Update script
format is a yaml map where the key is the path and the value is..well the value. e.g.:
---
a.b.c: true,
a.b.e:
- name: bob
`,
2015-10-06 05:07:09 +00:00
Run: writeProperty,
}
cmdWrite.PersistentFlags().BoolVarP(&writeInplace, "inplace", "i", false, "update the yaml file inplace")
cmdWrite.PersistentFlags().StringVarP(&writeScript, "script", "s", "", "yaml script for updating yaml")
2015-10-13 10:42:36 +00:00
return cmdWrite
2015-10-06 05:07:09 +00:00
}
2015-09-29 06:29:32 +00:00
2015-10-06 05:07:09 +00:00
func readProperty(cmd *cobra.Command, args []string) {
2015-10-10 23:00:22 +00:00
print(read(args))
}
func read(args []string) interface{} {
var parsedData map[interface{}]interface{}
2015-09-29 06:29:32 +00:00
2015-10-11 06:06:52 +00:00
readData(args[0], &parsedData, inputJSON)
2015-09-27 03:18:33 +00:00
2015-10-06 05:07:09 +00:00
if len(args) == 1 {
return parsedData
2015-09-29 06:29:32 +00:00
}
2015-10-06 05:07:09 +00:00
var paths = parsePath(args[1])
2015-09-27 03:18:33 +00:00
return readMap(parsedData, paths[0], paths[1:len(paths)])
2015-09-29 00:56:12 +00:00
}
2015-10-06 05:07:09 +00:00
func writeProperty(cmd *cobra.Command, args []string) {
updatedData := updateYaml(args)
if writeInplace {
ioutil.WriteFile(args[0], []byte(yamlToString(updatedData)), 0644)
} else {
2015-10-10 23:00:22 +00:00
print(updatedData)
}
}
func updateYaml(args []string) interface{} {
var writeCommands map[string]interface{}
if writeScript != "" {
2015-10-11 06:06:52 +00:00
readData(writeScript, &writeCommands, false)
} else if len(args) < 3 {
2015-10-06 05:39:19 +00:00
die("Must provide <filename> <path_to_update> <value>")
} else {
writeCommands = make(map[string]interface{})
writeCommands[args[1]] = parseValue(args[2])
2015-09-29 06:29:32 +00:00
}
2015-10-06 05:07:09 +00:00
var parsedData map[interface{}]interface{}
2015-10-11 06:06:52 +00:00
readData(args[0], &parsedData, inputJSON)
for path, value := range writeCommands {
var paths = parsePath(path)
write(parsedData, paths[0], paths[1:len(paths)], value)
}
return parsedData
2015-10-06 05:07:09 +00:00
}
func parseValue(argument string) interface{} {
var value, err interface{}
2015-10-03 07:25:13 +00:00
var inQuotes = argument[0] == '"'
if !inQuotes {
value, err = strconv.ParseFloat(argument, 64)
if err == nil {
return value
}
value, err = strconv.ParseBool(argument)
if err == nil {
return value
}
2015-10-03 07:25:13 +00:00
return argument
}
2015-10-03 07:25:13 +00:00
return argument[1 : len(argument)-1]
}
2015-10-10 23:00:22 +00:00
func print(context interface{}) {
var out string
if outputToJSON {
out = jsonToString(context)
} else {
out = yamlToString(context)
}
fmt.Println(out)
}
func yamlToString(context interface{}) string {
2015-09-29 01:05:28 +00:00
out, err := yaml.Marshal(context)
if err != nil {
2015-10-06 05:39:19 +00:00
die("error printing yaml: %v", err)
2015-09-29 01:05:28 +00:00
}
2015-09-29 06:29:32 +00:00
outStr := string(out)
// trim the trailing new line as it's easier for a script to add
// it in if required than to remove it
2015-10-06 05:07:09 +00:00
if trimOutput {
return strings.Trim(outStr, "\n ")
}
return outStr
2015-09-27 03:18:33 +00:00
}
2015-10-11 06:06:52 +00:00
func readData(filename string, parsedData interface{}, readAsJSON bool) {
if filename == "" {
2015-10-06 05:39:19 +00:00
die("Must provide filename")
2015-09-28 02:00:38 +00:00
}
2015-10-05 04:48:34 +00:00
var rawData []byte
if filename == "-" {
2015-10-05 04:48:34 +00:00
rawData = readStdin()
} else {
rawData = readFile(filename)
2015-10-05 04:48:34 +00:00
}
2015-09-26 22:20:42 +00:00
2015-10-11 06:06:52 +00:00
var err interface{}
if readAsJSON {
fromJSONBytes([]byte(rawData), parsedData.(*map[interface{}]interface{}))
} else {
err = yaml.Unmarshal([]byte(rawData), parsedData)
}
2015-09-28 02:00:38 +00:00
if err != nil {
2015-10-11 06:06:52 +00:00
die("error parsing data: ", err)
2015-09-28 02:00:38 +00:00
}
2015-09-27 03:18:33 +00:00
}
2015-09-26 22:15:49 +00:00
2015-10-05 04:48:34 +00:00
func readStdin() []byte {
bytes, err := ioutil.ReadAll(os.Stdin)
if err != nil {
2015-10-06 05:39:19 +00:00
die("error reading stdin", err)
2015-10-05 04:48:34 +00:00
}
return bytes
}
func readFile(filename string) []byte {
var rawData, readError = ioutil.ReadFile(filename)
if readError != nil {
2015-10-06 05:39:19 +00:00
die("error: %v", readError)
2015-09-28 02:00:38 +00:00
}
return rawData
2015-09-27 03:11:10 +00:00
}
2015-10-06 05:39:19 +00:00
func die(message ...interface{}) {
fmt.Println(message)
os.Exit(1)
}