package api

import (
	"bytes"
	"encoding/json"
	"fmt"
	"io"
	"net/http"
	"time"
)

const BaseURL = "https://api.servicetrade.com/api"

type Session struct {
	Client       *http.Client
	Cookie       string
	LastAccessed time.Time
}

func NewSession() *Session {
	return &Session{
		Client:       &http.Client{},
		LastAccessed: time.Now(),
	}
}

func (s *Session) ValidateSession() error {
	url := fmt.Sprintf("%s/auth/validate", BaseURL)
	req, err := http.NewRequest("GET", url, nil)
	if err != nil {
		return err
	}
	req.Header.Set("Cookie", s.Cookie)

	resp, err := s.Client.Do(req)
	if err != nil {
		return err
	}
	defer resp.Body.Close()

	if resp.StatusCode != http.StatusOK {
		return fmt.Errorf("session validation failed")
	}

	s.LastAccessed = time.Now()
	return nil
}

func (s *Session) DoRequest(method, endpoint string, body io.Reader) (*http.Response, error) {
	url := fmt.Sprintf("%s%s", BaseURL, endpoint)
	fmt.Println(url)
	req, err := http.NewRequest(method, url, body)
	if err != nil {
		return nil, fmt.Errorf("error creating request: %v", err)
	}
	req.Header.Set("Cookie", s.Cookie)
	req.Header.Set("Content-Type", "application/json")

	return s.Client.Do(req)
}

func (s *Session) Login(email, password string) error {
	payload := map[string]string{
		"username": email,
		"password": password,
	}
	payloadBytes, _ := json.Marshal(payload)

	resp, err := s.DoRequest("POST", "/auth", bytes.NewBuffer(payloadBytes))
	if err != nil {
		return err
	}
	defer resp.Body.Close()

	if resp.StatusCode != http.StatusOK {
		body, _ := io.ReadAll(resp.Body)
		return fmt.Errorf("failed to authenticate: %s, response: %s", resp.Status, string(body))
	}

	for _, cookie := range resp.Cookies() {
		if cookie.Name == "PHPSESSID" {
			s.Cookie = cookie.String()
			return nil
		}
	}

	return fmt.Errorf("failed to retrieve session cookie")
}

func (s *Session) Logout() error {
	resp, err := s.DoRequest("DELETE", "/auth", nil)
	if err != nil {
		return err
	}
	defer resp.Body.Close()

	if resp.StatusCode != http.StatusOK && resp.StatusCode != http.StatusNoContent {
		body, _ := io.ReadAll(resp.Body)
		return fmt.Errorf("failed to end session: %s, response: %s", resp.Status, string(body))
	}

	s.Cookie = ""
	return nil
}