add address book name &
add share address book
This commit is contained in:
ljw
@@ -22,6 +22,12 @@ func (s *AddressBookService) InfoByUserIdAndId(userid uint, id string) *model.Ad
|
||||
global.DB.Where("user_id = ? and id = ?", userid, id).First(p)
|
||||
return p
|
||||
}
|
||||
|
||||
func (s *AddressBookService) InfoByUserIdAndIdAndCid(userid uint, id string, cid uint) *model.AddressBook {
|
||||
p := &model.AddressBook{}
|
||||
global.DB.Where("user_id = ? and id = ? and collection_id = ?", userid, id, cid).First(p)
|
||||
return p
|
||||
}
|
||||
func (s *AddressBookService) InfoByRowId(id uint) *model.AddressBook {
|
||||
p := &model.AddressBook{}
|
||||
global.DB.Where("row_id = ?", id).First(p)
|
||||
@@ -96,7 +102,7 @@ func (s *AddressBookService) UpdateAddressBook(abs []*model.AddressBook, userId
|
||||
|
||||
}
|
||||
|
||||
func (t *AddressBookService) List(page, pageSize uint, where func(tx *gorm.DB)) (res *model.AddressBookList) {
|
||||
func (s *AddressBookService) List(page, pageSize uint, where func(tx *gorm.DB)) (res *model.AddressBookList) {
|
||||
res = &model.AddressBookList{}
|
||||
res.Page = int64(page)
|
||||
res.PageSize = int64(pageSize)
|
||||
@@ -111,34 +117,34 @@ func (t *AddressBookService) List(page, pageSize uint, where func(tx *gorm.DB))
|
||||
}
|
||||
|
||||
// Create 创建
|
||||
func (t *AddressBookService) Create(u *model.AddressBook) error {
|
||||
func (s *AddressBookService) Create(u *model.AddressBook) error {
|
||||
res := global.DB.Create(u).Error
|
||||
return res
|
||||
}
|
||||
func (t *AddressBookService) Delete(u *model.AddressBook) error {
|
||||
func (s *AddressBookService) Delete(u *model.AddressBook) error {
|
||||
return global.DB.Delete(u).Error
|
||||
}
|
||||
|
||||
// Update 更新
|
||||
func (t *AddressBookService) Update(u *model.AddressBook) error {
|
||||
return global.DB.Model(u).Updates(u).Error
|
||||
func (s *AddressBookService) Update(u *model.AddressBook) error {
|
||||
return global.DB.Model(u).Select("*").Omit("created_at").Updates(u).Error
|
||||
}
|
||||
|
||||
// ShareByWebClient 分享
|
||||
func (t *AddressBookService) ShareByWebClient(m *model.ShareRecord) error {
|
||||
func (s *AddressBookService) ShareByWebClient(m *model.ShareRecord) error {
|
||||
m.ShareToken = uuid.New().String()
|
||||
return global.DB.Create(m).Error
|
||||
}
|
||||
|
||||
// SharedPeer
|
||||
func (t *AddressBookService) SharedPeer(shareToken string) *model.ShareRecord {
|
||||
func (s *AddressBookService) SharedPeer(shareToken string) *model.ShareRecord {
|
||||
m := &model.ShareRecord{}
|
||||
global.DB.Where("share_token = ?", shareToken).First(m)
|
||||
return m
|
||||
}
|
||||
|
||||
// PlatformFromOs
|
||||
func (t *AddressBookService) PlatformFromOs(os string) string {
|
||||
func (s *AddressBookService) PlatformFromOs(os string) string {
|
||||
if strings.Contains(os, "Android") || strings.Contains(os, "android") {
|
||||
return "Android"
|
||||
}
|
||||
@@ -153,3 +159,152 @@ func (t *AddressBookService) PlatformFromOs(os string) string {
|
||||
}
|
||||
return ""
|
||||
}
|
||||
func (s *AddressBookService) ListByUserIdAndCollectionId(userId, cid, page, pageSize uint) (res *model.AddressBookList) {
|
||||
res = s.List(page, pageSize, func(tx *gorm.DB) {
|
||||
tx.Where("user_id = ? and collection_id = ?", userId, cid)
|
||||
})
|
||||
return
|
||||
}
|
||||
func (s *AddressBookService) ListCollection(page, pageSize uint, where func(tx *gorm.DB)) (res *model.AddressBookCollectionList) {
|
||||
res = &model.AddressBookCollectionList{}
|
||||
res.Page = int64(page)
|
||||
res.PageSize = int64(pageSize)
|
||||
tx := global.DB.Model(&model.AddressBookCollection{})
|
||||
if where != nil {
|
||||
where(tx)
|
||||
}
|
||||
tx.Count(&res.Total)
|
||||
tx.Scopes(Paginate(page, pageSize))
|
||||
tx.Find(&res.AddressBookCollection)
|
||||
return
|
||||
}
|
||||
func (s *AddressBookService) ListCollectionByIds(ids []uint) (res []*model.AddressBookCollection) {
|
||||
global.DB.Where("id in ?", ids).Find(&res)
|
||||
return res
|
||||
}
|
||||
|
||||
func (s *AddressBookService) ListCollectionByUserId(userId uint) (res *model.AddressBookCollectionList) {
|
||||
res = s.ListCollection(1, 100, func(tx *gorm.DB) {
|
||||
tx.Where("user_id = ?", userId)
|
||||
})
|
||||
return
|
||||
}
|
||||
func (s *AddressBookService) CollectionInfoById(id uint) *model.AddressBookCollection {
|
||||
p := &model.AddressBookCollection{}
|
||||
global.DB.Where("id = ?", id).First(p)
|
||||
return p
|
||||
}
|
||||
|
||||
func (s *AddressBookService) CollectionReadRules(user *model.User) (res []*model.AddressBookCollectionRule) {
|
||||
// personalRules
|
||||
var personalRules []*model.AddressBookCollectionRule
|
||||
tx2 := global.DB.Model(&model.AddressBookCollectionRule{})
|
||||
tx2.Where("type = ? and to_id = ? and rule > 0", model.ShareAddressBookRuleTypePersonal, user.Id).Find(&personalRules)
|
||||
res = append(res, personalRules...)
|
||||
|
||||
//group
|
||||
var groupRules []*model.AddressBookCollectionRule
|
||||
tx3 := global.DB.Model(&model.AddressBookCollectionRule{})
|
||||
tx3.Where("type = ? and to_id = ? and rule > 0", model.ShareAddressBookRuleTypeGroup, user.GroupId).Find(&groupRules)
|
||||
res = append(res, groupRules...)
|
||||
return
|
||||
}
|
||||
|
||||
func (s *AddressBookService) UserMaxRule(user *model.User, uid, cid uint) int {
|
||||
// ismy?
|
||||
if user.Id == uid {
|
||||
return model.ShareAddressBookRuleRuleFullControl
|
||||
}
|
||||
max := 0
|
||||
personalRules := &model.AddressBookCollectionRule{}
|
||||
tx := global.DB.Model(personalRules)
|
||||
tx.Where("type = ? and collection_id = ? and to_id = ?", model.ShareAddressBookRuleTypePersonal, cid, user.Id).First(&personalRules)
|
||||
if personalRules.Id != 0 {
|
||||
max = personalRules.Rule
|
||||
if max == model.ShareAddressBookRuleRuleFullControl {
|
||||
return max
|
||||
}
|
||||
}
|
||||
|
||||
groupRules := &model.AddressBookCollectionRule{}
|
||||
tx2 := global.DB.Model(groupRules)
|
||||
tx2.Where("type = ? and collection_id = ? and to_id = ?", model.ShareAddressBookRuleTypeGroup, cid, user.GroupId).First(&groupRules)
|
||||
if groupRules.Id != 0 {
|
||||
if groupRules.Rule > max {
|
||||
max = groupRules.Rule
|
||||
}
|
||||
if max == model.ShareAddressBookRuleRuleFullControl {
|
||||
return max
|
||||
}
|
||||
}
|
||||
return max
|
||||
}
|
||||
|
||||
func (s *AddressBookService) CheckUserReadPrivilege(user *model.User, uid, cid uint) bool {
|
||||
return s.UserMaxRule(user, uid, cid) >= model.ShareAddressBookRuleRuleRead
|
||||
}
|
||||
func (s *AddressBookService) CheckUserWritePrivilege(user *model.User, uid, cid uint) bool {
|
||||
return s.UserMaxRule(user, uid, cid) >= model.ShareAddressBookRuleRuleReadWrite
|
||||
}
|
||||
func (s *AddressBookService) CheckUserFullControlPrivilege(user *model.User, uid, cid uint) bool {
|
||||
return s.UserMaxRule(user, uid, cid) >= model.ShareAddressBookRuleRuleFullControl
|
||||
}
|
||||
|
||||
func (s *AddressBookService) CreateCollection(t *model.AddressBookCollection) error {
|
||||
return global.DB.Create(t).Error
|
||||
}
|
||||
|
||||
func (s *AddressBookService) UpdateCollection(t *model.AddressBookCollection) error {
|
||||
return global.DB.Model(t).Updates(t).Error
|
||||
}
|
||||
|
||||
func (s *AddressBookService) DeleteCollection(t *model.AddressBookCollection) error {
|
||||
//删除集合下的所有规则、地址簿，再删除集合
|
||||
tx := global.DB.Begin()
|
||||
tx.Where("collection_id = ?", t.Id).Delete(&model.AddressBookCollectionRule{})
|
||||
tx.Where("collection_id = ?", t.Id).Delete(&model.AddressBook{})
|
||||
tx.Delete(t)
|
||||
return tx.Commit().Error
|
||||
}
|
||||
|
||||
func (s *AddressBookService) RuleInfoById(u uint) *model.AddressBookCollectionRule {
|
||||
p := &model.AddressBookCollectionRule{}
|
||||
global.DB.Where("id = ?", u).First(p)
|
||||
return p
|
||||
}
|
||||
func (s *AddressBookService) RulePersonalInfoByToIdAndCid(toid, cid uint) *model.AddressBookCollectionRule {
|
||||
p := &model.AddressBookCollectionRule{}
|
||||
global.DB.Where("type = ? and to_id = ? and collection_id = ?", model.ShareAddressBookRuleTypePersonal, toid, cid).First(p)
|
||||
return p
|
||||
}
|
||||
func (s *AddressBookService) CreateRule(t *model.AddressBookCollectionRule) error {
|
||||
return global.DB.Create(t).Error
|
||||
}
|
||||
|
||||
func (s *AddressBookService) ListRules(page uint, size uint, f func(tx *gorm.DB)) *model.AddressBookCollectionRuleList {
|
||||
res := &model.AddressBookCollectionRuleList{}
|
||||
res.Page = int64(page)
|
||||
res.PageSize = int64(size)
|
||||
tx := global.DB.Model(&model.AddressBookCollectionRule{})
|
||||
if f != nil {
|
||||
f(tx)
|
||||
}
|
||||
tx.Count(&res.Total)
|
||||
tx.Scopes(Paginate(page, size))
|
||||
tx.Find(&res.AddressBookCollectionRule)
|
||||
return res
|
||||
}
|
||||
|
||||
func (s *AddressBookService) UpdateRule(t *model.AddressBookCollectionRule) error {
|
||||
return global.DB.Model(t).Updates(t).Error
|
||||
}
|
||||
|
||||
func (s *AddressBookService) DeleteRule(t *model.AddressBookCollectionRule) error {
|
||||
return global.DB.Delete(t).Error
|
||||
}
|
||||
|
||||
// CheckCollectionOwner 检查Collection的所有者
|
||||
func (s *AddressBookService) CheckCollectionOwner(uid uint, cid uint) bool {
|
||||
p := s.CollectionInfoById(cid)
|
||||
return p.UserId == uid
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user