288 lines
6.8 KiB
Bash
288 lines
6.8 KiB
Bash
#!/bin/bash
|
||
# ========================================
|
||
# 智能矿山标签监测系统 - Git初始化脚本
|
||
# ========================================
|
||
# 用于在已部署的服务器上初始化Git仓库，
|
||
# 以便后续使用 upgrade.sh 进行升级
|
||
# ========================================
|
||
|
||
set -e
|
||
|
||
# 颜色定义
|
||
RED='\033[0;31m'
|
||
GREEN='\033[0;32m'
|
||
YELLOW='\033[1;33m'
|
||
BLUE='\033[0;34m'
|
||
NC='\033[0m'
|
||
|
||
info() {
|
||
echo -e "${BLUE}[INFO]${NC} $1"
|
||
}
|
||
|
||
success() {
|
||
echo -e "${GREEN}[SUCCESS]${NC} $1"
|
||
}
|
||
|
||
warn() {
|
||
echo -e "${YELLOW}[WARN]${NC} $1"
|
||
}
|
||
|
||
error() {
|
||
echo -e "${RED}[ERROR]${NC} $1"
|
||
}
|
||
|
||
# Git仓库地址
|
||
GIT_REPO="http://10.224.35.147:3000/redline/program.git"
|
||
|
||
# 获取项目目录
|
||
PROJECT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||
|
||
echo "======================================"
|
||
echo " Git仓库初始化脚本"
|
||
echo "======================================"
|
||
echo ""
|
||
echo "项目目录: $PROJECT_DIR"
|
||
echo "远程仓库: $GIT_REPO"
|
||
echo ""
|
||
|
||
# 检查是否已安装Git
|
||
if ! command -v git &> /dev/null; then
|
||
error "Git未安装"
|
||
echo "请先安装Git:"
|
||
echo " Ubuntu/Debian: sudo apt-get install -y git"
|
||
echo " CentOS/RHEL: sudo yum install -y git"
|
||
exit 1
|
||
fi
|
||
|
||
info "Git版本: $(git --version)"
|
||
echo ""
|
||
|
||
# 检查是否已经是Git仓库
|
||
if git -C "$PROJECT_DIR" rev-parse --is-inside-work-tree &> /dev/null; then
|
||
warn "当前目录已经是Git仓库"
|
||
echo ""
|
||
echo "当前状态:"
|
||
echo " 分支: $(git -C "$PROJECT_DIR" rev-parse --abbrev-ref HEAD 2>/dev/null || echo 'unknown')"
|
||
echo " 提交: $(git -C "$PROJECT_DIR" rev-parse --short HEAD 2>/dev/null || echo 'unknown')"
|
||
echo " 远程: $(git -C "$PROJECT_DIR" remote get-url origin 2>/dev/null || echo '未配置')"
|
||
echo ""
|
||
|
||
read -p "是否重新初始化? (会保留现有数据) (y/N) " -n 1 -r
|
||
echo
|
||
if [[ ! $REPLY =~ ^[Yy]$ ]]; then
|
||
info "操作已取消"
|
||
exit 0
|
||
fi
|
||
else
|
||
info "开始初始化Git仓库..."
|
||
fi
|
||
|
||
# 步骤1: 创建.gitignore（如果不存在）
|
||
info "检查.gitignore..."
|
||
if [ ! -f "${PROJECT_DIR}/.gitignore" ]; then
|
||
cat > "${PROJECT_DIR}/.gitignore" << 'GITIGNORE'
|
||
# Python
|
||
__pycache__/
|
||
*.py[cod]
|
||
*$py.class
|
||
*.so
|
||
.Python
|
||
venv/
|
||
env/
|
||
.venv/
|
||
|
||
# 数据库
|
||
*.db-journal
|
||
*.db-wal
|
||
|
||
# 环境变量（敏感信息）
|
||
.env
|
||
|
||
# IDE
|
||
.vscode/
|
||
.idea/
|
||
*.swp
|
||
*.swo
|
||
*~
|
||
|
||
# 系统文件
|
||
.DS_Store
|
||
Thumbs.db
|
||
|
||
# 日志
|
||
*.log
|
||
logs/
|
||
|
||
# 备份
|
||
backups/
|
||
|
||
# 临时文件
|
||
.upgrade_staging/
|
||
|
||
# 编译产物
|
||
*.egg-info/
|
||
dist/
|
||
build/
|
||
*.egg
|
||
|
||
# 测试
|
||
.pytest_cache/
|
||
.coverage
|
||
htmlcov/
|
||
GITIGNORE
|
||
|
||
success "已创建.gitignore"
|
||
else
|
||
info ".gitignore已存在"
|
||
fi
|
||
|
||
echo ""
|
||
|
||
# 步骤2: 初始化Git仓库（如果需要）
|
||
if ! git -C "$PROJECT_DIR" rev-parse --is-inside-work-tree &> /dev/null; then
|
||
info "初始化Git仓库..."
|
||
cd "$PROJECT_DIR"
|
||
git init
|
||
success "Git仓库已初始化"
|
||
else
|
||
info "Git仓库已存在"
|
||
fi
|
||
|
||
echo ""
|
||
|
||
# 步骤3: 配置远程仓库
|
||
info "配置远程仓库..."
|
||
cd "$PROJECT_DIR"
|
||
|
||
if git remote -v | grep -q "origin" 2>/dev/null; then
|
||
local CURRENT_REMOTE=$(git remote get-url origin)
|
||
if [ "$CURRENT_REMOTE" != "$GIT_REPO" ]; then
|
||
info "更新远程仓库地址: $CURRENT_REMOTE -> $GIT_REPO"
|
||
git remote set-url origin "$GIT_REPO"
|
||
else
|
||
info "远程仓库地址已是: $GIT_REPO"
|
||
fi
|
||
else
|
||
git remote add origin "$GIT_REPO"
|
||
success "已添加远程仓库: origin"
|
||
fi
|
||
|
||
echo ""
|
||
|
||
# 步骤4: 拉取最新代码
|
||
info "从远程仓库拉取最新代码..."
|
||
|
||
# 先获取最新代码
|
||
git fetch origin 2>&1 || {
|
||
error "无法从远程仓库拉取代码"
|
||
echo "请检查:"
|
||
echo " 1. 网络连接是否正常"
|
||
echo " 2. Git仓库地址是否正确: $GIT_REPO"
|
||
echo " 3. 是否有访问权限"
|
||
exit 1
|
||
}
|
||
|
||
# 检查是否有本地修改
|
||
if ! git diff --quiet &>/dev/null || ! git status --porcelain | grep -q "^?? " &>/dev/null; then
|
||
warn "检测到本地修改或未跟踪的文件"
|
||
echo ""
|
||
echo "本地状态:"
|
||
git status --short
|
||
echo ""
|
||
|
||
# 保存本地配置
|
||
info "保存本地配置文件..."
|
||
local STAGING_DIR="${PROJECT_DIR}/.init_staging"
|
||
mkdir -p "$STAGING_DIR"
|
||
|
||
for file in "config.py" "mining_data.db" ".env"; do
|
||
if [ -f "${PROJECT_DIR}/${file}" ]; then
|
||
cp -f "${PROJECT_DIR}/${file}" "${STAGING_DIR}/${file}"
|
||
info " 已保存: $file"
|
||
fi
|
||
done
|
||
|
||
# 保存outputs目录
|
||
if [ -d "${PROJECT_DIR}/outputs" ]; then
|
||
cp -r "${PROJECT_DIR}/outputs" "${STAGING_DIR}/outputs" 2>/dev/null || true
|
||
info " 已保存: outputs/"
|
||
fi
|
||
|
||
# stash所有本地修改
|
||
git stash --include-untracked 2>/dev/null || true
|
||
fi
|
||
|
||
# 检查是否是首次拉取（空仓库）
|
||
if git log --oneline master &>/dev/null; then
|
||
# 已有本地提交，尝试合并
|
||
info "合并远程代码..."
|
||
git checkout master 2>/dev/null || git checkout -b master
|
||
|
||
# 如果是初次合并，可能需要强制重置
|
||
if ! git merge --abort 2>/dev/null; then
|
||
git reset --hard origin/master 2>/dev/null || \
|
||
git checkout master -- . 2>/dev/null || true
|
||
fi
|
||
else
|
||
# 首次拉取，设置本地master到远程master
|
||
info "设置本地master分支跟踪远程..."
|
||
git checkout -B master origin/master 2>/dev/null || \
|
||
git checkout master 2>/dev/null || true
|
||
fi
|
||
|
||
# 恢复本地配置
|
||
if [ -d "$STAGING_DIR" ]; then
|
||
info "恢复本地配置文件..."
|
||
|
||
for file in "config.py" "mining_data.db" ".env"; do
|
||
if [ -f "${STAGING_DIR}/${file}" ]; then
|
||
cp -f "${STAGING_DIR}/${file}" "${PROJECT_DIR}/${file}"
|
||
info " 已恢复: $file"
|
||
fi
|
||
done
|
||
|
||
if [ -d "${STAGING_DIR}/outputs" ]; then
|
||
mkdir -p "${PROJECT_DIR}/outputs"
|
||
cp -r "${STAGING_DIR}/outputs"/* "${PROJECT_DIR}/outputs/" 2>/dev/null || true
|
||
info " 已恢复: outputs/"
|
||
fi
|
||
|
||
rm -rf "$STAGING_DIR"
|
||
fi
|
||
|
||
success "代码拉取完成"
|
||
|
||
echo ""
|
||
|
||
# 步骤5: 显示当前状态
|
||
info "当前Git状态:"
|
||
echo " 分支: $(git rev-parse --abbrev-ref HEAD)"
|
||
echo " 提交: $(git rev-parse --short HEAD)"
|
||
echo " 时间: $(git log -1 --format='%ci')"
|
||
echo " 信息: $(git log -1 --format='%s')"
|
||
echo ""
|
||
|
||
# 步骤6: 确保upgrade.sh可执行
|
||
if [ -f "${PROJECT_DIR}/upgrade.sh" ]; then
|
||
chmod +x "${PROJECT_DIR}/upgrade.sh"
|
||
success "upgrade.sh 已设置为可执行"
|
||
else
|
||
warn "upgrade.sh 不存在，请检查代码是否完整"
|
||
fi
|
||
|
||
echo ""
|
||
|
||
# 完成
|
||
echo "======================================"
|
||
echo -e "${GREEN} Git初始化完成!${NC}"
|
||
echo "======================================"
|
||
echo ""
|
||
echo "后续升级只需执行:"
|
||
echo ""
|
||
echo " bash upgrade.sh # 升级到最新版本"
|
||
echo " bash upgrade.sh status # 查看版本状态"
|
||
echo " bash upgrade.sh rollback # 回滚到上一版本"
|
||
echo ""
|
||
echo "首次升级前建议先执行备份:"
|
||
echo " bash upgrade.sh backup"
|
||
echo "" |