package exceptions

import (
	"fmt"
	"github.com/tursom/GoCollections/lang"
	"io"
	"os"
	"strings"
)

type RuntimeException struct {
	lang.BaseObject
	message          string
	exceptionMessage string
	stackTrace       []StackTrace
	cause            Exception
}

func NewRuntimeException(message any, exceptionMessage string, config *ExceptionConfig) RuntimeException {
	if config == nil {
		config = DefaultExceptionConfig()
	}

	var stackTrace []StackTrace = nil
	if config.GetStackTrace {
		stackTrace = GetStackTraceSkipDeep(config.SkipStack + 1)
	}

	if len(exceptionMessage) == 0 {
		exceptionMessage = "exception caused:"
	}

	var causeException Exception = nil
	if config.Cause != nil {
		switch config.Cause.(type) {
		case Exception:
			causeException = config.Cause.(Exception)
		default:
			causeException = NewPackageException(config.Cause, "exception caused:", DefaultExceptionConfig().
				SetGetStackTrace(false))
		}
	}

	return RuntimeException{
		BaseObject:       lang.NewBaseObject(),
		message:          fmt.Sprint(message),
		exceptionMessage: exceptionMessage,
		stackTrace:       stackTrace,
		cause:            causeException,
	}
}

func (o RuntimeException) Cause() Exception {
	return o.cause
}

func (o RuntimeException) Error() string {
	builder := strings.Builder{}
	o.BuildPrintStackTrace(&builder)
	return builder.String()
}

func (o RuntimeException) ErrorMessage() string {
	return o.exceptionMessage
}

func (o RuntimeException) StackTrace() []StackTrace {
	return o.stackTrace
}

func (o RuntimeException) PrintStackTrace() {
	o.PrintStackTraceTo(os.Stderr)
}

func (o RuntimeException) PrintStackTraceTo(writer io.Writer) {
	bytes := []byte(o.Error())
	writeBytes := 0
	for writeBytes < len(bytes) {
		write, err := writer.Write(bytes[writeBytes:])
		if err != nil {
			Print(err)
			return
		}
		writeBytes += write
	}
}

func (o RuntimeException) BuildPrintStackTrace(builder *strings.Builder) {
	BuildStackTrace(builder, o, o.exceptionMessage)
	if o.cause != nil {
		builder.WriteString("caused by: ")
		o.cause.BuildPrintStackTrace(builder)
	}
}