#!/bin/bash
set -e

install_() {
    install -m 755 china-block.sh   /usr/local/sbin/china-block.sh
    install -m 755 china-unblock.sh /usr/local/sbin/china-unblock.sh
    install -m 644 china-block.service /etc/systemd/system/china-block.service
    systemctl daemon-reload
    systemctl enable --now china-block
    echo "[+] china-block service installed and started."
}

uninstall() {
    systemctl disable --now china-block 2>/dev/null || true
    rm -f /usr/local/sbin/china-block.sh
    rm -f /usr/local/sbin/china-unblock.sh
    rm -f /etc/systemd/system/china-block.service
    systemctl daemon-reload
    echo "[+] china-block service removed."
}

case "${1:-install}" in
    install)   install_ ;;
    uninstall) uninstall ;;
    *) echo "Usage: $0 [install|uninstall]"; exit 1 ;;
esac