package auth

import (
	"encoding/json"
	"fmt"
	"os"
	"path/filepath"

	"github.com/frenocorp/pop/internal/config"
)

type Session struct {
	UID         string `json:"uid"`
	AccessToken string `json:"access_token"`
	RefreshToken string `json:"refresh_token"`
	ExpiresAt   int64  `json:"expires_at"`
	TwoFAEnabled bool   `json:"two_factor_enabled"`
}

type SessionManager struct {
	configDir string
	sessionFile string
}

func NewSessionManager() *SessionManager {
	cfg := config.NewConfigManager()
	return &SessionManager{
		configDir:   cfg.ConfigDir(),
		sessionFile: filepath.Join(cfg.ConfigDir(), "session.json"),
	}
}

func (m *SessionManager) Login() error {
	// TODO: Implement interactive login with 2FA support
	// This will call the ProtonMail API and store the session
	session := Session{
		UID:         "placeholder-uid",
		AccessToken: "placeholder-token",
		RefreshToken: "placeholder-refresh",
		ExpiresAt:   0,
		TwoFAEnabled: false,
	}

	if err := os.MkdirAll(m.configDir, 0755); err != nil {
		return fmt.Errorf("failed to create config dir: %w", err)
	}

	data, err := json.MarshalIndent(session, "", "  ")
	if err != nil {
		return fmt.Errorf("failed to marshal session: %w", err)
	}

	if err := os.WriteFile(m.sessionFile, data, 0600); err != nil {
		return fmt.Errorf("failed to write session file: %w", err)
	}

	fmt.Println("Logged in successfully")
	return nil
}

func (m *SessionManager) Logout() error {
	if err := os.Remove(m.sessionFile); err != nil {
		return fmt.Errorf("failed to remove session file: %w", err)
	}

	fmt.Println("Logged out successfully")
	return nil
}

func (m *SessionManager) GetSession() (*Session, error) {
	data, err := os.ReadFile(m.sessionFile)
	if err != nil {
		return nil, fmt.Errorf("failed to read session file: %w", err)
	}

	var session Session
	if err := json.Unmarshal(data, &session); err != nil {
		return nil, fmt.Errorf("failed to parse session: %w", err)
	}

	return &session, nil
}

func (m *SessionManager) IsAuthenticated() (bool, error) {
	_, err := m.GetSession()
	if err != nil {
		return false, err
	}
	return true, nil
}