1 files changed, 114 insertions, 0 deletions
diff --git a/client/go/cmd/document.go b/client/go/cmd/document.go
new file mode 100644
index 00000000000..f2d7b80289d
--- /dev/null
+++ b/client/go/cmd/document.go
@@ -0,0 +1,114 @@
+// Copyright Verizon Media. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+// vespa document command
+// author: bratseth
+
+package cmd
+
+import (
+    "bytes"
+    "encoding/json"
+    "github.com/spf13/cobra"
+    "github.com/vespa-engine/vespa/util"
+    "io/ioutil"
+    "net/http"
+    "net/url"
+    "os"
+    "strings"
+    "time"
+)
+
+func init() {
+    rootCmd.AddCommand(documentCmd)
+    documentCmd.AddCommand(documentPostCmd)
+    documentCmd.AddCommand(documentGetCmd)
+}
+
+var documentCmd = &cobra.Command{
+    Use:   "document",
+    Short: "Issue document operations",
+    Long:  `TODO: Example vespa document mynamespace/mydocumenttype/myid document.json`,
+    // TODO: Check args
+    Run: func(cmd *cobra.Command, args []string) {
+        post("", args[0])
+    },
+}
+
+var documentPostCmd = &cobra.Command{
+    Use:   "post",
+    Short: "Posts the document in the given file",
+    Long:  `TODO`,
+    // TODO: Check args
+    Run: func(cmd *cobra.Command, args []string) {
+        if len(args) == 1 {
+            post("", args[0])
+        } else {
+            post(args[0], args[1])
+        }
+    },
+}
+
+var documentGetCmd = &cobra.Command{
+    Use:   "get",
+    Short: "Gets a document",
+    Long:  `TODO`,
+    // TODO: Check args
+    Run: func(cmd *cobra.Command, args []string) {
+        get(args[0])
+    },
+}
+
+func get(documentId string) {
+    // TODO
+}
+
+func post(documentId string, jsonFile string) {
+    header := http.Header{}
+    header.Add("Content-Type", "application/json")
+
+    fileReader, fileError := os.Open(jsonFile)
+    if fileError != nil {
+        util.Error("Could not open file at " + jsonFile)
+        util.Detail(fileError.Error())
+        return
+    }
+
+    documentData := util.ReaderToBytes(fileReader)
+
+    if documentId == "" {
+        var doc map[string]interface{}
+        json.Unmarshal(documentData, &doc)
+        documentId = doc["id"].(string)
+        if documentId == "" {
+            documentId = doc["put"].(string) // document feeder format
+        }
+        if documentId == "" {
+            util.Error("No document id given neither as argument or an 'id' key in the json file")
+            return
+        }
+    }
+
+    url, _ := url.Parse(getTarget(documentContext).document + "/document/v1/" + documentId)
+
+    request := &http.Request{
+        URL: url,
+        Method: "POST",
+        Header: header,
+        Body: ioutil.NopCloser(bytes.NewReader(documentData)),
+    }
+    serviceDescription := "Container (document API)"
+    response := util.HttpDo(request, time.Second * 60, serviceDescription)
+    if (response == nil) {
+        return
+    }
+
+    defer response.Body.Close()
+    if response.StatusCode == 200 {
+        util.Success("Success") // TODO: Change to something including document id
+    } else if response.StatusCode / 100 == 4 {
+        util.Error("Invalid document (" + response.Status + "):")
+        util.PrintReader(response.Body)
+    } else {
+        util.Error("Error from", strings.ToLower(serviceDescription), "at", request.URL.Host, "(" + response.Status + "):")
+        util.PrintReader(response.Body)
+    }
+}