twonly-app/lib/src/database/tables/messages.table.dart

import 'package:drift/drift.dart';
import 'package:twonly/src/database/tables/contacts.table.dart';
import 'package:twonly/src/database/tables/groups.table.dart';
import 'package:twonly/src/database/tables/mediafiles.table.dart';

enum MessageType { media, text, contacts, restoreFlameCounter, askAboutUser }

@DataClassName('Message')
class Messages extends Table {
  TextColumn get groupId =>
      text().references(Groups, #groupId, onDelete: KeyAction.cascade)();
  TextColumn get messageId => text()();

  // in case senderId is null, it was send by user itself
  IntColumn get senderId =>
      integer().nullable().references(Contacts, #userId)();

  TextColumn get type => text()();

  TextColumn get content => text().nullable()();
  TextColumn get mediaId => text().nullable().references(
    MediaFiles,
    #mediaId,
    onDelete: KeyAction.setNull,
  )();

  BlobColumn get additionalMessageData => blob().nullable()();

  BoolColumn get mediaStored => boolean().withDefault(const Constant(false))();
  BoolColumn get mediaReopened =>
      boolean().withDefault(const Constant(false))();

  BlobColumn get downloadToken => blob().nullable()();

  TextColumn get quotesMessageId => text().nullable()();

  BoolColumn get isDeletedFromSender =>
      boolean().withDefault(const Constant(false))();

  DateTimeColumn get openedAt => dateTime().nullable()();
  DateTimeColumn get openedByAll => dateTime().nullable()();
  DateTimeColumn get createdAt => dateTime().withDefault(currentDateAndTime)();
  DateTimeColumn get modifiedAt => dateTime().nullable()();
  DateTimeColumn get ackByUser => dateTime().nullable()();
  DateTimeColumn get ackByServer => dateTime().nullable()();

  @override
  Set<Column> get primaryKey => {messageId};
}

enum MessageActionType {
  openedAt,
  ackByUserAt,
  ackByServerAt,
}

@DataClassName('MessageAction')
class MessageActions extends Table {
  TextColumn get messageId =>
      text().references(Messages, #messageId, onDelete: KeyAction.cascade)();

  IntColumn get contactId =>
      integer().references(Contacts, #userId, onDelete: KeyAction.cascade)();

  TextColumn get type => textEnum<MessageActionType>()();

  DateTimeColumn get actionAt => dateTime().withDefault(currentDateAndTime)();

  @override
  Set<Column> get primaryKey => {messageId, contactId, type};
}

@DataClassName('MessageHistory')
class MessageHistories extends Table {
  IntColumn get id => integer().autoIncrement()();

  TextColumn get messageId =>
      text().references(Messages, #messageId, onDelete: KeyAction.cascade)();

  IntColumn get contactId => integer().nullable().references(
    Contacts,
    #userId,
    onDelete: KeyAction.cascade,
  )();

  TextColumn get content => text().nullable()();

  DateTimeColumn get createdAt => dateTime().withDefault(currentDateAndTime)();
}