7a4bb863dd
ioutil has been deprecated since go 1.16, this fixes all of the deprecated functions we are using: ioutil.ReadFile -> os.ReadFile ioutil.ReadAll -> io.ReadAll ioutil.WriteFile -> os.WriteFile ioutil.TempFile -> os.CreateTemp ioutil.TempDir -> os.MkdirTemp All of the above are a simple name change, the function arguments and results are exactly the same as before. ioutil.ReadDir -> os.ReadDir now returns a os.DirEntry but the IsDir and Name functions work the same. The difference is that the FileInfo must be retrieved with the Info() function which can also return an error. These were identified by running: golangci-lint run --build-tags=integration ./...
104 lines
3.3 KiB
Go
104 lines
3.3 KiB
Go
package main
|
|
|
|
import (
|
|
"crypto/rand"
|
|
"fmt"
|
|
"math"
|
|
"math/big"
|
|
"os"
|
|
|
|
"github.com/osbuild/osbuild-composer/internal/upload/oci"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var (
|
|
tenancy string
|
|
region string
|
|
userID string
|
|
privateKeyFile string
|
|
fingerprint string
|
|
bucketName string
|
|
bucketNamespace string
|
|
fileName string
|
|
objectName string
|
|
compartment string
|
|
)
|
|
|
|
var uploadCmd = &cobra.Command{
|
|
Example: "This tool uses the $HOME/.oci/config file to create the OCI client and can be\noverridden using CLI flags.",
|
|
SilenceUsage: true,
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
uploader, err := uploaderFromConfig()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
file, err := os.Open(fileName)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer file.Close()
|
|
|
|
imageID, err := uploader.Upload(objectName, bucketName, bucketNamespace, file, compartment, fileName)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to upload the image: %v", err)
|
|
}
|
|
|
|
fmt.Printf("Image %s was uploaded and created successfully\n", imageID)
|
|
return nil
|
|
},
|
|
}
|
|
|
|
func main() {
|
|
i, _ := rand.Int(rand.Reader, big.NewInt(math.MaxInt64))
|
|
uploadCmd.Flags().StringVarP(&tenancy, "tenancy", "t", "", "target tenancy")
|
|
uploadCmd.Flags().StringVarP(®ion, "region", "r", "", "target region")
|
|
uploadCmd.Flags().StringVarP(&userID, "user-id", "u", "", "user OCI ID")
|
|
uploadCmd.Flags().StringVarP(&bucketName, "bucket-name", "b", "", "target OCI bucket name")
|
|
uploadCmd.Flags().StringVarP(&bucketNamespace, "bucket-namespace", "", "", "target OCI bucket namespace")
|
|
uploadCmd.Flags().StringVarP(&fileName, "filename", "f", "", "image file to upload")
|
|
uploadCmd.Flags().StringVarP(&objectName, "object-name", "o", fmt.Sprintf("osbuild-upload-%v", i), "the target name of the uploaded object in the bucket")
|
|
uploadCmd.Flags().StringVarP(&privateKeyFile, "private-key", "p", "", "private key for authenticating OCI API requests")
|
|
uploadCmd.Flags().StringVarP(&fingerprint, "fingerprint", "", "", "the private key's fingerprint")
|
|
uploadCmd.Flags().StringVarP(&compartment, "compartment-id", "c", "", "the compartment ID of the target image")
|
|
_ = uploadCmd.MarkFlagRequired("bucket-name")
|
|
_ = uploadCmd.MarkFlagRequired("bucket-namespace")
|
|
_ = uploadCmd.MarkFlagRequired("compartment-id")
|
|
_ = uploadCmd.MarkFlagRequired("filename")
|
|
if err := uploadCmd.Execute(); err != nil {
|
|
fmt.Fprintln(os.Stderr, err)
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
func uploaderFromConfig() (oci.Uploader, error) {
|
|
if privateKeyFile != "" {
|
|
if tenancy == "" || region == "" || userID == "" || fingerprint == "" {
|
|
return nil, fmt.Errorf("when suppling a private key the following args are mandatory as well:" +
|
|
" fingerprint, tenancy, region, and user-id")
|
|
}
|
|
pk, err := os.ReadFile(privateKeyFile)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to read private key file %w", err)
|
|
}
|
|
uploader, err := oci.NewClient(
|
|
&oci.ClientParams{
|
|
Tenancy: tenancy,
|
|
User: userID,
|
|
Region: region,
|
|
PrivateKey: string(pk),
|
|
Fingerprint: fingerprint,
|
|
})
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to create an OCI client %w", err)
|
|
}
|
|
return uploader, nil
|
|
}
|
|
|
|
fmt.Printf("Creating an uploader from default config\n")
|
|
uploader, err := oci.NewClient(nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return uploader, nil
|
|
}
|