package db

import (
	"github.com/tidbcloud/jiraport/pkg/entity"
)

func (c *Client) AddMessage(notificationID uint, msgIndex *entity.MessageIndex) error {
	msg := entity.Message{
		NotificationID: notificationID,
		MessageIndex:   msgIndex,
	}
	err := c.dataDB.Omit("Notification").Create(&msg).Error
	if err != nil {
		return err
	}
	return nil
}

func (c *Client) GetMessage(msgIndex *entity.MessageIndex) (*entity.Message, error) {
	var msg entity.Message
	err := c.dataDB.
		Where(&entity.Message{MessageIndex: msgIndex}).
		Order("id DESC").
		First(&msg).
		Error
	if err != nil {
		return nil, err
	}
	return &msg, nil
}