add support for a release event type - issue #7

Signed-off-by: thecreed <[email protected]>

Author: thecreed

httpinterface/api.go

@@ -1,7 +1,6 @@
 package httpinterface
 
 import (
-	"encoding/json"
 	"fmt"
 	"html/template"
 	"net/http"
@@ -106,28 +105,12 @@ func WebHookListener(w http.ResponseWriter, r *http.Request) {
 		}
 	}
 
-	var payload WebhookPayload
-	err = json.Unmarshal(bodyInBytes, &payload)
-	if err != nil {
-		Respond(w, 400, map[string]any{
-			"error": err.Error(),
-		})
-		return
-	}
+	eventType := r.Header.Get("X-GitHub-Event")
+	err = VerifyEvent(eventType, bodyInBytes, project.Branch)
 
-	if payload.Ref == "" {
+	if err != nil {
 		Respond(w, 400, map[string]interface{}{
-			"error": "invalid payload: cannot find ref inside given payload",
-		})
-		return
-	}
-
-	branchStringArr := strings.Split(payload.Ref, "/")
-	branchString := branchStringArr[len(branchStringArr)-1]
-
-	if project.Branch != branchString {
-		Respond(w, 200, map[string]interface{}{
-			"message": "request recieved but the push event is not for the configured branch",
+			"error": fmt.Sprintf("error: %v.", err),
 		})
 		return
 	}

httpinterface/util.go

@@ -6,6 +6,7 @@ import (
 	"crypto/sha256"
 	"encoding/hex"
 	"encoding/json"
+	"fmt"
 	"io"
 	"net/http"
 	"strings"
@@ -32,6 +33,55 @@ func VerifySignature(payload []byte, hash, key string) (bool, error) {
 
 }
 
+// VirifyType - check if a given event  type is supported
+func VerifyEvent(eventType string, bodyInBytes []byte, configBranchName string) error {
+	switch eventType {
+
+	case "push":
+
+		var payload WebhookPayload
+		err := json.Unmarshal(bodyInBytes, &payload)
+		if err != nil {
+			return fmt.Errorf("Couldnt unmarshal push event  - %v", err)
+		}
+
+		if payload.Ref == "" {
+			return fmt.Errorf("invalid payload: cannot find ref inside given payload")
+		}
+
+		branchStringArr := strings.Split(payload.Ref, "/")
+		branchString := branchStringArr[len(branchStringArr)-1]
+
+		if configBranchName != branchString {
+			return fmt.Errorf("request recieved but the push event is not for the configured branch")
+		}
+
+		return nil
+
+	case "release":
+		supportedReleaseActions := map[string]bool{
+			"published": true,
+			"created":   false,
+			"released":  false,
+		}
+
+		var payload ReleaseWebhookPayload
+		err := json.Unmarshal(bodyInBytes, &payload)
+
+		if err != nil {
+			return fmt.Errorf("couldnt unmarshal release event  - %v", err)
+		}
+		supported, exists := supportedReleaseActions[payload.Action]
+		if exists && supported {
+			return nil
+		}
+		return fmt.Errorf("release event  action %s is not enabled", payload.Action)
+
+	default:
+		return fmt.Errorf("event type %s: is not supported", eventType)
+	}
+}
+
 func StreamToByte(stream io.Reader) ([]byte, error) {
 	buf := new(bytes.Buffer)
 	_, err := buf.ReadFrom(stream)

httpinterface/webhookPayload.go

@@ -169,3 +169,7 @@ type WebhookPayload struct {
 	Repository RepositoryT `json:"repository"`
 	Sender     SenderT     `json:"sender"`
 }
+
+type ReleaseWebhookPayload struct {
+	Action string `json:"action"`
+}