package elastic

import (
	"context"
	"crypto/tls"
	"cybertron/configs"
	"encoding/json"
	elasticsearch8 "github.com/elastic/go-elasticsearch/v8"
	"github.com/elastic/go-elasticsearch/v8/typedapi/core/search"
	"github.com/elastic/go-elasticsearch/v8/typedapi/types"
	"log"
	"net/http"
)

type ElasticSearchClient struct {
	client *elasticsearch8.TypedClient
	Config configs.ElasticConfig
}

func NewElasticClient(elasticConfig configs.ElasticConfig) (*ElasticSearchClient, error) {
	cfg := elasticsearch8.Config{
		Addresses: elasticConfig.Addresses,
		APIKey:    elasticConfig.APIKey,
		Transport: &http.Transport{
			TLSClientConfig: &tls.Config{
				InsecureSkipVerify: true,
			},
		},
	}

	client, err := elasticsearch8.NewTypedClient(cfg)
	return &ElasticSearchClient{
		client: client,
		Config: elasticConfig,
	}, err
}

func (el *ElasticSearchClient) IndexDocument(document interface{}) {
	_, err := el.client.Index(el.Config.Index).
		Request(document).
		Do(context.TODO())

	if err != nil {
		log.Fatal(err)
	}
}

func (el *ElasticSearchClient) SearchDocuments(query *types.Query, fields []string) ([]map[string]interface{}, error) {
	res, err := el.client.Search().
		Index(el.Config.Index).
		Request(&search.Request{
			Query: query,
		}).Source_(fields).
		Do(context.TODO())
	if err != nil {
		log.Println("Error getting response: %s", err)
	}
	var results []map[string]interface{}
	for _, hit := range res.Hits.Hits {
		var doc map[string]interface{}

		err := json.Unmarshal(hit.Source_, &doc)
		doc["id"] = *hit.Id_
		if err != nil {
			log.Printf("Error unmarshalling document: %s", err)
			continue
		}
		results = append(results, doc)
	}

	return results, nil
}

func (el *ElasticSearchClient) GetDocument(documentID string) (interface{}, error) {
	// Retrieve the document by its ID
	res, err := el.client.Get(el.Config.Index, documentID).Do(context.Background())
	var document interface{}
	if err != nil {
		log.Printf("Error getting response: %s", err)
		return nil, err
	}

	// Check if the document exists
	if !res.Found {
		log.Printf("Document with ID %s not found", documentID)
		return document, nil
	}

	// Unmarshal the JSON response into the provided document interface
	err = json.Unmarshal(res.Source_, &document)
	if err != nil {
		log.Printf("Error unmarshalling document: %s", err)
		return nil, err
	}

	return document, nil
}