feat: 添加数据库资产、命令拦截器、授权资产等功能，修复GitHub Actions工作流

server/api/tools_api.go

@@ -0,0 +1,161 @@
+package api
+
+import (
+	"bufio"
+	"bytes"
+	"fmt"
+	"io"
+	"net"
+	"net/http"
+	"os/exec"
+	"runtime"
+	"strconv"
+	"strings"
+	"time"
+
+	"github.com/labstack/echo/v4"
+	"golang.org/x/text/encoding/simplifiedchinese"
+	"golang.org/x/text/transform"
+)
+
+type ToolsApi struct{}
+
+func (api ToolsApi) TcpingEndpoint(c echo.Context) error {
+	host := c.QueryParam("host")
+	port := c.QueryParam("port")
+	attemptsStr := c.QueryParam("attempts")
+
+	attempts := 4
+	if attemptsStr != "" {
+		attempts, _ = strconv.Atoi(attemptsStr)
+	}
+
+	c.Response().Header().Set("Content-Type", "text/event-stream")
+	c.Response().Header().Set("Cache-Control", "no-cache")
+	c.Response().Header().Set("Connection", "keep-alive")
+
+	flusher, ok := c.Response().Writer.(http.Flusher)
+	if !ok {
+		return echo.NewHTTPError(500, "Streaming unsupported")
+	}
+
+	for i := 0; i < attempts; i++ {
+		start := time.Now()
+		conn, err := net.DialTimeout("tcp", host+":"+port, 5*time.Second)
+		elapsed := time.Since(start)
+
+		var result string
+		if err != nil {
+			result = fmt.Sprintf("TCP ping %s:%s - Failed: %v", host, port, err)
+		} else {
+			conn.Close()
+			result = fmt.Sprintf("TCP ping %s:%s - Connected, time=%v", host, port, elapsed.Round(time.Millisecond))
+		}
+
+		fmt.Fprintf(c.Response(), "data: %s\n\n", result)
+		flusher.Flush()
+
+		if i < attempts-1 {
+			time.Sleep(1 * time.Second)
+		}
+	}
+
+	fmt.Fprintf(c.Response(), "event: done\ndata: completed\n\n")
+	flusher.Flush()
+
+	return nil
+}
+
+func (api ToolsApi) PingEndpoint(c echo.Context) error {
+	host := c.QueryParam("host")
+	attemptsStr := c.QueryParam("attempts")
+
+	attempts := 4
+	if attemptsStr != "" {
+		attempts, _ = strconv.Atoi(attemptsStr)
+	}
+
+	c.Response().Header().Set("Content-Type", "text/event-stream")
+	c.Response().Header().Set("Cache-Control", "no-cache")
+	c.Response().Header().Set("Connection", "keep-alive")
+
+	flusher, ok := c.Response().Writer.(http.Flusher)
+	if !ok {
+		return echo.NewHTTPError(500, "Streaming unsupported")
+	}
+
+	var cmd *exec.Cmd
+	if runtime.GOOS == "windows" {
+		cmd = exec.Command("ping", "-n", strconv.Itoa(attempts), host)
+	} else {
+		cmd = exec.Command("ping", "-c", strconv.Itoa(attempts), host)
+	}
+
+	stdout, err := cmd.StdoutPipe()
+	if err != nil {
+		fmt.Fprintf(c.Response(), "data: Error: %v\n\n", err)
+		flusher.Flush()
+		return nil
+	}
+
+	if err := cmd.Start(); err != nil {
+		fmt.Fprintf(c.Response(), "data: Error: %v\n\n", err)
+		flusher.Flush()
+		return nil
+	}
+
+	var reader io.Reader = stdout
+	if runtime.GOOS == "windows" {
+		reader = transform.NewReader(stdout, simplifiedchinese.GBK.NewDecoder())
+	}
+
+	scanner := bufio.NewScanner(reader)
+	for scanner.Scan() {
+		line := scanner.Text()
+		if strings.TrimSpace(line) != "" {
+			fmt.Fprintf(c.Response(), "data: %s\n\n", line)
+			flusher.Flush()
+		}
+	}
+
+	cmd.Wait()
+
+	fmt.Fprintf(c.Response(), "event: done\ndata: completed\n\n")
+	flusher.Flush()
+
+	return nil
+}
+
+func (api ToolsApi) TelnetEndpoint(c echo.Context) error {
+	host := c.QueryParam("host")
+	port := c.QueryParam("port")
+
+	c.Response().Header().Set("Content-Type", "text/event-stream")
+	c.Response().Header().Set("Cache-Control", "no-cache")
+	c.Response().Header().Set("Connection", "keep-alive")
+
+	flusher, ok := c.Response().Writer.(http.Flusher)
+	if !ok {
+		return echo.NewHTTPError(500, "Streaming unsupported")
+	}
+
+	start := time.Now()
+	conn, err := net.DialTimeout("tcp", host+":"+port, 10*time.Second)
+	elapsed := time.Since(start)
+
+	var result bytes.Buffer
+	if err != nil {
+		result.WriteString(fmt.Sprintf("Telnet %s:%s - Connection failed: %v", host, port, err))
+	} else {
+		conn.Close()
+		result.WriteString(fmt.Sprintf("Telnet %s:%s - Connected successfully in %v", host, port, elapsed.Round(time.Millisecond)))
+	}
+
+	fmt.Fprintf(c.Response(), "data: %s\n\n", result.String())
+	flusher.Flush()
+
+	fmt.Fprintf(c.Response(), "event: done\ndata: completed\n\n")
+	flusher.Flush()
+
+	return nil
+}