package funcs

import (
	"context"
	"fmt"

	"github.com/hairyhenderson/gomplate/v4/conv"
	"github.com/hairyhenderson/gomplate/v4/regexp"
)

// CreateReFuncs -
func CreateReFuncs(ctx context.Context) map[string]any {
	ns := &ReFuncs{ctx}
	return map[string]any{
		"regexp": func() any { return ns },
	}
}

// ReFuncs -
type ReFuncs struct {
	ctx context.Context
}

// Find -
func (ReFuncs) Find(re, input any) (string, error) {
	return regexp.Find(conv.ToString(re), conv.ToString(input))
}

// FindAll -
func (ReFuncs) FindAll(args ...any) ([]string, error) {
	re := ""
	n := 0
	input := ""

	switch len(args) {
	case 2:
		n = -1
		re = conv.ToString(args[0])
		input = conv.ToString(args[1])
	case 3:
		re = conv.ToString(args[0])

		var err error
		n, err = conv.ToInt(args[1])
		if err != nil {
			return nil, fmt.Errorf("n must be an integer: %w", err)
		}

		input = conv.ToString(args[2])
	default:
		return nil, fmt.Errorf("wrong number of args: want 2 or 3, got %d", len(args))
	}

	return regexp.FindAll(re, n, input)
}

// Match -
func (ReFuncs) Match(re, input any) (bool, error) {
	return regexp.Match(conv.ToString(re), conv.ToString(input))
}

// QuoteMeta -
func (ReFuncs) QuoteMeta(in any) string {
	return regexp.QuoteMeta(conv.ToString(in))
}

// Replace -
func (ReFuncs) Replace(re, replacement, input any) (string, error) {
	return regexp.Replace(conv.ToString(re),
		conv.ToString(replacement),
		conv.ToString(input))
}

// ReplaceLiteral -
func (ReFuncs) ReplaceLiteral(re, replacement, input any) (string, error) {
	return regexp.ReplaceLiteral(conv.ToString(re),
		conv.ToString(replacement),
		conv.ToString(input))
}

// Split -
func (ReFuncs) Split(args ...any) ([]string, error) {
	re := ""
	n := -1
	input := ""

	switch len(args) {
	case 2:
		re = conv.ToString(args[0])
		input = conv.ToString(args[1])
	case 3:
		re = conv.ToString(args[0])
		var err error
		n, err = conv.ToInt(args[1])
		if err != nil {
			return nil, fmt.Errorf("n must be an integer: %w", err)
		}

		input = conv.ToString(args[2])
	default:
		return nil, fmt.Errorf("wrong number of args: want 2 or 3, got %d", len(args))
	}

	return regexp.Split(re, n, input)
}