package Database
|
|
|
|
import (
|
|
"github.com/gofrs/uuid"
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/clause"
|
|
)
|
|
|
|
// MessageData holds the content of the message
|
|
// encrypted through the Message.SymmetricKey
|
|
type MessageData struct {
|
|
Base
|
|
Data string ` json:"data"` // Stored encrypted
|
|
AttachmentID *uuid.UUID ` json:"attachment_id"`
|
|
Attachment Attachment ` json:"attachment"`
|
|
SenderID string `gorm:"not null" json:"sender_id"` // Stored encrypted
|
|
SymmetricKey string `gorm:"not null" json:"symmetric_key"` // Stored encrypted
|
|
}
|
|
|
|
func GetMessageDataById(id string) (MessageData, error) {
|
|
var (
|
|
messageData MessageData
|
|
err error
|
|
)
|
|
|
|
err = DB.Preload(clause.Associations).
|
|
First(&messageData, "id = ?", id).
|
|
Error
|
|
|
|
return messageData, err
|
|
}
|
|
|
|
func (messageData *MessageData) CreateMessageData() error {
|
|
var (
|
|
err error
|
|
)
|
|
|
|
err = DB.Session(&gorm.Session{FullSaveAssociations: true}).
|
|
Create(messageData).
|
|
Error
|
|
|
|
return err
|
|
}
|
|
|
|
func (messageData *MessageData) DeleteMessageData() error {
|
|
return DB.Session(&gorm.Session{FullSaveAssociations: true}).
|
|
Delete(messageData).
|
|
Error
|
|
}
|