cnst/cnix
1
0
Fork 0
Code Issues Pull Requests Activity

cleanup script fix and toothpick modules fix

Browse Source
This commit is contained in:
cnst
2024-09-15 17:54:20 +02:00
parent ef91994c62
commit b408f513d0
8 changed files with 305 additions and 201 deletions
Download Patch File Download Diff File Expand all files Collapse all files

184
.scripts/cleanup-boot.sh
View File

@@ -5,13 +5,7 @@
# Exit on any error, undefined variable, or pipeline failure
set -euo pipefail
# Ensure Bash 4 or newer is being used
if ((BASH_VERSINFO[0] < 4)); then
echo "Error: This script requires Bash version 4 or higher."
exit 1
fi
# Default number of versions and files to keep
# Default number of kernel and initrd files to keep
KEEP_KERNEL=4
KEEP_INITRD=6
@@ -30,10 +24,6 @@ usage() {
echo " --keep-kernel N Keep the latest N kernel files (default: 4)."
echo " --keep-initrd N Keep the latest N initrd files (default: 6)."
echo " --help, -h Show this help message."
echo
echo "Examples:"
echo " ./cleanup-boot.sh --dry-run"
echo " ./cleanup-boot.sh --keep-kernel 3 --keep-initrd 5"
exit 1
}
@@ -97,164 +87,66 @@ log "Starting cleanup script. Keeping $KEEP_KERNEL kernel files and $KEEP_INITRD
# Collect all .efi files in /boot/EFI/nixos
mapfile -d '' -t efi_files < <(find /boot/EFI/nixos -type f -name '*.efi' -print0)
declare -A kernels_files_by_version
declare -A initrds_files_by_version
declare -A versions_mtime
# Initialize arrays for kernel and initrd files
kernel_files=()
initrd_files=()
# Parse filenames to group kernel and initrd files by their version number
# Parse filenames and collect kernel and initrd files based on patterns
for file in "${efi_files[@]}"; do
basename=$(basename "$file")
# Pattern 1: <hash>-linux-<version>-bzImage.efi (kernel)
# Pattern: <hash>-linux-<version>-bzImage.efi (kernel)
if [[ "$basename" =~ ^(.*)-linux-([0-9]+\.[0-9]+\.[0-9]+(-[a-zA-Z0-9\.\+]+)?)-bzImage\.efi$ ]]; then
version="${BASH_REMATCH[2]}"
type="kernel"
kernel_files+=("$file")
# Pattern 1: <hash>-initrd-linux-<version>-initrd.efi (initrd)
# Pattern: <hash>-initrd-linux-<version>-initrd.efi (initrd)
elif [[ "$basename" =~ ^(.*)-initrd-linux-([0-9]+\.[0-9]+\.[0-9]+(-[a-zA-Z0-9\.\+]+)?)-initrd\.efi$ ]]; then
version="${BASH_REMATCH[2]}"
type="initrd"
initrd_files+=("$file")
# Pattern 2: kernel-<version>-<hash>.efi (kernel)
# Pattern: kernel-<version>-<hash>.efi (kernel)
elif [[ "$basename" =~ ^kernel-([0-9]+\.[0-9]+\.[0-9]+(-[a-zA-Z0-9\.\+]+)?)-([a-zA-Z0-9]+)\.efi$ ]]; then
version="${BASH_REMATCH[1]}"
type="kernel"
kernel_files+=("$file")
# Pattern 2: initrd-<version>-<hash>.efi (initrd)
# Pattern: initrd-<version>-<hash>.efi (initrd)
elif [[ "$basename" =~ ^initrd-([0-9]+\.[0-9]+\.[0-9]+(-[a-zA-Z0-9\.\+]+)?)-([a-zA-Z0-9]+)\.efi$ ]]; then
version="${BASH_REMATCH[1]}"
type="initrd"
initrd_files+=("$file")
else
log "Warning: Unrecognized filename format: $basename"
continue
fi
# Get file modification time
if ! file_mtime=$(stat -c '%Y' "$file" 2>/dev/null); then
log "Warning: Failed to get modification time for $file"
continue
fi
# Append the file to the list of files per version and type
entry="$file_mtime|$file"
if [[ "$type" == "kernel" ]]; then
kernels_files_by_version["$version"]+="$entry"$'\n'
elif [[ "$type" == "initrd" ]]; then
initrds_files_by_version["$version"]+="$entry"$'\n'
fi
# Update the latest modification time for the version
version_mtime="${versions_mtime["$version"]:-0}"
if [[ "$file_mtime" -gt "$version_mtime" ]]; then
versions_mtime["$version"]="$file_mtime"
fi
done
# Collect all versions
all_versions=("${!versions_mtime[@]}")
# Function to process and delete old files
process_files() {
local -n files=$1 # Pass array by reference
local keep_count=$2
local file_type=$3
# Sort versions by their latest modification time (newest first)
mapfile -t sorted_versions < <(
for version in "${all_versions[@]}"; do
echo "${versions_mtime[$version]}:$version"
done | sort -rn -k1,1 | awk -F: '{print $2}'
)
log "Processing $file_type files..."
version_count=${#sorted_versions[@]}
if [ "${#files[@]}" -gt "$keep_count" ]; then
# Sort files by modification time and keep the latest N files
sorted_files=$(for f in "${files[@]}"; do echo "$(stat -c '%Y' "$f"):$f"; done | sort -rn -k1,1)
log "Found $version_count versions."
# Collect files to delete
mapfile -t files_to_delete < <(echo "$sorted_files" | tail -n +"$((keep_count + 1))" | cut -d: -f2)
if [ "$version_count" -eq 0 ]; then
log "No versions found."
exit 0
fi
# Initialize delete_files array
delete_files=()
# Log the files identified for deletion
log "Files identified for deletion:"
# Process initrd files
for version in "${sorted_versions[@]}"; do
initrd_files="${initrds_files_by_version["$version"]:-}"
if [ -n "$initrd_files" ]; then
IFS=''
mapfile -t initrd_files_array <<<"$initrd_files"
unset IFS
# Sort initrd files by mtime and filename (newest first)
mapfile -t sorted_initrd_files < <(printf '%s\n' "${initrd_files_array[@]}" | sort -rn -k1,1 -k2,2)
initrd_file_count=${#sorted_initrd_files[@]}
if [ "$KEEP_INITRD" -gt 0 ] && [ "$initrd_file_count" -gt "$KEEP_INITRD" ]; then
files_to_delete=("${sorted_initrd_files[@]:$KEEP_INITRD}")
for entry in "${files_to_delete[@]}"; do
file="${entry#*|}"
delete_files+=("$file")
log "$file"
done
fi
fi
done
# Process kernel files
for version in "${sorted_versions[@]}"; do
kernel_files="${kernels_files_by_version["$version"]:-}"
if [ -n "$kernel_files" ]; then
IFS=''
mapfile -t kernel_files_array <<<"$kernel_files"
unset IFS
# Sort kernel files by mtime and filename (newest first)
mapfile -t sorted_kernel_files < <(printf '%s\n' "${kernel_files_array[@]}" | sort -rn -k1,1 -k2,2)
kernel_file_count=${#sorted_kernel_files[@]}
if [ "$KEEP_KERNEL" -gt 0 ] && [ "$kernel_file_count" -gt "$KEEP_KERNEL" ]; then
files_to_delete=("${sorted_kernel_files[@]:$KEEP_KERNEL}")
for entry in "${files_to_delete[@]}"; do
file="${entry#*|}"
delete_files+=("$file")
log "$file"
done
fi
fi
done
if [ ${#delete_files[@]} -eq 0 ]; then
log "No files to delete."
exit 0
fi
# Confirm dry run mode
if [ "$DRY_RUN" = true ]; then
log "Dry run mode enabled. No files will be deleted."
fi
# Remove old files
for file in "${delete_files[@]}"; do
# Skip empty entries
if [[ -z "$file" ]]; then
continue
fi
# Resolve absolute path
resolved_file=$(realpath "$file")
# Security check: Ensure the file is within /boot/EFI/nixos
if [[ "$resolved_file" != /boot/EFI/nixos/* ]]; then
log "Warning: Attempted to delete file outside of /boot/EFI/nixos: $resolved_file"
continue
fi
if [ "$DRY_RUN" = false ]; then
if rm -f -- "$resolved_file"; then
log "Deleted: $resolved_file"
else
log "Failed to delete: $resolved_file"
fi
for file in "${files_to_delete[@]}"; do
log "Deleting $file"
if [ "$DRY_RUN" = false ]; then
rm -f -- "$file"
else
log "Dry run - would delete: $file"
fi
done
else
log "Dry run - would delete: $resolved_file"
log "No $file_type files to delete. Current count: ${#files[@]}"
fi
done
}
# Process initrd and kernel files
process_files kernel_files "$KEEP_KERNEL" "kernel"
process_files initrd_files "$KEEP_INITRD" "initrd"
log "Cleanup script completed."