From b7259bc293641fbe83e44d0f886cea52b8816769 Mon Sep 17 00:00:00 2001
From: nsz <nsz@port70.net>
Date: Wed, 21 Dec 2011 03:46:27 +0100
Subject: [PATCH] epoint-client += query documents by id

---
 cmd/epoint-client/epoint-client.go | 133 ++++++++++++++++++++++++++++-
 1 file changed, 132 insertions(+), 1 deletion(-)

diff --git a/cmd/epoint-client/epoint-client.go b/cmd/epoint-client/epoint-client.go
index 019a4d3..71bbe04 100644
--- a/cmd/epoint-client/epoint-client.go
+++ b/cmd/epoint-client/epoint-client.go
@@ -21,12 +21,13 @@ import (
 
 var db *store.Conn
 
-const usage = `usage: ./epoint-client [k|d|s|v|c] [args..] < [seed|document]
+const usage = `usage: ./epoint-client [i|h|d|q|s|v|c] [args..] < [seed|document]
 server is http://localhost:8080 by default
 
 i - make issuer key, use seed for generation, args: denomination
 h - make holder key, use seed for generation, args: issuer
 d - make draft, use seed for signing key, args: targetid value
+q - query document, args: k|d|c id [server]
 s - submit a (key|draft|cert) document, args: k|d|c [server]
 v - verify a document (prints body of the document if ok)
 c - connect to server and get server key, args: [server]
@@ -172,6 +173,122 @@ func d(r []byte, target, value string) (err error) {
 	return
 }
 
+func q(cmd, id, server string) (err error) {
+	log.Printf("document id: %s, server: %s", id, server)
+	m := map[string]string{
+		"k": "key",
+		"d": "draft",
+		"c": "cert",
+	}
+	k, ok := m[cmd]
+	if !ok {
+		err = fmt.Errorf("unknown query command: %s", cmd)
+		return
+	}
+	d, err := db.Get(k, id)
+	if err != nil {
+		if _, ok := err.(store.NotFoundError); !ok {
+			return
+		}
+	} else {
+		_, err = os.Stdout.Write(d)
+		log.Printf("found %s in local store", id)
+		return
+	}
+	resp, err := http.Get(server+"/"+k+"/"+id)
+	if err != nil {
+		return
+	}
+	d, err = readall(resp.Body)
+	if err != nil {
+		return
+	}
+	err = resp.Body.Close()
+	if err != nil {
+		return
+	}
+	_, err = os.Stdout.Write(d)
+	if resp.StatusCode != 200 {
+		err = fmt.Errorf("request failed: %s", resp.Status)
+		return
+	}
+	if err != nil {
+		return
+	}
+	log.Printf("got %s from the server", id)
+	switch cmd {
+	case "k":
+		e, err1 := key.Parse(d)
+		err = err1
+		if err != nil {
+			return
+		}
+		if id != key.Id(e) {
+			err = fmt.Errorf("id mismatch, expected %s, got %s", id, key.Id(e))
+			return
+		}
+		err = db.Set("key", id, d)
+	case "d":
+		i, s, err1 := document.Parse(d)
+		err = err1
+		if err != nil {
+			return
+		}
+		if id != document.Id(s) {
+			err = fmt.Errorf("id mismatch, expected %s, got %s", id, document.Id(s))
+			return
+		}
+		draft := i.(*document.Draft)
+		b, err1 := db.Get("key", draft.Drawer)
+		err = err1
+		if err != nil {
+			return
+		}
+		e, err1 := key.Parse(b)
+		err = err1
+		if err != nil {
+			return
+		}
+		err = document.Verify(s, openpgp.EntityList{e})
+		if err != nil {
+			return
+		}
+		err = db.Set("draft", id, d)
+	case "c":
+		i, s, err1 := document.Parse(d)
+		err = err1
+		if err != nil {
+			return
+		}
+		if id != document.Id(s) {
+			err = fmt.Errorf("id mismatch, expected %s, got %s", id, document.Id(s))
+			return
+		}
+		cert, err1 := document.ToCert(i)
+		err = err1
+		if err != nil {
+			return
+		}
+		// TODO: check serverkey
+		b, err1 := db.Get("key", cert.AuthorizedBy)
+		err = err1
+		if err != nil {
+			return
+		}
+		e, err1 := key.Parse(b)
+		err = err1
+		if err != nil {
+			return
+		}
+		err = document.Verify(s, openpgp.EntityList{e})
+		if err != nil {
+			return
+		}
+		err = db.Set("cert", id, d)
+	}
+	return
+}
+
 func s(d []byte, cmd, server string) (err error) {
 	m := map[string]string{
 		"k": "key",
@@ -402,6 +519,20 @@ func main() {
 			log.Fatal(usage)
 		}
 		err = d(read(), os.Args[2], os.Args[3])
+	case "q":
+		cmd := ""
+		id := ""
+		if len(os.Args) == 5 {
+			cmd = os.Args[2]
+			id = os.Args[3]
+			server = os.Args[4]
+		} else if len(os.Args) == 4 {
+			cmd = os.Args[2]
+			id = os.Args[3]
+		} else {
+			log.Fatal(usage)
+		}
+		err = q(cmd, id, server)
 	case "s":
 		cmd := ""
 		if len(os.Args) == 4 {
-- 
2.20.1