storage_memory.go

// Copyright 2017-2026 Allow2 Pty Ltd. All rights reserved.
// Use of this source code is governed by the Allow2 API and SDK Licence.

package allow2service

import (
	"sync"
)

// MemoryTokenStorage is an in-memory token storage.
// Suitable for testing and development. Tokens are lost when the process exits.
type MemoryTokenStorage struct {
	mu    sync.RWMutex
	store map[string]*OAuthTokens
}

// NewMemoryTokenStorage creates a new MemoryTokenStorage.
func NewMemoryTokenStorage() *MemoryTokenStorage {
	return &MemoryTokenStorage{
		store: make(map[string]*OAuthTokens),
	}
}

// Store persists tokens for a user.
func (s *MemoryTokenStorage) Store(userID string, tokens *OAuthTokens) error {
	s.mu.Lock()
	defer s.mu.Unlock()

	// Copy to avoid external mutation
	copy := &OAuthTokens{
		AccessToken:  tokens.AccessToken,
		RefreshToken: tokens.RefreshToken,
		ExpiresAt:    tokens.ExpiresAt,
	}
	s.store[userID] = copy
	return nil
}

// Retrieve returns tokens for a user, or nil if none stored.
func (s *MemoryTokenStorage) Retrieve(userID string) (*OAuthTokens, error) {
	s.mu.RLock()
	defer s.mu.RUnlock()

	tokens, ok := s.store[userID]
	if !ok {
		return nil, nil
	}

	// Copy to avoid external mutation
	copy := &OAuthTokens{
		AccessToken:  tokens.AccessToken,
		RefreshToken: tokens.RefreshToken,
		ExpiresAt:    tokens.ExpiresAt,
	}
	return copy, nil
}

// Delete removes tokens for a user.
func (s *MemoryTokenStorage) Delete(userID string) error {
	s.mu.Lock()
	defer s.mu.Unlock()

	delete(s.store, userID)
	return nil
}

// Exists checks whether tokens exist for a user.
func (s *MemoryTokenStorage) Exists(userID string) (bool, error) {
	s.mu.RLock()
	defer s.mu.RUnlock()

	_, ok := s.store[userID]
	return ok, nil
}