feat: export image as png

Author: TeeWrath

lib/features/workspace/providers/workspace_provider.dart

@@ -1,6 +1,4 @@
-// lib/features/workspace/providers/workspace_provider.dart (Fully Modified)
-
-import 'dart:ui';
+import 'dart:ui' as ui;
 import 'dart:convert'; // For jsonDecode/jsonEncode
 import 'dart:math';
 import 'package:cookethflow/core/helpers/file_helper.dart';
@@ -11,6 +9,7 @@ import 'package:cookethflow/core/utils/enums.dart';
 import 'package:cookethflow/core/utils/state_handler.dart';
 import 'package:cookethflow/features/dashboard/providers/dashboard_provider.dart';
 import 'package:cookethflow/features/models/canvas_models/canvas_object.dart';
+import 'package:cookethflow/features/models/canvas_models/canvas_painter.dart';
 import 'package:cookethflow/features/models/canvas_models/objects/circle_object.dart';
 import 'package:cookethflow/features/models/canvas_models/objects/connector_object.dart';
 import 'package:cookethflow/features/models/canvas_models/objects/cylinder_object.dart';
@@ -78,8 +77,7 @@ class WorkspaceProvider extends StateHandler {
   String? _connectorSourceId;
   Alignment? _connectorSourceAlignment;
   Offset? _connectorDragPosition;
-  
-  // Create an instance of FileServices
+
   final FileServices _fileServices = FileServices();
 
   bool get isLoading => _isLoading;
@@ -118,40 +116,34 @@ class WorkspaceProvider extends StateHandler {
     return _tempQuillController;
   }
 
-  // --- NEW EXPORT METHOD ---
   Future<void> exportWorkspaceAsJson() async {
     if (_currentWorkspace == null) {
       print("Cannot export: No workspace is currently loaded.");
       return;
     }
 
     try {
-      // 1. Gather all necessary data
       final workspaceData = _currentWorkspace!.toJson();
       final canvasObjectsData =
           _canvasObjects.values.map((obj) => obj.toJson()).toList();
 
-      // 2. Structure the data into a single map
       final exportData = {
         'workspace': {
           'name': workspaceData["name"],
-          'data': workspaceData["data"]
+          'data': workspaceData["data"],
         },
         'canvasObjects': canvasObjectsData,
       };
 
-      // 3. Encode the map into a formatted JSON string
-      const jsonEncoder = JsonEncoder.withIndent('  '); // For pretty printing
+      const jsonEncoder = JsonEncoder.withIndent('  ');
       final jsonString = jsonEncoder.convert(exportData);
 
-      // 4. Generate a safe and unique file name
       final safeWorkspaceName = _currentWorkspace!.name
           .replaceAll(RegExp(r'[^\w\s-]'), '')
           .replaceAll(' ', '_');
       final uniqueId = const Uuid().v4().substring(0, 8);
-      final fileName = '${safeWorkspaceName}_$uniqueId'; // Service adds extension
+      final fileName = '${safeWorkspaceName}_$uniqueId';
 
-      // 5. Use the FileServices to trigger the download
       final result = await _fileServices.exportFile(
         defaultName: fileName,
         jsonString: jsonString,
@@ -167,6 +159,87 @@ class WorkspaceProvider extends StateHandler {
     }
   }
 
+  Future<void> exportWorkspaceAsPng() async {
+    if (_currentWorkspace == null || _canvasObjects.isEmpty) {
+      print("Cannot export: No workspace or content to export.");
+      return;
+    }
+    try {
+      Rect? contentBounds;
+      for (final object in _canvasObjects.values) {
+        if (contentBounds == null) {
+          contentBounds = object.getBounds();
+        } else {
+          contentBounds = contentBounds.expandToInclude(object.getBounds());
+        }
+      }
+
+      if (contentBounds == null) {
+        print("No objects on canvas to export.");
+        return;
+      }
+
+      const double padding = 50.0;
+      final imageBounds = Rect.fromLTRB(
+        contentBounds.left - padding,
+        contentBounds.top - padding,
+        contentBounds.right + padding,
+        contentBounds.bottom + padding,
+      );
+
+      final recorder = ui.PictureRecorder();
+      final canvas = Canvas(recorder);
+
+      canvas.translate(-imageBounds.left, -imageBounds.top);
+
+      final backgroundPaint = Paint()..color = _currentWorkspaceColor;
+      canvas.drawRect(
+          Rect.fromLTWH(0, 0, imageBounds.width, imageBounds.height),
+          backgroundPaint);
+
+      final painter = CanvasPainter(
+        canvasObjects: _canvasObjects,
+        userCursors: {},
+        currentlySelectedObjectId: null,
+        interactionMode: InteractionMode.none,
+        handleRadius: 0,
+        connectionPointRadius: _connectionPointRadius,
+        connectorDragPosition: null,
+        connectorSourceId: null,
+        connectorSourceAlignment: null,
+      );
+      painter.paint(canvas, imageBounds.size);
+
+      final picture = recorder.endRecording();
+      final img = await picture.toImage(
+        imageBounds.width.toInt(),
+        imageBounds.height.toInt(),
+      );
+      final byteData = await img.toByteData(format: ui.ImageByteFormat.png);
+      final pngBytes = byteData!.buffer.asUint8List();
+
+      final safeWorkspaceName = _currentWorkspace!.name
+          .replaceAll(RegExp(r'[^\w\s-]'), '')
+          .replaceAll(' ', '_');
+      final uniqueId = const Uuid().v4().substring(0, 8);
+      final fileName = '${safeWorkspaceName}_$uniqueId';
+
+      final result = await _fileServices.exportPNG(
+        defaultName: fileName,
+        pngBytes: pngBytes,
+      );
+
+      if (result == 'success') {
+        print("Workspace exported successfully as $fileName.png");
+      } else {
+        print("PNG export failed or was cancelled: $result");
+      }
+    } catch (e) {
+      print("An error occurred during PNG export: $e");
+    }
+  }
+
+
   void setStickyNoteMode(Color color) {
     _currentMode = DrawMode.stickyNote;
     _nextObjectColor = color;
@@ -279,7 +352,7 @@ class WorkspaceProvider extends StateHandler {
                     }
                   }
                 }
-                
+
                 if (payload['deleted_ids'] != null) {
                   final List<String> deletedIds = List<String>.from(
                     payload['deleted_ids'],
@@ -1000,4 +1073,5 @@ class WorkspaceProvider extends StateHandler {
       return a == b;
     }
   }
-}
+}
+

lib/features/workspace/widgets/export_dialog.dart

@@ -1,6 +1,5 @@
-// lib/features/workspace/widgets/export_dialog.dart (Fully Modified)
-
 import 'package:cookethflow/core/providers/supabase_provider.dart';
+import 'package:cookethflow/core/theme/colors.dart';
 import 'package:cookethflow/features/workspace/providers/workspace_provider.dart';
 import 'package:flutter/material.dart';
 import 'package:flutter_screenutil/flutter_screenutil.dart';
@@ -9,13 +8,9 @@ import 'package:cookethflow/core/helpers/responsive_layout.helper.dart' as rh;
 
 class ExportDialog extends StatefulWidget {
   final SupabaseService su;
-  final WorkspaceProvider wp; // MODIFIED: Added WorkspaceProvider
+  final WorkspaceProvider wp;
 
-  const ExportDialog({
-    super.key, 
-    required this.su, 
-    required this.wp, // MODIFIED: Added to constructor
-  });
+  const ExportDialog({super.key, required this.su, required this.wp});
 
   @override
   State<ExportDialog> createState() => _ExportDialogState();
@@ -36,7 +31,9 @@ class _ExportDialogState extends State<ExportDialog> {
         padding: EdgeInsets.all(24.r),
         decoration: BoxDecoration(
           color:
-              widget.su.isDark ? Color.fromRGBO(48, 48, 48, 1) : Colors.white,
+              widget.su.isDark
+                  ? const Color.fromRGBO(48, 48, 48, 1)
+                  : Colors.white,
           borderRadius: BorderRadius.circular(16.r),
           boxShadow: [
             BoxShadow(
@@ -87,7 +84,6 @@ class _ExportDialogState extends State<ExportDialog> {
               ],
             ),
             SizedBox(height: 24.h),
-
             Row(
               mainAxisAlignment: MainAxisAlignment.spaceBetween,
               children: [
@@ -154,20 +150,22 @@ class _ExportDialogState extends State<ExportDialog> {
               ],
             ),
             SizedBox(height: 32.h),
-
             SizedBox(
               width: double.infinity,
               child: ElevatedButton(
                 onPressed: () {
-                  // MODIFIED: Implement export logic
                   if (_selectedFormat == 'JSON') {
                     widget.wp.exportWorkspaceAsJson();
                   } else if (_selectedFormat == 'PNG') {
-                    // TODO: Implement PNG export logic
-                    print('Exporting as PNG is not implemented yet.');
+                    widget.wp.exportWorkspaceAsPng();
                   } else if (_selectedFormat == 'SVG') {
-                    // TODO: Implement SVG export logic
                     print('Exporting as SVG is not implemented yet.');
+                    ScaffoldMessenger.of(context).showSnackBar(
+                      SnackBar(
+                        content: Text('Coming soon!'),
+                        backgroundColor: primaryColor,
+                      ),
+                    );
                   }
                   Navigator.of(context).pop();
                 },
@@ -199,4 +197,4 @@ class _ExportDialogState extends State<ExportDialog> {
       ),
     );
   }
-}
+}