mago/viewobject.go at master · maliboot/mago · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
package mago

import "github.com/maliboot/mago/helper"

type ViewObject interface {
}

type AbstractViewObject struct {
	ViewObject `json:"-"`
}

type ViewObjectTemplate interface {
	Zero
	Error() error
	Body() interface{}
}

type VO[T any] struct {
	Data   *T
	Err    error
	IsZero bool
}

func NewVO[T any](err error, data interface{}) *VO[T] {
	return NewConvertorVO[T](err, data, 0)
}

func NewLowerCamelVO[T any](err error, data interface{}) *VO[T] {
	return NewConvertorVO[T](err, data, 1)
}

func NewSnakeVO[T any](err error, data interface{}) *VO[T] {
	return NewConvertorVO[T](err, data, 2)
}

func NewZeroVO[T any]() *VO[T] {
	vo := &VO[T]{
		IsZero: true,
	}

	return vo
}

func NewConvertorVO[T any](err error, data interface{}, convertType int) *VO[T] {
	vo := &VO[T]{
		Err: err,
	}
	if data == nil {
		vo.IsZero = true
		return vo
	}

	myData, ok := data.(*T)
	if ok {
		vo.Data = myData
	} else {
		switch convertType {
		case 1:
			vo.Data, _ = helper.LowerCamelConvertor[T](data)
			break
		case 2:
			vo.Data, _ = helper.SnakeConvertor[T](data)
			break
		default:
			vo.Data, _ = helper.Convertor[T](data)
		}
	}

	return vo
}

func (vo *VO[T]) Error() error {
	return vo.Err
}

func (vo *VO[T]) Zero() bool {
	return vo.IsZero
}

func (vo *VO[T]) Body() interface{} {
	return vo.Data
}