artifactapi/internal/virtual/merger.go

package virtual

import (
	"fmt"

	"git.unkin.net/unkin/artifactapi/pkg/models"
)

type MemberIndex struct {
	RemoteName string
	Body       []byte
}

type IndexMerger interface {
	MergeIndexes(members []MemberIndex, proxyBaseURL string) ([]byte, error)
}

var mergers = map[models.PackageType]IndexMerger{}

func RegisterMerger(pt models.PackageType, m IndexMerger) {
	mergers[pt] = m
}

func GetMerger(pt models.PackageType) (IndexMerger, error) {
	m, ok := mergers[pt]
	if !ok {
		return nil, fmt.Errorf("no merger registered for package type %q", pt)
	}
	return m, nil
}