Origin target ip address validation

internal/ent/schema/origin.go

@@ -10,6 +10,8 @@ import (
 
 	"go.infratographer.com/x/entx"
 	"go.infratographer.com/x/gidx"
+
+	"go.infratographer.com/load-balancer-api/internal/ent/schema/validations"
 )
 
 // Origin holds the schema definition for the Origin entity.
@@ -39,6 +41,7 @@ func (Origin) Fields() []ent.Field {
 			),
 		field.String("target").
 			NotEmpty().
+			Validate(validations.IPAddress).
 			// Comment("origin address").
 			Annotations(
 				entgql.OrderField("target"),

internal/ent/schema/validations/doc.go

@@ -0,0 +1,2 @@
+// Package validations contains validation functions for ent fields
+package validations

internal/ent/schema/validations/errors.go

@@ -0,0 +1,6 @@
+package validations
+
+import "errors"
+
+// ErrInvalidIPAddress is returned when the given string is not a valid IP address
+var ErrInvalidIPAddress = errors.New("invalid ip address")

internal/ent/schema/validations/validations.go

@@ -0,0 +1,15 @@
+// Package validations contains validation functions for ent fields
+package validations
+
+import (
+	"net"
+)
+
+// IPAddress validates if the given string is a valid IP address
+func IPAddress(ip string) error {
+	if net.ParseIP(ip) == nil {
+		return ErrInvalidIPAddress
+	}
+
+	return nil
+}

internal/graphapi/origin_test.go

@@ -141,6 +141,17 @@ func TestMutate_OriginCreate(t *testing.T) {
 				Active:     false,
 			},
 		},
+		{
+			TestName: "invalid target ip",
+			Input: graphclient.CreateLoadBalancerOriginInput{
+				Name:       "original",
+				Target:     "not a valid target ip",
+				PortNumber: 22,
+				PoolID:     pool1.ID,
+				Active:     newBool(false),
+			},
+			errorMsg: "invalid ip address",
+		},
 	}
 
 	for _, tt := range testCases {