Merge f6fb2bd into dbdff0b

proto/wantslist.proto

@@ -2,6 +2,10 @@ syntax = "proto3";
 
 package wantslist;
 
+message Config {
+  repeated WantList lists = 1;
+}
+
 message WantListEntry {
   int32 want = 1;
   int32 index = 2; 
@@ -13,7 +17,9 @@ message WantList {
 }
 
 
-message AddWantListRequest {}
+message AddWantListRequest {
+  WantList add = 1;
+}
 
 message AddWantListResponse {}
 
@@ -23,5 +29,5 @@ message GetWantListResponse {}
 
 service WantService {
   rpc AddWantList(AddWantListRequest) returns (AddWantListResponse) {};
-  rpc GetWantList(GetWantListRequest) returns (GetWantListResponse) {};
+  //rpc GetWantList(GetWantListRequest) returns (GetWantListResponse) {};
 }

wantslist.go

@@ -11,17 +11,25 @@ import (
 	"google.golang.org/grpc"
 
 	pbg "github.com/brotherlogic/goserver/proto"
+	pb "github.com/brotherlogic/wantslist/proto"
+)
+
+const (
+	// KEY - where the wantslists are stored
+	KEY = "/github.com/brotherlogic/wantslist/config"
 )
 
 //Server main server type
 type Server struct {
 	*goserver.GoServer
+	config *pb.Config
 }
 
 // Init builds the server
 func Init() *Server {
 	s := &Server{
 		&goserver.GoServer{},
+		&pb.Config{},
 	}
 	return s
 }
@@ -36,8 +44,29 @@ func (s *Server) ReportHealth() bool {
 	return true
 }
 
+func (s *Server) save(ctx context.Context) {
+	s.KSclient.Save(ctx, KEY, s.config)
+}
+
+func (s *Server) load(ctx context.Context) error {
+	config := &pb.Config{}
+	data, _, err := s.KSclient.Read(ctx, KEY, config)
+
+	if err != nil {
+		return err
+	}
+
+	s.config = data.(*pb.Config)
+	return nil
+}
+
 // Mote promotes/demotes this server
 func (s *Server) Mote(ctx context.Context, master bool) error {
+	if master {
+		err := s.load(ctx)
+		return err
+	}
+
 	return nil
 }
 

wantslistapi.go

@@ -0,0 +1,10 @@
+package main
+
+import "golang.org/x/net/context"
+import pb "github.com/brotherlogic/wantslist/proto"
+
+//AddWantList adds a want list
+func (s *Server) AddWantList(ctx context.Context, req *pb.AddWantListRequest) (*pb.AddWantListResponse, error) {
+	s.config.Lists = append(s.config.Lists, req.Add)
+	return &pb.AddWantListResponse{}, nil
+}

wantslistapi_test.go

@@ -0,0 +1,18 @@
+package main
+
+import (
+	"context"
+	"testing"
+
+	pb "github.com/brotherlogic/wantslist/proto"
+)
+
+func TestAddWantsList(t *testing.T) {
+	s := InitTestServer()
+
+	s.AddWantList(context.Background(), &pb.AddWantListRequest{Add: &pb.WantList{Name: "hello"}})
+
+	if len(s.config.Lists) != 1 {
+		t.Errorf("Wrong number of lists: %v", len(s.config.Lists))
+	}
+}