import { DataSource } from 'typeorm';
import { MessageEntity } from '../../../entities';
import { GetMessageByIdQuery } from '../../types';
import { rowToMessage } from '../../mappers';

export async function handleGetMessageById(query: GetMessageByIdQuery, dataSource: DataSource) {
  const repo = dataSource.getRepository(MessageEntity);
  const row = await repo.findOne({ where: { id: query.payload.messageId } });

  return row ? rowToMessage(row) : null;
}