blob: a41441d91b7ebbe1135399e2776d5edc3fe36445 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
|
#!/usr/bin/env bash
# common.sh - Shared functions for archangel installer
# Source this file: source "$(dirname "$0")/lib/common.sh"
#############################
# Output Functions
#############################
# Colors (optional, gracefully degrade if not supported)
if [[ -t 1 ]]; then
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[0;33m'
BLUE='\033[0;34m'
BOLD='\033[1m'
NC='\033[0m' # No Color
else
RED=''
GREEN=''
YELLOW=''
BLUE=''
BOLD=''
NC=''
fi
info() { echo -e "${GREEN}[INFO]${NC} $1"; }
warn() { echo -e "${YELLOW}[WARN]${NC} $1"; }
error() { echo -e "${RED}[ERROR]${NC} $1"; exit 1; }
step() { echo ""; echo -e "${BOLD}==> $1${NC}"; }
prompt() { echo -e "${BLUE}$1${NC}"; }
# Log to file if LOG_FILE is set
log() {
local msg="[$(date +'%Y-%m-%d %H:%M:%S')] $1"
if [[ -n "$LOG_FILE" ]]; then
echo "$msg" >> "$LOG_FILE"
fi
}
#############################
# Validation Functions
#############################
require_root() {
[[ $EUID -ne 0 ]] && error "This script must be run as root"
}
command_exists() {
command -v "$1" &>/dev/null
}
require_command() {
command_exists "$1" || error "Required command not found: $1"
}
#############################
# FZF Prompts
#############################
# Check if fzf is available
has_fzf() {
command_exists fzf
}
# Generic fzf selection
# Usage: result=$(fzf_select "prompt" "option1" "option2" ...)
fzf_select() {
local prompt="$1"
shift
local options=("$@")
if has_fzf; then
printf '%s\n' "${options[@]}" | fzf --prompt="$prompt " --height=15 --reverse
else
# Fallback to simple select
PS3="$prompt "
select opt in "${options[@]}"; do
if [[ -n "$opt" ]]; then
echo "$opt"
break
fi
done
fi
}
# Multi-select with fzf
# Usage: readarray -t results < <(fzf_multi "prompt" "opt1" "opt2" ...)
fzf_multi() {
local prompt="$1"
shift
local options=("$@")
if has_fzf; then
printf '%s\n' "${options[@]}" | fzf --prompt="$prompt " --height=20 --reverse --multi
else
# Fallback: just return all options (user must edit)
printf '%s\n' "${options[@]}"
fi
}
#############################
# Disk Utilities
#############################
# Get disk size in human-readable format
get_disk_size() {
local disk="$1"
lsblk -dno SIZE "$disk" 2>/dev/null | tr -d ' '
}
# Get disk model
get_disk_model() {
local disk="$1"
lsblk -dno MODEL "$disk" 2>/dev/null | tr -d ' ' | head -c 20
}
# Check if disk is in use (mounted or has holders)
disk_in_use() {
local disk="$1"
[[ -n "$(lsblk -no MOUNTPOINT "$disk" 2>/dev/null | grep -v '^$')" ]] && return 0
[[ -n "$(ls /sys/block/"$(basename "$disk")"/holders/ 2>/dev/null)" ]] && return 0
return 1
}
# List available disks (not in use)
list_available_disks() {
local disks=()
for disk in /dev/nvme[0-9]n[0-9] /dev/sd[a-z] /dev/vd[a-z]; do
[[ -b "$disk" ]] || continue
disk_in_use "$disk" && continue
local size=$(get_disk_size "$disk")
local model=$(get_disk_model "$disk")
disks+=("$disk ($size, $model)")
done
printf '%s\n' "${disks[@]}"
}
|
