705 lines
25 KiB
Go
705 lines
25 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"html/template"
|
|
"io"
|
|
"log"
|
|
"fmt"
|
|
"math/rand"
|
|
"net/http"
|
|
"os"
|
|
"sort"
|
|
"strings"
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
type Task struct {
|
|
ID string `json:"id"`
|
|
Name string `json:"name"`
|
|
URL string `json:"url"`
|
|
PageLoaded string `json:"page_loaded"`
|
|
OutOfStock string `json:"out_of_stock"`
|
|
InStock bool `json:"in_stock"`
|
|
LastCheck time.Time `json:"last_check"`
|
|
Status string `json:"status"`
|
|
Notified bool `json:"notified"`
|
|
NotifyEnabled bool `json:"notify_enabled"`
|
|
Order int `json:"order"`
|
|
History []HistoryItem `json:"history"`
|
|
}
|
|
|
|
type Config struct {
|
|
GotifyURL string `json:"gotify_url"`
|
|
GotifyToken string `json:"gotify_token"`
|
|
Interval int `json:"interval"`
|
|
Timeout int `json:"timeout"`
|
|
NotifyEnabled bool `json:"notify_enabled"`
|
|
}
|
|
|
|
type HistoryItem struct {
|
|
State string `json:"state"`
|
|
Time time.Time `json:"time"`
|
|
}
|
|
|
|
var (
|
|
tasks = make(map[string]*Task)
|
|
config = &Config{Interval: 60, Timeout: 20, NotifyEnabled: true}
|
|
mu sync.RWMutex
|
|
configMu sync.RWMutex
|
|
authToken = os.Getenv("AUTH_TOKEN")
|
|
client *http.Client
|
|
)
|
|
|
|
func main() {
|
|
loadEnv()
|
|
loadTasks()
|
|
loadConfig()
|
|
updateClient()
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
go checker(ctx)
|
|
|
|
http.HandleFunc("/", handleIndex)
|
|
http.HandleFunc("/api/tasks", auth(handleTasks))
|
|
http.HandleFunc("/api/task", auth(handleTask))
|
|
http.HandleFunc("/api/task/toggle-notify", auth(handleToggleNotify))
|
|
http.HandleFunc("/api/config", auth(handleConfig))
|
|
http.HandleFunc("/api/test-notification", auth(handleTestNotification))
|
|
|
|
port := os.Getenv("PORT")
|
|
if port == "" {
|
|
port = "8080"
|
|
}
|
|
log.Println("启动服务:", port)
|
|
log.Fatal(http.ListenAndServe(":"+port, nil))
|
|
}
|
|
|
|
func loadEnv() {
|
|
data, err := os.ReadFile(".env")
|
|
if err != nil {
|
|
return
|
|
}
|
|
for _, line := range strings.Split(string(data), "\n") {
|
|
line = strings.TrimSpace(line)
|
|
if line == "" || strings.HasPrefix(line, "#") {
|
|
continue
|
|
}
|
|
parts := strings.SplitN(line, "=", 2)
|
|
if len(parts) == 2 {
|
|
os.Setenv(parts[0], parts[1])
|
|
}
|
|
}
|
|
}
|
|
|
|
func auth(next http.HandlerFunc) http.HandlerFunc {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
if authToken != "" && r.Header.Get("Authorization") != "Bearer "+authToken {
|
|
http.Error(w, "Unauthorized", 401)
|
|
return
|
|
}
|
|
next(w, r)
|
|
}
|
|
}
|
|
|
|
func handleIndex(w http.ResponseWriter, r *http.Request) {
|
|
tmpl := `<!DOCTYPE html>
|
|
<html>
|
|
<head>
|
|
<meta charset="utf-8">
|
|
<meta name="viewport" content="width=device-width,initial-scale=1">
|
|
<title>库存监控</title>
|
|
<style>
|
|
*{margin:0;padding:0;box-sizing:border-box}
|
|
:root{
|
|
--bg:#f5f5f5;--card:#fff;--text:#333;--border:#ddd;--th-bg:#f8f9fa;
|
|
--green:#4CAF50;--red:#f44336;--orange:#ff9800;--blue:#2196F3;--gray:#999
|
|
}
|
|
@media(prefers-color-scheme:dark){
|
|
:root{
|
|
--bg:#1a1a1a;--card:#2d2d2d;--text:#e0e0e0;--border:#404040;--th-bg:#383838
|
|
}
|
|
}
|
|
body{font:14px sans-serif;padding:20px;background:var(--bg);color:var(--text);transition:all .3s}
|
|
.container{max-width:1400px;margin:0 auto}
|
|
h1{margin-bottom:20px;text-align:center}
|
|
.header{display:flex;justify-content:center;gap:10px;margin-bottom:20px}
|
|
.add-btn,.config-btn{background:var(--green);color:#fff;border:none;padding:10px 20px;cursor:pointer;border-radius:4px}
|
|
.config-btn{background:var(--blue)}
|
|
table{width:100%;background:var(--card);border-collapse:collapse;box-shadow:0 2px 8px rgba(0,0,0,.15);border-radius:8px;overflow:hidden}
|
|
th,td{padding:12px;text-align:left;border-bottom:1px solid var(--border)}
|
|
th{background:var(--th-bg);font-weight:600}
|
|
tbody tr:hover{background:rgba(33,150,243,.05)}
|
|
.status-cell{text-align:center}
|
|
.status{display:inline-block;width:10px;height:10px;border-radius:50%}
|
|
.status.ok{background:var(--green)}
|
|
.status.error{background:var(--red)}
|
|
.status.checking{background:var(--orange)}
|
|
.stock-yes{color:var(--green);font-weight:600}
|
|
.stock-no{color:var(--gray)}
|
|
.btn{padding:5px 10px;margin:0 2px;cursor:pointer;border:none;border-radius:3px;background:var(--blue);color:#fff;font-size:12px}
|
|
.btn-del{background:var(--red)}
|
|
.btn-icon{background:transparent;color:var(--text);font-size:16px;padding:5px 8px}
|
|
.btn-icon.active{color:var(--green)}
|
|
.uptime-bar{display:flex;gap:2px;height:30px}
|
|
.uptime-item{flex:1;border-radius:2px;cursor:pointer;position:relative;transition:transform .2s}
|
|
.uptime-item:hover{transform:scaleY(1.2)}
|
|
.uptime-item.stock{background:var(--green)}
|
|
.uptime-item.no-stock{background:#555}
|
|
.uptime-item.error{background:var(--orange)}
|
|
.uptime-item.unknown{background:#333}
|
|
.uptime-tooltip{position:absolute;bottom:35px;left:50%;transform:translateX(-50%);background:#000;color:#fff;padding:4px 8px;border-radius:4px;font-size:11px;white-space:nowrap;opacity:0;pointer-events:none;transition:opacity .2s;z-index:10}
|
|
.uptime-item:hover .uptime-tooltip{opacity:1}
|
|
.modal{display:none;position:fixed;top:0;left:0;right:0;bottom:0;background:rgba(0,0,0,.6);align-items:center;justify-content:center;z-index:1000}
|
|
.modal.show{display:flex}
|
|
.modal-content{background:var(--card);padding:24px;border-radius:8px;width:90%;max-width:500px;box-shadow:0 4px 20px rgba(0,0,0,.3)}
|
|
.form-group{margin-bottom:15px}
|
|
.form-group.checkbox-group{display:flex;align-items:center;justify-content:flex-end;gap:10px}
|
|
.form-group.checkbox-group label{margin:0;font-weight:600}
|
|
.form-group.checkbox-group input[type="checkbox"]{width:auto;margin:0}
|
|
label{display:block;margin-bottom:5px;font-weight:600}
|
|
input{width:100%;padding:10px;border:1px solid var(--border);border-radius:4px;background:var(--card);color:var(--text)}
|
|
.form-actions{display:flex;gap:10px;justify-content:flex-end;margin-top:20px}
|
|
a{color:var(--blue)}
|
|
@media(max-width:768px){
|
|
.container{padding:10px}
|
|
table{font-size:12px}
|
|
th,td{padding:8px}
|
|
}
|
|
</style>
|
|
</head>
|
|
<body>
|
|
<div class="container">
|
|
<h1>库存监控</h1>
|
|
<div class="header">
|
|
<button class="config-btn" onclick="showConfigModal()">⚙ 系统设置</button>
|
|
<button class="add-btn" onclick="showModal()">+ 添加监控</button>
|
|
</div>
|
|
<table id="taskTable">
|
|
<thead>
|
|
<tr>
|
|
<th width="60">状态</th>
|
|
<th width="150">名称</th>
|
|
<th>监控历史</th>
|
|
<th width="80">库存</th>
|
|
<th width="150">最后检测</th>
|
|
<th width="180">操作</th>
|
|
</tr>
|
|
</thead>
|
|
<tbody></tbody>
|
|
</table>
|
|
</div>
|
|
|
|
<div class="modal" id="modal">
|
|
<div class="modal-content">
|
|
<h2 id="modalTitle">添加监控</h2>
|
|
<form id="taskForm">
|
|
<input type="hidden" id="taskId">
|
|
<div class="form-group">
|
|
<label>名称</label>
|
|
<input id="name" required>
|
|
</div>
|
|
<div class="form-group">
|
|
<label>URL</label>
|
|
<input id="url" required>
|
|
</div>
|
|
<div class="form-group">
|
|
<label>页面加载标识</label>
|
|
<input id="pageLoaded" placeholder="如: froots/js-js" required>
|
|
</div>
|
|
<div class="form-group">
|
|
<label>缺货标识</label>
|
|
<input id="outOfStock" placeholder="如: Out of Stock" required>
|
|
</div>
|
|
<div class="form-group">
|
|
<label>排序</label>
|
|
<input id="order" type="number" value="0">
|
|
</div>
|
|
<div class="form-actions">
|
|
<button type="button" class="btn" onclick="hideModal()">取消</button>
|
|
<button type="submit" class="btn">保存</button>
|
|
</div>
|
|
</form>
|
|
</div>
|
|
</div>
|
|
|
|
<div class="modal" id="configModal">
|
|
<div class="modal-content">
|
|
<h2>系统设置</h2>
|
|
<form id="configForm">
|
|
<div class="form-group checkbox-group">
|
|
<label>启用通知推送</label>
|
|
<input type="checkbox" id="notifyEnabled">
|
|
</div>
|
|
<div class="form-group">
|
|
<label>Gotify URL</label>
|
|
<input id="gotifyUrl" placeholder="https://gotify.example.com">
|
|
</div>
|
|
<div class="form-group">
|
|
<label>Gotify Token</label>
|
|
<input id="gotifyToken" placeholder="通知令牌">
|
|
</div>
|
|
<div class="form-group">
|
|
<label>检测间隔(秒)</label>
|
|
<input id="interval" type="number" min="10" required>
|
|
</div>
|
|
<div class="form-group">
|
|
<label>超时时间(秒)</label>
|
|
<input id="timeout" type="number" min="5" required>
|
|
</div>
|
|
<div class="form-actions">
|
|
<button type="button" class="btn" onclick="testNotification()">测试通知</button>
|
|
<button type="button" class="btn" onclick="hideConfigModal()">取消</button>
|
|
<button type="submit" class="btn">保存</button>
|
|
</div>
|
|
</form>
|
|
</div>
|
|
</div>
|
|
|
|
<script>
|
|
const token = localStorage.getItem('token') || prompt('请输入访问令牌:');
|
|
if(token) localStorage.setItem('token', token);
|
|
const headers = {'Authorization': 'Bearer ' + token};
|
|
|
|
function loadTasks() {
|
|
fetch('/api/tasks', {headers})
|
|
.then(r => r.json())
|
|
.then(tasks => {
|
|
const tbody = document.querySelector('#taskTable tbody');
|
|
tbody.innerHTML = tasks.map(t => {
|
|
const statusClass = t.status || 'checking';
|
|
const stockText = t.in_stock ? '<span class="stock-yes">有货</span>' : '<span class="stock-no">无货</span>';
|
|
const lastCheck = t.last_check ? new Date(t.last_check).toLocaleString('zh-CN') : '-';
|
|
const notifyIcon = t.notify_enabled ? '🔔' : '🔕';
|
|
const notifyClass = t.notify_enabled ? 'active' : '';
|
|
|
|
const history = t.history || [];
|
|
const uptimeBar = history.map(h => {
|
|
const time = new Date(h.time).toLocaleString('zh-CN');
|
|
const label = h.state === 'stock' ? '有货' : h.state === 'no-stock' ? '无货' : h.state === 'error' ? '错误' : '未知';
|
|
return '<div class="uptime-item ' + h.state + '"><span class="uptime-tooltip">' + time + '<br>' + label + '</span></div>';
|
|
}).join('');
|
|
return '<tr>' +
|
|
'<td class="status-cell"><span class="status ' + statusClass + '"></span></td>' +
|
|
'<td>' + t.name + '</td>' +
|
|
'<td><div class="uptime-bar">' + uptimeBar + '</div></td>' +
|
|
'<td>' + stockText + '</td>' +
|
|
'<td>' + lastCheck + '</td>' +
|
|
'<td>' +
|
|
'<button class="btn btn-icon ' + notifyClass + '" onclick="toggleNotify(\'' + t.id + '\')" title="通知开关">' + notifyIcon + '</button>' +
|
|
'<a href="' + t.url + '" target="_blank" class="btn btn-icon" title="访问">🔗</a>' +
|
|
'<button class="btn" onclick="editTask(\'' + t.id + '\')">编辑</button>' +
|
|
'<button class="btn btn-del" onclick="delTask(\'' + t.id + '\')">删除</button>' +
|
|
'</td></tr>';
|
|
}).join('');
|
|
});
|
|
}
|
|
|
|
function toggleNotify(id) {
|
|
fetch('/api/task/toggle-notify?id=' + id, {method: 'POST', headers})
|
|
.then(() => loadTasks());
|
|
}
|
|
|
|
function showModal(task) {
|
|
document.getElementById('modalTitle').textContent = task ? '编辑监控' : '添加监控';
|
|
document.getElementById('taskId').value = task?.id || '';
|
|
document.getElementById('name').value = task?.name || '';
|
|
document.getElementById('url').value = task?.url || '';
|
|
document.getElementById('pageLoaded').value = task?.page_loaded || '';
|
|
document.getElementById('outOfStock').value = task?.out_of_stock || '';
|
|
document.getElementById('order').value = task?.order || 0;
|
|
document.getElementById('modal').classList.add('show');
|
|
}
|
|
|
|
function hideModal() {
|
|
document.getElementById('modal').classList.remove('show');
|
|
}
|
|
|
|
function showConfigModal() {
|
|
fetch('/api/config', {headers})
|
|
.then(r => r.json())
|
|
.then(cfg => {
|
|
document.getElementById('gotifyUrl').value = cfg.gotify_url || '';
|
|
document.getElementById('gotifyToken').value = cfg.gotify_token || '';
|
|
document.getElementById('interval').value = cfg.interval || 60;
|
|
document.getElementById('timeout').value = cfg.timeout || 20;
|
|
document.getElementById('notifyEnabled').checked = cfg.notify_enabled !== false;
|
|
document.getElementById('configModal').classList.add('show');
|
|
});
|
|
}
|
|
|
|
function hideConfigModal() {
|
|
document.getElementById('configModal').classList.remove('show');
|
|
}
|
|
|
|
function testNotification() {
|
|
const testConfig = {
|
|
gotify_url: document.getElementById('gotifyUrl').value,
|
|
gotify_token: document.getElementById('gotifyToken').value
|
|
};
|
|
|
|
fetch('/api/test-notification', {
|
|
method: 'POST',
|
|
headers: {...headers, 'Content-Type': 'application/json'},
|
|
body: JSON.stringify(testConfig)
|
|
}).then(r => r.ok ? alert('测试通知已发送,请检查 Gotify') : alert('发送失败,查看后台日志'));
|
|
}
|
|
|
|
function editTask(id) {
|
|
fetch('/api/tasks', {headers})
|
|
.then(r => r.json())
|
|
.then(tasks => {
|
|
const task = tasks.find(t => t.id === id);
|
|
if(task) showModal(task);
|
|
});
|
|
}
|
|
|
|
function delTask(id) {
|
|
if(!confirm('确认删除?')) return;
|
|
fetch('/api/task?id=' + id, {method: 'DELETE', headers})
|
|
.then(() => loadTasks());
|
|
}
|
|
|
|
document.getElementById('taskForm').onsubmit = e => {
|
|
e.preventDefault();
|
|
const data = {
|
|
id: document.getElementById('taskId').value,
|
|
name: document.getElementById('name').value,
|
|
url: document.getElementById('url').value,
|
|
page_loaded: document.getElementById('pageLoaded').value,
|
|
out_of_stock: document.getElementById('outOfStock').value,
|
|
order: parseInt(document.getElementById('order').value) || 0
|
|
};
|
|
fetch('/api/task', {
|
|
method: 'POST',
|
|
headers: {...headers, 'Content-Type': 'application/json'},
|
|
body: JSON.stringify(data)
|
|
}).then(() => {
|
|
hideModal();
|
|
loadTasks();
|
|
});
|
|
};
|
|
|
|
document.getElementById('configForm').onsubmit = e => {
|
|
e.preventDefault();
|
|
const data = {
|
|
gotify_url: document.getElementById('gotifyUrl').value,
|
|
gotify_token: document.getElementById('gotifyToken').value,
|
|
interval: parseInt(document.getElementById('interval').value),
|
|
timeout: parseInt(document.getElementById('timeout').value),
|
|
notify_enabled: document.getElementById('notifyEnabled').checked
|
|
};
|
|
fetch('/api/config', {
|
|
method: 'POST',
|
|
headers: {...headers, 'Content-Type': 'application/json'},
|
|
body: JSON.stringify(data)
|
|
}).then(() => {
|
|
hideConfigModal();
|
|
alert('设置已保存并生效');
|
|
});
|
|
};
|
|
|
|
loadTasks();
|
|
setInterval(loadTasks, 30000);
|
|
</script>
|
|
</body>
|
|
</html>`
|
|
w.Header().Set("Content-Type", "text/html")
|
|
template.Must(template.New("").Parse(tmpl)).Execute(w, nil)
|
|
}
|
|
|
|
func handleTasks(w http.ResponseWriter, r *http.Request) {
|
|
mu.RLock()
|
|
list := make([]*Task, 0, len(tasks))
|
|
for _, t := range tasks {
|
|
list = append(list, t)
|
|
}
|
|
mu.RUnlock()
|
|
|
|
sort.Slice(list, func(i, j int) bool {
|
|
return list[i].Order < list[j].Order
|
|
})
|
|
|
|
json.NewEncoder(w).Encode(list)
|
|
}
|
|
|
|
func handleTask(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method == "DELETE" {
|
|
id := r.URL.Query().Get("id")
|
|
mu.Lock()
|
|
delete(tasks, id)
|
|
mu.Unlock()
|
|
saveTasks()
|
|
return
|
|
}
|
|
|
|
var task Task
|
|
json.NewDecoder(r.Body).Decode(&task)
|
|
isNew := task.ID == ""
|
|
if isNew {
|
|
task.ID = time.Now().Format("20060102150405")
|
|
task.NotifyEnabled = true
|
|
task.History = make([]HistoryItem, 90)
|
|
now := time.Now()
|
|
for i := 0; i < 90; i++ {
|
|
task.History[i] = HistoryItem{
|
|
State: "unknown",
|
|
Time: now.Add(time.Duration(i-90) * time.Minute),
|
|
}
|
|
}
|
|
} else {
|
|
mu.RLock()
|
|
if existing, ok := tasks[task.ID]; ok {
|
|
task.History = existing.History
|
|
task.InStock = existing.InStock
|
|
task.LastCheck = existing.LastCheck
|
|
task.Status = existing.Status
|
|
task.Notified = existing.Notified
|
|
task.NotifyEnabled = existing.NotifyEnabled
|
|
}
|
|
mu.RUnlock()
|
|
}
|
|
|
|
mu.Lock()
|
|
tasks[task.ID] = &task
|
|
mu.Unlock()
|
|
saveTasks()
|
|
}
|
|
|
|
func handleToggleNotify(w http.ResponseWriter, r *http.Request) {
|
|
id := r.URL.Query().Get("id")
|
|
mu.Lock()
|
|
if task, ok := tasks[id]; ok {
|
|
task.NotifyEnabled = !task.NotifyEnabled
|
|
}
|
|
mu.Unlock()
|
|
saveTasks()
|
|
}
|
|
|
|
func handleConfig(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method == "POST" {
|
|
var cfg Config
|
|
json.NewDecoder(r.Body).Decode(&cfg)
|
|
configMu.Lock()
|
|
config = &cfg
|
|
configMu.Unlock()
|
|
updateClient()
|
|
saveConfig()
|
|
return
|
|
}
|
|
|
|
configMu.RLock()
|
|
defer configMu.RUnlock()
|
|
json.NewEncoder(w).Encode(config)
|
|
}
|
|
|
|
func checker(ctx context.Context) {
|
|
for {
|
|
select {
|
|
case <-ctx.Done():
|
|
return
|
|
default:
|
|
}
|
|
|
|
configMu.RLock()
|
|
interval := config.Interval
|
|
configMu.RUnlock()
|
|
|
|
mu.RLock()
|
|
list := make([]*Task, 0, len(tasks))
|
|
for _, t := range tasks {
|
|
list = append(list, t)
|
|
}
|
|
mu.RUnlock()
|
|
|
|
for _, task := range list {
|
|
go checkTask(task)
|
|
time.Sleep(time.Duration(2+rand.Intn(5)) * time.Second)
|
|
}
|
|
|
|
jitter := rand.Intn(20) - 10
|
|
time.Sleep(time.Duration(interval+jitter) * time.Second)
|
|
}
|
|
}
|
|
|
|
func checkTask(task *Task) {
|
|
body, err := fetch(task.URL)
|
|
now := time.Now()
|
|
|
|
status := "ok"
|
|
inStock := false
|
|
|
|
if err != nil || !strings.Contains(body, task.PageLoaded) {
|
|
status = "error"
|
|
} else {
|
|
inStock = !strings.Contains(body, task.OutOfStock)
|
|
}
|
|
|
|
state := "unknown"
|
|
if status == "ok" {
|
|
state = "no-stock"
|
|
if inStock {
|
|
state = "stock"
|
|
}
|
|
} else if status == "error" {
|
|
state = "error"
|
|
}
|
|
|
|
mu.Lock()
|
|
wasInStock := task.InStock
|
|
wasNotified := task.Notified
|
|
taskNotifyEnabled := task.NotifyEnabled
|
|
task.Status = status
|
|
task.InStock = inStock
|
|
task.LastCheck = now
|
|
|
|
if task.History == nil {
|
|
task.History = []HistoryItem{}
|
|
}
|
|
task.History = append(task.History, HistoryItem{State: state, Time: now})
|
|
if len(task.History) > 90 {
|
|
task.History = task.History[len(task.History)-90:]
|
|
}
|
|
|
|
if !inStock {
|
|
task.Notified = false
|
|
}
|
|
mu.Unlock()
|
|
saveTasks()
|
|
|
|
configMu.RLock()
|
|
globalNotifyEnabled := config.NotifyEnabled
|
|
configMu.RUnlock()
|
|
|
|
if globalNotifyEnabled && taskNotifyEnabled && inStock && !wasInStock && !wasNotified {
|
|
notify(task.Name + " 有货了！", task.URL)
|
|
mu.Lock()
|
|
task.Notified = true
|
|
mu.Unlock()
|
|
saveTasks()
|
|
}
|
|
}
|
|
|
|
func fetch(url string) (string, error) {
|
|
req, _ := http.NewRequest("GET", url, nil)
|
|
req.Header.Set("User-Agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36")
|
|
req.Header.Set("Accept", "text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8")
|
|
req.Header.Set("Accept-Language", "en-US,en;q=0.9")
|
|
|
|
resp, err := client.Do(req)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
defer resp.Body.Close()
|
|
body, _ := io.ReadAll(resp.Body)
|
|
return string(body), nil
|
|
}
|
|
|
|
func notify(title, msg string) {
|
|
configMu.RLock()
|
|
url, token := config.GotifyURL, config.GotifyToken
|
|
configMu.RUnlock()
|
|
|
|
if url == "" || token == "" {
|
|
return
|
|
}
|
|
|
|
go func() {
|
|
endpoint := strings.TrimRight(url, "/") + "/message"
|
|
payload := fmt.Sprintf(`{"title":"%s","message":"%s","priority":10}`, title, msg)
|
|
req, _ := http.NewRequest("POST", endpoint, strings.NewReader(payload))
|
|
req.Header.Set("Content-Type", "application/json")
|
|
req.Header.Set("X-Gotify-Key", token)
|
|
|
|
resp, err := http.DefaultClient.Do(req)
|
|
if err != nil {
|
|
log.Printf("发送通知失败: %v", err)
|
|
return
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
if resp.StatusCode != 200 {
|
|
body, _ := io.ReadAll(resp.Body)
|
|
log.Printf("通知返回错误 %d: %s", resp.StatusCode, body)
|
|
}
|
|
}()
|
|
}
|
|
|
|
func handleTestNotification(w http.ResponseWriter, r *http.Request) {
|
|
var testConfig struct {
|
|
GotifyURL string `json:"gotify_url"`
|
|
GotifyToken string `json:"gotify_token"`
|
|
}
|
|
json.NewDecoder(r.Body).Decode(&testConfig)
|
|
|
|
if testConfig.GotifyURL == "" || testConfig.GotifyToken == "" {
|
|
http.Error(w, "缺少配置", 400)
|
|
return
|
|
}
|
|
|
|
endpoint := strings.TrimRight(testConfig.GotifyURL, "/") + "/message"
|
|
payload := `{"title":"测试通知","message":"Gotify 配置正常","priority":10}`
|
|
req, _ := http.NewRequest("POST", endpoint, strings.NewReader(payload))
|
|
req.Header.Set("Content-Type", "application/json")
|
|
req.Header.Set("X-Gotify-Key", testConfig.GotifyToken)
|
|
|
|
resp, err := http.DefaultClient.Do(req)
|
|
if err != nil {
|
|
http.Error(w, err.Error(), 500)
|
|
return
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
if resp.StatusCode != 200 {
|
|
body, _ := io.ReadAll(resp.Body)
|
|
http.Error(w, string(body), resp.StatusCode)
|
|
return
|
|
}
|
|
|
|
json.NewEncoder(w).Encode(map[string]string{"status": "ok"})
|
|
}
|
|
|
|
func updateClient() {
|
|
configMu.RLock()
|
|
timeout := config.Timeout
|
|
configMu.RUnlock()
|
|
client = &http.Client{Timeout: time.Duration(timeout) * time.Second}
|
|
}
|
|
|
|
func saveJSON(filename string, v interface{}) {
|
|
os.MkdirAll("data", 0755)
|
|
data, _ := json.MarshalIndent(v, "", " ")
|
|
os.WriteFile("data/"+filename, data, 0644)
|
|
}
|
|
func loadTasks() {
|
|
data, err := os.ReadFile("data/tasks.json")
|
|
if err != nil {
|
|
return
|
|
}
|
|
var list []*Task
|
|
json.Unmarshal(data, &list)
|
|
mu.Lock()
|
|
for _, t := range list {
|
|
tasks[t.ID] = t
|
|
}
|
|
mu.Unlock()
|
|
}
|
|
func loadConfig() {
|
|
data, err := os.ReadFile("data/config.json")
|
|
if err != nil {
|
|
return
|
|
}
|
|
configMu.Lock()
|
|
json.Unmarshal(data, config)
|
|
configMu.Unlock()
|
|
}
|
|
|
|
func saveConfig() {
|
|
configMu.RLock()
|
|
defer configMu.RUnlock()
|
|
saveJSON("config.json", config)
|
|
}
|