Add create command structure and parsing
This commit is contained in:
parent
bd976c1791
commit
f9af813d49
24
pcsdk/pcsdk.go
Normal file
24
pcsdk/pcsdk.go
Normal file
@ -0,0 +1,24 @@
|
||||
package pcsdk
|
||||
|
||||
import (
|
||||
"net/netip"
|
||||
|
||||
"github.com/google/uuid"
|
||||
"github.com/thefeli73/polemos/state"
|
||||
)
|
||||
|
||||
type ProxyCommandCreate struct {
|
||||
Command CommandCreate `json:"create"`
|
||||
// signature Signature
|
||||
}
|
||||
|
||||
type CommandCreate struct {
|
||||
IncomingPort uint16 `json:"incoming_port"`
|
||||
DestinationPort uint16 `json:"destination_port"`
|
||||
DestinationIP netip.Addr `json:"destination_ip"`
|
||||
Id string `json:"id"`
|
||||
}
|
||||
|
||||
func NewCommandCreate(iport uint16, oport uint16, oip netip.Addr, id state.CustomUUID) CommandCreate {
|
||||
return CommandCreate{iport, oport, oip, uuid.UUID.String(uuid.UUID(id))}
|
||||
}
|
29
pcsdk/pcsdk_test.go
Normal file
29
pcsdk/pcsdk_test.go
Normal file
@ -0,0 +1,29 @@
|
||||
package pcsdk
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"net/netip"
|
||||
"testing"
|
||||
|
||||
"github.com/google/uuid"
|
||||
"github.com/thefeli73/polemos/state"
|
||||
)
|
||||
|
||||
func TestCommandCreateJsonParse(t *testing.T) {
|
||||
ip, _ := netip.ParseAddr("127.0.0.99")
|
||||
uuid, _ := uuid.Parse("87e79cbc-6df6-4462-8412-85d6c473e3b1")
|
||||
c := NewCommandCreate(5555, 6666, ip, state.CustomUUID(uuid))
|
||||
m := ProxyCommandCreate{c}
|
||||
msg, err := json.Marshal(m)
|
||||
if err != nil {
|
||||
t.Fatalf(`%q`, err)
|
||||
}
|
||||
|
||||
expected := "{\"create\":{\"incoming_port\":5555,\"destination_port\":6666,\"destination_ip\":\"127.0.0.99\",\"id\":\"87e79cbc-6df6-4462-8412-85d6c473e3b1\"}}"
|
||||
if string(msg) != expected {
|
||||
t.Fatalf(
|
||||
"\nExpected:\t %q\nGot:\t\t %q\n", expected, msg)
|
||||
}
|
||||
}
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user