first commit!

linux-desktop/build.sh

@@ -0,0 +1,2 @@
+pip install pyinstaller
+pyinstaller --onefile src/linux_udp_client.py

linux-desktop/src/linux_udp_client.py

@@ -0,0 +1,72 @@
+import gi
+gi.require_version("Gtk", "3.0")
+from gi.repository import Gtk, GLib
+
+import socket
+import json
+import threading
+import webbrowser
+
+UDP_PORT = 9876
+
+class UDPClient(Gtk.Window):
+    def __init__(self):
+        super().__init__(title="AI服务器发现工具")
+        self.set_border_width(20)
+        self.set_default_size(400, 150)
+
+        self.label = Gtk.Label(label="监听中... 请确保 AI 服务器在线")
+        self.add(self.label)
+
+        self.popup_shown = False  # 每次启动只弹一次
+
+        threading.Thread(target=self.listen_udp, daemon=True).start()
+
+    def listen_udp(self):
+        sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+        sock.bind(("", UDP_PORT))
+        print(f"Listening on UDP port {UDP_PORT}...")
+
+        while True:
+            data, addr = sock.recvfrom(1024)
+            try:
+                msg = json.loads(data.decode("utf-8"))
+                if msg.get("type") == "ai_server_announce":
+                    ip = msg.get("ip")
+                    name = msg.get("name", "Unknown")
+                    url = f"http://{ip}:80"  # 明确指定80端口
+
+                    if not self.popup_shown:
+                        self.popup_shown = True
+                        GLib.idle_add(self.update_ui, name, url)
+
+            except Exception as e:
+                print("Error parsing message:", e)
+
+    def update_ui(self, name, url):
+        self.label.set_text(f"发现设备：{name}\n{url}")
+        dialog = Gtk.MessageDialog(
+            transient_for=self,
+            flags=0,
+            message_type=Gtk.MessageType.QUESTION,
+            buttons=Gtk.ButtonsType.YES_NO,
+            text="是否访问 AI 服务器？",
+        )
+        dialog.format_secondary_text(url)
+        response = dialog.run()
+        dialog.destroy()
+
+        if response == Gtk.ResponseType.YES:
+            webbrowser.open(url)
+
+        return False
+
+def main():
+    app = UDPClient()
+    app.connect("destroy", Gtk.main_quit)
+    app.show_all()
+    Gtk.main()
+
+if __name__ == "__main__":
+    main()
+

server/broadcaster.go

@@ -0,0 +1,140 @@
+package main
+
+import (
+	"encoding/json"
+	"flag"
+	"fmt"
+	"net"
+	"os"
+	"os/exec"
+	"runtime"
+	"strings"
+	"time"
+)
+
+type BroadcastMessage struct {
+	Type string `json:"type"`
+	IP   string `json:"ip"`
+	Port int    `json:"port"`
+	Name string `json:"name"`
+}
+
+func getGatewayIP() (string, error) {
+	var cmd *exec.Cmd
+	if runtime.GOOS == "linux" {
+		cmd = exec.Command("sh", "-c", "ip route | grep default | awk '{print $3}'")
+	} else if runtime.GOOS == "darwin" {
+		cmd = exec.Command("sh", "-c", "netstat -rn | grep default | awk '{print $2}'")
+	} else {
+		return "", fmt.Errorf("unsupported OS")
+	}
+
+	out, err := cmd.Output()
+	if err != nil {
+		return "", err
+	}
+	return strings.TrimSpace(string(out)), nil
+}
+
+func ipInSameSubnet(ip1 net.IP, ip2 net.IP, mask net.IPMask) bool {
+	if mask == nil {
+		return false
+	}
+	net1 := ip1.Mask(mask)
+	net2 := ip2.Mask(mask)
+	return net1.Equal(net2)
+}
+
+func getLocalIP() string {
+	gatewayStr, err := getGatewayIP()
+	if err != nil {
+		fmt.Println("Failed to get gateway IP:", err)
+		return "127.0.0.1"
+	}
+	gateway := net.ParseIP(gatewayStr)
+	if gateway == nil {
+		fmt.Println("Invalid gateway IP")
+		return "127.0.0.1"
+	}
+
+	interfaces, err := net.Interfaces()
+	if err != nil {
+		fmt.Println("Failed to get network interfaces:", err)
+		return "127.0.0.1"
+	}
+
+	for _, iface := range interfaces {
+		if iface.Flags&net.FlagUp == 0 {
+			continue // interface down
+		}
+		addrs, err := iface.Addrs()
+		if err != nil {
+			continue
+		}
+		for _, addr := range addrs {
+			var ip net.IP
+			var mask net.IPMask
+
+			switch v := addr.(type) {
+			case *net.IPNet:
+				ip = v.IP
+				mask = v.Mask
+			case *net.IPAddr:
+				ip = v.IP
+			}
+
+			if ip == nil || ip.IsLoopback() || ip.To4() == nil {
+				continue
+			}
+
+			if ipInSameSubnet(ip, gateway, mask) {
+				return ip.String()
+			}
+		}
+	}
+	return "127.0.0.1"
+}
+
+func main() {
+	// 参数
+	var (
+		port     = flag.Int("port", 8000, "Device service port")
+		interval = flag.Int("interval", 2, "Broadcast interval in seconds")
+		name     = flag.String("name", "My-AI-Server", "Device name")
+	)
+	flag.Parse()
+
+	ip := getLocalIP()
+	if ip == "127.0.0.1" {
+		fmt.Println("Warning: failed to find suitable IP address, using loopback")
+	}
+
+	addr := net.UDPAddr{
+		IP:   net.IPv4bcast,
+		Port: 9876,
+	}
+	conn, err := net.DialUDP("udp4", nil, &addr)
+	if err != nil {
+		fmt.Println("Failed to dial UDP broadcast:", err)
+		os.Exit(1)
+	}
+	defer conn.Close()
+
+	for {
+		msg := BroadcastMessage{
+			Type: "ai_server_announce",
+			IP:   ip,
+			Port: *port,
+			Name: *name,
+		}
+		data, _ := json.Marshal(msg)
+		_, err := conn.Write(data)
+		if err != nil {
+			fmt.Println("Broadcast failed:", err)
+		} else {
+			fmt.Printf("Broadcasted: %s:%d (%s)\n", ip, *port, *name)
+		}
+		time.Sleep(time.Duration(*interval) * time.Second)
+	}
+}
+

server/build.sh

@@ -0,0 +1,4 @@
+# Linux/macOS
+go build -o broadcaster broadcaster.go
+./broadcaster -port=9876 -interval=2 -name="T1-GRT309024X2"
+