fix drone agent issue

This commit is contained in:
Brad Rydzewski
2016-05-03 23:36:23 -07:00
parent 4040dfceb8
commit 1d508fb792
5 changed files with 12 additions and 2 deletions

View File

@@ -18,7 +18,7 @@ func Agents(cli *cli.Context) gin.HandlerFunc {
logrus.Fatalf("failed to generate token from DRONE_AGENT_SECRET")
}
t := token.New(secret, "")
t := token.New(token.AgentToken, secret)
s, err := t.Sign(secret)
if err != nil {
logrus.Fatalf("failed to generate token from DRONE_AGENT_SECRET. %s", err)

View File

@@ -18,8 +18,10 @@ func AuthorizeAgent(c *gin.Context) {
})
if err != nil {
c.String(500, "invalid or empty token. %s", err)
c.Abort()
} else if parsed.Kind != token.AgentToken {
c.String(403, "invalid token. please use an agent token")
c.Abort()
} else {
c.Next()
}