115 lines
2.8 KiB
Go
115 lines
2.8 KiB
Go
package handler
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
|
|
"github.com/drone/drone/server/database"
|
|
"github.com/drone/drone/server/session"
|
|
"github.com/drone/drone/shared/model"
|
|
"github.com/gorilla/pat"
|
|
)
|
|
|
|
type UserHandler struct {
|
|
commits database.CommitManager
|
|
repos database.RepoManager
|
|
users database.UserManager
|
|
sess session.Session
|
|
}
|
|
|
|
func NewUserHandler(users database.UserManager, repos database.RepoManager, commits database.CommitManager, sess session.Session) *UserHandler {
|
|
return &UserHandler{commits, repos, users, sess}
|
|
}
|
|
|
|
// GetUser gets the authenticated user.
|
|
// GET /api/user
|
|
func (h *UserHandler) GetUser(w http.ResponseWriter, r *http.Request) error {
|
|
// get the user form the session
|
|
u := h.sess.User(r)
|
|
if u == nil {
|
|
return notAuthorized{}
|
|
}
|
|
// Normally the Token would not be serialized to json.
|
|
// In this case it is appropriate because the user is
|
|
// requesting their own data, and will need to display
|
|
// the Token on the website.
|
|
data := struct {
|
|
*model.User
|
|
Token string `json:"token"`
|
|
}{u, u.Token}
|
|
return json.NewEncoder(w).Encode(&data)
|
|
}
|
|
|
|
// PutUser updates the authenticated user.
|
|
// PUT /api/user
|
|
func (h *UserHandler) PutUser(w http.ResponseWriter, r *http.Request) error {
|
|
// get the user form the session
|
|
u := h.sess.User(r)
|
|
if u == nil {
|
|
return notAuthorized{}
|
|
}
|
|
|
|
// unmarshal the repository from the payload
|
|
defer r.Body.Close()
|
|
in := model.User{}
|
|
if err := json.NewDecoder(r.Body).Decode(&in); err != nil {
|
|
return badRequest{err}
|
|
}
|
|
|
|
// update the user email
|
|
if len(in.Email) != 0 {
|
|
u.SetEmail(in.Email)
|
|
}
|
|
// update the user full name
|
|
if len(in.Name) != 0 {
|
|
u.Name = in.Name
|
|
}
|
|
|
|
// update the database
|
|
if err := h.users.Update(u); err != nil {
|
|
return internalServerError{err}
|
|
}
|
|
|
|
return json.NewEncoder(w).Encode(u)
|
|
}
|
|
|
|
// GetRepos gets the authenticated user's repositories.
|
|
// GET /api/user/repos
|
|
func (h *UserHandler) GetRepos(w http.ResponseWriter, r *http.Request) error {
|
|
// get the user from the session
|
|
u := h.sess.User(r)
|
|
if u == nil {
|
|
return notAuthorized{}
|
|
}
|
|
|
|
// get the user repositories
|
|
repos, err := h.repos.List(u.ID)
|
|
if err != nil {
|
|
return badRequest{err}
|
|
}
|
|
return json.NewEncoder(w).Encode(&repos)
|
|
}
|
|
|
|
// GetFeed gets the authenticated user's commit feed.
|
|
// GET /api/user/feed
|
|
func (h *UserHandler) GetFeed(w http.ResponseWriter, r *http.Request) error {
|
|
// get the user from the session
|
|
u := h.sess.User(r)
|
|
if u == nil {
|
|
return notAuthorized{}
|
|
}
|
|
|
|
// get the user commits
|
|
commits, err := h.commits.ListUser(u.ID)
|
|
if err != nil {
|
|
return badRequest{err}
|
|
}
|
|
return json.NewEncoder(w).Encode(&commits)
|
|
}
|
|
|
|
func (h *UserHandler) Register(r *pat.Router) {
|
|
r.Get("/v1/user/repos", errorHandler(h.GetRepos))
|
|
r.Get("/v1/user/feed", errorHandler(h.GetFeed))
|
|
r.Get("/v1/user", errorHandler(h.GetUser))
|
|
r.Put("/v1/user", errorHandler(h.PutUser))
|
|
}
|