mirror of
https://hubproxy.babadafafafafa.cn/https://github.com/yunionio/cloudpods.git
synced 2026-09-20 08:03:53 +08:00
853153c739
* refactor(notify): ALL * feat(climc): Notify v2 * feat: support notify for codegen.py * delete deprecated code and Maintain SDK compatibility * Adjust the template for mail validation to support captcha * fix: The issue of notify refactoring * add copyright for notifyv2 cli * Adjust according to the advice of make check * Remove utils package in notify * fix import format * remove redundant Template module * fix validateCreate and validateUpdate of notifytemplate * fix update issues of receivers * fix: Add SetStageComplete in tasks
107 lines
2.7 KiB
Go
107 lines
2.7 KiB
Go
// Copyright 2019 Yunion
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package models
|
|
|
|
import (
|
|
"yunion.io/x/pkg/tristate"
|
|
|
|
"yunion.io/x/onecloud/pkg/cloudcommon/db"
|
|
)
|
|
|
|
type SSubContactManager struct {
|
|
db.SStandaloneResourceBaseManager
|
|
}
|
|
|
|
// +onecloud:swagger-gen-ignore
|
|
type SSubContact struct {
|
|
db.SStandaloneResourceBase
|
|
|
|
// id of receiver user
|
|
ReceiverID string `width:"128" nullable:"false" index:"true"`
|
|
Type string `width:"16" nullable:"false" index:"true"`
|
|
Contact string `width:"128" nullable:"false"`
|
|
ParentContactType string `width:"16" nullable:"false"`
|
|
Enabled tristate.TriState `nullable:"false" default:"false"`
|
|
Verified tristate.TriState `nullable:"false" default:"false"`
|
|
}
|
|
|
|
var SubContactManager *SSubContactManager
|
|
|
|
var (
|
|
vTrue = true
|
|
vFalse = false
|
|
|
|
pTrue = &vTrue
|
|
pFalse = &vFalse
|
|
)
|
|
|
|
func init() {
|
|
SubContactManager = &SSubContactManager{
|
|
SStandaloneResourceBaseManager: db.NewStandaloneResourceBaseManager(
|
|
SSubContact{},
|
|
"subcontacts_tbl",
|
|
"subcontact",
|
|
"subcontacts",
|
|
),
|
|
}
|
|
SubContactManager.SetVirtualObject(SubContactManager)
|
|
}
|
|
|
|
func (scm *SSubContactManager) fetchMapByReceiverID(receiverID string) (map[string]*SSubContact, error) {
|
|
q := scm.Query().Equals("receiver_id", receiverID)
|
|
scontacts := make([]SSubContact, 0, 3)
|
|
err := db.FetchModelObjects(scm, q, &scontacts)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
ret := make(map[string]*SSubContact, len(scontacts))
|
|
for i := range scontacts {
|
|
ret[scontacts[i].Type] = &scontacts[i]
|
|
}
|
|
return ret, nil
|
|
}
|
|
|
|
func (sc *SSubContact) Enable() error {
|
|
return sc.Update(nil, pTrue, nil)
|
|
}
|
|
|
|
func (sc *SSubContact) Disable() error {
|
|
return sc.Update(nil, pFalse, nil)
|
|
}
|
|
|
|
func (sc *SSubContact) Verify() error {
|
|
return sc.Update(nil, nil, pTrue)
|
|
}
|
|
|
|
func (sc *SSubContact) Disverify() error {
|
|
return sc.Update(nil, nil, pFalse)
|
|
}
|
|
|
|
func (sc *SSubContact) Update(contact *string, enabled *bool, verified *bool) error {
|
|
_, err := db.Update(sc, func() error {
|
|
if contact != nil {
|
|
sc.Contact = *contact
|
|
}
|
|
if enabled != nil {
|
|
sc.Enabled = tristate.NewFromBool(*enabled)
|
|
}
|
|
if verified != nil {
|
|
sc.Verified = tristate.NewFromBool(*verified)
|
|
}
|
|
return nil
|
|
})
|
|
return err
|
|
}
|