/*
 * @FileName:   body.go
 * @Author:		JJXu
 * @CreateTime:	2022/3/2 上午1:23
 * @Description:
 */

package simpleRequest

type EntryMark string

func (b EntryMark) string() string {
	return string(b)
}

const (
	StringEntryType EntryMark = "__STRING_ENTRY__"
	BytesEntryType  EntryMark = "__BYTES_ENTRY__"
	ModelEntryType  EntryMark = "__MODEL_ENTRY__"
	MapEntryType    EntryMark = "__MAP_ENTRY__"
)

type BodyConf struct {
	simpleReq *SimpleRequest
}

func (s *BodyConf) Set(key string, value any) *BodyConf {
	s.simpleReq.BodyEntries[key] = value
	return s
}
func (s *BodyConf) Sets(data map[string]any) *BodyConf {
	s.simpleReq.BodyEntryMark = MapEntryType
	for k, v := range data {
		s.simpleReq.BodyEntries[k] = v
	}
	return s
}
func (s *BodyConf) SetString(strData string) *BodyConf {
	s.simpleReq.BodyEntryMark = StringEntryType
	s.simpleReq.BodyEntries[StringEntryType.string()] = strData
	return s
}

func (s *BodyConf) SetBytes(byteData []byte) *BodyConf {
	s.simpleReq.BodyEntryMark = BytesEntryType
	s.simpleReq.BodyEntries[BytesEntryType.string()] = byteData
	return s
}

func (s *BodyConf) SetModel(model any) *BodyConf {
	s.simpleReq.BodyEntryMark = ModelEntryType
	s.simpleReq.BodyEntries[ModelEntryType.string()] = model
	return s
}