pigway-device-agent/install.sh

#!/bin/bash
set -euo pipefail
cd "$(dirname "$0")"

for required in app/pigway_pi_control.py app/oled_font_5x7.bin config/pigway-pi-control.conf systemd/pigway-pi-control.service; do
  if [ ! -f "$required" ]; then
    echo "ERROR: required file missing: $required" >&2
    exit 1
  fi
done

echo "[1/8] 检查 Raspberry Pi I2C..."
[ -e /dev/i2c-1 ] || { echo "错误：/dev/i2c-1 不存在。先 sudo raspi-config -> Interface Options -> I2C -> Enable，然后重启。"; exit 1; }

echo "[2/8] 等待 APT/DPKG..."
for i in $(seq 1 60); do
  if ! fuser /var/lib/dpkg/lock-frontend /var/lib/apt/lists/lock /var/cache/apt/archives/lock >/dev/null 2>&1; then break; fi
  [ "$i" = 60 ] && { echo "APT 锁等待超时，请稍后重试。"; exit 1; }
  echo "软件包管理器忙，等待 5 秒... ($i/60)"
  sleep 5
done

echo "[3/8] 安装依赖..."
apt-get update
apt-get install -y python3 python3-pil python3-smbus2 i2c-tools

echo "[4/8] 检查硬件地址..."
SCAN="$(i2cdetect -y 1)"; echo "$SCAN"
echo "$SCAN" | grep -qi '3c' || echo "警告：未扫描到 OLED 0x3C"
echo "$SCAN" | grep -qi '0d' || echo "警告：未扫描到控制 MCU 0x0D"

echo "[5/8] 安装程序与配置..."
if systemctl list-unit-files 2>/dev/null | grep '^temp-control.service' >/dev/null; then
  echo "检测到旧 temp-control.service，正在停止并禁用..."
  systemctl disable --now temp-control.service 2>/dev/null || true
fi
pkill -f '/temp_control/temp_control' 2>/dev/null || true
mkdir -p /usr/local/share/pigway-pi-control
install -m 0644 app/oled_font_5x7.bin /usr/local/share/pigway-pi-control/oled_font_5x7.bin

install -m 0755 app/pigway_pi_control.py /usr/local/sbin/pigway-pi-control

# Keep the complete existing configuration, including comments and custom sections.
# New options use runtime defaults until the user chooses to add them.
if [ -f /etc/pigway-pi-control.conf ]; then
  cp /etc/pigway-pi-control.conf "/etc/pigway-pi-control.conf.bak.$(date +%Y%m%d-%H%M%S)"
else
  install -m 0644 config/pigway-pi-control.conf /etc/pigway-pi-control.conf
fi

# Preserve the existing file verbatim while exposing newly configurable RGB
# defaults. Missing keys are inserted into [led]; existing user values win.
python3 - /etc/pigway-pi-control.conf <<'PY'
import re,sys
from pathlib import Path

path=Path(sys.argv[1])
text=path.read_text()
defaults=(
    ("cpu_r",255),("cpu_g",0),("cpu_b",0),
    ("power_r",255),("power_g",40),("power_b",0),
    ("memory_r",255),("memory_g",0),("memory_b",255),
    ("storage_r",255),("storage_g",255),("storage_b",255),
    ("network_r",0),("network_g",0),("network_b",255),
    ("service_r",0),("service_g",255),("service_b",96),
    ("warning_flash_on_ms",1500),("warning_flash_off_ms",1500),
    ("critical_flash_on_ms",750),("critical_flash_off_ms",750),
    ("emergency_flash_on_ms",300),("emergency_flash_off_ms",300),
)
missing=[(key,value) for key,value in defaults
         if not re.search(rf"(?m)^\s*{re.escape(key)}\s*=",text)]
if missing:
    block="\n# RGB object colors (0..255) and flash timing (milliseconds).\n"
    block+="".join(f"{key} = {value}\n" for key,value in missing)
    match=re.search(r"(?m)^\[led\]\s*$",text)
    if match:
        next_section=re.search(r"(?m)^\[.+\]\s*$",text[match.end():])
        pos=match.end()+(next_section.start() if next_section else len(text)-match.end())
        text=text[:pos].rstrip()+"\n"+block+"\n"+text[pos:].lstrip("\n")
    else:
        text=text.rstrip()+"\n\n[led]\n"+block
    path.write_text(text)
PY
install -m 0644 systemd/pigway-pi-control.service /etc/systemd/system/pigway-pi-control.service

echo "[6/8] 启用/重启服务..."
systemctl daemon-reload
systemctl enable pigway-pi-control.service >/dev/null
systemctl restart pigway-pi-control.service

echo "[7/8] 安装后自检..."
FAIL=0
check() { local label="$1"; shift; if "$@"; then printf "  %-30s PASS\n" "$label"; else printf "  %-30s FAIL\n" "$label"; FAIL=1; fi; }
APP=/usr/local/sbin/pigway-pi-control
check "Version 3.5.0" grep -Fq 'APP_VERSION="3.5.0"' "$APP"
check "OLED physical 128x32" grep -Fq 'W,H=128,32' "$APP"
check "OLED MUX 0x1F" grep -Fq '0xA8,0x1F' "$APP"
check "OLED pages 0..3" grep -Fq 'cmd(0x22);cmd(0);cmd(3)' "$APP"
check "UI CPU/TMP/DSK" grep -Fq 'f"TMP {temp:.1f}C"' "$APP"
check "UI IP4/IP6" grep -Fq 'ip_label' "$APP"
check "UI right column x=72" grep -Fq 'draw_text_5x7(pix,72,y,right[row])' "$APP"
legacy_stopped() { ! systemctl is-active --quiet temp-control.service; }
check "Legacy service stopped" legacy_stopped
check "Service running" systemctl is-active --quiet pigway-pi-control.service
check "P0 health monitor" grep -Fq 'throttled_state' "$APP"
check "Diagnostic command" grep -Fq -- '--diagnose' "$APP"
if [ "$FAIL" -ne 0 ]; then echo "错误：安装后自检未全部通过。"; exit 1; fi
echo "[8/8] 状态..."


systemctl --no-pager --full status pigway-pi-control.service || true
echo
echo "完成。程序：/usr/local/sbin/pigway-pi-control"
echo "配置：/etc/pigway-pi-control.conf"