yx-dataset-server/vendor/github.com/aliyun/credentials-go/credentials/oidc_credential_provider.go

package credentials

import (
	"os"

	"github.com/alibabacloud-go/tea/tea"
)

type oidcCredentialsProvider struct{}

var providerOIDC = new(oidcCredentialsProvider)

func newOidcCredentialsProvider() Provider {
	return &oidcCredentialsProvider{}
}

func (p *oidcCredentialsProvider) resolve() (*Config, error) {
	roleArn, ok1 := os.LookupEnv(ENVRoleArn)
	oidcProviderArn, ok2 := os.LookupEnv(ENVOIDCProviderArn)
	oidcTokenFilePath, ok3 := os.LookupEnv(ENVOIDCTokenFile)
	if !ok1 || !ok2 || !ok3 {
		return nil, nil
	}

	config := &Config{
		Type:              tea.String("oidc_role_arn"),
		RoleArn:           tea.String(roleArn),
		OIDCProviderArn:   tea.String(oidcProviderArn),
		OIDCTokenFilePath: tea.String(oidcTokenFilePath),
		RoleSessionName:   tea.String("defaultSessionName"),
	}
	roleSessionName, ok := os.LookupEnv(ENVRoleSessionName)
	if ok {
		config.RoleSessionName = tea.String(roleSessionName)
	}
	return config, nil
}