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.
105 lines
3.1 KiB
105 lines
3.1 KiB
package Models
|
|
|
|
import (
|
|
"database/sql/driver"
|
|
"errors"
|
|
|
|
"github.com/gofrs/uuid"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// BeforeUpdate prevents updating the email if it has not changed
|
|
// This stops a unique constraint error
|
|
func (u *User) BeforeUpdate(tx *gorm.DB) (err error) {
|
|
if !tx.Statement.Changed("Username") {
|
|
tx.Statement.Omit("Username")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// MessageExpiry holds values for how long messages should expire by default
|
|
type MessageExpiry []uint8
|
|
|
|
const (
|
|
// MessageExpiryFifteenMin expires after 15 minutes
|
|
MessageExpiryFifteenMin = "fifteen_min"
|
|
// MessageExpiryThirtyMin expires after 30 minutes
|
|
MessageExpiryThirtyMin = "thirty_min"
|
|
// MessageExpiryOneHour expires after one hour
|
|
MessageExpiryOneHour = "one_hour"
|
|
// MessageExpiryThreeHour expires after three hours
|
|
MessageExpiryThreeHour = "three_hour"
|
|
// MessageExpirySixHour expires after six hours
|
|
MessageExpirySixHour = "six_hour"
|
|
// MessageExpiryTwelveHour expires after twelve hours
|
|
MessageExpiryTwelveHour = "twelve_hour"
|
|
// MessageExpiryOneDay expires after one day
|
|
MessageExpiryOneDay = "one_day"
|
|
// MessageExpiryThreeDay expires after three days
|
|
MessageExpiryThreeDay = "three_day"
|
|
// MessageExpiryNoExpiry never expires
|
|
MessageExpiryNoExpiry = "no_expiry"
|
|
)
|
|
|
|
// MessageExpiryValues list of all expiry values for validation
|
|
var MessageExpiryValues = []string{
|
|
MessageExpiryFifteenMin,
|
|
MessageExpiryThirtyMin,
|
|
MessageExpiryOneHour,
|
|
MessageExpiryThreeHour,
|
|
MessageExpirySixHour,
|
|
MessageExpiryTwelveHour,
|
|
MessageExpiryOneDay,
|
|
MessageExpiryThreeDay,
|
|
MessageExpiryNoExpiry,
|
|
}
|
|
|
|
// Scan new value into MessageExpiry
|
|
func (e *MessageExpiry) Scan(value interface{}) error {
|
|
var (
|
|
strValue = value.(string)
|
|
m string
|
|
)
|
|
|
|
for _, m = range MessageExpiryValues {
|
|
if strValue != m {
|
|
continue
|
|
}
|
|
*e = MessageExpiry(strValue)
|
|
return nil
|
|
}
|
|
|
|
return errors.New("Invalid MessageExpiry value")
|
|
}
|
|
|
|
// Value gets value out of MessageExpiry column
|
|
func (e MessageExpiry) Value() (driver.Value, error) {
|
|
return string(e), nil
|
|
}
|
|
|
|
func (e MessageExpiry) String() string {
|
|
return string(e)
|
|
}
|
|
|
|
// User holds user data
|
|
type User struct {
|
|
Base
|
|
Username string `gorm:"not null;unique" json:"username"`
|
|
Password string `gorm:"not null" json:"password"`
|
|
ConfirmPassword string `gorm:"-" json:"confirm_password"`
|
|
AsymmetricPrivateKey string `gorm:"not null" json:"asymmetric_private_key"` // Stored encrypted
|
|
AsymmetricPublicKey string `gorm:"not null" json:"asymmetric_public_key"`
|
|
SymmetricKey string `gorm:"not null" json:"symmetric_key"` // Stored encrypted
|
|
AttachmentID *uuid.UUID ` json:"attachment_id"`
|
|
Attachment Attachment ` json:"attachment"`
|
|
MessageExpiryDefault MessageExpiry `gorm:"default:no_expiry" json:"-" sql:"type:ENUM(
|
|
'fifteen_min',
|
|
'thirty_min',
|
|
'one_hour',
|
|
'three_hour',
|
|
'six_hour',
|
|
'twelve_hour',
|
|
'one_day',
|
|
'three_day'
|
|
)"` // Stored encrypted
|
|
}
|