added repocli

2026-03-02 09:07:57 +02:00
parent e67f42a821
commit 1b025180f0
25 changed files with 1320 additions and 0 deletions
@@ -0,0 +1,159 @@
+package repocli
+
+import (
+	"github.com/stretchr/testify/require"
+
+	"bytes"
+	"context"
+	"encoding/json"
+	"fmt"
+	"math/rand"
+	"testing"
+	"time"
+)
+
+func xxxTestClientGetManifest(t *testing.T) {
+	rawrepo := "mirror.gcr.io/alpine"
+	tags := []string{
+		"3.20.0",
+		"sha256:29e5ba63e79337818e6c63cfcc68e2ab4e9ca483853b2de303bfbfba9372426c",
+	}
+	for _, tag := range tags {
+		cli := NewClient()
+		ctx, _ := context.WithTimeout(context.Background(), 10*time.Second)
+		exist, mime, man, err := cli.GetManifest(ctx, rawrepo, tag)
+		require.NoError(t, err)
+		require.True(t, exist)
+
+		fmt.Printf("Type: %s\n", mime)
+		buffer := bytes.NewBuffer(nil)
+		err = json.Indent(buffer, man, " ", "    ")
+		require.NoError(t, err)
+		//fmt.Printf("%s\n", buffer.String())
+	}
+}
+
+func xxxTestClientManifestExists(t *testing.T) {
+	rawrepo := "mirror.gcr.io/alpine"
+	tags := []string{
+		"3.20.0",
+		"sha256:29e5ba63e79337818e6c63cfcc68e2ab4e9ca483853b2de303bfbfba9372426c",
+	}
+	for _, tag := range tags {
+
+		cli := NewClient()
+		ctx, _ := context.WithTimeout(context.Background(), 10*time.Second)
+		exist, mime, size, csum, err := cli.ManifestExists(ctx, rawrepo, tag)
+		require.NoError(t, err)
+		require.True(t, exist)
+
+		fmt.Printf("MIME: %s\n", mime)
+		fmt.Printf("Size: %d\n", size)
+		fmt.Printf("Sum: %s\n", csum)
+		fmt.Printf("Typ: %d\n", DigestType(csum))
+	}
+}
+
+func xxxTestClientBlobExists(t *testing.T) {
+	rawrepos := []string{
+		"mirror.gcr.io/alpine",
+	}
+	for _, rawrepo := range rawrepos {
+		cli := NewClient()
+		ctx, _ := context.WithTimeout(context.Background(), 10*time.Second)
+		digest := "sha256:3b8747b05489980f63da1d2b8e5a444c55777f69540394397b0bc1c76c3e41f2"
+		exist, size, err := cli.BlobExists(ctx, rawrepo, digest)
+		require.NoError(t, err)
+		require.True(t, exist)
+
+		fmt.Printf("Size: %d\n", size)
+	}
+}
+
+func xxxTestClientGetBlob(t *testing.T) {
+	rawrepos := []string{
+		"mirror.gcr.io/alpine",
+	}
+	for _, rawrepo := range rawrepos {
+		cli := NewClient()
+		ctx, _ := context.WithTimeout(context.Background(), 10*time.Second)
+		buffer := bytes.NewBuffer(nil)
+		digest := "sha256:3b8747b05489980f63da1d2b8e5a444c55777f69540394397b0bc1c76c3e41f2"
+		exist, err := cli.GetBlob(ctx, rawrepo, buffer, digest)
+		require.NoError(t, err)
+		require.True(t, exist)
+		fmt.Printf("Size: %d\n", len(buffer.Bytes()))
+	}
+}
+
+func xxxxTestClientGetUpload(t *testing.T) {
+	rawrepos := []string{
+		"mstore:mstore@localhost:1025/alpine:3.20.0",
+	}
+	cli := NewClient()
+	cli.UseMiddleware(NewBasicAuthMiddleware("mstore", "mstore"))
+
+	for _, rawrepo := range rawrepos {
+		var err error
+		var loc string
+		{
+			ctx, _ := context.WithTimeout(context.Background(), 10*time.Second)
+			loc, err = cli.GetUpload(ctx, rawrepo)
+			require.NoError(t, err)
+			fmt.Printf("Upload Location: %s\n", loc)
+		}
+		{
+			srcsize := 1024 + 145
+			srcdata := make([]byte, srcsize)
+			_, err = rand.Read(srcdata)
+			require.NoError(t, err)
+
+			src := bytes.NewReader(srcdata)
+			//digest := SHA256Digest(srcdata)
+
+			ctx, _ := context.WithTimeout(context.Background(), 10*time.Second)
+			bloc, err := cli.PatchUpload(ctx, rawrepo, src, loc, int64(len(srcdata)))
+			require.NoError(t, err)
+			fmt.Printf("Path Location: %s\n", bloc)
+		}
+		{
+			srcsize := 1024 + 145
+			srcdata := make([]byte, srcsize)
+			_, err = rand.Read(srcdata)
+			require.NoError(t, err)
+
+			src := bytes.NewReader(srcdata)
+			digest := SHA256Digest(srcdata)
+
+			ctx, _ := context.WithTimeout(context.Background(), 10*time.Second)
+			bloc, err := cli.PutUpload(ctx, rawrepo, src, loc, digest, int64(len(srcdata)))
+			require.NoError(t, err)
+			fmt.Printf("Put blob Location: %s\n", bloc)
+		}
+
+	}
+}
+
+func xxxxTestClientGetToken(t *testing.T) {
+	var token string
+	var err error
+	{
+		cli := NewClient()
+		cli.UseMiddleware(NewBasicAuthMiddleware("onborodin", "2Albert334"))
+		ctx, _ := context.WithTimeout(context.Background(), 10*time.Second)
+		token, err = cli.GetToken(ctx, "https://auth.docker.io/token")
+		require.NoError(t, err)
+
+	}
+	fmt.Printf("Token: %s\n", token)
+	{
+		rawrepo := "docker.io/onborodin/toolbox:0.18"
+		cli := NewClient()
+		cli.UseMiddleware(NewBearerAuthMiddleware(token))
+		ctx, _ := context.WithTimeout(context.Background(), 10*time.Second)
+		loc, err := cli.GetUpload(ctx, rawrepo)
+		require.NoError(t, err)
+		fmt.Printf("Upload Location: %s\n", loc)
+	}
+
+}