Revize 830652de

Export in BE

1. Conflicts:
2. Backend/Core/Services/AnnotationService/AnnotationServiceEF.cs
3. Backend/Core/Services/AnnotationService/IAnnotationService.cs

Backend/Core/Services/AnnotationService/AnnotationServiceEF.cs

@@ -16 +16 @@
 using System.Text.RegularExpressions;
 using Newtonsoft.Json;
 using Core.GraphUtils;
+using Models.Documents;
 
 namespace Core.Services.AnnotationService
 {
@@ -775 +776 @@
 
             return true;
         }
-
+        
         public void SetTagIsFinal(Guid annotationId, Guid occurenceId, bool isFinal)
         {
             Annotation annotation;
@@ -802 +803 @@
 
             context.SaveChanges();
         }
+
+        public MemoryStream Export(ExportRequest request)
+        {
+            if (request.ExportAllDone && request.DocumentIds == null)
+            {
+                throw new InvalidOperationException("No documents specified");
+            }
+
+            // Get documents
+            IEnumerable<Document> documentsToExport = null;
+            if (request.ExportAllDone)
+            {
+                documentsToExport = context.Documents
+                    .Include(d => d.Content);
+            }
+            else
+            {
+                documentsToExport = context.Documents
+                    .Include(d => d.Content)
+                    .Where(d => request.DocumentIds.Contains(d.Id));
+            }
+
+            // Get annotations
+            Dictionary<Document, List<Annotation>> annotationsToExport = new();
+            foreach (var document in documentsToExport)
+            {
+                annotationsToExport[document] = context.Annotations
+                    .Where(a => a.Document == document && a.State == EState.DONE)
+                    .Include(a => a.Document)
+                    .Include(a => a.User)
+                    .ToList();
+            }
+
+            // Get final annotations
+            Dictionary<Document, FinalAnnotation> finalAnnotationsToExport = new();
+            foreach (var document in documentsToExport)
+            {
+                if (!context.FinalAnnotations.Any(fa => fa.Document == document))
+                {
+                    finalAnnotationsToExport[document] = null;
+                }
+                else
+                {
+                    finalAnnotationsToExport[document] = context.FinalAnnotations
+                        .Single(fa => fa.Document == document);
+                }
+            }
+
+            // Get tags for each annotation
+            Dictionary<Annotation, List<AnnotationTag>> tagsToExport = new();
+            foreach (var document in annotationsToExport.Keys)
+            {
+                foreach (var annotation in annotationsToExport[document])
+                {
+                    tagsToExport[annotation] = context.AnnotationTags
+                        .Include(at => at.Annotation)
+                        .Include(at => at.Tag).ThenInclude(t => t.Category)
+                        .Include(at => at.SubTag)
+                        .Where(at => at.Annotation == annotation)
+                        .ToList();
+                }
+            }
+
+            // Get tags for each final annotation
+            Dictionary<FinalAnnotation, List<FinalAnnotationTag>> finalTagsToExport = new();
+            foreach (var document in finalAnnotationsToExport.Keys)
+            {
+                var finalAnnotation = finalAnnotationsToExport[document];
+                finalTagsToExport[finalAnnotation] = context.FinalAnnotationTags
+                    .Include(at => at.Annotation)
+                    .Include(at => at.Tag).ThenInclude(t => t.Category)
+                    .Include(at => at.SubTag)
+                    .Where(at => at.Annotation == finalAnnotation)
+                    .ToList();
+            }
+
+            MemoryStream output = ZipUtils.Export.FullExport(documentsToExport.ToList(), annotationsToExport, finalAnnotationsToExport, tagsToExport, finalTagsToExport);
+            return output;
+        }
     }
 }