1
0
Fork 0

wipe: Add new repo wipe subcommand

This new command allows users to delete projects from the worktree
and from the `.repo` directory. It is a destructive operation.

It handles shared projects by refusing to wipe them unless the
`--force` flag is used. It also checks for uncommitted changes
before wiping.

Bug: 393383056
Change-Id: Ia30d8ffdc781a3f179af56310ce31c9dae331bbe
This commit is contained in:
Gavin Mak 2025-07-15 19:06:08 +00:00
parent 5d95ba8d85
commit 05efa68da6
2 changed files with 277 additions and 0 deletions

97
subcmds/wipe.py Normal file
View file

@ -0,0 +1,97 @@
# Copyright (C) 2025 The Android Open Source Project
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import os
import shutil
from command import Command
from command import UsageError
from git_command import GitCommand
from project import Project
class Wipe(Command, GitCommand):
"""Delete projects from the worktree and .repo"""
COMMON = True
helpSummary = "Wipe projects from the worktree"
helpUsage = """
%prog <project>...
"""
helpDescription = """
The `repo wipe` command removes the specified projects from the worktree,
and deletes the project's git data from `.repo`.
This is a destructive operation and cannot be undone.
"""
def _Options(self, p):
p.add_option(
"-f",
"--force",
action="store_true",
help="force wipe in the case of shared projects",
)
def Execute(self, opt, args):
if not args:
raise UsageError("no projects specified")
# We need all projects to correctly handle shared object directories.
all_projects = self.GetAllProjects(all_manifests=True)
projects_to_wipe = self.GetProjects(args, all_manifests=True)
names_to_wipe = {p.name for p in projects_to_wipe}
# Build a map from objdir to the names of projects that use it.
objdir_map = {}
for p in all_projects:
objdir_map.setdefault(p.objdir, set()).add(p.name)
objdirs_to_delete = set()
for project in projects_to_wipe:
users = objdir_map.get(project.objdir, {project.name})
is_shared_with_other_projects = not users.issubset(names_to_wipe)
if is_shared_with_other_projects and not opt.force:
# Find an example project it's shared with for a helpful error.
other_user = list(users - names_to_wipe)[0]
raise UsageError(
f"project '{project.name}' shares object directory with "
f"'{other_user}' (and possibly others). Use --force to wipe."
)
self._WipeProject(project)
if not is_shared_with_other_projects:
objdirs_to_delete.add(project.objdir)
for objdir in objdirs_to_delete:
if os.path.exists(objdir):
print(f"Deleting objects directory: {objdir}")
shutil.rmtree(objdir)
def _WipeProject(self, project):
"""Wipes a single project's worktree and git directory."""
if not project:
return
# Delete the worktree.
if project.worktree and os.path.exists(project.worktree):
print(f"Deleting worktree: {project.worktree}")
shutil.rmtree(project.worktree)
# Delete the git directory.
if os.path.exists(project.gitdir):
print(f"Deleting git directory: {project.gitdir}")
shutil.rmtree(project.gitdir)