From cd4e48414045301e5e15f97f8537d0f449634b12 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ma=C3=ABl=20Chiotti?=
 <44336112+maelchiotti@users.noreply.github.com>
Date: Sun, 5 Jan 2025 17:48:35 +0100
Subject: [PATCH] Add rich text notes class and migration (#354)

* [303] feat: add rich text notes class and migration

* [303] style: fix comment
---
 lib/common/actions/notes/add.dart             |   5 +-
 lib/common/constants/notes.dart               |  22 +--
 lib/common/logs/app_logger.dart               |  16 +-
 lib/common/preferences/preference_key.dart    |   3 +
 lib/models/deprecated/note.dart               |  47 ++++++
 lib/models/note/note.dart                     | 147 ++++--------------
 lib/models/note/rich_text/rich_text_note.dart | 116 ++++++++++++++
 lib/pages/editor/editor_page.dart             |  11 +-
 lib/services/backup/backup_service.dart       |   4 +-
 lib/services/database_service.dart            |  12 +-
 lib/services/migration/migration_service.dart |  69 ++++++++
 lib/services/notes/notes_service.dart         |  32 ++--
 12 files changed, 329 insertions(+), 155 deletions(-)
 create mode 100644 lib/models/deprecated/note.dart
 create mode 100644 lib/models/note/rich_text/rich_text_note.dart
 create mode 100644 lib/services/migration/migration_service.dart

diff --git a/lib/common/actions/notes/add.dart b/lib/common/actions/notes/add.dart
index 67ad9952..0750df86 100644
--- a/lib/common/actions/notes/add.dart
+++ b/lib/common/actions/notes/add.dart
@@ -1,10 +1,11 @@
 import 'package:flutter/material.dart';
 import 'package:flutter_riverpod/flutter_riverpod.dart';
-import 'select.dart';
+
 import '../../../models/note/note.dart';
 import '../../../navigation/navigator_utils.dart';
 import '../../../providers/notes/notes_provider.dart';
 import '../../../providers/notifiers/notifiers.dart';
+import 'select.dart';
 
 /// Adds a note.
 ///
@@ -14,7 +15,7 @@ Future<void> addNote(BuildContext context, WidgetRef ref, {String? content}) asy
     exitNotesSelectionMode(context, ref);
   }
 
-  final note = content == null ? Note.empty() : Note.content(content);
+  final note = content == null ? RichTextNote.empty() : RichTextNote.content(content);
 
   // If some content was provided, immediately save the note without waiting for changes in the editor
   if (content != null) {
diff --git a/lib/common/constants/notes.dart b/lib/common/constants/notes.dart
index d1858b33..96a30537 100644
--- a/lib/common/constants/notes.dart
+++ b/lib/common/constants/notes.dart
@@ -2,7 +2,7 @@ import '../../models/note/note.dart';
 import '../../utils/localizations_utils.dart';
 
 /// Note displayed on the very first run of the application to welcome the user.
-final welcomeNote = Note(
+final welcomeNote = RichTextNote(
   deleted: false,
   pinned: true,
   createdTime: DateTime.now(),
@@ -14,7 +14,7 @@ final welcomeNote = Note(
 /// Notes used when running integration tests.
 final integrationTestNotes = List.generate(
   100,
-  (index) => Note(
+  (index) => RichTextNote(
     deleted: false,
     pinned: false,
     createdTime: DateTime(2000, 01, 01, 12).subtract(Duration(minutes: index)),
@@ -25,7 +25,7 @@ final integrationTestNotes = List.generate(
 )..addAll(
     List.generate(
       100,
-      (index) => Note(
+      (index) => RichTextNote(
         deleted: true,
         pinned: false,
         createdTime: DateTime(2000, 01, 01, 12).subtract(Duration(minutes: index)),
@@ -38,7 +38,7 @@ final integrationTestNotes = List.generate(
 
 /// Notes used when taking screenshots of the application for the stores.
 final screenshotNotes = [
-  Note(
+  RichTextNote(
     deleted: false,
     pinned: true,
     createdTime: DateTime(2000, 01, 01, 12),
@@ -47,7 +47,7 @@ final screenshotNotes = [
     content:
         '[{"insert":"Simple","attributes":{"b":true}},{"insert":", "},{"insert":"local","attributes":{"i":true}},{"insert":", "},{"insert":"material design","attributes":{"u":true}},{"insert":" notes\\n"}]',
   ),
-  Note(
+  RichTextNote(
     deleted: false,
     pinned: false,
     createdTime: DateTime(2000, 01, 01, 11, 55),
@@ -56,7 +56,7 @@ final screenshotNotes = [
     content:
         '[{"insert":"Write text notes"},{"insert":"\\n","attributes":{"block":"cl","checked":true}},{"insert":"Use formatting options and undo/redo"},{"insert":"\\n","attributes":{"block":"cl","checked":true}},{"insert":"Use quick action to add from your homescreen"},{"insert":"\\n","attributes":{"block":"cl","checked":true}}]',
   ),
-  Note(
+  RichTextNote(
     deleted: false,
     pinned: false,
     createdTime: DateTime(2000, 01, 01, 11, 50),
@@ -64,7 +64,7 @@ final screenshotNotes = [
     title: "Organize",
     content: '[{"insert":"Search, sort and display in a list or a grid\\nPin and recover from the bin\\n"}]',
   ),
-  Note(
+  RichTextNote(
     deleted: false,
     pinned: false,
     createdTime: DateTime(2000, 01, 01, 11, 50),
@@ -72,7 +72,7 @@ final screenshotNotes = [
     title: "Categorize",
     content: '[{"insert":"Categorize notes with labels\\nPin, hide and colorize labels\\n"}]',
   ),
-  Note(
+  RichTextNote(
     deleted: false,
     pinned: false,
     createdTime: DateTime(2000, 01, 01, 11, 45),
@@ -81,7 +81,7 @@ final screenshotNotes = [
     content:
         '[{"insert":"Create a note from shared text\\nShare notes as text and export as Markdown\\nBackup notes as JSON\\n"}]',
   ),
-  Note(
+  RichTextNote(
     deleted: false,
     pinned: false,
     createdTime: DateTime(2000, 01, 01, 11, 40),
@@ -90,7 +90,7 @@ final screenshotNotes = [
     content:
         "[{\"insert\":\"Choose your language\\nChoose your theme (including black and dynamic)\\nHide features you don't need\\n\"}]",
   ),
-  Note(
+  RichTextNote(
     deleted: false,
     pinned: false,
     createdTime: DateTime(2000, 01, 01, 11, 35),
@@ -98,7 +98,7 @@ final screenshotNotes = [
     title: "Protect",
     content: '[{"insert":"Your data never leaves your device\\nEncrypt your exports\\n"}]',
   ),
-  Note(
+  RichTextNote(
     deleted: true,
     pinned: false,
     createdTime: DateTime(2000, 01, 01, 12),
diff --git a/lib/common/logs/app_logger.dart b/lib/common/logs/app_logger.dart
index 9d8f4d7f..69fd2485 100644
--- a/lib/common/logs/app_logger.dart
+++ b/lib/common/logs/app_logger.dart
@@ -3,17 +3,17 @@ import 'dart:ui';
 
 import 'package:flutter/material.dart';
 import 'package:flutter/services.dart';
-import '../constants/constants.dart';
-import '../enums/mime_type.dart';
-import '../extensions/date_time_extensions.dart';
-import '../extensions/string_extension.dart';
-import '../../l10n/app_localizations/app_localizations.g.dart';
-import '../../utils/files_utils.dart';
-import '../../utils/snack_bar_utils.dart';
 import 'package:logger/logger.dart' hide FileOutput;
 import 'package:path/path.dart';
 import 'package:path_provider/path_provider.dart';
 
+import '../../l10n/app_localizations/app_localizations.g.dart';
+import '../../utils/files_utils.dart';
+import '../../utils/snack_bar_utils.dart';
+import '../constants/constants.dart';
+import '../enums/mime_type.dart';
+import '../extensions/date_time_extensions.dart';
+import '../extensions/string_extension.dart';
 import 'filters/debug_filter.dart';
 import 'filters/release_filter.dart';
 
@@ -71,7 +71,7 @@ class AppLogger {
       _fileLogger.e(details.exceptionAsString().firstLine, error: details.exception, stackTrace: details.stack);
     };
     PlatformDispatcher.instance.onError = (exception, stackTrace) {
-      _fileLogger.e(exception.toString().firstLine, error: exception, stackTrace: stackTrace);
+      e(exception.toString().firstLine ?? 'Unknown exception', exception, stackTrace);
 
       return true;
     };
diff --git a/lib/common/preferences/preference_key.dart b/lib/common/preferences/preference_key.dart
index dee49b86..a782d434 100644
--- a/lib/common/preferences/preference_key.dart
+++ b/lib/common/preferences/preference_key.dart
@@ -56,6 +56,9 @@ enum PreferenceKey<T> {
   sortMethod<String>('editedDate', backup: false),
   sortAscending<bool>(false, backup: false),
   layout<String>('list', backup: false),
+
+  // Database
+  databaseVersion(1),
   ;
 
   /// Default value of this preference.
diff --git a/lib/models/deprecated/note.dart b/lib/models/deprecated/note.dart
new file mode 100644
index 00000000..19e30c41
--- /dev/null
+++ b/lib/models/deprecated/note.dart
@@ -0,0 +1,47 @@
+import 'package:isar/isar.dart';
+import 'package:json_annotation/json_annotation.dart';
+
+import '../label/label.dart';
+
+part 'note.g.dart';
+
+// ignore_for_file: must_be_immutable, public_member_api_docs
+
+List<String> _labelToJson(IsarLinks<Label> labels) => labels.map((label) => label.name).toList();
+
+/// Legacy model of the rich text note.
+@JsonSerializable()
+@Collection()
+@Deprecated('Legacy model of the rich text note')
+class Note {
+  @JsonKey(includeFromJson: false, includeToJson: false)
+  Id id = Isar.autoIncrement;
+
+  @JsonKey(includeFromJson: false, includeToJson: false)
+  @ignore
+  bool selected = false;
+
+  bool deleted;
+
+  bool pinned;
+
+  DateTime createdTime;
+
+  DateTime editedTime;
+
+  String title;
+
+  String content;
+
+  @JsonKey(includeFromJson: false, includeToJson: true, toJson: _labelToJson)
+  IsarLinks<Label> labels = IsarLinks<Label>();
+
+  Note({
+    required this.deleted,
+    required this.pinned,
+    required this.createdTime,
+    required this.editedTime,
+    required this.title,
+    required this.content,
+  });
+}
diff --git a/lib/models/note/note.dart b/lib/models/note/note.dart
index 9c6303ab..49ba43b1 100644
--- a/lib/models/note/note.dart
+++ b/lib/models/note/note.dart
@@ -1,29 +1,25 @@
 import 'dart:convert';
 
 import 'package:collection/collection.dart';
-import 'package:equatable/equatable.dart';
 import 'package:fleather/fleather.dart';
 import 'package:isar/isar.dart';
 import 'package:json_annotation/json_annotation.dart';
-import '../../common/constants/constants.dart';
+
+import '../../../common/constants/constants.dart';
+import '../../../utils/encryption_utils.dart';
 import '../../common/preferences/enums/sort_method.dart';
 import '../../common/preferences/preference_key.dart';
 import '../label/label.dart';
-import '../../utils/encryption_utils.dart';
 
 part 'note.g.dart';
 
-// ignore_for_file: must_be_immutable
-
-List<String> _labelToJson(IsarLinks<Label> labels) => labels.map((label) => label.name).toList();
+part 'rich_text/rich_text_note.dart';
 
-/// Rich text note with title, content and metadata.
-@JsonSerializable()
-@Collection(inheritance: false)
-class Note extends Equatable implements Comparable<Note> {
-  /// Empty content in fleather data representation.
-  static const String _emptyContent = '[{"insert":"\\n"}]';
+/// Converts the [labels] to a JSON-compatible list of strings.
+List<String> labelToJson(IsarLinks<Label> labels) => labels.map((label) => label.name).toList();
 
+/// Base class of the notes.
+sealed class Note implements Comparable<Note> {
   /// The ID of the note.
   ///
   /// Excluded from JSON because it's fully managed by Isar.
@@ -54,11 +50,8 @@ class Note extends Equatable implements Comparable<Note> {
   /// The title of the note.
   String title;
 
-  /// The content of the note, as rich text in the fleather representation.
-  String content;
-
   /// The labels used to categorize the note.
-  @JsonKey(includeFromJson: false, includeToJson: true, toJson: _labelToJson)
+  @JsonKey(includeFromJson: false, includeToJson: true, toJson: labelToJson)
   IsarLinks<Label> labels = IsarLinks<Label>();
 
   /// Default constructor of a note.
@@ -68,113 +61,26 @@ class Note extends Equatable implements Comparable<Note> {
     required this.createdTime,
     required this.editedTime,
     required this.title,
-    required this.content,
   });
 
-  /// Note with empty title and content.
-  factory Note.empty() => Note(
-        deleted: false,
-        pinned: false,
-        createdTime: DateTime.now(),
-        editedTime: DateTime.now(),
-        title: '',
-        content: _emptyContent,
-      );
-
-  /// Note with the provided [content].
-  factory Note.content(String content) => Note(
-        deleted: false,
-        pinned: false,
-        createdTime: DateTime.now(),
-        editedTime: DateTime.now(),
-        title: '',
-        content: content,
-      );
-
-  /// Note from [json] data.
-  factory Note.fromJson(Map<String, dynamic> json) => _$NoteFromJson(json);
-
-  /// Note from [json] data, encrypted with [password].
-  factory Note.fromJsonEncrypted(Map<String, dynamic> json, String password) => _$NoteFromJson(json)
-    ..title = (json['title'] as String).isEmpty ? '' : EncryptionUtils().decrypt(password, json['title'] as String)
-    ..content = EncryptionUtils().decrypt(password, json['content'] as String);
-
-  /// Note to JSON.
-  Map<String, dynamic> toJson() => _$NoteToJson(this);
-
-  /// Returns this note with the [title] and the [content] encrypted with the [password].
-  Note encrypted(String password) => this
-    ..title = isTitleEmpty ? '' : EncryptionUtils().encrypt(password, title)
-    ..content = EncryptionUtils().encrypt(password, content);
-
-  /// Returns the visible [labels] of the note as a sorted list.
-  @ignore
-  List<Label> get labelsVisibleSorted => labels.toList().where((label) => label.visible).sorted();
-
-  /// Returns the names of the visible [labels] of the note as a sorted list.
-  @ignore
-  List<String> get labelsNamesVisibleSorted => labelsVisibleSorted.map((label) => label.name).toList();
+  /// Returns this note with title and the content encrypted using the [password].
+  Note encrypted(String password);
 
   /// Note content as plain text.
   @ignore
-  String get plainText => document.toPlainText();
-
-  /// Note content for the preview of the notes tiles.
-  ///
-  /// Formats the following rich text elements:
-  ///   - Checkboxes (TODO: only partially, see https://github.com/maelchiotti/LocalMaterialNotes/issues/121)
-  ///
-  /// Skips the following rich text elements:
-  ///   - Horizontal rules
-  @ignore
-  String get contentPreview {
-    var content = '';
-
-    for (final child in document.root.children) {
-      final operations = child.toDelta().toList();
-
-      for (var i = 0; i < operations.length; i++) {
-        final operation = operations[i];
-
-        // Skip horizontal rules
-        if (operation.data is Map &&
-            (operation.data as Map).containsKey('_type') &&
-            (operation.data as Map)['_type'] == 'hr') {
-          continue;
-        }
-
-        final nextOperation = i == operations.length - 1 ? null : operations[i + 1];
-
-        final checklist = nextOperation != null &&
-            nextOperation.attributes != null &&
-            nextOperation.attributes!.containsKey('block') &&
-            nextOperation.attributes!['block'] == 'cl';
-
-        if (checklist) {
-          final checked = nextOperation.attributes!.containsKey('checked');
-          content += '${checked ? '✅' : '⬜'} ${operation.value}';
-        } else {
-          content += operation.value.toString();
-        }
-      }
-    }
-
-    return content.trim();
-  }
+  String get plainText;
 
   /// Note content as markdown.
   @ignore
-  String get markdown => parchmentMarkdownCodec.encode(document);
+  String get markdown;
 
-  /// Note title and content to be shared as a single text.
-  ///
-  /// Uses the [contentPreview] for the content.
+  /// Note content for the preview of the notes tiles.
   @ignore
-  String get shareText => '$title\n\n$contentPreview';
+  String get contentPreview;
 
-  /// Document containing the fleather content representation.
+  /// Note title and content to be shared as a single text.
   @ignore
-  ParchmentDocument get document => ParchmentDocument.fromJson(jsonDecode(content) as List);
+  String get shareText;
 
   /// Whether the title is empty.
   @ignore
@@ -182,18 +88,24 @@ class Note extends Equatable implements Comparable<Note> {
 
   /// Whether the content is empty.
   @ignore
-  bool get isContentEmpty => content == _emptyContent;
+  bool get isContentEmpty;
 
   /// Whether the preview of the content is empty.
   @ignore
   bool get isContentPreviewEmpty => contentPreview.isEmpty;
 
-  /// Whether the note is empty.
-  ///
-  /// Checks both the title and the content.
+  /// Whether the note is empty (title and content).
   @ignore
   bool get isEmpty => isTitleEmpty && isContentEmpty;
 
+  /// Returns the visible [labels] of the note as a sorted list.
+  @ignore
+  List<Label> get labelsVisibleSorted => labels.toList().where((label) => label.visible).sorted();
+
+  /// Returns the names of the visible [labels] of the note as a sorted list.
+  @ignore
+  List<String> get labelsNamesVisibleSorted => labelsVisibleSorted.map((label) => label.name).toList();
+
   /// Notes are sorted according to:
   ///   1. Their pin state.
   ///   2. The sort method chosen by the user.
@@ -221,6 +133,9 @@ class Note extends Equatable implements Comparable<Note> {
   }
 
   @override
+  bool operator ==(Object other) => other is Note && other.runtimeType == runtimeType && other.id == id;
+
   @ignore
-  List<Object?> get props => [id];
+  @override
+  int get hashCode => id.hashCode;
 }
diff --git a/lib/models/note/rich_text/rich_text_note.dart b/lib/models/note/rich_text/rich_text_note.dart
new file mode 100644
index 00000000..f2b21c8c
--- /dev/null
+++ b/lib/models/note/rich_text/rich_text_note.dart
@@ -0,0 +1,116 @@
+part of '../note.dart';
+
+/// Rich text note.
+@JsonSerializable()
+@Collection()
+class RichTextNote extends Note {
+  /// Empty content in fleather data representation.
+  static const String _emptyContent = '[{"insert":"\\n"}]';
+
+  /// The content of the note, as rich text in the fleather representation.
+  String content;
+
+  /// Note with rich text content.
+  RichTextNote({
+    required super.deleted,
+    required super.pinned,
+    required super.createdTime,
+    required super.editedTime,
+    required super.title,
+    required this.content,
+  });
+
+  /// Rich text note with empty title and content.
+  factory RichTextNote.empty() => RichTextNote(
+        deleted: false,
+        pinned: false,
+        createdTime: DateTime.now(),
+        editedTime: DateTime.now(),
+        title: '',
+        content: _emptyContent,
+      );
+
+  /// Rich text note with the provided [content].
+  factory RichTextNote.content(String content) => RichTextNote(
+        deleted: false,
+        pinned: false,
+        createdTime: DateTime.now(),
+        editedTime: DateTime.now(),
+        title: '',
+        content: content,
+      );
+
+  /// Rich text note from [json] data.
+  factory RichTextNote.fromJson(Map<String, dynamic> json) => _$RichTextNoteFromJson(json);
+
+  /// Rich text note from [json] data, encrypted with [password].
+  factory RichTextNote.fromJsonEncrypted(Map<String, dynamic> json, String password) => _$RichTextNoteFromJson(json)
+    ..title = (json['title'] as String).isEmpty ? '' : EncryptionUtils().decrypt(password, json['title'] as String)
+    ..content = EncryptionUtils().decrypt(password, json['content'] as String);
+
+  /// Rich text note to JSON.
+  Map<String, dynamic> toJson() => _$RichTextNoteToJson(this);
+
+  /// Returns this note with the [title] and the [content] encrypted with the [password].
+  @override
+  Note encrypted(String password) => this
+    ..title = isTitleEmpty ? '' : EncryptionUtils().encrypt(password, title)
+    ..content = EncryptionUtils().encrypt(password, content);
+
+  /// Document containing the fleather content representation.
+  @ignore
+  ParchmentDocument get document => ParchmentDocument.fromJson(jsonDecode(content) as List);
+
+  @ignore
+  @override
+  String get plainText => document.toPlainText();
+
+  @ignore
+  @override
+  String get markdown => parchmentMarkdownCodec.encode(document);
+
+  @ignore
+  @override
+  String get contentPreview {
+    var content = '';
+
+    for (final child in document.root.children) {
+      final operations = child.toDelta().toList();
+
+      for (var i = 0; i < operations.length; i++) {
+        final operation = operations[i];
+
+        // Skip horizontal rules
+        if (operation.data is Map &&
+            (operation.data as Map).containsKey('_type') &&
+            (operation.data as Map)['_type'] == 'hr') {
+          continue;
+        }
+
+        final nextOperation = i == operations.length - 1 ? null : operations[i + 1];
+
+        final checklist = nextOperation != null &&
+            nextOperation.attributes != null &&
+            nextOperation.attributes!.containsKey('block') &&
+            nextOperation.attributes!['block'] == 'cl';
+
+        if (checklist) {
+          final checked = nextOperation.attributes!.containsKey('checked');
+          content += '${checked ? '✅' : '⬜'} ${operation.value}';
+        } else {
+          content += operation.value.toString();
+        }
+      }
+    }
+
+    return content.trim();
+  }
+
+  @ignore
+  @override
+  String get shareText => '$title\n\n$contentPreview';
+
+  @ignore
+  @override
+  bool get isContentEmpty => content == _emptyContent;
+}
diff --git a/lib/pages/editor/editor_page.dart b/lib/pages/editor/editor_page.dart
index 3534f948..ebcabe41 100644
--- a/lib/pages/editor/editor_page.dart
+++ b/lib/pages/editor/editor_page.dart
@@ -78,7 +78,10 @@ class _EditorState extends ConsumerState<NotesEditorPage> {
     fleatherControllerCanUndoNotifier.value = editorController.canUndo;
     fleatherControllerCanRedoNotifier.value = editorController.canRedo;
 
-    note.content = jsonEncode(editorController.document.toDelta().toJson());
+    switch (note) {
+      case RichTextNote note:
+        note.content = jsonEncode(editorController.document.toDelta().toJson());
+    }
 
     ref.read(notesProvider.notifier).edit(note);
   }
@@ -105,7 +108,11 @@ class _EditorState extends ConsumerState<NotesEditorPage> {
 
         titleController = TextEditingController(text: currentNote.title);
 
-        editorController = FleatherController(document: currentNote.document);
+        switch (currentNote) {
+          case final RichTextNote note:
+            editorController = FleatherController(document: note.document);
+        }
+
         editorController.addListener(() => _synchronizeContent(currentNote));
 
         fleatherControllerNotifier.value = editorController;
diff --git a/lib/services/backup/backup_service.dart b/lib/services/backup/backup_service.dart
index 36e3d49f..e5c6186f 100644
--- a/lib/services/backup/backup_service.dart
+++ b/lib/services/backup/backup_service.dart
@@ -120,7 +120,7 @@ class ManualBackupService {
 
       try {
         for (final noteAsJsonEncrypted in notesAsJson) {
-          notes.add(Note.fromJsonEncrypted(noteAsJsonEncrypted, password));
+          notes.add(RichTextNote.fromJsonEncrypted(noteAsJsonEncrypted, password));
         }
       } catch (exception, stackTrace) {
         logger.e(exception.toString(), exception, stackTrace);
@@ -134,7 +134,7 @@ class ManualBackupService {
       }
     } else {
       for (final noteAsJson in notesAsJson) {
-        notes.add(Note.fromJson(noteAsJson));
+        notes.add(RichTextNote.fromJson(noteAsJson));
       }
     }
 
diff --git a/lib/services/database_service.dart b/lib/services/database_service.dart
index ade55114..4e78f8c3 100644
--- a/lib/services/database_service.dart
+++ b/lib/services/database_service.dart
@@ -1,10 +1,13 @@
 import 'package:flutter_mimir/flutter_mimir.dart';
 import 'package:isar/isar.dart';
+import 'package:path_provider/path_provider.dart';
+
+import '../models/deprecated/note.dart';
 import '../models/label/label.dart';
 import '../models/note/note.dart';
 import 'labels/labels_service.dart';
+import 'migration/migration_service.dart';
 import 'notes/notes_service.dart';
-import 'package:path_provider/path_provider.dart';
 
 /// Abstract service for the database.
 ///
@@ -28,12 +31,17 @@ class DatabaseService {
     final databaseName = 'materialnotes';
     final databaseDirectory = (await getApplicationDocumentsDirectory()).path;
 
+    // Initialize the database
     database = await Isar.open(
-      [NoteSchema, LabelSchema],
+      [NoteSchema, RichTextNoteSchema, LabelSchema],
       name: databaseName,
       directory: databaseDirectory,
     );
 
+    // Perform migrations if needed
+    await MigrationService().migrateIfNeeded();
+
+    // Initialize mimir
     mimir = await Mimir.defaultInstance;
 
     // Initialize the models services
diff --git a/lib/services/migration/migration_service.dart b/lib/services/migration/migration_service.dart
new file mode 100644
index 00000000..39f4ce5e
--- /dev/null
+++ b/lib/services/migration/migration_service.dart
@@ -0,0 +1,69 @@
+import 'package:isar/isar.dart';
+
+import '../../common/constants/constants.dart';
+import '../../common/preferences/preference_key.dart';
+import '../../models/deprecated/note.dart';
+import '../../models/note/note.dart';
+import '../database_service.dart';
+
+/// Database migration service.
+class MigrationService {
+  final _databaseService = DatabaseService();
+
+  /// Migrates the database to the latest version if needed.
+  Future<void> migrateIfNeeded() async {
+    final databaseVersion = PreferenceKey.databaseVersion.getPreferenceOrDefault();
+
+    switch (databaseVersion) {
+      case 1:
+        await _migrateToV2();
+    }
+  }
+
+  /// Migrates the database to version 2.
+  ///
+  /// Moves all legacy notes from the legacy `notes` collection to the new `richTextNotes` collection
+  /// to allow adding different types of notes.
+  Future<void> _migrateToV2() async {
+    logger.i('Migrating the database from version 1 to version 2');
+
+    final database = _databaseService.database;
+
+    // Convert the legacy notes to rich text notes
+    final legacyNotes = await database.notes.where().findAll();
+    final richTextNotes = legacyNotes
+        .map(
+          (legacyNote) => RichTextNote(
+            deleted: legacyNote.deleted,
+            pinned: legacyNote.pinned,
+            createdTime: legacyNote.createdTime,
+            editedTime: legacyNote.editedTime,
+            title: legacyNote.title,
+            content: legacyNote.content,
+          ),
+        )
+        .toList();
+
+    // Add all the notes to the new collection
+    var addedRichTextNotes = <int>[];
+    await database.writeTxn(() async {
+      addedRichTextNotes = await database.richTextNotes.putAll(richTextNotes);
+    });
+
+    // Check that the migration succeeded
+    final legacyNotesCount = await database.notes.count();
+    final addedRichTextNotesCount = addedRichTextNotes.length;
+    assert(
+      legacyNotesCount == addedRichTextNotesCount,
+      'The count of legacy notes ($legacyNotesCount) is different from the count of rich text notes ($addedRichTextNotesCount) after the migration to v2',
+    );
+
+    // Remove all the legacy notes
+    await database.writeTxn(() async {
+      await database.notes.clear();
+    });
+
+    // Update the database version
+    await PreferenceKey.databaseVersion.set(2);
+  }
+}
diff --git a/lib/services/notes/notes_service.dart b/lib/services/notes/notes_service.dart
index 2fd5c413..ade895b3 100644
--- a/lib/services/notes/notes_service.dart
+++ b/lib/services/notes/notes_service.dart
@@ -23,7 +23,7 @@ class NotesService {
   NotesService._internal();
 
   final Isar _database = DatabaseService().database;
-  final _notes = DatabaseService().database.notes;
+  final _richTextNotes = DatabaseService().database.richTextNotes;
 
   late final MimirIndex _index;
 
@@ -86,16 +86,16 @@ class NotesService {
   }
 
   /// Returns the total number of notes.
-  Future<int> get count async => await _notes.count();
+  Future<int> get count async => await _richTextNotes.count();
 
   /// Returns all the notes.
-  Future<List<Note>> getAll() async => _notes.where().findAll();
+  Future<List<Note>> getAll() async => _richTextNotes.where().findAll();
 
   /// Returns all the notes that are not deleted.
-  Future<List<Note>> getAllNotDeleted() async => _notes.where().deletedEqualTo(false).findAll();
+  Future<List<Note>> getAllNotDeleted() async => _richTextNotes.where().deletedEqualTo(false).findAll();
 
   /// Returns all the notes that are deleted.
-  Future<List<Note>> getAllDeleted() async => _notes.where().deletedEqualTo(true).findAll();
+  Future<List<Note>> getAllDeleted() async => _richTextNotes.where().deletedEqualTo(true).findAll();
 
   /// Returns all the notes containing the [label].
   Future<List<Note>> filterByLabel(Label label) async {
@@ -127,7 +127,7 @@ class NotesService {
     );
     final notesIds = searchResults.map((Map<String, dynamic> noteIndex) => noteIndex['id'] as int).toList();
 
-    final notes = (await _notes.getAll(notesIds));
+    final notes = (await _richTextNotes.getAll(notesIds));
     final notesNotNull = notes.nonNulls.toList();
 
     // Check that all search results correspond to an existing note
@@ -142,7 +142,10 @@ class NotesService {
   /// Puts the [note] in the database.
   Future<void> put(Note note) async {
     await _database.writeTxn(() async {
-      await _notes.put(note);
+      switch (note) {
+        case final RichTextNote note:
+          await _richTextNotes.put(note);
+      }
     });
 
     await _updateIndex(note);
@@ -151,7 +154,12 @@ class NotesService {
   /// Puts the [notes] in the database.
   Future<void> putAll(List<Note> notes) async {
     await _database.writeTxn(() async {
-      await _notes.putAll(notes);
+      for (final note in notes) {
+        switch (note) {
+          case final RichTextNote note:
+            await _richTextNotes.put(note);
+        }
+      }
     });
 
     await _updateAllIndexes(notes);
@@ -201,7 +209,7 @@ class NotesService {
   /// Deletes the [note] from the database.
   Future<void> delete(Note note) async {
     await _database.writeTxn(() async {
-      await _notes.delete(note.id);
+      await _richTextNotes.delete(note.id);
     });
 
     await _deleteIndex(note);
@@ -210,7 +218,7 @@ class NotesService {
   /// Deletes the [notes] from the database.
   Future<void> deleteAll(List<Note> notes) async {
     await _database.writeTxn(() async {
-      await _notes.deleteAll(notes.map((note) => note.id).toList());
+      await _richTextNotes.deleteAll(notes.map((note) => note.id).toList());
     });
 
     await _deleteAllIndexes(notes);
@@ -221,7 +229,7 @@ class NotesService {
     final notes = await getAllDeleted();
 
     await _database.writeTxn(() async {
-      await _notes.where().deletedEqualTo(true).deleteAll();
+      await _richTextNotes.where().deletedEqualTo(true).deleteAll();
     });
 
     _deleteAllIndexes(notes);
@@ -232,7 +240,7 @@ class NotesService {
     final notes = await getAllNotDeleted();
 
     await _database.writeTxn(() async {
-      await _notes.where().deleteAll();
+      await _richTextNotes.where().deleteAll();
     });
 
     _deleteAllIndexes(notes);