package usecasehelper

import (
	"errors"
	"fmt"
	"strings"
	"time"

	pl "simrs-vx/pkg/logger"

	"gorm.io/gorm"
)

func SafeToResponse[T any](data *T) any {
	if data == nil {
		return nil
	}

	// Use type assertion to call ToResponse if the type has it
	if converter, ok := any(data).(interface{ ToResponse() any }); ok {
		return converter.ToResponse()
	}

	if converter, ok := any(*data).(interface{ ToResponse() any }); ok {
		return converter.ToResponse()
	}

	return nil
}

func HandleReadError(err error, event *pl.Event, itemType string, id interface{}, data any) error {
	if err == nil {
		pl.SetLogInfo(event, data, "complete")
		return nil
	}

	if errors.Is(err, gorm.ErrRecordNotFound) {
		event.Status = "failed"
		event.ErrInfo = pl.ErrorInfo{
			Code:   "data-notFound",
			Detail: fmt.Sprintf("%s with ID %v not found", itemType, id),
			Raw:    err,
		}
	} else {
		event.Status = "failed"
		event.ErrInfo = pl.ErrorInfo{
			Code:   "data-read-detail-fail",
			Detail: fmt.Sprintf("%s read failed", itemType),
			Raw:    err,
		}
	}

	return pl.SetLogError(event, nil)
}

func HandleSearchError(err error, event *pl.Event, itemType string, query interface{}, data any) error {
	if err == nil {
		pl.SetLogInfo(event, data, "complete")
		return nil
	}

	if errors.Is(err, gorm.ErrRecordNotFound) {
		event.Status = "failed"
		event.ErrInfo = pl.ErrorInfo{
			Code:   "data-notFound",
			Detail: fmt.Sprintf("%s with query '%v' not found", itemType, query),
			Raw:    err,
		}
	} else {
		event.Status = "failed"
		event.ErrInfo = pl.ErrorInfo{
			Code:   "data-search-fail",
			Detail: fmt.Sprintf("%s search failed", itemType),
			Raw:    err,
		}
	}

	return pl.SetLogError(event, nil)
}

func GetMiddlewareErrorCode(mwType MWType) string {
	if strings.Contains(string(mwType), "Pre") {
		return "MW_PRE_FAILED"
	}
	return "MW_POST_FAILED"
}

// GetLogData returns whichever of data or input is non-nil (prefers data)
func GetLogData(input interface{}, data interface{}) interface{} {
	if data != nil {
		return data
	}
	return input
}

func HandleMiddlewareError(event *pl.Event, mwType, mwName string, logData interface{}, err error) error {
	event.Status = "failed"
	event.ErrInfo = pl.ErrorInfo{
		Code:   GetMiddlewareErrorCode(MWType(mwType)),
		Detail: fmt.Sprintf("%s middleware %s failed", mwType, mwName),
		Raw:    err,
	}
	return pl.SetLogError(event, logData)
}

func IsDataNotFoundError(err error) bool {
	if err == nil {
		return false
	}
	return strings.Contains(err.Error(), "code: data-notFound")
}

func AddPrefix(prefix string, str string) string {
	return prefix + str
}

func GetPreloads(input string) []string {
	result := []string{}
	parts := strings.Split(input, ",")
	for _, p := range parts {
		subParts := strings.Split(p, "-")
		for i := range subParts {
			subParts[i] = kebabToPascal(subParts[i])
		}
		result = append(result, strings.Join(subParts, "."))
	}
	return result
}

func kebabToPascal(input string) string {
	parts := strings.Split(input, "-")
	for i, p := range parts {
		if len(p) > 0 {
			r := []rune(p)
			r[0] = []rune(strings.ToUpper(string(r[0])))[0]
			parts[i] = string(r)
		}
	}
	return strings.Join(parts, "")
}

func GetTimeNow() *time.Time {
	tmp := time.Now()
	return &tmp
}