From 5ebbc6bb0ec6b95a5a78c9dc4e8a79a01a05d864 Mon Sep 17 00:00:00 2001
From: Krishna Srinivas <krishna.srinivas@gmail.com>
Date: Wed, 23 Sep 2015 02:27:13 -0700
Subject: [PATCH] Add DiscoverServers and GetControllerNetInfo controller APIs

---
 controller-rpc.go  | 59 ++++++++++++++++++++++++++++++++++++++++++++++
 rpc-definitions.go | 25 +++++++++++++++++++-
 2 files changed, 83 insertions(+), 1 deletion(-)

diff --git a/controller-rpc.go b/controller-rpc.go
index fb5d0d498..450f257cd 100644
--- a/controller-rpc.go
+++ b/controller-rpc.go
@@ -211,6 +211,65 @@ func (s *controllerRPCService) AddServer(r *http.Request, args *ControllerArgs,
 	return nil
 }
 
+func (s *controllerRPCService) DiscoverServers(r *http.Request, args *DiscoverArgs, rep *DiscoverRep) error {
+	c := make(chan DiscoverRepEntry)
+	defer close(c)
+	for _, host := range args.Hosts {
+		go func(c chan DiscoverRepEntry, host string) {
+			u := &url.URL{}
+			if args.SSL {
+				u.Scheme = "https"
+			} else {
+				u.Scheme = "http"
+			}
+			if args.Port != 0 {
+				u.Host = host + ":" + string(args.Port)
+			} else {
+				u.Host = host + ":9002"
+			}
+			u.Path = "/rpc"
+
+			op := rpcOperation{
+				Method:  "Server.Version",
+				Request: ServerArg{},
+			}
+			versionrep := VersionRep{}
+			request, err := newRPCRequest(u.String(), op, nil)
+			if err != nil {
+				c <- DiscoverRepEntry{host, err.ToGoError().Error()}
+				return
+			}
+			var resp *http.Response
+			resp, err = request.Do()
+			if err != nil {
+				c <- DiscoverRepEntry{host, err.ToGoError().Error()}
+				return
+			}
+			if err := json.DecodeClientResponse(resp.Body, &versionrep); err != nil {
+				c <- DiscoverRepEntry{host, err.Error()}
+				return
+			}
+			c <- DiscoverRepEntry{host, ""}
+		}(c, host)
+	}
+	for range args.Hosts {
+		entry := <-c
+		rep.Entry = append(rep.Entry, entry)
+	}
+	return nil
+}
+
+func (s *controllerRPCService) GetControllerNetInfo(r *http.Request, args *ServerArg, res *ControllerNetInfoRep) error {
+	addrs, err := net.InterfaceAddrs()
+	if err != nil {
+		return err
+	}
+	for _, addr := range addrs {
+		res.NetInfo = append(res.NetInfo, addr.String())
+	}
+	return nil
+}
+
 func (s *controllerRPCService) GetServerMemStats(r *http.Request, args *ControllerArgs, res *MemStatsRep) error {
 	err := proxyRequest("Server.MemStats", args.Host, args.SSL, res)
 	if err != nil {
diff --git a/rpc-definitions.go b/rpc-definitions.go
index 98b9e6dea..f9860086a 100644
--- a/rpc-definitions.go
+++ b/rpc-definitions.go
@@ -46,6 +46,11 @@ type ServerRep struct {
 	ID   string `json:"id"`
 }
 
+// ControllerNetInfoRep array of ip/mask in the form: 172.17.42.1/16
+type ControllerNetInfoRep struct {
+	NetInfo []string `json:"netinfo"`
+}
+
 // DefaultRep default reply
 type DefaultRep struct {
 	Error   error  `json:"error"`
@@ -111,7 +116,25 @@ type AuthRep struct {
 	SecretAccessKey string `json:"secretAccessKey"`
 }
 
-// BucketStats bucket name and total used by the bucket
+// DiscoverArgs array of IP addresses / names to discover
+type DiscoverArgs struct {
+	Hosts []string `json:"hosts"`
+	Port  int      `json:"port"`
+	SSL   bool     `json:"bool"`
+}
+
+// DiscoverRepEntry : Error is "" if there is no error
+type DiscoverRepEntry struct {
+	Host  string `json:"host"`
+	Error string `json:"error"`
+}
+
+// DiscoverRep list of discovered hosts
+type DiscoverRep struct {
+	Entry []DiscoverRepEntry `json:"entry"`
+}
+
+// BucketStats bucket-name and storage used
 type BucketStats struct {
 	Name string `json:"name"`
 	Used uint64 `json:"used"`