2014-06-05 00:25:38 +03:00
|
|
|
package handler
|
|
|
|
|
|
|
|
import (
|
|
|
|
"encoding/json"
|
|
|
|
"fmt"
|
|
|
|
"net/http"
|
|
|
|
|
2014-07-13 05:01:58 +03:00
|
|
|
"github.com/drone/drone/plugin/remote"
|
2014-09-29 04:36:24 +03:00
|
|
|
"github.com/drone/drone/server/datastore"
|
2014-06-13 01:28:05 +03:00
|
|
|
"github.com/drone/drone/shared/httputil"
|
2014-07-10 03:26:21 +03:00
|
|
|
"github.com/drone/drone/shared/model"
|
2014-06-13 01:28:05 +03:00
|
|
|
"github.com/drone/drone/shared/sshutil"
|
2014-09-29 04:36:24 +03:00
|
|
|
"github.com/goji/context"
|
|
|
|
"github.com/zenazn/goji/web"
|
2014-06-05 00:25:38 +03:00
|
|
|
)
|
|
|
|
|
2014-09-29 04:36:24 +03:00
|
|
|
// GetRepo accepts a request to retrieve a commit
|
|
|
|
// from the datastore for the given repository, branch and
|
|
|
|
// commit hash.
|
|
|
|
//
|
|
|
|
// GET /api/repos/:host/:owner/:name
|
|
|
|
//
|
|
|
|
func GetRepo(c web.C, w http.ResponseWriter, r *http.Request) {
|
|
|
|
var (
|
|
|
|
admin = r.FormValue("admin")
|
|
|
|
role = ToRole(c)
|
|
|
|
repo = ToRepo(c)
|
|
|
|
)
|
|
|
|
|
|
|
|
// if the user is not requesting (or cannot access)
|
|
|
|
// admin data then we just return the repo as-is
|
|
|
|
if len(admin) == 0 || role.Admin == false {
|
|
|
|
json.NewEncoder(w).Encode(repo)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
// else we should return restricted fields
|
|
|
|
json.NewEncoder(w).Encode(struct {
|
2014-07-10 03:26:21 +03:00
|
|
|
*model.Repo
|
2014-09-02 10:18:17 +03:00
|
|
|
PublicKey string `json:"public_key"`
|
|
|
|
Params string `json:"params"`
|
|
|
|
}{repo, repo.PublicKey, repo.Params})
|
2014-06-05 00:25:38 +03:00
|
|
|
}
|
|
|
|
|
2014-09-29 04:36:24 +03:00
|
|
|
// DelRepo accepts a request to inactivate the named
|
|
|
|
// repository. This will disable all builds in the system
|
|
|
|
// for this repository.
|
|
|
|
//
|
|
|
|
// DEL /api/repos/:host/:owner/:name
|
|
|
|
//
|
|
|
|
func DelRepo(c web.C, w http.ResponseWriter, r *http.Request) {
|
|
|
|
var ctx = context.FromC(c)
|
|
|
|
var repo = ToRepo(c)
|
|
|
|
|
|
|
|
// disable everything
|
|
|
|
repo.Active = false
|
|
|
|
repo.PullRequest = false
|
|
|
|
repo.PostCommit = false
|
2014-06-05 00:25:38 +03:00
|
|
|
|
2014-09-29 04:36:24 +03:00
|
|
|
if err := datastore.PutRepo(ctx, repo); err != nil {
|
|
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
|
|
return
|
2014-06-05 00:25:38 +03:00
|
|
|
}
|
2014-09-29 04:36:24 +03:00
|
|
|
w.WriteHeader(http.StatusNoContent)
|
|
|
|
}
|
2014-06-05 00:25:38 +03:00
|
|
|
|
2014-09-29 04:36:24 +03:00
|
|
|
// PostRepo accapets a request to activate the named repository
|
|
|
|
// in the datastore. It returns a 201 status created if successful
|
|
|
|
//
|
|
|
|
// POST /api/repos/:host/:owner/:name
|
|
|
|
//
|
|
|
|
func PostRepo(c web.C, w http.ResponseWriter, r *http.Request) {
|
|
|
|
var ctx = context.FromC(c)
|
|
|
|
var repo = ToRepo(c)
|
|
|
|
var user = ToUser(c)
|
2014-06-05 00:25:38 +03:00
|
|
|
|
|
|
|
// update the repo active flag and fields
|
|
|
|
repo.Active = true
|
|
|
|
repo.PullRequest = true
|
|
|
|
repo.PostCommit = true
|
|
|
|
repo.UserID = user.ID
|
2014-09-06 21:10:00 +03:00
|
|
|
repo.Timeout = 3600 // default to 1 hour
|
2014-06-05 00:25:38 +03:00
|
|
|
|
2014-09-29 04:36:24 +03:00
|
|
|
// generates the rsa key
|
2014-06-05 00:25:38 +03:00
|
|
|
key, err := sshutil.GeneratePrivateKey()
|
|
|
|
if err != nil {
|
2014-09-29 04:36:24 +03:00
|
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
|
|
return
|
2014-06-05 00:25:38 +03:00
|
|
|
}
|
|
|
|
repo.PublicKey = sshutil.MarshalPublicKey(&key.PublicKey)
|
|
|
|
repo.PrivateKey = sshutil.MarshalPrivateKey(key)
|
|
|
|
|
2014-09-29 04:36:24 +03:00
|
|
|
var remote = remote.Lookup(repo.Host)
|
2014-09-02 10:18:17 +03:00
|
|
|
if remote == nil {
|
2014-09-29 04:36:24 +03:00
|
|
|
w.WriteHeader(http.StatusNotFound)
|
|
|
|
return
|
2014-06-05 00:25:38 +03:00
|
|
|
}
|
|
|
|
|
2014-09-29 04:36:24 +03:00
|
|
|
// setup the post-commit hook with the remote system and
|
|
|
|
// if necessary, register the public key
|
2014-10-07 17:57:01 +03:00
|
|
|
var hook = fmt.Sprintf("%s/api/hook/%s", httputil.GetURL(r), repo.Remote)
|
2014-09-02 10:18:17 +03:00
|
|
|
if err := remote.Activate(user, repo, hook); err != nil {
|
2014-09-29 04:36:24 +03:00
|
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
|
|
return
|
2014-06-05 00:25:38 +03:00
|
|
|
}
|
|
|
|
|
2014-09-29 04:36:24 +03:00
|
|
|
if err := datastore.PutRepo(ctx, repo); err != nil {
|
|
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
|
|
return
|
2014-06-05 00:25:38 +03:00
|
|
|
}
|
|
|
|
w.WriteHeader(http.StatusCreated)
|
2014-09-29 04:36:24 +03:00
|
|
|
json.NewEncoder(w).Encode(repo)
|
2014-06-05 00:25:38 +03:00
|
|
|
}
|
|
|
|
|
2014-09-29 04:36:24 +03:00
|
|
|
// PutRepo accapets a request to update the named repository
|
|
|
|
// in the datastore. It expects a JSON input and returns the
|
|
|
|
// updated repository in JSON format if successful.
|
|
|
|
//
|
|
|
|
// PUT /api/repos/:host/:owner/:name
|
|
|
|
//
|
|
|
|
func PutRepo(c web.C, w http.ResponseWriter, r *http.Request) {
|
|
|
|
var ctx = context.FromC(c)
|
|
|
|
var repo = ToRepo(c)
|
|
|
|
var user = ToUser(c)
|
2014-06-05 00:25:38 +03:00
|
|
|
|
|
|
|
// unmarshal the repository from the payload
|
|
|
|
defer r.Body.Close()
|
|
|
|
in := struct {
|
|
|
|
PostCommit *bool `json:"post_commits"`
|
|
|
|
PullRequest *bool `json:"pull_requests"`
|
|
|
|
Privileged *bool `json:"privileged"`
|
|
|
|
Params *string `json:"params"`
|
|
|
|
Timeout *int64 `json:"timeout"`
|
2014-09-06 22:45:27 +03:00
|
|
|
PublicKey *string `json:"public_key"`
|
|
|
|
PrivateKey *string `json:"private_key"`
|
2014-06-05 00:25:38 +03:00
|
|
|
}{}
|
|
|
|
if err := json.NewDecoder(r.Body).Decode(&in); err != nil {
|
2014-09-29 04:36:24 +03:00
|
|
|
w.WriteHeader(http.StatusBadRequest)
|
|
|
|
return
|
2014-06-05 00:25:38 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
if in.Params != nil {
|
|
|
|
repo.Params = *in.Params
|
|
|
|
}
|
|
|
|
if in.PostCommit != nil {
|
|
|
|
repo.PostCommit = *in.PostCommit
|
|
|
|
}
|
|
|
|
if in.PullRequest != nil {
|
|
|
|
repo.PullRequest = *in.PullRequest
|
|
|
|
}
|
|
|
|
if in.Privileged != nil && user.Admin {
|
|
|
|
repo.Privileged = *in.Privileged
|
|
|
|
}
|
|
|
|
if in.Timeout != nil && user.Admin {
|
|
|
|
repo.Timeout = *in.Timeout
|
|
|
|
}
|
2014-09-06 22:45:27 +03:00
|
|
|
if in.PrivateKey != nil && in.PublicKey != nil {
|
|
|
|
repo.PublicKey = *in.PublicKey
|
|
|
|
repo.PrivateKey = *in.PrivateKey
|
|
|
|
}
|
2014-09-29 04:36:24 +03:00
|
|
|
if err := datastore.PutRepo(ctx, repo); err != nil {
|
|
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
|
|
return
|
2014-06-05 00:25:38 +03:00
|
|
|
}
|
2014-09-29 04:36:24 +03:00
|
|
|
json.NewEncoder(w).Encode(repo)
|
2014-06-05 00:25:38 +03:00
|
|
|
}
|