From 4f37d5de48cad23ed04eb552bae96ddfe768a354 Mon Sep 17 00:00:00 2001
From: ChillerDragon <ChillerDragon@gmail.com>
Date: Fri, 21 Jun 2024 10:40:49 +0800
Subject: [PATCH] Fix nil result and add ip port cli args

---
 README.md    |  2 +-
 teeworlds.go | 27 +++++++++++++++++++++++----
 2 files changed, 24 insertions(+), 5 deletions(-)

diff --git a/README.md b/README.md
index 46bd2ec..69254d4 100644
--- a/README.md
+++ b/README.md
@@ -8,7 +8,7 @@ WORK IN PROGRESS
 
 ```
 go build
-./teeworlds
+./teeworlds 127.0.0.1 8303
 ```
 
 ## tests
diff --git a/teeworlds.go b/teeworlds.go
index b998304..bb8adc5 100644
--- a/teeworlds.go
+++ b/teeworlds.go
@@ -5,6 +5,7 @@ import (
 	"fmt"
 	"net"
 	"os"
+	"strconv"
 	"time"
 
 	"github.com/teeworlds-go/teeworlds/messages7"
@@ -16,8 +17,8 @@ const (
 	maxPacksize = 1400
 )
 
-func getConnection() (net.Conn, error) {
-	conn, err := net.Dial("udp", "127.0.0.1:8303")
+func getConnection(serverIp string, serverPort int) (net.Conn, error) {
+	conn, err := net.Dial("udp", fmt.Sprintf("%s:%d", serverIp, serverPort))
 	if err != nil {
 		fmt.Printf("Some error %v", err)
 	}
@@ -45,7 +46,25 @@ func readNetwork(ch chan<- []byte, conn net.Conn) {
 func main() {
 	ch := make(chan []byte, maxPacksize)
 
-	conn, err := getConnection()
+	serverIp := "127.0.0.1"
+	serverPort := 8303
+
+	if len(os.Args) > 1 {
+		if os.Args[1][0] == '-' {
+			fmt.Println("usage: ./teeworlds [serverIp] [serverPort]")
+			os.Exit(1)
+		}
+		serverIp = os.Args[1]
+	}
+	if len(os.Args) > 2 {
+		var err error
+		serverPort, err = strconv.Atoi(os.Args[2])
+		if err != nil {
+			panic(err)
+		}
+	}
+
+	conn, err := getConnection(serverIp, serverPort)
 	if err != nil {
 		fmt.Printf("error connecting %v\n", err)
 		os.Exit(1)
@@ -71,7 +90,7 @@ func main() {
 			if err != nil {
 				panic(err)
 			}
-			if result.Response != nil {
+			if result != nil && result.Response != nil {
 
 				// example of inspecting incoming trafic
 				for i, msg := range result.Packet.Messages {