From f650f8880688a5b518cf734dcf39960feff10d95 Mon Sep 17 00:00:00 2001
From: Arthur <worldnomad@gmail.com>
Date: Tue, 31 Oct 2023 10:21:02 -0700
Subject: [PATCH] 14132 api / graphql

---
 netbox/extras/api/serializers.py | 29 +++++++++++++++++++++++------
 netbox/extras/api/urls.py        |  1 +
 netbox/extras/api/views.py       | 11 +++++++++++
 netbox/extras/graphql/types.py   | 10 +++++++++-
 4 files changed, 44 insertions(+), 7 deletions(-)

diff --git a/netbox/extras/api/serializers.py b/netbox/extras/api/serializers.py
index c1fad99ee..c5ee1faba 100644
--- a/netbox/extras/api/serializers.py
+++ b/netbox/extras/api/serializers.py
@@ -39,6 +39,7 @@ __all__ = (
     'CustomFieldSerializer',
     'CustomLinkSerializer',
     'DashboardSerializer',
+    'EventRuleSerializer',
     'ExportTemplateSerializer',
     'ImageAttachmentSerializer',
     'JournalEntrySerializer',
@@ -58,22 +59,38 @@ __all__ = (
 
 
 #
-# Webhooks
+# Event Rules
 #
 
-class WebhookSerializer(NetBoxModelSerializer):
-    url = serializers.HyperlinkedIdentityField(view_name='extras-api:webhook-detail')
+class EventRuleSerializer(NetBoxModelSerializer):
+    url = serializers.HyperlinkedIdentityField(view_name='extras-api:eventrule-detail')
     content_types = ContentTypeField(
         queryset=ContentType.objects.filter(FeatureQuery('webhooks').get_query()),
         many=True
     )
 
     class Meta:
-        model = Webhook
+        model = EventRule
         fields = [
             'id', 'url', 'display', 'content_types', 'name', 'type_create', 'type_update', 'type_delete',
-            'type_job_start', 'type_job_end', 'payload_url', 'enabled', 'http_method', 'http_content_type',
-            'additional_headers', 'body_template', 'secret', 'conditions', 'ssl_verification', 'ca_file_path',
+            'type_job_start', 'type_job_end', 'enabled', 'conditions', 'event_type',
+            'custom_fields', 'tags', 'created', 'last_updated',
+        ]
+
+
+#
+# Webhooks
+#
+
+class WebhookSerializer(NetBoxModelSerializer):
+    url = serializers.HyperlinkedIdentityField(view_name='extras-api:webhook-detail')
+
+    class Meta:
+        model = Webhook
+        fields = [
+            'id', 'url', 'display',
+            'payload_url', 'http_method', 'http_content_type',
+            'additional_headers', 'body_template', 'secret', 'ssl_verification', 'ca_file_path',
             'custom_fields', 'tags', 'created', 'last_updated',
         ]
 
diff --git a/netbox/extras/api/urls.py b/netbox/extras/api/urls.py
index 5f2b324e6..1616b8554 100644
--- a/netbox/extras/api/urls.py
+++ b/netbox/extras/api/urls.py
@@ -7,6 +7,7 @@ from . import views
 router = NetBoxRouter()
 router.APIRootView = views.ExtrasRootView
 
+router.register('event-rules', views.EventRuleViewSet)
 router.register('webhooks', views.WebhookViewSet)
 router.register('custom-fields', views.CustomFieldViewSet)
 router.register('custom-field-choice-sets', views.CustomFieldChoiceSetViewSet)
diff --git a/netbox/extras/api/views.py b/netbox/extras/api/views.py
index f518275e0..75c69da1d 100644
--- a/netbox/extras/api/views.py
+++ b/netbox/extras/api/views.py
@@ -37,6 +37,17 @@ class ExtrasRootView(APIRootView):
         return 'Extras'
 
 
+#
+# EventRules
+#
+
+class EventRuleViewSet(NetBoxModelViewSet):
+    metadata_class = ContentTypeMetadata
+    queryset = EventRule.objects.all()
+    serializer_class = serializers.EventRuleSerializer
+    filterset_class = filtersets.EventRuleFilterSet
+
+
 #
 # Webhooks
 #
diff --git a/netbox/extras/graphql/types.py b/netbox/extras/graphql/types.py
index 068da02f2..4981ddd72 100644
--- a/netbox/extras/graphql/types.py
+++ b/netbox/extras/graphql/types.py
@@ -8,6 +8,7 @@ __all__ = (
     'CustomFieldChoiceSetType',
     'CustomFieldType',
     'CustomLinkType',
+    'EventRuleType',
     'ExportTemplateType',
     'ImageAttachmentType',
     'JournalEntryType',
@@ -110,5 +111,12 @@ class WebhookType(OrganizationalObjectType):
 
     class Meta:
         model = models.Webhook
-        exclude = ('content_types', )
         filterset_class = filtersets.WebhookFilterSet
+
+
+class EventRuleType(OrganizationalObjectType):
+
+    class Meta:
+        model = models.EventRule
+        exclude = ('content_types', )
+        filterset_class = filtersets.EventRuleFilterSet