Process project name

Author: Saif Al-Din Ali

src/migrate/azext_migrate/custom.py

@@ -172,7 +172,7 @@ def new_local_server_replication(cmd,
         subscription_id = get_subscription_id(cmd.cli_ctx)
         print(f"Selected Subscription Id: '{subscription_id}'")
 
-    rg_uri, machine_id, subscription_id = validate_server_parameters(
+    rg_uri, machine_id, subscription_id, project_name = validate_server_parameters(
         cmd,
         machine_id,
         machine_index,
@@ -239,7 +239,7 @@ def new_local_server_replication(cmd,
                 "Please verify your appliance setup and provided "
                 "-machine_id.")
 
-        amh_solution, migrate_project, machine_props = process_amh_solution(
+        amh_solution, migrate_project, machine_props, project_name = process_amh_solution(
             cmd,
             machine,
             site_object,

src/migrate/azext_migrate/helpers/replication/new/_process_inputs.py

@@ -228,7 +228,7 @@ def process_amh_solution(cmd,
             f"'{resource_group_name}' and project '{project_name}'. "
             "Please verify your appliance setup."
         )
-    return amh_solution, migrate_project, machine_props
+    return amh_solution, migrate_project, machine_props, project_name
 
 
 def process_replication_vault(cmd,

src/migrate/azext_migrate/helpers/replication/new/_validate.py

@@ -206,6 +206,11 @@ def validate_server_parameters(  # pylint: disable=too-many-locals,too-many-bran
         # machine_id was provided directly
         # Check if it's in Microsoft.Migrate format and needs to be resolved
         if "/Microsoft.Migrate/MigrateProjects/" in machine_id or "/Microsoft.Migrate/migrateprojects/" in machine_id:
+            # Extract project_name from the Microsoft.Migrate machine ID
+            migrate_id_parts = machine_id.split("/")
+            if len(migrate_id_parts) >= 9 and not project_name:
+                project_name = migrate_id_parts[8]
+
             # This is a Migrate Project machine ID, need to resolve to OffAzure machine ID
             migrate_machine = get_resource_by_id(
                 cmd, machine_id, APIVersion.Microsoft_Migrate.value)
@@ -257,7 +262,7 @@ def validate_server_parameters(  # pylint: disable=too-many-locals,too-many-bran
             f"/subscriptions/{subscription_id}/"
             f"resourceGroups/{resource_group_name}")
 
-    return rg_uri, machine_id, subscription_id
+    return rg_uri, machine_id, subscription_id, project_name
 
 
 def validate_required_parameters(machine_id,

src/migrate/azext_migrate/tests/latest/test_migrate_commands.py

@@ -3246,7 +3246,7 @@ def test_process_amh_solution_project_from_discovery(self, mock_get_resource):
 
         mock_get_resource.side_effect = [mock_project, mock_amh]
 
-        amh, project, props = process_amh_solution(
+        amh, _, _, _ = process_amh_solution(
             mock_cmd, mock_machine, mock_site, None, 'rg1', 'machine1',
             '/subscriptions/sub1/resourceGroups/rg1'
         )