hatchet-dev · abelanger5 · Jun 19, 2024 · Jun 19, 2024 · Jun 19, 2024 · Jun 19, 2024
diff --git a/api-contracts/openapi/components/schemas/_index.yaml b/api-contracts/openapi/components/schemas/_index.yaml
@@ -78,6 +78,8 @@ Event:
   $ref: "./event.yaml#/Event"
 EventData:
   $ref: "./event.yaml#/EventData"
+CreateEventRequest:
+  $ref: "./event.yaml#/CreateEventRequest"
 EventWorkflowRunSummary:
   $ref: "./event.yaml#/EventWorkflowRunSummary"
 EventOrderByField:

diff --git a/api-contracts/openapi/components/schemas/event.yaml b/api-contracts/openapi/components/schemas/event.yaml
@@ -30,6 +30,21 @@ EventData:
   required:
     - data
 
+CreateEventRequest:
+  properties:
+    key:
+      type: string
+      description: The key for the event.
+    data:
+      type: object
+      description: The data for the event.
+    additionalMetadata:
+      type: object
+      description: Additional metadata for the event.
+  required:
+    - key
+    - data
+
 ReplayEventRequest:
   properties:
     eventIds:

diff --git a/api-contracts/openapi/paths/event/event.yaml b/api-contracts/openapi/paths/event/event.yaml
@@ -98,6 +98,55 @@ withTenant:
     summary: List events
     tags:
       - Event
+  post:
+    x-resources: ["tenant"]
+    description: Creates a new event.
+    operationId: event:create
+    parameters:
+      - description: The tenant id
+        in: path
+        name: tenant
+        required: true
+        schema:
+          type: string
+          format: uuid
+          minLength: 36
+          maxLength: 36
+    requestBody:
+      content:
+        application/json:
+          schema:
+            $ref: "../../components/schemas/_index.yaml#/CreateEventRequest"
+      description: The event to create
+      required: true
+    responses:
+      "200":
+        content:
+          application/json:
+            schema:
+              $ref: "../../components/schemas/_index.yaml#/Event"
+        description: Successfully created the event
+      "400":
+        content:
+          application/json:
+            schema:
+              $ref: "../../components/schemas/_index.yaml#/APIErrors"
+        description: A malformed or bad request
+      "403":
+        content:
+          application/json:
+            schema:
+              $ref: "../../components/schemas/_index.yaml#/APIErrors"
+        description: Forbidden
+      "429":
+        content:
+          application/json:
+            schema:
+              $ref: "../../components/schemas/_index.yaml#/APIErrors"
+        description: Resource limit exceeded
+    summary: Create event
+    tags:
+      - Event
 
 eventData:
   get:

diff --git a/api/v1/server/handlers/events/create.go b/api/v1/server/handlers/events/create.go
@@ -0,0 +1,57 @@
+package events
+
+import (
+	"encoding/json"
+
+	"github.com/labstack/echo/v4"
+
+	"github.com/hatchet-dev/hatchet/api/v1/server/oas/apierrors"
+	"github.com/hatchet-dev/hatchet/api/v1/server/oas/gen"
+	"github.com/hatchet-dev/hatchet/api/v1/server/oas/transformers"
+	"github.com/hatchet-dev/hatchet/pkg/repository/metered"
+	"github.com/hatchet-dev/hatchet/pkg/repository/prisma/db"
+	"github.com/hatchet-dev/hatchet/pkg/repository/prisma/sqlchelpers"
+)
+
+func (t *EventService) EventCreate(ctx echo.Context, request gen.EventCreateRequestObject) (gen.EventCreateResponseObject, error) {
+	tenant := ctx.Get("tenant").(*db.TenantModel)
+
+	// marshal the data object to bytes
+	dataBytes, err := json.Marshal(request.Body.Data)
+
+	if err != nil {
+		return nil, err
+	}
+
+	var additionalMetadata []byte
+
+	if request.Body.AdditionalMetadata != nil {
+		additionalMetadata, err = json.Marshal(request.Body.AdditionalMetadata)
+
+		if err != nil {
+			return nil, err
+		}
+	}
+
+	newEvent, err := t.config.Ingestor.IngestEvent(ctx.Request().Context(), tenant.ID, request.Body.Key, dataBytes, additionalMetadata)
+
+	if err != nil {
+		if err == metered.ErrResourceExhausted {
+			return gen.EventCreate429JSONResponse(
+				apierrors.NewAPIErrors("Event limit exceeded"),
+			), nil
+		}
+
+		return nil, err
+	}
+
+	dbNewEvent, err := t.config.APIRepository.Event().GetEventById(sqlchelpers.UUIDToStr(newEvent.ID))
+
+	if err != nil {
+		return nil, err
+	}
+
+	return gen.EventCreate200JSONResponse(
+		*transformers.ToEvent(dbNewEvent),
+	), nil
+}