From 8827093bc65bbd1a1121b6508110bc733910571d Mon Sep 17 00:00:00 2001
From: Chris Goller <goller@gmail.com>
Date: Thu, 23 Mar 2017 13:16:02 -0700
Subject: [PATCH] Update databases/retention policy link responses

---
 server/databases.go | 34 +++++++++++++++++++++++++---------
 server/sources.go   |  2 ++
 2 files changed, 27 insertions(+), 9 deletions(-)

diff --git a/server/databases.go b/server/databases.go
index e51afc86f..ccad99a60 100644
--- a/server/databases.go
+++ b/server/databases.go
@@ -10,8 +10,8 @@ import (
 )
 
 type dbLinks struct {
-	Self string `json:"self"` // Self link mapping to this resource
-	RPs  string `json:"rps"`  // URL for retention policies for this database
+	Self string `json:"self"`              // Self link mapping to this resource
+	RPs  string `json:"retentionPolicies"` // URL for retention policies for this database
 }
 
 type dbResponse struct {
@@ -22,6 +22,16 @@ type dbResponse struct {
 	Links         dbLinks `json:"links"`                   // Links are URI locations related to the database
 }
 
+func NewDBResponse(srcID int, name string) dbResponse {
+	base := "/chronograf/v1/sources"
+	return dbResponse{
+		Name: name,
+		Links: dbLinks{
+			Self: fmt.Sprintf("%s/%d/dbs/%s", base, srcID, name),
+		},
+	}
+}
+
 type dbsResponse struct {
 	Databases []dbResponse `json:"databases"`
 }
@@ -39,6 +49,13 @@ type rpResponse struct {
 	Links         rpLinks `json:"links"`         // Links are URI locations related to the database
 }
 
+func (r *rpResponse) WithLinks(srcID int, dbName string) {
+	base := "/chronograf/v1/sources"
+	r.Links = rpLinks{
+		Self: fmt.Sprintf("%s/%d/dbs/%s/rps/%s", base, srcID, dbName, r.Name),
+	}
+}
+
 type rpsResponse struct {
 	RetentionPolicies []rpResponse `json:"retentionPolicies"`
 }
@@ -60,7 +77,6 @@ func (h *Service) GetDatabases(w http.ResponseWriter, r *http.Request) {
 	}
 
 	db := h.Databases
-
 	if err = db.Connect(ctx, &src); err != nil {
 		msg := fmt.Sprintf("Unable to connect to source %d: %v", srcID, err)
 		Error(w, http.StatusBadRequest, msg, h.Logger)
@@ -75,9 +91,7 @@ func (h *Service) GetDatabases(w http.ResponseWriter, r *http.Request) {
 
 	dbs := make([]dbResponse, len(databases))
 	for i, d := range databases {
-		dbs[i] = dbResponse{
-			Name: d.Name,
-		}
+		dbs[i] = NewDBResponse(srcID, d.Name)
 	}
 
 	res := dbsResponse{
@@ -127,7 +141,7 @@ func (h *Service) NewDatabase(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
-	res := dbResponse{Name: database.Name}
+	res := NewDBResponse(srcID, database.Name)
 	encodeJSON(w, http.StatusCreated, res, h.Logger)
 }
 
@@ -198,13 +212,15 @@ func (h *Service) RetentionPolicies(w http.ResponseWriter, r *http.Request) {
 
 	rps := make([]rpResponse, len(allRP))
 	for i, rp := range allRP {
-		rps[i] = rpResponse{
+		rp := rpResponse{
 			Name:          rp.Name,
 			Duration:      rp.Duration,
 			Replication:   rp.Replication,
 			ShardDuration: rp.ShardDuration,
 			Default:       rp.Default,
 		}
+		rp.WithLinks(srcID, dbID)
+		rps[i] = rp
 	}
 
 	res := rpsResponse{
@@ -302,7 +318,7 @@ func (h *Service) UpdateRetentionPolicy(w http.ResponseWriter, r *http.Request)
 		return
 	}
 
-  // TODO: this needs to be the actual RP information
+	// TODO: this needs to be the actual RP information
 	res := rpResponse{Name: rp.Name}
 	encodeJSON(w, http.StatusCreated, res, h.Logger)
 }
diff --git a/server/sources.go b/server/sources.go
index 55c505229..d56a85ecd 100644
--- a/server/sources.go
+++ b/server/sources.go
@@ -18,6 +18,7 @@ type sourceLinks struct {
 	Permissions string `json:"permissions"`     // URL for all allowed permissions for this source
 	Users       string `json:"users"`           // URL for all users associated with this source
 	Roles       string `json:"roles,omitempty"` // URL for all users associated with this source
+	Databases   string `json:"databases"`       // URL for the databases contained within this soure
 }
 
 type sourceResponse struct {
@@ -43,6 +44,7 @@ func newSourceResponse(src chronograf.Source) sourceResponse {
 			Kapacitors:  fmt.Sprintf("%s/%d/kapacitors", httpAPISrcs, src.ID),
 			Permissions: fmt.Sprintf("%s/%d/permissions", httpAPISrcs, src.ID),
 			Users:       fmt.Sprintf("%s/%d/users", httpAPISrcs, src.ID),
+			Databases:   fmt.Sprintf("%s/%d/dbs", httpAPISrcs, src.ID),
 		},
 	}