aboutsummaryrefslogtreecommitdiffstats
path: root/Software/Visual_Studio/PPC/Tango.PPC.Common/RemoteJobUpload/DefaultRemoteJobUploadService.cs
blob: a0205215c6bfa58e3a40db769eb8bd3acce99b47 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using Tango.Core;
using Tango.Core.DI;
using Tango.Core.IO;
using Tango.Integration.ExternalBridge;
using Tango.PPC.Common.ExternalBridge;
using Tango.PPC.Shared.RemoteJobUpload;
using Tango.Settings;

namespace Tango.PPC.Common.RemoteJobUpload
{
    [TangoCreateWhenRegistered]
    public class DefaultRemoteJobUploadService : ExtendedObject, IRemoteJobUploadService, IExternalBridgeRequestHandler
    {
        private class PendingJobUpload
        {
            public String ID { get; set; }
            public String Name { get; set; }
            public TemporaryFile File { get; set; }
            public RemoteJobUploadType Type { get; set; }
        }

        private PPCSettings _settings;
        private TemporaryFolder _jobsFolder;
        private List<PendingJobUpload> _jobs;

        public event EventHandler<RemoteJobReceivedEventArgs> JobReceived;

        public bool Enabled
        {
            get { return _settings.EnableRemoteJobUpload; }
            set { _settings.EnableRemoteJobUpload = value; }
        }

        public DefaultRemoteJobUploadService(IPPCExternalBridgeService externalBridge)
        {
            try
            {
                _jobs = new List<PendingJobUpload>();
                externalBridge.RegisterRequestHandler(this);
                _settings = SettingsManager.Default.GetOrCreate<PPCSettings>();
                _jobsFolder = TemporaryManager.CreateFolder();
            }
            catch (Exception ex)
            {
                LogManager.Log(ex, "Error initializing the remote job upload service.");
                Enabled = false;
            }
        }

        [ExternalBridgeRequestHandlerMethod(typeof(RemoteJobUploadRequest), RequestHandlerLoggingMode.LogRequestNameAndContent)]
        public async Task OnRemoteJobUploadRequest(RemoteJobUploadRequest request, String token, ExternalBridgeReceiver receiver)
        {
            this.ThrowIfDisabled();

            PendingJobUpload job = new PendingJobUpload();
            job.ID = Guid.NewGuid().ToString();
            job.Name = request.Name;
            job.File = _jobsFolder.CreateImaginaryFile();
            job.Type = request.Type;

            _jobs.Add(job);

            await receiver.SendGenericResponse(new RemoteJobUploadResponse()
            {
                TargetFilePath = job.File.Path,
                ID = job.ID
            }, token);
        }

        [ExternalBridgeRequestHandlerMethod(typeof(RemoteJobUploadCompletedRequest), RequestHandlerLoggingMode.LogRequestNameAndContent)]
        public async Task OnRemoteJobUploadCompletedRequest(RemoteJobUploadCompletedRequest request, String token, ExternalBridgeReceiver receiver)
        {
            var completion = new TaskCompletionSource<object>();

            var job = _jobs.SingleOrDefault(x => x.ID == request.ID);

            if (job == null)
            {
                throw new FileNotFoundException("The specified job id could not be found.");
            }

            _jobs.Remove(job);

            JobReceived?.Invoke(this, new RemoteJobReceivedEventArgs(async () => 
            {
                job.File.Delete();
                await receiver.SendGenericResponse(new RemoteJobUploadCompletedResponse(), token);
                completion.SetResult(true);

            }, async (error) => 
            {
                job.File.Delete();
                await receiver.SendGenericResponse(new RemoteJobUploadCompletedResponse() { Error = error }, token);
                completion.SetResult(true);

            }) { Type = job.Type, FilePath = job.File.Path, Name = job.Name });

            await completion.Task;
        }

        public void OnReceiverDisconnected(ExternalBridgeReceiver receiver)
        {
            
        }
    }
}