store.go

// Package inmem implements an in memory storage backend for tests.
package inmem

import (
	"context"
	"fmt"
	"sort"
	"strings"
	"sync"
	"time"

	"github.com/blang/semver/v4"
	"github.com/gopasspw/gopass/internal/backend"
	"golang.org/x/exp/maps"
)

// InMem is a thread-safe in-memory store.
type InMem struct {
	sync.Mutex
	data map[string][]byte
}

// New creates a new mock.
func New() *InMem {
	return &InMem{
		data: make(map[string][]byte, 128),
	}
}

// Get retrieves a value.
func (m *InMem) Get(ctx context.Context, name string) ([]byte, error) {
	m.Lock()
	defer m.Unlock()

	if m.data == nil {
		return nil, fmt.Errorf("entry not found")
	}

	sec, found := m.data[name]
	if !found {
		// not found
		return nil, fmt.Errorf("entry not found")
	}

	// found
	return sec, nil
}

// Set writes a value.
func (m *InMem) Set(ctx context.Context, name string, value []byte) error {
	m.Lock()
	defer m.Unlock()

	m.data[name] = value

	return nil
}

// Delete removes a value.
func (m *InMem) Delete(ctx context.Context, name string) error {
	m.Lock()
	defer m.Unlock()

	delete(m.data, name)

	return nil
}

// Exists checks is a value exists.
func (m *InMem) Exists(ctx context.Context, name string) bool {
	m.Lock()
	defer m.Unlock()

	_, found := m.data[name]

	return found
}

// List shows all values.
func (m *InMem) List(ctx context.Context, prefix string) ([]string, error) {
	m.Lock()
	defer m.Unlock()

	keys := maps.Keys(m.data)
	sort.Strings(keys)

	return keys, nil
}

// IsDir returns true if the entry is a directory.
func (m *InMem) IsDir(ctx context.Context, name string) bool {
	m.Lock()
	defer m.Unlock()

	for k := range m.data {
		if strings.HasPrefix(k, name+"/") {
			return true
		}
	}

	return false
}

// Prune removes a directory.
func (m *InMem) Prune(ctx context.Context, prefix string) error {
	m.Lock()
	defer m.Unlock()

	deleted := 0

	for k := range m.data {
		if strings.HasPrefix(k, prefix+"/") {
			delete(m.data, k)
			deleted++
		}
	}

	if deleted < 1 {
		return fmt.Errorf("not found")
	}

	return nil
}

// Name returns the name of this backend.
func (m *InMem) Name() string {
	return "inmem"
}

// Version returns the version of this backend.
func (m *InMem) Version(context.Context) semver.Version {
	return semver.Version{Major: 1}
}

// String implement fmt.Stringer.
func (m *InMem) String() string {
	return "inmem()"
}

// Path returns inmem.
func (m *InMem) Path() string {
	return "inmem"
}

// Fsck always returns nil.
func (m *InMem) Fsck(ctx context.Context) error {
	return nil
}

// Add does nothing.
func (m *InMem) Add(ctx context.Context, args ...string) error {
	return nil
}

// TryAdd does nothing.
func (m *InMem) TryAdd(ctx context.Context, args ...string) error {
	return nil
}

// Commit does nothing.
func (m *InMem) Commit(ctx context.Context, msg string) error {
	return nil
}

// TryCommit does nothing.
func (m *InMem) TryCommit(ctx context.Context, msg string) error {
	return nil
}

// Push does nothing.
func (m *InMem) Push(ctx context.Context, origin, branch string) error {
	return nil
}

// TryPush does nothing.
func (m *InMem) TryPush(ctx context.Context, origin, branch string) error {
	return nil
}

// Pull does nothing.
func (m *InMem) Pull(ctx context.Context, origin, branch string) error {
	return nil
}

// Cmd does nothing.
func (m *InMem) Cmd(ctx context.Context, name string, args ...string) error {
	return nil
}

// Init does nothing.
func (m *InMem) Init(context.Context, string, string) error {
	return nil
}

// InitConfig does nothing.
func (m *InMem) InitConfig(context.Context, string, string) error {
	return nil
}

// AddRemote does nothing.
func (m *InMem) AddRemote(ctx context.Context, remote, url string) error {
	return nil
}

// RemoveRemote does nothing.
func (m *InMem) RemoveRemote(ctx context.Context, remote string) error {
	return nil
}

// Revisions is not implemented.
func (m *InMem) Revisions(context.Context, string) ([]backend.Revision, error) {
	return []backend.Revision{
		{
			Hash: "latest",
			Date: time.Now(),
		},
	}, nil
}

// GetRevision is not implemented.
func (m *InMem) GetRevision(context.Context, string, string) ([]byte, error) {
	return []byte("foo\nbar"), nil
}

// Status is not implemented.
func (m *InMem) Status(context.Context) ([]byte, error) {
	return []byte(""), nil
}

// Compact is not implemented.
func (m *InMem) Compact(context.Context) error {
	return nil
}

// Link is not implemented.
func (m *InMem) Link(context.Context, string, string) error {
	return nil
}

// Move is not implemented.
func (m *InMem) Move(context.Context, string, string, bool) error {
	return nil
}