unkinben
cca640958d
When a helm repo's base URL includes a path (e.g. stakater.github.io/ stakater-charts), the merger extracted the full path from download URLs causing duplication: /stakater-charts/stakater-charts/reloader.tgz. Now strips the base URL's path prefix from same-host download URLs so the proxy constructs the correct upstream URL.
130 lines
2.9 KiB
Go
130 lines
2.9 KiB
Go
package virtual
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"gopkg.in/yaml.v3"
|
|
|
|
"git.unkin.net/unkin/artifactapi/pkg/models"
|
|
)
|
|
|
|
func init() {
|
|
RegisterMerger(models.PackageHelm, &HelmMerger{})
|
|
}
|
|
|
|
type HelmMerger struct{}
|
|
|
|
type helmIndex struct {
|
|
APIVersion string `yaml:"apiVersion"`
|
|
Entries map[string][]helmChartVersion `yaml:"entries"`
|
|
Generated string `yaml:"generated,omitempty"`
|
|
}
|
|
|
|
type helmChartVersion struct {
|
|
Name string `yaml:"name"`
|
|
Version string `yaml:"version"`
|
|
URLs []string `yaml:"urls"`
|
|
rest map[string]any
|
|
}
|
|
|
|
func (m *HelmMerger) MergeIndexes(members []MemberIndex, proxyBaseURL string) ([]byte, error) {
|
|
merged := &helmIndex{
|
|
APIVersion: "v1",
|
|
Entries: make(map[string][]helmChartVersion),
|
|
}
|
|
|
|
seen := map[string]map[string]bool{}
|
|
|
|
for _, member := range members {
|
|
var idx helmIndex
|
|
if err := yaml.Unmarshal(member.Body, &idx); err != nil {
|
|
continue
|
|
}
|
|
|
|
for chart, versions := range idx.Entries {
|
|
if seen[chart] == nil {
|
|
seen[chart] = map[string]bool{}
|
|
}
|
|
for _, ver := range versions {
|
|
key := chart + ":" + ver.Version
|
|
if seen[chart][ver.Version] {
|
|
continue
|
|
}
|
|
seen[chart][ver.Version] = true
|
|
|
|
if proxyBaseURL != "" {
|
|
routePrefix := "remote"
|
|
if member.RepoType == "local" {
|
|
routePrefix = "local"
|
|
}
|
|
baseHost := extractHost(member.BaseURL)
|
|
|
|
for i, u := range ver.URLs {
|
|
if strings.HasPrefix(u, "http://") || strings.HasPrefix(u, "https://") {
|
|
if baseHost != "" && extractHost(u) != baseHost {
|
|
continue
|
|
}
|
|
relPath := extractPathRelativeToBase(u, member.BaseURL)
|
|
ver.URLs[i] = fmt.Sprintf("%s/api/v1/%s/%s/%s",
|
|
strings.TrimRight(proxyBaseURL, "/"),
|
|
routePrefix,
|
|
member.RemoteName,
|
|
relPath)
|
|
} else {
|
|
ver.URLs[i] = fmt.Sprintf("%s/api/v1/%s/%s/%s",
|
|
strings.TrimRight(proxyBaseURL, "/"),
|
|
routePrefix,
|
|
member.RemoteName,
|
|
u)
|
|
}
|
|
}
|
|
}
|
|
|
|
merged.Entries[chart] = append(merged.Entries[chart], ver)
|
|
_ = key
|
|
}
|
|
}
|
|
}
|
|
|
|
return yaml.Marshal(merged)
|
|
}
|
|
|
|
func extractHost(rawURL string) string {
|
|
idx := strings.Index(rawURL, "://")
|
|
if idx == -1 {
|
|
return ""
|
|
}
|
|
rest := rawURL[idx+3:]
|
|
slashIdx := strings.Index(rest, "/")
|
|
if slashIdx == -1 {
|
|
return rest
|
|
}
|
|
return rest[:slashIdx]
|
|
}
|
|
|
|
func extractPathRelativeToBase(rawURL, baseURL string) string {
|
|
fullPath := extractPath(rawURL)
|
|
basePath := extractPath(baseURL)
|
|
if basePath != "" {
|
|
basePath = strings.TrimRight(basePath, "/") + "/"
|
|
if strings.HasPrefix(fullPath, basePath) {
|
|
return fullPath[len(basePath):]
|
|
}
|
|
}
|
|
return fullPath
|
|
}
|
|
|
|
func extractPath(rawURL string) string {
|
|
idx := strings.Index(rawURL, "://")
|
|
if idx == -1 {
|
|
return rawURL
|
|
}
|
|
rest := rawURL[idx+3:]
|
|
slashIdx := strings.Index(rest, "/")
|
|
if slashIdx == -1 {
|
|
return ""
|
|
}
|
|
return rest[slashIdx+1:]
|
|
}
|