|
package Models
|
|
|
|
import (
|
|
"database/sql/driver"
|
|
|
|
"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"
|
|
)
|
|
|
|
// Scan new value into MessageExpiry
|
|
func (e *MessageExpiry) Scan(value interface{}) error {
|
|
*e = MessageExpiry(value.(string))
|
|
return nil
|
|
}
|
|
|
|
// Value gets value out of MessageExpiry column
|
|
func (e MessageExpiry) Value() (driver.Value, error) {
|
|
return string(e), nil
|
|
}
|
|
|
|
// 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"`
|
|
MessageExpiryDefault MessageExpiry `gorm:"default:no_expiry" json:"message_expiry_default" sql:"type:ENUM('fifteen_min', 'thirty_min', 'one_hour', 'three_hour', 'six_hour', 'twelve_hour', 'one_day', 'three_day')"` // Stored encrypted
|
|
|
|
}
|