caskin/server_object_data.go at main · AWaterColorPen/caskin · 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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
package caskin

// CreateObjectData
// if there does not exist the item then create a new one
// 1. current user has item's write permission
// 2. create a new item into database
func (s *server) CreateObjectData(user User, domain Domain, item ObjectData, ty ObjectType) error {
	if err := s.CheckCreateObjectData(user, domain, item, ty); err != nil {
		return err
	}
	item.SetDomainID(domain.GetID())
	return s.DB.Create(item)
}

// RecoverObjectData
// if there exist the item but soft deleted then recover it
// 1. current user has item's write permission
// 2. recover the soft delete one item at database
func (s *server) RecoverObjectData(user User, domain Domain, item ObjectData) error {
	if err := s.CheckRecoverObjectData(user, domain, item); err != nil {
		return err
	}
	return s.DB.Recover(item)
}

// DeleteObjectData
// if there exist the item
// 1. current user has item's write permission
// 3. soft delete one item in database
func (s *server) DeleteObjectData(user User, domain Domain, item ObjectData) error {
	if err := s.CheckDeleteObjectData(user, domain, item); err != nil {
		return err
	}
	item.SetDomainID(domain.GetID())
	return s.DB.DeleteByID(item, item.GetID())
}

// UpdateObjectData
// if there exist the item
// 1. current user has item's write permission and
// 2. update item's properties
func (s *server) UpdateObjectData(user User, domain Domain, item ObjectData, ty ObjectType) error {
	if err := s.CheckUpdateObjectData(user, domain, item, ty); err != nil {
		return err
	}
	item.SetDomainID(domain.GetID())
	return s.DB.Update(item)
}

// GetObjectData
// get items
// 1. current user has item's read permission
// func (s *server) GetObjectData(user User, domain Domain, item ObjectData) ([]ObjectData, error) {
//
// 	GetInDomain[item](s.DB, domain)
// 	roles, err := s.DB.GetRoleInDomain(domain)
// 	if err != nil {
// 		return nil, err
// 	}
// 	return Filter(s.Enforcer, user, domain, Read, roles), nil
// }

func (s *server) CheckWriteObjectData(user User, domain Domain, item ObjectData, ty ObjectType) error {
	if err := s.CheckObjectData(user, domain, item, Write); err != nil {
		return err
	}
	o := DefaultFactory().NewObject()
	o.SetID(item.GetObjectID())
	if err := s.DB.Take(o); err != nil {
		return ErrInValidObject
	}
	if o.GetObjectType() != ty {
		return ErrInValidObjectType
	}
	return nil
}

func (s *server) CheckCreateObjectData(user User, domain Domain, item ObjectData, ty ObjectType) error {
	if err := s.DBCreateCheck(item); err != nil {
		return err
	}
	return s.CheckWriteObjectData(user, domain, item, ty)
}

func (s *server) CheckRecoverObjectData(user User, domain Domain, item ObjectData) error {
	if err := s.DBRecoverCheck(item); err != nil {
		return err
	}
	return s.CheckObjectData(user, domain, item, Write)
}

func (s *server) CheckDeleteObjectData(user User, domain Domain, item ObjectData) error {
	if err := s.IDInterfaceDeleteCheck(item); err != nil {
		return err
	}
	return s.CheckObjectData(user, domain, item, Write)
}

func (s *server) CheckUpdateObjectData(user User, domain Domain, item ObjectData, ty ObjectType) error {
	old := newByE(item)
	if err := s.IDInterfaceUpdateCheck(item, old); err != nil {
		return err
	}
	if err := s.CheckWriteObjectData(user, domain, old, ty); err != nil {
		return err
	}
	if item.GetObjectID() != old.GetObjectID() {
		return s.CheckWriteObjectData(user, domain, item, ty)
	}
	return nil
}

func (s *server) CheckModifyObjectData(user User, domain Domain, item ObjectData) error {
	if err := s.IDInterfaceModifyCheck(item); err != nil {
		return err
	}
	return s.CheckObjectData(user, domain, item, Write)
}

func (s *server) CheckGetObjectData(user User, domain Domain, item ObjectData) error {
	if err := s.IDInterfaceGetCheck(item); err != nil {
		return err
	}
	return s.CheckObjectData(user, domain, item, Read)
}