110 lines
		
	
	
		
			2.3 KiB
		
	
	
	
		
			Go
		
	
	
	
			
		
		
	
	
			110 lines
		
	
	
		
			2.3 KiB
		
	
	
	
		
			Go
		
	
	
	
package actions
 | 
						|
 | 
						|
import (
 | 
						|
	"fmt"
 | 
						|
	"net/url"
 | 
						|
	"os"
 | 
						|
	"strings"
 | 
						|
)
 | 
						|
 | 
						|
var ErrInvalidGitHubConfigURL = fmt.Errorf("invalid config URL, should point to an enterprise, org, or repository")
 | 
						|
 | 
						|
type GitHubScope int
 | 
						|
 | 
						|
const (
 | 
						|
	GitHubScopeUnknown GitHubScope = iota
 | 
						|
	GitHubScopeEnterprise
 | 
						|
	GitHubScopeOrganization
 | 
						|
	GitHubScopeRepository
 | 
						|
)
 | 
						|
 | 
						|
type GitHubConfig struct {
 | 
						|
	ConfigURL *url.URL
 | 
						|
	Scope     GitHubScope
 | 
						|
 | 
						|
	Enterprise   string
 | 
						|
	Organization string
 | 
						|
	Repository   string
 | 
						|
 | 
						|
	IsHosted bool
 | 
						|
}
 | 
						|
 | 
						|
func ParseGitHubConfigFromURL(in string) (*GitHubConfig, error) {
 | 
						|
	u, err := url.Parse(strings.Trim(in, "/"))
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
 | 
						|
	isHosted := isHostedGitHubURL(u)
 | 
						|
 | 
						|
	configURL := &GitHubConfig{
 | 
						|
		ConfigURL: u,
 | 
						|
		IsHosted:  isHosted,
 | 
						|
	}
 | 
						|
 | 
						|
	invalidURLError := fmt.Errorf("%q: %w", u.String(), ErrInvalidGitHubConfigURL)
 | 
						|
 | 
						|
	pathParts := strings.Split(strings.Trim(u.Path, "/"), "/")
 | 
						|
 | 
						|
	switch len(pathParts) {
 | 
						|
	case 1: // Organization
 | 
						|
		if pathParts[0] == "" {
 | 
						|
			return nil, invalidURLError
 | 
						|
		}
 | 
						|
 | 
						|
		configURL.Scope = GitHubScopeOrganization
 | 
						|
		configURL.Organization = pathParts[0]
 | 
						|
 | 
						|
	case 2: // Repository or enterprise
 | 
						|
		if strings.ToLower(pathParts[0]) == "enterprises" {
 | 
						|
			configURL.Scope = GitHubScopeEnterprise
 | 
						|
			configURL.Enterprise = pathParts[1]
 | 
						|
			break
 | 
						|
		}
 | 
						|
 | 
						|
		configURL.Scope = GitHubScopeRepository
 | 
						|
		configURL.Organization = pathParts[0]
 | 
						|
		configURL.Repository = pathParts[1]
 | 
						|
	default:
 | 
						|
		return nil, invalidURLError
 | 
						|
	}
 | 
						|
 | 
						|
	return configURL, nil
 | 
						|
}
 | 
						|
 | 
						|
func (c *GitHubConfig) GitHubAPIURL(path string) *url.URL {
 | 
						|
	result := &url.URL{
 | 
						|
		Scheme: c.ConfigURL.Scheme,
 | 
						|
		Host:   c.ConfigURL.Host, // default for Enterprise mode
 | 
						|
		Path:   "/api/v3",        // default for Enterprise mode
 | 
						|
	}
 | 
						|
 | 
						|
	isHosted := isHostedGitHubURL(c.ConfigURL)
 | 
						|
 | 
						|
	if isHosted {
 | 
						|
		result.Host = fmt.Sprintf("api.%s", c.ConfigURL.Host)
 | 
						|
		result.Path = ""
 | 
						|
 | 
						|
		if strings.EqualFold("www.github.com", c.ConfigURL.Host) {
 | 
						|
			// re-routing www.github.com to api.github.com
 | 
						|
			result.Host = "api.github.com"
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	result.Path += path
 | 
						|
 | 
						|
	return result
 | 
						|
}
 | 
						|
 | 
						|
func isHostedGitHubURL(u *url.URL) bool {
 | 
						|
	_, forceGhes := os.LookupEnv("GITHUB_ACTIONS_FORCE_GHES")
 | 
						|
	if forceGhes {
 | 
						|
		return false
 | 
						|
	}
 | 
						|
 | 
						|
	return strings.EqualFold(u.Host, "github.com") ||
 | 
						|
		strings.EqualFold(u.Host, "www.github.com") ||
 | 
						|
		strings.EqualFold(u.Host, "github.localhost") ||
 | 
						|
		strings.HasSuffix(u.Host, ".ghe.com")
 | 
						|
}
 |