OpenPeerChat-flutter/lib/database/message_db.dart

import 'dart:async';
import 'package:path/path.dart';
import 'package:sqflite/sqflite.dart';
import 'model.dart';

/// It is the database for the messages.

class MessageDB {
  static final MessageDB instance = MessageDB._init();

  static Database? _database;

  MessageDB._init();

  Future<Database> get database async {
    if (_database != null) return _database!;
    _database = await _initDB('p2p.db');
    return _database!;
  }

  Future<Database> _initDB(String filePath) async {
    final dbPath = await getDatabasesPath();
    final path = join(dbPath, filePath);
    return await openDatabase(path, version: 2, onCreate: _createDB);
  }

  Future _createDB(Database db, int version) async {
    await db.execute(
        'CREATE TABLE $messagesTableName(_id PRIMARY KEY, type TEXT NOT NULL,msg TEXT NOT NULL);');
    await db.execute(
        'CREATE TABLE $conversationsTableName(_id PRIMARY KEY, converser TEXT NOT NULL,type TEXT NOT NULL,msg TEXT NOT NULL,timestamp TEXT NOT NULL, ack TEXT NOT NULL);');
    await db.execute(
        'CREATE TABLE $publicKeyTableName( ${PublicKeyFields.converser} TEXT NOT NULL,${PublicKeyFields.publicKey} TEXT NOT NULL);');
  }
  Future<void> insertPublicKey(PublicKeyFromDB publicKey) async {
    final db = await instance.database;

    await db.insert(publicKeyTableName, publicKey.toJson(),
        conflictAlgorithm: ConflictAlgorithm.replace);
  }

  Future<List<PublicKeyFromDB>> readAllFromPublicKeyTable() async {
    final db = await instance.database;

    final result = await db.query(publicKeyTableName);

    return result.map((json) => PublicKeyFromDB.fromJson(json)).toList();
  }

  Future<PublicKeyFromDB?> getPublicKey(String converser) async {
    final db = await instance.database;

    final result = await db.query(publicKeyTableName,
        where: '${PublicKeyFields.converser} = ?', whereArgs: [converser]);

    if (result.isNotEmpty) {
      return PublicKeyFromDB.fromJson(result.first);
    } else {
      return null;
    }
  }


  void insertIntoMessagesTable(MessageFromDB message) async {
    final db = await instance.database;
    await db.insert(messagesTableName, message.toJson());
    return;
  }

  void insertIntoConversationsTable(ConversationFromDB message) async {
    final db = await instance.database;
    await db.insert(conversationsTableName, message.toJson());
    return;
  }

  // Function to get last message id from the database
  // in the conversation table
  // The type can be modified based on the type of message
  // that is being sent or received
  Future<String> getLastMessageId({required String type}) async {
    final db = await instance.database;
    final message = await db.query(
      conversationsTableName,
      where: '${ConversationTableFields.type}=?',
      whereArgs: [type],
      orderBy: '${ConversationTableFields.timestamp} DESC',
      limit: 1,
    );
    if (message.isEmpty) return "-1"; // If error in database.
    return MessageFromDB.fromJson(message[0]).id;
  }

  Future<MessageFromDB?> readFromMessagesTable(int id) async {
    final db = await instance.database;
    final maps = await db.query(
      messagesTableName,
      columns: MessageTableFields.values,
      where: '${MessageTableFields.id}=?',
      whereArgs: [id],
    );
    if (maps.isNotEmpty) {
      return MessageFromDB.fromJson(maps.first);
    } else {
      return null;
    }
  }

  Future<ConversationFromDB?> readFromConversationsTable(int id) async {
    final db = await instance.database;
    final maps = await db.query(
      conversationsTableName,
      columns: ConversationTableFields.values,
      where: '${ConversationTableFields.id}=?',
      whereArgs: [id],
    );
    if (maps.isNotEmpty) {
      return ConversationFromDB.fromJson(maps.first);
    } else {
      return null;
    }
  }

  Future<List<MessageFromDB>> readAllFromMessagesTable() async {
    final db = await instance.database;
    final result = await db.query(
      messagesTableName,
    );
    return result.map((json) => MessageFromDB.fromJson(json)).toList();
  }

  Future<List<ConversationFromDB>> readAllFromConversationsTable() async {
    final db = await instance.database;
    final result = await db.query(
      conversationsTableName,
    );
    return result.map((json) => ConversationFromDB.fromJson(json)).toList();
  }

  Future<int> updateMessageTable(MessageFromDB msg) async {
    final db = await instance.database;
    return db.update(messagesTableName, msg.toJson(),
        where: '${MessageTableFields.id}=?', whereArgs: [msg.id]);
  }

  Future<int> updateConversationTable(ConversationFromDB msg) async {
    final db = await instance.database;
    return db.update(conversationsTableName, msg.toJson(),
        where: '${ConversationTableFields.id}=?', whereArgs: [msg.id]);
  }

  Future<int> deleteFromMessagesTable(String id) async {
    final db = await instance.database;
    return db.delete(messagesTableName,
        where: '${MessageTableFields.id}=?', whereArgs: [id]);
  }

  Future<int> deleteFromConversationsTable(String id) async {
    final db = await instance.database;
    return db.delete(conversationsTableName,
        where: '${ConversationTableFields.id}=?', whereArgs: [id]);
  }

  Future close() async {
    final db = await instance.database;
    db.close();
  }
}