SimpleCloudNotifier/server/models/message.go

package models

import (
	"database/sql"
	"github.com/blockloop/scan"
	"gogs.mikescher.com/BlackForestBytes/goext/langext"
	"time"
)

const (
	ContentLengthTrim  = 1900
	ContentLengthShort = 200
)

type Message struct {
	SCNMessageID    SCNMessageID
	SenderUserID    UserID
	OwnerUserID     UserID
	ChannelName     string
	ChannelID       ChannelID
	SenderName      *string
	SenderIP        string
	TimestampReal   time.Time
	TimestampClient *time.Time
	Title           string
	Content         *string
	Priority        int
	UserMessageID   *string
}

func (m Message) FullJSON() MessageJSON {
	return MessageJSON{
		SCNMessageID:  m.SCNMessageID,
		SenderUserID:  m.SenderUserID,
		OwnerUserID:   m.OwnerUserID,
		ChannelName:   m.ChannelName,
		ChannelID:     m.ChannelID,
		SenderName:    m.SenderName,
		SenderIP:      m.SenderIP,
		Timestamp:     m.Timestamp().Format(time.RFC3339Nano),
		Title:         m.Title,
		Content:       m.Content,
		Priority:      m.Priority,
		UserMessageID: m.UserMessageID,
		Trimmed:       false,
	}
}

func (m Message) TrimmedJSON() MessageJSON {
	return MessageJSON{
		SCNMessageID:  m.SCNMessageID,
		SenderUserID:  m.SenderUserID,
		OwnerUserID:   m.OwnerUserID,
		ChannelName:   m.ChannelName,
		ChannelID:     m.ChannelID,
		SenderName:    m.SenderName,
		SenderIP:      m.SenderIP,
		Timestamp:     m.Timestamp().Format(time.RFC3339Nano),
		Title:         m.Title,
		Content:       m.TrimmedContent(),
		Priority:      m.Priority,
		UserMessageID: m.UserMessageID,
		Trimmed:       m.NeedsTrim(),
	}
}

func (m Message) Timestamp() time.Time {
	return langext.Coalesce(m.TimestampClient, m.TimestampReal)
}

func (m Message) NeedsTrim() bool {
	return m.Content != nil && len(*m.Content) > ContentLengthTrim
}

func (m Message) TrimmedContent() *string {
	if m.Content == nil {
		return nil
	}
	if !m.NeedsTrim() {
		return m.Content
	}
	return langext.Ptr(langext.Coalesce(m.Content, "")[0:ContentLengthTrim-3] + "...")
}

func (m Message) ShortContent() string {
	if m.Content == nil {
		return ""
	}
	if len(*m.Content) < ContentLengthShort {
		return *m.Content
	}
	return (*m.Content)[0:ContentLengthShort-3] + "..."
}

type MessageJSON struct {
	SCNMessageID  SCNMessageID `json:"scn_message_id"`
	SenderUserID  UserID       `json:"sender_user_id"`
	OwnerUserID   UserID       `json:"owner_user_id"`
	ChannelName   string       `json:"channel_name"`
	ChannelID     ChannelID    `json:"channel_id"`
	SenderName    *string      `json:"sender_name"`
	SenderIP      string       `json:"sender_ip"`
	Timestamp     string       `json:"timestamp"`
	Title         string       `json:"title"`
	Content       *string      `json:"body"`
	Priority      int          `json:"priority"`
	UserMessageID *string      `json:"usr_message_id"`
	Trimmed       bool         `json:"trimmed"`
}

type MessageDB struct {
	SCNMessageID    SCNMessageID `db:"scn_message_id"`
	SenderUserID    UserID       `db:"sender_user_id"`
	OwnerUserID     UserID       `db:"owner_user_id"`
	ChannelName     string       `db:"channel_name"`
	ChannelID       ChannelID    `db:"channel_id"`
	SenderName      *string      `db:"sender_name"`
	SenderIP        string       `db:"sender_ip"`
	TimestampReal   int64        `db:"timestamp_real"`
	TimestampClient *int64       `db:"timestamp_client"`
	Title           string       `db:"title"`
	Content         *string      `db:"content"`
	Priority        int          `db:"priority"`
	UserMessageID   *string      `db:"usr_message_id"`
}

func (m MessageDB) Model() Message {
	return Message{
		SCNMessageID:    m.SCNMessageID,
		SenderUserID:    m.SenderUserID,
		OwnerUserID:     m.OwnerUserID,
		ChannelName:     m.ChannelName,
		ChannelID:       m.ChannelID,
		SenderName:      m.SenderName,
		SenderIP:        m.SenderIP,
		TimestampReal:   time.UnixMilli(m.TimestampReal),
		TimestampClient: timeOptFromMilli(m.TimestampClient),
		Title:           m.Title,
		Content:         m.Content,
		Priority:        m.Priority,
		UserMessageID:   m.UserMessageID,
	}
}

func DecodeMessage(r *sql.Rows) (Message, error) {
	var data MessageDB
	err := scan.RowStrict(&data, r)
	if err != nil {
		return Message{}, err
	}
	return data.Model(), nil
}

func DecodeMessages(r *sql.Rows) ([]Message, error) {
	var data []MessageDB
	err := scan.RowsStrict(&data, r)
	if err != nil {
		return nil, err
	}
	return langext.ArrMap(data, func(v MessageDB) Message { return v.Model() }), nil
}