godoxy-yusing/internal/notif/body.go

package notif

import (
	"bytes"
	"strings"

	"github.com/bytedance/sonic"
	gperr "github.com/yusing/goutils/errs"
)

type (
	LogField struct {
		Name  string `json:"name"`
		Value string `json:"value"`
	}
	LogFormat struct {
		string
	}
	LogBody interface {
		Format(format *LogFormat) ([]byte, error)
	}
)

type (
	FieldsBody  []LogField
	ListBody    []string
	MessageBody string
	errorBody   struct {
		Error error
	}
)

var (
	LogFormatMarkdown = &LogFormat{"markdown"}
	LogFormatPlain    = &LogFormat{"plain"}
	LogFormatRawJSON  = &LogFormat{"json"} // internal use only
)

func MakeLogFields(fields ...LogField) LogBody {
	return FieldsBody(fields)
}

func ErrorBody(err error) LogBody {
	return errorBody{Error: err}
}

func (f *LogFormat) Parse(format string) error {
	switch format {
	case "":
		f.string = LogFormatMarkdown.string
	case LogFormatPlain.string, LogFormatMarkdown.string:
		f.string = format
	default:
		return gperr.Multiline().
			Addf("invalid log format %s, supported formats:", format).
			AddLines(
				LogFormatPlain,
				LogFormatMarkdown,
			)
	}
	return nil
}

func (f *FieldsBody) Add(name, value string) {
	*f = append(*f, LogField{Name: name, Value: value})
}

func (f FieldsBody) Format(format *LogFormat) ([]byte, error) {
	switch format {
	case LogFormatMarkdown:
		var msg bytes.Buffer
		for _, field := range f {
			msg.WriteString("#### ")
			msg.WriteString(field.Name)
			msg.WriteRune('\n')
			msg.WriteString(field.Value)
			msg.WriteRune('\n')
		}
		return msg.Bytes(), nil
	case LogFormatPlain:
		var msg bytes.Buffer
		for _, field := range f {
			msg.WriteString(field.Name)
			msg.WriteString(": ")
			msg.WriteString(field.Value)
			msg.WriteRune('\n')
		}
		return msg.Bytes(), nil
	case LogFormatRawJSON:
		return sonic.Marshal(f)
	}
	return f.Format(LogFormatMarkdown)
}

func (l ListBody) Format(format *LogFormat) ([]byte, error) {
	switch format {
	case LogFormatPlain:
		return []byte(strings.Join(l, "\n")), nil
	case LogFormatMarkdown:
		var msg bytes.Buffer
		for _, item := range l {
			msg.WriteString("* ")
			msg.WriteString(item)
			msg.WriteRune('\n')
		}
		return msg.Bytes(), nil
	case LogFormatRawJSON:
		return sonic.Marshal(l)
	}
	return l.Format(LogFormatMarkdown)
}

func (m MessageBody) Format(format *LogFormat) ([]byte, error) {
	switch format {
	case LogFormatPlain, LogFormatMarkdown:
		return []byte(m), nil
	case LogFormatRawJSON:
		return sonic.Marshal(m)
	}
	return m.Format(LogFormatMarkdown)
}

func (e errorBody) Format(format *LogFormat) ([]byte, error) {
	switch format {
	case LogFormatRawJSON:
		return sonic.Marshal(e.Error)
	case LogFormatPlain:
		return gperr.Plain(e.Error), nil
	case LogFormatMarkdown:
		return gperr.Markdown(e.Error), nil
	}
	return gperr.Markdown(e.Error), nil
}