hanayo/oauth.go

49 lines
1.0 KiB
Go
Raw Normal View History

2019-02-23 13:29:15 +00:00
package main
import (
"database/sql"
"github.com/gin-gonic/gin"
"github.com/osuYozora/hanayo/routers/oauth"
)
type oauthRequestHandler struct{}
func (o oauthRequestHandler) CheckLoggedInOrRedirect(c *gin.Context) bool {
ctx := getContext(c)
if ctx.User.ID == 0 {
resp403(c)
return false
}
return true
}
func (o oauthRequestHandler) DisplayAuthorizePage(client oauth.Client, c *gin.Context) {
var creatorName string
db.Get(&creatorName, "SELECT username FROM users WHERE id = ? LIMIT 1", client.CreatorID)
c.Header("X-Frame-Options", "deny")
simple(c, getSimpleByFilename("oauth.html"), nil, map[string]interface{}{
"Client": client,
"CreatorName": creatorName,
})
}
func (o oauthRequestHandler) CheckCSRF(c *gin.Context, s string) bool {
b, _ := CSRF.Validate(getContext(c).User.ID, s)
return b
}
func (o oauthRequestHandler) GetDB() *sql.DB {
return db.DB
}
func (o oauthRequestHandler) GetUserID(c *gin.Context) int {
return getContext(c).User.ID
}
func setUpOauth() {
oauth.Initialise(oauthRequestHandler{})
}