woodpecker/controller/user.go

package controller

import (
	"net/http"

	"github.com/gin-gonic/gin"

	"github.com/drone/drone/model"
	"github.com/drone/drone/router/middleware/context"
	"github.com/drone/drone/router/middleware/session"
	"github.com/drone/drone/shared/token"
	"github.com/hashicorp/golang-lru"
)

var cache *lru.Cache

func init() {
	var err error
	cache, err = lru.New(1028)
	if err != nil {
		panic(err)
	}
}

func GetSelf(c *gin.Context) {
	c.IndentedJSON(200, session.User(c))
}

func GetFeed(c *gin.Context) {
	user := session.User(c)
	db := context.Database(c)
	feed, err := model.GetUserFeed(db, user, 25, 0)
	if err != nil {
		c.AbortWithStatus(http.StatusInternalServerError)
		return
	}
	c.IndentedJSON(http.StatusOK, feed)
}

func GetRepos(c *gin.Context) {
	user := session.User(c)
	db := context.Database(c)
	repos, err := model.GetRepoList(db, user)
	if err != nil {
		c.AbortWithStatus(http.StatusInternalServerError)
		return
	}
	c.IndentedJSON(http.StatusOK, repos)
}

func GetRemoteRepos(c *gin.Context) {
	user := session.User(c)
	remote := context.Remote(c)

	// attempt to get the repository list from the
	// cache since the operation is expensive
	v, ok := cache.Get(user.Login)
	if ok {
		c.IndentedJSON(http.StatusOK, v)
		return
	}

	repos, err := remote.Repos(user)
	if err != nil {
		c.AbortWithStatus(http.StatusInternalServerError)
		return
	}
	cache.Add(user.Login, repos)
	c.IndentedJSON(http.StatusOK, repos)
}

func PostToken(c *gin.Context) {
	user := session.User(c)

	token := token.New(token.UserToken, user.Login)
	tokenstr, err := token.Sign(user.Hash)
	if err != nil {
		c.AbortWithError(http.StatusInternalServerError, err)
	} else {
		c.String(http.StatusOK, tokenstr)
	}
}