Add schools remove-from-team command for batch team removal

Add a CLI command that removes one or more schools from a specified
team/subteam for a given academic year (defaults to pending). Includes
validation to prevent removal if the school has associated records.

Usage: `bin/mavis schools remove-from-team <Workgroup> <SubteamName> <URN...> [--academic-year YEAR]`

MAV-2717

Author: murugapl

app/lib/mavis_cli.rb

@@ -44,6 +44,7 @@ def self.progress_bar(total)
 require_relative "mavis_cli/schools/add_to_team"
 require_relative "mavis_cli/schools/create"
 require_relative "mavis_cli/schools/move_patients"
+require_relative "mavis_cli/schools/remove_from_team"
 require_relative "mavis_cli/schools/remove_programme_year_group"
 require_relative "mavis_cli/schools/show"
 require_relative "mavis_cli/sessions/configure"

app/lib/mavis_cli/schools/remove_from_team.rb

@@ -0,0 +1,74 @@
+# frozen_string_literal: true
+
+module MavisCLI
+  module Schools
+    class RemoveFromTeam < Dry::CLI::Command
+      desc "Remove an existing school from a team"
+
+      argument :team_workgroup,
+               required: true,
+               desc: "The workgroup of the team"
+      argument :subteam_name, required: true, desc: "The name of the subteam"
+      argument :urns,
+               type: :array,
+               required: true,
+               desc: "The URN of the school (including site, if applicable)"
+
+      option :academic_year,
+             type: :integer,
+             desc:
+               "The academic year to remove the school from (defaults to pending academic year)"
+
+      def call(team_workgroup:, subteam_name:, urns:, academic_year: nil, **)
+        MavisCLI.load_rails
+
+        team = Team.find_by(workgroup: team_workgroup)
+
+        if team.nil?
+          warn "Could not find team with workgroup #{team_workgroup}."
+          return
+        end
+
+        subteam = team.subteams.find_by(name: subteam_name)
+
+        if subteam.nil?
+          warn "Could not find subteam with name #{subteam_name}."
+          return
+        end
+
+        academic_year ||= AcademicYear.pending
+
+        ActiveRecord::Base.transaction do
+          urns.each do |urn|
+            location = Location.school.find_by_urn_and_site(urn)
+
+            if location.nil?
+              warn "Could not find location with URN #{urn}"
+              next
+            end
+
+            team_location =
+              TeamLocation
+                .includes(:team)
+                .where(team:, academic_year:, subteam:, location:)
+                .sole
+
+            unless team_location.safe_to_destroy?
+              warn "Location #{location.id} (URN: #{urn}) cannot be removed as it has associated records."
+              next
+            end
+
+            team_location.destroy!
+
+            puts "Location #{location.id} (URN: #{urn}) has been removed from subteam #{subteam_name} " \
+                   "for academic year #{academic_year}."
+          end
+        end
+      end
+    end
+  end
+
+  register "schools" do |prefix|
+    prefix.register "remove-from-team", Schools::RemoveFromTeam
+  end
+end

app/models/team_location.rb

@@ -62,6 +62,10 @@ def phone = subteam&.phone || team&.phone
   def phone_instructions =
     subteam&.phone_instructions || team&.phone_instructions
 
+  def safe_to_destroy?
+    !sessions.exists? && !consent_forms.exists? && !patient_locations.exists?
+  end
+
   private
 
   def subteam_belongs_to_team

spec/features/cli_schools_remove_from_team_spec.rb

@@ -0,0 +1,64 @@
+# frozen_string_literal: true
+
+require_relative "../../app/lib/mavis_cli"
+
+describe "mavis schools remove-from-team" do
+  context "with valid arguments" do
+    it "runs successfully" do
+      given_schools_exist
+      when_i_run_the_command
+      then_the_schools_are_removed_from_the_team
+    end
+  end
+
+  private
+
+  def given_schools_exist
+    team = create(:team, workgroup: "TeamA")
+    subteam = create(:subteam, team:, name: "SubteamA")
+    @school_a =
+      create(
+        :school,
+        urn: "123456",
+        name: "MainSchool",
+        site: nil,
+        team:,
+        subteam:
+      )
+    @school_b =
+      create(
+        :school,
+        urn: "654321",
+        name: "OtherSchool",
+        site: nil,
+        team:,
+        subteam:
+      )
+    expect(@school_a.teams.count).to eq(1)
+    expect(@school_b.teams.count).to eq(1)
+  end
+
+  def command
+    Dry::CLI.new(MavisCLI).call(
+      arguments: %w[
+        schools
+        remove-from-team
+        TeamA
+        SubteamA
+        123456
+        654321
+        --academic-year
+        2025
+      ]
+    )
+  end
+
+  def when_i_run_the_command
+    @output = capture_output { command }
+  end
+
+  def then_the_schools_are_removed_from_the_team
+    expect(@school_a.teams.count).to eq(0)
+    expect(@school_b.teams.count).to eq(0)
+  end
+end