package client

import (
	"dborg/internal/models"
	"encoding/json"
	"fmt"
	"net/url"
)

func (c *Client) SearchStealerLogs(params *models.SLParams) (*models.SLResponse, error) {
	queryParams := url.Values{}
	queryParams.Add("query", params.Query)

	if params.MaxHits > 0 && params.MaxHits != 10 {
		queryParams.Add("max_hits", fmt.Sprintf("%d", params.MaxHits))
	}
	if params.SortBy != "" {
		queryParams.Add("sort_by", params.SortBy)
	}
	if params.IngestStartDate != "" {
		queryParams.Add("ingest_start_date", params.IngestStartDate)
	}
	if params.IngestEndDate != "" {
		queryParams.Add("ingest_end_date", params.IngestEndDate)
	}
	if params.PostedStartDate != "" {
		queryParams.Add("posted_start_date", params.PostedStartDate)
	}
	if params.PostedEndDate != "" {
		queryParams.Add("posted_end_date", params.PostedEndDate)
	}
	if params.Format != "" && params.Format != "json" {
		queryParams.Add("format", params.Format)
	}

	data, err := c.Get("/sl/search", queryParams)
	if err != nil {
		return nil, err
	}

	if params.Format != "" && params.Format != "json" {
		return &models.SLResponse{
			Message: string(data),
		}, nil
	}

	var response models.SLResponse
	if err := json.Unmarshal(data, &response); err != nil {
		return nil, fmt.Errorf("failed to parse stealer logs response: %w", err)
	}

	return &response, nil
}