mirror of https://github.com/synctv-org/synctv
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
78 lines
1.8 KiB
Go
78 lines
1.8 KiB
Go
package providers
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/synctv-org/synctv/internal/provider"
|
|
"github.com/zijiren233/gencontainer/rwmap"
|
|
)
|
|
|
|
var (
|
|
enabledProviders rwmap.RWMap[provider.OAuth2Provider, struct{}]
|
|
allProviders rwmap.RWMap[provider.OAuth2Provider, provider.ProviderInterface]
|
|
)
|
|
|
|
func InitProvider(p provider.OAuth2Provider, c provider.Oauth2Option) (provider.ProviderInterface, error) {
|
|
pi, ok := allProviders.Load(p)
|
|
if !ok {
|
|
return nil, FormatErrNotImplemented(p)
|
|
}
|
|
pi.Init(c)
|
|
return pi, nil
|
|
}
|
|
|
|
func RegisterProvider(ps ...provider.ProviderInterface) {
|
|
for _, p := range ps {
|
|
allProviders.Store(p.Provider(), p)
|
|
}
|
|
}
|
|
|
|
func GetProvider(p provider.OAuth2Provider) (provider.ProviderInterface, error) {
|
|
_, ok := enabledProviders.Load(p)
|
|
if !ok {
|
|
return nil, FormatErrNotImplemented(p)
|
|
}
|
|
pi, ok := allProviders.Load(p)
|
|
if !ok {
|
|
return nil, FormatErrNotImplemented(p)
|
|
}
|
|
return pi, nil
|
|
}
|
|
|
|
func AllProvider() map[provider.OAuth2Provider]provider.ProviderInterface {
|
|
m := make(map[provider.OAuth2Provider]provider.ProviderInterface)
|
|
allProviders.Range(func(key string, value provider.ProviderInterface) bool {
|
|
m[key] = value
|
|
return true
|
|
})
|
|
return m
|
|
}
|
|
|
|
func EnabledProvider() *rwmap.RWMap[provider.OAuth2Provider, struct{}] {
|
|
return &enabledProviders
|
|
}
|
|
|
|
func EnableProvider(p provider.OAuth2Provider) error {
|
|
_, ok := allProviders.Load(p)
|
|
if !ok {
|
|
return FormatErrNotImplemented(p)
|
|
}
|
|
enabledProviders.Store(p, struct{}{})
|
|
return nil
|
|
}
|
|
|
|
func DisableProvider(p provider.OAuth2Provider) error {
|
|
_, ok := allProviders.Load(p)
|
|
if !ok {
|
|
return FormatErrNotImplemented(p)
|
|
}
|
|
enabledProviders.Delete(p)
|
|
return nil
|
|
}
|
|
|
|
type FormatErrNotImplemented string
|
|
|
|
func (f FormatErrNotImplemented) Error() string {
|
|
return fmt.Sprintf("%s not implemented", string(f))
|
|
}
|