Working on FSE/PPC FileSystem Provider/Service.

1 files changed, 202 insertions, 0 deletions

diff --git a/Software/Visual_Studio/PPC/Tango.PPC.Common/FileSystem/DefaultFileSystemService.cs b/Software/Visual_Studio/PPC/Tango.PPC.Common/FileSystem/DefaultFileSystemService.cs
new file mode 100644
index 000000000..ffbba2e7a
--- /dev/null
+++ b/Software/Visual_Studio/PPC/Tango.PPC.Common/FileSystem/DefaultFileSystemService.cs
@@ -0,0 +1,202 @@
+using System;
+using System.Collections.Generic;
+using System.IO;
+using System.IO.Compression;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+using Tango.Core;
+using Tango.Core.DI;
+using Tango.FileSystem;
+using Tango.FileSystem.Network;
+using Tango.Integration.ExternalBridge;
+using Tango.PPC.Common.ExternalBridge;
+using Tango.Transport;
+
+namespace Tango.PPC.Common.FileSystem
+{
+    [TangoCreateWhenRegistered]
+    public class DefaultFileSystemService : ExtendedObject, IFileSystemService, IExternalBridgeRequestHandler
+    {
+        private enum FileSystemOperationMode
+        {
+            Upload,
+            Download
+        }
+
+        private class FileSystemOperation
+        {
+            public FileSystemOperationMode Mode { get; set; }
+            public String Id { get; set; }
+            public String Path { get; set; }
+
+            public FileSystemOperation(FileSystemOperationMode mode, String path)
+            {
+                Mode = mode;
+                Id = Guid.NewGuid().ToString();
+                Path = path;
+            }
+        }
+
+        private FileSystemManager _manager;
+        private Dictionary<String, FileSystemOperation> _operations;
+
+        public bool Enabled { get; set; } = true;
+
+        public DefaultFileSystemService(IPPCExternalBridgeService externalBridge)
+        {
+            _manager = new FileSystemManager();
+            _operations = new Dictionary<string, FileSystemOperation>();
+            externalBridge.RegisterRequestHandler(this);
+        }
+
+        [ExternalBridgeRequestHandlerMethod(typeof(GetFileSystemItemRequest))]
+        public async void OnGetFileSystemItemRequest(GetFileSystemItemRequest request, String token, ExternalBridgeReceiver receiver)
+        {
+            try
+            {
+                FileSystemItemDTO dto = _manager.GetFolder(request);
+                await receiver.SendGenericResponse(new GetFileSystemItemResponse() { FileSystemItem = dto }, token);
+            }
+            catch (Exception ex)
+            {
+                await receiver.SendErrorResponse(ex, token);
+            }
+        }
+
+        [ExternalBridgeRequestHandlerMethod(typeof(FileUploadRequest))]
+        public async void OnFileUploadRequest(FileUploadRequest request, String token, ExternalBridgeReceiver receiver)
+        {
+            try
+            {
+                using (var stream = new FileStream(request.Path, FileMode.Create)) { }
+
+                FileSystemOperation operation = new FileSystemOperation(FileSystemOperationMode.Upload, request.Path);
+                _operations.Add(operation.Id, operation);
+
+                await receiver.SendGenericResponse(new FileUploadResponse() { OperationId = operation.Id }, token);
+            }
+            catch (Exception ex)
+            {
+                await receiver.SendErrorResponse(ex, token);
+            }
+        }
+
+        [ExternalBridgeRequestHandlerMethod(typeof(FileDownloadRequest))]
+        public async void OnFileDownloadRequest(FileDownloadRequest request, String token, ExternalBridgeReceiver receiver)
+        {
+            try
+            {
+                if (!File.Exists(request.Path))
+                {
+                    await receiver.SendErrorResponse(new FileNotFoundException("Could not find the specified file."), token);
+                    return;
+                }
+
+                FileSystemOperation operation = new FileSystemOperation(FileSystemOperationMode.Download, request.Path);
+
+                _operations.Add(operation.Id, operation);
+
+                await receiver.SendGenericResponse(new FileDownloadResponse()
+                {
+                    OperationId = operation.Id,
+                    Length = new FileInfo(request.Path).Length
+                }, token);
+            }
+            catch (Exception ex)
+            {
+                await receiver.SendErrorResponse(ex, token);
+            }
+        }
+
+        [ExternalBridgeRequestHandlerMethod(typeof(ChunkUploadRequest))]
+        public async void OnChunkUploadRequest(ChunkUploadRequest request, String token, ExternalBridgeReceiver receiver)
+        {
+            try
+            {
+                FileSystemOperation operation;
+                _operations.TryGetValue(request.OperationId, out operation);
+
+                if (operation == null)
+                {
+                    await receiver.SendErrorResponse(new ArgumentException("Invalid operation id."), token);
+                    return;
+                }
+
+                using (var stream = new FileStream(operation.Path, FileMode.Append))
+                {
+                    stream.Write(request.Data, 0, request.Data.Length);
+                }
+
+                await receiver.SendGenericResponse(new ChunkUploadResponse(), token);
+            }
+            catch (Exception ex)
+            {
+                await receiver.SendErrorResponse(ex, token);
+            }
+        }
+
+        [ExternalBridgeRequestHandlerMethod(typeof(ChunkDownloadRequest))]
+        public async void OnChunkDownloadRequest(ChunkDownloadRequest request, String token, ExternalBridgeReceiver receiver)
+        {
+            FileSystemOperation operation;
+            _operations.TryGetValue(request.OperationId, out operation);
+
+            if (operation == null)
+            {
+                await receiver.SendErrorResponse(new ArgumentException("Invalid operation id."), token);
+                return;
+            }
+
+            using (FileStream stream = new FileStream(operation.Path, FileMode.Open))
+            {
+                stream.Position = request.Position;
+                byte[] data = new byte[Math.Min(request.MaxChunkSize, stream.Length - stream.Position)];
+                await stream.ReadAsync(data, 0, data.Length);
+                await receiver.SendGenericResponse(new ChunkDownloadResponse()
+                {
+                    Data = data
+                }, token);
+            }
+        }
+
+        [ExternalBridgeRequestHandlerMethod(typeof(AbortOperationRequest))]
+        public async void OnAbortOperationRequest(AbortOperationRequest request, String token, ExternalBridgeReceiver receiver)
+        {
+            FileSystemOperation operation;
+            _operations.TryGetValue(request.OperationId, out operation);
+
+            if (operation == null)
+            {
+                await receiver.SendErrorResponse(new ArgumentException("Invalid operation id."), token);
+                return;
+            }
+
+            try
+            {
+                if (operation.Mode == FileSystemOperationMode.Upload)
+                {
+                    if (File.Exists(operation.Path))
+                    {
+                        File.Delete(operation.Path);
+                    }
+                    else if (Directory.Exists(operation.Path))
+                    {
+                        Directory.Delete(operation.Path, true);
+                    }
+                }
+
+                await receiver.SendGenericResponse(new AbortOperationResponse(), token);
+            }
+            catch (Exception ex)
+            {
+                await receiver.SendErrorResponse(ex, token);
+            }
+        }
+
+        public void OnReceiverDisconnected(ExternalBridgeReceiver receiver)
+        {
+
+        }
+    }
+}