Change directory back to pcsdk

pcsdk/commands.go

@@ -0,0 +1,97 @@
+package pcsdk
+
+import (
+	"bytes"
+	"encoding/json"
+	"errors"
+	"fmt"
+	"io/ioutil"
+	"net/http"
+	"net/netip"
+
+	"github.com/google/uuid"
+	"github.com/thefeli73/polemos/state"
+)
+
+type ExecuteCommand interface {
+	Execute(string) error
+}
+
+type response struct {
+	message string `json:"message"`
+}
+
+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 (c ProxyCommandCreate) Execute(url string) error {
+	data, err := json.Marshal(c)
+	if err != nil {
+		return errors.New(fmt.Sprintf("could not serialize: %s\n", err))
+	}
+
+	requestURL := fmt.Sprintf("%s/command", url)
+
+	bodyReader := bytes.NewReader(data)
+
+	res, err := http.DefaultClient.Post(requestURL, "application/json", bodyReader)
+	if err != nil {
+		return errors.New(fmt.Sprintf("error making http request: %s\n", err))
+	}
+
+	fmt.Println(res)
+
+	body, err := ioutil.ReadAll(res.Body)
+	fmt.Println(string(body))
+	if err != nil {
+		return errors.New(fmt.Sprintf("error reading response: %s\n", err))
+	}
+
+	if res.StatusCode != 202 {
+		return errors.New(fmt.Sprintf("error processing command: (%d) %s\n", res.StatusCode, body))
+	} else {
+		return nil
+	}
+}
+
+func NewCommandCreate(iport uint16, oport uint16, oip netip.Addr, id state.CustomUUID) ProxyCommandCreate {
+	c := CommandCreate{iport, oport, oip, uuid.UUID.String(uuid.UUID(id))}
+	return ProxyCommandCreate{c}
+}
+
+type ProxyCommandModify struct {
+	Command CommandModify `json:"modify"`
+}
+
+type CommandModify struct {
+	DestinationPort uint16     `json:"destination_port"`
+	DestinationIP   netip.Addr `json:"destination_ip"`
+	Id              string     `json:"id"`
+}
+
+func NewCommandModify(oport uint16, oip netip.Addr, id state.CustomUUID) ProxyCommandModify {
+	c := CommandModify{oport, oip, uuid.UUID.String(uuid.UUID(id))}
+	return ProxyCommandModify{c}
+}
+
+type ProxyCommandDelete struct {
+	Command CommandDelete `json:"delete"`
+}
+
+type CommandDelete struct {
+	Id string `json:"id"`
+}
+
+func NewCommandDelete(id state.CustomUUID) ProxyCommandDelete {
+	c := CommandDelete{uuid.UUID.String(uuid.UUID(id))}
+	return ProxyCommandDelete{c}
+}

pcsdk/commands_test.go

@@ -0,0 +1,57 @@
+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")
+	m := NewCommandCreate(5555, 6666, ip, state.CustomUUID(uuid))
+	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)
+	}
+}
+
+func TestCommandModifyJsonParse(t *testing.T) {
+	ip, _ := netip.ParseAddr("127.0.0.99")
+	uuid, _ := uuid.Parse("87e79cbc-6df6-4462-8412-85d6c473e3b1")
+	m := NewCommandModify(8888, ip, state.CustomUUID(uuid))
+	msg, err := json.Marshal(m)
+	if err != nil {
+		t.Fatalf(`%q`, err)
+	}
+
+	expected := "{\"modify\":{\"destination_port\":8888,\"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)
+	}
+}
+
+func TestCommandDeleteJsonParse(t *testing.T) {
+	uuid, _ := uuid.Parse("87e79cbc-6df6-4462-8412-85d6c473e3b1")
+	m := NewCommandDelete(state.CustomUUID(uuid))
+	msg, err := json.Marshal(m)
+	if err != nil {
+		t.Fatalf(`%q`, err)
+	}
+
+	expected := "{\"delete\":{\"id\":\"87e79cbc-6df6-4462-8412-85d6c473e3b1\"}}"
+	if string(msg) != expected {
+		t.Fatalf(
+			"\nExpected:\t %q\nGot:\t\t %q\n", expected, msg)
+	}
+}