package handlers

import (
	"Portifolio/internal/database"
	"Portifolio/internal/model"
	"database/sql"
	"encoding/json"
	"fmt"
	"net/http"

	_ "github.com/mattn/go-sqlite3"
)

func AddTradeHandler(db *sql.DB) http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		var req model.AddTradeRequest
		if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
			http.Error(w, "invalid json", http.StatusBadRequest)
			return
		}

		err := req.Validate()
		if err != nil {
			http.Error(w, fmt.Sprintf("failed to validate trade: %s", err), http.StatusInternalServerError)
			return
		}

		// check if currency is in the db.
		currency, err := database.GetCurrencyByCode(db, req.CurrencyCode)
		if err != nil {
			http.Error(w, fmt.Sprintf("failed to find currency: %s", err), http.StatusInternalServerError)
			return
		}

		// check if company is in the db.
		company, err := database.GetCompanyByID(db, req.TickerId)
		if err != nil {
			http.Error(w, fmt.Sprintf("failed to find currency: %s", err), http.StatusInternalServerError)
			return
		}

		trade := model.Trade{
			Symbol:       company.Symbol,
			Shares:       req.Shares,
			Product:      model.TradeProduct(req.Product),
			Type:         model.TradeType(req.Type),
			Price:        req.Price,
			CurrencyCode: currency.Code,
			Date:         req.Date,
		}

		database.InsertTrade(db, trade)
	}
}

func GetTradeListHandler(db *sql.DB) http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		tradeList, err := database.GetTrades(db)
		if err != nil {
			http.Error(w, fmt.Sprintf("failed to fetch trades: %s", err), http.StatusInternalServerError)
			return
		}

		w.Header().Set("Content-Type", "application/json")
		if err := json.NewEncoder(w).Encode(tradeList); err != nil {
			http.Error(w, fmt.Sprintf("failed to encode trades: %s", err), http.StatusInternalServerError)
			return
		}
	}
}

func GetPositionListHandler(db *sql.DB) http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		tradeList, err := database.GetPositions(db)
		if err != nil {
			http.Error(w, "failed to fetch trades", http.StatusInternalServerError)
			return
		}

		w.Header().Set("Content-Type", "application/json")
		if err := json.NewEncoder(w).Encode(tradeList); err != nil {
			http.Error(w, "failed to encode trades", http.StatusInternalServerError)
			return
		}
	}
}