机器人功能还差发送nodeinfo

2026-06-12 19:24:40 +08:00
parent 0db2b181cc
commit 609462252c
10 changed files with 629 additions and 83 deletions
@@ -9,15 +9,17 @@ func TestBuildTextMessageServiceEnvelopeRoundTrip(t *testing.T) {
 	}

 	raw, err := BuildTextMessageServiceEnvelope(TextMessageBuildOptions{
-		FromNodeNum: 0x12345678,
-		ToNodeNum:   NodeNumBroadcast,
-		PacketID:    0x87654321,
-		ChannelID:   "LongFast",
-		GatewayID:   "!12345678",
-		Text:        "hello from bot",
-		PSK:         key,
-		Encrypt:     true,
-		ViaMQTT:     true,
+		PacketBuildOptions: PacketBuildOptions{
+			FromNodeNum: 0x12345678,
+			ToNodeNum:   NodeNumBroadcast,
+			PacketID:    0x87654321,
+			ChannelID:   "LongFast",
+			GatewayID:   "!12345678",
+			PSK:         key,
+			Encrypt:     true,
+			ViaMQTT:     true,
+		},
+		Text: "hello from bot",
 	})
 	if err != nil {
 		t.Fatalf("BuildTextMessageServiceEnvelope() error = %v", err)
@@ -51,15 +53,17 @@ func TestBuildTextMessageServiceEnvelopeDirectRoundTrip(t *testing.T) {
 	}

 	raw, err := BuildTextMessageServiceEnvelope(TextMessageBuildOptions{
-		FromNodeNum: 0x12345678,
-		ToNodeNum:   0x10203040,
-		PacketID:    0x11111111,
-		ChannelID:   "LongFast",
-		GatewayID:   "!12345678",
-		Text:        "direct hello",
-		PSK:         key,
-		Encrypt:     true,
-		ViaMQTT:     true,
+		PacketBuildOptions: PacketBuildOptions{
+			FromNodeNum: 0x12345678,
+			ToNodeNum:   0x10203040,
+			PacketID:    0x11111111,
+			ChannelID:   "LongFast",
+			GatewayID:   "!12345678",
+			PSK:         key,
+			Encrypt:     true,
+			ViaMQTT:     true,
+		},
+		Text: "direct hello",
 	})
 	if err != nil {
 		t.Fatalf("BuildTextMessageServiceEnvelope() error = %v", err)
@@ -80,6 +84,62 @@ func TestBuildTextMessageServiceEnvelopeDirectRoundTrip(t *testing.T) {
 	}
 }

+func TestBuildNodeInfoServiceEnvelopeRoundTrip(t *testing.T) {
+	key, err := ExpandPSK("AQ==")
+	if err != nil {
+		t.Fatalf("ExpandPSK() error = %v", err)
+	}
+
+	raw, err := BuildNodeInfoServiceEnvelope(NodeInfoBuildOptions{
+		PacketBuildOptions: PacketBuildOptions{
+			FromNodeNum: 0x12345678,
+			ToNodeNum:   NodeNumBroadcast,
+			PacketID:    0x22222222,
+			ChannelID:   "LongFast",
+			GatewayID:   "!12345678",
+			PSK:         key,
+			Encrypt:     true,
+			ViaMQTT:     true,
+		},
+		NodeID:     "!12345678",
+		LongName:   "MQTT Bot",
+		ShortName:  "BT",
+		HWModel:    255,
+		Role:       0,
+		IsLicensed: false,
+		PublicKey:  []byte{1, 2, 3},
+	})
+	if err != nil {
+		t.Fatalf("BuildNodeInfoServiceEnvelope() error = %v", err)
+	}
+
+	valid, _, record := MQTTPP("msh/2/e/LongFast/!12345678", raw, key, Options{})
+	if !valid {
+		t.Fatalf("MQTTPP() valid = false, record = %#v", record)
+	}
+	if record["type"] != "nodeinfo" {
+		t.Fatalf("record type = %v", record["type"])
+	}
+	if record["long_name"] != "MQTT Bot" {
+		t.Fatalf("long_name = %v", record["long_name"])
+	}
+	if record["short_name"] != "BT" {
+		t.Fatalf("short_name = %v", record["short_name"])
+	}
+	if record["hw_model"] != "PRIVATE_HW" {
+		t.Fatalf("hw_model = %v", record["hw_model"])
+	}
+	if record["role"] != "CLIENT" {
+		t.Fatalf("role = %v", record["role"])
+	}
+	if record["is_licensed"] != false {
+		t.Fatalf("is_licensed = %v", record["is_licensed"])
+	}
+	if record["public_key"] != "010203" {
+		t.Fatalf("public_key = %v", record["public_key"])
+	}
+}
+
 func TestParseNodeID(t *testing.T) {
 	num, err := ParseNodeID("!1234abcd")
 	if err != nil {