package model_es

import (
	"bytes"
	"catface/app/global/variable"
	"catface/app/model"
	"context"
	"encoding/json"
	"fmt"

	"github.com/elastic/go-elasticsearch/v8/esapi"
)

func CreateAnimalESFactory(animal *model.Animal) *Animal {
	if animal == nil {
		return &Animal{}
	}
	return &Animal{
		Id:          animal.Id,
		Name:        animal.Name,
		NickNames:   animal.NickNamesList,
		Description: animal.Description,
	}
}

type Animal struct {
	Id          int64    `json:"id"`
	Name        string   `json:"name"`
	NickNames   []string `json:"nick_names"`
	Description string   `json:"description"`
}

func (a *Animal) IndexName() string {
	return "catface_animals"
}

func (a *Animal) InsertDocument() error {
	ctx := context.Background()

	data, err := json.Marshal(a)
	if err != nil {
		return err
	}

	req := esapi.IndexRequest{
		Index: a.IndexName(),
		// DocumentID: fmt.Sprintf("%d", a.Id),
		Body:    bytes.NewReader(data),
		Refresh: "true",
	}

	res, err := req.Do(ctx, variable.ElasticClient)
	if err != nil {
		return err
	}
	defer res.Body.Close()

	if res.IsError() {
		var e map[string]interface{}
		if err := json.NewDecoder(res.Body).Decode(&e); err != nil {
			return fmt.Errorf("error parsing the response body: %s", err)
		} else {
			return fmt.Errorf("[%s] %s: %s",
				res.Status(),
				e["error"].(map[string]interface{})["type"],
				e["error"].(map[string]interface{})["reason"],
			)
		}
	}

	return nil
}