notify/service/pagerduty/pager_duty.go

package pagerduty

import (
    "context"
    "errors"
    "fmt"

    "github.com/PagerDuty/go-pagerduty"
    "github.com/hanzoai/notify"
)

type Client interface {
    CreateIncidentWithContext(ctx context.Context, from string, options *pagerduty.CreateIncidentOptions) (*pagerduty.Incident, error) //nolint:lll // acceptable in this case, alternative makes the interface even less readable
}

// Compile-time check to verify that the PagerDuty type implements the notifier.Notifier interface.
var _ notify.Notifier = &PagerDuty{}

type PagerDuty struct {
    *Config
    Client Client
}

func New(token string, clientOptions ...pagerduty.ClientOptions) (*PagerDuty, error) {
    if token == "" {
        return nil, errors.New("access token is required")
    }

    pagerDuty := &PagerDuty{
        Config: NewConfig(),
        Client: pagerduty.NewClient(token, clientOptions...),
    }

    return pagerDuty, nil
}

func (s *PagerDuty) Send(ctx context.Context, subject, message string) error {
    if err := s.Config.OK(); err != nil {
        return fmt.Errorf("invalid configuration: %w", err)
    }

    incident := s.IncidentOptions(subject, message)

    for _, receiver := range s.Config.Receivers {
        // set the service ID to the receiver
        incident.Service.ID = receiver

        _, err := s.Client.CreateIncidentWithContext(ctx, s.Config.FromAddress, incident)
        if err != nil {
            return fmt.Errorf("create pager duty incident: %w", err)
        }
    }

    return nil
}

func (s *PagerDuty) IncidentOptions(subject, message string) *pagerduty.CreateIncidentOptions {
    return &pagerduty.CreateIncidentOptions{
        Title: subject,
        Service: &pagerduty.APIReference{
            ID:   "", // service ID will be set per receiver
            Type: APIReferenceType,
        },
        Body: &pagerduty.APIDetails{
            Type:    s.Config.NotificationType,
            Details: message,
        },
        Priority: s.Config.PriorityReference(),
        Urgency:  s.Config.Urgency,
    }
}