Files
tools/linux/repartition_disks.sh
Wang Defa b34e1ef872 feat: 为所有脚本添加远程库加载支持
## 功能概述

实现了智能远程库加载机制,允许脚本从远程 Gitea 仓库动态下载公共库,
同时保留本地库优先的策略,确保脚本可以独立运行而不依赖本地项目结构。

## 变更内容

### 新增文件

- **common/remote_loader.sh**: 核心远程加载器
  - 支持 curl/wget 双重下载方式
  - 临时文件自动清理机制
  - 环境变量可配置仓库 URL

- **examples/remote_example.sh**: 纯远程模式示例
  - 演示完全独立的远程加载脚本
  - 适合单文件分发场景

- **examples/hybrid_loader_template.sh**: 混合模式模板
  - 本地优先 + 远程回退策略
  - 适合仓库内脚本

- **examples/REMOTE_LOADER_README.md**: 完整使用文档
  - 三种使用模式详解
  - 故障排除指南
  - 最佳实践建议

### 更新脚本(版本 2.1.0)

所有主要脚本均升级到 2.1.0 版本,支持智能库加载:

- **linux/create_raid0_array.sh**
- **linux/repartition_disks.sh**
- **linux/install_oh_my_zsh.sh**
- **gcp/create_ai_projects.sh**
- **gcp/delete_all_projects.sh**
- **oci/create_instance.sh**
- **common/demo_usage.sh**

## 加载策略

1. 检查 FORCE_REMOTE=1 环境变量 → 强制远程
2. 检查本地库存在性 → 优先使用本地
3. 本地不存在 → 自动回退远程
4. 远程失败 → 友好错误提示

## 使用示例

### 本地模式(默认)
```bash
./linux/create_raid0_array.sh
```

### 强制远程模式
```bash
FORCE_REMOTE=1 ./linux/create_raid0_array.sh
```

### 自定义仓库 URL
```bash
REMOTE_LIB_URL=https://example.com/repo ./script.sh
```

## 技术特性

-  向后兼容:默认使用本地库,行为不变
-  离线可用:本地库存在时无需网络
-  独立分发:支持单文件脚本分发
-  错误处理:详细的错误信息和诊断建议
-  环境可配:REMOTE_LIB_URL 和 FORCE_REMOTE
-  自动清理:临时文件通过 trap 自动删除
2025-12-26 15:00:06 +08:00

152 lines
4.7 KiB
Bash
Raw Blame History

This file contains ambiguous Unicode characters
This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.
#!/bin/bash
# ============================================================================
# 文件名: repartition_disks.sh
# 描述: 删除所有现有分区并创建新的 GPT 分区表(除了系统盘)
# 作者: Cloud Tools Project
# 版本: 2.1.0(支持远程库加载)
# 警告: 此操作将删除目标磁盘上的所有数据
# ============================================================================
set -euo pipefail
# ============================================================================
# 远程库加载配置
# ============================================================================
# 远程仓库 URL可通过环境变量覆盖
readonly REMOTE_BASE_URL="${REMOTE_LIB_URL:-https://gitea.bcde.io/wangdefa/tools/raw/branch/main}"
# 获取脚本目录(用于本地加载)
readonly SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
readonly PROJECT_ROOT="$(dirname "$SCRIPT_DIR")"
#
# 智能加载公共库
#
# 加载策略:
# 1. 如果 FORCE_REMOTE=1强制使用远程库
# 2. 否则尝试使用本地库
# 3. 本地库不存在时自动回退到远程库
#
load_common_libs() {
local use_remote=false
# 检查是否强制远程
if [[ "${FORCE_REMOTE:-0}" == "1" ]]; then
echo "[INFO] 强制使用远程库 (FORCE_REMOTE=1)" >&2
use_remote=true
# 检查本地库是否存在
elif [[ -f "${PROJECT_ROOT}/common/logging.sh" ]] && [[ -f "${PROJECT_ROOT}/common/error_handler.sh" ]]; then
# shellcheck disable=SC1091
source "${PROJECT_ROOT}/common/logging.sh"
# shellcheck disable=SC1091
source "${PROJECT_ROOT}/common/error_handler.sh"
return 0
else
echo "[WARN] 本地库不存在,使用远程库" >&2
use_remote=true
fi
# 使用远程库
if [[ "$use_remote" == "true" ]]; then
if command -v curl &>/dev/null; then
echo "[INFO] 使用 curl 下载远程库..." >&2
# shellcheck disable=SC1090
if source <(curl -fsSL "${REMOTE_BASE_URL}/common/remote_loader.sh" 2>/dev/null); then
return 0
fi
elif command -v wget &>/dev/null; then
echo "[INFO] 使用 wget 下载远程库..." >&2
# shellcheck disable=SC1090
if source <(wget -qO- "${REMOTE_BASE_URL}/common/remote_loader.sh" 2>/dev/null); then
return 0
fi
fi
echo "[ERROR] 无法加载公共库" >&2
echo "[ERROR] - 本地库不存在" >&2
echo "[ERROR] - 远程下载失败(需要 curl 或 wget" >&2
echo "[ERROR] - 仓库 URL: ${REMOTE_BASE_URL}" >&2
exit 1
fi
}
# 加载公共库
load_common_libs
# 检查 root 权限
check_root
#
# 检查并安装依赖工具
#
# 参数:
# $1 - 工具名称
#
check_and_install() {
local tool=$1
if ! command -v "$tool" &> /dev/null; then
log_warning "$tool 未安装,正在尝试安装..."
if command -v apt-get &> /dev/null; then
run_command "安装 $tool" bash -c "apt-get update && apt-get install -y $tool"
elif command -v yum &> /dev/null; then
run_command "安装 $tool" yum install -y "$tool"
else
log_error "无法安装 $tool,请手动安装后重试"
exit 1
fi
fi
}
#
# 主函数 - 重新分区所有非系统盘
#
main() {
log_info "============ 磁盘重新分区程序 ============"
log_info "版本: 2.1.0 (支持远程库加载)"
# 检查依赖
check_and_install lsblk
check_and_install parted
check_and_install wipefs
log_success "所有依赖已满足"
# 获取系统盘
local system_disk
system_disk=$(mount | grep " / " | cut -d' ' -f1 | sed 's/[0-9]*//g')
log_info "检测到系统盘: ${system_disk}"
# 获取可用磁盘
local disks
disks=$(lsblk -dno NAME | sed -e '/^loop/d' -e '/^sr/d' -e "/^${system_disk##*/}$/d")
if [ -z "$disks" ]; then
log_warning "没有找到除系统盘以外的硬盘"
exit 0
fi
# 处理每个磁盘
for disk in $disks; do
log_info "正在处理硬盘: /dev/${disk}"
log_warning "删除 /dev/${disk} 上的所有分区..."
run_command "清除分区签名" wipefs -a "/dev/${disk}"
log_warning "在 /dev/${disk} 上创建新的 GPT 分区表..."
run_command "创建分区表" parted -s "/dev/${disk}" mklabel gpt
log_warning "在 /dev/${disk} 上创建新分区..."
run_command "创建分区" parted -s "/dev/${disk}" mkpart primary 0% 100%
log_info "刷新分区表..."
run_command "刷新分区" partprobe "/dev/${disk}"
log_success "完成处理 /dev/${disk}"
echo
done
log_success "所有操作完成"
}
# 执行主函数
main "$@"