datadunia
107 lines
2.9 KiB
Go
107 lines
2.9 KiB
Go
package main
|
|
|
|
import (
|
|
"database/sql/driver"
|
|
"time"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type JSON []byte
|
|
|
|
// Global database handle shared across the application
|
|
var db *gorm.DB
|
|
|
|
func (j JSON) Value() (driver.Value, error) {
|
|
if j == nil {
|
|
return nil, nil
|
|
}
|
|
return string(j), nil
|
|
}
|
|
|
|
func (j *JSON) Scan(value interface{}) error {
|
|
if value == nil {
|
|
*j = nil
|
|
return nil
|
|
}
|
|
switch v := value.(type) {
|
|
case []byte:
|
|
*j = JSON(v)
|
|
case string:
|
|
*j = JSON(v)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (j JSON) MarshalJSON() ([]byte, error) {
|
|
if j == nil {
|
|
return []byte("null"), nil
|
|
}
|
|
return j, nil
|
|
}
|
|
|
|
func (j *JSON) UnmarshalJSON(data []byte) error {
|
|
*j = JSON(data)
|
|
return nil
|
|
}
|
|
|
|
type Server struct {
|
|
ID uint `gorm:"primaryKey"`
|
|
Name string `gorm:"uniqueIndex;not null"`
|
|
Mode string `gorm:"not null;default:forward"`
|
|
PublicKey string `gorm:"not null"`
|
|
Endpoint string
|
|
Webhooks []Webhook `gorm:"foreignKey:ServerID"`
|
|
Peers []Peer `gorm:"foreignKey:ServerID"`
|
|
CreatedAt time.Time
|
|
UpdatedAt time.Time
|
|
}
|
|
|
|
type Peer struct {
|
|
ID uint `gorm:"primaryKey"`
|
|
ServerID uint `gorm:"not null;index"`
|
|
PublicKey string `gorm:"uniqueIndex;not null"`
|
|
IP string `gorm:"not null"`
|
|
AllowAccess JSON `gorm:"type:text"`
|
|
AllowInternet bool `gorm:"default:false"`
|
|
CreatedAt time.Time
|
|
UpdatedAt time.Time
|
|
// ExpiresAt defines when this peer should be considered expired and eligible for auto-deletion
|
|
ExpiresAt time.Time
|
|
// DataLimitGB defines the monthly data limit per peer (in GB). 0 means unlimited.
|
|
DataLimitGB int64
|
|
// CurrentDataUsageBytes tracks the amount of data used by this peer (in bytes)
|
|
CurrentDataUsageBytes int64
|
|
// Enabled indicates whether the peer is active. Auto-restrict disables the peer if over the limit.
|
|
Enabled bool `gorm:"default:true"`
|
|
}
|
|
|
|
type Webhook struct {
|
|
ID uint `gorm:"primaryKey"`
|
|
ServerID uint `gorm:"index"`
|
|
Name string `gorm:"not null"`
|
|
URL string `gorm:"not null"`
|
|
Template string `gorm:"default:default"`
|
|
CustomBody string `gorm:"type:text"`
|
|
DefaultPayload string `gorm:"type:text"`
|
|
VerifySSL bool `gorm:"default:true"`
|
|
CustomHeaders JSON `gorm:"type:text"`
|
|
SubscribedActions JSON `gorm:"type:text"`
|
|
IsEnabled bool `gorm:"default:true"`
|
|
IsGlobal bool `gorm:"default:false"`
|
|
CreatedAt time.Time
|
|
UpdatedAt time.Time
|
|
}
|
|
|
|
type SMTPSettings struct {
|
|
ID uint `gorm:"primaryKey"`
|
|
Enabled bool `gorm:"default:false"`
|
|
Server string `gorm:"default:smtp.gmail.com"`
|
|
Port int `gorm:"default:587"`
|
|
UseTLS bool `gorm:"default:true"`
|
|
Username string
|
|
Password string
|
|
FromEmail string
|
|
FromName string
|
|
UseAuth bool `gorm:"default:true"`
|
|
}
|