forge/detect.go at main · git-pkgs/forge · GitHub

History

163 lines (136 loc) · 4.06 KB

Raw

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

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

159

160

161

162

163

package forges

import (

"bytes"

"context"

"encoding/json"

"fmt"

"io"

"net/http"

"strings"

)

// DetectForgeType probes a domain to identify which forge software it runs.

// It checks HTTP response headers first, then falls back to API endpoints.

// The domain may include an http:// or https:// prefix; without one, https is

// assumed. If hc is nil, http.DefaultClient is used.

func DetectForgeType(ctx context.Context, domain string, hc ...*http.Client) (ForgeType, error) {

client := http.DefaultClient

if len(hc) > 0 && hc[0] != nil {

client = hc[0]

}

baseURL, _, err := normalizeBaseURL(domain)

if err != nil {

return Unknown, err

}

ft, err := detectFromHeaders(ctx, client, baseURL)

if err != nil {

return Unknown, fmt.Errorf("could not detect forge type for %s: %w", baseURL, err)

}

if ft != Unknown {

return ft, nil

}

return detectFromAPI(ctx, client, baseURL)

}

func detectFromHeaders(ctx context.Context, client *http.Client, baseURL string) (ForgeType, error) {

req, err := http.NewRequestWithContext(ctx, http.MethodGet, baseURL, nil)

if err != nil {

return Unknown, err

}

resp, err := client.Do(req)

if err != nil {

return Unknown, err

}

defer func() { _ = resp.Body.Close() }()

if resp.Header.Get("X-Forgejo-Version") != "" {

return Forgejo, nil

}

if resp.Header.Get("X-Gitea-Version") != "" {

return Gitea, nil

}

if resp.Header.Get("X-Gitlab-Meta") != "" {

return GitLab, nil

}

if resp.Header.Get("X-GitHub-Request-Id") != "" {

return GitHub, nil

}

return Unknown, nil

}

func detectFromAPI(ctx context.Context, client *http.Client, baseURL string) (ForgeType, error) {

// Try Gitea/Forgejo /api/v1/version

if ft, err := probeGiteaAPI(ctx, client, baseURL); err == nil {

return ft, nil

}

// Try GitLab /api/v4/version

if ok, err := probeURL(ctx, client, baseURL+"/api/v4/version"); err == nil && ok {

return GitLab, nil

}

// Try Gerrit /config/server/version

if ok, err := probeGerritAPI(ctx, client, baseURL); err == nil && ok {

return Gerrit, nil

}

// Try GitHub Enterprise /api/v3/meta

if ok, err := probeURL(ctx, client, baseURL+"/api/v3/meta"); err == nil && ok {

return GitHub, nil

}

// Try Tangled knot/appview XRPC endpoint.

if ok, err := probeURL(ctx, client, baseURL+"/xrpc/sh.tangled.knot.version"); err == nil && ok {

return Tangled, nil

}

return Unknown, fmt.Errorf("could not detect forge type for %s", baseURL)

}

func probeGiteaAPI(ctx context.Context, client *http.Client, baseURL string) (ForgeType, error) {

req, err := http.NewRequestWithContext(ctx, http.MethodGet, baseURL+"/api/v1/version", nil)

if err != nil {

return Unknown, err

}

resp, err := client.Do(req)

if err != nil {

return Unknown, err

}

defer func() { _ = resp.Body.Close() }()

if resp.StatusCode != http.StatusOK {

return Unknown, fmt.Errorf("status %d", resp.StatusCode)

}

var v struct {

Version string `json:"version"`

}

if err := json.NewDecoder(resp.Body).Decode(&v); err != nil {

return Unknown, err

}

if strings.Contains(strings.ToLower(v.Version), "forgejo") {

return Forgejo, nil

}

return Gitea, nil

}

func probeGerritAPI(ctx context.Context, client *http.Client, baseURL string) (bool, error) {

req, err := http.NewRequestWithContext(ctx, http.MethodGet, baseURL+"/config/server/version", nil)

if err != nil {

return false, err

}

resp, err := client.Do(req)

if err != nil {

return false, err

}

defer func() { _ = resp.Body.Close() }()

if resp.StatusCode != http.StatusOK {

return false, fmt.Errorf("status %d", resp.StatusCode)

}

body, err := io.ReadAll(resp.Body)

if err != nil {

return false, err

}

body = bytes.TrimPrefix(body, []byte(")]}'\n"))

var version string

return json.Unmarshal(body, &version) == nil && version != "", nil

}

func probeURL(ctx context.Context, client *http.Client, url string) (bool, error) {

req, err := http.NewRequestWithContext(ctx, http.MethodGet, url, nil)

if err != nil {

return false, err

}

resp, err := client.Do(req)

if err != nil {

return false, err

}

defer func() { _ = resp.Body.Close() }()

return resp.StatusCode == http.StatusOK, nil

}

forge/detect.go at main · git-pkgs/forge · GitHub

Uh oh!

Uh oh!

forge/detect.go at main · git-pkgs/forge · GitHub

Uh oh!

Uh oh!

Expand file tree

Breadcrumbs

detect.go

Latest commit

History

Breadcrumbs

detect.go

File metadata and controls