Handle custom and echo mission in same files

equinor · Aug 1, 2024 · 3f963f8 · 3f963f8
1 parent 1ba6de0
commit 3f963f8
Show file tree

Hide file tree

Showing 9 changed files with 145 additions and 284 deletions.
diff --git a/backend/api.test/EventHandlers/TestMissionEventHandler.cs b/backend/api.test/EventHandlers/TestMissionEventHandler.cs
@@ -52,6 +52,7 @@ public TestMissionEventHandler(DatabaseFixture fixture)
             var returnToHomeServiceLogger = new Mock<ILogger<ReturnToHomeService>>().Object;
             var missionDefinitionServiceLogger = new Mock<ILogger<MissionDefinitionService>>().Object;
             var lastMissionRunServiceLogger = new Mock<ILogger<LastMissionRunService>>().Object;
+            var sourceServiceLogger = new Mock<ILogger<SourceService>>().Object;
 
             var configuration = WebApplication.CreateBuilder().Configuration;
 
@@ -66,8 +67,8 @@ public TestMissionEventHandler(DatabaseFixture fixture)
 
             var echoServiceMock = new MockEchoService();
             var stidServiceMock = new MockStidService(context);
-            var customMissionServiceMock = new MockCustomMissionService();
-            var missionDefinitionService = new MissionDefinitionService(context, echoServiceMock, customMissionServiceMock, signalRService, accessRoleService, missionDefinitionServiceLogger, _missionRunService);
+            var sourceService = new SourceService(context, echoServiceMock, sourceServiceLogger);
+            var missionDefinitionService = new MissionDefinitionService(context, echoServiceMock, sourceService, signalRService, accessRoleService, missionDefinitionServiceLogger, _missionRunService);
             var robotModelService = new RobotModelService(context);
             var taskDurationServiceMock = new MockTaskDurationService();
             var isarServiceMock = new MockIsarService();

diff --git a/backend/api.test/Mocks/CustomMissionServiceMock.cs b/backend/api.test/Mocks/CustomMissionServiceMock.cs
diff --git a/backend/api.test/TestWebApplicationFactory.cs b/backend/api.test/TestWebApplicationFactory.cs
@@ -36,7 +36,6 @@ protected override void ConfigureWebHost(IWebHostBuilder builder)
                     services.AddScoped<IMapService, MockMapService>();
                     services.AddScoped<IBlobService, MockBlobService>();
                     services.AddScoped<IStidService, MockStidService>();
-                    services.AddScoped<ICustomMissionService, MockCustomMissionService>();
                     services.AddAuthorizationBuilder().AddFallbackPolicy(
                         TestAuthHandler.AuthenticationScheme, policy => policy.RequireAuthenticatedUser()
                     );

diff --git a/backend/api/Controllers/MissionSchedulingController.cs b/backend/api/Controllers/MissionSchedulingController.cs
@@ -14,7 +14,6 @@ namespace Api.Controllers
     [Route("missions")]
     public class MissionSchedulingController(
             IMissionDefinitionService missionDefinitionService,
-            ICustomMissionSchedulingService customMissionSchedulingService,
             IMissionRunService missionRunService,
             IInstallationService installationService,
             IEchoService echoService,
@@ -23,7 +22,8 @@ public class MissionSchedulingController(
             IStidService stidService,
             ILocalizationService localizationService,
             IRobotService robotService,
-            ISourceService sourceService
+            ISourceService sourceService,
+            IAreaService areaService
         ) : ControllerBase
 
     {
@@ -371,7 +371,41 @@ [FromBody] CustomMissionQuery customMissionQuery
             var missionTasks = customMissionQuery.Tasks.Select(task => new MissionTask(task)).ToList();
 
             MissionDefinition? customMissionDefinition;
-            try { customMissionDefinition = await customMissionSchedulingService.FindExistingOrCreateCustomMissionDefinition(customMissionQuery, missionTasks); }
+            try
+            {
+                Area? area = null;
+                if (customMissionQuery.AreaName != null) { area = await areaService.ReadByInstallationAndName(customMissionQuery.InstallationCode, customMissionQuery.AreaName); }
+
+                if (area == null)
+                {
+                    throw new AreaNotFoundException($"No area with name {customMissionQuery.AreaName} in installation {customMissionQuery.InstallationCode} was found");
+                }
+
+                var source = await sourceService.CheckForExistingCustomSource(missionTasks);
+
+                MissionDefinition? existingMissionDefinition = null;
+                if (source == null)
+                {
+                    source = await sourceService.CreateSourceIfDoesNotExist(missionTasks);
+                }
+                else
+                {
+                    var missionDefinitions = await missionDefinitionService.ReadBySourceId(source.SourceId);
+                    if (missionDefinitions.Count > 0) { existingMissionDefinition = missionDefinitions.First(); }
+                }
+
+                customMissionDefinition = existingMissionDefinition ?? new MissionDefinition
+                {
+                    Id = Guid.NewGuid().ToString(),
+                    Source = source,
+                    Name = customMissionQuery.Name,
+                    InspectionFrequency = customMissionQuery.InspectionFrequency,
+                    InstallationCode = customMissionQuery.InstallationCode,
+                    Area = area
+                };
+
+                if (existingMissionDefinition == null) { await missionDefinitionService.Create(customMissionDefinition); }
+            }
             catch (SourceException e) { return StatusCode(StatusCodes.Status502BadGateway, e.Message); }
             catch (AreaNotFoundException) { return NotFound($"No area with name {customMissionQuery.AreaName} in installation {customMissionQuery.InstallationCode} was found"); }
 
@@ -380,7 +414,30 @@ [FromBody] CustomMissionQuery customMissionQuery
             catch (RobotNotInSameInstallationAsMissionException e) { return Conflict(e.Message); }
 
             MissionRun? newMissionRun;
-            try { newMissionRun = await customMissionSchedulingService.QueueCustomMissionRun(customMissionQuery, customMissionDefinition.Id, robot.Id, missionTasks); }
+            try
+            {
+                var scheduledMission = new MissionRun
+                {
+                    Name = customMissionQuery.Name,
+                    Description = customMissionQuery.Description,
+                    MissionId = customMissionDefinition.Id,
+                    Comment = customMissionQuery.Comment,
+                    Robot = robot,
+                    Status = MissionStatus.Pending,
+                    MissionRunType = MissionRunType.Normal,
+                    DesiredStartTime = customMissionQuery.DesiredStartTime ?? DateTime.UtcNow,
+                    Tasks = missionTasks,
+                    InstallationCode = customMissionQuery.InstallationCode,
+                    Area = customMissionDefinition.Area,
+                    Map = new MapMetadata()
+                };
+
+                await mapService.AssignMapToMission(scheduledMission);
+
+                if (scheduledMission.Tasks.Any()) { scheduledMission.CalculateEstimatedDuration(); }
+
+                newMissionRun = await missionRunService.Create(scheduledMission);
+            }
             catch (Exception e) when (e is UnsupportedRobotCapabilityException) { return BadRequest(e.Message); }
             catch (Exception e) when (e is MissionNotFoundException) { return NotFound(e.Message); }
             catch (Exception e) when (e is RobotNotFoundException) { return NotFound(e.Message); }

diff --git a/backend/api/Program.cs b/backend/api/Program.cs
@@ -70,7 +70,6 @@
 builder.Services.AddScoped<ISourceService, SourceService>();
 
 builder.Services.AddScoped<IMissionSchedulingService, MissionSchedulingService>();
-builder.Services.AddScoped<ICustomMissionSchedulingService, CustomMissionSchedulingService>();
 
 builder.Services.AddScoped<IIsarService, IsarService>();
 builder.Services.AddScoped<IEchoService, EchoService>();
@@ -106,7 +105,6 @@
     builder.Services.AddScoped<ITimeseriesService, TimeseriesService>();
 builder.Services.AddScoped<RobotController>();
 builder.Services.AddScoped<EmergencyActionController>();
-builder.Services.AddScoped<ICustomMissionService, CustomMissionService>();
 builder.Services.AddScoped<InspectionFindingService>();
 
 builder.Services.AddTransient<ISignalRService, SignalRService>();

diff --git a/backend/api/Services/ActionServices/CustomMissionSchedulingService.cs b/backend/api/Services/ActionServices/CustomMissionSchedulingService.cs