/
oauth.go
124 lines (113 loc) · 2.92 KB
/
oauth.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
package oauth
import (
"encoding/json"
"errors"
"fmt"
"io/ioutil"
"os"
"github.com/dotenx/dotenx/ao-api/models"
"github.com/dotenx/dotenx/ao-api/oauth/provider"
"github.com/dotenx/dotenx/ao-api/pkg/utils"
"github.com/dotenx/goth"
)
var providers []models.OauthProvider
var gitProviders []models.OauthProvider
var gothProviders map[string]*goth.Provider
var gitGothProviders map[string]*goth.Provider
var gothNotSupported []string
func init() {
// gothNotSupported = append(gothNotSupported, "typeform", "ebay")
gothNotSupported = append(gothNotSupported, "typeform")
jsonFile, err := os.Open("providers.json")
if err != nil {
fmt.Println(err)
}
defer jsonFile.Close()
byteValue, _ := ioutil.ReadAll(jsonFile)
err = json.Unmarshal(byteValue, &providers)
if err != nil {
fmt.Println(err)
}
fmt.Println("############")
fmt.Println(providers)
fmt.Println("############")
initGitProviders()
}
func initGitProviders() {
jsonFile, err := os.Open("git_providers.json")
if err != nil {
fmt.Println(err)
}
defer jsonFile.Close()
byteValue, _ := ioutil.ReadAll(jsonFile)
err = json.Unmarshal(byteValue, &gitProviders)
if err != nil {
fmt.Println(err)
}
}
// GetProviders returns a slice of providers formed from the corresponding config section
func GetProviders(cbURIBase string) (map[string]*goth.Provider, error) {
gothProviders = make(map[string]*goth.Provider)
if providers == nil {
return gothProviders, nil
}
for _, v := range providers {
if utils.ContainsString(gothNotSupported, v.Name) {
continue
}
uri := cbURIBase + v.Name
p, err := provider.New(v.Name, &v.Secret, &v.Key, uri, v.Scopes...)
if err != nil {
return gothProviders, err
}
gothProviders[v.Name] = p
}
return gothProviders, nil
}
func GetGitProviders(cbURIBase string) (map[string]*goth.Provider, error) {
gitGothProviders = make(map[string]*goth.Provider)
if gitProviders == nil {
return gitGothProviders, nil
}
for _, v := range gitProviders {
if utils.ContainsString(gothNotSupported, v.Name) {
continue
}
uri := cbURIBase + v.Name
p, err := provider.New(v.Name, &v.Secret, &v.Key, uri, v.Scopes...)
if err != nil {
return gitGothProviders, err
}
gitGothProviders[v.Name] = p
}
return gitGothProviders, nil
}
func GetProviderByName(name string) (*goth.Provider, error) {
p, ok := gothProviders[name]
if !ok {
return nil, errors.New("Provider not found")
}
return p, nil
}
func GetGitProviderByName(name string) (*goth.Provider, error) {
p, ok := gitGothProviders[name]
if !ok {
return nil, errors.New("provider not found")
}
return p, nil
}
func GetProviderModelByName(name string) (*models.OauthProvider, error) {
for _, v := range providers {
if v.Name == name {
return &v, nil
}
}
return nil, errors.New("provider not found")
}
func GetProvidersMap() map[string]models.OauthProvider {
res := make(map[string]models.OauthProvider)
for _, p := range providers {
res[p.Name] = p
}
return res
}