package langext

import (
	"bytes"
	"encoding/json"
	"fmt"
)

type H map[string]any

type A []any

func TryPrettyPrintJson(str string) string {
	var prettyJSON bytes.Buffer
	if err := json.Indent(&prettyJSON, []byte(str), "", "    "); err != nil {
		return str
	}
	return prettyJSON.String()
}

func PrettyPrintJson(str string) (string, bool) {
	var prettyJSON bytes.Buffer
	if err := json.Indent(&prettyJSON, []byte(str), "", "    "); err != nil {
		return str, false
	}
	return prettyJSON.String(), true
}

func PatchJson[JV string | []byte](rawjson JV, key string, value any) (JV, error) {
	var err error

	var jsonpayload map[string]any
	err = json.Unmarshal([]byte(rawjson), &jsonpayload)
	if err != nil {
		return *new(JV), fmt.Errorf("failed to unmarshal payload: %w", err)
	}

	jsonpayload[key] = value

	newjson, err := json.Marshal(jsonpayload)
	if err != nil {
		return *new(JV), fmt.Errorf("failed to re-marshal payload: %w", err)
	}

	return JV(newjson), nil
}

func PatchRemJson[JV string | []byte](rawjson JV, key string) (JV, error) {
	var err error

	var jsonpayload map[string]any
	err = json.Unmarshal([]byte(rawjson), &jsonpayload)
	if err != nil {
		return *new(JV), fmt.Errorf("failed to unmarshal payload: %w", err)
	}

	delete(jsonpayload, key)

	newjson, err := json.Marshal(jsonpayload)
	if err != nil {
		return *new(JV), fmt.Errorf("failed to re-marshal payload: %w", err)
	}

	return JV(newjson), nil
}

func MarshalJsonOrPanic(v any) string {
	bin, err := json.Marshal(v)
	if err != nil {
		panic(err)
	}
	return string(bin)
}

func MarshalJsonOrDefault(v any, def string) string {
	bin, err := json.Marshal(v)
	if err != nil {
		return def
	}
	return string(bin)
}

func MarshalJsonOrNil(v any) *string {
	bin, err := json.Marshal(v)
	if err != nil {
		return nil
	}
	return Ptr(string(bin))
}

func MarshalJsonIndentOrPanic(v any, prefix, indent string) string {
	bin, err := json.MarshalIndent(v, prefix, indent)
	if err != nil {
		panic(err)
	}
	return string(bin)
}

func MarshalJsonIndentOrDefault(v any, prefix, indent string, def string) string {
	bin, err := json.MarshalIndent(v, prefix, indent)
	if err != nil {
		return def
	}
	return string(bin)
}

func MarshalJsonIndentOrNil(v any, prefix, indent string) *string {
	bin, err := json.MarshalIndent(v, prefix, indent)
	if err != nil {
		return nil
	}
	return Ptr(string(bin))
}