passed/storage/storage.go

package storage

import (
	"errors"
	"log"
	"math/rand/v2"
	"os"
	"strings"
	"time"
)

var (
	ErrNotFound = errors.New("password not found")
	ErrFull     = errors.New("storage is filled")
)

type entry struct {
	Password  []byte
	ExpiresAt time.Time
}

type Store interface {
	CreatePassword(password []byte, expiresAt time.Time) (string, error)
	GetPassword(id string) ([]byte, error)
	HasPassword(id string) (bool, error)
	Close() error
}

func NewStore() (Store, error) {
	storeType := os.Getenv("PASSED_STORE_TYPE")
	storeType = strings.ToLower(storeType)

	switch storeType {
	case "ram":
		return NewRamStore(20 * time.Second), nil
	case "dir":
		path := os.Getenv("PASSED_STORE_DIR_PATH")
		if path == "" {
			log.Printf("No PASSED_STORE_DIR_PATH provided, defaulting to \"passwords\".")
			path = "passwords"
		}

		return NewDirStore(60*time.Second, path), nil

	default:
		log.Printf("No PASSED_STORE_TYPE provided, defaulting to memory store.")
		return NewRamStore(20 * time.Second), nil
	}
}

func generateId(length int) string {
	runes := []rune("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ")

	str := make([]rune, length)
	for i := range str {
		str[i] = runes[rand.IntN(len(runes))]
	}

	return string(str)
}