mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-29 22:06:38 +08:00
ryan
57 lines
1.9 KiB
Go
57 lines
1.9 KiB
Go
package model
|
|
|
|
import "time"
|
|
|
|
type ProxyRoute struct {
|
|
ID uint `json:"id" gorm:"primaryKey"`
|
|
Domain string `json:"domain" gorm:"uniqueIndex;size:255;not null"`
|
|
OriginURL string `json:"origin_url" gorm:"size:2048;not null"`
|
|
OriginHost string `json:"origin_host" gorm:"size:255"`
|
|
Enabled bool `json:"enabled" gorm:"not null;default:true"`
|
|
EnableHTTPS bool `json:"enable_https" gorm:"column:enable_https;not null;default:false"`
|
|
CertID *uint `json:"cert_id"`
|
|
RedirectHTTP bool `json:"redirect_http" gorm:"not null;default:false"`
|
|
CustomHeaders string `json:"custom_headers" gorm:"type:text;not null;default:'[]'"`
|
|
Remark string `json:"remark" gorm:"size:255"`
|
|
CreatedAt time.Time `json:"created_at"`
|
|
UpdatedAt time.Time `json:"updated_at"`
|
|
}
|
|
|
|
func ListProxyRoutes() (routes []*ProxyRoute, err error) {
|
|
err = DB.Order("id desc").Find(&routes).Error
|
|
return routes, err
|
|
}
|
|
|
|
func GetEnabledProxyRoutes() (routes []*ProxyRoute, err error) {
|
|
err = DB.Where("enabled = ?", true).Order("domain asc").Find(&routes).Error
|
|
return routes, err
|
|
}
|
|
|
|
func GetProxyRouteByID(id uint) (*ProxyRoute, error) {
|
|
route := &ProxyRoute{}
|
|
err := DB.First(route, id).Error
|
|
return route, err
|
|
}
|
|
|
|
func (route *ProxyRoute) Insert() error {
|
|
return DB.Create(route).Error
|
|
}
|
|
|
|
func (route *ProxyRoute) Update() error {
|
|
return DB.Model(&ProxyRoute{}).Where("id = ?", route.ID).Updates(map[string]any{
|
|
"domain": route.Domain,
|
|
"origin_url": route.OriginURL,
|
|
"origin_host": route.OriginHost,
|
|
"enabled": route.Enabled,
|
|
"enable_https": route.EnableHTTPS,
|
|
"cert_id": route.CertID,
|
|
"redirect_http": route.RedirectHTTP,
|
|
"custom_headers": route.CustomHeaders,
|
|
"remark": route.Remark,
|
|
}).Error
|
|
}
|
|
|
|
func (route *ProxyRoute) Delete() error {
|
|
return DB.Delete(route).Error
|
|
}
|