Working on FSE/PPC FileSystem Provider/Service.

3 files changed, 222 insertions, 1 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)
+        {
+
+        }
+    }
+}
diff --git a/Software/Visual_Studio/PPC/Tango.PPC.Common/FileSystem/IFileSystemService.cs b/Software/Visual_Studio/PPC/Tango.PPC.Common/FileSystem/IFileSystemService.cs
new file mode 100644
index 000000000..050bb1cd6
--- /dev/null
+++ b/Software/Visual_Studio/PPC/Tango.PPC.Common/FileSystem/IFileSystemService.cs
@@ -0,0 +1,13 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+
+namespace Tango.PPC.Common.FileSystem
+{
+    public interface IFileSystemService
+    {
+        bool Enabled { get; set; }
+    }
+}
diff --git a/Software/Visual_Studio/PPC/Tango.PPC.Common/Tango.PPC.Common.csproj b/Software/Visual_Studio/PPC/Tango.PPC.Common/Tango.PPC.Common.csproj
index 4551fe427..664e9e228 100644
--- a/Software/Visual_Studio/PPC/Tango.PPC.Common/Tango.PPC.Common.csproj
+++ b/Software/Visual_Studio/PPC/Tango.PPC.Common/Tango.PPC.Common.csproj
@@ -152,6 +152,8 @@
     <Compile Include="ExtensionMethods\ObservableCollectionExtensions.cs" />
     <Compile Include="ExternalBridge\IPPCExternalBridgeService.cs" />
     <Compile Include="ExternalBridge\PPCExternalBridgeService.cs" />
+    <Compile Include="FileSystem\DefaultFileSystemService.cs" />
+    <Compile Include="FileSystem\IFileSystemService.cs" />
     <Compile Include="Helpers\KeyboardHelper.cs" />
     <Compile Include="HotSpot\DefaultHotSpotProvider.cs" />
     <Compile Include="HotSpot\IHotSpotProvider.cs" />
@@ -368,6 +370,10 @@
       <Project>{4399AF76-DB52-4CFB-8020-6F85BDB29FD5}</Project>
       <Name>Tango.Explorer</Name>
     </ProjectReference>
+    <ProjectReference Include="..\..\Tango.FileSystem\Tango.FileSystem.csproj">
+      <Project>{c6ebbbbe-2123-44dc-aef7-a0d47d736ac0}</Project>
+      <Name>Tango.FileSystem</Name>
+    </ProjectReference>
     <ProjectReference Include="..\..\Tango.Integration\Tango.Integration.csproj">
       <Project>{4206ac58-3b57-4699-8835-90bf6db01a61}</Project>
       <Name>Tango.Integration</Name>
@@ -446,7 +452,7 @@
   </Target>
   <ProjectExtensions>
     <VisualStudio>
-      <UserProperties BuildVersion_StartDate="2000/1/1" BuildVersion_UseGlobalSettings="False" BuildVersion_BuildVersioningStyle="None.None.Increment.TimeStamp" BuildVersion_UpdateAssemblyVersion="True" BuildVersion_AssemblyInfoFilename="Properties\AssemblyInfo.cs" />
+      <UserProperties BuildVersion_AssemblyInfoFilename="Properties\AssemblyInfo.cs" BuildVersion_UpdateAssemblyVersion="True" BuildVersion_BuildVersioningStyle="None.None.Increment.TimeStamp" BuildVersion_UseGlobalSettings="False" BuildVersion_StartDate="2000/1/1" />
     </VisualStudio>
   </ProjectExtensions>
 </Project>
\ No newline at end of file