prs/prs.desktop/Panels/DataEntry/DataEntryList.xaml.cs

using Comal.Classes;
using InABox.Clients;
using InABox.Core;
using Syncfusion.Pdf.Graphics;
using Syncfusion.Pdf.Parsing;
using Syncfusion.Pdf;
using System;
using System.Collections.Generic;
using System.Drawing.Imaging;
using System.Drawing;
using System.IO;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using System.Windows;
using System.Windows.Media.Imaging;
using InABox.DynamicGrid;
using InABox.WPF;
using Encoder = System.Drawing.Imaging.Encoder;
using Path = System.IO.Path;
using Image = System.Windows.Controls.Image;
using System.ComponentModel;
using System.Windows.Controls;
using System.Windows.Forms;
using Clipboard = System.Windows.Clipboard;
using DataFormats = System.Windows.DataFormats;
using DragDropEffects = System.Windows.DragDropEffects;
using DragEventArgs = System.Windows.DragEventArgs;
using MessageBox = System.Windows.MessageBox;
using UserControl = System.Windows.Controls.UserControl;

namespace PRSDesktop
{
    public static class PDFExtensions
    {
        public static IEnumerable<PdfPageBase> GetPages(this PdfDocumentBase doc)
        {
            if (doc is PdfLoadedDocument lDoc)
                return lDoc.Pages.Cast<PdfPageBase>();
            if (doc is PdfDocument pdfDoc)
                return pdfDoc.Pages.Cast<PdfPageBase>();
            throw new Exception($"Unsupported PDF Document type {doc.GetType()}");
        }
        public static PdfPageBase GetPage(this PdfDocumentBase doc, int index)
        {
            if (doc is PdfLoadedDocument lDoc)
                return lDoc.Pages[index];
            if (doc is PdfDocument pdfDoc)
                return pdfDoc.Pages[index];
            throw new Exception($"Unsupported PDF Document type {doc.GetType()}");
        }
        public static int PageCount(this PdfDocumentBase doc)
        {
            if (doc is PdfLoadedDocument lDoc)
                return lDoc.Pages.Count;
            if (doc is PdfDocument pdfDoc)
                return pdfDoc.Pages.Count;
            throw new Exception($"Unsupported PDF Document type {doc.GetType()}");
        }
        public static PdfLoadedDocument AsLoadedDocument(this PdfDocumentBase doc)
        {
            if (doc is PdfLoadedDocument lDoc)
                return lDoc;
            if (doc is PdfDocument pdfDoc)
            {
                using var ms = new MemoryStream();
                pdfDoc.Save(ms);
                var array = ms.ToArray();
                return new PdfLoadedDocument(array);
            }
            throw new Exception($"Unsupported PDF Document type {doc.GetType()}");
        }

        public static byte[] SaveToBytes(this PdfDocumentBase doc)
        {
            using var ms = new MemoryStream();
            doc.Save(ms);
            return ms.ToArray();
        }
    }

    /// <summary>
    /// Interaction logic for ScanPanel.xaml
    /// </summary>
    public partial class DataEntryList : UserControl, ICorePanel, IDockPanel
    {
        private List<DataEntryDocument> SelectedScans = new();

        public delegate void DateEntrySelectionHandler(String appliesTo, Guid entityID, bool allowprocess);

        public event DateEntrySelectionHandler? SelectionChanged;
        
        public DataEntryList()
        {
            InitializeComponent();
        }

        #region Panel

        public void Setup()
        {
            _dataEntryGrid.HiddenColumns.Add(x => x.Document.ID);
            _dataEntryGrid.Refresh(true, false);
            _historyGrid.Refresh(true, false);
        }

        public void Refresh()
        {
            if (_pages.SelectedIndex == 0)
                _dataEntryGrid.Refresh(false, true);
            else if (_pages.SelectedIndex == 1)
                _historyGrid.Refresh(false,true);
                
        }

        public void Shutdown(CancelEventArgs? cancel)
        {
        }

        #endregion

        #region View List

        private static List<byte[]> RenderTextFile(string textData)
        {
            var pdfDocument = new PdfDocument();
            var page = pdfDocument.Pages.Add();

            var font = new PdfStandardFont(PdfFontFamily.Courier, 14);
            var textElement = new PdfTextElement(textData, font);
            var layoutFormat = new PdfLayoutFormat
            {
                Layout = PdfLayoutType.Paginate,
                Break = PdfLayoutBreakType.FitPage
            };

            textElement.Draw(page, new RectangleF(0, 0, page.GetClientSize().Width, page.GetClientSize().Height), layoutFormat);

            using var docStream = new MemoryStream();
            pdfDocument.Save(docStream);

            var loadeddoc = new PdfLoadedDocument(docStream.ToArray());
            Bitmap[] bmpImages = loadeddoc.ExportAsImage(0, loadeddoc.Pages.Count - 1);

            var jpgEncoder = ImageUtils.GetEncoder(ImageFormat.Jpeg)!;
            var quality = Encoder.Quality;
            var encodeParams = new EncoderParameters(1);
            encodeParams.Param[0] = new EncoderParameter(quality, 100L);

            var images = new List<byte[]>();
            if (bmpImages != null)
                foreach (var image in bmpImages)
                {
                    using var data = new MemoryStream();
                    image.Save(data, jpgEncoder, encodeParams);
                    images.Add(data.ToArray());
                }
            return images;
        }

        private void UpdateViewList()
        {
            var selected = _dataEntryGrid.SelectedRows.Select(x => x.ToObject<DataEntryDocument>()).ToList();
            if (selected.Count == SelectedScans.Count && !selected.Any(x => SelectedScans.All(y => x.ID != y.ID)))
                return;
         
            SelectedScans = selected;
            ViewListPanel.Children.Clear();
            
            var docIDs = SelectedScans.Select(x => x.Document.ID).Distinct().ToArray();
            new Client<Document>()
                .Query(
                    new Filter<Document>(x => x.ID).InList(docIDs),
                    new Columns<Document>(x => x.ID).Add(x => x.Data).Add(x => x.FileName),
                    null,(o,e) => Dispatcher.BeginInvoke(() => LoadDocuments(o)));
        }

        private void LoadDocuments(CoreTable? data)
        {
            var documents = data?.ToObjects<Document>() ?? Array.Empty<Document>();
            var bitmaps = new Dictionary<Guid, List<BitmapImage>>();
            foreach (var document in documents)
            {
                List<byte[]> images;
                var extension = Path.GetExtension(document.FileName).ToLower();
                if (extension == ".pdf")
                {
                    try
                    {
                        images = ImageUtils.RenderPDFToImages(document.Data);
                    }
                    catch (Exception e)
                    {
                        MessageBox.Show($"Cannot load document '{document.FileName}': {e.Message}");
                        images = new List<byte[]>();
                    }
                }
                else if (extension == ".jpg" || extension == ".jpeg" || extension == ".png" || extension == ".bmp")
                {
                    images = new List<byte[]> { document.Data };
                }
                else
                {
                    images = ImageUtils.RenderTextFileToImages(Encoding.UTF8.GetString(document.Data));
                }

                foreach (var imageData in images)
                {
                    try
                    {
                        if (!bitmaps.TryGetValue(document.ID, out var list))
                        {
                            list = new List<BitmapImage>();
                            bitmaps[document.ID] = list;
                        }

                        list.Add(ImageUtils.LoadImage(imageData));
                    }
                    catch (Exception e)
                    {
                        MessageBox.Show($"Cannot load document '{document.FileName}': {e.Message}");
                    }
                }
            }
       
            foreach(var scan in SelectedScans)
            {
                if(bitmaps.TryGetValue(scan.Document.ID, out var list))
                {
                    foreach(var bitmap in list)
                    {
                        var image = new Image
                        {
                            Source = bitmap,
                            Margin = new Thickness(0, 0, 0, 20),
                            ContextMenu = ViewListPanel.ContextMenu
                        };
                        ViewListPanel.Children.Add(image);
                    }
                }
            }
        }

        #endregion

        #region Uploading

        private void DynamicTabItem_Drop(object sender, DragEventArgs e)
        {
            Task.Run(() =>
            {
                var docs = DataEntryReGroupWindow.HandleFileDrop(e);
                if (docs is not null)
                {
                    foreach(var (filename, doc) in docs)
                    {
                        _dataEntryGrid.UploadDocument(Path.ChangeExtension(filename, ".pdf"), doc.SaveToBytes(), Guid.Empty);
                    }
                    /*Dispatcher.Invoke(() =>
                    {
                        ScanGrid.ShowDocumentWindow(pages, filename);
                    });*/
                }
            });
        }

        private void DynamicTabItem_DragOver(object sender, DragEventArgs e)
        {
            if (e.Data.GetDataPresent(DataFormats.FileDrop) || e.Data.GetDataPresent("FileGroupDescriptor"))
            {
                e.Effects = DragDropEffects.Copy;
            }
            else
            {
                e.Effects = DragDropEffects.None;
            }
            e.Handled = true;
        }

        #endregion

        private void _documents_OnSelectItem(object sender, DynamicGridSelectionEventArgs e)
        {
            UpdateViewList();

            DoSelect(e.Rows);
        }

        private void DoSelect(CoreRow[] rows)
        {
            var appliesTo = rows?.Length == 1
                ? rows[0].Get<DataEntryDocument, string>(x => x.Tag.AppliesTo)
                : "";
            var entityid = rows?.Length == 1
                ? rows[0].Get<DataEntryDocument, Guid>(x => x.EntityID)
                : Guid.Empty;
            var archived = rows?.Length == 1
                ? rows[0].Get<DataEntryDocument, DateTime>(x => x.Archived)
                : DateTime.MinValue;
            SelectionChanged?.Invoke(appliesTo, entityid, archived.IsEmpty());
        }

        private void _historyGrid_OnOnSelectItem(object sender, DynamicGridSelectionEventArgs e)
        {
            DoSelect(e.Rows);
        }
        
        private void _Explode_OnClick(object sender, RoutedEventArgs e)
        {
           _dataEntryGrid.DoExplode();
        }

        private List<DataEntryTag>? _tags;

        private void _uploadMenu_OnOpened(object sender, RoutedEventArgs e)
        {
            if (Clipboard.ContainsText())
            {
                _pasteItem.Header = "Paste Text from Clipboard";
                _pasteItem.Visibility = Visibility.Visible;
            }
            else if (Clipboard.ContainsImage())
            {
                _pasteItem.Header = "Paste Image from Clipboard";
                _pasteItem.Visibility = Visibility.Visible;
            }
            else if (Clipboard.ContainsFileDropList())
            {
                int count = CheckAllowableFiles();
                if (count > 0)
                {
                    _pasteItem.Header = $@"Paste {count} File{(count > 1 ? "s" : "")} from Clipboard";
                    _pasteItem.Visibility = Visibility.Visible;
                }
            }
            else
                _pasteItem.Visibility = Visibility.Collapsed;
            
            _archive.Visibility = _dataEntryGrid.SelectedRows.Any() && Security.CanEdit<DataEntryDocument>()
                ? Visibility.Visible
                : Visibility.Collapsed;

            _archiveseparator.Visibility = _archive.Visibility;
            
            _tags ??= DataEntryGrid.GetVisibleTagList();
            _changeTag.Items.Clear();

            foreach (var tag in _tags)
                _changeTag.Items.Add(new MenuItem()
                {
                    Header = tag.Name, 
                    Command = new Command((_) => ChangeTag(tag)) { }
                });
            _changeTag.Items.Add(new Separator());
            _changeTag.Items.Add(new MenuItem()
            {
                Header= "Clear Tags",
                Command = new Command((_) => ChangeTag(new DataEntryTag()))
            });
            
            _changeTag.Visibility = _dataEntryGrid.SelectedRows.Any() && _tags.Any() && (Security.CanEdit<DataEntryDocument>() || Security.IsAllowed<CanSetupDataEntryTags>())
                ? Visibility.Visible
                : Visibility.Collapsed;
            
            _changetagseparator.Visibility = _archive.Visibility;
            
        }

        private void ChangeTag(object obj)
        {
            if (obj is DataEntryTag tag)
                _dataEntryGrid.DoChangeTags(tag.ID);
        }

        private void _addItem_OnClick(object sender, RoutedEventArgs e)
        {
            var ofd = new OpenFileDialog()
            {
                Filter = @"All Files (*.pdf, *.bmp, *.png, *.jpg, *.jpeg)|*.pdf;*.bmp;*.png;*.jpg;*.jpeg",
                Multiselect = true,
                Title = @"Select Files to Upload",
                InitialDirectory = Environment.GetFolderPath(Environment.SpecialFolder.MyDocuments)
            };
            if (ofd.ShowDialog() == DialogResult.OK)
            {
                foreach (var file in ofd.FileNames)
                    Upload(
                        Path.GetFileName(file),
                        new FileStream(file,FileMode.Open));
            }
        }

        private void _pasteItem_OnClick(object sender, RoutedEventArgs e)
        {
            if (Clipboard.ContainsText())
            {
                Upload(
                    $"Pasted Text {DateTime.Now:yyyy-MM-dd hh-mm-ss-fff}.txt",
                    new MemoryStream(new UTF8Encoding().GetBytes(Clipboard.GetText()))
                );
            }
            else if (Clipboard.ContainsImage())
            {
                var img = Clipboard.GetImage();
                if (img != null)
                {
                    var bmp = ImageUtils.BitmapSourceToBitmap(img);
                    using (var ms = new MemoryStream())
                    {
                        bmp.Save(ms, ImageFormat.Png);
                        Upload(
                            $"Pasted Image {DateTime.Now:yyyy-MM-dd hh-mm-ss-fff}.png",
                            ms
                        );
                    }
                }
                
            }
            else if (CheckAllowableFiles() > 0)
            {
                var files = Clipboard.GetFileDropList().OfType<String>().ToArray();
                foreach (var file in files)
                {
                    Upload(
                        Path.GetFileName(file), 
                        new FileStream(file,FileMode.Open)
                    );
                }
            }
        }

        private void Upload(string filename, Stream data)
        {
            var doc = DataEntryReGroupWindow.RenderToPDF(filename, data);
            _dataEntryGrid.UploadDocument(Path.ChangeExtension(filename,"pdf"), doc.SaveToBytes(), Guid.Empty);
        }
        
        private static int CheckAllowableFiles()
        {
            var extensions = Clipboard.GetFileDropList().OfType<String>().Select(x => Path.GetExtension(x.ToUpper())).ToArray();
            return extensions.Count(x => 
                String.Equals(x, "PDF")
                || String.Equals(x, "PNG")
                || String.Equals(x, "JPG")
                || String.Equals(x, "JPEG")
                || String.Equals(x, "BMP")
                || String.Equals(x, "TXT")
            );
        }
        
        private void _pages_OnSelectionChanged(object sender, SelectionChangedEventArgs e)
        {
            if (_pages.SelectedIndex == 0)
                _dataEntryGrid.Refresh(false,true);
            else if (_pages.SelectedIndex == 1)
                _historyGrid.Refresh(false,true);
        }

        private void _remove_OnClick(object sender, RoutedEventArgs e)
        {
            _dataEntryGrid.DoRemove();
        }

        private void _reopen_OnClick(object sender, RoutedEventArgs e)
        {
            _historyGrid.DoReopen();
        }

        private void _dataEntryGrid_OnContextMenuOpening(object sender, ContextMenuEventArgs e)
        {

        }
    }
}