From b6c14f8b21a277576a882061fb772a76e3d4511b Mon Sep 17 00:00:00 2001
From: Robin Lambertz <robinlambertz+dev@gmail.com>
Date: Thu, 11 Aug 2016 20:23:25 +0200
Subject: [PATCH] Add AddCollaborator API Endpoint (#2780)

* Add AddCollaborator API Endpoint

* Add optional Permission to AddCollaborator endpoint

* Use APIContext
---
 routers/api/v1/api.go                |  1 +
 routers/api/v1/repo/collaborators.go | 49 ++++++++++++++++++++++++++++
 2 files changed, 50 insertions(+)
 create mode 100644 routers/api/v1/repo/collaborators.go

diff --git a/routers/api/v1/api.go b/routers/api/v1/api.go
index f43afd6e0..198036207 100644
--- a/routers/api/v1/api.go
+++ b/routers/api/v1/api.go
@@ -234,6 +234,7 @@ func RegisterRoutes(m *macaron.Macaron) {
 					m.Combo("/:id").Patch(bind(api.EditHookOption{}), repo.EditHook).
 						Delete(repo.DeleteHook)
 				})
+				m.Put("/collaborators/:collaborator", bind(api.AddCollaboratorOption{}), repo.AddCollaborator)
 				m.Get("/raw/*", context.RepoRef(), repo.GetRawFile)
 				m.Get("/archive/*", repo.GetArchive)
 				m.Group("/branches", func() {
diff --git a/routers/api/v1/repo/collaborators.go b/routers/api/v1/repo/collaborators.go
new file mode 100644
index 000000000..83c27aece
--- /dev/null
+++ b/routers/api/v1/repo/collaborators.go
@@ -0,0 +1,49 @@
+// Copyright 2014 The Gogs Authors. All rights reserved.
+// Use of this source code is governed by a MIT-style
+// license that can be found in the LICENSE file.
+
+package repo
+
+import (
+	api "github.com/gogits/go-gogs-client"
+
+	"github.com/gogits/gogs/models"
+	"github.com/gogits/gogs/modules/context"
+)
+
+func AddCollaborator(ctx *context.APIContext, form api.AddCollaboratorOption) {
+	collaborator, err := models.GetUserByName(ctx.Params(":collaborator"))
+
+	if err != nil {
+		if models.IsErrUserNotExist(err) {
+			ctx.Error(422, "", err)
+		} else {
+			ctx.Error(500, "GetUserByName", err)
+		}
+		return
+	}
+
+	if err := ctx.Repo.Repository.AddCollaborator(collaborator); err != nil {
+		ctx.Error(500, "AddCollaborator", err)
+		return
+	}
+
+	mode := models.ACCESS_MODE_WRITE
+	if form.Permission != nil && *form.Permission == "pull" {
+		mode = models.ACCESS_MODE_READ
+	} else if form.Permission != nil && *form.Permission == "push" {
+		mode = models.ACCESS_MODE_WRITE
+	} else if form.Permission != nil && *form.Permission == "admin" {
+		mode = models.ACCESS_MODE_ADMIN
+	} else if form.Permission != nil {
+		ctx.Error(500, "Permission", "Invalid permission type")
+		return
+	}
+	if err := ctx.Repo.Repository.ChangeCollaborationAccessMode(collaborator.Id, mode); err != nil {
+		ctx.Error(500, "ChangeCollaborationAccessMode", err)
+		return
+	}
+
+	ctx.Status(204)
+	return
+}