v0.0.2

integration_test/ocr_provider_test.dart

@@ -0,0 +1,43 @@
+import 'package:flutter_test/flutter_test.dart';
+import 'package:snapfig/shared/services/ocr_core/ocr_core.dart';
+import 'package:flutter/services.dart' show rootBundle;
+import 'package:path/path.dart' as path_lib;
+import 'dart:io';
+
+/// assetPath: 예) 'test/unit/assets/test.pdf'
+/// 반환: 내부 저장소에 복사된 파일의 경로
+Future<String> copyAssetToLocal(String assetPath) async {
+  final byteData = await rootBundle.load(assetPath);
+  final tempDir = await Directory.systemTemp.createTemp('test_pdf_');
+  final fileName = path_lib.basename(assetPath);
+  final file = File(path_lib.join(tempDir.path, fileName));
+  await file.writeAsBytes(byteData.buffer.asUint8List());
+  return file.path;
+}
+
+void main() {
+  late OCRProvider provider;
+
+  setUp(() {
+    provider = OCRProviderImpl(
+      baseUrl: 'https://901b-39-115-116-188.ngrok-free.app',
+    );
+  });
+
+  test('process()는 OCRResult를 반환해야 한다', () async {
+    final pdfPath = await copyAssetToLocal('integration_test/assets/test.pdf');
+    final result = await provider.process(pdfPath);
+    expect(result, isA<OCRResult>());
+    expect(result.title, isNotEmpty);
+    expect(result.chapters, isNotEmpty);
+    expect(result.pages, isNotEmpty);
+    expect(result.metadata.totalPages, isNotEmpty);
+    expect(result.metadata.processingTime, isNotEmpty);
+    expect(result.metadata.totalFigures, isNotEmpty);
+  }, timeout: const Timeout(Duration(minutes: 15)));
+
+  test('process()에 잘못된 경로를 넣어도 예외가 발생하지 않는다', () async {
+    final result = await provider.process('invalid/path.pdf');
+    expect(result, isA<OCRResult>());
+  });
+}

integration_test/pdf_provider_test.dart

@@ -20,7 +20,13 @@ Future<String> copyAssetToLocal(String assetPath) async {
 class DummyOcrService implements OCRProvider {
   @override
   Future<OCRResult> process(String pdfPath) async {
-    return OCRResult();
+    return OCRResult(
+      title: 'test',
+      chapters: [],
+      pages: [],
+      figures: [],
+      metadata: Metadata(totalPages: 0, processingTime: 0, totalFigures: 0),
+    );
   }
 }
 

lib/features/pdf_viewer/models/pdf_viewer_view_model.dart

@@ -0,0 +1,231 @@
+import 'package:flutter/foundation.dart';
+import 'package:file_picker/file_picker.dart';
+
+import '../../../shared/services/pdf_core/pdf_core.dart';
+
+class PdfViewerViewModel extends ChangeNotifier {
+  // Private state
+  BasePdf? _pdfModel;
+  List<BasePage> _pages = [];
+  List<BaseLayout> _layouts = [];
+  Map<int, List<BaseLayout>> _layoutsByPage = {};
+
+  // UI state
+  double _zoomPercent = 100.0;
+  int _currentPage = 1;
+  String _pdfTitle = 'PDF Viewer';
+  String _searchQuery = '';
+  bool _sidebarVisible = true;
+  int _selectedTabIndex = 0;
+  bool _isLoading = false;
+  String? _errorMessage;
+
+  // Dependencies
+  final PDFProvider _pdfProvider;
+
+  PdfViewerViewModel({required PDFProvider pdfProvider})
+    : _pdfProvider = pdfProvider;
+
+  BasePdf? get pdfModel => _pdfModel;
+  List<BasePage> get pages => _pages;
+  List<BaseLayout> get layouts => _layouts;
+
+  double get zoomPercent => _zoomPercent;
+  int get currentPage => _currentPage;
+  String get pdfTitle => _pdfTitle;
+  String get searchQuery => _searchQuery;
+  bool get sidebarVisible => _sidebarVisible;
+  int get selectedTabIndex => _selectedTabIndex;
+  bool get isLoading => _isLoading;
+  String? get errorMessage => _errorMessage;
+
+  // Filtered data
+  List<BasePage> get filteredPages =>
+      _pages
+          .where((p) => p.pageIndex.toString().contains(_searchQuery))
+          .toList();
+
+  List<BaseLayout> get filteredLayouts =>
+      _layouts
+          .where((l) => l.type == LayoutType.figure)
+          .where(
+            (l) =>
+                l.figureId?.toLowerCase().contains(
+                  _searchQuery.toLowerCase(),
+                ) ??
+                false,
+          )
+          .toList();
+
+  // Get layouts for a specific page
+  List<BaseLayout> getLayoutsForPage(int pageIndex) {
+    return _layoutsByPage[pageIndex] ?? [];
+  }
+
+  // Get figure references for a specific page
+  List<BaseLayout> getFigureReferencesForPage(int pageIndex) {
+    return getLayoutsForPage(
+      pageIndex,
+    ).where((layout) => layout.type == LayoutType.figureReference).toList();
+  }
+
+  // Get all figure references
+  List<BaseLayout> get figureReferences =>
+      _layouts
+          .where((layout) => layout.type == LayoutType.figureReference)
+          .toList();
+
+  // Get all figures
+  List<BaseLayout> get figures =>
+      _layouts.where((layout) => layout.type == LayoutType.figure).toList();
+
+  // Find figure by ID
+  BaseLayout? findFigureById(String figureId) {
+    try {
+      return figures.firstWhere((figure) => figure.figureId == figureId);
+    } catch (e) {
+      return null;
+    }
+  }
+
+  // Find figure by referenced figure ID
+  BaseLayout? findFigureByReference(BaseLayout reference) {
+    if (reference.type != LayoutType.figureReference) return null;
+    final referencedId = reference.referencedFigureId;
+    if (referencedId == null) return null;
+    return findFigureById(referencedId);
+  }
+
+  // Get page containing a specific layout
+  BasePage? getPageForLayout(BaseLayout layout) {
+    for (final entry in _layoutsByPage.entries) {
+      if (entry.value.contains(layout)) {
+        return _pages.firstWhere((page) => page.pageIndex == entry.key);
+      }
+    }
+    return null;
+  }
+
+  // Navigate to a specific figure by finding its page
+  int? getPageNumberForFigure(BaseLayout figure) {
+    if (figure.type != LayoutType.figure) return null;
+
+    final page = getPageForLayout(figure);
+    if (page != null) {
+      return page.pageIndex +
+          1; // Convert to 1-based page number for PDF controller
+    }
+    return null;
+  }
+
+  // Methods
+  Future<void> loadPdf({required String path, required bool isAsset}) async {
+    try {
+      _setLoading(true);
+      _clearError();
+
+      // 2. Get PDF from provider by path
+      final model = await _pdfProvider.getPDF(path);
+
+      if (model == null) {
+        throw StateError('PDF not found in provider: $path');
+      }
+
+      // 3. Load pages and layouts
+      final pages = await model.getPages();
+      final allLayouts = <BaseLayout>[];
+      final layoutsByPage = <int, List<BaseLayout>>{};
+
+      for (final page in pages) {
+        final layouts = await page.getLayouts();
+        allLayouts.addAll(layouts);
+        layoutsByPage[page.pageIndex] = layouts;
+      }
+
+      // 4. Update state
+      _pdfModel = model;
+      _pages = pages;
+      _layouts = allLayouts;
+      _layoutsByPage = layoutsByPage;
+      _currentPage = 1;
+      _zoomPercent = 100.0;
+      _pdfTitle = model.name;
+
+      notifyListeners();
+    } catch (e) {
+      _setError('Failed to load PDF: $e');
+    } finally {
+      _setLoading(false);
+    }
+  }
+
+  Future<void> pickPdfFromDevice() async {
+    try {
+      final result = await FilePicker.platform.pickFiles(
+        type: FileType.custom,
+        allowedExtensions: ['pdf'],
+      );
+
+      if (result != null && result.files.single.path != null) {
+        await loadPdf(path: result.files.single.path!, isAsset: false);
+      }
+    } catch (e) {
+      _setError('Failed to pick PDF: $e');
+    }
+  }
+
+  void updateZoom(double scale) {
+    _zoomPercent = (scale * 100).roundToDouble();
+    notifyListeners();
+  }
+
+  void updateCurrentPage(int pageIndex) {
+    _currentPage = pageIndex;
+    notifyListeners();
+  }
+
+  void updateSearchQuery(String query) {
+    _searchQuery = query;
+    notifyListeners();
+  }
+
+  void toggleSidebar() {
+    _sidebarVisible = !_sidebarVisible;
+    notifyListeners();
+  }
+
+  void selectTab(int index) {
+    _selectedTabIndex = index;
+    notifyListeners();
+  }
+
+  void clearSearch() {
+    _searchQuery = '';
+    notifyListeners();
+  }
+
+  void clearError() {
+    _clearError();
+  }
+
+  // Private helpers
+  void _setLoading(bool loading) {
+    _isLoading = loading;
+    notifyListeners();
+  }
+
+  void _setError(String error) {
+    _errorMessage = error;
+    notifyListeners();
+  }
+
+  void _clearError() {
+    _errorMessage = null;
+    notifyListeners();
+  }
+
+  @override
+  void dispose() {
+    super.dispose();
+  }
+}