working commit

pkg/repocli/imager.go

@@ -3,6 +3,7 @@ package repocli
 import (
 	"bytes"
 	"context"
+	"encoding/json"
 	"errors"
 	"io"
 	"os"
@@ -12,19 +13,12 @@ import (
 	ocispec "github.com/opencontainers/image-spec/specs-go/v1"
 )
 
-type Imageer interface {
-	WriteManifest(mime string, manifest []byte) error
-	ReadManifest(digest string) (bool, string, []byte, error)
-	WriteBlob(digest string, reader io.Reader) error
-	ReadBlob(digest string, writer io.Writer) error
-}
-
 type Imager struct {
 	index ocispec.Index
 	place string
 }
 
-func NewImager(place string) *Imager {
+func NewEmptyImager(place string) *Imager {
 	imager := &Imager{
 		index: ocispec.Index{
 			MediaType: MediaTypeOIIv1,
@@ -32,7 +26,7 @@ func NewImager(place string) *Imager {
 		},
 		place: place,
 	}
-	//imager.SchemaVersion = 2
+	imager.index.SchemaVersion = 2
 	return imager
 }
 
@@ -42,32 +36,30 @@ func (ima *Imager) WriteManifest(ctx context.Context, digest, mime string, paylo
 	if err != nil {
 		return err
 	}
-	var subdir string
-	switch digestobj.Algorithm() {
-	case ocidigest.SHA256:
-		subdir = "sha256"
-	case ocidigest.SHA384:
-		subdir = "sha384"
-	case ocidigest.SHA512:
-		subdir = "sha512"
-	default:
-		err := errors.New("Unknown digest type")
-		return err
-	}
+	subdir := string(digestobj.Algorithm())
 	dir := filepath.Join(ima.place, subdir)
 	err = os.MkdirAll(dir, 0750)
 	if err != nil {
 		return err
 	}
-	fpath := filepath.Join(dir, digestobj.Encoded())
-	file, err := os.OpenFile(fpath, os.O_CREATE|os.O_WRONLY, 0640)
+	mpath := filepath.Join(dir, digestobj.Encoded())
+	mfile, err := os.OpenFile(mpath, os.O_CREATE|os.O_WRONLY|os.O_TRUNC, 0640)
 	if err != nil {
 		return err
 	}
+	defer mfile.Close()
 	reader := bytes.NewReader(payload)
-	size, err := io.Copy(file, reader)
-	if size != int64(len(payload)) {
-		err = errors.New("Mismatch sizes")
+
+	verifier := digestobj.Verifier()
+	mwriter := io.MultiWriter(verifier, mfile)
+	recsize, err := io.Copy(mwriter, reader)
+	size := int64(len(payload))
+	if size != recsize {
+		err = errors.New("Mismatch manigest sizes")
+		return err
+	}
+	if !verifier.Verified() {
+		err = errors.New("Mismatch manifest digests")
 		return err
 	}
 	descr := ocispec.Descriptor{
@@ -76,7 +68,26 @@ func (ima *Imager) WriteManifest(ctx context.Context, digest, mime string, paylo
 		Size:      size,
 	}
 	ima.index.Manifests = append(ima.index.Manifests, descr)
-	// TODO: flush index
+	indexdat, err := json.Marshal(ima.index)
+	if err != nil {
+		return err
+	}
+	// Flush index
+	ipath := filepath.Join(dir, "index.json")
+	ifile, err := os.OpenFile(ipath, os.O_CREATE|os.O_WRONLY|os.O_TRUNC, 0640)
+	if err != nil {
+		return err
+	}
+	defer ifile.Close()
+	reader = bytes.NewReader(indexdat)
+	recsize, err = io.Copy(ifile, reader)
+	if err != nil {
+		return err
+	}
+	if recsize != int64(len(indexdat)) {
+		err = errors.New("Mismatch index sizes")
+		return err
+	}
 	return err
 }
 
@@ -86,31 +97,20 @@ func (ima *Imager) WriteLayer(ctx context.Context, digest string, size int64, re
 	if err != nil {
 		return err
 	}
-	var subdir string
-	switch digestobj.Algorithm() {
-	case ocidigest.SHA256:
-		subdir = "sha256"
-	case ocidigest.SHA384:
-		subdir = "sha384"
-	case ocidigest.SHA512:
-		subdir = "sha512"
-	default:
-		err := errors.New("Unknown digest type")
-		return err
-	}
+	subdir := string(digestobj.Algorithm())
 	dir := filepath.Join(ima.place, subdir)
 	err = os.MkdirAll(dir, 0750)
 	if err != nil {
 		return err
 	}
 	fpath := filepath.Join(dir, digestobj.Encoded())
-	file, err := os.OpenFile(fpath, os.O_CREATE|os.O_WRONLY, 0640)
+	file, err := os.OpenFile(fpath, os.O_CREATE|os.O_WRONLY|os.O_TRUNC, 0640)
 	if err != nil {
 		return err
 	}
 	verifier := digestobj.Verifier()
-	mw := io.MultiWriter(verifier, file)
-	recsize, err := io.Copy(mw, reader)
+	mwriter := io.MultiWriter(verifier, file)
+	recsize, err := io.Copy(mwriter, reader)
 	if size != recsize {
 		err = errors.New("Mismatch sizes")
 		return err