go-eve-pi/esi/cached.go

package esi

import (
	"context"
	"crypto/sha256"
	"encoding/hex"
	"encoding/json"
	"fmt"
	"os"
	"time"

	"go-eve-pi/options"
	"go-eve-pi/types"

	logger "git.site.quack-lab.dev/dave/cylogger"
)

// CacheEntry is now defined in the types package

// CachedESI implements ESIInterface with caching
type CachedESI struct {
	direct ESIInterface
	db     interface {
		GetCacheEntry(urlHash string) (*types.CacheEntry, error)
		SaveCacheEntry(entry *types.CacheEntry) error
	}
	cacheValidity time.Duration
}

// NewCachedESI creates a new CachedESI instance
func NewCachedESI(direct ESIInterface, db interface {
	GetCacheEntry(urlHash string) (*types.CacheEntry, error)
	SaveCacheEntry(entry *types.CacheEntry) error
}) *CachedESI {
	// Parse cache validity ONCE at initialization
	cacheValidity, err := time.ParseDuration(options.GlobalOptions.CacheValidity)
	if err != nil {
		logger.Error("Invalid cache validity duration %s: %v", options.GlobalOptions.CacheValidity, err)
		os.Exit(1)
	}

	logger.Info("Initializing CachedESI with cache validity: %v", cacheValidity)
	return &CachedESI{
		direct:        direct,
		db:            db,
		cacheValidity: cacheValidity,
	}
}

// GetCharacterPlanets retrieves a list of planets for a character with caching
func (c *CachedESI) GetCharacterPlanets(ctx context.Context, characterID int, accessToken string) ([]Planet, error) {
	url := fmt.Sprintf("/v1/characters/%d/planets/", characterID)
	logger.Info("CachedESI.GetCharacterPlanets: Starting for character %d", characterID)

	fetchFunc := func() ([]Planet, error) {
		logger.Info("CachedESI.GetCharacterPlanets: Calling direct ESI for character %d planets", characterID)
		result, err := c.direct.GetCharacterPlanets(ctx, characterID, accessToken)
		if err != nil {
			logger.Error("CachedESI.GetCharacterPlanets: Direct ESI call failed for character %d: %v", characterID, err)
			return nil, err
		}
		logger.Info("CachedESI.GetCharacterPlanets: Direct ESI returned %d planets for character %d", len(result), characterID)
		return result, nil
	}

	result, err := getCachedResponse(c, url, fetchFunc)
	if err != nil {
		logger.Error("CachedESI.GetCharacterPlanets: Failed to get planets for character %d: %v", characterID, err)
		return nil, err
	}

	logger.Info("CachedESI.GetCharacterPlanets: Successfully retrieved %d planets for character %d", len(result), characterID)
	return result, nil
}

// GetPlanetDetails retrieves detailed information about a specific planet with caching
func (c *CachedESI) GetPlanetDetails(ctx context.Context, characterID, planetID int, accessToken string) (*PlanetDetail, error) {
	url := fmt.Sprintf("/v3/characters/%d/planets/%d/", characterID, planetID)
	logger.Info("CachedESI.GetPlanetDetails: Starting for character %d, planet %d", characterID, planetID)

	fetchFunc := func() (*PlanetDetail, error) {
		logger.Info("CachedESI.GetPlanetDetails: Calling direct ESI for character %d planet %d details", characterID, planetID)
		result, err := c.direct.GetPlanetDetails(ctx, characterID, planetID, accessToken)
		if err != nil {
			logger.Error("CachedESI.GetPlanetDetails: Direct ESI call failed for character %d planet %d: %v", characterID, planetID, err)
			return nil, err
		}
		if result != nil {
			logger.Info("CachedESI.GetPlanetDetails: Direct ESI returned planet details with %d pins for character %d planet %d", len(result.Pins), characterID, planetID)
		} else {
			logger.Info("CachedESI.GetPlanetDetails: Direct ESI returned nil planet details for character %d planet %d", characterID, planetID)
		}
		return result, nil
	}

	result, err := getCachedResponse(c, url, fetchFunc)
	if err != nil {
		logger.Error("CachedESI.GetPlanetDetails: Failed to get planet details for character %d, planet %d: %v", characterID, planetID, err)
		return nil, err
	}

	if result != nil {
		logger.Info("CachedESI.GetPlanetDetails: Successfully retrieved planet details with %d pins for character %d, planet %d", len(result.Pins), characterID, planetID)
	} else {
		logger.Info("CachedESI.GetPlanetDetails: Successfully retrieved nil planet details for character %d, planet %d", characterID, planetID)
	}
	return result, nil
}

// getCachedResponse handles caching logic with generics
func getCachedResponse[T any](c *CachedESI, url string, fetchFunc func() (T, error)) (T, error) {
	// Generate cache key
	hash := sha256.Sum256([]byte(url))
	urlHash := hex.EncodeToString(hash[:])
	logger.Debug("CachedESI: Generated cache key %s for URL: %s", urlHash[:8], url)

	// Check cache using the interface
	logger.Debug("CachedESI: Checking cache for URL: %s", url)
	cacheEntry, err := c.db.GetCacheEntry(urlHash)
	if err == nil {
		// Check if cache is still valid
		age := time.Since(cacheEntry.CachedAt)
		if age < c.cacheValidity {
			logger.Info("Cache HIT for URL: %s (cached %v ago, valid for %v more)", url, age, c.cacheValidity-age)
			// Parse cached response
			var result T
			if err := json.Unmarshal([]byte(cacheEntry.Response), &result); err == nil {
				logger.Debug("CachedESI: Successfully unmarshaled cached response for URL: %s", url)
				return result, nil
			}
			logger.Warning("Failed to unmarshal cached response for URL: %s: %v", url, err)
		} else {
			logger.Info("Cache EXPIRED for URL: %s (cached %v ago, validity: %v)", url, age, c.cacheValidity)
		}
	} else {
		logger.Info("Cache MISS for URL: %s (no cache entry found): %v", url, err)
	}

	// Cache miss or invalid, fetch from API
	logger.Info("Fetching from API for URL: %s", url)
	result, err := fetchFunc()
	if err != nil {
		logger.Error("CachedESI: API fetch failed for URL: %s: %v", url, err)
		var zero T
		return zero, err
	}
	logger.Debug("CachedESI: API fetch successful for URL: %s", url)

	// Store in cache
	logger.Debug("CachedESI: Marshaling response for caching URL: %s", url)
	responseBytes, err := json.Marshal(result)
	if err != nil {
		logger.Warning("Failed to marshal response for caching URL: %s: %v", url, err)
		return result, nil
	}
	logger.Debug("CachedESI: Response marshaled successfully for URL: %s (%d bytes)", url, len(responseBytes))

	cacheEntry = &types.CacheEntry{
		URLHash:  urlHash,
		Response: string(responseBytes),
		CachedAt: time.Now(),
	}

	logger.Debug("CachedESI: Saving cache entry for URL: %s", url)
	if err := c.db.SaveCacheEntry(cacheEntry); err != nil {
		logger.Warning("Failed to cache response for URL: %s: %v", url, err)
	} else {
		logger.Info("Cached response for URL: %s (valid for %v, %d bytes)", url, c.cacheValidity, len(responseBytes))
	}

	return result, nil
}

// GetPlanetName retrieves planet name with caching
func (c *CachedESI) GetPlanetName(ctx context.Context, planetID int) (*PlanetName, error) {
	url := fmt.Sprintf("/v1/universe/planets/%d/", planetID)
	logger.Info("CachedESI.GetPlanetName: Starting for planet %d", planetID)

	fetchFunc := func() (*PlanetName, error) {
		logger.Info("CachedESI.GetPlanetName: Calling direct ESI for planet %d name", planetID)
		result, err := c.direct.GetPlanetName(ctx, planetID)
		if err != nil {
			logger.Error("CachedESI.GetPlanetName: Direct ESI call failed for planet %d: %v", planetID, err)
			return nil, err
		}
		if result != nil {
			logger.Info("CachedESI.GetPlanetName: Direct ESI returned planet name '%s' for planet %d", result.Name, planetID)
		} else {
			logger.Info("CachedESI.GetPlanetName: Direct ESI returned nil planet name for planet %d", planetID)
		}
		return result, nil
	}

	result, err := getCachedResponse(c, url, fetchFunc)
	if err != nil {
		logger.Error("CachedESI.GetPlanetName: Failed to get planet name for planet %d: %v", planetID, err)
		return nil, err
	}

	if result != nil {
		logger.Info("CachedESI.GetPlanetName: Successfully retrieved planet name '%s' for planet %d", result.Name, planetID)
	} else {
		logger.Info("CachedESI.GetPlanetName: Successfully retrieved nil planet name for planet %d", planetID)
	}
	return result, nil
}