enhancement: make cdn url configurable during import

rocky-linux · Sep 5, 2021 · 360c4d5 · 360c4d5
1 parent fff8896
commit 360c4d5
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 1 deletion.
diff --git a/pkg/data/process.go b/pkg/data/process.go
@@ -53,4 +53,5 @@ type ProcessData struct {
 	ModuleFallbackStream string
 	AllowStreamBranches  bool
 	FsCreator            FsCreatorFunc
+	CdnUrl               string
 }
diff --git a/pkg/modes/git.go b/pkg/modes/git.go
@@ -240,7 +240,7 @@ func (g *GitMode) WriteSource(pd *data.ProcessData, md *data.ModeData) error {
 				body = fromBlobStorage
 				log.Printf("downloading %s from blob storage", hash)
 			} else {
-				url := fmt.Sprintf("https://git.centos.org/sources/%s/%s/%s", md.Name, branchName, hash)
+				url := fmt.Sprintf("%s/%s/%s/%s", pd.CdnUrl, md.Name, branchName, hash)
 				log.Printf("downloading %s", url)
 
 				req, err := http.NewRequest("GET", url, nil)

diff --git a/pkg/srpmproc/process.go b/pkg/srpmproc/process.go
@@ -83,6 +83,7 @@ type ProcessDataRequest struct {
 	NoStorageUpload      bool
 	NoStorageDownload    bool
 	SingleTag            string
+	CdnUrl               string
 }
 
 func gitlabify(str string) string {
@@ -126,6 +127,9 @@ func NewProcessData(req *ProcessDataRequest) (*data.ProcessData, error) {
 	if req.BranchPrefix == "" {
 		req.BranchPrefix = "r"
 	}
+	if req.CdnUrl == "" {
+		req.CdnUrl = "https://git.centos.org/sources"
+	}
 
 	// Validate required
 	if req.Package == "" {