repo.me/internal/interfaces/blobs/local.go

package blobs

import (
	"context"
	"errors"
	"io"
	"nf-repo/internal/interfaces"
	"nf-repo/internal/model"
	"nf-repo/internal/opt"
	"nf-repo/internal/verify"
	"os"
	"path"
	"sync"
)

type localHandler struct {
	base string
	sync.Mutex
}

func (l *localHandler) path(hash model.Hash) string {
	//return path.Join(l.base, hash.Hex)
	dir := path.Join(l.base, hash.Hex[:2], hash.Hex[2:4])
	_ = os.MkdirAll(dir, 0755)
	return path.Join(dir, hash.Hex)
}

func (l *localHandler) Get(ctx context.Context, repo string, hash model.Hash) (io.ReadCloser, error) {
	var (
		err error
		f   *os.File
	)

	l.Lock()
	defer l.Unlock()

	if f, err = os.Open(l.path(hash)); err != nil {
		if errors.Is(err, os.ErrNotExist) {
			return nil, opt.ErrNotFound
		}

		return nil, err
	}

	return f, nil
}

func (l *localHandler) Stat(ctx context.Context, repo string, hash model.Hash) (int64, error) {
	var (
		err  error
		info os.FileInfo
		sp   = l.path(hash)
	)

	l.Lock()
	defer l.Unlock()

	if info, err = os.Stat(sp); err != nil {
		if errors.Is(err, os.ErrNotExist) {
			return 0, opt.ErrNotFound
		}

		return 0, err
	}

	return info.Size(), nil
}

func (l *localHandler) Put(ctx context.Context, repo string, hash model.Hash, rc io.ReadCloser) error {
	var (
		err error
		f   *os.File
		nrc io.ReadCloser
	)

	l.Lock()
	defer l.Unlock()

	if nrc, err = verify.ReadCloser(rc, verify.SizeUnknown, hash); err != nil {
		return err
	}
	defer nrc.Close()

	if f, err = os.OpenFile(l.path(hash), os.O_CREATE|os.O_RDWR|os.O_TRUNC, 0644); err != nil {
		return err
	}

	if _, err = io.Copy(f, rc); err != nil {
		return err
	}

	return nil
}

func (l *localHandler) Delete(ctx context.Context, repo string, hash model.Hash) error {
	var (
		err      error
		info     os.FileInfo
		filename = l.path(hash)
	)

	l.Lock()
	defer l.Unlock()

	if info, err = os.Stat(filename); err != nil {
		if errors.Is(err, os.ErrNotExist) {
			return opt.ErrNotFound
		}

		return err
	}

	_ = info

	return os.Remove(filename)
}

func NewLocalBlobHandler(baseDir string) interfaces.BlobHandler {
	_ = os.MkdirAll(baseDir, 0755)
	return &localHandler{base: baseDir}
}