🎣 Open-Source Phishing Toolkit
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

86 lines
1.9 KiB

package models
import (
"errors"
log "github.com/gophish/gophish/logger"
)
// Webhook represents the webhook model
type Webhook struct {
Id int64 `json:"id" gorm:"column:id; primary_key:yes"`
Name string `json:"name"`
URL string `json:"url"`
Secret string `json:"secret"`
IsActive bool `json:"is_active"`
}
// ErrURLNotSpecified indicates there was no URL specified
var ErrURLNotSpecified = errors.New("URL can't be empty")
// ErrNameNotSpecified indicates there was no name specified
var ErrNameNotSpecified = errors.New("Name can't be empty")
// GetWebhooks returns the webhooks
func GetWebhooks() ([]Webhook, error) {
whs := []Webhook{}
err := db.Find(&whs).Error
return whs, err
}
// GetActiveWebhooks returns the active webhooks
func GetActiveWebhooks() ([]Webhook, error) {
whs := []Webhook{}
err := db.Where("is_active=?", true).Find(&whs).Error
return whs, err
}
// GetWebhook returns the webhook that the given id corresponds to.
// If no webhook is found, an error is returned.
func GetWebhook(id int64) (Webhook, error) {
wh := Webhook{}
err := db.Where("id=?", id).First(&wh).Error
return wh, err
}
// PostWebhook creates a new webhook in the database.
func PostWebhook(wh *Webhook) error {
err := wh.Validate()
if err != nil {
log.Error(err)
return err
}
err = db.Save(wh).Error
if err != nil {
log.Error(err)
}
return err
}
// PutWebhook edits an existing webhook in the database.
func PutWebhook(wh *Webhook) error {
err := wh.Validate()
if err != nil {
log.Error(err)
return err
}
err = db.Save(wh).Error
return err
}
// DeleteWebhook deletes an existing webhook in the database.
// An error is returned if a webhook with the given id isn't found.
func DeleteWebhook(id int64) error {
err := db.Where("id=?", id).Delete(&Webhook{}).Error
return err
}
func (wh *Webhook) Validate() error {
if wh.URL == "" {
return ErrURLNotSpecified
}
if wh.Name == "" {
return ErrNameNotSpecified
}
return nil
}