chore: add GitHub Actions release workflow, MIT license, and update gitignore

每次写日志时，都重新根据当前日期确定日志文件名

.github/workflows/release.yml

@@ -0,0 +1,53 @@
+name: Release
+
+on:
+  push:
+    tags:
+      - 'v*' # 当推送标签形如 v1.0.0 时触发自动构建
+
+permissions:
+  contents: write
+
+jobs:
+  build:
+    name: Build & Release
+    runs-on: ubuntu-latest
+
+    steps:
+      - name: Checkout Code
+        uses: actions/checkout@v4
+
+      - name: Set up Go
+        uses: actions/setup-go@v5
+        with:
+          go-version: '1.21'
+          cache: true
+
+      - name: Download Dependencies
+        run: go mod download
+
+      - name: Compile Cross-Platform Binaries
+        run: |
+          mkdir -p build_output
+          
+          echo "Building Windows amd64..."
+          GOOS=windows GOARCH=amd64 go build -ldflags="-s -w" -o build_output/smart-monitor_windows_amd64.exe ./cmd/smart-monitor
+          
+          echo "Building Windows arm64..."
+          GOOS=windows GOARCH=arm64 go build -ldflags="-s -w" -o build_output/smart-monitor_windows_arm64.exe ./cmd/smart-monitor
+          
+          echo "Building Linux amd64..."
+          GOOS=linux GOARCH=amd64 go build -ldflags="-s -w" -o build_output/smart-monitor_linux_amd64 ./cmd/smart-monitor
+          
+          echo "Building Linux arm64..."
+          GOOS=linux GOARCH=arm64 go build -ldflags="-s -w" -o build_output/smart-monitor_linux_arm64 ./cmd/smart-monitor
+
+      - name: Publish GitHub Release
+        uses: softprops/action-gh-release@v1
+        with:
+          files: build_output/*
+          draft: false
+          prerelease: false
+          generate_release_notes: true
+        env:
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}

.gitignore

@@ -0,0 +1,27 @@
+# Binaries for programs and plugins
+*.exe
+*.exe~
+*.dll
+*.so
+*.dylib
+
+# Test binary, built with `go test -c`
+*.test
+
+# Output directory for cross-platform builds
+/bin/
+/build/
+smart-monitor
+smart-monitor.exe
+
+# Output of the go coverage tool
+*.out
+
+# IDE configs
+.idea/
+.vscode/
+
+# Local logs or configuration testing directories
+logs/
+*.log
+config.json

LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 UnbalancedCat
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

README.md

@@ -1,122 +0,0 @@
-# 智能网络监控脚本 - 系统部署版
-
-## 📁 文件结构
-
-```
-smart_shutdown/
-├── smart_shutdown.ps1          # 原始监控脚本（用户模式）
-├── deploy_system.ps1           # 系统级部署脚本
-├── uninstall_system.ps1        # 系统级卸载脚本
-├── TEST_REPORT.md              # 测试报告
-├── README.md                   # 本文件
-└── logs/                       # 本地测试日志目录
-    └── network_monitor_20250606.log
-```
-
-## 🚀 快速部署
-
-### 系统级部署（推荐）
-```powershell
-# 以管理员身份运行
-.\deploy_system.ps1
-```
-
-**部署效果：**
-- 程序部署到：`C:\Program Files\SmartNetworkMonitor\`
-- 配置和日志：`C:\ProgramData\SmartNetworkMonitor\`
-- 开机自启动，无需用户登录
-- 以 SYSTEM 权限运行
-
-### 用户级运行
-```powershell
-# 以管理员身份运行（需要用户登录）
-.\smart_shutdown.ps1
-```
-
-## 🎯 系统部署后的文件位置
-
-### 程序文件位置
-```
-C:\Program Files\SmartNetworkMonitor\
-├── smart_shutdown_system.ps1   # 系统优化版监控脚本
-└── manage.ps1                  # 管理工具脚本
-```
-
-### 数据文件位置
-```
-C:\ProgramData\SmartNetworkMonitor\
-├── config.json                 # 配置文件
-└── logs\                       # 系统日志目录
-    └── network_monitor_YYYYMMDD.log
-```
-
-## 🔧 管理命令
-
-### 查看任务状态
-```powershell
-Get-ScheduledTask -TaskName "Smart Network Shutdown Monitor"
-```
-
-### 启动/停止监控
-```powershell
-Start-ScheduledTask -TaskName "Smart Network Shutdown Monitor"
-Stop-ScheduledTask -TaskName "Smart Network Shutdown Monitor"
-```
-
-### 查看日志
-```powershell
-# 查看今天的日志
-Get-Content "C:\ProgramData\SmartNetworkMonitor\logs\network_monitor_$(Get-Date -Format 'yyyyMMdd').log"
-
-# 实时监控日志
-Get-Content "C:\ProgramData\SmartNetworkMonitor\logs\network_monitor_$(Get-Date -Format 'yyyyMMdd').log" -Wait
-```
-
-### 查看系统事件日志
-```powershell
-Get-EventLog -LogName Application -Source SmartNetworkMonitor -Newest 20
-```
-
-## ⚙️ 配置说明
-
-系统部署后，配置文件位于：`C:\ProgramData\SmartNetworkMonitor\config.json`
-
-默认配置：
-```json
-{
-    "TargetIP": "192.168.3.3",
-    "MonitorWindowSeconds": 180,
-    "ShutdownCountdown": 60,
-    "NormalPingInterval": 15
-}
-```
-
-修改配置文件后，重启监控任务使配置生效：
-```powershell
-Stop-ScheduledTask -TaskName "Smart Network Shutdown Monitor"
-Start-ScheduledTask -TaskName "Smart Network Shutdown Monitor"
-```
-
-## 🗑️ 卸载
-
-```powershell
-# 以管理员身份运行
-.\uninstall_system.ps1
-```
-
-## 📊 特性
-
-- ✅ 开机自启动，无需用户登录
-- ✅ 系统级权限运行
-- ✅ 智能日志管理
-- ✅ 配置文件支持
-- ✅ 事件日志备份
-- ✅ 自动日志清理（30天）
-- ✅ 网络恢复自动取消关机
-- ✅ 完整的错误处理
-
-## 🔐 安全说明
-
-- 脚本以 SYSTEM 权限运行，具有执行关机的完整权限
-- 所有操作都会记录在日志和系统事件日志中
-- 支持通过任务计划程序进行管理和监控

TEST_REPORT.md

@@ -1,113 +0,0 @@
-# 智能网络监控脚本测试报告
-
-## 测试日期
-2025年6月6日
-
-## 脚本概述
-这是一个PowerShell智能网络监控脚本，用于监控指定IP的网络连通性，当网络持续中断超过设定时间后自动关闭计算机。
-
-## 主要功能
-1. **网络连通性监控** - 定期ping目标IP地址
-2. **智能故障检测** - 区分临时网络波动和持续故障
-3. **可配置监控窗口** - 防止因短暂网络中断导致误关机
-4. **关机倒计时机制** - 提供最后的取消机会
-5. **完整日志记录** - 记录所有操作和状态变化
-6. **手动取消功能** - 用户可以随时按键取消关机
-
-## 配置参数（原版）
-- **目标IP**: 192.168.3.3
-- **正常监控间隔**: 15秒
-- **监控窗口**: 180秒（3分钟）
-- **关机倒计时**: 60秒
-- **倒计时ping间隔**: 3秒
-- **Ping超时**: 3秒
-
-## 测试配置（测试版）
-- **目标IP**: 192.168.3.99（不存在的IP，模拟网络故障）
-- **正常监控间隔**: 3秒
-- **监控窗口**: 15秒
-- **关机倒计时**: 10秒
-- **倒计时ping间隔**: 2秒
-
-## 修复的问题
-1. **兼容性问题**: 修复了`Test-Connection`命令在旧版PowerShell中不支持`-TimeoutSeconds`参数的问题
-2. **编码问题**: 将特殊Unicode字符（✓ ✗ ℹ）替换为ASCII字符，避免控制台显示乱码
-
-## 测试结果
-
-### ✅ 功能测试全部通过
-
-#### 1. 脚本启动功能
-- ✅ 正常启动和初始化
-- ✅ 日志系统正常工作
-- ✅ 配置参数正确加载
-- ✅ 本地IP获取功能正常
-
-#### 2. 网络监控功能
-- ✅ 正常IP连通性检测（如8.8.8.8）
-- ✅ 故障IP检测（如192.168.3.99）
-- ✅ 网络异常处理机制
-- ✅ Ping命令兼容性修复
-
-#### 3. 监控窗口机制
-- ✅ 首次网络中断检测
-- ✅ 监控窗口计时功能
-- ✅ 剩余时间计算和显示
-- ✅ 监控窗口超时触发
-
-#### 4. 关机倒计时功能
-- ✅ 倒计时启动机制
-- ✅ 倒计时期间网络快速检测
-- ✅ 按键手动取消功能
-- ✅ 模拟关机操作
-
-#### 5. 日志记录功能
-- ✅ 日志目录自动创建
-- ✅ 按日期分类的日志文件
-- ✅ 完整的操作记录
-- ✅ 不同级别的日志分类（INFO、WARN、CRITICAL、FAIL、SUCCESS）
-
-#### 6. 用户界面和显示
-- ✅ 彩色控制台输出
-- ✅ 清晰的状态提示
-- ✅ 编码问题修复
-- ✅ 用户友好的信息显示
-
-## 实际测试执行记录
-
-### 测试1（13:02:43开始）
-- 监控窗口：15秒
-- 实际触发时间：18秒后进入关机倒计时
-- 关机倒计时：10秒
-- 结果：正常完成模拟关机，重置状态继续监控
-
-### 测试2（13:08:30开始）
-- 同样配置重复测试
-- 确认功能稳定性和一致性
-- 编码问题已完全修复
-
-## 实际使用建议
-
-### 安全配置
-1. **取消注释关机命令**：在`smart_shutdown.ps1`第217行，将`# Stop-Computer -Force`的注释去掉
-2. **以管理员身份运行**：脚本需要管理员权限执行关机命令
-3. **测试配置**：建议先用较短时间参数测试，确认无误后再使用正式配置
-
-### 最佳实践
-1. **监控窗口设置**：建议设置为3-5分钟，避免网络短暂波动造成误关机
-2. **目标IP选择**：建议使用网关IP或重要服务器IP
-3. **日志监控**：定期检查日志文件，了解网络状况
-4. **备用方案**：考虑配置多个目标IP进行冗余检测
-
-## 结论
-**脚本功能完整，测试全部通过，可以安全投入使用。**
-
-经过全面测试，智能网络监控脚本各项功能均正常工作：
-- 网络监控准确可靠
-- 故障检测机制有效
-- 用户交互友好
-- 日志记录完整
-- 编码问题已解决
-- 兼容性问题已修复
-
-脚本已准备好在生产环境中使用，能够有效监控网络连接并在必要时自动关闭计算机。

cmd/smart-monitor/main.go

@@ -0,0 +1,123 @@
+package main
+
+import (
+	"fmt"
+	"os"
+
+	"github.com/kardianos/service"
+	"github.com/spf13/cobra"
+	"smart-shutdown/pkg/config"
+	"smart-shutdown/pkg/daemon"
+	"smart-shutdown/pkg/logger"
+)
+
+func main() {
+	// 初始化全局系统日志
+	if err := logger.InitLogger(); err != nil {
+		fmt.Printf("无法初始化日志系统: %v\n", err)
+		os.Exit(1)
+	}
+
+	// 初始化配置加载
+	cfg, err := config.LoadConfig()
+	if err != nil {
+		logger.Crit("读取配置失败: %v", err)
+		os.Exit(1)
+	}
+
+	// 注册为跨平台系统服务
+	svc, err := daemon.GetService(cfg)
+	if err != nil {
+		logger.Crit("构建服务对象失败: %v", err)
+		os.Exit(1)
+	}
+
+	// 定义根命令，无后续子命令会直接触发
+	var rootCommand = &cobra.Command{
+		Use:   "smart-monitor",
+		Short: "智能网络监控自动关机",
+		Run: func(cmd *cobra.Command, args []string) {
+			// 如果没有输入子命令，尝试作为服务直接前台或者后台跑起来
+			// 这个分支也是系统启动服务（systemd/services.msc）自动拉起进程时的必然入口
+			err := svc.Run()
+			if err != nil {
+				logger.Fail("运行抛出异常: %v", err)
+			}
+		},
+	}
+
+	// 系统服务管理一键控制逻辑，使用 kardianos/service 提供的内置 Control方法
+	cmds := []*cobra.Command{
+		{
+			Use:   "install",
+			Short: "将本程序装载并注册为系统常驻服务 (例如 systemd / windows registry)",
+			Run: func(cmd *cobra.Command, args []string) {
+				handleServiceControl(svc, "install")
+			},
+		},
+		{
+			Use:   "uninstall",
+			Short: "从系统中彻底卸载此后台监控服务",
+			Run: func(cmd *cobra.Command, args []string) {
+				handleServiceControl(svc, "uninstall")
+			},
+		},
+		{
+			Use:   "start",
+			Short: "令已注册的系统服务开始运行",
+			Run: func(cmd *cobra.Command, args []string) {
+				handleServiceControl(svc, "start")
+			},
+		},
+		{
+			Use:   "stop",
+			Short: "停止系统服务",
+			Run: func(cmd *cobra.Command, args []string) {
+				handleServiceControl(svc, "stop")
+			},
+		},
+		{
+			Use:   "restart",
+			Short: "重启系统服务",
+			Run: func(cmd *cobra.Command, args []string) {
+				handleServiceControl(svc, "restart")
+			},
+		},
+		{
+			Use:   "status",
+			Short: "查询服务存活进程状态",
+			Run: func(cmd *cobra.Command, args []string) {
+				status, err := svc.Status()
+				if err != nil {
+					fmt.Printf("❌ 获取服务状态失败: %v\n", err)
+					return
+				}
+				switch status {
+				case service.StatusRunning:
+					fmt.Println("✅ 服务存活 [运行中]")
+				case service.StatusStopped:
+					fmt.Println("💤 服务处于 [已停止] 状态")
+				default:
+					fmt.Println("❓ 服务尚未在本机注册或者状态未知。请确认你是否执行过 smart-monitor install")
+				}
+			},
+		},
+	}
+
+	for _, c := range cmds {
+		rootCommand.AddCommand(c)
+	}
+
+	if err := rootCommand.Execute(); err != nil {
+		os.Exit(1)
+	}
+}
+
+func handleServiceControl(s service.Service, action string) {
+	err := service.Control(s, action)
+	if err != nil {
+		logger.Crit("执行动作 [%s] 遭到拒绝或失败: %v", action, err)
+		return
+	}
+	logger.Succ("成功对系统后台服务执行 [%s] 指令！", action)
+}

deploy_system.ps1

@@ -1,519 +0,0 @@
-<#
-.SYNOPSIS
-    智能网络监控脚本 - 系统级部署脚本
-
-.DESCRIPTION
-    将脚本部署到系统级目录，配置为开机自启动，无需用户登录
-    
-    部署位置：
-    - 程序文件: C:\Program Files\SmartNetworkMonitor\
-    - 配置文件: C:\ProgramData\SmartNetworkMonitor\
-    - 日志文件: C:\ProgramData\SmartNetworkMonitor\logs\
-#>
-
-# 定义系统级路径
-$ProgramPath = "C:\Program Files\SmartNetworkMonitor"
-$DataPath = "C:\ProgramData\SmartNetworkMonitor"
-$LogPath = "$DataPath\logs"
-
-Write-Host "==========================================" -ForegroundColor Cyan
-Write-Host "智能网络监控脚本 - 系统级部署" -ForegroundColor Cyan
-Write-Host "==========================================" -ForegroundColor Cyan
-Write-Host ""
-
-Write-Host "部署计划:" -ForegroundColor Yellow
-Write-Host "- 程序目录: $ProgramPath" -ForegroundColor White
-Write-Host "- 数据目录: $DataPath" -ForegroundColor White
-Write-Host "- 日志目录: $LogPath" -ForegroundColor White
-Write-Host ""
-
-# 检查源文件
-$SourcePath = $PSScriptRoot
-$RequiredFiles = @(
-    "smart_shutdown.ps1"
-)
-
-foreach ($file in $RequiredFiles) {
-    if (-not (Test-Path -Path (Join-Path $SourcePath $file))) {
-        Write-Error "找不到必需文件: $file"
-        exit 1
-    }
-}
-
-Write-Host "[OK] 源文件检查完成" -ForegroundColor Green
-
-# 创建目录结构
-Write-Host "正在创建目录结构..." -ForegroundColor Yellow
-
-try {
-    # 创建程序目录
-    if (-not (Test-Path -Path $ProgramPath)) {
-        New-Item -Path $ProgramPath -ItemType Directory -Force | Out-Null
-        Write-Host "[OK] 创建程序目录: $ProgramPath" -ForegroundColor Green
-    }
-
-    # 创建数据目录
-    if (-not (Test-Path -Path $DataPath)) {
-        New-Item -Path $DataPath -ItemType Directory -Force | Out-Null
-        Write-Host "[OK] 创建数据目录: $DataPath" -ForegroundColor Green
-    }
-
-    # 创建日志目录
-    if (-not (Test-Path -Path $LogPath)) {
-        New-Item -Path $LogPath -ItemType Directory -Force | Out-Null
-        Write-Host "[OK] 创建日志目录: $LogPath" -ForegroundColor Green
-    }
-}
-catch {
-    Write-Error "创建目录失败: $($_.Exception.Message)"
-    exit 1
-}
-
-# 创建系统优化版的主脚本
-Write-Host "正在创建系统版本的脚本..." -ForegroundColor Yellow
-
-$SystemScriptContent = @"
-#Requires -Version 5.1
-
-<#
-.SYNOPSIS
-    智能网络监控脚本 - 系统服务版本
-
-.DESCRIPTION
-    系统级网络监控脚本，在网络持续中断时自动关机
-    
-    特点：
-    - 开机自启动，无需用户登录
-    - 日志存储在系统数据目录
-    - 优化的错误处理和权限管理
-#>
-
-# ==================== 系统级配置参数 ====================
-
-# 监控的目标IP地址
-[string]`$TargetIP = "192.168.3.3"
-
-# 正常监控时的ping间隔（秒）
-[int]`$NormalPingInterval = 15
-
-# 监控窗口时长（秒） - 在此期间持续失败则触发关机
-[int]`$MonitorWindowSeconds = 180
-
-# 关机倒计时时长（秒）
-[int]`$ShutdownCountdown = 60
-
-# 关机倒计时阶段的ping间隔（秒）
-[int]`$CountdownPingInterval = 3
-
-# 系统级日志配置
-[string]`$LogDirectory = "$LogPath"
-[string]`$LogFile = Join-Path -Path `$LogDirectory -ChildPath "network_monitor_`$(Get-Date -Format 'yyyyMMdd').log"
-[string]`$ConfigFile = "$DataPath\config.json"
-[int]`$MaxLogDays = 30
-
-# 正常连接时的日志记录间隔计数器
-[int]`$normalLogInterval = 24  # 每24次循环记录一次状态 (约6分钟)
-[int]`$normalLogCounter = 0
-
-# =================================================
-
-# --- 函数定义 ---
-
-# 加载配置文件
-function Load-Configuration {
-    if (Test-Path -Path `$ConfigFile) {
-        try {
-            `$config = Get-Content -Path `$ConfigFile | ConvertFrom-Json
-            if (`$config.TargetIP) { `$script:TargetIP = `$config.TargetIP }
-            if (`$config.MonitorWindowSeconds) { `$script:MonitorWindowSeconds = `$config.MonitorWindowSeconds }
-            if (`$config.ShutdownCountdown) { `$script:ShutdownCountdown = `$config.ShutdownCountdown }
-            if (`$config.NormalPingInterval) { `$script:NormalPingInterval = `$config.NormalPingInterval }
-            Write-Log -Level "INFO" -Message "配置文件加载成功"
-        }
-        catch {
-            Write-Log -Level "WARN" -Message "配置文件格式错误，使用默认配置"
-        }
-    } else {
-        # 创建默认配置文件
-        `$defaultConfig = @{
-            TargetIP = `$TargetIP
-            MonitorWindowSeconds = `$MonitorWindowSeconds
-            ShutdownCountdown = `$ShutdownCountdown
-            NormalPingInterval = `$NormalPingInterval
-        }
-        try {
-            `$defaultConfig | ConvertTo-Json | Set-Content -Path `$ConfigFile
-            Write-Log -Level "INFO" -Message "创建默认配置文件: `$ConfigFile"
-        }
-        catch {
-            Write-Log -Level "WARN" -Message "无法创建配置文件，使用内置默认值"
-        }
-    }
-}
-
-# 日志写入函数
-function Write-Log {
-    param(
-        [Parameter(Mandatory=`$true)]
-        [ValidateSet("INFO", "SUCCESS", "FAIL", "WARN", "CRITICAL", "DEBUG")]
-        [string]`$Level,
-
-        [Parameter(Mandatory=`$true)]
-        [string]`$Message
-    )
-
-    `$logTimestamp = Get-Date -Format "yyyy-MM-dd HH:mm:ss"
-    `$logEntry = "[`$logTimestamp] [`$Level] `$Message"
-
-    # 尝试写入日志文件
-    try {
-        if (-not (Test-Path -Path `$LogDirectory)) {
-            New-Item -Path `$LogDirectory -ItemType Directory -ErrorAction Stop | Out-Null
-        }
-        Add-Content -Path `$LogFile -Value `$logEntry -ErrorAction Stop
-    }
-    catch {
-        # 如果写入失败，尝试输出到事件日志
-        try {
-            if (-not [System.Diagnostics.EventLog]::SourceExists("SmartNetworkMonitor")) {
-                New-EventLog -LogName "Application" -Source "SmartNetworkMonitor"
-            }
-            Write-EventLog -LogName "Application" -Source "SmartNetworkMonitor" -EventId 1001 -EntryType Information -Message `$logEntry
-        }
-        catch {
-            # 最后尝试输出到控制台（虽然在服务模式下不可见）
-            Write-Host "[`$logTimestamp] [CRITICAL] 日志写入失败！`$Message"
-        }
-    }
-
-    # 在服务模式下，也输出到事件日志作为备份
-    try {
-        if (`$Level -eq "CRITICAL" -or `$Level -eq "FAIL") {
-            Write-EventLog -LogName "Application" -Source "SmartNetworkMonitor" -EventId 1002 -EntryType Warning -Message `$logEntry -ErrorAction SilentlyContinue
-        }
-    }
-    catch {
-        # 忽略事件日志写入错误
-    }
-}
-
-# 清理旧日志文件
-function Remove-OldLogs {
-    try {
-        `$cutoffDate = (Get-Date).AddDays(-`$MaxLogDays)
-        `$oldLogs = Get-ChildItem -Path `$LogDirectory -Filter "network_monitor_*.log" | Where-Object { `$_.LastWriteTime -lt `$cutoffDate }
-        foreach (`$oldLog in `$oldLogs) {
-            Remove-Item -Path `$oldLog.FullName -Force
-            Write-Log -Level "INFO" -Message "删除旧日志文件: `$(`$oldLog.Name)"
-        }
-    }
-    catch {
-        Write-Log -Level "WARN" -Message "清理旧日志文件时出错: `$(`$_.Exception.Message)"
-    }
-}
-
-# =================================================
-
-# --- 主程序开始 ---
-
-# 初始化
-Load-Configuration
-Remove-OldLogs
-
-# 获取系统信息
-try {
-    `$runningAs = [System.Security.Principal.WindowsIdentity]::GetCurrent().Name
-    `$localIP = (Get-NetIPAddress -AddressFamily IPv4 | Where-Object { `$_.InterfaceAlias -notmatch "Loopback" } | Select-Object -First 1).IPAddress
-}
-catch {
-    `$runningAs = "Unknown"
-    `$localIP = "Unknown"
-}
-
-Write-Log -Level "INFO" -Message "========== 系统级网络监控脚本启动 =========="
-Write-Log -Level "INFO" -Message "运行账户: `$runningAs"
-Write-Log -Level "INFO" -Message "本机IP: `$localIP"
-Write-Log -Level "INFO" -Message "监控目标: `$TargetIP"
-Write-Log -Level "INFO" -Message "监控窗口: `${MonitorWindowSeconds}秒"
-Write-Log -Level "INFO" -Message "关机倒计时: `${ShutdownCountdown}秒"
-Write-Log -Level "INFO" -Message "日志目录: `$LogDirectory"
-Write-Log -Level "INFO" -Message "配置文件: `$ConfigFile"
-
-# 主循环
-`$failureStartTime = `$null
-
-while (`$true) {
-    # 测试网络连接
-    try {
-        `$pingResult = Test-Connection -ComputerName `$TargetIP -Count 1 -Quiet -ErrorAction SilentlyContinue
-        `$isOnline = `$pingResult
-    }
-    catch {
-        Write-Log -Level "FAIL" -Message "网络测试出现异常: `$(`$_.Exception.Message)"
-        `$isOnline = `$false
-    }
-
-    if (`$isOnline) {
-        if (`$failureStartTime) {
-            Write-Log -Level "SUCCESS" -Message "网络连接已恢复。"
-            `$failureStartTime = `$null
-        } else {
-            # 减少正常连接时的日志频率
-            `$normalLogCounter++
-            if (`$normalLogCounter -ge `$normalLogInterval) {
-                Write-Log -Level "INFO" -Message "网络连接正常 (定期状态报告)"
-                `$normalLogCounter = 0
-            }
-        }
-        
-        Start-Sleep -Seconds `$NormalPingInterval
-
-    } else {
-        # 网络不通时的处理逻辑
-        Write-Log -Level "FAIL" -Message "Ping失败 - 目标: `$TargetIP"
-        
-        if (-not `$failureStartTime) {
-            `$failureStartTime = Get-Date
-            Write-Log -Level "WARN" -Message "网络首次中断，开始进入监控窗口计时。"
-        }
-
-        `$failureDuration = (New-TimeSpan -Start `$failureStartTime -End (Get-Date)).TotalSeconds
-        Write-Log -Level "INFO" -Message "网络已持续中断 `$([math]::Round(`$failureDuration)) / `${MonitorWindowSeconds} 秒。"
-
-        if (`$failureDuration -ge `$MonitorWindowSeconds) {
-            Write-Log -Level "CRITICAL" -Message "网络持续中断已超过 `${MonitorWindowSeconds} 秒，开始关机流程。"
-            
-            `$shutdownCancelled = `$false
-            `$countdownEndTime = (Get-Date).AddSeconds(`$ShutdownCountdown)
-            
-            while ((Get-Date) -lt `$countdownEndTime -and -not `$shutdownCancelled) {
-                `$remainingSeconds = [math]::Ceiling((`$countdownEndTime - (Get-Date)).TotalSeconds)
-                if (`$remainingSeconds -le 0) { break }
-                
-                Write-Log -Level "WARN" -Message "距离关机还有 `$remainingSeconds 秒... 正在快速检测网络。"
-                
-                # 在倒计时中再次检测网络
-                try {
-                    if (Test-Connection -ComputerName `$TargetIP -Count 1 -Quiet -ErrorAction SilentlyContinue) {
-                        Write-Log -Level "SUCCESS" -Message "网络在倒计时期间恢复！取消关机。"
-                        `$failureStartTime = `$null
-                        `$shutdownCancelled = `$true
-                        break
-                    }
-                }
-                catch {
-                    Write-Log -Level "WARN" -Message "倒计时期间网络测试出现异常: `$(`$_.Exception.Message)"
-                }
-                
-                Start-Sleep -Seconds `$CountdownPingInterval
-            }
-            
-            if (-not `$shutdownCancelled) {
-                Write-Log -Level "CRITICAL" -Message "关机倒计时完成，执行系统关机命令。"
-                try {
-                    # 执行实际关机命令
-                    Stop-Computer -Force
-                    Write-Log -Level "CRITICAL" -Message "关机命令已执行。"
-                }
-                catch {
-                    Write-Log -Level "CRITICAL" -Message "关机命令执行失败: `$(`$_.Exception.Message)"
-                }
-                exit
-            }
-        } else {
-            Start-Sleep -Seconds `$NormalPingInterval
-        }
-    }
-}
-"@
-
-# 写入系统版本脚本
-$SystemScriptPath = Join-Path $ProgramPath "smart_shutdown_system.ps1"
-try {
-    $SystemScriptContent | Set-Content -Path $SystemScriptPath -Encoding UTF8
-    Write-Host "[OK] 创建系统脚本: $SystemScriptPath" -ForegroundColor Green
-}
-catch {
-    Write-Error "创建系统脚本失败: $($_.Exception.Message)"
-    exit 1
-}
-
-# 创建默认配置文件
-Write-Host "正在创建配置文件..." -ForegroundColor Yellow
-
-$DefaultConfig = @{
-    TargetIP = "192.168.3.3"
-    MonitorWindowSeconds = 180
-    ShutdownCountdown = 60
-    NormalPingInterval = 15
-}
-
-$ConfigPath = Join-Path $DataPath "config.json"
-try {
-    $DefaultConfig | ConvertTo-Json -Depth 10 | Set-Content -Path $ConfigPath -Encoding UTF8
-    Write-Host "[OK] 创建配置文件: $ConfigPath" -ForegroundColor Green
-}
-catch {
-    Write-Error "创建配置文件失败: $($_.Exception.Message)"
-    exit 1
-}
-
-# 配置任务计划程序
-Write-Host "正在配置任务计划程序..." -ForegroundColor Yellow
-
-try {
-    # 删除已存在的任务（如果有）
-    try {
-        Unregister-ScheduledTask -TaskName "Smart Network Shutdown Monitor" -Confirm:$false -ErrorAction SilentlyContinue
-    }
-    catch {
-        # 忽略错误，任务可能不存在
-    }
-
-    # 创建触发器（系统启动时）
-    $Trigger = New-ScheduledTaskTrigger -AtStartup
-
-    # 创建操作
-    $Action = New-ScheduledTaskAction -Execute "PowerShell.exe" -Argument "-WindowStyle Hidden -ExecutionPolicy Bypass -File `"$SystemScriptPath`""
-
-    # 创建设置
-    $Settings = New-ScheduledTaskSettingsSet -AllowStartIfOnBatteries -DontStopIfGoingOnBatteries -StartWhenAvailable -DontStopOnIdleEnd -ExecutionTimeLimit (New-TimeSpan -Days 365)
-
-    # 创建主体（以SYSTEM身份运行）
-    $Principal = New-ScheduledTaskPrincipal -UserId "SYSTEM" -LogonType ServiceAccount -RunLevel Highest
-
-    # 注册任务
-    Register-ScheduledTask -TaskName "Smart Network Shutdown Monitor" -Trigger $Trigger -Action $Action -Settings $Settings -Principal $Principal -Description "智能网络监控脚本 - 系统级运行"
-
-    Write-Host "[OK] 任务计划程序配置完成" -ForegroundColor Green
-}
-catch {
-    Write-Error "配置任务计划程序失败: $($_.Exception.Message)"
-    exit 1
-}
-
-# 创建管理脚本
-Write-Host "正在创建管理工具..." -ForegroundColor Yellow
-
-$ManagementScript = @"
-# 智能网络监控脚本 - 管理工具
-
-Write-Host "智能网络监控脚本 - 管理工具" -ForegroundColor Cyan
-Write-Host "==============================" -ForegroundColor Cyan
-Write-Host ""
-
-Write-Host "可用操作:" -ForegroundColor Yellow
-Write-Host "1. 查看任务状态"
-Write-Host "2. 启动监控任务"
-Write-Host "3. 停止监控任务"
-Write-Host "4. 查看今天的日志"
-Write-Host "5. 查看实时日志"
-Write-Host "6. 查看配置文件"
-Write-Host "7. 编辑配置文件"
-Write-Host ""
-
-`$choice = Read-Host "请选择操作 (1-7)"
-
-switch (`$choice) {
-    "1" {
-        Get-ScheduledTask -TaskName "Smart Network Shutdown Monitor" | Format-Table
-    }
-    "2" {
-        Start-ScheduledTask -TaskName "Smart Network Shutdown Monitor"
-        Write-Host "任务已启动" -ForegroundColor Green
-    }
-    "3" {
-        Stop-ScheduledTask -TaskName "Smart Network Shutdown Monitor"
-        Write-Host "任务已停止" -ForegroundColor Green
-    }
-    "4" {
-        `$logFile = "$LogPath\network_monitor_`$(Get-Date -Format 'yyyyMMdd').log"
-        if (Test-Path `$logFile) {
-            Get-Content `$logFile
-        } else {
-            Write-Host "今天的日志文件不存在" -ForegroundColor Red
-        }
-    }
-    "5" {
-        `$logFile = "$LogPath\network_monitor_`$(Get-Date -Format 'yyyyMMdd').log"
-        if (Test-Path `$logFile) {
-            Get-Content `$logFile -Wait
-        } else {
-            Write-Host "今天的日志文件不存在" -ForegroundColor Red
-        }
-    }
-    "6" {
-        Get-Content "$DataPath\config.json"
-    }
-    "7" {
-        notepad "$DataPath\config.json"
-    }
-    default {
-        Write-Host "无效选择" -ForegroundColor Red
-    }
-}
-
-Write-Host ""
-Write-Host "常用命令:" -ForegroundColor Yellow
-Write-Host "1. 查看任务状态:"
-Write-Host "   Get-ScheduledTask -TaskName 'Smart Network Shutdown Monitor'"
-Write-Host ""
-Write-Host "2. 启动任务:"
-Write-Host "   Start-ScheduledTask -TaskName 'Smart Network Shutdown Monitor'"
-Write-Host ""
-Write-Host "3. 停止任务:"
-Write-Host "   Stop-ScheduledTask -TaskName 'Smart Network Shutdown Monitor'"
-Write-Host ""
-Write-Host "4. 查看今天日志:"
-Write-Host "   Get-Content '$LogPath\network_monitor_`$(Get-Date -Format 'yyyyMMdd').log'"
-Write-Host ""
-Write-Host "5. 实时查看日志:"
-Write-Host "   Get-Content '$LogPath\network_monitor_`$(Get-Date -Format 'yyyyMMdd').log' -Wait"
-Write-Host ""
-Write-Host "6. 查看配置:"
-Write-Host "   Get-Content '$DataPath\config.json'"
-Write-Host ""
-Write-Host "7. 编辑配置文件:"
-Write-Host "   notepad '$DataPath\config.json'"
-Write-Host ""
-"@
-
-$ManagementScriptPath = Join-Path $ProgramPath "manage.ps1"
-try {
-    $ManagementScript | Set-Content -Path $ManagementScriptPath -Encoding UTF8
-    Write-Host "[OK] 创建管理脚本: $ManagementScriptPath" -ForegroundColor Green
-}
-catch {
-    Write-Error "创建管理脚本失败: $($_.Exception.Message)"
-    exit 1
-}
-
-Write-Host ""
-Write-Host "[INFO] 系统级部署完成！" -ForegroundColor Green
-Write-Host ""
-Write-Host "部署摘要:" -ForegroundColor Cyan
-Write-Host "[OK] 程序文件已部署到: $ProgramPath" -ForegroundColor White
-Write-Host "[OK] 数据目录已创建: $DataPath" -ForegroundColor White
-Write-Host "[OK] 日志目录已创建: $LogPath" -ForegroundColor White
-Write-Host "[OK] 任务计划程序已配置（系统启动时运行）" -ForegroundColor White
-Write-Host "[OK] SYSTEM 权限配置完成" -ForegroundColor White
-Write-Host ""
-Write-Host "管理工具:" -ForegroundColor Yellow
-Write-Host "- 管理脚本: $ManagementScriptPath" -ForegroundColor White
-Write-Host "- 配置文件: $DataPath\config.json" -ForegroundColor White
-Write-Host "- 今天日志: $LogPath\network_monitor_$(Get-Date -Format 'yyyyMMdd').log" -ForegroundColor White
-Write-Host ""
-Write-Host "下次重启后，脚本将自动运行！" -ForegroundColor Green
-Write-Host ""
-
-# 询问是否立即启动
-$response = Read-Host "是否立即启动监控任务？(Y/N)"
-if ($response -eq 'Y' -or $response -eq 'y' -or $response -eq '') {
-    try {
-        Start-ScheduledTask -TaskName "Smart Network Shutdown Monitor"
-        Write-Host "[OK] 监控任务已启动" -ForegroundColor Green
-    }
-    catch {
-        Write-Host "[X] 启动任务失败: $($_.Exception.Message)" -ForegroundColor Red
-    }
-}

go.mod

@@ -0,0 +1,22 @@
+module smart-shutdown
+
+go 1.25.0
+
+require (
+	github.com/fatih/color v1.19.0
+	github.com/kardianos/service v1.2.4
+	github.com/prometheus-community/pro-bing v0.8.0
+	github.com/spf13/cobra v1.10.2
+	gopkg.in/natefinch/lumberjack.v2 v2.2.1
+)
+
+require (
+	github.com/google/uuid v1.6.0 // indirect
+	github.com/inconshreveable/mousetrap v1.1.0 // indirect
+	github.com/mattn/go-colorable v0.1.14 // indirect
+	github.com/mattn/go-isatty v0.0.20 // indirect
+	github.com/spf13/pflag v1.0.9 // indirect
+	golang.org/x/net v0.49.0 // indirect
+	golang.org/x/sync v0.19.0 // indirect
+	golang.org/x/sys v0.42.0 // indirect
+)

go.sum

@@ -0,0 +1,31 @@
+github.com/cpuguy83/go-md2man/v2 v2.0.6/go.mod h1:oOW0eioCTA6cOiMLiUPZOpcVxMig6NIQQ7OS05n1F4g=
+github.com/fatih/color v1.19.0 h1:Zp3PiM21/9Ld6FzSKyL5c/BULoe/ONr9KlbYVOfG8+w=
+github.com/fatih/color v1.19.0/go.mod h1:zNk67I0ZUT1bEGsSGyCZYZNrHuTkJJB+r6Q9VuMi0LE=
+github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
+github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
+github.com/inconshreveable/mousetrap v1.1.0 h1:wN+x4NVGpMsO7ErUn/mUI3vEoE6Jt13X2s0bqwp9tc8=
+github.com/inconshreveable/mousetrap v1.1.0/go.mod h1:vpF70FUmC8bwa3OWnCshd2FqLfsEA9PFc4w1p2J65bw=
+github.com/kardianos/service v1.2.4 h1:XNlGtZOYNx2u91urOdg/Kfmc+gfmuIo1Dd3rEi2OgBk=
+github.com/kardianos/service v1.2.4/go.mod h1:E4V9ufUuY82F7Ztlu1eN9VXWIQxg8NoLQlmFe0MtrXc=
+github.com/mattn/go-colorable v0.1.14 h1:9A9LHSqF/7dyVVX6g0U9cwm9pG3kP9gSzcuIPHPsaIE=
+github.com/mattn/go-colorable v0.1.14/go.mod h1:6LmQG8QLFO4G5z1gPvYEzlUgJ2wF+stgPZH1UqBm1s8=
+github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=
+github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
+github.com/prometheus-community/pro-bing v0.8.0 h1:CEY/g1/AgERRDjxw5P32ikcOgmrSuXs7xon7ovx6mNc=
+github.com/prometheus-community/pro-bing v0.8.0/go.mod h1:Idyxz8raDO6TgkUN6ByiEGvWJNyQd40kN9ZUeho3lN0=
+github.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=
+github.com/spf13/cobra v1.10.2 h1:DMTTonx5m65Ic0GOoRY2c16WCbHxOOw6xxezuLaBpcU=
+github.com/spf13/cobra v1.10.2/go.mod h1:7C1pvHqHw5A4vrJfjNwvOdzYu0Gml16OCs2GRiTUUS4=
+github.com/spf13/pflag v1.0.9 h1:9exaQaMOCwffKiiiYk6/BndUBv+iRViNW+4lEMi0PvY=
+github.com/spf13/pflag v1.0.9/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg=
+go.yaml.in/yaml/v3 v3.0.4/go.mod h1:DhzuOOF2ATzADvBadXxruRBLzYTpT36CKvDb3+aBEFg=
+golang.org/x/net v0.49.0 h1:eeHFmOGUTtaaPSGNmjBKpbng9MulQsJURQUAfUwY++o=
+golang.org/x/net v0.49.0/go.mod h1:/ysNB2EvaqvesRkuLAyjI1ycPZlQHM3q01F02UY/MV8=
+golang.org/x/sync v0.19.0 h1:vV+1eWNmZ5geRlYjzm2adRgW2/mcpevXNg50YZtPCE4=
+golang.org/x/sync v0.19.0/go.mod h1:9KTHXmSnoGruLpwFjVSX0lNNA75CykiMECbovNTZqGI=
+golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.42.0 h1:omrd2nAlyT5ESRdCLYdm3+fMfNFE/+Rf4bDIQImRJeo=
+golang.org/x/sys v0.42.0/go.mod h1:4GL1E5IUh+htKOUEOaiffhrAeqysfVGipDYzABqnCmw=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/natefinch/lumberjack.v2 v2.2.1 h1:bBRl1b0OH9s/DuPhuXpNl+VtCaJXFZ5/uEFST95x9zc=
+gopkg.in/natefinch/lumberjack.v2 v2.2.1/go.mod h1:YD8tP3GAjkrDg1eZH7EGmyESg/lsYskCTPBJVb9jqSc=

pkg/config/config.go

@@ -0,0 +1,67 @@
+package config
+
+import (
+	"encoding/json"
+	"os"
+	"path/filepath"
+	"runtime"
+)
+
+// Config 存储程序的各类配置常量，支持通过读取 json 文件动态修改
+type Config struct {
+	TargetIP             string `json:"TargetIP"`             // 检测目标 IP，默认 192.168.3.3
+	MonitorWindowSeconds int    `json:"MonitorWindowSeconds"` // 监控窗口期内持续失败才触发关机
+	ShutdownCountdown    int    `json:"ShutdownCountdown"`    // 关机倒计时时间
+	NormalPingInterval   int    `json:"NormalPingInterval"`   // 正常状态下的 ping 间隔
+}
+
+// DefaultConfig 提供一套开箱即用的默认配置
+func DefaultConfig() *Config {
+	return &Config{
+		TargetIP:             "192.168.3.3",
+		MonitorWindowSeconds: 180,
+		ShutdownCountdown:    60,
+		NormalPingInterval:   15,
+	}
+}
+
+// GetConfigDir 根据当前操作系统返回标准的配置存放路径
+func GetConfigDir() string {
+	if runtime.GOOS == "windows" {
+		return `C:\ProgramData\SmartNetworkMonitor`
+	}
+	return `/etc/smart-network-monitor`
+}
+
+// GetLogDir 根据当前操作系统返回标准的日志存放路径
+func GetLogDir() string {
+	if runtime.GOOS == "windows" {
+		return `C:\ProgramData\SmartNetworkMonitor\logs`
+	}
+	return `/var/log/smart-network-monitor`
+}
+
+// LoadConfig 从系统标准路径读取 config.json 并反序列化。如果不存在，则返回默认配置。
+func LoadConfig() (*Config, error) {
+	configDir := GetConfigDir()
+	configPath := filepath.Join(configDir, "config.json")
+
+	data, err := os.ReadFile(configPath)
+	if err != nil {
+		if os.IsNotExist(err) {
+			// 文件不存在时，静默采用默认配置
+			return DefaultConfig(), nil
+		}
+		return nil, err
+	}
+
+	if len(data) >= 3 && data[0] == 0xef && data[1] == 0xbb && data[2] == 0xbf {
+		data = data[3:]
+	}
+
+	cfg := &Config{}
+	if err := json.Unmarshal(data, cfg); err != nil {
+		return nil, err
+	}
+	return cfg, nil
+}

pkg/daemon/service.go

@@ -0,0 +1,63 @@
+package daemon
+
+import (
+	"context"
+
+	"github.com/kardianos/service"
+	"smart-shutdown/pkg/config"
+	"smart-shutdown/pkg/logger"
+	"smart-shutdown/pkg/monitor"
+)
+
+type program struct {
+	exit   chan struct{}
+	cancel context.CancelFunc
+	cfg    *config.Config
+}
+
+func (p *program) Start(s service.Service) error {
+	// Start should not block. Do the actual work async.
+	logger.Info("准备在后台启动服务监控...")
+
+	ctx, cancel := context.WithCancel(context.Background())
+	p.cancel = cancel
+	p.exit = make(chan struct{})
+
+	go p.run(ctx)
+	return nil
+}
+
+func (p *program) run(ctx context.Context) {
+	// 挂载执行核心循环逻辑
+	monitor.Run(ctx, p.cfg)
+	close(p.exit)
+}
+
+func (p *program) Stop(s service.Service) error {
+	// Stop should not block. Return within a few seconds.
+	logger.Info("正在平滑停止后台服务监控...")
+	if p.cancel != nil {
+		p.cancel()
+	}
+	<-p.exit
+	return nil
+}
+
+// GetService 构建 service 实例供 CLI 控制（安装，启动，停止，卸载）和前台直接 Run。
+func GetService(cfg *config.Config) (service.Service, error) {
+	svcConfig := &service.Config{
+		Name:        "SmartNetworkMonitor",
+		DisplayName: "Smart Network Shutdown Monitor",
+		Description: "A daemon that pings target IP periodically and shuts down the computer if disconnected for too long.",
+	}
+
+	prg := &program{
+		cfg: cfg,
+	}
+
+	s, err := service.New(prg, svcConfig)
+	if err != nil {
+		return nil, err
+	}
+	return s, nil
+}

pkg/logger/logger.go

@@ -0,0 +1,70 @@
+package logger
+
+import (
+	"io"
+	"log"
+	"os"
+	"path/filepath"
+
+	"github.com/fatih/color"
+	"gopkg.in/natefinch/lumberjack.v2"
+	"smart-shutdown/pkg/config"
+)
+
+var (
+	infoLogger *log.Logger
+	succLogger *log.Logger
+	warnLogger *log.Logger
+	failLogger *log.Logger
+	critLogger *log.Logger
+)
+
+// InitLogger 初始化全局日志系统，将日志同时输出到终端色彩日志和滚动文件
+func InitLogger() error {
+	logDir := config.GetLogDir()
+	if err := os.MkdirAll(logDir, 0755); err != nil {
+		// 降级为当前目录，适用于未具有系统写权限的 Local 执行
+		logDir = "logs"
+		os.MkdirAll(logDir, 0755)
+	}
+
+	logFile := filepath.Join(logDir, "network_monitor.log")
+
+	// 使用 lumberjack 实现每天轮转和最大保留 30 天功能
+	ljLogger := &lumberjack.Logger{
+		Filename:   logFile,
+		MaxSize:    10,   // 每个切片最大兆字节 (MB)
+		MaxBackups: 30,   // 保留最近 30 个切片
+		MaxAge:     30,   // 保留最近 30 天的文件
+		Compress:   false, // 是否压缩
+	}
+
+	// 各级别终端色彩
+	cSucc := color.New(color.FgGreen).SprintFunc()
+	cFail := color.New(color.FgRed).SprintFunc()
+	cWarn := color.New(color.FgYellow).SprintFunc()
+	cCrit := color.New(color.FgHiRed, color.Bold).SprintFunc()
+
+	// 构建复合输出写入器：终端标准输出 + 日志文件
+	outGeneral := io.MultiWriter(os.Stdout, ljLogger)
+	outError := io.MultiWriter(os.Stderr, ljLogger)
+
+	// 时间戳格式前缀
+	flags := log.Ldate | log.Ltime 
+
+	// 实例化各个级别 Logger
+	infoLogger = log.New(outGeneral, "[INFO] ", flags)
+	succLogger = log.New(outGeneral, cSucc("[SUCCESS] "), flags)
+	warnLogger = log.New(outGeneral, cWarn("[WARN] "), flags)
+	failLogger = log.New(outError, cFail("[FAIL] "), flags)
+	critLogger = log.New(outError, cCrit("[CRITICAL] "), flags)
+
+	return nil
+}
+
+// 封装导出供业务调用的函数
+func Info(format string, v ...interface{}) { infoLogger.Printf(format, v...) }
+func Succ(format string, v ...interface{}) { succLogger.Printf(format, v...) }
+func Warn(format string, v ...interface{}) { warnLogger.Printf(format, v...) }
+func Fail(format string, v ...interface{}) { failLogger.Printf(format, v...) }
+func Crit(format string, v ...interface{}) { critLogger.Printf(format, v...) }

pkg/monitor/statemachine.go

@@ -0,0 +1,124 @@
+package monitor
+
+import (
+	"context"
+	"time"
+
+	"smart-shutdown/pkg/config"
+	"smart-shutdown/pkg/logger"
+	"smart-shutdown/pkg/pinger"
+	"smart-shutdown/pkg/system"
+)
+
+// Run 开启核心状态机循环监控，接收 ctx 以便主程序或者服务管理器控制安全退出
+func Run(ctx context.Context, cfg *config.Config) {
+	logger.Info("========== 智能关机监控已启动 ==========")
+	logger.Info("监控目标: %s", cfg.TargetIP)
+	logger.Info("监控窗口: %d秒", cfg.MonitorWindowSeconds)
+	logger.Info("关机倒计时: %d秒", cfg.ShutdownCountdown)
+
+	var failureStartTime time.Time
+	var inFailureWindow bool = false
+
+	normalStatusCounter := 0
+	// 每 24 次正常检测输出一次日志（约 6 分钟）
+	const normalStatusLogInterval = 24
+
+	for {
+		select {
+		case <-ctx.Done():
+			logger.Info("========== 收到停止信号，退出监控 ==========")
+			return
+		default:
+		}
+
+		isOnline := pinger.Ping(cfg.TargetIP, 3) // 3 秒超时探测一次
+
+		if isOnline {
+			if inFailureWindow {
+				logger.Succ("网络连接已恢复！重置断网计时器。")
+				inFailureWindow = false
+				normalStatusCounter = 0
+			} else {
+				normalStatusCounter++
+				if normalStatusCounter >= normalStatusLogInterval {
+					logger.Info("网络连接持续正常（已检测 %d 次）", normalStatusCounter)
+					normalStatusCounter = 0
+				}
+			}
+
+			// 正常状态下休眠等待下一次探测
+			time.Sleep(time.Duration(cfg.NormalPingInterval) * time.Second)
+
+		} else {
+			logger.Fail("Ping 测试失败 - 目标: %s", cfg.TargetIP)
+
+			if !inFailureWindow {
+				// 首次断网
+				failureStartTime = time.Now()
+				inFailureWindow = true
+				logger.Warn("网络首次中断，开始进入监控窗口计时。")
+			}
+
+			// 计算当前断网累计时间
+			failureDuration := time.Since(failureStartTime).Seconds()
+
+			if int(failureDuration) >= cfg.MonitorWindowSeconds {
+				logger.Crit("网络持续中断已超过 %d 秒，触发关机流程！", cfg.MonitorWindowSeconds)
+
+				// 进入倒计时阶段，倒计时期间高频检测（阻塞调用）
+				shutdownCancelled := startCountdown(ctx, cfg.TargetIP, cfg.ShutdownCountdown)
+
+				if !shutdownCancelled {
+					logger.Crit("倒计时结束，执行最终关机！")
+					system.ExecuteShutdown()
+					// 关机指令发出，理论上系统即将终止进程，为了优雅，此处也退出
+					return
+				} else {
+					// 倒计时中途网络恢复被取消了，重置状态重新进入监控环节
+					inFailureWindow = false
+				}
+			} else {
+				// 还在容忍窗口期内，仅仅打印日志并休眠
+				logger.Info("网络已持续中断 %.0f / %d 秒，继续监控...", failureDuration, cfg.MonitorWindowSeconds)
+				time.Sleep(time.Duration(cfg.NormalPingInterval) * time.Second)
+			}
+		}
+	}
+}
+
+// startCountdown 执行关机前倒计时的轮询，如果中途恢复网络返回 true (代表取消关机)。
+// 返回 false 代表断网到底，坚决关机。
+func startCountdown(ctx context.Context, targetIP string, countdownSec int) bool {
+	logger.Warn("============================================")
+	logger.Warn("准备关机！倒计时 %d 秒期间将快速探测网络...", countdownSec)
+	logger.Warn("============================================")
+
+	endTime := time.Now().Add(time.Duration(countdownSec) * time.Second)
+	// 倒计时期间提高探测频率：每 3 秒一次
+	ticker := time.NewTicker(3 * time.Second)
+	defer ticker.Stop()
+
+	for {
+		remaining := int(time.Until(endTime).Seconds())
+		if remaining <= 0 {
+			return false // 时间到，确认无法恢复
+		}
+
+		logger.Warn("距离关机还有 %d 秒... 正在高频检测网络", remaining)
+
+		select {
+		case <-ctx.Done():
+			logger.Info("收到退出系统服务信号，中止倒计时关机！")
+			return true
+		case <-ticker.C:
+			// 高频重试检测网络
+			if pinger.Ping(targetIP, 3) {
+				logger.Succ("============================================")
+				logger.Succ("网络在倒计时期间奇迹般地恢复了！立刻取消关机！")
+				logger.Succ("============================================")
+				return true
+			}
+		}
+	}
+}

pkg/pinger/pinger.go

@@ -0,0 +1,41 @@
+package pinger
+
+import (
+	"time"
+
+	probing "github.com/prometheus-community/pro-bing"
+	"smart-shutdown/pkg/logger"
+)
+
+// Ping 连通性测试，向指定的 targetIP 发送 ICMP 请求。
+// 如果规定时间内可达返回 true，超时或者全部丢包返回 false。
+func Ping(targetIP string, timeoutSec int) bool {
+	pinger, err := probing.NewPinger(targetIP)
+	if err != nil {
+		logger.Warn("解析目标 IP 失败 [%s]: %v", targetIP, err)
+		return false
+	}
+
+	// 大部分操作系统为了安全，普通程序发 ICMP 包会被拦截。开启 Privileged 会尝试使用 raw sockets 发行 
+	// 本程序由于本身作为系统服务（System 或 Root 权限）运行，故直接开启特权模式以保障发包成功率。
+	pinger.SetPrivileged(true)
+
+	// 只发送 1 个探测包
+	pinger.Count = 1
+	pinger.Timeout = time.Duration(timeoutSec) * time.Second
+
+	err = pinger.Run()
+	if err != nil {
+		logger.Fail("Ping 测试运行时遇到异常: %v", err)
+		return false
+	}
+
+	stats := pinger.Statistics()
+	// 只要成功收到至少 1 个回包，认为网络通畅
+	if stats.PacketsRecv > 0 {
+		return true
+	}
+
+	// 丢包或者无回音
+	return false
+}

pkg/system/shutdown.go

@@ -0,0 +1,28 @@
+package system
+
+import (
+	"os/exec"
+	"runtime"
+	"smart-shutdown/pkg/logger"
+)
+
+// ExecuteShutdown 跨平台执行立即无条件关机
+func ExecuteShutdown() error {
+	logger.Crit("执行系统关机操作！")
+	
+	// 注意：在实际正式环境使用前，如果想防误触可暂时注释掉下方的 `cmd.Run()` 
+	var cmd *exec.Cmd
+	if runtime.GOOS == "windows" {
+		cmd = exec.Command("shutdown", "-s", "-f", "-t", "0")
+	} else {
+		cmd = exec.Command("shutdown", "-h", "now")
+	}
+
+	err := cmd.Run()
+	if err != nil {
+		logger.Crit("系统关机指令执行失败: %v", err)
+		return err
+	}
+	
+	return nil
+}

smart_shutdown.ps1

@@ -1,234 +0,0 @@
-#Requires -Version 5.1
-#Requires -RunAsAdministrator
-
-<#
-.SYNOPSIS
-    一个智能网络监控脚本，当检测到与目标IP的网络持续中断后，会自动关闭计算机。
-
-.DESCRIPTION
-    该脚本会定期测试与指定目标IP的网络连通性。
-    如果在设定的“监控窗口”时间内连接持续失败，脚本将启动一个可取消的关机倒计时。
-    如果在倒计时期间网络恢复，关机将被中止。
-    所有操作都会被记录在日志文件中。
-#>
-
-# ==================== 配置参数 ====================
-
-# 监控的目标IP地址
-[string]$TargetIP = "192.168.3.3"
-
-# 正常监控时的ping间隔（秒）
-[int]$NormalPingInterval = 15
-
-# 监控窗口时长（秒） - 在此期间持续失败则触发关机
-[int]$MonitorWindowSeconds = 60
-
-# 关机倒计时时长（秒）
-[int]$ShutdownCountdown = 60
-
-# 关机倒计时阶段的ping间隔（秒）
-[int]$CountdownPingInterval = 3
-
-# ping超时时间（秒）
-[int]$PingTimeout = 3
-
-# 日志配置
-[string]$LogDirectory = Join-Path -Path $PSScriptRoot -ChildPath "logs"
-[string]$LogFile = Join-Path -Path $LogDirectory -ChildPath "network_monitor_$(Get-Date -Format 'yyyyMMdd').log"
-[int]$MaxLogDays = 30
-
-# =================================================
-
-# --- 函数定义 ---
-
-# 日志写入函数
-function Write-Log {
-    param(
-        [Parameter(Mandatory=$true)]
-        [ValidateSet("INFO", "SUCCESS", "FAIL", "WARN", "CRITICAL", "DEBUG")]
-        [string]$Level,
-
-        [Parameter(Mandatory=$true)]
-        [string]$Message
-    )
-
-    $logTimestamp = Get-Date -Format "yyyy-MM-dd HH:mm:ss"
-    $logEntry = "[$logTimestamp] [$Level] $Message"
-
-    # 尝试写入日志文件
-    try {
-        if (-not (Test-Path -Path $LogDirectory)) {
-            New-Item -Path $LogDirectory -ItemType Directory -ErrorAction Stop | Out-Null
-        }
-        Add-Content -Path $LogFile -Value $logEntry -ErrorAction Stop
-    }
-    catch {
-        Write-Host "[$logTimestamp] [CRITICAL] 日志文件写入失败！请检查权限或磁盘空间。" -ForegroundColor Red
-    }
-
-    # 根据日志级别在控制台输出不同颜色的信息
-    switch ($Level) {
-        "SUCCESS"  { Write-Host $logEntry -ForegroundColor Green }
-        "FAIL"     { Write-Host $logEntry -ForegroundColor Red }
-        "WARN"     { Write-Host $logEntry -ForegroundColor Yellow }
-        "CRITICAL" { Write-Host $logEntry -ForegroundColor DarkRed }
-        default    { Write-Host $logEntry }
-    }
-}
-
-# 获取本机的最佳IP地址
-function Get-LocalIP {
-    param($TargetIP)
-    try {
-        # 尝试寻找与目标IP在同一个子网的本机IP地址
-        $targetSubnet = ($TargetIP.Split('.')[0..2]) -join '.'
-        $ip = Get-NetIPAddress -AddressFamily IPv4 -AddressState Preferred | Where-Object { $_.IPAddress -like "$targetSubnet.*" } | Select-Object -First 1 -ExpandProperty IPAddress
-        
-        if ($ip) { return $ip }
-
-        # 如果找不到，则返回任意一个非环回的私有IP地址
-        $ip = Get-NetIPAddress -AddressFamily IPv4 -AddressState Preferred | Where-Object { $_.IPAddress -notlike "127.0.0.1" -and $_.InterfaceAlias -notlike "Loopback*" } | Select-Object -First 1 -ExpandProperty IPAddress
-        return $ip
-    }
-    catch {
-        return "未知"
-    }
-}
-
-
-# --- 脚本主逻辑 ---
-
-# 清理旧日志
-if (Test-Path -Path $LogDirectory) {
-    Get-ChildItem -Path $LogDirectory -Filter "*.log" | Where-Object { $_.LastWriteTime -lt (Get-Date).AddDays(-$MaxLogDays) } | Remove-Item
-}
-
-# 初始化
-Clear-Host
-$localIP = Get-LocalIP -TargetIP $TargetIP
-Write-Log -Level "INFO" -Message "========== 脚本启动 =========="
-Write-Log -Level "INFO" -Message "本机IP: $localIP"
-Write-Log -Level "INFO" -Message "监控目标: $TargetIP"
-Write-Log -Level "INFO" -Message "监控窗口: ${MonitorWindowSeconds}秒"
-Write-Log -Level "INFO" -Message "关机倒计时: ${ShutdownCountdown}秒"
-Write-Log -Level "INFO" -Message "日志文件: $LogFile"
-Write-Log -Level "INFO" -Message "脚本已启动，按 Ctrl+C 退出..."
-Write-Host "========================================"
-
-# 主循环
-$failureStartTime = $null
-$normalStatusCounter = 0  # 正常状态计数器，用于定期记录正常运行日志
-$normalStatusLogInterval = 24  # 每24次正常检测记录一次日志（约6分钟）
-
-while ($true) {# 测试网络连接
-    try {
-        # 使用兼容性更好的ping方法
-        $pingResult = Test-Connection -ComputerName $TargetIP -Count 1 -Quiet -ErrorAction SilentlyContinue
-        $isOnline = $pingResult
-    }    catch {
-        Write-Log -Level "FAIL" -Message "网络测试出现异常: $($_.Exception.Message)"
-        $isOnline = $false
-    }    if ($isOnline) {
-        if ($failureStartTime) {
-            Write-Log -Level "SUCCESS" -Message "网络连接已恢复。"
-            Write-Host "[OK] 网络连接已恢复，重置监控窗口" -ForegroundColor Green
-            $failureStartTime = $null # 重置失败计时器
-            $normalStatusCounter = 0  # 重置正常状态计数器
-        } else {
-            Write-Host "[OK] 网络连接正常" -ForegroundColor Green
-            # 正常连接时减少日志频率，只定期记录
-            $normalStatusCounter++
-            if ($normalStatusCounter -ge $normalStatusLogInterval) {
-                Write-Log -Level "INFO" -Message "网络连接持续正常（已检测 $normalStatusCounter 次）"
-                $normalStatusCounter = 0  # 重置计数器
-            }
-        }
-        # 只在控制台显示等待信息，不写入日志文件
-        Write-Host "等待 ${NormalPingInterval}秒后继续监控..." -ForegroundColor Cyan
-        Start-Sleep -Seconds $NormalPingInterval
-
-    } else {
-        # 如果网络不通
-        Write-Log -Level "FAIL" -Message "Ping失败 - 目标: $TargetIP"
-        
-        if (-not $failureStartTime) {
-            # 记录首次失败的时间
-            $failureStartTime = Get-Date
-            Write-Log -Level "WARN" -Message "网络首次中断，开始进入监控窗口计时。"
-        }        # 计算网络已中断的时间
-        $failureDuration = (New-TimeSpan -Start $failureStartTime -End (Get-Date)).TotalSeconds
-        Write-Log -Level "INFO" -Message "网络已持续中断 $([math]::Round($failureDuration)) / ${MonitorWindowSeconds} 秒。"
-        
-        # 显示详细状态
-        $remainingTime = $MonitorWindowSeconds - [math]::Round($failureDuration)
-        Write-Host "监控状态 - 已中断: $([math]::Round($failureDuration))/${MonitorWindowSeconds}秒，剩余: ${remainingTime}秒" -ForegroundColor Yellow
-
-        if ($failureDuration -ge $MonitorWindowSeconds) {
-            # 持续失败时间超过监控窗口，开始关机倒计时
-            Write-Log -Level "CRITICAL" -Message "网络持续中断已超过 ${MonitorWindowSeconds} 秒，开始关机流程。"
-            Write-Host "========================================" -ForegroundColor Yellow
-              $shutdownCancelled = $false
-            $countdownEndTime = (Get-Date).AddSeconds($ShutdownCountdown)
-              while ((Get-Date) -lt $countdownEndTime -and -not $shutdownCancelled) {
-                $remainingSeconds = [math]::Ceiling(($countdownEndTime - (Get-Date)).TotalSeconds)
-                if ($remainingSeconds -le 0) { break }
-                
-                Write-Log -Level "WARN" -Message "距离关机还有 $remainingSeconds 秒... 正在快速检测网络。"
-                Write-Host "----------------------------------------" -ForegroundColor Yellow
-                Write-Host "距离关机还有 $remainingSeconds 秒..." -ForegroundColor Yellow
-                Write-Host "正在快速检测网络连接... 按任意键可手动取消关机" -ForegroundColor Yellow                # 在倒计时中再次检测网络
-                try {
-                    if (Test-Connection -ComputerName $TargetIP -Count 1 -Quiet -ErrorAction SilentlyContinue) {
-                        Write-Log -Level "SUCCESS" -Message "网络在倒计时期间恢复！取消关机。"
-                        Write-Host "========================================" -ForegroundColor Green
-                        Write-Host "[OK] 网络连接已恢复！取消关机操作" -ForegroundColor Green
-                        Write-Host "========================================" -ForegroundColor Green
-                        $failureStartTime = $null
-                        $shutdownCancelled = $true
-                        break
-                    } else {
-                        Write-Host "[X] 网络仍然中断" -ForegroundColor Red
-                    }
-                }
-                catch {
-                    Write-Log -Level "WARN" -Message "倒计时期间网络测试出现异常: $($_.Exception.Message)"
-                    Write-Host "[X] 网络测试异常" -ForegroundColor Red
-                }
-                  # 检查是否有按键输入（手动取消）
-                $timeout = $CountdownPingInterval
-                $startTime = Get-Date
-                while (((Get-Date) - $startTime).TotalSeconds -lt $timeout -and -not $shutdownCancelled) {
-                    if ([Console]::KeyAvailable) {
-                        $null = [Console]::ReadKey($true)  # 读取按键但不使用
-                        Write-Log -Level "WARN" -Message "用户手动取消了关机操作"                        Write-Host ""
-                        Write-Host "========================================" -ForegroundColor Cyan
-                        Write-Host "[INFO] 用户已手动取消关机，返回监控模式" -ForegroundColor Cyan
-                        Write-Host "========================================" -ForegroundColor Cyan
-                        $failureStartTime = $null
-                        $shutdownCancelled = $true
-                        break                    }
-                    Start-Sleep -Milliseconds 100
-                }
-            }
-            
-            if (-not $shutdownCancelled) {
-                Write-Log -Level "CRITICAL" -Message "关机倒计时完成，执行系统关机命令。"
-                try {
-                    # 取消注释下一行以启用实际关机
-                    # Stop-Computer -Force
-                    Write-Host "模拟关机：Stop-Computer -Force （为防止意外，此行已注释）" -ForegroundColor Red
-                    Write-Log -Level "CRITICAL" -Message "关机命令已执行（模拟模式）。"
-                }
-                catch {
-                    Write-Log -Level "CRITICAL" -Message "关机命令执行失败: $($_.Exception.Message)"
-                }
-                # 脚本执行完关机命令后可以退出
-                exit
-            }
-        } else {
-            # 在监控窗口期内，继续等待
-            Write-Log -Level "INFO" -Message "等待 ${NormalPingInterval}秒后继续..."
-            Start-Sleep -Seconds $NormalPingInterval
-        }
-    }
-}

uninstall_system.ps1

@@ -1,91 +0,0 @@
-#Requires -RunAsAdministrator
-
-<#
-.SYNOPSIS
-    智能网络监控脚本 - 系统卸载脚本
-
-.DESCRIPTION
-    完全卸载系统级部署的智能网络监控脚本
-#>
-
-$ProgramPath = "C:\Program Files\SmartNetworkMonitor"
-$DataPath = "C:\ProgramData\SmartNetworkMonitor"
-
-Write-Host "==========================================" -ForegroundColor Red
-Write-Host "智能网络监控脚本 - 系统卸载" -ForegroundColor Red
-Write-Host "==========================================" -ForegroundColor Red
-Write-Host ""
-
-Write-Host "将要删除:" -ForegroundColor Yellow
-Write-Host "- 程序目录: $ProgramPath" -ForegroundColor White
-Write-Host "- 数据目录: $DataPath (包含日志文件)" -ForegroundColor White
-Write-Host "- 任务计划程序条目" -ForegroundColor White
-Write-Host "- 事件日志源" -ForegroundColor White
-Write-Host ""
-
-$confirmation = Read-Host "确定要继续吗？这将删除所有相关文件和配置 (y/N)"
-
-if ($confirmation -ne 'y' -and $confirmation -ne 'Y') {
-    Write-Host "卸载已取消" -ForegroundColor Green
-    exit 0
-}
-
-Write-Host ""
-Write-Host "正在卸载..." -ForegroundColor Yellow
-
-# 停止并删除任务计划程序
-try {
-    $task = Get-ScheduledTask -TaskName "Smart Network Shutdown Monitor" -ErrorAction SilentlyContinue
-    if ($task) {
-        Stop-ScheduledTask -TaskName "Smart Network Shutdown Monitor" -ErrorAction SilentlyContinue
-        Unregister-ScheduledTask -TaskName "Smart Network Shutdown Monitor" -Confirm:$false
-        Write-Host "[OK] 任务计划程序已删除" -ForegroundColor Green
-    }
-}
-catch {
-    Write-Host "[X] 删除任务计划程序失败: $($_.Exception.Message)" -ForegroundColor Red
-}
-
-# 删除事件日志源
-try {
-    if ([System.Diagnostics.EventLog]::SourceExists("SmartNetworkMonitor")) {
-        [System.Diagnostics.EventLog]::DeleteEventSource("SmartNetworkMonitor")
-        Write-Host "[OK] 事件日志源已删除" -ForegroundColor Green
-    }
-}
-catch {
-    Write-Host "[X] 删除事件日志源失败: $($_.Exception.Message)" -ForegroundColor Red
-}
-
-# 删除程序目录
-try {
-    if (Test-Path $ProgramPath) {
-        Remove-Item -Path $ProgramPath -Recurse -Force
-        Write-Host "[OK] 程序目录已删除: $ProgramPath" -ForegroundColor Green
-    }
-}
-catch {
-    Write-Host "[X] 删除程序目录失败: $($_.Exception.Message)" -ForegroundColor Red
-}
-
-# 询问是否删除数据目录（包含日志）
-Write-Host ""
-$deleteData = Read-Host "是否同时删除数据目录和所有日志文件？(y/N)"
-
-if ($deleteData -eq 'y' -or $deleteData -eq 'Y') {
-    try {
-        if (Test-Path $DataPath) {
-            Remove-Item -Path $DataPath -Recurse -Force
-            Write-Host "[OK] 数据目录已删除: $DataPath" -ForegroundColor Green
-        }
-    }
-    catch {
-        Write-Host "[X] 删除数据目录失败: $($_.Exception.Message)" -ForegroundColor Red
-    }
-} else {
-    Write-Host "[INFO] 数据目录保留: $DataPath" -ForegroundColor Cyan
-}
-
-Write-Host ""
-Write-Host "[OK] 卸载完成！" -ForegroundColor Green
-Read-Host "按任意键退出"