144 lines
4.2 KiB
Go
144 lines
4.2 KiB
Go
// Copyright 2019 Sorint.lab
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package common
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/pkg/errors"
|
|
gitsource "github.com/sorintlab/agola/internal/gitsources"
|
|
"github.com/sorintlab/agola/internal/gitsources/gitea"
|
|
"github.com/sorintlab/agola/internal/gitsources/gitlab"
|
|
"github.com/sorintlab/agola/internal/services/types"
|
|
)
|
|
|
|
func SourceSupportedAuthTypes(rsType types.RemoteSourceType) []types.RemoteSourceAuthType {
|
|
switch rsType {
|
|
case types.RemoteSourceTypeGitea:
|
|
return []types.RemoteSourceAuthType{types.RemoteSourceAuthTypePassword}
|
|
case types.RemoteSourceTypeGithub:
|
|
fallthrough
|
|
case types.RemoteSourceTypeGitlab:
|
|
return []types.RemoteSourceAuthType{types.RemoteSourceAuthTypeOauth2}
|
|
|
|
default:
|
|
panic(fmt.Errorf("unsupported remote source type: %q", rsType))
|
|
}
|
|
}
|
|
|
|
func SourceSupportsAuthType(rsType types.RemoteSourceType, authType types.RemoteSourceAuthType) bool {
|
|
supportedAuthTypes := SourceSupportedAuthTypes(rsType)
|
|
for _, st := range supportedAuthTypes {
|
|
if st == authType {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func newGitea(rs *types.RemoteSource, accessToken string) (*gitea.Client, error) {
|
|
return gitea.New(gitea.Opts{
|
|
URL: rs.APIURL,
|
|
SkipVerify: rs.SkipVerify,
|
|
Token: accessToken,
|
|
})
|
|
}
|
|
|
|
func newGitlab(rs *types.RemoteSource, accessToken string) (*gitlab.Client, error) {
|
|
return gitlab.New(gitlab.Opts{
|
|
URL: rs.APIURL,
|
|
SkipVerify: rs.SkipVerify,
|
|
Token: accessToken,
|
|
Oauth2ClientID: rs.Oauth2ClientID,
|
|
Oauth2Secret: rs.Oauth2ClientSecret,
|
|
})
|
|
}
|
|
|
|
func GetAccessToken(authType types.RemoteSourceAuthType, userAccessToken, oauth2AccessToken string) (string, error) {
|
|
switch authType {
|
|
case types.RemoteSourceAuthTypePassword:
|
|
return userAccessToken, nil
|
|
case types.RemoteSourceAuthTypeOauth2:
|
|
return oauth2AccessToken, nil
|
|
default:
|
|
return "", errors.Errorf("invalid remote source auth type %q", authType)
|
|
}
|
|
}
|
|
|
|
func GetGitSource(rs *types.RemoteSource, la *types.LinkedAccount) (gitsource.GitSource, error) {
|
|
var accessToken string
|
|
if la != nil {
|
|
var err error
|
|
accessToken, err = GetAccessToken(rs.AuthType, la.UserAccessToken, la.Oauth2AccessToken)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
var gitSource gitsource.GitSource
|
|
var err error
|
|
switch rs.Type {
|
|
case types.RemoteSourceTypeGitea:
|
|
gitSource, err = newGitea(rs, accessToken)
|
|
case types.RemoteSourceTypeGitlab:
|
|
gitSource, err = newGitlab(rs, accessToken)
|
|
default:
|
|
return nil, errors.Errorf("remote source %s isn't a valid git source", rs.Name)
|
|
}
|
|
|
|
return gitSource, err
|
|
}
|
|
|
|
func GetUserSource(rs *types.RemoteSource, accessToken string) (gitsource.UserSource, error) {
|
|
var userSource gitsource.UserSource
|
|
var err error
|
|
switch rs.AuthType {
|
|
case types.RemoteSourceAuthTypeOauth2:
|
|
userSource, err = GetOauth2Source(rs, accessToken)
|
|
case types.RemoteSourceAuthTypePassword:
|
|
userSource, err = GetPasswordSource(rs, accessToken)
|
|
default:
|
|
return nil, errors.Errorf("unknown remote source auth type")
|
|
}
|
|
|
|
return userSource, err
|
|
}
|
|
|
|
func GetOauth2Source(rs *types.RemoteSource, accessToken string) (gitsource.Oauth2Source, error) {
|
|
var oauth2Source gitsource.Oauth2Source
|
|
var err error
|
|
switch rs.Type {
|
|
case types.RemoteSourceTypeGitlab:
|
|
oauth2Source, err = newGitlab(rs, accessToken)
|
|
default:
|
|
return nil, errors.Errorf("remote source %s isn't a valid oauth2 source", rs.Name)
|
|
}
|
|
|
|
return oauth2Source, err
|
|
}
|
|
|
|
func GetPasswordSource(rs *types.RemoteSource, accessToken string) (gitsource.PasswordSource, error) {
|
|
var passwordSource gitsource.PasswordSource
|
|
var err error
|
|
switch rs.Type {
|
|
case types.RemoteSourceTypeGitea:
|
|
passwordSource, err = newGitea(rs, accessToken)
|
|
default:
|
|
return nil, errors.Errorf("remote source %s isn't a valid oauth2 source", rs.Name)
|
|
}
|
|
|
|
return passwordSource, err
|
|
}
|