feat: now server part supports multiple channels for the same type (#50)

model/channel.go

@@ -0,0 +1,106 @@
+package model
+
+import (
+	"errors"
+)
+
+const (
+	TypeEmail             = "email"
+	TypeWeChatTestAccount = "test"
+	TypeWeChatCorpAccount = "corp_app"
+	TypeCorp              = "corp"
+	TypeLark              = "lark"
+	TypeDing              = "ding"
+	TypeTelegram          = "telegram"
+	TypeDiscord           = "discord"
+	TypeBark              = "bark"
+	TypeClient            = "client"
+	TypeNone              = "none"
+)
+
+type Channel struct {
+	Id        int    `json:"id"`
+	Type      string `json:"type" gorm:"type:varchar(32)"`
+	UserId    int    `json:"user_id" gorm:"uniqueIndex:name_user_id"`
+	Name      string `json:"name" gorm:"type:varchar(32);uniqueIndex:name_user_id"`
+	Status    int    `json:"status" gorm:"default:1"` // enabled, disabled
+	Secret    string `json:"secret"`
+	AppId     string `json:"app_id"`
+	AccountId string `json:"account_id"`
+	URL       string `json:"url" gorm:"column:url"`
+	Other     string `json:"other"`
+}
+
+func GetChannelById(id int, userId int) (*Channel, error) {
+	if id == 0 || userId == 0 {
+		return nil, errors.New("id 或 userId 为空！")
+	}
+	c := Channel{Id: id, UserId: userId}
+	err := DB.Where(c).First(&c).Error
+	return &c, err
+}
+
+func GetChannelByName(name string, userId int) (*Channel, error) {
+	if name == "" || userId == 0 {
+		return nil, errors.New("name 或 userId 为空！")
+	}
+	c := Channel{Name: name, UserId: userId}
+	err := DB.Where(c).First(&c).Error
+	return &c, err
+}
+
+func GetTokenStoreChannels() (channels []*Channel, err error) {
+	err = DB.Where("type = ? or type = ?", TypeWeChatCorpAccount, TypeWeChatTestAccount).Find(&channels).Error
+	return channels, err
+}
+
+func GetTokenStoreChannelsByUserId(userId int) (channels []*Channel, err error) {
+	err = DB.Where("user_id = ?", userId).Where("type = ? or type = ?", TypeWeChatCorpAccount, TypeWeChatTestAccount).Find(&channels).Error
+	return channels, err
+}
+
+func GetChannelsByUserId(userId int, startIdx int, num int) (channels []*Channel, err error) {
+	err = DB.Where("user_id = ?", userId).Order("id desc").Limit(num).Offset(startIdx).Find(&channels).Error
+	return channels, err
+}
+
+func SearchChannels(userId int, keyword string) (channels []*Channel, err error) {
+	err = DB.Where("user_id = ?", userId).Select([]string{"id", "name"}).Where("id = ? or name LIKE", keyword, keyword+"%").Find(&channels).Error
+	return channels, err
+}
+
+func DeleteChannelById(id int, userId int) (c *Channel, err error) {
+	// Why we need userId here? In case user want to delete other's c.
+	if id == 0 || userId == 0 {
+		return nil, errors.New("id 或 userId 为空！")
+	}
+	c = &Channel{Id: id, UserId: userId}
+	err = DB.Where(c).First(&c).Error
+	if err != nil {
+		return nil, err
+	}
+	return c, c.Delete()
+}
+
+func (channel *Channel) Insert() error {
+	var err error
+	err = DB.Create(channel).Error
+	return err
+}
+
+func (channel *Channel) UpdateStatus(status int) error {
+	err := DB.Model(channel).Update("status", status).Error
+	return err
+}
+
+// Update Make sure your token's fields is completed, because this will update non-zero values
+func (channel *Channel) Update() error {
+	var err error
+	err = DB.Model(channel).Select("type", "name", "secret", "app_id", "account_id", "url", "other").Updates(channel).Error
+	return err
+}
+
+func (channel *Channel) Delete() error {
+	err := DB.Delete(channel).Error
+	return err
+}