notify/service/line/line.go

package line

import (
	"context"

	"github.com/line/line-bot-sdk-go/linebot"
	"github.com/pkg/errors"
)

// Line struct holds info about client and destination ID for communicating with line API
type Line struct {
	client      *linebot.Client
	receiverIDs []string
}

// New creates a new instance of Line notifier service
// For more info about line api credential:
// -> https://github.com/line/line-bot-sdk-go
func New(channelSecret, channelAccessToken string) (*Line, error) {
	bot, err := linebot.New(channelSecret, channelAccessToken)
	if err != nil {
		return nil, err
	}

	l := &Line{
		client: bot,
	}

	return l, nil
}

// AddReceivers receives user, group or room IDs then add them to internal receivers list
func (l *Line) AddReceivers(receiverIDs ...string) {
	l.receiverIDs = append(l.receiverIDs, receiverIDs...)
}

// Send receives message subject and body then sends it to all receivers set previously
// Subject will be on the first line followed by message on the next line
func (l *Line) Send(ctx context.Context, subject, message string) error {
	lineMessage := &linebot.TextMessage{
		Text: subject + "\n" + message,
	}

	for _, receiverID := range l.receiverIDs {
		select {
		case <-ctx.Done():
			return ctx.Err()
		default:
			_, err := l.client.PushMessage(receiverID, lineMessage).WithContext(ctx).Do()
			if err != nil {
				return errors.Wrapf(err, "failed to send message to LINE contact '%s'", receiverID)
			}
		}
	}

	return nil
}