2fas-server/internal/pass/pairing/memorystore.go

package pairing

import (
	"context"
	"errors"
	"sync"
	"time"
)

// MemoryStore keeps in memory pairing between extension and mobile.
//
// TODO: check ttlcache pkg, right now entries are not invalidated.
type MemoryStore struct {
	mu            sync.Mutex
	extensionsMap map[string]Item
}

type Item struct {
	ExtensionID string
	Expires     time.Time
	PairingInfo PairingInfo
}

func NewMemoryStore() *MemoryStore {
	return &MemoryStore{
		extensionsMap: make(map[string]Item),
	}
}

func (s *MemoryStore) AddExtension(_ context.Context, extensionID string) {
	s.setItem(extensionID, Item{ExtensionID: extensionID})
}

func (s *MemoryStore) ExtensionExists(_ context.Context, extensionID string) bool {
	_, ok := s.getItem(extensionID)
	return ok
}

func (s *MemoryStore) GetPairingInfo(ctx context.Context, extensionID string) (PairingInfo, error) {
	v, ok := s.getItem(extensionID)
	if !ok {
		return PairingInfo{}, errors.New("extension does not exists")
	}
	return v.PairingInfo, nil
}

func (s *MemoryStore) SetPairingInfo(ctx context.Context, extensionID string, pi PairingInfo) error {
	_, ok := s.getItem(extensionID)
	if !ok {
		return errors.New("extension does not exists")
	}
	s.setItem(extensionID, Item{
		ExtensionID: extensionID,
		Expires:     time.Time{},
		PairingInfo: pi,
	})
	return nil
}

func (s *MemoryStore) setItem(key string, item Item) {
	s.mu.Lock()
	defer s.mu.Unlock()
	s.extensionsMap[key] = item
}

func (s *MemoryStore) getItem(key string) (Item, bool) {
	s.mu.Lock()
	defer s.mu.Unlock()
	v, ok := s.extensionsMap[key]
	return v, ok
}