Envelope/Backend/Models/Users.go

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
}