mirror of
https://github.com/SecurityBrewery/catalyst.git
synced 2025-12-06 15:22:47 +01:00
133 lines
3.6 KiB
Go
133 lines
3.6 KiB
Go
package webhook
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
|
|
"github.com/labstack/echo/v5"
|
|
"github.com/pocketbase/dbx"
|
|
"github.com/pocketbase/pocketbase/apis"
|
|
"github.com/pocketbase/pocketbase/core"
|
|
"github.com/pocketbase/pocketbase/daos"
|
|
"github.com/pocketbase/pocketbase/migrations"
|
|
"github.com/pocketbase/pocketbase/models"
|
|
"github.com/pocketbase/pocketbase/models/schema"
|
|
)
|
|
|
|
const webhooksCollection = "webhooks"
|
|
|
|
type Webhook struct {
|
|
ID string `db:"id" json:"id"`
|
|
Name string `db:"name" json:"name"`
|
|
Collection string `db:"collection" json:"collection"`
|
|
Destination string `db:"destination" json:"destination"`
|
|
}
|
|
|
|
func BindHooks(app core.App) {
|
|
migrations.Register(func(db dbx.Builder) error {
|
|
return daos.New(db).SaveCollection(&models.Collection{
|
|
Name: webhooksCollection,
|
|
Type: models.CollectionTypeBase,
|
|
System: true,
|
|
Schema: schema.NewSchema(
|
|
&schema.SchemaField{
|
|
Name: "name",
|
|
Type: schema.FieldTypeText,
|
|
Required: true,
|
|
},
|
|
&schema.SchemaField{
|
|
Name: "collection",
|
|
Type: schema.FieldTypeText,
|
|
Required: true,
|
|
},
|
|
&schema.SchemaField{
|
|
Name: "destination",
|
|
Type: schema.FieldTypeUrl,
|
|
Required: true,
|
|
},
|
|
),
|
|
})
|
|
}, nil, "1690000000_webhooks.go")
|
|
|
|
app.OnRecordAfterCreateRequest().Add(func(e *core.RecordCreateEvent) error {
|
|
return event(app, "create", e.Collection.Name, e.Record, e.HttpContext)
|
|
})
|
|
app.OnRecordAfterUpdateRequest().Add(func(e *core.RecordUpdateEvent) error {
|
|
return event(app, "update", e.Collection.Name, e.Record, e.HttpContext)
|
|
})
|
|
app.OnRecordAfterDeleteRequest().Add(func(e *core.RecordDeleteEvent) error {
|
|
return event(app, "delete", e.Collection.Name, e.Record, e.HttpContext)
|
|
})
|
|
}
|
|
|
|
type Payload struct {
|
|
Action string `json:"action"`
|
|
Collection string `json:"collection"`
|
|
Record *models.Record `json:"record"`
|
|
Auth *models.Record `json:"auth,omitempty"`
|
|
Admin *models.Admin `json:"admin,omitempty"`
|
|
}
|
|
|
|
func event(app core.App, event, collection string, record *models.Record, ctx echo.Context) error {
|
|
auth, _ := ctx.Get(apis.ContextAuthRecordKey).(*models.Record)
|
|
admin, _ := ctx.Get(apis.ContextAdminKey).(*models.Admin)
|
|
|
|
var webhooks []Webhook
|
|
if err := app.Dao().DB().
|
|
Select().
|
|
From(webhooksCollection).
|
|
Where(dbx.HashExp{"collection": collection}).
|
|
All(&webhooks); err != nil {
|
|
return err
|
|
}
|
|
|
|
if len(webhooks) == 0 {
|
|
return nil
|
|
}
|
|
|
|
payload, err := json.Marshal(&Payload{
|
|
Action: event,
|
|
Collection: collection,
|
|
Record: record,
|
|
Auth: auth,
|
|
Admin: admin,
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, webhook := range webhooks {
|
|
if err := sendWebhook(ctx.Request().Context(), webhook, payload); err != nil {
|
|
app.Logger().Error("failed to send webhook", "action", event, "name", webhook.Name, "collection", webhook.Collection, "destination", webhook.Destination, "error", err.Error())
|
|
} else {
|
|
app.Logger().Info("webhook sent", "action", event, "name", webhook.Name, "collection", webhook.Collection, "destination", webhook.Destination)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func sendWebhook(ctx context.Context, webhook Webhook, payload []byte) error {
|
|
req, _ := http.NewRequestWithContext(ctx, http.MethodPost, webhook.Destination, bytes.NewReader(payload))
|
|
req.Header.Set("Content-Type", "application/json")
|
|
|
|
resp, err := http.DefaultClient.Do(req)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
defer resp.Body.Close()
|
|
|
|
if resp.StatusCode < 200 || resp.StatusCode >= 300 {
|
|
b, _ := io.ReadAll(resp.Body)
|
|
|
|
return fmt.Errorf("failed to send webhook: %s", string(b))
|
|
}
|
|
|
|
return nil
|
|
}
|