mirrors/Mole
1
0
Fork 0
mirror of https://github.com/tw93/Mole.git synced 2026-02-04 17:24:45 +00:00
Code Activity

Merge branch 'main' of https://github.com/tw93/Mole

Browse Source
This commit is contained in:
tw93
2026-01-26 15:46:05 +08:00
parent 0fbf88a6c6 f272d29490
commit b67d149cef
1 changed files with 42 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

84
bin/clean.sh
View File

@@ -69,10 +69,10 @@ if [[ -f "$HOME/.config/mole/whitelist" ]]; then
fi fi
case "$line" in case "$line" in
/ | /System | /System/* | /bin | /bin/* | /sbin | /sbin/* | /usr/bin | /usr/bin/* | /usr/sbin | /usr/sbin/* | /etc | /etc/* | /var/db | /var/db/*) / | /System | /System/* | /bin | /bin/* | /sbin | /sbin/* | /usr/bin | /usr/bin/* | /usr/sbin | /usr/sbin/* | /etc | /etc/* | /var/db | /var/db/*)
WHITELIST_WARNINGS+=("Protected system path: $line") WHITELIST_WARNINGS+=("Protected system path: $line")
continue continue
;; ;;
esac esac
duplicate="false" duplicate="false"
@@ -86,7 +86,7 @@ if [[ -f "$HOME/.config/mole/whitelist" ]]; then
fi fi
[[ "$duplicate" == "true" ]] && continue [[ "$duplicate" == "true" ]] && continue
WHITELIST_PATTERNS+=("$line") WHITELIST_PATTERNS+=("$line")
done < "$HOME/.config/mole/whitelist" done <"$HOME/.config/mole/whitelist"
else else
WHITELIST_PATTERNS=("${DEFAULT_WHITELIST_PATTERNS[@]}") WHITELIST_PATTERNS=("${DEFAULT_WHITELIST_PATTERNS[@]}")
fi fi
@@ -140,7 +140,7 @@ cleanup() {
fi fi
CLEANUP_DONE=true CLEANUP_DONE=true
stop_inline_spinner 2> /dev/null || true stop_inline_spinner 2>/dev/null || true
if [[ -t 1 ]]; then if [[ -t 1 ]]; then
printf "\r\033[K" >&2 || true printf "\r\033[K" >&2 || true
@@ -166,8 +166,8 @@ start_section() {
if [[ "$DRY_RUN" == "true" ]]; then if [[ "$DRY_RUN" == "true" ]]; then
ensure_user_file "$EXPORT_LIST_FILE" ensure_user_file "$EXPORT_LIST_FILE"
echo "" >> "$EXPORT_LIST_FILE" echo "" >>"$EXPORT_LIST_FILE"
echo "=== $1 ===" >> "$EXPORT_LIST_FILE" echo "=== $1 ===" >>"$EXPORT_LIST_FILE"
fi fi
} }
@@ -220,7 +220,7 @@ normalize_paths_for_cleanup() {
done done
fi fi
[[ "$is_child" == "true" ]] || result_paths+=("$path") [[ "$is_child" == "true" ]] || result_paths+=("$path")
done <<< "$sorted_paths" done <<<"$sorted_paths"
if [[ ${#result_paths[@]} -gt 0 ]]; then if [[ ${#result_paths[@]} -gt 0 ]]; then
printf '%s\n' "${result_paths[@]}" printf '%s\n' "${result_paths[@]}"
@@ -232,9 +232,9 @@ get_cleanup_path_size_kb() {
local path="$1" local path="$1"
if [[ -f "$path" && ! -L "$path" ]]; then if [[ -f "$path" && ! -L "$path" ]]; then
if command -v stat > /dev/null 2>&1; then if command -v stat >/dev/null 2>&1; then
local bytes local bytes
bytes=$(stat -f%z "$path" 2> /dev/null || echo "0") bytes=$(stat -f%z "$path" 2>/dev/null || echo "0")
if [[ "$bytes" =~ ^[0-9]+$ && "$bytes" -gt 0 ]]; then if [[ "$bytes" =~ ^[0-9]+$ && "$bytes" -gt 0 ]]; then
echo $(((bytes + 1023) / 1024)) echo $(((bytes + 1023) / 1024))
return 0 return 0
@@ -243,9 +243,9 @@ get_cleanup_path_size_kb() {
fi fi
if [[ -L "$path" ]]; then if [[ -L "$path" ]]; then
if command -v stat > /dev/null 2>&1; then if command -v stat >/dev/null 2>&1; then
local bytes local bytes
bytes=$(stat -f%z "$path" 2> /dev/null || echo "0") bytes=$(stat -f%z "$path" 2>/dev/null || echo "0")
if [[ "$bytes" =~ ^[0-9]+$ && "$bytes" -gt 0 ]]; then if [[ "$bytes" =~ ^[0-9]+$ && "$bytes" -gt 0 ]]; then
echo $(((bytes + 1023) / 1024)) echo $(((bytes + 1023) / 1024))
else else
@@ -465,9 +465,9 @@ safe_clean() {
[[ ! "$size" =~ ^[0-9]+$ ]] && size=0 [[ ! "$size" =~ ^[0-9]+$ ]] && size=0
if [[ "$size" -gt 0 ]]; then if [[ "$size" -gt 0 ]]; then
echo "$size 1" > "$temp_dir/result_${idx}" echo "$size 1" >"$temp_dir/result_${idx}"
else else
echo "0 0" > "$temp_dir/result_${idx}" echo "0 0" >"$temp_dir/result_${idx}"
fi fi
((idx++)) ((idx++))
@@ -492,17 +492,17 @@ safe_clean() {
[[ ! "$size" =~ ^[0-9]+$ ]] && size=0 [[ ! "$size" =~ ^[0-9]+$ ]] && size=0
local tmp_file="$temp_dir/result_${idx}.$$" local tmp_file="$temp_dir/result_${idx}.$$"
if [[ "$size" -gt 0 ]]; then if [[ "$size" -gt 0 ]]; then
echo "$size 1" > "$tmp_file" echo "$size 1" >"$tmp_file"
else else
echo "0 0" > "$tmp_file" echo "0 0" >"$tmp_file"
fi fi
mv "$tmp_file" "$temp_dir/result_${idx}" 2> /dev/null || true mv "$tmp_file" "$temp_dir/result_${idx}" 2>/dev/null || true
) & ) &
pids+=($!) pids+=($!)
((idx++)) ((idx++))
if ((${#pids[@]} >= MOLE_MAX_PARALLEL_JOBS)); then if ((${#pids[@]} >= MOLE_MAX_PARALLEL_JOBS)); then
wait "${pids[0]}" 2> /dev/null || true wait "${pids[0]}" 2>/dev/null || true
pids=("${pids[@]:1}") pids=("${pids[@]:1}")
((completed++)) ((completed++))
@@ -515,7 +515,7 @@ safe_clean() {
if [[ ${#pids[@]} -gt 0 ]]; then if [[ ${#pids[@]} -gt 0 ]]; then
for pid in "${pids[@]}"; do for pid in "${pids[@]}"; do
wait "$pid" 2> /dev/null || true wait "$pid" 2>/dev/null || true
((completed++)) ((completed++))
if [[ "$show_spinner" == "true" && -t 1 ]]; then if [[ "$show_spinner" == "true" && -t 1 ]]; then
@@ -536,7 +536,7 @@ safe_clean() {
for path in "${existing_paths[@]}"; do for path in "${existing_paths[@]}"; do
local result_file="$temp_dir/result_${idx}" local result_file="$temp_dir/result_${idx}"
if [[ -f "$result_file" ]]; then if [[ -f "$result_file" ]]; then
read -r size count < "$result_file" 2> /dev/null || true read -r size count <"$result_file" 2>/dev/null || true
local removed=0 local removed=0
if [[ "$DRY_RUN" != "true" ]]; then if [[ "$DRY_RUN" != "true" ]]; then
if safe_remove "$path" true; then if safe_remove "$path" true; then
@@ -635,9 +635,9 @@ safe_clean() {
local size=0 local size=0
if [[ -n "${temp_dir:-}" && -f "$temp_dir/result_${idx}" ]]; then if [[ -n "${temp_dir:-}" && -f "$temp_dir/result_${idx}" ]]; then
read -r size count < "$temp_dir/result_${idx}" 2> /dev/null || true read -r size count <"$temp_dir/result_${idx}" 2>/dev/null || true
else else
size=$(get_cleanup_path_size_kb "$path" 2> /dev/null || echo "0") size=$(get_cleanup_path_size_kb "$path" 2>/dev/null || echo "0")
fi fi
[[ "$size" == "0" || -z "$size" ]] && { [[ "$size" == "0" || -z "$size" ]] && {
@@ -645,7 +645,7 @@ safe_clean() {
continue continue
} }
echo "$(dirname "$path")|$size|$path" >> "$paths_temp" echo "$(dirname "$path")|$size|$path" >>"$paths_temp"
((idx++)) ((idx++))
done done
fi fi
@@ -676,9 +676,9 @@ safe_clean() {
' | while IFS='|' read -r display_path total_size child_count; do ' | while IFS='|' read -r display_path total_size child_count; do
local size_human=$(bytes_to_human "$((total_size * 1024))") local size_human=$(bytes_to_human "$((total_size * 1024))")
if [[ $child_count -gt 1 ]]; then if [[ $child_count -gt 1 ]]; then
echo "$display_path # $size_human, $child_count items" >> "$EXPORT_LIST_FILE" echo "$display_path # $size_human, $child_count items" >>"$EXPORT_LIST_FILE"
else else
echo "$display_path # $size_human" >> "$EXPORT_LIST_FILE" echo "$display_path # $size_human" >>"$EXPORT_LIST_FILE"
fi fi
done done
@@ -718,7 +718,7 @@ start_cleanup() {
SYSTEM_CLEAN=false SYSTEM_CLEAN=false
ensure_user_file "$EXPORT_LIST_FILE" ensure_user_file "$EXPORT_LIST_FILE"
cat > "$EXPORT_LIST_FILE" << EOF cat >"$EXPORT_LIST_FILE" <<EOF
# Mole Cleanup Preview - $(date '+%Y-%m-%d %H:%M:%S') # Mole Cleanup Preview - $(date '+%Y-%m-%d %H:%M:%S')
# #
# How to protect files: # How to protect files:
@@ -734,7 +734,7 @@ EOF
fi fi
if [[ -t 0 ]]; then if [[ -t 0 ]]; then
if sudo -n true 2> /dev/null; then if sudo -n true 2>/dev/null; then
SYSTEM_CLEAN=true SYSTEM_CLEAN=true
echo -e "${GREEN}${ICON_SUCCESS}${NC} Admin access already available" echo -e "${GREEN}${ICON_SUCCESS}${NC} Admin access already available"
echo "" echo ""
@@ -774,7 +774,7 @@ EOF
else else
echo "" echo ""
echo "Running in non-interactive mode" echo "Running in non-interactive mode"
if sudo -n true 2> /dev/null; then if sudo -n true 2>/dev/null; then
SYSTEM_CLEAN=true SYSTEM_CLEAN=true
echo " ${ICON_LIST} System-level cleanup enabled, sudo session active" echo " ${ICON_LIST} System-level cleanup enabled, sudo session active"
else else
@@ -1025,7 +1025,7 @@ perform_cleanup() {
echo "# Potential cleanup: ${freed_gb}GB" echo "# Potential cleanup: ${freed_gb}GB"
echo "# Items: $files_cleaned" echo "# Items: $files_cleaned"
echo "# Categories: $total_items" echo "# Categories: $total_items"
} >> "$EXPORT_LIST_FILE" } >>"$EXPORT_LIST_FILE"
summary_details+=("Detailed file list: ${GRAY}$EXPORT_LIST_FILE${NC}") summary_details+=("Detailed file list: ${GRAY}$EXPORT_LIST_FILE${NC}")
summary_details+=("Use ${GRAY}mo clean --whitelist${NC} to add protection rules") summary_details+=("Use ${GRAY}mo clean --whitelist${NC} to add protection rules")
@@ -1077,18 +1077,18 @@ perform_cleanup() {
main() { main() {
for arg in "$@"; do for arg in "$@"; do
case "$arg" in case "$arg" in
"--debug") "--debug")
export MO_DEBUG=1 export MO_DEBUG=1
;; ;;
"--dry-run" | "-n") "--dry-run" | "-n")
DRY_RUN=true DRY_RUN=true
export MOLE_DRY_RUN=1 export MOLE_DRY_RUN=1
;; ;;
"--whitelist") "--whitelist")
source "$SCRIPT_DIR/../lib/manage/whitelist.sh" source "$SCRIPT_DIR/../lib/manage/whitelist.sh"
manage_whitelist "clean" manage_whitelist "clean"
exit 0 exit 0
;; ;;
esac esac
done done