♻️ Refactored post draft system

lib/services/compose_storage_db.dart

@@ -1,183 +1,16 @@
 import 'dart:convert';
 import 'package:drift/drift.dart';
-import 'package:riverpod_annotation/riverpod_annotation.dart';
 import 'package:island/database/drift_db.dart';
+import 'package:island/models/post.dart';
 import 'package:island/pods/database.dart';
-import 'package:island/services/file.dart';
-import 'package:island/models/file.dart';
-import 'package:island/pods/config.dart';
-import 'package:island/pods/network.dart';
+import 'package:riverpod_annotation/riverpod_annotation.dart';
 
 part 'compose_storage_db.g.dart';
 
-class ComposeDraftModel {
-  final String id;
-  final String title;
-  final String description;
-  final String content;
-  final List<UniversalFile> attachments;
-  final int visibility;
-  final DateTime lastModified;
-
-  ComposeDraftModel({
-    required this.id,
-    required this.title,
-    required this.description,
-    required this.content,
-    required this.attachments,
-    required this.visibility,
-    required this.lastModified,
-  });
-
-  Map<String, dynamic> toJson() => {
-    'id': id,
-    'title': title,
-    'description': description,
-    'content': content,
-    'attachments': attachments.map((e) => e.toJson()).toList(),
-    'visibility': visibility,
-    'lastModified': lastModified.toIso8601String(),
-  };
-
-  factory ComposeDraftModel.fromJson(Map<String, dynamic> json) => ComposeDraftModel(
-    id: json['id'] as String,
-    title: json['title'] as String? ?? '',
-    description: json['description'] as String? ?? '',
-    content: json['content'] as String? ?? '',
-    attachments: (json['attachments'] as List? ?? [])
-        .map((e) => UniversalFile.fromJson(e as Map<String, dynamic>))
-        .toList(),
-    visibility: json['visibility'] as int? ?? 0,
-    lastModified: DateTime.parse(json['lastModified'] as String),
-  );
-
-  factory ComposeDraftModel.fromDbRow(ComposeDraft row) => ComposeDraftModel(
-    id: row.id,
-    title: row.title,
-    description: row.description,
-    content: row.content,
-    attachments: (jsonDecode(row.attachmentIds) as List)
-        .map((e) => UniversalFile.fromJson(e as Map<String, dynamic>))
-        .toList(),
-    visibility: row.visibility,
-    lastModified: row.lastModified,
-  );
-
-  ComposeDraftsCompanion toDbCompanion() => ComposeDraftsCompanion(
-    id: Value(id),
-    title: Value(title),
-    description: Value(description),
-    content: Value(content),
-    attachmentIds: Value(jsonEncode(attachments.map((e) => e.toJson()).toList())),
-    visibility: Value(visibility),
-    lastModified: Value(lastModified),
-  );
-
-  ComposeDraftModel copyWith({
-    String? id,
-    String? title,
-    String? description,
-    String? content,
-    List<UniversalFile>? attachments,
-    int? visibility,
-    DateTime? lastModified,
-  }) {
-    return ComposeDraftModel(
-      id: id ?? this.id,
-      title: title ?? this.title,
-      description: description ?? this.description,
-      content: content ?? this.content,
-      attachments: attachments ?? this.attachments,
-      visibility: visibility ?? this.visibility,
-      lastModified: lastModified ?? this.lastModified,
-    );
-  }
-
-  bool get isEmpty =>
-      title.isEmpty &&
-      description.isEmpty &&
-      content.isEmpty &&
-      attachments.isEmpty;
-}
-
-class ArticleDraftModel {
-  final String id;
-  final String title;
-  final String description;
-  final String content;
-  final int visibility;
-  final DateTime lastModified;
-
-  ArticleDraftModel({
-    required this.id,
-    required this.title,
-    required this.description,
-    required this.content,
-    required this.visibility,
-    required this.lastModified,
-  });
-
-  Map<String, dynamic> toJson() => {
-    'id': id,
-    'title': title,
-    'description': description,
-    'content': content,
-    'visibility': visibility,
-    'lastModified': lastModified.toIso8601String(),
-  };
-
-  factory ArticleDraftModel.fromJson(Map<String, dynamic> json) => ArticleDraftModel(
-    id: json['id'] as String,
-    title: json['title'] as String? ?? '',
-    description: json['description'] as String? ?? '',
-    content: json['content'] as String? ?? '',
-    visibility: json['visibility'] as int? ?? 0,
-    lastModified: DateTime.parse(json['lastModified'] as String),
-  );
-
-  factory ArticleDraftModel.fromDbRow(ArticleDraft row) => ArticleDraftModel(
-    id: row.id,
-    title: row.title,
-    description: row.description,
-    content: row.content,
-    visibility: row.visibility,
-    lastModified: row.lastModified,
-  );
-
-  ArticleDraftsCompanion toDbCompanion() => ArticleDraftsCompanion(
-    id: Value(id),
-    title: Value(title),
-    description: Value(description),
-    content: Value(content),
-    visibility: Value(visibility),
-    lastModified: Value(lastModified),
-  );
-
-  ArticleDraftModel copyWith({
-    String? id,
-    String? title,
-    String? description,
-    String? content,
-    int? visibility,
-    DateTime? lastModified,
-  }) {
-    return ArticleDraftModel(
-      id: id ?? this.id,
-      title: title ?? this.title,
-      description: description ?? this.description,
-      content: content ?? this.content,
-      visibility: visibility ?? this.visibility,
-      lastModified: lastModified ?? this.lastModified,
-    );
-  }
-
-  bool get isEmpty => title.isEmpty && description.isEmpty && content.isEmpty;
-}
-
 @riverpod
 class ComposeStorageNotifier extends _$ComposeStorageNotifier {
   @override
-  Map<String, ComposeDraftModel> build() {
+  Map<String, SnPost> build() {
     _loadDrafts();
     return {};
   }
@@ -185,10 +18,9 @@ class ComposeStorageNotifier extends _$ComposeStorageNotifier {
   void _loadDrafts() async {
     try {
       final database = ref.read(databaseProvider);
-      final dbDrafts = await database.getAllComposeDrafts();
-      final drafts = <String, ComposeDraftModel>{};
-      for (final dbDraft in dbDrafts) {
-        final draft = ComposeDraftModel.fromDbRow(dbDraft);
+      final dbDrafts = await database.getAllPostDrafts();
+      final drafts = <String, SnPost>{};
+      for (final draft in dbDrafts) {
         drafts[draft.id] = draft;
       }
       state = drafts;
@@ -198,52 +30,22 @@ class ComposeStorageNotifier extends _$ComposeStorageNotifier {
     }
   }
 
-  Future<void> saveDraft(ComposeDraftModel draft) async {
-    if (draft.isEmpty) {
-      await deleteDraft(draft.id);
-      return;
-    }
-
-    // Upload all attachments that are not yet uploaded
-    final uploadedAttachments = <UniversalFile>[];
-    final serverUrl = ref.read(serverUrlProvider);
-    final token = ref.read(tokenProvider);
-    
-    for (final attachment in draft.attachments) {
-      if (!attachment.isOnCloud) {
-        try {
-          final completer = putMediaToCloud(
-            fileData: attachment,
-            atk: token?.token ?? '',
-            baseUrl: serverUrl,
-          );
-          final uploadedFile = await completer.future;
-          if (uploadedFile != null) {
-            uploadedAttachments.add(UniversalFile.fromAttachment(uploadedFile));
-          } else {
-            uploadedAttachments.add(attachment);
-          }
-        } catch (e) {
-          // If upload fails, keep the original file
-          uploadedAttachments.add(attachment);
-        }
-      } else {
-        uploadedAttachments.add(attachment);
-      }
-    }
-
-    final updatedDraft = draft.copyWith(
-      attachments: uploadedAttachments,
-      lastModified: DateTime.now(),
-    );
+  Future<void> saveDraft(SnPost draft) async {
+    final updatedDraft = draft.copyWith(updatedAt: DateTime.now());
     state = {...state, updatedDraft.id: updatedDraft};
-    
+
     try {
       final database = ref.read(databaseProvider);
-      await database.saveComposeDraft(updatedDraft.toDbCompanion());
+      await database.addPostDraft(
+        PostDraftsCompanion(
+          id: Value(updatedDraft.id),
+          post: Value(jsonEncode(updatedDraft.toJson())),
+          lastModified: Value(updatedDraft.updatedAt ?? DateTime.now()),
+        ),
+      );
     } catch (e) {
       // Revert state on error
-      final newState = Map<String, ComposeDraftModel>.from(state);
+      final newState = Map<String, SnPost>.from(state);
       newState.remove(updatedDraft.id);
       state = newState;
       rethrow;
@@ -252,13 +54,13 @@ class ComposeStorageNotifier extends _$ComposeStorageNotifier {
 
   Future<void> deleteDraft(String id) async {
     final oldDraft = state[id];
-    final newState = Map<String, ComposeDraftModel>.from(state);
+    final newState = Map<String, SnPost>.from(state);
     newState.remove(id);
     state = newState;
-    
+
     try {
       final database = ref.read(databaseProvider);
-      await database.deleteComposeDraft(id);
+      await database.deletePostDraft(id);
     } catch (e) {
       // Revert state on error
       if (oldDraft != null) {
@@ -268,22 +70,22 @@ class ComposeStorageNotifier extends _$ComposeStorageNotifier {
     }
   }
 
-  ComposeDraftModel? getDraft(String id) {
+  SnPost? getDraft(String id) {
     return state[id];
   }
 
-  List<ComposeDraftModel> getAllDrafts() {
+  List<SnPost> getAllDrafts() {
     final drafts = state.values.toList();
-    drafts.sort((a, b) => b.lastModified.compareTo(a.lastModified));
+    drafts.sort((a, b) => b.updatedAt!.compareTo(a.updatedAt!));
     return drafts;
   }
 
   Future<void> clearAllDrafts() async {
     state = {};
-    
+
     try {
       final database = ref.read(databaseProvider);
-      await database.clearAllComposeDrafts();
+      await database.clearAllPostDrafts();
     } catch (e) {
       // If clearing fails, we might want to reload from database
       _loadDrafts();
@@ -291,90 +93,3 @@ class ComposeStorageNotifier extends _$ComposeStorageNotifier {
     }
   }
 }
-
-@riverpod
-class ArticleStorageNotifier extends _$ArticleStorageNotifier {
-  @override
-  Map<String, ArticleDraftModel> build() {
-    _loadDrafts();
-    return {};
-  }
-
-  void _loadDrafts() async {
-    try {
-      final database = ref.read(databaseProvider);
-      final dbDrafts = await database.getAllArticleDrafts();
-      final drafts = <String, ArticleDraftModel>{};
-      for (final dbDraft in dbDrafts) {
-        final draft = ArticleDraftModel.fromDbRow(dbDraft);
-        drafts[draft.id] = draft;
-      }
-      state = drafts;
-    } catch (e) {
-      // If there's an error loading drafts, start with empty state
-      state = {};
-    }
-  }
-
-  Future<void> saveDraft(ArticleDraftModel draft) async {
-    if (draft.isEmpty) {
-      await deleteDraft(draft.id);
-      return;
-    }
-
-    final updatedDraft = draft.copyWith(lastModified: DateTime.now());
-    state = {...state, updatedDraft.id: updatedDraft};
-    
-    try {
-      final database = ref.read(databaseProvider);
-      await database.saveArticleDraft(updatedDraft.toDbCompanion());
-    } catch (e) {
-      // Revert state on error
-      final newState = Map<String, ArticleDraftModel>.from(state);
-      newState.remove(updatedDraft.id);
-      state = newState;
-      rethrow;
-    }
-  }
-
-  Future<void> deleteDraft(String id) async {
-    final oldDraft = state[id];
-    final newState = Map<String, ArticleDraftModel>.from(state);
-    newState.remove(id);
-    state = newState;
-    
-    try {
-      final database = ref.read(databaseProvider);
-      await database.deleteArticleDraft(id);
-    } catch (e) {
-      // Revert state on error
-      if (oldDraft != null) {
-        state = {...state, id: oldDraft};
-      }
-      rethrow;
-    }
-  }
-
-  ArticleDraftModel? getDraft(String id) {
-    return state[id];
-  }
-
-  List<ArticleDraftModel> getAllDrafts() {
-    final drafts = state.values.toList();
-    drafts.sort((a, b) => b.lastModified.compareTo(a.lastModified));
-    return drafts;
-  }
-
-  Future<void> clearAllDrafts() async {
-    state = {};
-    
-    try {
-      final database = ref.read(databaseProvider);
-      await database.clearAllArticleDrafts();
-    } catch (e) {
-      // If clearing fails, we might want to reload from database
-      _loadDrafts();
-      rethrow;
-    }
-  }
-}