replace zxq.co/ripple/hanayo

vendor/github.com/RangelReale/osin/example/teststorage.go

@@ -0,0 +1,108 @@
+package example
+
+import (
+	"fmt"
+	"github.com/RangelReale/osin"
+)
+
+type TestStorage struct {
+	clients   map[string]osin.Client
+	authorize map[string]*osin.AuthorizeData
+	access    map[string]*osin.AccessData
+	refresh   map[string]string
+}
+
+func NewTestStorage() *TestStorage {
+	r := &TestStorage{
+		clients:   make(map[string]osin.Client),
+		authorize: make(map[string]*osin.AuthorizeData),
+		access:    make(map[string]*osin.AccessData),
+		refresh:   make(map[string]string),
+	}
+
+	r.clients["1234"] = &osin.DefaultClient{
+		Id:          "1234",
+		Secret:      "aabbccdd",
+		RedirectUri: "http://localhost:14000/appauth",
+	}
+
+	return r
+}
+
+func (s *TestStorage) Clone() osin.Storage {
+	return s
+}
+
+func (s *TestStorage) Close() {
+}
+
+func (s *TestStorage) GetClient(id string) (osin.Client, error) {
+	fmt.Printf("GetClient: %s\n", id)
+	if c, ok := s.clients[id]; ok {
+		return c, nil
+	}
+	return nil, osin.ErrNotFound
+}
+
+func (s *TestStorage) SetClient(id string, client osin.Client) error {
+	fmt.Printf("SetClient: %s\n", id)
+	s.clients[id] = client
+	return nil
+}
+
+func (s *TestStorage) SaveAuthorize(data *osin.AuthorizeData) error {
+	fmt.Printf("SaveAuthorize: %s\n", data.Code)
+	s.authorize[data.Code] = data
+	return nil
+}
+
+func (s *TestStorage) LoadAuthorize(code string) (*osin.AuthorizeData, error) {
+	fmt.Printf("LoadAuthorize: %s\n", code)
+	if d, ok := s.authorize[code]; ok {
+		return d, nil
+	}
+	return nil, osin.ErrNotFound
+}
+
+func (s *TestStorage) RemoveAuthorize(code string) error {
+	fmt.Printf("RemoveAuthorize: %s\n", code)
+	delete(s.authorize, code)
+	return nil
+}
+
+func (s *TestStorage) SaveAccess(data *osin.AccessData) error {
+	fmt.Printf("SaveAccess: %s\n", data.AccessToken)
+	s.access[data.AccessToken] = data
+	if data.RefreshToken != "" {
+		s.refresh[data.RefreshToken] = data.AccessToken
+	}
+	return nil
+}
+
+func (s *TestStorage) LoadAccess(code string) (*osin.AccessData, error) {
+	fmt.Printf("LoadAccess: %s\n", code)
+	if d, ok := s.access[code]; ok {
+		return d, nil
+	}
+	return nil, osin.ErrNotFound
+}
+
+func (s *TestStorage) RemoveAccess(code string) error {
+	fmt.Printf("RemoveAccess: %s\n", code)
+	delete(s.access, code)
+	return nil
+}
+
+func (s *TestStorage) LoadRefresh(code string) (*osin.AccessData, error) {
+	fmt.Printf("LoadRefresh: %s\n", code)
+	if d, ok := s.refresh[code]; ok {
+		return s.LoadAccess(d)
+	}
+	return nil, osin.ErrNotFound
+}
+
+func (s *TestStorage) RemoveRefresh(code string) error {
+	fmt.Printf("RemoveRefresh: %s\n", code)
+	delete(s.refresh, code)
+	return nil
+}