From 4afc73fb2a6ad0de380621b9b5194f77dc0e5442 Mon Sep 17 00:00:00 2001
From: LandaMm <qwer.009771@gmail.com>
Date: Fri, 18 Apr 2025 13:44:33 +0200
Subject: [PATCH] fix: correct import + ping pong route

---
 main.go | 38 ++++++++++++++++++++++++++++++++------
 1 file changed, 32 insertions(+), 6 deletions(-)

diff --git a/main.go b/main.go
index 52f1dc1..396f3d0 100644
--- a/main.go
+++ b/main.go
@@ -7,27 +7,42 @@ import (
 	"os"
 	"os/signal"
 	"syscall"
+	"time"
 
-	"github.com/LandaMm/hsp-go/server"
+	"github.com/LandaMm/hsp-go/hsp"
+	"github.com/LandaMm/hsp-go/hsp/client"
+	"github.com/LandaMm/hsp-go/hsp/server"
 )
 
-func FileUploadRoute(req *server.Request) *server.Response {
+func PingPongRoute(req *hsp.Request) *hsp.Response {
+	log.Println("Ping pong request:", req)
+	text, err := req.ExtractText()
+	if err != nil {
+		return hsp.NewStatusResponse(hsp.STATUS_INTERNALERR)
+	}
+
+	log.Println("Received text from req:", text)
+
+	return hsp.NewStatusResponse(hsp.STATUS_SUCCESS)
+}
+
+func FileUploadRoute(req *hsp.Request) *hsp.Response {
 	log.Println("[MAIN] File Upload request:", req)
 	bytes, err := req.ExtractBytes()
 	if err != nil {
-		return server.NewStatusResponse(server.STATUS_INTERNALERR)
+		return hsp.NewStatusResponse(hsp.STATUS_INTERNALERR)
 	}
 
 	filename := "received.bin"
 	err = os.WriteFile(filename, bytes, 0644)
 	if err != nil {
 		log.Fatalln("Failed to write packet payload into a file:", err)
-		return server.NewStatusResponse(server.STATUS_INTERNALERR)
+		return hsp.NewStatusResponse(hsp.STATUS_INTERNALERR)
 	}
 
 	log.Println("Received new request from client:", req.Conn().RemoteAddr().String())
 
-	res := server.NewTextResponse("Hello, World!")
+	res := hsp.NewTextResponse("Hello, World!")
 	res.AddHeader("filename", filename)
 
 	return res
@@ -42,6 +57,7 @@ func main() {
 	router := server.NewRouter()
 
 	router.AddRoute("/file-upload", FileUploadRoute)
+	router.AddRoute("/ping-pong", PingPongRoute)
 
 	srv.SetListener(handler)
 
@@ -68,7 +84,17 @@ func main() {
 
 	signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)
 
+	go func() {
+		time.Sleep(2 * time.Second)
+		c := client.NewClient()
+		rsp, err := c.SendText("localhost:3000/ping-pong", "Hello, guys!")
+		if err != nil {
+			log.Fatalf("Failed to received response from ping-pong: %s\n", err)
+		}
+		log.Println("[CLIENT] received response from remote server:", rsp)
+	}()
+
 	if err := srv.Start(); err != nil {
 		log.Fatalln("Failed to start server")
-	}
+	}	
 }