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    int64
	SenderUserID    int64
	OwnerUserID     int64
	ChannelName     string
	ChannelID       int64
	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,
		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,
		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  int64   `json:"scn_message_id"`
	SenderUserID  int64   `json:"sender_user_id"`
	OwnerUserID   int64   `json:"owner_user_id"`
	ChannelName   string  `json:"channel_name"`
	ChannelID     int64   `json:"channel_id"`
	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    int64   `db:"scn_message_id"`
	SenderUserID    int64   `db:"sender_user_id"`
	OwnerUserID     int64   `db:"owner_user_id"`
	ChannelName     string  `db:"channel_name"`
	ChannelID       int64   `db:"channel_id"`
	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,
		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
}