Albirew/nyaa-pantsu
Archivé
1
0
Bifurcation 0
Ce dépôt a été archivé le 2022-05-07. Vous pouvez voir ses fichiers ou le cloner, mais pas ouvrir de ticket ou de demandes d'ajout, ni soumettre de changements.
nyaa-pantsu/models/oauth_server.go
akuma06 03ea72595d OAuth API [done] (#1275)
* Initial Commit for OAuth API

This builds and run and return the right error.
Need to test it and then adding all users as possible client

* Added mising dependency

* just compile already...

* Fixing template test

* Imrpovements

Moved db stuff in models
Added some tests
Added form in modpanel to add/update a client
Added controllers for add/update of client

* Added Forms + speed improvements

Controller oauth client listing + html
Controller oauth client delete + messages
Messages on comment delete
New ES config that disable ES if set to false. Improve load speed on local development
Fix a load config bug
Fix index admin & translation string sign_out broken by @ewhal

* Sanitize empty strig in form array + css

Multiple empty array of strings are sanitized for the oauth client create form
Added some css for the form display

* Upload and Create form works

* Fix splitting response types

* Removing required on secret when updating

* fix travis error

* Fix travis template test

* Update dependency

* Moved to jinzhu instead of azhao

* randomizen secret on creation

* Final touch on oath api

improved display name
fix grant form csrf
fix login csrf on oauth

* Fix gorm test

* fix template test

* Fixing deleted dependency issue

* Make travis faster

* Fix typo

* Fix csrf for api calls

* This shouldn't be exempt

* Removing hard coded hash

@ewhal Don't forget to replace the hash in tokens.go with another one

* Added an example on how to use OAuth middleware

* Renamed fosite utils to oauth2 utils
2017-07-28 13:46:40 +10:00

118 lignes
2,5 Kio
Go

package models
import (
"context"
"encoding/json"
"net/url"
"strings"
"time"
"github.com/NyaaPantsu/nyaa/utils/log"
"github.com/NyaaPantsu/nyaa/utils/oauth2/client"
"github.com/ory/fosite"
"github.com/pkg/errors"
)
const (
TableOpenID = "oidc"
TableAccess = "access"
TableRefresh = "refresh"
TableCode = "code"
)
const oauth_prefix = "hydra_oauth2_"
type OauthAbstract struct {
Signature string `gorm:"column:signature;primary_key;not null"`
RequestID string `gorm:"column:request_id;not null"`
RequestedAt time.Time `gorm:"column:requested_at;not null;type:timestamp;default:CURRENT_TIMESTAMP"`
ClientID string `gorm:"column:client_id;not null"`
Scope string `gorm:"column:scope;not null"`
GrantedScope string `gorm:"column:granted_scope;not null"`
FormData string `gorm:"column:form_data;not null"`
SessionData []byte `gorm:"column:session_data;not null"`
}
type OpenID struct {
OauthAbstract
}
type Access struct {
OauthAbstract
}
type Code struct {
OauthAbstract
}
type Refresh struct {
OauthAbstract
}
func (o OpenID) TableName() string {
return oauth_prefix + TableOpenID
}
func (a Access) TableName() string {
return oauth_prefix + TableAccess
}
func (r Refresh) TableName() string {
return oauth_prefix + TableRefresh
}
func (c Code) TableName() string {
return oauth_prefix + TableCode
}
func (s *OauthAbstract) ToRequest(session fosite.Session, cm Manager) (*fosite.Request, error) {
if session != nil {
if err := json.Unmarshal(s.SessionData, session); err != nil {
return nil, errors.WithStack(err)
}
} else {
log.Debugf("Got an empty session in toRequest")
}
c, err := cm.GetClient(context.Background(), s.ClientID)
if err != nil {
return nil, err
}
val, err := url.ParseQuery(s.FormData)
if err != nil {
return nil, errors.WithStack(err)
}
r := &fosite.Request{
ID: s.RequestID,
RequestedAt: s.RequestedAt,
Client: c,
Scopes: fosite.Arguments(strings.Split(s.Scope, "|")),
GrantedScopes: fosite.Arguments(strings.Split(s.GrantedScope, "|")),
Form: val,
Session: session,
}
return r, nil
}
type Manager interface {
Storage
Authenticate(id string, secret []byte) (*client.Client, error)
}
type Storage interface {
fosite.Storage
CreateClient(c *client.Client) error
UpdateClient(c *client.Client) error
DeleteClient(id string) error
GetClients() (map[string]client.Client, error)
GetConcreteClient(id string) (*client.Client, error)
}