Impl ContributionConllectionServiceServer

Author: gedorinku

app/run.go

@@ -61,7 +61,7 @@ func Run() error {
 			server.NewRoleServiceServer(store),
 			server.NewDepartmentServiceServer(store),
 			server.NewInvitationServiceServer(store, cli),
-			server.NewContributionConllectionServiceServer(store),
+			server.NewContributionConllectionServiceServer(store, cfg),
 		),
 	)
 

app/server/contribution_conllections_server.go

@@ -4,11 +4,13 @@ import (
 	"context"
 
 	"github.com/izumin5210/grapi/pkg/grapiserver"
-	"google.golang.org/grpc/codes"
-	"google.golang.org/grpc/status"
+	"google.golang.org/grpc/grpclog"
 
 	api_pb "github.com/ProgrammingLab/prolab-accounts/api"
+	"github.com/ProgrammingLab/prolab-accounts/app/config"
 	"github.com/ProgrammingLab/prolab-accounts/app/di"
+	"github.com/ProgrammingLab/prolab-accounts/infra/record"
+	"github.com/ProgrammingLab/prolab-accounts/model"
 )
 
 // ContributionConllectionServiceServer is a composite interface of api_pb.ContributionConllectionServiceServer and grapiserver.Server.
@@ -18,17 +20,67 @@ type ContributionConllectionServiceServer interface {
 }
 
 // NewContributionConllectionServiceServer creates a new ContributionConllectionServiceServer instance.
-func NewContributionConllectionServiceServer(store di.StoreComponent) ContributionConllectionServiceServer {
+func NewContributionConllectionServiceServer(store di.StoreComponent, cfg *config.Config) ContributionConllectionServiceServer {
 	return &contributionConllectionServiceServerImpl{
 		StoreComponent: store,
+		cfg:            cfg,
 	}
 }
 
 type contributionConllectionServiceServerImpl struct {
 	di.StoreComponent
+	cfg *config.Config
 }
 
 func (s *contributionConllectionServiceServerImpl) ListContributionConllections(ctx context.Context, req *api_pb.ListContributionConllectionsRequest) (*api_pb.ListContributionConllectionsResponse, error) {
-	// TODO: Not yet implemented.
-	return nil, status.Error(codes.Unimplemented, "TODO: You should implement it!")
+	gs := s.GitHubStore(ctx)
+	grpclog.Info(req.GetUsersCount())
+	cols, err := gs.ListContributionCollections(int(req.GetUsersCount()))
+	if err != nil {
+		return nil, err
+	}
+
+	resp := contributionConllectionsToResponse(cols, s.cfg)
+	return &api_pb.ListContributionConllectionsResponse{
+		ContributionConllections: resp,
+	}, nil
+}
+
+func contributionConllectionsToResponse(cols []*model.GitHubContributionCollection, cfg *config.Config) []*api_pb.ContributionConllection {
+	resp := make([]*api_pb.ContributionConllection, 0, len(cols))
+	for _, c := range cols {
+		rc := contributionConllectionToResponse(c, cfg)
+		if rc == nil {
+			continue
+		}
+		resp = append(resp, rc)
+	}
+
+	return resp
+}
+
+func contributionConllectionToResponse(col *model.GitHubContributionCollection, cfg *config.Config) *api_pb.ContributionConllection {
+	if len(col.Days) == 0 {
+		return nil
+	}
+
+	u := col.Days[0].R.User
+	return &api_pb.ContributionConllection{
+		User:       userToResponse(u, false, cfg),
+		TotalCount: int32(col.TotalCount),
+		Days:       contributionDaysToResponse(col.Days),
+	}
+}
+
+func contributionDaysToResponse(days []*record.GithubContributionDay) []*api_pb.ContributionDay {
+	resp := make([]*api_pb.ContributionDay, 0, len(days))
+	for _, d := range days {
+		rd := &api_pb.ContributionDay{
+			Date:  timeToResponse(d.Date),
+			Count: int32(d.Count),
+		}
+		resp = append(resp, rd)
+	}
+
+	return resp
 }