From ab7e4b8a4bf0b53ccf09b0a8e05daadf89c710fa Mon Sep 17 00:00:00 2001
From: Simone Gotti <simone.gotti@gmail.com>
Date: Fri, 3 May 2019 00:11:11 +0200
Subject: [PATCH] gateway: add user remote repos API

---
 internal/services/gateway/api/remoterepo.go | 113 ++++++++++++++++++++
 internal/services/gateway/gateway.go        |   7 +-
 2 files changed, 118 insertions(+), 2 deletions(-)
 create mode 100644 internal/services/gateway/api/remoterepo.go

diff --git a/internal/services/gateway/api/remoterepo.go b/internal/services/gateway/api/remoterepo.go
new file mode 100644
index 0000000..60a3a85
--- /dev/null
+++ b/internal/services/gateway/api/remoterepo.go
@@ -0,0 +1,113 @@
+// Copyright 2019 Sorint.lab
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package api
+
+import (
+	"net/http"
+
+	gitsource "github.com/sorintlab/agola/internal/gitsources"
+	csapi "github.com/sorintlab/agola/internal/services/configstore/api"
+	"github.com/sorintlab/agola/internal/services/gateway/command"
+	"github.com/sorintlab/agola/internal/services/types"
+	"github.com/sorintlab/agola/internal/util"
+
+	"github.com/gorilla/mux"
+	"github.com/pkg/errors"
+	"go.uber.org/zap"
+)
+
+type RemoteRepoResponse struct {
+	ID   string `json:"id,omitempty"`
+	Path string `json:"path,omitempty"`
+}
+
+func createRemoteRepoResponse(r *gitsource.RepoInfo) *RemoteRepoResponse {
+	rr := &RemoteRepoResponse{
+		ID:   r.ID,
+		Path: r.Path,
+	}
+
+	return rr
+}
+
+type UserRemoteReposHandler struct {
+	log               *zap.SugaredLogger
+	ch                *command.CommandHandler
+	configstoreClient *csapi.Client
+}
+
+func NewUserRemoteReposHandler(logger *zap.Logger, ch *command.CommandHandler, configstoreClient *csapi.Client) *UserRemoteReposHandler {
+	return &UserRemoteReposHandler{log: logger.Sugar(), ch: ch, configstoreClient: configstoreClient}
+}
+
+func (h *UserRemoteReposHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
+	ctx := r.Context()
+	vars := mux.Vars(r)
+	remoteSourceID := vars["remotesourceid"]
+
+	userIDVal := ctx.Value("userid")
+	if userIDVal == nil {
+		httpError(w, util.NewErrBadRequest(errors.Errorf("user not authenticated")))
+		return
+	}
+	userID := userIDVal.(string)
+	h.log.Infof("userID: %q", userID)
+
+	user, resp, err := h.configstoreClient.GetUser(ctx, userID)
+	if httpErrorFromRemote(w, resp, err) {
+		h.log.Errorf("err: %+v", err)
+		return
+	}
+
+	rs, resp, err := h.configstoreClient.GetRemoteSource(ctx, remoteSourceID)
+	if httpErrorFromRemote(w, resp, err) {
+		h.log.Errorf("err: %+v", err)
+		return
+	}
+	h.log.Infof("rs: %s", util.Dump(rs))
+
+	var la *types.LinkedAccount
+	for _, v := range user.LinkedAccounts {
+		if v.RemoteSourceID == rs.ID {
+			la = v
+			break
+		}
+	}
+	h.log.Infof("la: %s", util.Dump(la))
+	if la == nil {
+		httpError(w, util.NewErrBadRequest(errors.Errorf("user doesn't have a linked account for remote source %q", rs.Name)))
+		return
+	}
+
+	gitsource, err := h.ch.GetGitSource(ctx, rs, user.Name, la)
+	if err != nil {
+		httpError(w, util.NewErrBadRequest(errors.Wrapf(err, "failed to create gitsource client")))
+		return
+	}
+
+	remoteRepos, err := gitsource.ListUserRepos()
+	if err != nil {
+		httpError(w, util.NewErrBadRequest(errors.Wrapf(err, "failed to get user repositories from gitsource")))
+		return
+	}
+
+	repos := make([]*RemoteRepoResponse, len(remoteRepos))
+	for i, r := range remoteRepos {
+		repos[i] = createRemoteRepoResponse(r)
+	}
+	if err := httpResponse(w, http.StatusOK, repos); err != nil {
+		h.log.Errorf("err: %+v", err)
+	}
+}
diff --git a/internal/services/gateway/gateway.go b/internal/services/gateway/gateway.go
index e8df76c..8067c3e 100644
--- a/internal/services/gateway/gateway.go
+++ b/internal/services/gateway/gateway.go
@@ -197,6 +197,7 @@ func (g *Gateway) Run(ctx context.Context) error {
 	logsHandler := api.NewLogsHandler(logger, g.runserviceClient)
 
 	reposHandler := api.NewReposHandler(logger, g.c.GitServerURL)
+	userRemoteReposHandler := api.NewUserRemoteReposHandler(logger, g.ch, g.configstoreClient)
 
 	loginUserHandler := api.NewLoginUserHandler(logger, g.ch)
 	authorizeHandler := api.NewAuthorizeHandler(logger, g.ch)
@@ -268,13 +269,15 @@ func (g *Gateway) Run(ctx context.Context) error {
 	apirouter.Handle("/runs/{runid}/tasks/{taskid}/actions", runTaskActionsHandler).Methods("PUT")
 	apirouter.Handle("/runs", authForcedHandler(runsHandler)).Methods("GET")
 
+	router.Handle("/repos/{rest:.*}", reposHandler).Methods("GET", "POST")
+
+	apirouter.Handle("/user/remoterepos/{remotesourceid}", authForcedHandler(userRemoteReposHandler)).Methods("GET")
+
 	router.Handle("/login", loginUserHandler).Methods("POST")
 	router.Handle("/authorize", authorizeHandler).Methods("POST")
 	router.Handle("/register", registerHandler).Methods("POST")
 	router.Handle("/oauth2/callback", oauth2callbackHandler).Methods("GET")
 
-	router.Handle("/repos/{rest:.*}", reposHandler).Methods("GET", "POST")
-
 	router.Handle("/webhooks", webhooksHandler).Methods("POST")
 	router.PathPrefix("/").HandlerFunc(handlers.NewWebBundleHandlerFunc(g.c.APIExposedURL))