feat: hspnet cli
This commit is contained in:
165
cmd/hspnet.go
Normal file
165
cmd/hspnet.go
Normal file
@ -0,0 +1,165 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"flag"
|
||||
"fmt"
|
||||
"net"
|
||||
"os"
|
||||
"os/signal"
|
||||
"strings"
|
||||
"syscall"
|
||||
|
||||
"github.com/LandaMm/hsp-go/hsp"
|
||||
"github.com/LandaMm/hsp-go/hsp/client"
|
||||
"github.com/LandaMm/hsp-go/hsp/server"
|
||||
"github.com/chzyer/readline"
|
||||
)
|
||||
|
||||
func Index(req *hsp.Request) *hsp.Response {
|
||||
pkt := req.GetRawPacket()
|
||||
|
||||
fmt.Printf("%s %s\n", strings.ToUpper(req.GetRequestKind()), req.GetRoute())
|
||||
fmt.Println("Headers:")
|
||||
|
||||
for k, v := range pkt.Headers {
|
||||
fmt.Printf("\t%s: %s\n", k, v)
|
||||
}
|
||||
|
||||
df, err := req.GetDataFormat()
|
||||
if err != nil {
|
||||
h, ok := req.GetHeader(hsp.H_DATA_FORMAT)
|
||||
if !ok {
|
||||
fmt.Println("ERR: Data-Format header is not present")
|
||||
return hsp.NewErrorResponse(err)
|
||||
}
|
||||
fmt.Printf("Invalid Data Format: %s\n", h)
|
||||
return hsp.NewErrorResponse(err)
|
||||
}
|
||||
|
||||
fmt.Printf("Payload (%s):\n", df.String())
|
||||
|
||||
switch df.Format {
|
||||
case "text":
|
||||
body, err := req.ExtractText()
|
||||
if err != nil {
|
||||
fmt.Println("ERR: Failed to extract text from payload:", err)
|
||||
return hsp.NewErrorResponse(err)
|
||||
}
|
||||
|
||||
fmt.Println(body)
|
||||
break
|
||||
case "json":
|
||||
var out map[string]any
|
||||
err := req.ExtractJson(&out)
|
||||
if err != nil {
|
||||
fmt.Println("ERR: Failed to extract json from payload:", err)
|
||||
return hsp.NewErrorResponse(err)
|
||||
}
|
||||
|
||||
fmt.Println(json.MarshalIndent(out, "", " "))
|
||||
break
|
||||
case "bytes":
|
||||
bts, err := req.ExtractBytes()
|
||||
if err != nil {
|
||||
fmt.Println("ERR: Failed to extract bytes from payload:", err)
|
||||
return hsp.NewErrorResponse(err)
|
||||
}
|
||||
|
||||
fmt.Println(bts)
|
||||
break;
|
||||
default:
|
||||
fmt.Printf("ERR: Unsupported data format: %s\n", df.String())
|
||||
}
|
||||
|
||||
return hsp.NewStatusResponse(hsp.STATUS_SUCCESS)
|
||||
}
|
||||
|
||||
func StartServer(addr *hsp.Adddress) {
|
||||
srv := server.NewServer(*addr)
|
||||
fmt.Printf("Server created on address: %s\n", srv.Addr.String())
|
||||
|
||||
handler := make(chan net.Conn, 1)
|
||||
|
||||
router := server.NewRouter()
|
||||
|
||||
router.AddRoute("/", Index)
|
||||
|
||||
srv.SetListener(handler)
|
||||
|
||||
go func() {
|
||||
for {
|
||||
conn := <-handler
|
||||
if err := router.Handle(conn); err != nil {
|
||||
fmt.Println("ERR: Couldn't handle connection:", err.Error())
|
||||
}
|
||||
}
|
||||
}()
|
||||
|
||||
sigs := make(chan os.Signal, 1)
|
||||
|
||||
go func() {
|
||||
s := <-sigs
|
||||
if s == syscall.SIGINT || s == syscall.SIGTERM {
|
||||
fmt.Println("Gracefully shutting down the server")
|
||||
if err := srv.Stop(); err != nil {
|
||||
fmt.Println("Failed to close the server:", err)
|
||||
}
|
||||
}
|
||||
}()
|
||||
|
||||
signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)
|
||||
|
||||
if err := srv.Start(); err != nil {
|
||||
fmt.Println("ERR: Failed to start server:", err)
|
||||
}
|
||||
}
|
||||
|
||||
func StartSession(addr *hsp.Adddress) {
|
||||
fmt.Println("Starting session on", addr.String())
|
||||
|
||||
rl, err := readline.New("> ")
|
||||
if err != nil {
|
||||
fmt.Println("ERR: Failed to open readline session:", err)
|
||||
}
|
||||
|
||||
defer rl.Close()
|
||||
|
||||
c := client.NewClient()
|
||||
|
||||
for {
|
||||
line, err := rl.Readline()
|
||||
if err != nil {
|
||||
break
|
||||
}
|
||||
|
||||
c.SendText(addr.String(), line)
|
||||
}
|
||||
}
|
||||
|
||||
func main() {
|
||||
var listening bool
|
||||
flag.BoolVar(&listening, "server", false, "start a simple server")
|
||||
|
||||
var host string
|
||||
var service string
|
||||
|
||||
flag.StringVar(&host, "host", "localhost", "specify target host (default: localhost)")
|
||||
flag.StringVar(&service, "port", "998", "specify target port (default: 998)")
|
||||
|
||||
flag.Parse()
|
||||
|
||||
a := fmt.Sprintf("%s:%s", host, service)
|
||||
addr, err := hsp.ParseAddress(a)
|
||||
if err != nil {
|
||||
fmt.Printf("ERR: Invalid address %s: %v\n", a, err)
|
||||
return
|
||||
}
|
||||
|
||||
if listening {
|
||||
StartServer(addr)
|
||||
return
|
||||
}
|
||||
|
||||
StartSession(addr)
|
||||
}
|
||||
166
cmd/main.go
166
cmd/main.go
@ -1,166 +0,0 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"log"
|
||||
"net"
|
||||
"os"
|
||||
"os/signal"
|
||||
"syscall"
|
||||
"time"
|
||||
|
||||
"github.com/LandaMm/hsp-go/hsp"
|
||||
"github.com/LandaMm/hsp-go/hsp/client"
|
||||
"github.com/LandaMm/hsp-go/hsp/server"
|
||||
)
|
||||
|
||||
func PingPongRoute(req *hsp.Request) *hsp.Response {
|
||||
log.Println("Ping pong request:", req.GetRawPacket())
|
||||
df, err := req.GetDataFormat()
|
||||
if err != nil {
|
||||
return hsp.NewErrorResponse(err)
|
||||
}
|
||||
|
||||
log.Println("Data format of the request:", df)
|
||||
|
||||
text, err := req.ExtractText()
|
||||
if err != nil {
|
||||
return hsp.NewErrorResponse(err)
|
||||
}
|
||||
|
||||
log.Println("Received text from req:", text)
|
||||
|
||||
// return hsp.NewStatusResponse(hsp.STATUS_SUCCESS)
|
||||
res, err := hsp.NewJsonResponse(map[string]any{
|
||||
"success": true,
|
||||
"message": text,
|
||||
})
|
||||
|
||||
if err != nil {
|
||||
return hsp.NewErrorResponse(err)
|
||||
}
|
||||
|
||||
return res
|
||||
}
|
||||
|
||||
func FileUploadStreamer(req *hsp.Request, c chan []byte) {
|
||||
log.Println("[MAIN] File Upload request:", req.GetRawPacket())
|
||||
|
||||
filename := "received.bin"
|
||||
|
||||
file, err := os.OpenFile(filename, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
|
||||
if err != nil {
|
||||
log.Fatalf("Failed to open file: %v", err)
|
||||
}
|
||||
|
||||
defer file.Close()
|
||||
|
||||
for buf := range c {
|
||||
if _, err := file.Write(buf); err != nil {
|
||||
log.Printf("Write error: %v", err)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func JsonMockRoute(req *hsp.Request) *hsp.Response {
|
||||
log.Println("[JSON MOCK] New request:", req.GetRawPacket())
|
||||
|
||||
var test map[string]any
|
||||
|
||||
if err := req.ExtractJson(&test); err != nil {
|
||||
return hsp.NewErrorResponse(err)
|
||||
}
|
||||
|
||||
log.Println("Json data from request:", test)
|
||||
|
||||
res, err := hsp.NewJsonResponse(test)
|
||||
if err != nil {
|
||||
return hsp.NewErrorResponse(err)
|
||||
}
|
||||
|
||||
return res
|
||||
}
|
||||
|
||||
func main() {
|
||||
addr, err := hsp.ParseAddress("127.0.0.1")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
srv := server.NewServer(*addr)
|
||||
fmt.Printf("Server created on address: %s\n", srv.Addr.String())
|
||||
|
||||
handler := make(chan net.Conn, 1)
|
||||
|
||||
router := server.NewRouter()
|
||||
|
||||
router.AddStreamer("/file-upload", FileUploadStreamer)
|
||||
router.AddRoute("/ping-pong", PingPongRoute)
|
||||
router.AddRoute("/mock-json", JsonMockRoute)
|
||||
|
||||
srv.SetListener(handler)
|
||||
|
||||
go func() {
|
||||
for {
|
||||
conn := <-handler
|
||||
if err := router.Handle(conn); err != nil {
|
||||
log.Println("[MAIN] Error handling connection:", err.Error())
|
||||
}
|
||||
}
|
||||
}()
|
||||
|
||||
sigs := make(chan os.Signal, 1)
|
||||
|
||||
go func() {
|
||||
s := <-sigs
|
||||
if s == syscall.SIGINT || s == syscall.SIGTERM {
|
||||
log.Println("Gracefully shutting down the server")
|
||||
if err := srv.Stop(); err != nil {
|
||||
log.Fatalln("Failed to close the server:", err)
|
||||
}
|
||||
}
|
||||
}()
|
||||
|
||||
signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)
|
||||
|
||||
go func() {
|
||||
time.Sleep(3 * time.Second)
|
||||
c := client.NewClient()
|
||||
rsp, err := c.SendText("localhost/ping-pong", "Hello, guys!")
|
||||
if err != nil {
|
||||
log.Fatalf("[CLIENT] Failed to received response from ping-pong: %s\n", err)
|
||||
}
|
||||
log.Println("[CLIENT] received response from remote server:", rsp)
|
||||
bts, err := json.Marshal(rsp)
|
||||
if err != nil {
|
||||
log.Fatalln("[CLIENT] Failed to serialize response from server")
|
||||
}
|
||||
|
||||
log.Println("[CLIENT] Serialized response from server:", string(bts))
|
||||
}()
|
||||
|
||||
go func() {
|
||||
time.Sleep(1 * time.Second)
|
||||
c := client.NewClient()
|
||||
rsp, err := c.SendJson("localhost/mock-json", map[string]any{
|
||||
"mock": true,
|
||||
"message": "Hello, everyone!",
|
||||
})
|
||||
|
||||
if err != nil {
|
||||
log.Fatalf("[CLIENT] Failed to received response from ping-pong: %s\n", err)
|
||||
}
|
||||
|
||||
log.Println("[CLIENT] received response from remote server:", rsp)
|
||||
bts, err := json.Marshal(rsp)
|
||||
if err != nil {
|
||||
log.Fatalln("[CLIENT] Failed to serialize response from server")
|
||||
}
|
||||
|
||||
log.Println("[CLIENT] Serialized response from server:", string(bts))
|
||||
}()
|
||||
|
||||
if err := srv.Start(); err != nil {
|
||||
log.Fatalln("Failed to start server:", err)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user