Compare commits
19
Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
3ad7faba3b | ||
|
|
0c8ac02f82 | ||
|
|
6449de0cdb | ||
|
|
d27cc47d8c | ||
|
|
bb6a446cad | ||
|
|
09e0996b9d | ||
|
|
97bbe3ced6 | ||
|
|
46af20e96c | ||
|
|
c3114492bb | ||
|
|
ff04cb0c1a | ||
|
|
16ece31d57 | ||
|
|
8d8b201498 | ||
|
|
5744bf5585 | ||
|
|
9a070b77af | ||
|
|
886cb2af27 | ||
|
|
0381fe3f2f | ||
|
|
c11aadaa68 | ||
|
|
80db7a482a | ||
|
|
d84db86486 |
@@ -2,6 +2,8 @@ name: Release
|
|||||||
|
|
||||||
on:
|
on:
|
||||||
push:
|
push:
|
||||||
|
branches:
|
||||||
|
- main
|
||||||
tags:
|
tags:
|
||||||
- 'v*'
|
- 'v*'
|
||||||
|
|
||||||
@@ -15,6 +17,53 @@ jobs:
|
|||||||
steps:
|
steps:
|
||||||
- name: Checkout code
|
- name: Checkout code
|
||||||
uses: actions/checkout@v4
|
uses: actions/checkout@v4
|
||||||
|
with:
|
||||||
|
fetch-depth: 0
|
||||||
|
|
||||||
|
- name: Determine Version Tag & Generate Notes
|
||||||
|
id: versioning
|
||||||
|
run: |
|
||||||
|
PREV_TAG=$(git tag -l "v*" --sort=-v:refname | head -n1)
|
||||||
|
echo "Previous tag: $PREV_TAG"
|
||||||
|
|
||||||
|
if [[ "${{ github.ref }}" == refs/tags/v* ]]; then
|
||||||
|
TAG="${GITHUB_REF#refs/tags/}"
|
||||||
|
else
|
||||||
|
if [ -z "$PREV_TAG" ]; then
|
||||||
|
TAG="v1.0.0"
|
||||||
|
else
|
||||||
|
VER_NUM="${PREV_TAG#v}"
|
||||||
|
IFS='.' read -r MAJOR MINOR PATCH <<< "$VER_NUM"
|
||||||
|
PATCH=$((PATCH + 1))
|
||||||
|
TAG="v${MAJOR}.${MINOR}.${PATCH}"
|
||||||
|
fi
|
||||||
|
echo "Auto incremented patch version to $TAG"
|
||||||
|
|
||||||
|
git config user.name "github-actions[bot]"
|
||||||
|
git config user.email "github-actions[bot]@users.noreply.github.com"
|
||||||
|
git tag "$TAG"
|
||||||
|
git push origin "$TAG"
|
||||||
|
fi
|
||||||
|
echo "tag=$TAG" >> $GITHUB_OUTPUT
|
||||||
|
|
||||||
|
# 生成详细的 Release Changelog 正文
|
||||||
|
cat << NOTES > release_notes.md
|
||||||
|
### 🚀 What's Changed in ${TAG}
|
||||||
|
|
||||||
|
NOTES
|
||||||
|
|
||||||
|
if [ -n "$PREV_TAG" ]; then
|
||||||
|
git log "${PREV_TAG}..HEAD" --pretty=format:"- %s (%h by @kyaring)" >> release_notes.md
|
||||||
|
else
|
||||||
|
git log -n 10 --pretty=format:"- %s (%h by @kyaring)" >> release_notes.md
|
||||||
|
fi
|
||||||
|
|
||||||
|
echo "" >> release_notes.md
|
||||||
|
echo "" >> release_notes.md
|
||||||
|
if [ -n "$PREV_TAG" ]; then
|
||||||
|
echo "**Full Changelog**: https://github.com/${{ github.repository }}/compare/${PREV_TAG}...${TAG}" >> release_notes.md
|
||||||
|
fi
|
||||||
|
cat release_notes.md
|
||||||
|
|
||||||
- name: Set up Go
|
- name: Set up Go
|
||||||
uses: actions/setup-go@v5
|
uses: actions/setup-go@v5
|
||||||
@@ -40,6 +89,8 @@ jobs:
|
|||||||
- name: Create GitHub Release
|
- name: Create GitHub Release
|
||||||
uses: softprops/action-gh-release@v2
|
uses: softprops/action-gh-release@v2
|
||||||
with:
|
with:
|
||||||
|
tag_name: ${{ steps.versioning.outputs.tag }}
|
||||||
|
name: Release ${{ steps.versioning.outputs.tag }}
|
||||||
|
body_path: release_notes.md
|
||||||
files: |
|
files: |
|
||||||
dist/*
|
dist/*
|
||||||
generate_release_notes: true
|
|
||||||
|
|||||||
@@ -73,17 +73,27 @@
|
|||||||
|
|
||||||
---
|
---
|
||||||
|
|
||||||
## 🚀 极速一键部署 (推荐)
|
## 🚀 极速一键部署与平滑升级
|
||||||
|
|
||||||
|
### 1. 一键全自动安装 (推荐)
|
||||||
无论是在受控端 VPS 还是中心端,只需一行命令即可交互式安装与启动:
|
无论是在受控端 VPS 还是中心端,只需一行命令即可交互式安装与启动:
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
bash <(curl -sSL https://raw.githubusercontent.com/kyaring/vps-commander/main/deploy/install.sh)
|
bash <(curl -fsSLk https://gitea.king.nyc.mn/openclaw/vps-commander/raw/branch/main/deploy/install.sh)
|
||||||
```
|
```
|
||||||
|
|
||||||
- **安装 Agent (受控 VPS)**:输入 Hub 的 WSS 地址与通信密钥,全自动配置开机自启并秒级连入集群。
|
- **安装 Agent (受控 VPS)**:输入 Hub 的 WSS 地址与通信密钥,全自动配置开机自启并秒级连入集群。
|
||||||
- **安装 Hub (中心服务端)**:自动生成安全 API Key / Web 面板密码并以 Systemd 托管常驻。
|
- **安装 Hub (中心服务端)**:自动生成安全 API Key / Web 面板密码并以 Systemd 托管常驻。
|
||||||
|
|
||||||
|
### 2. 受控端 Agent 一键平滑升级 (免重输配置)
|
||||||
|
当新版本发布后,无需重新配置通信密钥或重新安装,在任意受控机上直接运行升级脚本:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
curl -fsSLk https://gitea.king.nyc.mn/openclaw/vps-commander/raw/branch/main/deploy/upgrade-agent.sh | bash
|
||||||
|
```
|
||||||
|
- 自动检测并拉取 Gitea 最新版本架构包(AMD64 / ARM64);
|
||||||
|
- 自动备份旧版本二进制(`.bak`);
|
||||||
|
- 升级异常(校验失败或无法启动)时**自动秒级回滚**,保证节点永不失联。
|
||||||
|
|
||||||
---
|
---
|
||||||
|
|
||||||
## 🛠️ 手动编译部署
|
## 🛠️ 手动编译部署
|
||||||
@@ -140,20 +150,51 @@ Agent 启动后会通过 WSS 自动连入 Hub,在 Hub 管理面板与 `/api/v1
|
|||||||
|
|
||||||
---
|
---
|
||||||
|
|
||||||
## 🤖 接入 ChatGPT / AI 客户端
|
## 🤖 接入 ChatGPT (新版插件 Apps & MCP 模式)
|
||||||
|
|
||||||
### 方式 1:ChatGPT Actions (推荐)
|
当前 ChatGPT 已升级为以 **插件 / 应用 (Apps & MCP)** 规范连接外部服务。VPS-Commander 原生内置完整的 **MCP SSE + OAuth2** 协议支持,无需部署额外适配层。
|
||||||
1. 在 ChatGPT 创建自定义 GPT(GPT Builder)-> **Actions** -> **Create new action**。
|
|
||||||
2. **Schema 导入**:在 `Import from URL` 填入 `https://your-domain.com/openapi.json`。
|
|
||||||
3. **Authentication (身份验证)**:
|
|
||||||
- Authentication Type 选择 **API Key**。
|
|
||||||
- Auth Type 选择 **Bearer**。
|
|
||||||
- 填入你配置的 `VPS_COMMANDER_API_KEY`。
|
|
||||||
|
|
||||||
### 方式 2:MCP (Model Context Protocol)
|
### 接入配置步骤:
|
||||||
- **SSE 端点**:`https://your-domain.com/mcp/sse`
|
|
||||||
- **消息交互**:`https://your-domain.com/mcp/message`
|
1. **进入 ChatGPT 插件/应用配置页**:
|
||||||
- **认证**:支持 URL 参数 `?token=...` 或标准 Header `Authorization: Bearer <TOKEN>`。
|
- 打开 ChatGPT -> **Settings (设置)** -> **Apps & Integrations (应用与集成) / Plugins** -> **Create / Add App (添加应用)**。
|
||||||
|
2. **连接方式选择 MCP**:
|
||||||
|
- 连接协议选择 **MCP (Model Context Protocol)** 或 **Server-Sent Events (SSE)**。
|
||||||
|
- **MCP Server URL**:填入 Hub 的 SSE 统一入口:
|
||||||
|
```text
|
||||||
|
https://your-domain.com/mcp/sse
|
||||||
|
```
|
||||||
|
3. **身份认证 (Authentication)**:
|
||||||
|
- **方式 A (OAuth 2.0 自动授权,推荐)**:
|
||||||
|
- Hub 内置了标准 RFC 8414 OAuth Metadata (`/.well-known/oauth-authorization-server`);
|
||||||
|
- 授权端点与 Token 端点由 ChatGPT 自动识别,一键点击「Authorize」即可完成免密安全绑定。
|
||||||
|
- **方式 B (Bearer API Key)**:
|
||||||
|
- 若选择 API Key,填入 Hub 环境变量 `VPS_COMMANDER_API_KEY` 即可。
|
||||||
|
4. **自动发现 4 大运维工具**:
|
||||||
|
- 握手成功后,ChatGPT 会自动加载并列出 4 个原生运维工具:
|
||||||
|
- `list_devices`:列出受控节点、CPU/内存/磁盘剩余绝对值及 Docker 状态;
|
||||||
|
- `exec_command`:在目标机器(如 `m4-live-agent`)执行 Shell 指令;
|
||||||
|
- `read_file`:分页读取目标文件;
|
||||||
|
- `write_file`:远程编辑或创建文件。
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
### 传统客户端支持 (Claude Desktop / Cursor / VS Code)
|
||||||
|
|
||||||
|
支持通过本地 stdio 二进制连接 Hub:
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"mcpServers": {
|
||||||
|
"vps-commander": {
|
||||||
|
"command": "/path/to/vps-commander-mcp-stdio-linux-amd64",
|
||||||
|
"args": [
|
||||||
|
"-hub", "https://your-domain.com",
|
||||||
|
"-key", "YOUR_API_KEY"
|
||||||
|
]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
---
|
---
|
||||||
|
|
||||||
|
|||||||
+41
-5
@@ -10,6 +10,7 @@ import (
|
|||||||
"log"
|
"log"
|
||||||
"net/http"
|
"net/http"
|
||||||
"os"
|
"os"
|
||||||
|
"runtime"
|
||||||
"sync"
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
@@ -69,10 +70,11 @@ func run(hub, name, token string, insecure bool) error {
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
exec := executor.Local{MaxOutput: 1024 * 1024}
|
exec := executor.Local{MaxOutput: 1024 * 1024}
|
||||||
|
mcpRuntime := cluster.NewMCPRuntime()
|
||||||
|
|
||||||
// 初始握手附带画像
|
// 初始握手附带画像
|
||||||
initProf := sysinfo.Collect("APPLICATION")
|
initProf := sysinfo.Collect("APPLICATION")
|
||||||
if err := writeJSON(cluster.Message{Event: "hello", Device: name, Profile: &initProf, TS: time.Now().Unix()}); err != nil {
|
if err := writeJSON(cluster.Message{Event: "hello", Device: name, Arch: runtime.GOARCH, OS: runtime.GOOS, Profile: &initProf, TS: time.Now().Unix()}); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -85,7 +87,7 @@ func run(hub, name, token string, insecure bool) error {
|
|||||||
select {
|
select {
|
||||||
case <-t.C:
|
case <-t.C:
|
||||||
prof := sysinfo.Collect("APPLICATION")
|
prof := sysinfo.Collect("APPLICATION")
|
||||||
_ = writeJSON(cluster.Message{Event: "ping", Device: name, Profile: &prof, TS: time.Now().Unix()})
|
_ = writeJSON(cluster.Message{Event: "ping", Device: name, Arch: runtime.GOARCH, OS: runtime.GOOS, Profile: &prof, TS: time.Now().Unix()})
|
||||||
case <-stopPing:
|
case <-stopPing:
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -101,13 +103,23 @@ func run(hub, name, token string, insecure bool) error {
|
|||||||
if msg.Event == "registered" {
|
if msg.Event == "registered" {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
if msg.Event == "mcp_sync" {
|
||||||
|
var services []cluster.MCPService
|
||||||
|
if err := decodePayload(msg.Payload, &services); err != nil {
|
||||||
|
log.Printf("mcp sync failed: %v", err)
|
||||||
|
} else {
|
||||||
|
mcpRuntime.Sync(services)
|
||||||
|
log.Printf("MCP services synced: %d", len(services))
|
||||||
|
}
|
||||||
|
continue
|
||||||
|
}
|
||||||
if msg.Event == "revoked" {
|
if msg.Event == "revoked" {
|
||||||
log.Printf("device revoked by admin: %s", msg.Error)
|
log.Printf("device revoked by admin: %s", msg.Error)
|
||||||
return errors.New("revoked by admin")
|
return errors.New("revoked by admin")
|
||||||
}
|
}
|
||||||
if msg.Event == "ping" {
|
if msg.Event == "ping" {
|
||||||
prof := sysinfo.Collect("APPLICATION")
|
prof := sysinfo.Collect("APPLICATION")
|
||||||
if err := writeJSON(cluster.Message{Event: "pong", Device: name, Profile: &prof, TS: msg.TS}); err != nil {
|
if err := writeJSON(cluster.Message{Event: "pong", Device: name, Arch: runtime.GOARCH, OS: runtime.GOOS, Profile: &prof, TS: msg.TS}); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
continue
|
continue
|
||||||
@@ -116,7 +128,7 @@ func run(hub, name, token string, insecure bool) error {
|
|||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
go func(msg cluster.Message) {
|
go func(msg cluster.Message) {
|
||||||
if err := handle(writeJSON, exec, msg); err != nil {
|
if err := handle(writeJSON, exec, mcpRuntime, msg); err != nil {
|
||||||
log.Printf("request %s failed: %v", msg.ID, err)
|
log.Printf("request %s failed: %v", msg.ID, err)
|
||||||
_ = c.Close()
|
_ = c.Close()
|
||||||
}
|
}
|
||||||
@@ -124,8 +136,32 @@ func run(hub, name, token string, insecure bool) error {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func handle(writeJSON func(cluster.Message) error, exec executor.Local, msg cluster.Message) error {
|
func handle(writeJSON func(cluster.Message) error, exec executor.Local, mcpRuntime *cluster.MCPRuntime, msg cluster.Message) error {
|
||||||
switch msg.Action {
|
switch msg.Action {
|
||||||
|
case "mcp_list":
|
||||||
|
return writeJSON(cluster.Message{ID: msg.ID, Status: "success", Payload: mcpRuntime.List()})
|
||||||
|
case "mcp_call":
|
||||||
|
var p struct {
|
||||||
|
Server string `json:"server"`
|
||||||
|
Tool string `json:"tool"`
|
||||||
|
Args any `json:"args"`
|
||||||
|
}
|
||||||
|
if err := decodePayload(msg.Payload, &p); err != nil {
|
||||||
|
return sendError(writeJSON, msg.ID, err)
|
||||||
|
}
|
||||||
|
if p.Server == "" || p.Tool == "" {
|
||||||
|
return sendError(writeJSON, msg.ID, errors.New("server and tool required"))
|
||||||
|
}
|
||||||
|
ctx, cancel := context.WithTimeout(context.Background(), 300*time.Second)
|
||||||
|
defer cancel()
|
||||||
|
start := time.Now()
|
||||||
|
result, err := mcpRuntime.Call(ctx, p.Server, p.Tool, p.Args)
|
||||||
|
duration := time.Since(start).Milliseconds()
|
||||||
|
if err != nil {
|
||||||
|
_ = writeJSON(cluster.Message{ID: msg.ID, Status: "error", Error: err.Error(), Payload: map[string]any{"server": p.Server, "tool": p.Tool, "duration_ms": duration}})
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
return writeJSON(cluster.Message{ID: msg.ID, Status: "success", Payload: map[string]any{"server": p.Server, "tool": p.Tool, "result": result, "duration_ms": duration}})
|
||||||
case "exec":
|
case "exec":
|
||||||
var p cluster.ExecPayload
|
var p cluster.ExecPayload
|
||||||
if err := decodePayload(msg.Payload, &p); err != nil {
|
if err := decodePayload(msg.Payload, &p); err != nil {
|
||||||
|
|||||||
+1
-1
@@ -55,7 +55,7 @@ func main() {
|
|||||||
s := &api.Server{
|
s := &api.Server{
|
||||||
Auth: authManager,
|
Auth: authManager,
|
||||||
Exec: executor.Local{MaxOutput: 1024 * 1024},
|
Exec: executor.Local{MaxOutput: 1024 * 1024},
|
||||||
Cluster: func() *cluster.Manager { m := cluster.NewManager(clusterSecret); m.SetStore(store); return m }(),
|
Cluster: cluster.NewManager(clusterSecret, store),
|
||||||
Store: store,
|
Store: store,
|
||||||
LocalName: local,
|
LocalName: local,
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -9,6 +9,7 @@ import (
|
|||||||
"fmt"
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
"net/http"
|
"net/http"
|
||||||
|
"net/url"
|
||||||
"os"
|
"os"
|
||||||
"strings"
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
@@ -147,3 +148,31 @@ func main() {
|
|||||||
os.Stdout.WriteString("\n")
|
os.Stdout.WriteString("\n")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (h *HubRemoteBackend) ListAgentMCP(ctx context.Context, device string) (any, error) {
|
||||||
|
var res any
|
||||||
|
err := h.get(ctx, "/api/v1/mcp/list?device="+url.QueryEscape(device), &res)
|
||||||
|
return res, err
|
||||||
|
}
|
||||||
|
func (h *HubRemoteBackend) CallAgentMCP(ctx context.Context, clientIP, device, server, tool string, args any) (any, error) {
|
||||||
|
payload := map[string]any{"device": device, "server": server, "tool": tool, "arguments": args}
|
||||||
|
var res any
|
||||||
|
return res, h.post("/api/v1/mcp/call", payload, &res)
|
||||||
|
}
|
||||||
|
func (h *HubRemoteBackend) get(ctx context.Context, path string, out any) error {
|
||||||
|
req, err := http.NewRequestWithContext(ctx, http.MethodGet, strings.TrimRight(h.BaseURL, "/")+path, nil)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
req.Header.Set("Authorization", "Bearer "+h.APIKey)
|
||||||
|
resp, err := h.HTTPClient.Do(req)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer resp.Body.Close()
|
||||||
|
if resp.StatusCode >= 400 {
|
||||||
|
b, _ := io.ReadAll(resp.Body)
|
||||||
|
return fmt.Errorf("HTTP %d: %s", resp.StatusCode, string(b))
|
||||||
|
}
|
||||||
|
return json.NewDecoder(resp.Body).Decode(out)
|
||||||
|
}
|
||||||
|
|||||||
+17
-19
@@ -2,12 +2,13 @@
|
|||||||
set -e
|
set -e
|
||||||
|
|
||||||
# VPS-Commander 一键安装与配置脚本 (Hub / Agent)
|
# VPS-Commander 一键安装与配置脚本 (Hub / Agent)
|
||||||
# Repo: https://github.com/kyaring/vps-commander
|
# Repo: https://gitea.king.nyc.mn/openclaw/vps-commander
|
||||||
|
|
||||||
REPO="kyaring/vps-commander"
|
GITEA_HOST="gitea.king.nyc.mn"
|
||||||
|
REPO="openclaw/vps-commander"
|
||||||
INSTALL_DIR="/opt/vps-commander"
|
INSTALL_DIR="/opt/vps-commander"
|
||||||
CONF_DIR="/etc/vps-commander"
|
CONF_DIR="/etc/vps-commander"
|
||||||
GITHUB_URL="https://github.com"
|
GITEA_URL="https://gitea.king.nyc.mn"
|
||||||
|
|
||||||
# 颜色输出
|
# 颜色输出
|
||||||
RED='\033[0;31m'
|
RED='\033[0;31m'
|
||||||
@@ -37,12 +38,12 @@ command -v curl >/dev/null 2>&1 || (apt-get update && apt-get install -y curl ||
|
|||||||
|
|
||||||
# 获取最新版本 tag
|
# 获取最新版本 tag
|
||||||
info "正在获取 VPS-Commander 最新版本信息..."
|
info "正在获取 VPS-Commander 最新版本信息..."
|
||||||
LATEST_TAG=$(curl -fsSL -k "https://api.github.com/repos/${REPO}/releases/latest" | grep '"tag_name":' | sed -E 's/.*"([^"]+)".*/\1/')
|
LATEST_TAG=$(curl -fsSL -k "${GITEA_URL}/api/v1/repos/${REPO}/releases/latest" 2>/dev/null | grep -o '"tag_name":"[^"]*"' | head -n1 | cut -d'"' -f4)
|
||||||
if [ -z "$LATEST_TAG" ]; then
|
if [ -z "$LATEST_TAG" ]; then
|
||||||
warn "无法通过 GitHub API 自动获取 Tag,将尝试使用默认版本 v1.0.0"
|
warn "无法通过 Gitea API 自动获取 Tag,回退至 v1.1.5"
|
||||||
LATEST_TAG="v1.0.0"
|
LATEST_TAG="v1.1.5"
|
||||||
fi
|
fi
|
||||||
info "检测到最新版本: ${LATEST_TAG}"
|
info "检测到最新版本: ${LATEST_TAG} (${TARGET_ARCH})"
|
||||||
|
|
||||||
mkdir -p "$INSTALL_DIR" "$INSTALL_DIR/data" "$CONF_DIR"
|
mkdir -p "$INSTALL_DIR" "$INSTALL_DIR/data" "$CONF_DIR"
|
||||||
chmod 750 "$INSTALL_DIR" "$CONF_DIR"
|
chmod 750 "$INSTALL_DIR" "$CONF_DIR"
|
||||||
@@ -58,10 +59,10 @@ select_mode() {
|
|||||||
install_agent() {
|
install_agent() {
|
||||||
info "开始安装 VPS-Commander Agent (${TARGET_ARCH})..."
|
info "开始安装 VPS-Commander Agent (${TARGET_ARCH})..."
|
||||||
BIN_NAME="vps-commander-agent-linux-${TARGET_ARCH}"
|
BIN_NAME="vps-commander-agent-linux-${TARGET_ARCH}"
|
||||||
DOWNLOAD_URL="${GITHUB_URL}/${REPO}/releases/download/${LATEST_TAG}/${BIN_NAME}"
|
DOWNLOAD_URL="${GITEA_URL}/${REPO}/releases/download/${LATEST_TAG}/${BIN_NAME}"
|
||||||
|
|
||||||
info "正在下载 Agent: ${DOWNLOAD_URL}"
|
info "正在从 Gitea 下载 Agent: ${DOWNLOAD_URL}"
|
||||||
curl -fsSL -k -o "${INSTALL_DIR}/vps-commander-agent" "${DOWNLOAD_URL}" || error "下载失败,请检查网络或 Release 是否存在"
|
curl -fsSL -k -o "${INSTALL_DIR}/vps-commander-agent" "${DOWNLOAD_URL}" || error "下载失败,请检查 Gitea 网络或 Release 是否存在"
|
||||||
chmod +x "${INSTALL_DIR}/vps-commander-agent"
|
chmod +x "${INSTALL_DIR}/vps-commander-agent"
|
||||||
|
|
||||||
echo -e "\n--- 配置 Agent 参数 ---"
|
echo -e "\n--- 配置 Agent 参数 ---"
|
||||||
@@ -77,7 +78,6 @@ install_agent() {
|
|||||||
read -rp "Cluster Secret 不能为空,请重新输入: " CLUSTER_SECRET
|
read -rp "Cluster Secret 不能为空,请重新输入: " CLUSTER_SECRET
|
||||||
done
|
done
|
||||||
|
|
||||||
# 写入凭据文件
|
|
||||||
cat << ENV > "${CONF_DIR}/agent.env"
|
cat << ENV > "${CONF_DIR}/agent.env"
|
||||||
VPS_COMMANDER_HUB_WS_URL=${HUB_WS}
|
VPS_COMMANDER_HUB_WS_URL=${HUB_WS}
|
||||||
VPS_COMMANDER_AGENT_NAME=${AGENT_NAME}
|
VPS_COMMANDER_AGENT_NAME=${AGENT_NAME}
|
||||||
@@ -85,7 +85,6 @@ VPS_COMMANDER_CLUSTER_SECRET=${CLUSTER_SECRET}
|
|||||||
ENV
|
ENV
|
||||||
chmod 0600 "${CONF_DIR}/agent.env"
|
chmod 0600 "${CONF_DIR}/agent.env"
|
||||||
|
|
||||||
# 配置 systemd
|
|
||||||
cat << SVC > /etc/systemd/system/vps-commander-agent.service
|
cat << SVC > /etc/systemd/system/vps-commander-agent.service
|
||||||
[Unit]
|
[Unit]
|
||||||
Description=VPS-Commander Agent
|
Description=VPS-Commander Agent
|
||||||
@@ -125,20 +124,19 @@ SVC
|
|||||||
install_hub() {
|
install_hub() {
|
||||||
info "开始安装 VPS-Commander Hub (${TARGET_ARCH})..."
|
info "开始安装 VPS-Commander Hub (${TARGET_ARCH})..."
|
||||||
BIN_NAME="vps-commander-hub-linux-${TARGET_ARCH}"
|
BIN_NAME="vps-commander-hub-linux-${TARGET_ARCH}"
|
||||||
DOWNLOAD_URL="${GITHUB_URL}/${REPO}/releases/download/${LATEST_TAG}/${BIN_NAME}"
|
DOWNLOAD_URL="${GITEA_URL}/${REPO}/releases/download/${LATEST_TAG}/${BIN_NAME}"
|
||||||
|
|
||||||
info "正在下载 Hub: ${DOWNLOAD_URL}"
|
info "正在从 Gitea 下载 Hub: ${DOWNLOAD_URL}"
|
||||||
curl -fsSL -k -o "${INSTALL_DIR}/vps-commander-hub" "${DOWNLOAD_URL}" || error "下载失败,请检查网络或 Release 是否存在"
|
curl -fsSL -k -o "${INSTALL_DIR}/vps-commander-hub" "${DOWNLOAD_URL}" || error "下载失败,请检查 Gitea 网络或 Release 是否存在"
|
||||||
chmod +x "${INSTALL_DIR}/vps-commander-hub"
|
chmod +x "${INSTALL_DIR}/vps-commander-hub"
|
||||||
|
|
||||||
echo -e "\n--- 配置 Hub 参数 ---"
|
echo -e "\n--- 配置 Hub 参数 ---"
|
||||||
read -rp "请输入监听端口 (默认 9521): " HUB_PORT
|
read -rp "请输入监听端口 (默认 9521): " HUB_PORT
|
||||||
HUB_PORT=${HUB_PORT:-9521}
|
HUB_PORT=${HUB_PORT:-9521}
|
||||||
|
|
||||||
# 自动生成随机凭据
|
GEN_API_KEY=$(head -c 24 /dev/urandom | xxd -p 2>/dev/null || openssl rand -hex 24)
|
||||||
GEN_API_KEY=$(head -c 24 /dev/urandom | xxd -p)
|
GEN_SECRET=$(head -c 24 /dev/urandom | xxd -p 2>/dev/null || openssl rand -hex 24)
|
||||||
GEN_SECRET=$(head -c 24 /dev/urandom | xxd -p)
|
GEN_PASS=$(head -c 16 /dev/urandom | xxd -p 2>/dev/null || openssl rand -hex 16)
|
||||||
GEN_PASS=$(head -c 16 /dev/urandom | xxd -p)
|
|
||||||
|
|
||||||
read -rp "请输入 API Key (默认随机生成): " API_KEY
|
read -rp "请输入 API Key (默认随机生成): " API_KEY
|
||||||
API_KEY=${API_KEY:-$GEN_API_KEY}
|
API_KEY=${API_KEY:-$GEN_API_KEY}
|
||||||
|
|||||||
Executable
+81
@@ -0,0 +1,81 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
set -e
|
||||||
|
|
||||||
|
# VPS-Commander Agent 一键安全升级脚本
|
||||||
|
# Repo: https://gitea.king.nyc.mn/openclaw/vps-commander
|
||||||
|
|
||||||
|
GITEA_HOST="gitea.king.nyc.mn"
|
||||||
|
REPO="openclaw/vps-commander"
|
||||||
|
INSTALL_DIR="/opt/vps-commander"
|
||||||
|
GITEA_URL="https://gitea.king.nyc.mn"
|
||||||
|
|
||||||
|
# 颜色输出
|
||||||
|
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} $*"; }
|
||||||
|
success() { echo -e "${GREEN}[SUCCESS]${NC} $*"; }
|
||||||
|
warn() { echo -e "${YELLOW}[WARN]${NC} $*"; }
|
||||||
|
error() { echo -e "${RED}[ERROR]${NC} $*"; exit 1; }
|
||||||
|
|
||||||
|
# 检查 root 权限
|
||||||
|
[[ $EUID -ne 0 ]] && error "必须使用 root 权限运行此脚本"
|
||||||
|
|
||||||
|
# 检查是否安装过 agent
|
||||||
|
if [ ! -f "${INSTALL_DIR}/vps-commander-agent" ]; then
|
||||||
|
error "未检测到已安装的 Agent (${INSTALL_DIR}/vps-commander-agent),请先使用 install.sh 完整安装。"
|
||||||
|
fi
|
||||||
|
|
||||||
|
# 架构检测
|
||||||
|
ARCH=$(uname -m)
|
||||||
|
case "$ARCH" in
|
||||||
|
x86_64|amd64) TARGET_ARCH="amd64" ;;
|
||||||
|
aarch64|arm64) TARGET_ARCH="arm64" ;;
|
||||||
|
*) error "暂不支持的系统架构: $ARCH" ;;
|
||||||
|
esac
|
||||||
|
|
||||||
|
info "正在获取最新 Release 版本信息..."
|
||||||
|
LATEST_TAG=$(curl -fsSL -k "${GITEA_URL}/api/v1/repos/${REPO}/releases/latest" 2>/dev/null | grep -o '"tag_name":"[^"]*"' | head -n1 | cut -d'"' -f4)
|
||||||
|
if [ -z "$LATEST_TAG" ]; then
|
||||||
|
warn "无法通过 Gitea API 自动获取 Tag,回退至 v1.1.5"
|
||||||
|
LATEST_TAG="v1.1.5"
|
||||||
|
fi
|
||||||
|
info "目标升级版本: ${LATEST_TAG} (${TARGET_ARCH})"
|
||||||
|
|
||||||
|
BIN_NAME="vps-commander-agent-linux-${TARGET_ARCH}"
|
||||||
|
DOWNLOAD_URL="${GITEA_URL}/${REPO}/releases/download/${LATEST_TAG}/${BIN_NAME}"
|
||||||
|
TMP_BIN="${INSTALL_DIR}/vps-commander-agent.tmp"
|
||||||
|
BAK_BIN="${INSTALL_DIR}/vps-commander-agent.bak"
|
||||||
|
|
||||||
|
info "正在从 Gitea 下载新版本二进制文件..."
|
||||||
|
curl -fsSL -k -o "${TMP_BIN}" "${DOWNLOAD_URL}" || error "下载失败,请检查 Gitea 网络或 Release 是否存在"
|
||||||
|
|
||||||
|
# 完整性校验:文件大小不能小于 1MB
|
||||||
|
FILE_SIZE=$(stat -c%s "${TMP_BIN}" 2>/dev/null || stat -f%z "${TMP_BIN}" 2>/dev/null || echo 0)
|
||||||
|
if [ "$FILE_SIZE" -lt 1000000 ]; then
|
||||||
|
rm -f "${TMP_BIN}"
|
||||||
|
error "下载的文件不完整或损坏 (大小: ${FILE_SIZE} 字节),已终止升级"
|
||||||
|
fi
|
||||||
|
|
||||||
|
chmod +x "${TMP_BIN}"
|
||||||
|
|
||||||
|
info "备份旧版本并原子替换..."
|
||||||
|
cp -f "${INSTALL_DIR}/vps-commander-agent" "${BAK_BIN}" 2>/dev/null || true
|
||||||
|
mv -f "${TMP_BIN}" "${INSTALL_DIR}/vps-commander-agent"
|
||||||
|
|
||||||
|
info "重启 vps-commander-agent 服务..."
|
||||||
|
systemctl restart vps-commander-agent
|
||||||
|
sleep 2
|
||||||
|
|
||||||
|
if systemctl is-active --quiet vps-commander-agent; then
|
||||||
|
success "Agent 成功平滑升级至 ${LATEST_TAG} 并已恢复运行!"
|
||||||
|
rm -f "${BAK_BIN}"
|
||||||
|
else
|
||||||
|
warn "服务启动异常,正在自动回滚原版本..."
|
||||||
|
mv -f "${BAK_BIN}" "${INSTALL_DIR}/vps-commander-agent"
|
||||||
|
systemctl restart vps-commander-agent
|
||||||
|
error "升级失败,已安全回滚至上一版本,请查看日志: journalctl -u vps-commander-agent -n 20"
|
||||||
|
fi
|
||||||
+156
-1
@@ -76,6 +76,157 @@ func (s *Server) AuditsJSON(w http.ResponseWriter, r *http.Request) {
|
|||||||
jsonOut(w, logs)
|
jsonOut(w, logs)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type MCPServiceRequest struct {
|
||||||
|
ID string `json:"id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
Description string `json:"description"`
|
||||||
|
Transport string `json:"transport"`
|
||||||
|
Command string `json:"command"`
|
||||||
|
Args []string `json:"args"`
|
||||||
|
Env map[string]string `json:"env"`
|
||||||
|
URL string `json:"url"`
|
||||||
|
Headers map[string]string `json:"headers"`
|
||||||
|
Scope string `json:"scope"`
|
||||||
|
TargetNodes []string `json:"target_nodes"`
|
||||||
|
Enabled bool `json:"enabled"`
|
||||||
|
}
|
||||||
|
type MCPCallRequest struct {
|
||||||
|
Device string `json:"device"`
|
||||||
|
Server string `json:"server"`
|
||||||
|
Tool string `json:"tool"`
|
||||||
|
Arguments any `json:"arguments"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Server) MCPServicesJSON(w http.ResponseWriter, r *http.Request) {
|
||||||
|
if s.Store == nil {
|
||||||
|
http.Error(w, "storage unavailable", 500)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
switch r.Method {
|
||||||
|
case http.MethodGet:
|
||||||
|
v, err := s.Store.ListMCPServices()
|
||||||
|
if err != nil {
|
||||||
|
http.Error(w, err.Error(), 500)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
jsonOut(w, v)
|
||||||
|
case http.MethodPost:
|
||||||
|
var q MCPServiceRequest
|
||||||
|
if err := json.NewDecoder(http.MaxBytesReader(w, r.Body, 256*1024)).Decode(&q); err != nil {
|
||||||
|
http.Error(w, "bad json", 400)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if q.ID == "" || q.Name == "" || q.Transport == "stdio" && q.Command == "" || (q.Transport != "stdio" && q.URL == "") {
|
||||||
|
http.Error(w, "invalid MCP service", 400)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
args, _ := json.Marshal(q.Args)
|
||||||
|
env, _ := json.Marshal(q.Env)
|
||||||
|
headers, _ := json.Marshal(q.Headers)
|
||||||
|
nodes, _ := json.Marshal(q.TargetNodes)
|
||||||
|
if q.Scope == "" {
|
||||||
|
q.Scope = "all"
|
||||||
|
}
|
||||||
|
if q.Scope != "all" && q.Scope != "custom" {
|
||||||
|
http.Error(w, "invalid scope", 400)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
v := storage.MCPService{ID: q.ID, Name: q.Name, Description: q.Description, Transport: q.Transport, Command: q.Command, ArgsJSON: string(args), EnvJSON: string(env), URL: q.URL, HeadersJSON: string(headers), Scope: q.Scope, TargetNodesJSON: string(nodes), Enabled: q.Enabled}
|
||||||
|
if err := s.Store.SaveMCPService(v); err != nil {
|
||||||
|
http.Error(w, err.Error(), 500)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if s.Cluster != nil {
|
||||||
|
s.Cluster.SyncMCPServices()
|
||||||
|
}
|
||||||
|
_ = s.Store.Audit(clientIP(r), "", "mcp_service_save", q.ID, 0, 0, "")
|
||||||
|
jsonOut(w, v)
|
||||||
|
default:
|
||||||
|
http.Error(w, "method not allowed", 405)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
func (s *Server) DeleteMCPService(w http.ResponseWriter, r *http.Request) {
|
||||||
|
if r.Method != http.MethodDelete {
|
||||||
|
http.Error(w, "method not allowed", 405)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
id := r.URL.Query().Get("id")
|
||||||
|
if id == "" {
|
||||||
|
http.Error(w, "id required", 400)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if err := s.Store.DeleteMCPService(id); err != nil {
|
||||||
|
http.Error(w, err.Error(), 500)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if s.Cluster != nil {
|
||||||
|
s.Cluster.SyncMCPServices()
|
||||||
|
}
|
||||||
|
_ = s.Store.Audit(clientIP(r), "", "mcp_service_delete", id, 0, 0, "")
|
||||||
|
jsonOut(w, map[string]any{"ok": true})
|
||||||
|
}
|
||||||
|
func (s *Server) ListAgentMCPJSON(w http.ResponseWriter, r *http.Request) {
|
||||||
|
if r.Method != http.MethodGet {
|
||||||
|
http.Error(w, "method not allowed", 405)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
device := r.URL.Query().Get("device")
|
||||||
|
if device == "" {
|
||||||
|
http.Error(w, "device required", 400)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if s.isRemote(device) && (s.Cluster == nil || !s.Cluster.Online(device)) {
|
||||||
|
http.Error(w, "device offline", 409)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if s.isRemote(device) {
|
||||||
|
v, err := s.Cluster.ListMCP(device)
|
||||||
|
if err != nil {
|
||||||
|
http.Error(w, err.Error(), 404)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
jsonOut(w, map[string]any{"device": device, "services": v})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
jsonOut(w, map[string]any{"device": s.LocalName, "services": []any{}})
|
||||||
|
}
|
||||||
|
func (s *Server) CallAgentMCPJSON(w http.ResponseWriter, r *http.Request) {
|
||||||
|
if r.Method != http.MethodPost {
|
||||||
|
http.Error(w, "method not allowed", 405)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
var q MCPCallRequest
|
||||||
|
if err := json.NewDecoder(http.MaxBytesReader(w, r.Body, 256*1024)).Decode(&q); err != nil {
|
||||||
|
http.Error(w, "bad json", 400)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if q.Device == "" || q.Server == "" || q.Tool == "" {
|
||||||
|
http.Error(w, "device, server and tool required", 400)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if s.isRemote(q.Device) && (s.Cluster == nil || !s.Cluster.Online(q.Device)) {
|
||||||
|
if s.Store != nil {
|
||||||
|
_ = s.Store.Audit(clientIP(r), q.Device, "mcp_call", q.Server+"/"+q.Tool, 1, 0, "device offline")
|
||||||
|
}
|
||||||
|
http.Error(w, "device offline", 409)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if !s.isRemote(q.Device) {
|
||||||
|
http.Error(w, "agent MCP requires a remote device", 400)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
start := time.Now()
|
||||||
|
v, err := s.Cluster.CallMCP(r.Context(), q.Device, q.Server, q.Tool, q.Arguments)
|
||||||
|
ms := time.Since(start).Milliseconds()
|
||||||
|
if err != nil {
|
||||||
|
_ = s.Store.Audit(clientIP(r), q.Device, "mcp_call", q.Server+"/"+q.Tool, 1, ms, err.Error())
|
||||||
|
http.Error(w, err.Error(), 502)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
_ = s.Store.Audit(clientIP(r), q.Device, "mcp_call", q.Server+"/"+q.Tool, 0, ms, "")
|
||||||
|
jsonOut(w, v)
|
||||||
|
}
|
||||||
|
|
||||||
func (s *Server) Routes() http.Handler {
|
func (s *Server) Routes() http.Handler {
|
||||||
mux := http.NewServeMux()
|
mux := http.NewServeMux()
|
||||||
mux.HandleFunc("/healthz", s.health)
|
mux.HandleFunc("/healthz", s.health)
|
||||||
@@ -83,6 +234,10 @@ func (s *Server) Routes() http.Handler {
|
|||||||
mux.HandleFunc("/api/v1/exec", s.exec)
|
mux.HandleFunc("/api/v1/exec", s.exec)
|
||||||
mux.HandleFunc("/api/v1/file/read", s.fileRead)
|
mux.HandleFunc("/api/v1/file/read", s.fileRead)
|
||||||
mux.HandleFunc("/api/v1/file/write", s.fileWrite)
|
mux.HandleFunc("/api/v1/file/write", s.fileWrite)
|
||||||
|
mux.HandleFunc("/api/v1/mcp/services", s.MCPServicesJSON)
|
||||||
|
mux.HandleFunc("/api/v1/mcp/service", s.DeleteMCPService)
|
||||||
|
mux.HandleFunc("/api/v1/mcp/list", s.ListAgentMCPJSON)
|
||||||
|
mux.HandleFunc("/api/v1/mcp/call", s.CallAgentMCPJSON)
|
||||||
mux.HandleFunc("/agent/ws", s.agentWS)
|
mux.HandleFunc("/agent/ws", s.agentWS)
|
||||||
mux.HandleFunc("/openapi.json", s.openapi)
|
mux.HandleFunc("/openapi.json", s.openapi)
|
||||||
return mux
|
return mux
|
||||||
@@ -179,7 +334,7 @@ func (s *Server) exec(w http.ResponseWriter, r *http.Request) {
|
|||||||
var res executor.Result
|
var res executor.Result
|
||||||
if s.isRemote(req.Device) {
|
if s.isRemote(req.Device) {
|
||||||
if s.Cluster == nil || !s.Cluster.Online(req.Device) {
|
if s.Cluster == nil || !s.Cluster.Online(req.Device) {
|
||||||
http.Error(w, "device offline", 404)
|
http.Error(w, "device offline", 409)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
rr, err := s.Cluster.Exec(ctx, req.Device, cluster.ExecPayload{Command: req.Command, Workdir: req.Workdir, Timeout: req.Timeout})
|
rr, err := s.Cluster.Exec(ctx, req.Device, cluster.ExecPayload{Command: req.Command, Workdir: req.Workdir, Timeout: req.Timeout})
|
||||||
|
|||||||
@@ -188,3 +188,38 @@ func (b *MCPBackend) WriteFile(ctx context.Context, clientIP, device, path, cont
|
|||||||
Bytes: n,
|
Bytes: n,
|
||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (b *MCPBackend) ListAgentMCP(ctx context.Context, device string) (any, error) {
|
||||||
|
if device == "" {
|
||||||
|
return nil, errors.New("device required")
|
||||||
|
}
|
||||||
|
if b.server.isRemote(device) && (b.server.Cluster == nil || !b.server.Cluster.Online(device)) {
|
||||||
|
return nil, fmt.Errorf("device %s is offline", device)
|
||||||
|
}
|
||||||
|
if !b.server.isRemote(device) {
|
||||||
|
return map[string]any{"device": b.server.LocalName, "services": []any{}}, nil
|
||||||
|
}
|
||||||
|
return b.server.Cluster.ListMCP(device)
|
||||||
|
}
|
||||||
|
func (b *MCPBackend) CallAgentMCP(ctx context.Context, clientIP, device, server, tool string, args any) (any, error) {
|
||||||
|
if device == "" || server == "" || tool == "" {
|
||||||
|
return nil, errors.New("device, server and tool required")
|
||||||
|
}
|
||||||
|
if b.server.isRemote(device) && (b.server.Cluster == nil || !b.server.Cluster.Online(device)) {
|
||||||
|
return nil, fmt.Errorf("device %s is offline", device)
|
||||||
|
}
|
||||||
|
if !b.server.isRemote(device) {
|
||||||
|
return nil, errors.New("agent MCP requires a remote device")
|
||||||
|
}
|
||||||
|
start := time.Now()
|
||||||
|
v, err := b.server.Cluster.CallMCP(ctx, device, server, tool, args)
|
||||||
|
ms := time.Since(start).Milliseconds()
|
||||||
|
if b.server.Store != nil {
|
||||||
|
if err != nil {
|
||||||
|
_ = b.server.Store.Audit(clientIP, device, "mcp_call", server+"/"+tool, 1, ms, err.Error())
|
||||||
|
} else {
|
||||||
|
_ = b.server.Store.Audit(clientIP, device, "mcp_call", server+"/"+tool, 0, ms, "")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return v, err
|
||||||
|
}
|
||||||
|
|||||||
@@ -157,6 +157,69 @@
|
|||||||
"type": "integer"
|
"type": "integer"
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
},
|
||||||
|
"MCPService": {
|
||||||
|
"type": "object",
|
||||||
|
"properties": {
|
||||||
|
"id": {
|
||||||
|
"type": "string"
|
||||||
|
},
|
||||||
|
"name": {
|
||||||
|
"type": "string"
|
||||||
|
},
|
||||||
|
"description": {
|
||||||
|
"type": "string"
|
||||||
|
},
|
||||||
|
"transport": {
|
||||||
|
"type": "string"
|
||||||
|
},
|
||||||
|
"command": {
|
||||||
|
"type": "string"
|
||||||
|
},
|
||||||
|
"args_json": {
|
||||||
|
"type": "string"
|
||||||
|
},
|
||||||
|
"env_json": {
|
||||||
|
"type": "string"
|
||||||
|
},
|
||||||
|
"url": {
|
||||||
|
"type": "string"
|
||||||
|
},
|
||||||
|
"headers_json": {
|
||||||
|
"type": "string"
|
||||||
|
},
|
||||||
|
"scope": {
|
||||||
|
"type": "string"
|
||||||
|
},
|
||||||
|
"target_nodes_json": {
|
||||||
|
"type": "string"
|
||||||
|
},
|
||||||
|
"enabled": {
|
||||||
|
"type": "boolean"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"MCPCallRequest": {
|
||||||
|
"type": "object",
|
||||||
|
"required": [
|
||||||
|
"device",
|
||||||
|
"server",
|
||||||
|
"tool"
|
||||||
|
],
|
||||||
|
"properties": {
|
||||||
|
"device": {
|
||||||
|
"type": "string"
|
||||||
|
},
|
||||||
|
"server": {
|
||||||
|
"type": "string"
|
||||||
|
},
|
||||||
|
"tool": {
|
||||||
|
"type": "string"
|
||||||
|
},
|
||||||
|
"arguments": {
|
||||||
|
"type": "object"
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
@@ -281,6 +344,135 @@
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
},
|
||||||
|
"/api/v1/mcp/services": {
|
||||||
|
"get": {
|
||||||
|
"operationId": "list_mcp_services",
|
||||||
|
"security": [
|
||||||
|
{
|
||||||
|
"bearerAuth": []
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"responses": {
|
||||||
|
"200": {
|
||||||
|
"description": "MCP services",
|
||||||
|
"content": {
|
||||||
|
"application/json": {
|
||||||
|
"schema": {
|
||||||
|
"type": "array",
|
||||||
|
"items": {
|
||||||
|
"$ref": "#/components/schemas/MCPService"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"post": {
|
||||||
|
"operationId": "save_mcp_service",
|
||||||
|
"security": [
|
||||||
|
{
|
||||||
|
"bearerAuth": []
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"requestBody": {
|
||||||
|
"required": true,
|
||||||
|
"content": {
|
||||||
|
"application/json": {
|
||||||
|
"schema": {
|
||||||
|
"$ref": "#/components/schemas/MCPService"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"responses": {
|
||||||
|
"200": {
|
||||||
|
"description": "Saved MCP service"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"/api/v1/mcp/service": {
|
||||||
|
"delete": {
|
||||||
|
"operationId": "delete_mcp_service",
|
||||||
|
"security": [
|
||||||
|
{
|
||||||
|
"bearerAuth": []
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"parameters": [
|
||||||
|
{
|
||||||
|
"name": "id",
|
||||||
|
"in": "query",
|
||||||
|
"required": true,
|
||||||
|
"schema": {
|
||||||
|
"type": "string"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"responses": {
|
||||||
|
"200": {
|
||||||
|
"description": "Deleted"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"/api/v1/mcp/list": {
|
||||||
|
"get": {
|
||||||
|
"operationId": "list_agent_mcp",
|
||||||
|
"security": [
|
||||||
|
{
|
||||||
|
"bearerAuth": []
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"parameters": [
|
||||||
|
{
|
||||||
|
"name": "device",
|
||||||
|
"in": "query",
|
||||||
|
"required": true,
|
||||||
|
"schema": {
|
||||||
|
"type": "string"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"responses": {
|
||||||
|
"200": {
|
||||||
|
"description": "Mounted MCP services"
|
||||||
|
},
|
||||||
|
"409": {
|
||||||
|
"description": "Device offline"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"/api/v1/mcp/call": {
|
||||||
|
"post": {
|
||||||
|
"operationId": "call_agent_mcp",
|
||||||
|
"security": [
|
||||||
|
{
|
||||||
|
"bearerAuth": []
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"requestBody": {
|
||||||
|
"required": true,
|
||||||
|
"content": {
|
||||||
|
"application/json": {
|
||||||
|
"schema": {
|
||||||
|
"$ref": "#/components/schemas/MCPCallRequest"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"responses": {
|
||||||
|
"200": {
|
||||||
|
"description": "MCP result"
|
||||||
|
},
|
||||||
|
"409": {
|
||||||
|
"description": "Device offline"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
+175
-31
@@ -16,6 +16,8 @@ import (
|
|||||||
"github.com/wjyhk/vps-commander/internal/sysinfo"
|
"github.com/wjyhk/vps-commander/internal/sysinfo"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
const deviceOfflineAfter = 70 * time.Second
|
||||||
|
|
||||||
type Manager struct {
|
type Manager struct {
|
||||||
Secret string
|
Secret string
|
||||||
Store *storage.Store
|
Store *storage.Store
|
||||||
@@ -26,10 +28,17 @@ type Manager struct {
|
|||||||
type Node struct {
|
type Node struct {
|
||||||
Name string
|
Name string
|
||||||
Conn *websocket.Conn
|
Conn *websocket.Conn
|
||||||
|
ConnectionID string
|
||||||
WriteMu sync.Mutex
|
WriteMu sync.Mutex
|
||||||
Pending sync.Map
|
Pending sync.Map
|
||||||
LastSeen atomic.Int64
|
LastSeen atomic.Int64
|
||||||
|
LastHeartbeat atomic.Int64
|
||||||
|
Connected atomic.Bool
|
||||||
LastProfile *sysinfo.Profile
|
LastProfile *sysinfo.Profile
|
||||||
|
Arch string
|
||||||
|
OS string
|
||||||
|
IsLocal bool
|
||||||
|
MCP *MCPRuntime
|
||||||
}
|
}
|
||||||
|
|
||||||
type Result struct {
|
type Result struct {
|
||||||
@@ -47,8 +56,30 @@ type FileResult struct {
|
|||||||
EOF bool `json:"eof"`
|
EOF bool `json:"eof"`
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewManager(secret string) *Manager {
|
func NewManager(secret string, store *storage.Store) *Manager {
|
||||||
return &Manager{Secret: secret, nodes: make(map[string]*Node)}
|
m := &Manager{Secret: secret, Store: store, nodes: make(map[string]*Node)}
|
||||||
|
if store != nil {
|
||||||
|
if records, err := store.ListDevices(); err == nil {
|
||||||
|
for _, d := range records {
|
||||||
|
if d.IsLocal {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
n := &Node{Name: d.Name, Arch: d.Arch, OS: d.OS, MCP: NewMCPRuntime()}
|
||||||
|
n.LastSeen.Store(d.LastSeen * int64(time.Second))
|
||||||
|
n.LastHeartbeat.Store(d.LastHeartbeat * int64(time.Second))
|
||||||
|
if d.ProfileJSON != "" {
|
||||||
|
var p sysinfo.Profile
|
||||||
|
if json.Unmarshal([]byte(d.ProfileJSON), &p) == nil {
|
||||||
|
n.LastProfile = &p
|
||||||
|
}
|
||||||
|
}
|
||||||
|
n.Connected.Store(false)
|
||||||
|
m.nodes[d.Name] = n
|
||||||
|
_ = store.UpdateDeviceStatus(d.Name, "offline", d.LastHeartbeat)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return m
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m *Manager) SetStore(s *storage.Store) {
|
func (m *Manager) SetStore(s *storage.Store) {
|
||||||
@@ -83,15 +114,19 @@ func (m *Manager) ServeWS(w http.ResponseWriter, r *http.Request) {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
n := &Node{Name: name, Conn: c}
|
n := &Node{Name: name, Conn: c, ConnectionID: fmt.Sprintf("%s-%d", name, time.Now().UnixNano()), MCP: NewMCPRuntime()}
|
||||||
n.LastSeen.Store(time.Now().UnixNano())
|
n.LastSeen.Store(time.Now().UnixNano())
|
||||||
|
n.LastHeartbeat.Store(time.Now().UnixNano())
|
||||||
|
n.Connected.Store(true)
|
||||||
m.mu.Lock()
|
m.mu.Lock()
|
||||||
if old := m.nodes[name]; old != nil {
|
old := m.nodes[name]
|
||||||
_ = old.Conn.Close()
|
|
||||||
}
|
|
||||||
m.nodes[name] = n
|
m.nodes[name] = n
|
||||||
m.mu.Unlock()
|
m.mu.Unlock()
|
||||||
|
if old != nil && old.Conn != nil {
|
||||||
|
_ = old.Conn.Close()
|
||||||
|
}
|
||||||
_ = n.write(Message{Event: "registered", Device: name, TS: time.Now().Unix()})
|
_ = n.write(Message{Event: "registered", Device: name, TS: time.Now().Unix()})
|
||||||
|
m.syncMCP(n)
|
||||||
go m.heartbeat(n)
|
go m.heartbeat(n)
|
||||||
go m.readLoop(n)
|
go m.readLoop(n)
|
||||||
}
|
}
|
||||||
@@ -107,33 +142,59 @@ func (n *Node) write(msg Message) error {
|
|||||||
|
|
||||||
func (m *Manager) readLoop(n *Node) {
|
func (m *Manager) readLoop(n *Node) {
|
||||||
defer func() {
|
defer func() {
|
||||||
m.mu.Lock()
|
n.Connected.Store(false)
|
||||||
if m.nodes[n.Name] == n {
|
conn := n.Conn
|
||||||
delete(m.nodes, n.Name)
|
m.mu.RLock()
|
||||||
|
current := m.nodes[n.Name] == n
|
||||||
|
m.mu.RUnlock()
|
||||||
|
if conn != nil {
|
||||||
|
_ = conn.Close()
|
||||||
}
|
}
|
||||||
m.mu.Unlock()
|
|
||||||
_ = n.Conn.Close()
|
|
||||||
n.Pending.Range(func(key, v any) bool {
|
n.Pending.Range(func(key, v any) bool {
|
||||||
if ch, ok := v.(chan Message); ok {
|
if ch, ok := v.(chan Message); ok {
|
||||||
select {
|
select {
|
||||||
case ch <- Message{Status: "error", Error: "device disconnected"}:
|
case ch <- Message{Status: "error", Error: "device disconnected"}:
|
||||||
default:
|
default:
|
||||||
}
|
}
|
||||||
n.Pending.Delete(key)
|
|
||||||
}
|
}
|
||||||
|
n.Pending.Delete(key)
|
||||||
return true
|
return true
|
||||||
})
|
})
|
||||||
|
if current && m.Store != nil {
|
||||||
|
_ = m.Store.UpdateDeviceStatus(n.Name, "offline", n.LastHeartbeat.Load()/int64(time.Second))
|
||||||
|
}
|
||||||
}()
|
}()
|
||||||
_ = n.Conn.SetReadDeadline(time.Now().Add(75 * time.Second))
|
conn := n.Conn
|
||||||
for {
|
if conn == nil {
|
||||||
var msg Message
|
|
||||||
if err := n.Conn.ReadJSON(&msg); err != nil {
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
n.LastSeen.Store(time.Now().UnixNano())
|
_ = conn.SetReadDeadline(time.Now().Add(75 * time.Second))
|
||||||
_ = n.Conn.SetReadDeadline(time.Now().Add(75 * time.Second))
|
for {
|
||||||
|
var msg Message
|
||||||
|
if err := conn.ReadJSON(&msg); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
now := time.Now().UnixNano()
|
||||||
|
n.LastSeen.Store(now)
|
||||||
|
_ = conn.SetReadDeadline(time.Now().Add(75 * time.Second))
|
||||||
|
if msg.Arch != "" {
|
||||||
|
n.Arch = msg.Arch
|
||||||
|
}
|
||||||
|
if msg.OS != "" {
|
||||||
|
n.OS = msg.OS
|
||||||
|
}
|
||||||
if msg.Profile != nil {
|
if msg.Profile != nil {
|
||||||
n.LastProfile = msg.Profile
|
n.LastProfile = msg.Profile
|
||||||
|
n.LastHeartbeat.Store(now)
|
||||||
|
if m.Store != nil {
|
||||||
|
b, _ := json.Marshal(msg.Profile)
|
||||||
|
_ = m.Store.UpsertRemote(n.Name, "online", n.Arch, n.OS, now/int64(time.Second), string(b))
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
n.LastHeartbeat.Store(now)
|
||||||
|
if m.Store != nil {
|
||||||
|
_ = m.Store.UpdateDeviceStatus(n.Name, "online", now/int64(time.Second))
|
||||||
|
}
|
||||||
}
|
}
|
||||||
if msg.Event == "pong" || msg.Event == "ping" || msg.Event == "hello" || msg.Event == "profile" {
|
if msg.Event == "pong" || msg.Event == "ping" || msg.Event == "hello" || msg.Event == "profile" {
|
||||||
continue
|
continue
|
||||||
@@ -153,21 +214,26 @@ func (m *Manager) readLoop(n *Node) {
|
|||||||
|
|
||||||
func (m *Manager) Online(name string) bool {
|
func (m *Manager) Online(name string) bool {
|
||||||
m.mu.RLock()
|
m.mu.RLock()
|
||||||
defer m.mu.RUnlock()
|
|
||||||
n := m.nodes[name]
|
n := m.nodes[name]
|
||||||
return n != nil && time.Since(time.Unix(0, n.LastSeen.Load())) < 70*time.Second
|
m.mu.RUnlock()
|
||||||
|
if n == nil || !n.Connected.Load() {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
return time.Since(time.Unix(0, n.LastSeen.Load())) < deviceOfflineAfter
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m *Manager) RevokeDevice(name, reason string) error {
|
func (m *Manager) RevokeDevice(name, reason string) error {
|
||||||
m.mu.Lock()
|
m.mu.Lock()
|
||||||
n, exists := m.nodes[name]
|
n := m.nodes[name]
|
||||||
if exists {
|
|
||||||
delete(m.nodes, name)
|
delete(m.nodes, name)
|
||||||
|
m.mu.Unlock()
|
||||||
|
if n != nil {
|
||||||
|
n.Connected.Store(false)
|
||||||
|
if n.Conn != nil {
|
||||||
_ = n.write(Message{Event: "revoked", Error: "revoked by admin"})
|
_ = n.write(Message{Event: "revoked", Error: "revoked by admin"})
|
||||||
_ = n.Conn.Close()
|
_ = n.Conn.Close()
|
||||||
}
|
}
|
||||||
m.mu.Unlock()
|
}
|
||||||
|
|
||||||
if m.Store != nil {
|
if m.Store != nil {
|
||||||
return m.Store.RevokeDevice(name, reason)
|
return m.Store.RevokeDevice(name, reason)
|
||||||
}
|
}
|
||||||
@@ -179,16 +245,15 @@ func (m *Manager) Devices() []map[string]any {
|
|||||||
defer m.mu.RUnlock()
|
defer m.mu.RUnlock()
|
||||||
out := make([]map[string]any, 0, len(m.nodes))
|
out := make([]map[string]any, 0, len(m.nodes))
|
||||||
for name, n := range m.nodes {
|
for name, n := range m.nodes {
|
||||||
status := "online"
|
status := "offline"
|
||||||
lastSeen := time.Unix(0, n.LastSeen.Load())
|
lastSeen := time.Unix(0, n.LastSeen.Load())
|
||||||
if time.Since(lastSeen) >= 70*time.Second {
|
if n.Connected.Load() && time.Since(lastSeen) < deviceOfflineAfter {
|
||||||
status = "offline"
|
status = "online"
|
||||||
}
|
}
|
||||||
item := map[string]any{
|
item := map[string]any{
|
||||||
"name": name,
|
"name": name, "status": status, "is_local": n.IsLocal,
|
||||||
"status": status,
|
"arch": n.Arch, "os": n.OS,
|
||||||
"is_local": false,
|
"last_heartbeat": n.LastHeartbeat.Load() / int64(time.Second),
|
||||||
"last_heartbeat": lastSeen.Unix(),
|
|
||||||
}
|
}
|
||||||
if n.LastProfile != nil {
|
if n.LastProfile != nil {
|
||||||
item["profile"] = n.LastProfile
|
item["profile"] = n.LastProfile
|
||||||
@@ -263,6 +328,85 @@ func (m *Manager) WriteFile(ctx context.Context, name string, p FileWritePayload
|
|||||||
return v.Bytes, nil
|
return v.Bytes, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (m *Manager) syncMCP(n *Node) {
|
||||||
|
if n == nil || n.MCP == nil || m.Store == nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
stored, err := m.Store.ListMCPServicesForNode(n.Name)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
services := make([]MCPService, 0, len(stored))
|
||||||
|
for _, v := range stored {
|
||||||
|
services = append(services, decodeStoredMCPService(v))
|
||||||
|
}
|
||||||
|
_ = n.write(Message{Event: "mcp_sync", Payload: services, TS: time.Now().Unix()})
|
||||||
|
}
|
||||||
|
|
||||||
|
func decodeStoredMCPService(v storage.MCPService) MCPService {
|
||||||
|
out := MCPService{ID: v.ID, Name: v.Name, Description: v.Description, Transport: v.Transport, Command: v.Command, URL: v.URL}
|
||||||
|
_ = json.Unmarshal([]byte(v.ArgsJSON), &out.Args)
|
||||||
|
_ = json.Unmarshal([]byte(v.EnvJSON), &out.Env)
|
||||||
|
_ = json.Unmarshal([]byte(v.HeadersJSON), &out.Headers)
|
||||||
|
return out
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *Manager) SyncMCPServices() {
|
||||||
|
m.mu.RLock()
|
||||||
|
nodes := make([]*Node, 0, len(m.nodes))
|
||||||
|
for _, n := range m.nodes {
|
||||||
|
nodes = append(nodes, n)
|
||||||
|
}
|
||||||
|
m.mu.RUnlock()
|
||||||
|
for _, n := range nodes {
|
||||||
|
if n.Connected.Load() {
|
||||||
|
m.syncMCP(n)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *Manager) ListMCP(name string) ([]MCPService, error) {
|
||||||
|
m.mu.RLock()
|
||||||
|
n := m.nodes[name]
|
||||||
|
m.mu.RUnlock()
|
||||||
|
if n == nil || !m.Online(name) {
|
||||||
|
return nil, fmt.Errorf("device %s is offline", name)
|
||||||
|
}
|
||||||
|
msg, err := m.call(context.Background(), name, "mcp_list", nil)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
var out []MCPService
|
||||||
|
b, _ := json.Marshal(msg.Payload)
|
||||||
|
if err := json.Unmarshal(b, &out); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if n.MCP == nil {
|
||||||
|
n.MCP = NewMCPRuntime()
|
||||||
|
}
|
||||||
|
n.MCP.Sync(out)
|
||||||
|
return out, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *Manager) CallMCP(ctx context.Context, name, server, tool string, args any) (MCPCallResult, error) {
|
||||||
|
m.mu.RLock()
|
||||||
|
n := m.nodes[name]
|
||||||
|
m.mu.RUnlock()
|
||||||
|
if n == nil || !m.Online(name) {
|
||||||
|
return MCPCallResult{}, fmt.Errorf("device %s is offline", name)
|
||||||
|
}
|
||||||
|
msg, err := m.call(ctx, name, "mcp_call", map[string]any{"server": server, "tool": tool, "args": args})
|
||||||
|
if err != nil {
|
||||||
|
return MCPCallResult{}, err
|
||||||
|
}
|
||||||
|
var out MCPCallResult
|
||||||
|
b, _ := json.Marshal(msg.Payload)
|
||||||
|
if err := json.Unmarshal(b, &out); err != nil {
|
||||||
|
return MCPCallResult{}, err
|
||||||
|
}
|
||||||
|
return out, nil
|
||||||
|
}
|
||||||
|
|
||||||
func (m *Manager) heartbeat(n *Node) {
|
func (m *Manager) heartbeat(n *Node) {
|
||||||
t := time.NewTicker(30 * time.Second)
|
t := time.NewTicker(30 * time.Second)
|
||||||
defer t.Stop()
|
defer t.Stop()
|
||||||
|
|||||||
@@ -0,0 +1,329 @@
|
|||||||
|
package cluster
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bufio"
|
||||||
|
"bytes"
|
||||||
|
"context"
|
||||||
|
"encoding/json"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"net/http"
|
||||||
|
"os"
|
||||||
|
"os/exec"
|
||||||
|
"strings"
|
||||||
|
"sync"
|
||||||
|
)
|
||||||
|
|
||||||
|
type MCPService struct {
|
||||||
|
ID string `json:"id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
Description string `json:"description,omitempty"`
|
||||||
|
Transport string `json:"transport"`
|
||||||
|
Command string `json:"command,omitempty"`
|
||||||
|
Args []string `json:"args,omitempty"`
|
||||||
|
Env map[string]string `json:"env,omitempty"`
|
||||||
|
URL string `json:"url,omitempty"`
|
||||||
|
Headers map[string]string `json:"headers,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type MCPCallResult struct {
|
||||||
|
Server string `json:"server"`
|
||||||
|
Tool string `json:"tool"`
|
||||||
|
Result any `json:"result,omitempty"`
|
||||||
|
Error string `json:"error,omitempty"`
|
||||||
|
DurationMS int64 `json:"duration_ms"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type mcpRPC struct {
|
||||||
|
JSONRPC string `json:"jsonrpc"`
|
||||||
|
ID any `json:"id,omitempty"`
|
||||||
|
Method string `json:"method"`
|
||||||
|
Params any `json:"params,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type mcpRPCResponse struct {
|
||||||
|
JSONRPC string `json:"jsonrpc"`
|
||||||
|
ID any `json:"id,omitempty"`
|
||||||
|
Result json.RawMessage `json:"result,omitempty"`
|
||||||
|
Error *struct {
|
||||||
|
Code int `json:"code"`
|
||||||
|
Message string `json:"message"`
|
||||||
|
Data any `json:"data,omitempty"`
|
||||||
|
} `json:"error,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type MCPRuntime struct {
|
||||||
|
mu sync.RWMutex
|
||||||
|
services map[string]MCPService
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewMCPRuntime() *MCPRuntime { return &MCPRuntime{services: make(map[string]MCPService)} }
|
||||||
|
|
||||||
|
func (r *MCPRuntime) Sync(services []MCPService) {
|
||||||
|
next := make(map[string]MCPService, len(services))
|
||||||
|
for _, s := range services {
|
||||||
|
if s.ID == "" || s.Transport == "" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
next[s.ID] = s
|
||||||
|
}
|
||||||
|
r.mu.Lock()
|
||||||
|
r.services = next
|
||||||
|
r.mu.Unlock()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *MCPRuntime) List() []MCPService {
|
||||||
|
r.mu.RLock()
|
||||||
|
defer r.mu.RUnlock()
|
||||||
|
out := make([]MCPService, 0, len(r.services))
|
||||||
|
for _, s := range r.services {
|
||||||
|
out = append(out, s)
|
||||||
|
}
|
||||||
|
return out
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *MCPRuntime) Get(id string) (MCPService, bool) {
|
||||||
|
r.mu.RLock()
|
||||||
|
defer r.mu.RUnlock()
|
||||||
|
s, ok := r.services[id]
|
||||||
|
return s, ok
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *MCPRuntime) Call(ctx context.Context, server, tool string, args any) (any, error) {
|
||||||
|
s, ok := r.Get(server)
|
||||||
|
if !ok {
|
||||||
|
return nil, fmt.Errorf("MCP service %q is not mounted", server)
|
||||||
|
}
|
||||||
|
switch strings.ToLower(s.Transport) {
|
||||||
|
case "stdio":
|
||||||
|
return callStdio(ctx, s, tool, args)
|
||||||
|
case "http", "sse":
|
||||||
|
return callHTTP(ctx, s, tool, args)
|
||||||
|
default:
|
||||||
|
return nil, fmt.Errorf("unsupported MCP transport %q", s.Transport)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func callStdio(ctx context.Context, s MCPService, tool string, args any) (any, error) {
|
||||||
|
if s.Command == "" {
|
||||||
|
return nil, errors.New("MCP stdio command is empty")
|
||||||
|
}
|
||||||
|
cmd := exec.CommandContext(ctx, s.Command, s.Args...)
|
||||||
|
if len(s.Env) > 0 {
|
||||||
|
cmd.Env = append(os.Environ(), envList(s.Env)...)
|
||||||
|
}
|
||||||
|
stdin, err := cmd.StdinPipe()
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
stdout, err := cmd.StdoutPipe()
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
cmd.Stderr = io.Discard
|
||||||
|
if err := cmd.Start(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer func() { _ = cmd.Process.Kill(); _ = cmd.Wait() }()
|
||||||
|
|
||||||
|
enc := json.NewEncoder(stdin)
|
||||||
|
dec := bufio.NewReaderSize(stdout, 64*1024)
|
||||||
|
id := int64(1)
|
||||||
|
if err := enc.Encode(mcpRPC{JSONRPC: "2.0", ID: id, Method: "initialize", Params: map[string]any{
|
||||||
|
"protocolVersion": "2024-11-05", "capabilities": map[string]any{},
|
||||||
|
"clientInfo": map[string]string{"name": "vps-commander-agent", "version": "1.0"},
|
||||||
|
}}); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if _, err := readRPC(ctx, dec, id); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if err := enc.Encode(mcpRPC{JSONRPC: "2.0", Method: "notifications/initialized", Params: map[string]any{}}); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
id++
|
||||||
|
if err := enc.Encode(mcpRPC{JSONRPC: "2.0", ID: id, Method: "tools/call", Params: map[string]any{"name": tool, "arguments": args}}); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return readRPC(ctx, dec, id)
|
||||||
|
}
|
||||||
|
|
||||||
|
func readRPC(ctx context.Context, r *bufio.Reader, wantID int64) (any, error) {
|
||||||
|
for {
|
||||||
|
lineCh := make(chan []byte, 1)
|
||||||
|
errCh := make(chan error, 1)
|
||||||
|
go func() {
|
||||||
|
b, err := r.ReadBytes('\n')
|
||||||
|
if err != nil && len(b) == 0 {
|
||||||
|
errCh <- err
|
||||||
|
return
|
||||||
|
}
|
||||||
|
lineCh <- bytes.TrimSpace(b)
|
||||||
|
}()
|
||||||
|
var line []byte
|
||||||
|
select {
|
||||||
|
case <-ctx.Done():
|
||||||
|
return nil, ctx.Err()
|
||||||
|
case err := <-errCh:
|
||||||
|
return nil, err
|
||||||
|
case line = <-lineCh:
|
||||||
|
}
|
||||||
|
if len(line) == 0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
var resp mcpRPCResponse
|
||||||
|
if json.Unmarshal(line, &resp) != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if resp.ID == nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if n, ok := resp.ID.(float64); !ok || int64(n) != wantID {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if resp.Error != nil {
|
||||||
|
return nil, fmt.Errorf("MCP error %d: %s", resp.Error.Code, resp.Error.Message)
|
||||||
|
}
|
||||||
|
var out any
|
||||||
|
if len(resp.Result) > 0 {
|
||||||
|
if err := json.Unmarshal(resp.Result, &out); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return out, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func callHTTP(ctx context.Context, s MCPService, tool string, args any) (any, error) {
|
||||||
|
if s.URL == "" {
|
||||||
|
return nil, errors.New("MCP HTTP URL is empty")
|
||||||
|
}
|
||||||
|
client := &http.Client{Timeout: 0}
|
||||||
|
var session string
|
||||||
|
post := func(req mcpRPC) (any, http.Header, error) {
|
||||||
|
b, _ := json.Marshal(req)
|
||||||
|
h := make(http.Header)
|
||||||
|
h.Set("Content-Type", "application/json")
|
||||||
|
h.Set("Accept", "application/json, text/event-stream")
|
||||||
|
for k, v := range s.Headers {
|
||||||
|
h.Set(k, v)
|
||||||
|
}
|
||||||
|
if session != "" {
|
||||||
|
h.Set("Mcp-Session-Id", session)
|
||||||
|
}
|
||||||
|
hr, err := http.NewRequestWithContext(ctx, http.MethodPost, s.URL, bytes.NewReader(b))
|
||||||
|
if err != nil {
|
||||||
|
return nil, nil, err
|
||||||
|
}
|
||||||
|
hr.Header = h
|
||||||
|
resp, err := client.Do(hr)
|
||||||
|
if err != nil {
|
||||||
|
return nil, nil, err
|
||||||
|
}
|
||||||
|
defer resp.Body.Close()
|
||||||
|
if resp.StatusCode < 200 || resp.StatusCode >= 300 {
|
||||||
|
return nil, resp.Header, fmt.Errorf("MCP HTTP status %s", resp.Status)
|
||||||
|
}
|
||||||
|
if v := resp.Header.Get("Mcp-Session-Id"); v != "" {
|
||||||
|
session = v
|
||||||
|
}
|
||||||
|
data, err := io.ReadAll(io.LimitReader(resp.Body, 4*1024*1024))
|
||||||
|
if err != nil {
|
||||||
|
return nil, nil, err
|
||||||
|
}
|
||||||
|
if len(bytes.TrimSpace(data)) == 0 {
|
||||||
|
return nil, resp.Header, nil
|
||||||
|
}
|
||||||
|
if bytes.Contains(data, []byte("event:")) {
|
||||||
|
data = parseSSEData(data)
|
||||||
|
}
|
||||||
|
var rr mcpRPCResponse
|
||||||
|
if err := json.Unmarshal(bytes.TrimSpace(data), &rr); err != nil {
|
||||||
|
return nil, resp.Header, err
|
||||||
|
}
|
||||||
|
if rr.Error != nil {
|
||||||
|
return nil, resp.Header, fmt.Errorf("MCP error %d: %s", rr.Error.Code, rr.Error.Message)
|
||||||
|
}
|
||||||
|
var out any
|
||||||
|
if len(rr.Result) > 0 {
|
||||||
|
if err := json.Unmarshal(rr.Result, &out); err != nil {
|
||||||
|
return nil, resp.Header, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return out, resp.Header, nil
|
||||||
|
}
|
||||||
|
if strings.EqualFold(s.Transport, "sse") {
|
||||||
|
endpoint, err := discoverSSEEndpoint(ctx, s)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
old := s.URL
|
||||||
|
s.URL = endpoint
|
||||||
|
defer func() { s.URL = old }()
|
||||||
|
}
|
||||||
|
if _, _, err := post(mcpRPC{JSONRPC: "2.0", ID: int64(1), Method: "initialize", Params: map[string]any{"protocolVersion": "2024-11-05", "capabilities": map[string]any{}, "clientInfo": map[string]string{"name": "vps-commander-agent", "version": "1.0"}}}); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if _, _, err := post(mcpRPC{JSONRPC: "2.0", Method: "notifications/initialized", Params: map[string]any{}}); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return func() (any, error) {
|
||||||
|
v, _, err := post(mcpRPC{JSONRPC: "2.0", ID: int64(2), Method: "tools/call", Params: map[string]any{"name": tool, "arguments": args}})
|
||||||
|
return v, err
|
||||||
|
}()
|
||||||
|
}
|
||||||
|
|
||||||
|
func discoverSSEEndpoint(ctx context.Context, s MCPService) (string, error) {
|
||||||
|
req, err := http.NewRequestWithContext(ctx, http.MethodGet, s.URL, nil)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
for k, v := range s.Headers {
|
||||||
|
req.Header.Set(k, v)
|
||||||
|
}
|
||||||
|
resp, err := http.DefaultClient.Do(req)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
defer resp.Body.Close()
|
||||||
|
if resp.StatusCode < 200 || resp.StatusCode >= 300 {
|
||||||
|
return "", fmt.Errorf("MCP SSE status %s", resp.Status)
|
||||||
|
}
|
||||||
|
sc := bufio.NewScanner(io.LimitReader(resp.Body, 256*1024))
|
||||||
|
isEndpoint := false
|
||||||
|
for sc.Scan() {
|
||||||
|
line := sc.Text()
|
||||||
|
if strings.HasPrefix(line, "event:") {
|
||||||
|
isEndpoint = strings.TrimSpace(strings.TrimPrefix(line, "event:")) == "endpoint"
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if isEndpoint && strings.HasPrefix(line, "data:") {
|
||||||
|
u := strings.TrimSpace(strings.TrimPrefix(line, "data:"))
|
||||||
|
if strings.HasPrefix(u, "/") {
|
||||||
|
base := *resp.Request.URL
|
||||||
|
base.Path = u
|
||||||
|
base.RawQuery = ""
|
||||||
|
return base.String(), nil
|
||||||
|
}
|
||||||
|
return u, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return "", errors.New("MCP SSE endpoint event not received")
|
||||||
|
}
|
||||||
|
|
||||||
|
func parseSSEData(b []byte) []byte {
|
||||||
|
for _, line := range strings.Split(string(b), "\n") {
|
||||||
|
if strings.HasPrefix(line, "data:") {
|
||||||
|
return []byte(strings.TrimSpace(strings.TrimPrefix(line, "data:")))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return b
|
||||||
|
}
|
||||||
|
func envList(m map[string]string) []string {
|
||||||
|
out := make([]string, 0, len(m))
|
||||||
|
for k, v := range m {
|
||||||
|
out = append(out, k+"="+v)
|
||||||
|
}
|
||||||
|
return out
|
||||||
|
}
|
||||||
@@ -0,0 +1,80 @@
|
|||||||
|
package cluster
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"net/http"
|
||||||
|
"net/http/httptest"
|
||||||
|
"os"
|
||||||
|
"os/exec"
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestMCPRuntimeStdioOnDemand(t *testing.T) {
|
||||||
|
if _, err := exec.LookPath("python3"); err != nil {
|
||||||
|
t.Skip("python3 unavailable")
|
||||||
|
}
|
||||||
|
dir := t.TempDir()
|
||||||
|
script := dir + "/mcp.py"
|
||||||
|
code := `import json,sys
|
||||||
|
for line in sys.stdin:
|
||||||
|
r=json.loads(line)
|
||||||
|
if r.get("method")=="initialize":
|
||||||
|
print(json.dumps({"jsonrpc":"2.0","id":r["id"],"result":{"protocolVersion":"2024-11-05"}}),flush=True)
|
||||||
|
elif r.get("method")=="tools/call":
|
||||||
|
a=r.get("params",{}).get("arguments",{})
|
||||||
|
print(json.dumps({"jsonrpc":"2.0","id":r["id"],"result":{"content":[{"type":"text","text":"hello "+a.get("name","")}]}}),flush=True)
|
||||||
|
`
|
||||||
|
if err := os.WriteFile(script, []byte(code), 0755); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
r := NewMCPRuntime()
|
||||||
|
r.Sync([]MCPService{{ID: "test", Transport: "stdio", Command: "python3", Args: []string{script}}})
|
||||||
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||||||
|
defer cancel()
|
||||||
|
v, err := r.Call(ctx, "test", "hello", map[string]any{"name": "vpc"})
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
b, _ := json.Marshal(v)
|
||||||
|
if string(b) == "" {
|
||||||
|
t.Fatal("empty result")
|
||||||
|
}
|
||||||
|
if !contains(string(b), "hello vpc") {
|
||||||
|
t.Fatalf("unexpected %s", b)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestMCPRuntimeHTTP(t *testing.T) {
|
||||||
|
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
b, _ := io.ReadAll(r.Body)
|
||||||
|
var q mcpRPC
|
||||||
|
_ = json.Unmarshal(b, &q)
|
||||||
|
w.Header().Set("Content-Type", "application/json")
|
||||||
|
fmt.Fprintf(w, `{"jsonrpc":"2.0","id":%s,"result":{"ok":true}}`, jsonID(q.ID))
|
||||||
|
}))
|
||||||
|
defer ts.Close()
|
||||||
|
r := NewMCPRuntime()
|
||||||
|
r.Sync([]MCPService{{ID: "http", Transport: "http", URL: ts.URL}})
|
||||||
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||||||
|
defer cancel()
|
||||||
|
v, err := r.Call(ctx, "http", "ping", map[string]any{})
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
if !contains(fmt.Sprint(v), "true") {
|
||||||
|
t.Fatalf("unexpected %v", v)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
func jsonID(v any) string { b, _ := json.Marshal(v); return string(b) }
|
||||||
|
func contains(s, x string) bool {
|
||||||
|
for i := 0; i+len(x) <= len(s); i++ {
|
||||||
|
if s[i:i+len(x)] == x {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
@@ -7,6 +7,8 @@ type Message struct {
|
|||||||
Event string `json:"event,omitempty"`
|
Event string `json:"event,omitempty"`
|
||||||
Action string `json:"action,omitempty"`
|
Action string `json:"action,omitempty"`
|
||||||
Device string `json:"device,omitempty"`
|
Device string `json:"device,omitempty"`
|
||||||
|
Arch string `json:"arch,omitempty"`
|
||||||
|
OS string `json:"os,omitempty"`
|
||||||
Status string `json:"status,omitempty"`
|
Status string `json:"status,omitempty"`
|
||||||
Payload any `json:"payload,omitempty"`
|
Payload any `json:"payload,omitempty"`
|
||||||
Profile *sysinfo.Profile `json:"profile,omitempty"`
|
Profile *sysinfo.Profile `json:"profile,omitempty"`
|
||||||
|
|||||||
@@ -76,6 +76,11 @@ type ActionBackend interface {
|
|||||||
WriteFile(ctx context.Context, clientIP, device, path, content string) (any, error)
|
WriteFile(ctx context.Context, clientIP, device, path, content string) (any, error)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type AgentMCPBackend interface {
|
||||||
|
ListAgentMCP(ctx context.Context, device string) (any, error)
|
||||||
|
CallAgentMCP(ctx context.Context, clientIP, device, server, tool string, args any) (any, error)
|
||||||
|
}
|
||||||
|
|
||||||
type Server struct {
|
type Server struct {
|
||||||
Backend ActionBackend
|
Backend ActionBackend
|
||||||
Auth *auth.Manager
|
Auth *auth.Manager
|
||||||
@@ -272,6 +277,14 @@ func (s *Server) handleRequest(ctx context.Context, r *http.Request, req JSONRPC
|
|||||||
Required: []string{"path", "content"},
|
Required: []string{"path", "content"},
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
|
{
|
||||||
|
Name: "list_agent_mcp", Description: "列出指定 Agent 当前通过 Hub 动态挂载的 MCP 服务及状态",
|
||||||
|
InputSchema: InputSchema{Type: "object", Properties: map[string]PropertyDef{"device": {Type: "string", Description: "目标 Agent 节点名称"}}, Required: []string{"device"}},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Name: "call_agent_mcp", Description: "通过 Hub 调度指定 Agent 上已挂载的 MCP 工具",
|
||||||
|
InputSchema: InputSchema{Type: "object", Properties: map[string]PropertyDef{"device": {Type: "string", Description: "目标 Agent 节点名称"}, "server": {Type: "string", Description: "MCP 服务 ID"}, "tool": {Type: "string", Description: "MCP 工具名称"}, "arguments": {Type: "object", Description: "MCP 工具参数 JSON 对象"}}, Required: []string{"device", "server", "tool"}},
|
||||||
|
},
|
||||||
}
|
}
|
||||||
return JSONRPCResponse{
|
return JSONRPCResponse{
|
||||||
JSONRPC: "2.0",
|
JSONRPC: "2.0",
|
||||||
@@ -372,6 +385,43 @@ func (s *Server) callTool(ctx context.Context, clientIP string, params CallToolP
|
|||||||
}
|
}
|
||||||
return res, false
|
return res, false
|
||||||
|
|
||||||
|
case "list_agent_mcp":
|
||||||
|
var args struct {
|
||||||
|
Device string `json:"device"`
|
||||||
|
}
|
||||||
|
_ = json.Unmarshal(params.Arguments, &args)
|
||||||
|
b, ok := s.Backend.(AgentMCPBackend)
|
||||||
|
if !ok {
|
||||||
|
return map[string]string{"error": "agent MCP unavailable"}, true
|
||||||
|
}
|
||||||
|
res, err := b.ListAgentMCP(ctx, args.Device)
|
||||||
|
if err != nil {
|
||||||
|
return map[string]string{"error": err.Error()}, true
|
||||||
|
}
|
||||||
|
return res, false
|
||||||
|
|
||||||
|
case "call_agent_mcp":
|
||||||
|
var args struct {
|
||||||
|
Device string `json:"device"`
|
||||||
|
Server string `json:"server"`
|
||||||
|
Tool string `json:"tool"`
|
||||||
|
Arguments json.RawMessage `json:"arguments"`
|
||||||
|
}
|
||||||
|
_ = json.Unmarshal(params.Arguments, &args)
|
||||||
|
b, ok := s.Backend.(AgentMCPBackend)
|
||||||
|
if !ok {
|
||||||
|
return map[string]string{"error": "agent MCP unavailable"}, true
|
||||||
|
}
|
||||||
|
var av any
|
||||||
|
if len(args.Arguments) > 0 {
|
||||||
|
_ = json.Unmarshal(args.Arguments, &av)
|
||||||
|
}
|
||||||
|
res, err := b.CallAgentMCP(ctx, clientIP, args.Device, args.Server, args.Tool, av)
|
||||||
|
if err != nil {
|
||||||
|
return map[string]string{"error": err.Error()}, true
|
||||||
|
}
|
||||||
|
return res, false
|
||||||
|
|
||||||
default:
|
default:
|
||||||
return map[string]string{"error": "unknown tool: " + params.Name}, true
|
return map[string]string{"error": "unknown tool: " + params.Name}, true
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -75,8 +75,8 @@ func TestMCPInitializeAndListTools(t *testing.T) {
|
|||||||
t.Fatalf("expected map result")
|
t.Fatalf("expected map result")
|
||||||
}
|
}
|
||||||
tools, ok := resMap["tools"].([]any)
|
tools, ok := resMap["tools"].([]any)
|
||||||
if !ok || len(tools) != 4 {
|
if !ok || len(tools) != 6 {
|
||||||
t.Fatalf("expected 4 tools, got %d", len(tools))
|
t.Fatalf("expected 6 tools, got %d", len(tools))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -0,0 +1,51 @@
|
|||||||
|
package storage
|
||||||
|
|
||||||
|
import (
|
||||||
|
"path/filepath"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestDeviceLifecyclePersistence(t *testing.T) {
|
||||||
|
s, err := Open(filepath.Join(t.TempDir(), "commander.db"))
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
defer s.DB.Close()
|
||||||
|
|
||||||
|
if err := s.UpsertRemote("node1", "online", "amd64", "linux", 100, "{\"role\":\"APPLICATION\"}"); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
rows, err := s.ListDevices()
|
||||||
|
if err != nil || len(rows) != 1 {
|
||||||
|
t.Fatalf("rows=%d err=%v", len(rows), err)
|
||||||
|
}
|
||||||
|
if rows[0].Status != "online" || rows[0].Arch != "amd64" || rows[0].OS != "linux" {
|
||||||
|
t.Fatalf("unexpected row: %+v", rows[0])
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := s.UpdateDeviceStatus("node1", "offline", 120); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
rows, err = s.ListDevices()
|
||||||
|
if err != nil || len(rows) != 1 {
|
||||||
|
t.Fatalf("rows=%d err=%v", len(rows), err)
|
||||||
|
}
|
||||||
|
if rows[0].Status != "offline" || rows[0].OfflineAt == nil {
|
||||||
|
t.Fatalf("offline state not persisted: %+v", rows[0])
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := s.RevokeDevice("node1", "test"); err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
rows, err = s.ListDevices()
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
if len(rows) != 0 {
|
||||||
|
t.Fatalf("revoked device still present: %+v", rows)
|
||||||
|
}
|
||||||
|
revoked, err := s.IsRevoked("node1")
|
||||||
|
if err != nil || !revoked {
|
||||||
|
t.Fatalf("revoked=%v err=%v", revoked, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
+239
-33
@@ -2,6 +2,8 @@ package storage
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"database/sql"
|
"database/sql"
|
||||||
|
"encoding/json"
|
||||||
|
"errors"
|
||||||
_ "modernc.org/sqlite"
|
_ "modernc.org/sqlite"
|
||||||
"os"
|
"os"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
@@ -18,8 +20,6 @@ func Open(path string) (*Store, error) {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
// SQLite has a single-writer model. Keep one application connection so
|
|
||||||
// concurrent callers queue instead of opening competing writer connections.
|
|
||||||
db.SetMaxOpenConns(1)
|
db.SetMaxOpenConns(1)
|
||||||
db.SetMaxIdleConns(1)
|
db.SetMaxIdleConns(1)
|
||||||
s := &Store{db}
|
s := &Store{db}
|
||||||
@@ -31,28 +31,235 @@ func Open(path string) (*Store, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (s *Store) init() error {
|
func (s *Store) init() error {
|
||||||
if _, err := s.DB.Exec(`PRAGMA journal_mode = WAL`); err != nil {
|
if _, err := s.DB.Exec("PRAGMA journal_mode = WAL"); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
if _, err := s.DB.Exec(`PRAGMA busy_timeout = 5000`); err != nil {
|
if _, err := s.DB.Exec("PRAGMA busy_timeout = 5000"); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
_, err := s.DB.Exec(`CREATE TABLE IF NOT EXISTS devices (
|
_, err := s.DB.Exec("CREATE TABLE IF NOT EXISTS devices (" +
|
||||||
name TEXT PRIMARY KEY, status TEXT NOT NULL, is_local INTEGER NOT NULL,
|
"name TEXT PRIMARY KEY, status TEXT NOT NULL DEFAULT 'offline', is_local INTEGER NOT NULL DEFAULT 0, " +
|
||||||
arch TEXT, os TEXT, updated_at INTEGER NOT NULL
|
"arch TEXT, os TEXT, last_seen INTEGER NOT NULL DEFAULT 0, last_heartbeat INTEGER NOT NULL DEFAULT 0, " +
|
||||||
); CREATE TABLE IF NOT EXISTS audit_logs (
|
"offline_at INTEGER, profile_json TEXT, created_at INTEGER NOT NULL DEFAULT 0, updated_at INTEGER NOT NULL DEFAULT 0); " +
|
||||||
id INTEGER PRIMARY KEY AUTOINCREMENT, timestamp INTEGER NOT NULL,
|
"CREATE TABLE IF NOT EXISTS audit_logs (" +
|
||||||
caller_ip TEXT, device TEXT, action TEXT, command TEXT,
|
"id INTEGER PRIMARY KEY AUTOINCREMENT, timestamp INTEGER NOT NULL, caller_ip TEXT, device TEXT, action TEXT, command TEXT, " +
|
||||||
exit_code INTEGER, duration_ms INTEGER, error_msg TEXT
|
"exit_code INTEGER, duration_ms INTEGER, error_msg TEXT); " +
|
||||||
);`)
|
"CREATE TABLE IF NOT EXISTS revoked_devices (name TEXT PRIMARY KEY, revoked_at INTEGER NOT NULL, reason TEXT); CREATE TABLE IF NOT EXISTS mcp_services (id TEXT PRIMARY KEY, name TEXT NOT NULL, description TEXT, transport TEXT NOT NULL, command TEXT, args_json TEXT, env_json TEXT, url TEXT, headers_json TEXT, scope TEXT NOT NULL DEFAULT 'all', target_nodes_json TEXT, enabled INTEGER NOT NULL DEFAULT 1, created_at INTEGER NOT NULL, updated_at INTEGER NOT NULL); CREATE INDEX IF NOT EXISTS idx_mcp_services_enabled ON mcp_services(enabled);")
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return s.migrateDevices()
|
||||||
|
}
|
||||||
|
|
||||||
|
type MCPService struct {
|
||||||
|
ID string `json:"id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
Description string `json:"description,omitempty"`
|
||||||
|
Transport string `json:"transport"`
|
||||||
|
Command string `json:"command,omitempty"`
|
||||||
|
ArgsJSON string `json:"args_json,omitempty"`
|
||||||
|
EnvJSON string `json:"env_json,omitempty"`
|
||||||
|
URL string `json:"url,omitempty"`
|
||||||
|
HeadersJSON string `json:"headers_json,omitempty"`
|
||||||
|
Scope string `json:"scope"`
|
||||||
|
TargetNodesJSON string `json:"target_nodes_json,omitempty"`
|
||||||
|
Enabled bool `json:"enabled"`
|
||||||
|
CreatedAt int64 `json:"created_at"`
|
||||||
|
UpdatedAt int64 `json:"updated_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Store) SaveMCPService(v MCPService) error {
|
||||||
|
now := time.Now().Unix()
|
||||||
|
if v.CreatedAt <= 0 {
|
||||||
|
v.CreatedAt = now
|
||||||
|
}
|
||||||
|
v.UpdatedAt = now
|
||||||
|
if v.Scope == "" {
|
||||||
|
v.Scope = "all"
|
||||||
|
}
|
||||||
|
if v.Transport == "" {
|
||||||
|
return errors.New("transport required")
|
||||||
|
}
|
||||||
|
_, err := s.DB.Exec(`INSERT INTO mcp_services(id,name,description,transport,command,args_json,env_json,url,headers_json,scope,target_nodes_json,enabled,created_at,updated_at)
|
||||||
|
VALUES(?,?,?,?,?,?,?,?,?,?,?,?,?,?) ON CONFLICT(id) DO UPDATE SET name=excluded.name,description=excluded.description,transport=excluded.transport,command=excluded.command,args_json=excluded.args_json,env_json=excluded.env_json,url=excluded.url,headers_json=excluded.headers_json,scope=excluded.scope,target_nodes_json=excluded.target_nodes_json,enabled=excluded.enabled,updated_at=excluded.updated_at`,
|
||||||
|
v.ID, v.Name, v.Description, v.Transport, v.Command, v.ArgsJSON, v.EnvJSON, v.URL, v.HeadersJSON, v.Scope, v.TargetNodesJSON, boolInt(v.Enabled), v.CreatedAt, v.UpdatedAt)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
func boolInt(v bool) int {
|
||||||
|
if v {
|
||||||
|
return 1
|
||||||
|
}
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
func (s *Store) DeleteMCPService(id string) error {
|
||||||
|
_, err := s.DB.Exec(`DELETE FROM mcp_services WHERE id=?`, id)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
func (s *Store) ListMCPServices() ([]MCPService, error) {
|
||||||
|
return s.listMCP(`SELECT id,name,description,transport,COALESCE(command,''),COALESCE(args_json,''),COALESCE(env_json,''),COALESCE(url,''),COALESCE(headers_json,''),scope,COALESCE(target_nodes_json,''),enabled,created_at,updated_at FROM mcp_services ORDER BY id`)
|
||||||
|
}
|
||||||
|
func (s *Store) ListMCPServicesForNode(node string) ([]MCPService, error) {
|
||||||
|
all, err := s.listMCP(`SELECT id,name,description,transport,COALESCE(command,''),COALESCE(args_json,''),COALESCE(env_json,''),COALESCE(url,''),COALESCE(headers_json,''),scope,COALESCE(target_nodes_json,''),enabled,created_at,updated_at FROM mcp_services WHERE enabled=1 ORDER BY id`)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
out := make([]MCPService, 0, len(all))
|
||||||
|
for _, v := range all {
|
||||||
|
if v.Scope != "custom" || containsNode(v.TargetNodesJSON, node) {
|
||||||
|
out = append(out, v)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return out, nil
|
||||||
|
}
|
||||||
|
func containsNode(raw, node string) bool {
|
||||||
|
var a []string
|
||||||
|
if json.Unmarshal([]byte(raw), &a) != nil {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
for _, v := range a {
|
||||||
|
if v == node {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
func (s *Store) listMCP(q string) ([]MCPService, error) {
|
||||||
|
rows, err := s.DB.Query(q)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer rows.Close()
|
||||||
|
out := []MCPService{}
|
||||||
|
for rows.Next() {
|
||||||
|
var v MCPService
|
||||||
|
var en int
|
||||||
|
if err := rows.Scan(&v.ID, &v.Name, &v.Description, &v.Transport, &v.Command, &v.ArgsJSON, &v.EnvJSON, &v.URL, &v.HeadersJSON, &v.Scope, &v.TargetNodesJSON, &en, &v.CreatedAt, &v.UpdatedAt); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
v.Enabled = en != 0
|
||||||
|
out = append(out, v)
|
||||||
|
}
|
||||||
|
return out, rows.Err()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Store) migrateDevices() error {
|
||||||
|
cols := map[string]bool{}
|
||||||
|
rows, err := s.DB.Query("PRAGMA table_info(devices)")
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer rows.Close()
|
||||||
|
for rows.Next() {
|
||||||
|
var cid, notnull, pk int
|
||||||
|
var name, typ string
|
||||||
|
var dflt sql.NullString
|
||||||
|
if err := rows.Scan(&cid, &name, &typ, ¬null, &dflt, &pk); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
cols[name] = true
|
||||||
|
}
|
||||||
|
if err := rows.Err(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
migrations := []struct{ name, sql string }{
|
||||||
|
{"last_seen", "ALTER TABLE devices ADD COLUMN last_seen INTEGER NOT NULL DEFAULT 0"},
|
||||||
|
{"last_heartbeat", "ALTER TABLE devices ADD COLUMN last_heartbeat INTEGER NOT NULL DEFAULT 0"},
|
||||||
|
{"offline_at", "ALTER TABLE devices ADD COLUMN offline_at INTEGER"},
|
||||||
|
{"profile_json", "ALTER TABLE devices ADD COLUMN profile_json TEXT"},
|
||||||
|
{"created_at", "ALTER TABLE devices ADD COLUMN created_at INTEGER NOT NULL DEFAULT 0"},
|
||||||
|
}
|
||||||
|
for _, m := range migrations {
|
||||||
|
if !cols[m.name] {
|
||||||
|
if _, err := s.DB.Exec(m.sql); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_, err = s.DB.Exec("UPDATE devices SET created_at=CASE WHEN created_at=0 THEN updated_at ELSE created_at END, " +
|
||||||
|
"last_seen=CASE WHEN last_seen=0 THEN updated_at ELSE last_seen END, " +
|
||||||
|
"last_heartbeat=CASE WHEN last_heartbeat=0 THEN updated_at ELSE last_heartbeat END " +
|
||||||
|
"WHERE created_at=0 OR last_seen=0 OR last_heartbeat=0")
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *Store) UpsertLocal(name, arch, osName string) error {
|
func (s *Store) UpsertLocal(name, arch, osName string) error {
|
||||||
_, err := s.DB.Exec(`INSERT INTO devices(name,status,is_local,arch,os,updated_at)
|
now := time.Now().Unix()
|
||||||
VALUES(?,?,?,?,?,?) ON CONFLICT(name) DO UPDATE SET status=excluded.status,
|
_, err := s.DB.Exec("INSERT INTO devices(name,status,is_local,arch,os,last_seen,last_heartbeat,offline_at,profile_json,created_at,updated_at) "+
|
||||||
arch=excluded.arch, os=excluded.os, updated_at=excluded.updated_at`,
|
"VALUES(?,?,?,?,?,?,?,?,?,?,?) ON CONFLICT(name) DO UPDATE SET status='online', is_local=1, arch=excluded.arch, os=excluded.os, offline_at=NULL, updated_at=excluded.updated_at",
|
||||||
name, "online", 1, arch, osName, time.Now().Unix())
|
name, "online", 1, arch, osName, now, now, nil, nil, now, now)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Store) UpsertRemote(name, status, arch, osName string, lastHeartbeat int64, profileJSON string) error {
|
||||||
|
now := time.Now().Unix()
|
||||||
|
if lastHeartbeat <= 0 {
|
||||||
|
lastHeartbeat = now
|
||||||
|
}
|
||||||
|
_, err := s.DB.Exec("INSERT INTO devices(name,status,is_local,arch,os,last_seen,last_heartbeat,offline_at,profile_json,created_at,updated_at) "+
|
||||||
|
"VALUES(?,?,?,?,?,?,?,?,?,?,?) ON CONFLICT(name) DO UPDATE SET status=excluded.status, is_local=0, arch=excluded.arch, os=excluded.os, "+
|
||||||
|
"last_seen=excluded.last_seen, last_heartbeat=excluded.last_heartbeat, offline_at=NULL, profile_json=excluded.profile_json, updated_at=excluded.updated_at",
|
||||||
|
name, status, 0, arch, osName, lastHeartbeat, lastHeartbeat, nil, profileJSON, now, now)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Store) UpdateRemoteProfile(name string, lastHeartbeat int64, arch, osName, profileJSON string) error {
|
||||||
|
now := time.Now().Unix()
|
||||||
|
if lastHeartbeat <= 0 {
|
||||||
|
lastHeartbeat = now
|
||||||
|
}
|
||||||
|
_, err := s.DB.Exec("UPDATE devices SET status='online', arch=?, os=?, last_seen=?, last_heartbeat=?, offline_at=NULL, profile_json=?, updated_at=? WHERE name=? AND is_local=0",
|
||||||
|
arch, osName, lastHeartbeat, lastHeartbeat, profileJSON, now, name)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Store) UpdateDeviceStatus(name, status string, lastHeartbeat int64) error {
|
||||||
|
now := time.Now().Unix()
|
||||||
|
if lastHeartbeat <= 0 {
|
||||||
|
lastHeartbeat = now
|
||||||
|
}
|
||||||
|
var offlineAt any
|
||||||
|
if status == "offline" {
|
||||||
|
offlineAt = now
|
||||||
|
}
|
||||||
|
_, err := s.DB.Exec("UPDATE devices SET status=?, last_seen=?, last_heartbeat=?, offline_at=?, updated_at=? WHERE name=? AND is_local=0",
|
||||||
|
status, lastHeartbeat, lastHeartbeat, offlineAt, now, name)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
type DeviceRecord struct {
|
||||||
|
Name, Status, Arch, OS, ProfileJSON string
|
||||||
|
IsLocal bool
|
||||||
|
LastSeen, LastHeartbeat int64
|
||||||
|
OfflineAt *int64
|
||||||
|
UpdatedAt int64
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Store) ListDevices() ([]DeviceRecord, error) {
|
||||||
|
rows, err := s.DB.Query("SELECT name,status,is_local,COALESCE(arch,''),COALESCE(os,''),last_seen,last_heartbeat,offline_at,COALESCE(profile_json,''),updated_at FROM devices ORDER BY name")
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer rows.Close()
|
||||||
|
var out []DeviceRecord
|
||||||
|
for rows.Next() {
|
||||||
|
var d DeviceRecord
|
||||||
|
var local int
|
||||||
|
var offline sql.NullInt64
|
||||||
|
if err := rows.Scan(&d.Name, &d.Status, &local, &d.Arch, &d.OS, &d.LastSeen, &d.LastHeartbeat, &offline, &d.ProfileJSON, &d.UpdatedAt); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
d.IsLocal = local != 0
|
||||||
|
if offline.Valid {
|
||||||
|
v := offline.Int64
|
||||||
|
d.OfflineAt = &v
|
||||||
|
}
|
||||||
|
out = append(out, d)
|
||||||
|
}
|
||||||
|
return out, rows.Err()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Store) DeleteDevice(name string) error {
|
||||||
|
_, err := s.DB.Exec("DELETE FROM devices WHERE name=? AND is_local=0", name)
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -69,8 +276,8 @@ type AuditLog struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (s *Store) Audit(ip, device, action, command string, code int, ms int64, msg string) error {
|
func (s *Store) Audit(ip, device, action, command string, code int, ms int64, msg string) error {
|
||||||
_, err := s.DB.Exec(`INSERT INTO audit_logs(timestamp,caller_ip,device,action,command,exit_code,duration_ms,error_msg)
|
_, err := s.DB.Exec("INSERT INTO audit_logs(timestamp,caller_ip,device,action,command,exit_code,duration_ms,error_msg) VALUES(?,?,?,?,?,?,?,?)",
|
||||||
VALUES(?,?,?,?,?,?,?,?)`, time.Now().Unix(), ip, device, action, command, code, ms, msg)
|
time.Now().Unix(), ip, device, action, command, code, ms, msg)
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -81,8 +288,7 @@ func (s *Store) RecentAudits(limit int) ([]AuditLog, error) {
|
|||||||
if limit > 500 {
|
if limit > 500 {
|
||||||
limit = 500
|
limit = 500
|
||||||
}
|
}
|
||||||
rows, err := s.DB.Query(`SELECT id,timestamp,caller_ip,device,action,command,exit_code,duration_ms,error_msg
|
rows, err := s.DB.Query("SELECT id,timestamp,caller_ip,device,action,command,exit_code,duration_ms,error_msg FROM audit_logs ORDER BY id DESC LIMIT ?", limit)
|
||||||
FROM audit_logs ORDER BY id DESC LIMIT ?`, limit)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
@@ -98,29 +304,29 @@ func (s *Store) RecentAudits(limit int) ([]AuditLog, error) {
|
|||||||
return out, rows.Err()
|
return out, rows.Err()
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *Store) InitRevocationTable() error {
|
func (s *Store) RevokeDevice(name, reason string) error {
|
||||||
_, err := s.DB.Exec(`CREATE TABLE IF NOT EXISTS revoked_devices (
|
tx, err := s.DB.Begin()
|
||||||
name TEXT PRIMARY KEY,
|
if err != nil {
|
||||||
revoked_at INTEGER NOT NULL,
|
|
||||||
reason TEXT
|
|
||||||
);`)
|
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
defer tx.Rollback()
|
||||||
func (s *Store) RevokeDevice(name, reason string) error {
|
if _, err := tx.Exec("INSERT INTO revoked_devices(name,revoked_at,reason) VALUES(?,?,?) ON CONFLICT(name) DO UPDATE SET revoked_at=excluded.revoked_at,reason=excluded.reason",
|
||||||
_, err := s.DB.Exec(`INSERT INTO revoked_devices(name, revoked_at, reason)
|
name, time.Now().Unix(), reason); err != nil {
|
||||||
VALUES(?, ?, ?) ON CONFLICT(name) DO UPDATE SET revoked_at=excluded.revoked_at, reason=excluded.reason`,
|
|
||||||
name, time.Now().Unix(), reason)
|
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
if _, err := tx.Exec("DELETE FROM devices WHERE name=? AND is_local=0", name); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return tx.Commit()
|
||||||
|
}
|
||||||
|
|
||||||
func (s *Store) UnrevokeDevice(name string) error {
|
func (s *Store) UnrevokeDevice(name string) error {
|
||||||
_, err := s.DB.Exec(`DELETE FROM revoked_devices WHERE name = ?`, name)
|
_, err := s.DB.Exec("DELETE FROM revoked_devices WHERE name = ?", name)
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *Store) IsRevoked(name string) (bool, error) {
|
func (s *Store) IsRevoked(name string) (bool, error) {
|
||||||
var count int
|
var count int
|
||||||
err := s.DB.QueryRow(`SELECT COUNT(*) FROM revoked_devices WHERE name = ?`, name).Scan(&count)
|
err := s.DB.QueryRow("SELECT COUNT(*) FROM revoked_devices WHERE name = ?", name).Scan(&count)
|
||||||
return count > 0, err
|
return count > 0, err
|
||||||
}
|
}
|
||||||
|
|||||||
+39
-30
@@ -12,13 +12,14 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type Profile struct {
|
type Profile struct {
|
||||||
Role string `json:"role"`
|
Role string `json:"role,omitempty"`
|
||||||
CPUCores int `json:"cpu_cores"`
|
CPUCores int `json:"cpu_cores"`
|
||||||
CPUUsagePercent int `json:"cpu_usage_percent"`
|
CPUUsagePercent int `json:"cpu_usage_percent"`
|
||||||
|
Load1M float64 `json:"load_1m"`
|
||||||
MemPercent int `json:"mem_percent"`
|
MemPercent int `json:"mem_percent"`
|
||||||
MemAvailable string `json:"mem_available"`
|
MemAvailable string `json:"mem_available"`
|
||||||
SwapPercent int `json:"swap_percent"`
|
|
||||||
DiskPercent int `json:"disk_percent"`
|
DiskPercent int `json:"disk_percent"`
|
||||||
|
DiskFree string `json:"disk_free"`
|
||||||
IOPSI float64 `json:"io_psi"`
|
IOPSI float64 `json:"io_psi"`
|
||||||
Docker bool `json:"docker"`
|
Docker bool `json:"docker"`
|
||||||
UpdatedAt int64 `json:"updated_at"`
|
UpdatedAt int64 `json:"updated_at"`
|
||||||
@@ -35,8 +36,9 @@ func Collect(defaultRole string) Profile {
|
|||||||
|
|
||||||
cores := runtime.NumCPU()
|
cores := runtime.NumCPU()
|
||||||
cpuPercent := getCPUUsage()
|
cpuPercent := getCPUUsage()
|
||||||
memPct, memAvl, swapPct := getMemAndSwap()
|
load1m := getLoad1M()
|
||||||
diskPct := getDiskPercent()
|
memPct, memAvl := getMemInfo()
|
||||||
|
diskPct, diskFree := getDiskInfo()
|
||||||
psi := getIOPSI()
|
psi := getIOPSI()
|
||||||
docker := hasDocker()
|
docker := hasDocker()
|
||||||
|
|
||||||
@@ -44,23 +46,34 @@ func Collect(defaultRole string) Profile {
|
|||||||
Role: role,
|
Role: role,
|
||||||
CPUCores: cores,
|
CPUCores: cores,
|
||||||
CPUUsagePercent: cpuPercent,
|
CPUUsagePercent: cpuPercent,
|
||||||
|
Load1M: load1m,
|
||||||
MemPercent: memPct,
|
MemPercent: memPct,
|
||||||
MemAvailable: memAvl,
|
MemAvailable: memAvl,
|
||||||
SwapPercent: swapPct,
|
|
||||||
DiskPercent: diskPct,
|
DiskPercent: diskPct,
|
||||||
|
DiskFree: diskFree,
|
||||||
IOPSI: psi,
|
IOPSI: psi,
|
||||||
Docker: docker,
|
Docker: docker,
|
||||||
UpdatedAt: time.Now().Unix(),
|
UpdatedAt: time.Now().Unix(),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func getLoad1M() float64 {
|
||||||
|
if data, err := os.ReadFile("/proc/loadavg"); err == nil {
|
||||||
|
fields := strings.Fields(string(data))
|
||||||
|
if len(fields) > 0 {
|
||||||
|
if val, err := strconv.ParseFloat(fields[0], 64); err == nil {
|
||||||
|
return val
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return 0.0
|
||||||
|
}
|
||||||
|
|
||||||
func getCPUUsage() int {
|
func getCPUUsage() int {
|
||||||
// 尝试从 /proc/stat 快速采样或 /proc/loadavg
|
|
||||||
pct := sampleCPUStat()
|
pct := sampleCPUStat()
|
||||||
if pct >= 0 {
|
if pct >= 0 {
|
||||||
return pct
|
return pct
|
||||||
}
|
}
|
||||||
// 回退到 loadavg / cores * 100
|
|
||||||
if data, err := os.ReadFile("/proc/loadavg"); err == nil {
|
if data, err := os.ReadFile("/proc/loadavg"); err == nil {
|
||||||
fields := strings.Fields(string(data))
|
fields := strings.Fields(string(data))
|
||||||
if len(fields) > 0 {
|
if len(fields) > 0 {
|
||||||
@@ -127,14 +140,14 @@ func sampleCPUStat() int {
|
|||||||
return int(busy)
|
return int(busy)
|
||||||
}
|
}
|
||||||
|
|
||||||
func getMemAndSwap() (memPct int, memAvl string, swapPct int) {
|
func getMemInfo() (memPct int, memAvl string) {
|
||||||
f, err := os.Open("/proc/meminfo")
|
f, err := os.Open("/proc/meminfo")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return 0, "N/A", 0
|
return 0, "N/A"
|
||||||
}
|
}
|
||||||
defer f.Close()
|
defer f.Close()
|
||||||
|
|
||||||
var total, free, avail, buffers, cached, swapTotal, swapFree uint64
|
var total, free, avail, buffers, cached uint64
|
||||||
scanner := bufio.NewScanner(f)
|
scanner := bufio.NewScanner(f)
|
||||||
for scanner.Scan() {
|
for scanner.Scan() {
|
||||||
line := scanner.Text()
|
line := scanner.Text()
|
||||||
@@ -159,10 +172,6 @@ func getMemAndSwap() (memPct int, memAvl string, swapPct int) {
|
|||||||
buffers = v
|
buffers = v
|
||||||
case "Cached":
|
case "Cached":
|
||||||
cached = v
|
cached = v
|
||||||
case "SwapTotal":
|
|
||||||
swapTotal = v
|
|
||||||
case "SwapFree":
|
|
||||||
swapFree = v
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -182,34 +191,35 @@ func getMemAndSwap() (memPct int, memAvl string, swapPct int) {
|
|||||||
memAvl = "N/A"
|
memAvl = "N/A"
|
||||||
}
|
}
|
||||||
|
|
||||||
if swapTotal > 0 {
|
return memPct, memAvl
|
||||||
swapUsed := swapTotal - swapFree
|
|
||||||
swapPct = int(float64(swapUsed) / float64(swapTotal) * 100)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return memPct, memAvl, swapPct
|
func getDiskInfo() (diskPct int, diskFree string) {
|
||||||
}
|
|
||||||
|
|
||||||
func getDiskPercent() int {
|
|
||||||
var stat syscall.Statfs_t
|
var stat syscall.Statfs_t
|
||||||
if err := syscall.Statfs("/", &stat); err != nil {
|
if err := syscall.Statfs("/", &stat); err != nil {
|
||||||
return 0
|
return 0, "N/A"
|
||||||
}
|
}
|
||||||
total := stat.Blocks * uint64(stat.Bsize)
|
total := stat.Blocks * uint64(stat.Bsize)
|
||||||
free := stat.Bfree * uint64(stat.Bsize)
|
free := stat.Bavail * uint64(stat.Bsize)
|
||||||
if total == 0 {
|
if total == 0 {
|
||||||
return 0
|
return 0, "N/A"
|
||||||
}
|
}
|
||||||
used := total - free
|
used := total - free
|
||||||
pct := int(float64(used) / float64(total) * 100)
|
diskPct = int(float64(used) / float64(total) * 100)
|
||||||
if pct > 100 {
|
if diskPct > 100 {
|
||||||
pct = 100
|
diskPct = 100
|
||||||
}
|
}
|
||||||
return pct
|
|
||||||
|
if free >= 1024*1024*1024 {
|
||||||
|
diskFree = fmt.Sprintf("%.1fG", float64(free)/(1024*1024*1024))
|
||||||
|
} else {
|
||||||
|
diskFree = fmt.Sprintf("%dM", free/(1024*1024))
|
||||||
|
}
|
||||||
|
|
||||||
|
return diskPct, diskFree
|
||||||
}
|
}
|
||||||
|
|
||||||
func getIOPSI() float64 {
|
func getIOPSI() float64 {
|
||||||
// 读取 /proc/pressure/io
|
|
||||||
data, err := os.ReadFile("/proc/pressure/io")
|
data, err := os.ReadFile("/proc/pressure/io")
|
||||||
if err == nil {
|
if err == nil {
|
||||||
lines := strings.Split(string(data), "\n")
|
lines := strings.Split(string(data), "\n")
|
||||||
@@ -227,7 +237,6 @@ func getIOPSI() float64 {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
// 回退:/proc/loadavg 的 1min load
|
|
||||||
if data, err := os.ReadFile("/proc/loadavg"); err == nil {
|
if data, err := os.ReadFile("/proc/loadavg"); err == nil {
|
||||||
fields := strings.Fields(string(data))
|
fields := strings.Fields(string(data))
|
||||||
if len(fields) > 0 {
|
if len(fields) > 0 {
|
||||||
|
|||||||
@@ -43,10 +43,50 @@ func (p *Panel) Handler() http.Handler {
|
|||||||
mux.HandleFunc("/panel/api/audits", p.protected(p.Runner.AuditsJSON))
|
mux.HandleFunc("/panel/api/audits", p.protected(p.Runner.AuditsJSON))
|
||||||
mux.HandleFunc("/panel/api/exec", p.protected(p.Runner.PanelExec))
|
mux.HandleFunc("/panel/api/exec", p.protected(p.Runner.PanelExec))
|
||||||
mux.HandleFunc("/panel/api/key/rotate", p.protected(p.Runner.RotateAPIKey))
|
mux.HandleFunc("/panel/api/key/rotate", p.protected(p.Runner.RotateAPIKey))
|
||||||
|
mux.HandleFunc("/panel/api/mcp/services", p.protected(p.mcpServices))
|
||||||
|
mux.HandleFunc("/panel/api/mcp/service", p.protected(p.deleteMCPService))
|
||||||
|
mux.HandleFunc("/panel/api/mcp/list", p.protected(p.listAgentMCP))
|
||||||
|
mux.HandleFunc("/panel/api/mcp/call", p.protected(p.callAgentMCP))
|
||||||
mux.HandleFunc("/", p.index)
|
mux.HandleFunc("/", p.index)
|
||||||
return mux
|
return mux
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type mcpPanelRunner interface {
|
||||||
|
MCPServicesJSON(http.ResponseWriter, *http.Request)
|
||||||
|
DeleteMCPService(http.ResponseWriter, *http.Request)
|
||||||
|
ListAgentMCPJSON(http.ResponseWriter, *http.Request)
|
||||||
|
CallAgentMCPJSON(http.ResponseWriter, *http.Request)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *Panel) mcpServices(w http.ResponseWriter, r *http.Request) {
|
||||||
|
if v, ok := p.Runner.(mcpPanelRunner); ok {
|
||||||
|
v.MCPServicesJSON(w, r)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
http.Error(w, "MCP API unavailable", 501)
|
||||||
|
}
|
||||||
|
func (p *Panel) deleteMCPService(w http.ResponseWriter, r *http.Request) {
|
||||||
|
if v, ok := p.Runner.(mcpPanelRunner); ok {
|
||||||
|
v.DeleteMCPService(w, r)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
http.Error(w, "MCP API unavailable", 501)
|
||||||
|
}
|
||||||
|
func (p *Panel) listAgentMCP(w http.ResponseWriter, r *http.Request) {
|
||||||
|
if v, ok := p.Runner.(mcpPanelRunner); ok {
|
||||||
|
v.ListAgentMCPJSON(w, r)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
http.Error(w, "MCP API unavailable", 501)
|
||||||
|
}
|
||||||
|
func (p *Panel) callAgentMCP(w http.ResponseWriter, r *http.Request) {
|
||||||
|
if v, ok := p.Runner.(mcpPanelRunner); ok {
|
||||||
|
v.CallAgentMCPJSON(w, r)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
http.Error(w, "MCP API unavailable", 501)
|
||||||
|
}
|
||||||
|
|
||||||
func (p *Panel) handleDevices(w http.ResponseWriter, r *http.Request) {
|
func (p *Panel) handleDevices(w http.ResponseWriter, r *http.Request) {
|
||||||
if r.Method == http.MethodDelete {
|
if r.Method == http.MethodDelete {
|
||||||
p.Runner.DeleteDevice(w, r)
|
p.Runner.DeleteDevice(w, r)
|
||||||
|
|||||||
+23
-19
@@ -17,25 +17,29 @@ header strong{font-size:18px}header span{margin-left:12px;color:var(--text-muted
|
|||||||
:root[data-theme="dark"] .svg-sun{display:block;color:#facc15}
|
:root[data-theme="dark"] .svg-sun{display:block;color:#facc15}
|
||||||
:root[data-theme="light"] .svg-moon{display:block;color:#475569}
|
:root[data-theme="light"] .svg-moon{display:block;color:#475569}
|
||||||
|
|
||||||
/* 设备卡片与画像升级 */
|
/* 设备卡片与画像升级 (方案 1: 经典主副标题分行) */
|
||||||
.device-card{min-width:0;padding:16px;border:1px solid var(--border);border-radius:12px;background:var(--bg-sub);display:flex;flex-direction:column;gap:10px}
|
.devices{display:grid;grid-template-columns:repeat(auto-fill,minmax(280px,1fr));gap:12px}
|
||||||
.device-head{display:flex;align-items:center;justify-content:space-between;gap:8px;min-width:0}
|
.device-card{min-width:0;padding:16px;border:1px solid var(--border);border-radius:12px;background:var(--bg-sub);display:flex;flex-direction:column;gap:12px}
|
||||||
.device-title{display:flex;align-items:center;gap:8px;min-width:0;flex-wrap:wrap}
|
|
||||||
.device-name{font-size:15px;color:var(--text-main);overflow-wrap:anywhere;word-break:break-all}
|
.v1-head{display:flex;flex-direction:column;gap:4px;min-width:0}
|
||||||
.role-badge{font-size:11px;padding:2px 6px;border-radius:5px;font-weight:600;letter-spacing:.5px;border:1px solid var(--border-strong)}
|
.v1-row1{display:flex;align-items:center;justify-content:space-between;gap:8px;min-width:0}
|
||||||
.role-control{background:rgba(59,130,246,.15);color:#60a5fa;border-color:rgba(59,130,246,.3)}
|
.v1-title{display:flex;align-items:center;gap:8px;min-width:0;flex:1}
|
||||||
.role-application{background:rgba(16,185,129,.15);color:#34d399;border-color:rgba(16,185,129,.3)}
|
.v1-name{font-size:16px;font-weight:700;color:var(--text-main);white-space:nowrap;overflow:hidden;text-overflow:ellipsis;display:block}
|
||||||
.role-database{background:rgba(245,158,11,.15);color:#fbbf24;border-color:rgba(245,158,11,.3)}
|
.v1-row2{font-size:12px;color:var(--text-muted);padding-left:16px;white-space:nowrap;overflow:hidden;text-overflow:ellipsis}
|
||||||
.btn-revoke{padding:3px 8px;font-size:12px;border-radius:6px;background:transparent;border:1px solid var(--border-strong);color:var(--dot-offline);cursor:pointer}
|
|
||||||
|
.btn-revoke{padding:2px 8px;font-size:11px;border-radius:5px;background:transparent;border:1px solid var(--border-strong);color:var(--dot-offline);cursor:pointer;flex-shrink:0}
|
||||||
.btn-revoke:hover{background:rgba(239,68,68,.15);border-color:var(--dot-offline)}
|
.btn-revoke:hover{background:rgba(239,68,68,.15);border-color:var(--dot-offline)}
|
||||||
.device-meta{font-size:12px;color:var(--text-muted);display:flex;align-items:center;gap:6px;flex-wrap:wrap}
|
|
||||||
|
|
||||||
/* 资源画像 2列紧凑栅格 */
|
/* 资源画像 2列紧凑栅格 */
|
||||||
.profile-grid{display:grid;grid-template-columns:repeat(2,minmax(0,1fr));gap:6px;padding:10px;background:var(--bg-card);border:1px solid var(--border);border-radius:8px;font-size:12px}
|
.profile-grid{display:grid;grid-template-columns:repeat(2,minmax(0,1fr));gap:6px 12px;padding:10px 12px;background:var(--bg-card);border:1px solid var(--border);border-radius:8px;font-size:12px}
|
||||||
.p-item{display:flex;justify-content:space-between;align-items:center;gap:6px}
|
.p-item{display:flex;justify-content:space-between;align-items:center;gap:6px;min-height:20px}
|
||||||
.p-label{color:var(--text-muted)}
|
.p-label{color:var(--text-muted);font-weight:500;font-size:11px}
|
||||||
.p-val{font-family:ui-monospace,monospace;font-weight:600;color:var(--text-main)}
|
.p-val{font-family:ui-monospace,SFMono-Regular,Menlo,Monaco,Consolas,monospace;font-weight:600;font-size:12px;color:var(--text-main);white-space:nowrap}
|
||||||
.p-docker{grid-column:1/-1;display:flex;justify-content:flex-end}
|
.badge-docker-text{padding:1px 6px;border-radius:4px;font-size:11px;font-weight:700}
|
||||||
.badge-docker{font-size:11px;padding:2px 8px;border-radius:4px;font-weight:600}
|
.badge-docker-text.yes{background:rgba(34,197,94,.15);color:var(--dot-online);border:1px solid rgba(34,197,94,.3)}
|
||||||
.badge-docker.yes{background:rgba(34,197,94,.15);color:var(--dot-online);border:1px solid rgba(34,197,94,.3)}
|
.badge-docker-text.no{background:rgba(148,163,184,.15);color:var(--text-muted);border:1px solid var(--border)}
|
||||||
.badge-docker.no{background:rgba(148,163,184,.15);color:var(--text-muted);border:1px solid var(--border)}
|
.device-card.offline{opacity:.62;filter:saturate(.45)}
|
||||||
|
.device-card.offline .btn-revoke{opacity:1}
|
||||||
|
|
||||||
|
.mcp-toolbar{display:flex;justify-content:space-between;align-items:center;gap:12px;margin-bottom:14px}.mcp-grid{display:grid;grid-template-columns:repeat(auto-fit,minmax(280px,1fr));gap:12px}.mcp-item{border:1px solid var(--line,#30343b);border-radius:10px;padding:14px;background:var(--card,#11151a)}.mcp-item h3{margin:0 0 6px;font-size:15px}.mcp-meta{font-size:12px;opacity:.72;line-height:1.7}.mcp-actions{display:flex;gap:8px;margin-top:10px}.mcp-form{display:grid;grid-template-columns:1fr 1fr;gap:10px}.mcp-form .full{grid-column:1/-1}.mcp-form label{display:flex;flex-direction:column;gap:5px;font-size:12px;opacity:.9}.mcp-form input,.mcp-form select,.mcp-form textarea{width:100%;box-sizing:border-box}.mcp-form textarea{min-height:72px;font-family:monospace}.mcp-nodes{display:flex;flex-wrap:wrap;gap:8px}.mcp-nodes label{flex-direction:row;align-items:center}.mcp-status{font-size:12px;margin-top:8px;min-height:18px}
|
||||||
|
@media(max-width:700px){.mcp-form{grid-template-columns:1fr}.mcp-form .full{grid-column:auto}.mcp-toolbar{align-items:flex-start;flex-direction:column}}
|
||||||
|
|||||||
+41
-20
@@ -32,46 +32,48 @@ async function removeDevice(name){
|
|||||||
}
|
}
|
||||||
|
|
||||||
async function loadDevices(){
|
async function loadDevices(){
|
||||||
let ds=await api("/devices");
|
let all=await api("/devices");
|
||||||
|
let ds=all.filter(d=>!d.is_local);
|
||||||
$("#devices").innerHTML=ds.map(d=>{
|
$("#devices").innerHTML=ds.map(d=>{
|
||||||
const p = d.profile || {};
|
const p = d.profile || {};
|
||||||
const role = p.role || (d.is_local ? "CONTROL" : "APPLICATION");
|
const role = p.role || "APPLICATION";
|
||||||
const isOnline = d.status === "online";
|
const isOnline = d.status === "online";
|
||||||
|
const statusText = isOnline ? "在线" : "离线";
|
||||||
|
const lastText = d.last_heartbeat ? new Date(d.last_heartbeat*1000).toLocaleString() : "";
|
||||||
|
const archOs = (d.arch || d.os) ? `${esc(d.arch||"")}/${esc(d.os||"")} · ` : "";
|
||||||
const cpuStr = p.cpu_cores ? `${p.cpu_cores}C · ${p.cpu_usage_percent ?? 0}%` : '-';
|
const cpuStr = p.cpu_cores ? `${p.cpu_cores}C · ${p.cpu_usage_percent ?? 0}%` : '-';
|
||||||
const memStr = p.mem_percent !== undefined ? `${p.mem_percent}%` + (p.mem_available ? ` (${p.mem_available})` : '') : '-';
|
const loadStr = p.load_1m !== undefined ? p.load_1m.toFixed(2) : '-';
|
||||||
const swapStr = p.swap_percent !== undefined ? `${p.swap_percent}%` : '-';
|
const memStr = p.mem_available ? `${p.mem_available} (${p.mem_percent}%)` : (p.mem_percent !== undefined ? `${p.mem_percent}%` : '-');
|
||||||
const diskStr = p.disk_percent !== undefined ? `${p.disk_percent}%` : '-';
|
const diskStr = p.disk_free ? `${p.disk_free} (${p.disk_percent}%)` : (p.disk_percent !== undefined ? `${p.disk_percent}%` : '-');
|
||||||
const psiStr = p.io_psi !== undefined ? p.io_psi.toFixed(2) : '-';
|
const psiStr = p.io_psi !== undefined ? p.io_psi.toFixed(2) : '-';
|
||||||
const dockerBadge = p.docker ? '<span class="badge-docker yes">Docker YES</span>' : '<span class="badge-docker no">Docker NO</span>';
|
const dockerVal = p.docker ? 'YES' : 'NO';
|
||||||
|
const dockerClass = p.docker ? 'yes' : 'no';
|
||||||
|
|
||||||
return `
|
return `
|
||||||
<div class="device-card ${isOnline ? "online" : "offline"}">
|
<div class="device-card ${isOnline ? "online" : "offline"}">
|
||||||
<div class="device-head">
|
<div class="v1-head">
|
||||||
<div class="device-title">
|
<div class="v1-row1">
|
||||||
|
<div class="v1-title">
|
||||||
<span class="dot"></span>
|
<span class="dot"></span>
|
||||||
<strong class="device-name">${esc(d.name)}</strong>
|
<strong class="v1-name" title="${esc(d.name)}">${esc(d.name)}</strong>
|
||||||
<span class="role-badge role-${role.toLowerCase()}">${esc(role)}</span>
|
|
||||||
</div>
|
</div>
|
||||||
${d.is_local ? '' : `<button class="btn-revoke" onclick="removeDevice('${esc(d.name)}')">注销</button>`}
|
<button class="btn-revoke" onclick="removeDevice('${esc(d.name)}')">注销</button>
|
||||||
</div>
|
</div>
|
||||||
<div class="device-meta">
|
<div class="v1-row2">${archOs}${statusText}${isOnline ? "" : (lastText ? " · 最后活跃: "+lastText : "")}</div>
|
||||||
<span>${esc(d.status)}</span> ·
|
|
||||||
<span>${d.is_local ? "local" : "remote"}</span> ·
|
|
||||||
<span>${esc(d.arch||"")}/${esc(d.os||"")}</span>
|
|
||||||
</div>
|
</div>
|
||||||
<div class="profile-grid">
|
<div class="profile-grid">
|
||||||
<div class="p-item"><span class="p-label">CPU</span><span class="p-val">${cpuStr}</span></div>
|
<div class="p-item"><span class="p-label">CPU</span><span class="p-val">${cpuStr}</span></div>
|
||||||
<div class="p-item"><span class="p-label">Mem</span><span class="p-val">${memStr}</span></div>
|
<div class="p-item"><span class="p-label">LOAD</span><span class="p-val">${loadStr}</span></div>
|
||||||
<div class="p-item"><span class="p-label">Swap</span><span class="p-val">${swapStr}</span></div>
|
<div class="p-item"><span class="p-label">MEM</span><span class="p-val">${memStr}</span></div>
|
||||||
<div class="p-item"><span class="p-label">Disk</span><span class="p-val">${diskStr}</span></div>
|
<div class="p-item"><span class="p-label">DISK</span><span class="p-val">${diskStr}</span></div>
|
||||||
<div class="p-item"><span class="p-label">IO PSI</span><span class="p-val">${psiStr}</span></div>
|
<div class="p-item"><span class="p-label">IO PSI</span><span class="p-val">${psiStr}</span></div>
|
||||||
<div class="p-item p-docker">${dockerBadge}</div>
|
<div class="p-item"><span class="p-label">DOCKER</span><span class="p-val badge-docker-text ${dockerClass}">${dockerVal}</span></div>
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
`;
|
`;
|
||||||
}).join("");
|
}).join("");
|
||||||
|
|
||||||
$("#device").innerHTML=ds.map(d=>'<option>'+esc(d.name)+'</option>').join("");
|
$("#device").innerHTML=ds.map(d=>`<option value="${esc(d.name)}" ${d.status==="offline" ? "disabled" : ""}>${esc(d.name)}${d.status==="offline" ? " [离线]" : ""}</option>`).join("");
|
||||||
}
|
}
|
||||||
|
|
||||||
async function loadAudits(){
|
async function loadAudits(){
|
||||||
@@ -107,3 +109,22 @@ $("#logout").onclick=async()=>{
|
|||||||
location.href="/panel/login"
|
location.href="/panel/login"
|
||||||
};
|
};
|
||||||
loadDevices().then(loadAudits).catch(console.error);
|
loadDevices().then(loadAudits).catch(console.error);
|
||||||
|
|
||||||
|
let mcpEditingId="";
|
||||||
|
async function loadMCPServices(){
|
||||||
|
const services=await api("/mcp/services");
|
||||||
|
const devices=(await api("/devices")).filter(d=>!d.is_local);
|
||||||
|
$("#mcpNodes").innerHTML=devices.map(d=>`<label><input type="checkbox" value="${esc(d.name)}">${esc(d.name)}</label>`).join("");
|
||||||
|
$("#mcpList").innerHTML=services.map(s=>`<div class="mcp-item"><h3>${esc(s.name)} <small>${esc(s.id)}</small></h3><div class="mcp-meta">${esc(s.transport)} · ${s.scope==='custom'?'指定节点':'全部节点'} · ${s.enabled?'启用':'停用'}${s.url?' · '+esc(s.url):s.command?' · '+esc(s.command):''}</div><div class="mcp-actions"><button type="button" onclick="editMCP('${esc(s.id)}')">编辑</button><button type="button" onclick="deleteMCP('${esc(s.id)}')">删除</button></div></div>`).join("")||'<div class="mcp-meta">暂无 MCP 服务配置。</div>';
|
||||||
|
}
|
||||||
|
function resetMCPForm(){
|
||||||
|
mcpEditingId=""; $("#mcpEditor").hidden=false; $("#mcpId").disabled=false; ["mcpId","mcpName","mcpCommand","mcpUrl"].forEach(id=>$("#"+id).value=""); $("#mcpTransport").value="stdio"; $("#mcpEnabled").value="1"; $("#mcpArgs").value="[]"; $("#mcpHeaders").value="{}"; $("#mcpEnv").value="{}"; $("#mcpScope").value="all"; $("#mcpStatus").textContent=""; $("#mcpNodes").querySelectorAll("input").forEach(x=>x.checked=false);
|
||||||
|
}
|
||||||
|
async function editMCP(id){
|
||||||
|
const services=await api("/mcp/services"); const s=services.find(x=>x.id===id); if(!s)return; resetMCPForm(); mcpEditingId=id; $("#mcpId").value=s.id; $("#mcpId").disabled=true; $("#mcpName").value=s.name||""; $("#mcpTransport").value=s.transport; $("#mcpEnabled").value=s.enabled?"1":"0"; $("#mcpCommand").value=s.command||""; $("#mcpUrl").value=s.url||""; $("#mcpArgs").value=s.args_json||"[]"; $("#mcpHeaders").value=s.headers_json||"{}"; $("#mcpEnv").value=s.env_json||"{}"; $("#mcpScope").value=s.scope||"all"; let nodes=[];try{nodes=JSON.parse(s.target_nodes_json||"[]")}catch(e){};$("#mcpNodes").querySelectorAll("input").forEach(x=>x.checked=nodes.includes(x.value));
|
||||||
|
}
|
||||||
|
async function deleteMCP(id){if(!confirm(`确认删除 MCP 服务 [${id}]?`))return;try{await api("/mcp/service?id="+encodeURIComponent(id),{method:"DELETE"});await loadMCPServices()}catch(e){alert("删除失败: "+e.message)}}
|
||||||
|
$("#mcpNew").onclick=resetMCPForm; $("#mcpCancel").onclick=()=>$("#mcpEditor").hidden=true;
|
||||||
|
$("#mcpSave").onclick=async()=>{try{const scope=$("#mcpScope").value;const body={id:$("#mcpId").value.trim(),name:$("#mcpName").value.trim(),transport:$("#mcpTransport").value,command:$("#mcpCommand").value.trim(),url:$("#mcpUrl").value.trim(),scope,enabled:$("#mcpEnabled").value==="1",args:JSON.parse($("#mcpArgs").value||"[]"),headers:JSON.parse($("#mcpHeaders").value||"{}"),env:JSON.parse($("#mcpEnv").value||"{}"),target_nodes:[...$("#mcpNodes").querySelectorAll("input:checked")].map(x=>x.value)};if(!body.id||!body.name)throw Error("服务标识和显示名称不能为空");await api("/mcp/services",{method:"POST",headers:{"Content-Type":"application/json"},body:JSON.stringify(body)});$("#mcpStatus").textContent="已保存并向在线节点同步。";await loadMCPServices()}catch(e){$("#mcpStatus").textContent="保存失败: "+e.message}};
|
||||||
|
$("#mcpTransport").onchange=()=>{const stdio=$("#mcpTransport").value==="stdio";$("#mcpCommand").parentElement.style.display=stdio?"":"none";$("#mcpArgs").parentElement.style.display=stdio?"":"none";$("#mcpUrl").parentElement.style.display=stdio?"none":""};
|
||||||
|
loadMCPServices().catch(console.error);
|
||||||
|
|||||||
@@ -1 +1,2 @@
|
|||||||
<!doctype html><html lang="zh-CN"><head><meta charset="utf-8"><meta name="viewport" content="width=device-width,initial-scale=1"><script>(function(){try{var t=localStorage.getItem("vpc_theme");if(t!=="light"&&t!=="dark")t=matchMedia("(prefers-color-scheme:light)").matches?"light":"dark";document.documentElement.dataset.theme=t}catch(e){document.documentElement.dataset.theme="dark"}})();</script><title>VPS-Commander</title><link rel="stylesheet" href="/static/app.css"></head><body><header><div class="brand-wrap"><img src="/static/icon.png" alt="logo" class="brand-logo"><div><strong>VPS-Commander</strong><span>Private Infrastructure Console</span></div></div><div class="header-actions"><button id="themeToggle" type="button" class="btn-icon" title="切换主题" aria-label="切换主题"><svg class="theme-svg svg-sun" viewBox="0 0 24 24" width="18" height="18" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"><circle cx="12" cy="12" r="5"></circle><line x1="12" y1="1" x2="12" y2="3"></line><line x1="12" y1="21" x2="12" y2="23"></line><line x1="4.22" y1="4.22" x2="5.64" y2="5.64"></line><line x1="18.36" y1="18.36" x2="19.78" y2="19.78"></line><line x1="1" y1="12" x2="3" y2="12"></line><line x1="21" y1="12" x2="23" y2="12"></line><line x1="4.22" y1="19.78" x2="5.64" y2="18.36"></line><line x1="18.36" y1="5.64" x2="19.78" y2="4.22"></line></svg><svg class="theme-svg svg-moon" viewBox="0 0 24 24" width="18" height="18" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"><path d="M21 12.79A9 9 0 1 1 11.21 3 7 7 0 0 0 21 12.79z"></path></svg></button><button id="logout" type="button">退出</button></div></header><main><section class="grid"><article class="card wide"><div class="cardhead"><h2>设备看板</h2><button id="refresh" type="button">刷新</button></div><div id="devices" class="devices"></div></article><article class="card"><h2>Actions 向导</h2><p>当前 Hub 的 OpenAPI Schema:</p><div class="actions"><button id="copySchema" type="button">复制 Schema 地址</button><button id="copyPrompt" type="button">复制 System Prompt</button><button id="rotateKey" type="button">轮换 API Key</button></div><pre id="schemaHint">/openapi.json</pre><pre id="newKey"></pre></article><article class="card wide"><div class="cardhead"><h2>快速控制台</h2><select id="device"></select></div><textarea id="command" placeholder="输入 Shell 命令,例如 uname -a"></textarea><div><button id="run" type="button">执行</button></div><pre id="output"></pre></article><article class="card wide"><div class="cardhead"><h2>审计日志</h2><button id="auditRefresh" type="button">刷新</button></div><div class="tablewrap"><table><thead><tr><th>时间</th><th>设备</th><th>动作</th><th>命令/路径</th><th>退出码</th><th>耗时</th><th>来源</th></tr></thead><tbody id="audits"></tbody></table></div></article></section></main><script src="/static/app.js"></script></body></html>
|
<!doctype html><html lang="zh-CN"><head><meta charset="utf-8"><meta name="viewport" content="width=device-width,initial-scale=1"><script>(function(){try{var t=localStorage.getItem("vpc_theme");if(t!=="light"&&t!=="dark")t=matchMedia("(prefers-color-scheme:light)").matches?"light":"dark";document.documentElement.dataset.theme=t}catch(e){document.documentElement.dataset.theme="dark"}})();</script><title>VPS-Commander</title><link rel="stylesheet" href="/static/app.css"></head><body><header><div class="brand-wrap"><img src="/static/icon.png" alt="logo" class="brand-logo"><div><strong>VPS-Commander</strong><span>Private Infrastructure Console</span></div></div><div class="header-actions"><button id="themeToggle" type="button" class="btn-icon" title="切换主题" aria-label="切换主题"><svg class="theme-svg svg-sun" viewBox="0 0 24 24" width="18" height="18" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"><circle cx="12" cy="12" r="5"></circle><line x1="12" y1="1" x2="12" y2="3"></line><line x1="12" y1="21" x2="12" y2="23"></line><line x1="4.22" y1="4.22" x2="5.64" y2="5.64"></line><line x1="18.36" y1="18.36" x2="19.78" y2="19.78"></line><line x1="1" y1="12" x2="3" y2="12"></line><line x1="21" y1="12" x2="23" y2="12"></line><line x1="4.22" y1="19.78" x2="5.64" y2="18.36"></line><line x1="18.36" y1="5.64" x2="19.78" y2="4.22"></line></svg><svg class="theme-svg svg-moon" viewBox="0 0 24 24" width="18" height="18" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"><path d="M21 12.79A9 9 0 1 1 11.21 3 7 7 0 0 0 21 12.79z"></path></svg></button><button id="logout" type="button">退出</button></div></header><main><section class="grid"><article class="card wide"><div class="cardhead"><h2>设备看板</h2><button id="refresh" type="button">刷新</button></div><div id="devices" class="devices"></div></article><article class="card"><h2>Actions 向导</h2><p>当前 Hub 的 OpenAPI Schema:</p><div class="actions"><button id="copySchema" type="button">复制 Schema 地址</button><button id="copyPrompt" type="button">复制 System Prompt</button><button id="rotateKey" type="button">轮换 API Key</button></div><pre id="schemaHint">/openapi.json</pre><pre id="newKey"></pre></article><article class="card wide"><div class="mcp-toolbar"><h2>MCP 挂载</h2><button id="mcpNew" type="button">新增 MCP</button></div><div id="mcpList" class="mcp-grid"></div><div id="mcpEditor" hidden><div class="mcp-form"><label>服务标识<input id="mcpId" placeholder="docker"></label><label>显示名称<input id="mcpName" placeholder="Docker 容器管理"></label><label>传输类型<select id="mcpTransport"><option value="stdio">命令行 / stdio</option><option value="http">HTTP / Streamable HTTP</option><option value="sse">SSE</option></select></label><label>启用<select id="mcpEnabled"><option value="1">启用</option><option value="0">停用</option></select></label><label class="full">Command<input id="mcpCommand" placeholder="docker-mcp"></label><label class="full">Arguments(JSON 数组)<textarea id="mcpArgs">[]</textarea></label><label class="full">URL<input id="mcpUrl" placeholder="http://127.0.0.1:8080/mcp"></label><label class="full">Headers(JSON 对象,可选)<textarea id="mcpHeaders">{}</textarea></label><label class="full">环境变量(JSON 对象,可选)<textarea id="mcpEnv">{}</textarea></label><label>生效范围<select id="mcpScope"><option value="all">全部节点</option><option value="custom">指定节点</option></select></label><div><span>指定节点</span><div id="mcpNodes" class="mcp-nodes"></div></div></div><div class="mcp-actions"><button id="mcpSave" type="button">保存并热更新</button><button id="mcpCancel" type="button">取消</button></div><div id="mcpStatus" class="mcp-status"></div></div></article>
|
||||||
|
<article class="card wide"><div class="cardhead"><h2>快速控制台</h2><select id="device"></select></div><textarea id="command" placeholder="输入 Shell 命令,例如 uname -a"></textarea><div><button id="run" type="button">执行</button></div><pre id="output"></pre></article><article class="card wide"><div class="cardhead"><h2>审计日志</h2><button id="auditRefresh" type="button">刷新</button></div><div class="tablewrap"><table><thead><tr><th>时间</th><th>设备</th><th>动作</th><th>命令/路径</th><th>退出码</th><th>耗时</th><th>来源</th></tr></thead><tbody id="audits"></tbody></table></div></article></section></main><script src="/static/app.js"></script></body></html>
|
||||||
|
|||||||
+18
-19
@@ -17,25 +17,24 @@ header strong{font-size:18px}header span{margin-left:12px;color:var(--text-muted
|
|||||||
:root[data-theme="dark"] .svg-sun{display:block;color:#facc15}
|
:root[data-theme="dark"] .svg-sun{display:block;color:#facc15}
|
||||||
:root[data-theme="light"] .svg-moon{display:block;color:#475569}
|
:root[data-theme="light"] .svg-moon{display:block;color:#475569}
|
||||||
|
|
||||||
/* 设备卡片与画像升级 */
|
/* 设备卡片与画像升级 (方案 1: 经典主副标题分行) */
|
||||||
.device-card{min-width:0;padding:16px;border:1px solid var(--border);border-radius:12px;background:var(--bg-sub);display:flex;flex-direction:column;gap:10px}
|
.devices{display:grid;grid-template-columns:repeat(auto-fill,minmax(280px,1fr));gap:12px}
|
||||||
.device-head{display:flex;align-items:center;justify-content:space-between;gap:8px;min-width:0}
|
.device-card{min-width:0;padding:16px;border:1px solid var(--border);border-radius:12px;background:var(--bg-sub);display:flex;flex-direction:column;gap:12px}
|
||||||
.device-title{display:flex;align-items:center;gap:8px;min-width:0;flex-wrap:wrap}
|
|
||||||
.device-name{font-size:15px;color:var(--text-main);overflow-wrap:anywhere;word-break:break-all}
|
.v1-head{display:flex;flex-direction:column;gap:4px;min-width:0}
|
||||||
.role-badge{font-size:11px;padding:2px 6px;border-radius:5px;font-weight:600;letter-spacing:.5px;border:1px solid var(--border-strong)}
|
.v1-row1{display:flex;align-items:center;justify-content:space-between;gap:8px;min-width:0}
|
||||||
.role-control{background:rgba(59,130,246,.15);color:#60a5fa;border-color:rgba(59,130,246,.3)}
|
.v1-title{display:flex;align-items:center;gap:8px;min-width:0;flex:1}
|
||||||
.role-application{background:rgba(16,185,129,.15);color:#34d399;border-color:rgba(16,185,129,.3)}
|
.v1-name{font-size:16px;font-weight:700;color:var(--text-main);white-space:nowrap;overflow:hidden;text-overflow:ellipsis;display:block}
|
||||||
.role-database{background:rgba(245,158,11,.15);color:#fbbf24;border-color:rgba(245,158,11,.3)}
|
.v1-row2{font-size:12px;color:var(--text-muted);padding-left:16px;white-space:nowrap;overflow:hidden;text-overflow:ellipsis}
|
||||||
.btn-revoke{padding:3px 8px;font-size:12px;border-radius:6px;background:transparent;border:1px solid var(--border-strong);color:var(--dot-offline);cursor:pointer}
|
|
||||||
|
.btn-revoke{padding:2px 8px;font-size:11px;border-radius:5px;background:transparent;border:1px solid var(--border-strong);color:var(--dot-offline);cursor:pointer;flex-shrink:0}
|
||||||
.btn-revoke:hover{background:rgba(239,68,68,.15);border-color:var(--dot-offline)}
|
.btn-revoke:hover{background:rgba(239,68,68,.15);border-color:var(--dot-offline)}
|
||||||
.device-meta{font-size:12px;color:var(--text-muted);display:flex;align-items:center;gap:6px;flex-wrap:wrap}
|
|
||||||
|
|
||||||
/* 资源画像 2列紧凑栅格 */
|
/* 资源画像 2列紧凑栅格 */
|
||||||
.profile-grid{display:grid;grid-template-columns:repeat(2,minmax(0,1fr));gap:6px;padding:10px;background:var(--bg-card);border:1px solid var(--border);border-radius:8px;font-size:12px}
|
.profile-grid{display:grid;grid-template-columns:repeat(2,minmax(0,1fr));gap:6px 12px;padding:10px 12px;background:var(--bg-card);border:1px solid var(--border);border-radius:8px;font-size:12px}
|
||||||
.p-item{display:flex;justify-content:space-between;align-items:center;gap:6px}
|
.p-item{display:flex;justify-content:space-between;align-items:center;gap:6px;min-height:20px}
|
||||||
.p-label{color:var(--text-muted)}
|
.p-label{color:var(--text-muted);font-weight:500;font-size:11px}
|
||||||
.p-val{font-family:ui-monospace,monospace;font-weight:600;color:var(--text-main)}
|
.p-val{font-family:ui-monospace,SFMono-Regular,Menlo,Monaco,Consolas,monospace;font-weight:600;font-size:12px;color:var(--text-main);white-space:nowrap}
|
||||||
.p-docker{grid-column:1/-1;display:flex;justify-content:flex-end}
|
.badge-docker-text{padding:1px 6px;border-radius:4px;font-size:11px;font-weight:700}
|
||||||
.badge-docker{font-size:11px;padding:2px 8px;border-radius:4px;font-weight:600}
|
.badge-docker-text.yes{background:rgba(34,197,94,.15);color:var(--dot-online);border:1px solid rgba(34,197,94,.3)}
|
||||||
.badge-docker.yes{background:rgba(34,197,94,.15);color:var(--dot-online);border:1px solid rgba(34,197,94,.3)}
|
.badge-docker-text.no{background:rgba(148,163,184,.15);color:var(--text-muted);border:1px solid var(--border)}
|
||||||
.badge-docker.no{background:rgba(148,163,184,.15);color:var(--text-muted);border:1px solid var(--border)}
|
|
||||||
+20
-19
@@ -32,40 +32,41 @@ async function removeDevice(name){
|
|||||||
}
|
}
|
||||||
|
|
||||||
async function loadDevices(){
|
async function loadDevices(){
|
||||||
let ds=await api("/devices");
|
let all=await api("/devices");
|
||||||
|
let ds=all.filter(d=>!d.is_local);
|
||||||
$("#devices").innerHTML=ds.map(d=>{
|
$("#devices").innerHTML=ds.map(d=>{
|
||||||
const p = d.profile || {};
|
const p = d.profile || {};
|
||||||
const role = p.role || (d.is_local ? "CONTROL" : "APPLICATION");
|
const role = p.role || "APPLICATION";
|
||||||
const isOnline = d.status === "online";
|
const isOnline = d.status === "online";
|
||||||
|
const statusText = isOnline ? "在线" : "离线";
|
||||||
|
const archOs = (d.arch || d.os) ? `${esc(d.arch||"")}/${esc(d.os||"")} · ` : "";
|
||||||
const cpuStr = p.cpu_cores ? `${p.cpu_cores}C · ${p.cpu_usage_percent ?? 0}%` : '-';
|
const cpuStr = p.cpu_cores ? `${p.cpu_cores}C · ${p.cpu_usage_percent ?? 0}%` : '-';
|
||||||
const memStr = p.mem_percent !== undefined ? `${p.mem_percent}%` + (p.mem_available ? ` (${p.mem_available})` : '') : '-';
|
const loadStr = p.load_1m !== undefined ? p.load_1m.toFixed(2) : '-';
|
||||||
const swapStr = p.swap_percent !== undefined ? `${p.swap_percent}%` : '-';
|
const memStr = p.mem_available ? `${p.mem_available} (${p.mem_percent}%)` : (p.mem_percent !== undefined ? `${p.mem_percent}%` : '-');
|
||||||
const diskStr = p.disk_percent !== undefined ? `${p.disk_percent}%` : '-';
|
const diskStr = p.disk_free ? `${p.disk_free} (${p.disk_percent}%)` : (p.disk_percent !== undefined ? `${p.disk_percent}%` : '-');
|
||||||
const psiStr = p.io_psi !== undefined ? p.io_psi.toFixed(2) : '-';
|
const psiStr = p.io_psi !== undefined ? p.io_psi.toFixed(2) : '-';
|
||||||
const dockerBadge = p.docker ? '<span class="badge-docker yes">Docker YES</span>' : '<span class="badge-docker no">Docker NO</span>';
|
const dockerVal = p.docker ? 'YES' : 'NO';
|
||||||
|
const dockerClass = p.docker ? 'yes' : 'no';
|
||||||
|
|
||||||
return `
|
return `
|
||||||
<div class="device-card ${isOnline ? "online" : "offline"}">
|
<div class="device-card ${isOnline ? "online" : "offline"}">
|
||||||
<div class="device-head">
|
<div class="v1-head">
|
||||||
<div class="device-title">
|
<div class="v1-row1">
|
||||||
|
<div class="v1-title">
|
||||||
<span class="dot"></span>
|
<span class="dot"></span>
|
||||||
<strong class="device-name">${esc(d.name)}</strong>
|
<strong class="v1-name" title="${esc(d.name)}">${esc(d.name)}</strong>
|
||||||
<span class="role-badge role-${role.toLowerCase()}">${esc(role)}</span>
|
|
||||||
</div>
|
</div>
|
||||||
${d.is_local ? '' : `<button class="btn-revoke" onclick="removeDevice('${esc(d.name)}')">注销</button>`}
|
<button class="btn-revoke" onclick="removeDevice('${esc(d.name)}')">注销</button>
|
||||||
</div>
|
</div>
|
||||||
<div class="device-meta">
|
<div class="v1-row2">${archOs}${statusText}</div>
|
||||||
<span>${esc(d.status)}</span> ·
|
|
||||||
<span>${d.is_local ? "local" : "remote"}</span> ·
|
|
||||||
<span>${esc(d.arch||"")}/${esc(d.os||"")}</span>
|
|
||||||
</div>
|
</div>
|
||||||
<div class="profile-grid">
|
<div class="profile-grid">
|
||||||
<div class="p-item"><span class="p-label">CPU</span><span class="p-val">${cpuStr}</span></div>
|
<div class="p-item"><span class="p-label">CPU</span><span class="p-val">${cpuStr}</span></div>
|
||||||
<div class="p-item"><span class="p-label">Mem</span><span class="p-val">${memStr}</span></div>
|
<div class="p-item"><span class="p-label">LOAD</span><span class="p-val">${loadStr}</span></div>
|
||||||
<div class="p-item"><span class="p-label">Swap</span><span class="p-val">${swapStr}</span></div>
|
<div class="p-item"><span class="p-label">MEM</span><span class="p-val">${memStr}</span></div>
|
||||||
<div class="p-item"><span class="p-label">Disk</span><span class="p-val">${diskStr}</span></div>
|
<div class="p-item"><span class="p-label">DISK</span><span class="p-val">${diskStr}</span></div>
|
||||||
<div class="p-item"><span class="p-label">IO PSI</span><span class="p-val">${psiStr}</span></div>
|
<div class="p-item"><span class="p-label">IO PSI</span><span class="p-val">${psiStr}</span></div>
|
||||||
<div class="p-item p-docker">${dockerBadge}</div>
|
<div class="p-item"><span class="p-label">DOCKER</span><span class="p-val badge-docker-text ${dockerClass}">${dockerVal}</span></div>
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
`;
|
`;
|
||||||
|
|||||||
Reference in New Issue
Block a user