package user

import (
	"fmt"
	"github.com/fossyy/filekeeper/utils"
	"sync"
	"time"

	"github.com/fossyy/filekeeper/db"
	"github.com/fossyy/filekeeper/logger"
	"github.com/google/uuid"
)

type Cache struct {
	users map[string]*UserWithExpired
	mu    sync.Mutex
}

type UserWithExpired struct {
	UserID   uuid.UUID
	Username string
	Email    string
	Password string
	AccessAt time.Time
}

var log *logger.AggregatedLogger
var UserCache *Cache

// TESTTING VAR
var database db.Database

func init() {
	log = logger.Logger()
	database = db.NewPostgresDB(utils.Getenv("DB_USERNAME"), utils.Getenv("DB_PASSWORD"), utils.Getenv("DB_HOST"), utils.Getenv("DB_PORT"), utils.Getenv("DB_NAME"))

	UserCache = &Cache{users: make(map[string]*UserWithExpired)}
	ticker := time.NewTicker(time.Hour * 8)

	go func() {
		for {
			<-ticker.C
			currentTime := time.Now()
			cacheClean := 0
			log.Info(fmt.Sprintf("Cache cleanup initiated at %02d:%02d:%02d", currentTime.Hour(), currentTime.Minute(), currentTime.Second()))

			UserCache.mu.Lock()
			for _, user := range UserCache.users {
				if currentTime.Sub(user.AccessAt) > time.Hour*8 {
					delete(UserCache.users, user.Email)
					cacheClean++
				}
			}
			UserCache.mu.Unlock()

			log.Info(fmt.Sprintf("Cache cleanup completed: %d entries removed. Finished at %s", cacheClean, time.Since(currentTime)))
		}
	}()
}

func Get(email string) (*UserWithExpired, error) {
	UserCache.mu.Lock()
	defer UserCache.mu.Unlock()

	if user, ok := UserCache.users[email]; ok {
		return user, nil
	}

	userData, err := database.GetUser(email)
	if err != nil {
		return nil, err
	}

	UserCache.users[email] = &UserWithExpired{
		UserID:   userData.UserID,
		Username: userData.Username,
		Email:    userData.Email,
		Password: userData.Password,
		AccessAt: time.Now(),
	}

	return UserCache.users[email], nil
}

func DeleteCache(email string) {
	UserCache.mu.Lock()
	defer UserCache.mu.Unlock()

	delete(UserCache.users, email)
}