package strings

import (
	"encoding/base64"

	"github.com/pkg/errors"

	"github.com/Luzifer/twitch-bot/v3/plugins"
)

func Register(args plugins.RegistrationArguments) error {
	args.RegisterTemplateFunction("b64urlenc", plugins.GenericTemplateFunctionGetter(base64URLEncode))
	args.RegisterTemplateFunction("b64urldec", plugins.GenericTemplateFunctionGetter(base64URLDecode))
	return nil
}

func base64URLEncode(v string) string {
	return base64.URLEncoding.EncodeToString([]byte(v))
}

func base64URLDecode(v string) (string, error) {
	data, err := base64.URLEncoding.DecodeString(v)
	if err != nil {
		return "", errors.Wrap(err, "decoding string")
	}
	return string(data), nil
}