feat: query for manual update via base image from a collection

[<35;92;38M35;102;41M35;104;41M35;106;41M35;107;4

Signed-off-by: Damiano Mason <damiano.mason@secomind.com>

Author: Dam-99

backend/lib/edgehog/os_management/os_management.ex

@@ -31,7 +31,13 @@ defmodule Edgehog.OSManagement do
     root_level_errors? true
 
     mutations do
-      create OTAOperation, :create_manual_ota_operation, :manual do
+      create OTAOperation, :create_manual_ota_operation_no_existing_base_image, :manual do
+        relay_id_translations input: [device_id: :device]
+      end
+
+      create OTAOperation,
+             :create_manual_ota_operation_existing_base_image,
+             :manual_from_collection do
         relay_id_translations input: [device_id: :device]
       end
 

backend/lib/edgehog/os_management/ota_operation/ota_operation.ex

@@ -96,6 +96,25 @@ defmodule Edgehog.OSManagement.OTAOperation do
       change Changes.SendUpdateRequest
     end
 
+    create :manual_from_collection do
+      description "Initiates an OTA update by selecting one from an available collection"
+
+      accept [:base_image_url]
+
+      argument :device_id, :id do
+        description "The ID identifying the Device the OTA Operation will be sent to"
+      end
+
+      change manage_relationship(:device_id, :device,
+               type: :append,
+               eager_validate_with: Edgehog.Devices
+             )
+
+      change set_attribute(:manual?, true)
+
+      change Changes.SendUpdateRequest
+    end
+
     destroy :destroy do
       require_atomic? false
 

frontend/src/api/schema.graphql

@@ -5005,16 +5005,33 @@ type Tag {
   name: String!
 }
 
-"The result of the :create_manual_ota_operation mutation"
-type CreateManualOtaOperationResult {
+"The result of the :create_manual_ota_operation_existing_base_image mutation"
+type CreateManualOtaOperationExistingBaseImageResult {
   "The successful result of the mutation"
   result: OtaOperation
 
   "Any errors generated, if the mutation failed"
   errors: [MutationError!]!
 }
 
-input CreateManualOtaOperationInput {
+input CreateManualOtaOperationExistingBaseImageInput {
+  "The URL of the base image being installed on the device"
+  baseImageUrl: String!
+
+  "The ID identifying the Device the OTA Operation will be sent to"
+  deviceId: ID
+}
+
+"The result of the :create_manual_ota_operation_no_existing_base_image mutation"
+type CreateManualOtaOperationNoExistingBaseImageResult {
+  "The successful result of the mutation"
+  result: OtaOperation
+
+  "Any errors generated, if the mutation failed"
+  errors: [MutationError!]!
+}
+
+input CreateManualOtaOperationNoExistingBaseImageInput {
   "The ID identifying the Device the OTA Operation will be sent to"
   deviceId: ID!
 
@@ -6257,9 +6274,14 @@ type RootMutationType {
   deleteChannel(id: ID!): DeleteChannelResult
 
   "Initiates an OTA update with a user provided OS image"
-  createManualOtaOperation(
-    input: CreateManualOtaOperationInput!
-  ): CreateManualOtaOperationResult
+  createManualOtaOperationNoExistingBaseImage(
+    input: CreateManualOtaOperationNoExistingBaseImageInput!
+  ): CreateManualOtaOperationNoExistingBaseImageResult
+
+  "Initiates an OTA update by selecting one from an available collection"
+  createManualOtaOperationExistingBaseImage(
+    input: CreateManualOtaOperationExistingBaseImageInput!
+  ): CreateManualOtaOperationExistingBaseImageResult
 
   "Creates a new device group."
   createDeviceGroup(input: CreateDeviceGroupInput!): CreateDeviceGroupResult