Adding support for listing a user keys

api/handler.go

@@ -94,6 +94,21 @@ func RemoveKey(w http.ResponseWriter, r *http.Request) {
 	fmt.Fprintf(w, "Key \"%s\" successfully removed", kName)
 }
 
+func ListKeys(w http.ResponseWriter, r *http.Request) {
+	uName := r.URL.Query().Get(":name")
+	keys, err := user.ListKeys(uName)
+	if err != nil {
+		http.Error(w, err.Error(), http.StatusNotFound)
+		return
+	}
+	out, err := json.Marshal(&keys)
+	if err != nil {
+		http.Error(w, err.Error(), http.StatusInternalServerError)
+		return
+	}
+	w.Write(out)
+}
+
 func NewUser(w http.ResponseWriter, r *http.Request) {
 	var usr user.User
 	if err := parseBody(r.Body, &usr); err != nil {

api/handler_test.go

@@ -390,6 +390,25 @@ func (s *S) TestRemoveKeyShouldReturnErrorWithLineBreakAtEnd(c *C) {
 	c.Assert(b, Equals, "User \"Gandalf\" does not exists\n")
 }
 
+func (s *S) TestListKeysGivesExpectedSuccessResponse(c *C) {
+	keys := map[string]string{"key1": "ssh-key somekey gandalf@host1", "key2": "ssh-key somekey gandalf@host2"}
+	u, err := user.New("Gandalf", keys)
+	c.Assert(err, IsNil)
+	defer db.Session.User().RemoveId(u.Name)
+	url := "/user/Gandalf/keys?:name=Gandalf"
+	request, err := http.NewRequest("GET", url, nil)
+	c.Assert(err, IsNil)
+	recorder := httptest.NewRecorder()
+	ListKeys(recorder, request)
+	c.Assert(recorder.Code, Equals, 200)
+	body, err := ioutil.ReadAll(recorder.Body)
+	c.Assert(err, IsNil)
+	var data map[string]string
+	err = json.Unmarshal(body, &data)
+	c.Assert(err, IsNil)
+	c.Assert(data, DeepEquals, keys)
+}
+
 func (s *S) TestRemoveUser(c *C) {
 	u, err := user.New("username", map[string]string{})
 	c.Assert(err, IsNil)

user/user.go

@@ -123,3 +123,14 @@ func RemoveKey(uName, kName string) error {
 	}
 	return removeKey(kContent, uName)
 }
+
+// ListKeys lists all user's keys
+//
+// If the user is not found, returns an error
+func ListKeys(uName string) (map[string]string, error) {
+	var u User
+	if err := db.Session.User().FindId(uName).One(&u); err != nil {
+		return nil, fmt.Errorf(`User "%s" does not exists`, uName)
+	}
+	return u.Keys, nil
+}