musadaq-saas/musadaq-app/lib/features/scanner/controllers/scanner_controller.dart

import 'dart:io';
import 'package:flutter/material.dart';
import 'package:get/get.dart';
import 'package:firebase_messaging/firebase_messaging.dart';
import 'package:path_provider/path_provider.dart';
import 'package:path/path.dart' as path;
import '../../../core/utils/logger.dart';
import '../../../core/utils/app_snackbar.dart';
import '../../../core/services/image_processing_service.dart';
import '../../../core/services/invoice_upload_service.dart';
import '../../../core/network/dio_client.dart';

class ScannerController extends GetxController {
  var capturedImages = <File>[].obs;
  var isProcessing = false.obs;
  var uploadProgress = 0.0.obs;
  var companies = <Map<String, dynamic>>[].obs;
  var isLoadingCompanies = false.obs;
  var currentBatchId = ''.obs;
  var processedImagesCount = 0.obs;
  var totalImagesCount = 0.obs;
  var isBatchDone = false.obs;

  final InvoiceUploadService _uploadService = InvoiceUploadService();

  @override
  void onInit() {
    super.onInit();
    fetchCompanies();
    _initFcmListener();
  }

  void _initFcmListener() {
    FirebaseMessaging.onMessage.listen((RemoteMessage message) {
      final data = message.data;
      if (data['type'] == 'batch_progress' && data['batch_id'] == currentBatchId.value) {
        processedImagesCount.value = int.tryParse(data['processed'].toString()) ?? 0;
        totalImagesCount.value = int.tryParse(data['total'].toString()) ?? 0;
        
        if (processedImagesCount.value >= totalImagesCount.value) {
           isBatchDone.value = true;
        }
      }
    });
  }

  Future<void> fetchCompanies() async {
    isLoadingCompanies.value = true;
    try {
      final res = await DioClient().client.get('companies');
      if (res.data['success'] == true && res.data['data'] != null) {
        companies.value = List<Map<String, dynamic>>.from(res.data['data']);
      }
    } catch (e) {
      AppLogger.error('Failed to fetch companies', e);
    } finally {
      isLoadingCompanies.value = false;
    }
  }

  Future<void> addImage(String imagePath) async {
    File originalFile = File(imagePath);
    // Add to UI immediately so the user doesn't wait
    capturedImages.add(originalFile);
    int index = capturedImages.length - 1;

    // Process in background without showing full-screen loader
    ImageProcessingService.processInvoiceImage(originalFile).then((processedFile) {
      if (processedFile != null && index < capturedImages.length) {
        capturedImages[index] = processedFile;
        AppLogger.print('Finished processing image in background. Replaced in batch.');
      }
    }).catchError((e) {
        AppLogger.error('Failed to process image in background', e);
    });
  }

  void removeImage(int index) {
    if (index >= 0 && index < capturedImages.length) {
      capturedImages.removeAt(index);
    }
  }

  Future<void> uploadBatch(String fallbackCompanyId) async {
    if (capturedImages.isEmpty) {
      AppSnackbar.showWarning('تنبيه', 'الرجاء تصوير فاتورة واحدة على الأقل');
      return;
    }

    try {
      isProcessing.value = true;
      uploadProgress.value = 0.0;
      
      // Fetch a valid company ID dynamically to prevent 403 Forbidden
      String companyId = fallbackCompanyId;
      if (companyId == 'mock_company_id_123' || companyId.isEmpty) {
        final res = await DioClient().client.get('companies');
        if (res.data['success'] == true && res.data['data'] != null && res.data['data'].isNotEmpty) {
           companyId = res.data['data'][0]['id'];
           AppLogger.print('Dynamically fetched company: $companyId');
        } else {
           AppSnackbar.showError('خطأ', 'لا توجد شركات مسجلة في حسابك');
           isProcessing.value = false;
           return;
        }
      }

      AppLogger.print('Uploading batch of ${capturedImages.length} images to company $companyId...');
      
      final batchId = await _uploadService.uploadBatch(
        companyId: companyId,
        images: capturedImages,
        onProgress: (current, total) {
          uploadProgress.value = current / total;
        },
      );
      
      if (batchId != null) {
        currentBatchId.value = batchId;
        totalImagesCount.value = capturedImages.length;
        processedImagesCount.value = 0;
        capturedImages.clear();
        uploadProgress.value = 0.0;
        
        Get.dialog(
          AlertDialog(
            shape: RoundedRectangleBorder(borderRadius: BorderRadius.circular(12)),
            title: const Center(
              child: Text('جاري المعالجة ⏳', 
                style: TextStyle(fontFamily: 'El Messiri', fontWeight: FontWeight.bold, fontSize: 18)
              )
            ),
            content: Obx(() => Column(
              mainAxisSize: MainAxisSize.min,
              children: [
                const Text('يتم الآن تدقيق الفواتير عبر الذكاء الاصطناعي...',
                  textAlign: TextAlign.center,
                  style: TextStyle(fontFamily: 'El Messiri', fontSize: 14)
                ),
                const SizedBox(height: 20),
                if (!isBatchDone.value) ...[
                  LinearProgressIndicator(
                    value: totalImagesCount.value > 0 ? processedImagesCount.value / totalImagesCount.value : 0,
                    backgroundColor: Colors.grey[200],
                    valueColor: const AlwaysStoppedAnimation<Color>(Color(0xFF0F4C81)),
                  ),
                  const SizedBox(height: 10),
                  Text('${processedImagesCount.value} من ${totalImagesCount.value}',
                    style: const TextStyle(fontFamily: 'El Messiri', fontWeight: FontWeight.bold)
                  ),
                ] else ...[
                  const Icon(Icons.check_circle, color: Colors.green, size: 50),
                  const SizedBox(height: 10),
                  const Text('اكتملت المعالجة بنجاح!',
                    style: TextStyle(fontFamily: 'El Messiri', color: Colors.green, fontWeight: FontWeight.bold)
                  ),
                ],
              ],
            )),
            actions: [
              TextButton(
                onPressed: () {
                  Get.back(); // close dialog
                  Get.back(); // go back to dashboard
                },
                child: const Text('إغلاق', style: TextStyle(fontFamily: 'El Messiri', fontWeight: FontWeight.bold)),
              )
            ],
          ),
          barrierDismissible: false,
        );
      } else {
        AppSnackbar.showError('خطأ', 'فشل رفع الفواتير، يرجى المحاولة لاحقاً');
      }
    } catch (e) {
      AppLogger.error('Failed to upload batch', e);
      AppSnackbar.showError('خطأ', 'حدث خطأ غير متوقع أثناء الرفع');
    } finally {
      isProcessing.value = false;
    }
  }

  Future<String> getSavePath() async {
    final directory = await getTemporaryDirectory();
    final fileName = 'invoice_${DateTime.now().millisecondsSinceEpoch}.jpg';
    return path.join(directory.path, fileName);
  }
}