-
Notifications
You must be signed in to change notification settings - Fork 18
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Chore: add sendgrid webhook endpoints to receive webhook events (#1150)
* Chore: Add sendgrid webhook receiver to trigger workflow Signed-off-by: Daishan Peng <[email protected]>
- Loading branch information
1 parent
c1041f6
commit 6d035af
Showing
8 changed files
with
206 additions
and
111 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,45 @@ | ||
| package sendgrid | ||
|
|
||
| import ( | ||
| "fmt" | ||
| "net/http" | ||
|
|
||
| "github.com/obot-platform/obot/apiclient/types" | ||
| "github.com/obot-platform/obot/pkg/api" | ||
| "github.com/obot-platform/obot/pkg/emailtrigger" | ||
| "github.com/sendgrid/sendgrid-go/helpers/inbound" | ||
| kclient "sigs.k8s.io/controller-runtime/pkg/client" | ||
| ) | ||
|
|
||
| type InboundWebhookHandler struct { | ||
| emailTrigger *emailtrigger.EmailHandler | ||
| username string | ||
| password string | ||
| } | ||
|
|
||
| func NewInboundWebhookHandler(c kclient.Client, hostname string, username, password string) *InboundWebhookHandler { | ||
| emailTrigger := emailtrigger.EmailTrigger(c, hostname) | ||
| return &InboundWebhookHandler{emailTrigger: emailTrigger, username: username, password: password} | ||
| } | ||
|
|
||
| func (h *InboundWebhookHandler) InboundWebhookHandler(req api.Context) error { | ||
| if h.username != "" && h.password != "" { | ||
| username, password, ok := req.Request.BasicAuth() | ||
| if !ok || username != h.username || password != h.password { | ||
| return types.NewErrHttp(http.StatusUnauthorized, "Invalid credentials") | ||
| } | ||
| } | ||
|
|
||
| inboundEmail, err := inbound.Parse(req.Request) | ||
| if err != nil { | ||
| return types.NewErrHttp(http.StatusBadRequest, fmt.Sprintf("Failed to parse inbound email: %v", err)) | ||
| } | ||
|
|
||
| subject := inboundEmail.Headers["Subject"] | ||
| if err := h.emailTrigger.Handler(req.Context(), inboundEmail.Envelope.From, inboundEmail.Envelope.To, subject, []byte(inboundEmail.TextBody)); err != nil { | ||
| return types.NewErrHttp(http.StatusInternalServerError, fmt.Sprintf("Failed to handle inbound email: %v", err)) | ||
| } | ||
|
|
||
| req.WriteHeader(http.StatusOK) | ||
| return nil | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,132 @@ | ||
| package emailtrigger | ||
|
|
||
| import ( | ||
| "context" | ||
| "encoding/json" | ||
| "fmt" | ||
| "net/mail" | ||
| "path" | ||
| "strings" | ||
|
|
||
| "github.com/obot-platform/obot/logger" | ||
| "github.com/obot-platform/obot/pkg/alias" | ||
| v1 "github.com/obot-platform/obot/pkg/storage/apis/obot.obot.ai/v1" | ||
| "github.com/obot-platform/obot/pkg/system" | ||
| apierror "k8s.io/apimachinery/pkg/api/errors" | ||
| metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" | ||
| kclient "sigs.k8s.io/controller-runtime/pkg/client" | ||
| ) | ||
|
|
||
| var log = logger.Package() | ||
|
|
||
| type EmailHandler struct { | ||
| c kclient.Client | ||
| hostname string | ||
| } | ||
|
|
||
| func EmailTrigger(c kclient.Client, hostname string) *EmailHandler { | ||
| return &EmailHandler{ | ||
| c: c, | ||
| hostname: hostname, | ||
| } | ||
| } | ||
|
|
||
| func (h *EmailHandler) Handler(ctx context.Context, from string, to []string, subject string, data []byte) error { | ||
| for _, to := range to { | ||
| toAddr, err := mail.ParseAddress(to) | ||
| if err != nil { | ||
| return fmt.Errorf("parse to address: %w", err) | ||
| } | ||
|
|
||
| name, host, ok := strings.Cut(toAddr.Address, "@") | ||
| if !ok { | ||
| return fmt.Errorf("invalid to address: %s", toAddr.Address) | ||
| } | ||
|
|
||
| if host != h.hostname { | ||
| log.Infof("Skipping mail for %s: not for this host", toAddr.Address) | ||
| continue | ||
| } | ||
|
|
||
| name, ns, _ := strings.Cut(name, "+") | ||
| if ns == "" { | ||
| ns = system.DefaultNamespace | ||
| } | ||
|
|
||
| var emailReceiver v1.EmailReceiver | ||
| if err = alias.Get(ctx, h.c, &emailReceiver, ns, name); apierror.IsNotFound(err) { | ||
| log.Infof("Skipping mail for %s: no receiver found", toAddr.Address) | ||
| continue | ||
| } else if err != nil { | ||
| return fmt.Errorf("get email receiver: %w", err) | ||
| } | ||
|
|
||
| if !matches(from, emailReceiver) { | ||
| log.Infof("Skipping mail for %s: sender not allowed", toAddr.Address) | ||
| continue | ||
| } | ||
|
|
||
| if err = h.dispatchEmail(ctx, emailReceiver, string(data), from, to, subject); err != nil { | ||
| return fmt.Errorf("dispatch email: %w", err) | ||
| } | ||
| } | ||
|
|
||
| return nil | ||
| } | ||
|
|
||
| func (h *EmailHandler) dispatchEmail(ctx context.Context, email v1.EmailReceiver, body string, from, to, subject string) error { | ||
| var input struct { | ||
| Type string `json:"type"` | ||
| From string `json:"from"` | ||
| To string `json:"to"` | ||
| Subject string `json:"subject"` | ||
| Body string `json:"body"` | ||
| } | ||
|
|
||
| input.Type = "email" | ||
| input.From = from | ||
| input.To = to | ||
| input.Subject = subject | ||
| input.Body = body | ||
|
|
||
| inputJSON, err := json.Marshal(input) | ||
| if err != nil { | ||
| return fmt.Errorf("marshal input: %w", err) | ||
| } | ||
|
|
||
| var workflow v1.Workflow | ||
| if err = alias.Get(ctx, h.c, &workflow, email.Namespace, email.Spec.Workflow); err != nil { | ||
| return err | ||
| } | ||
|
|
||
| return h.c.Create(ctx, &v1.WorkflowExecution{ | ||
| ObjectMeta: metav1.ObjectMeta{ | ||
| GenerateName: system.WorkflowExecutionPrefix, | ||
| Namespace: workflow.Namespace, | ||
| }, | ||
| Spec: v1.WorkflowExecutionSpec{ | ||
| WorkflowName: workflow.Name, | ||
| EmailReceiverName: email.Name, | ||
| ThreadName: workflow.Spec.ThreadName, | ||
| Input: string(inputJSON), | ||
| }, | ||
| }) | ||
| } | ||
|
|
||
| func matches(address string, email v1.EmailReceiver) bool { | ||
| if len(email.Spec.AllowedSenders) == 0 { | ||
| return true | ||
| } | ||
|
|
||
| for _, allowedSender := range email.Spec.AllowedSenders { | ||
| if allowedSender == address { | ||
| return true | ||
| } | ||
| matched, _ := path.Match(allowedSender, address) | ||
| if matched { | ||
| return true | ||
| } | ||
| } | ||
|
|
||
| return false | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.