rocky-linux/peridot
4
0
Fork 0
mirror of https://github.com/rocky-linux/peridot.git synced 2024-11-21 12:41:27 +00:00
Code Issues Projects Releases Activity

Merge pull request #194 from rocky-linux/feat/import-skip-upload

Browse Source 
feat: allow rpm import to skip upload OR import step
This commit is contained in:
resf-prow[bot] 2024-10-18 15:50:49 +00:00 committed by GitHub
commit 549b3a8294
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 43 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
peridot/cmd/v1/peridot/BUILD.bazel
View File

@ -64,8 +64,8 @@ pkg_rpm(
srcs = [":peridot-files"],
license = "MIT",
summary = "Peridot Command Line Interface",
version = "0.2.0",
release = "0",
version = "0.2.1",
release = "1",
architecture = "x86_64",
description = "A command line interface to interact with the Peridot build system",
source_date_epoch = 0,

52
peridot/cmd/v1/peridot/build_rpm_import.go
View File

@ -31,13 +31,16 @@
package main
import (
"crypto/sha256"
"encoding/base64"
"github.com/spf13/cobra"
"io/ioutil"
"encoding/hex"
"log"
"openapi.peridot.resf.org/peridotopenapi"
"os"
"strings"
"time"
"github.com/spf13/cobra"
"openapi.peridot.resf.org/peridotopenapi"
)
type LookasideUploadTask struct {
@ -57,9 +60,11 @@ var buildRpmImport = &cobra.Command{
}
var buildRpmImportForceOverride bool
var skipStep string
func init() {
buildRpmImport.Flags().BoolVar(&buildRpmImportForceOverride, "force-override", true, "Force override even if version exists (default: true)")
buildRpmImport.Flags().StringVarP(&skipStep, "skip", "s", "", "which step to skip")
}
func isFile(path string) bool {
@ -74,6 +79,20 @@ func buildRpmImportMn(_ *cobra.Command, args []string) {
// Ensure project id exists
projectId := mustGetProjectID()
var skipUpload = false
var skipImport = false
if skipStep != "" {
switch strings.ToLower(skipStep) {
case "upload":
skipUpload = true
case "import":
skipImport = true
default:
log.Fatalf("invalid skip step: %s", skipStep)
}
}
// Ensure all args are valid files
for _, arg := range args {
if !isFile(arg) {
@ -85,16 +104,27 @@ func buildRpmImportMn(_ *cobra.Command, args []string) {
var blobs []string
projectCl := getClient(serviceProject).(peridotopenapi.ProjectServiceApi)
for _, arg := range args {
bts, err := ioutil.ReadFile(arg)
errFatal(err)
base64EncodedBytes := base64.StdEncoding.EncodeToString(bts)
res, _, err := projectCl.LookasideFileUpload(getContext()).Body(peridotopenapi.V1LookasideFileUploadRequest{
File: &base64EncodedBytes,
}).Execute()
bts, err := os.ReadFile(arg)
errFatal(err)
log.Printf("Uploaded %s to lookaside", arg)
blobs = append(blobs, res.GetDigest())
hash := sha256.Sum256(bts)
shasum := hex.EncodeToString(hash[:])
if !skipUpload {
base64EncodedBytes := base64.StdEncoding.EncodeToString(bts)
_, _, err := projectCl.LookasideFileUpload(getContext()).Body(peridotopenapi.V1LookasideFileUploadRequest{
File: &base64EncodedBytes,
}).Execute()
errFatal(err)
log.Printf("Uploaded %s to lookaside", arg)
}
log.Printf("Will upload %s to lookaside for %s", shasum, arg)
blobs = append(blobs, shasum)
}
if skipImport {
return
}
taskCl := getClient(serviceTask).(peridotopenapi.TaskServiceApi)