drone-telegram/plugin.go

package main

import (
	"errors"
	"fmt"
	"log"
	"os"
	"strconv"
	"strings"

	"gopkg.in/telegram-bot-api.v4"
)

type (
	// Repo information.
	Repo struct {
		Owner string
		Name  string
	}

	// Build information.
	Build struct {
		Event   string
		Number  int
		Commit  string
		Message string
		Branch  string
		Author  string
		Status  string
		Link    string
	}

	// Config for the plugin.
	Config struct {
		Token    string
		Debug    bool
		To       []string
		Message  []string
		Photo    []string
		Document []string
		Format   string
	}

	// Plugin values.
	Plugin struct {
		Repo   Repo
		Build  Build
		Config Config
	}
)

func trimElement(keys []string) []string {
	var newKeys []string

	for _, value := range keys {
		value = strings.Trim(value, " ")
		if len(value) == 0 {
			continue
		}
		newKeys = append(newKeys, value)
	}

	return newKeys
}

func fileExist(keys []string) []string {
	var newKeys []string

	for _, value := range keys {
		if _, err := os.Stat(value); os.IsNotExist(err) {
			continue
		}
		newKeys = append(newKeys, value)
	}

	return newKeys
}

func parseID(keys []string) []int64 {
	var newKeys []int64

	for _, value := range keys {
		id, err := strconv.ParseInt(value, 10, 64)
		if err != nil {
			log.Println(err.Error())

			continue
		}
		newKeys = append(newKeys, id)
	}

	return newKeys
}

// Exec executes the plugin.
func (p Plugin) Exec() error {

	if len(p.Config.Token) == 0 || len(p.Config.To) == 0 {
		log.Println("missing telegram token or user list")

		return errors.New("missing telegram token or user list")
	}

	var message []string
	if len(p.Config.Message) > 0 {
		message = p.Config.Message
	} else {
		message = p.Message(p.Repo, p.Build)
	}

	bot, err := tgbotapi.NewBotAPI(p.Config.Token)

	if err != nil {
		log.Println("Initialize New Bot Error:", err.Error())

		return err
	}

	bot.Debug = p.Config.Debug

	// parse ids
	ids := parseID(p.Config.To)
	photos := fileExist(trimElement(p.Config.Photo))
	documents := fileExist(trimElement(p.Config.Document))

	// send message.
	for _, user := range ids {
		for _, value := range trimElement(message) {
			msg := tgbotapi.NewMessage(user, value)
			msg.ParseMode = p.Config.Format
			p.Send(bot, msg)
		}

		for _, value := range photos {
			msg := tgbotapi.NewPhotoUpload(user, value)
			p.Send(bot, msg)
		}

		for _, value := range documents {
			msg := tgbotapi.NewDocumentUpload(user, value)
			p.Send(bot, msg)
		}
	}

	return nil
}

// Send bot message.
func (p Plugin) Send(bot *tgbotapi.BotAPI, msg tgbotapi.Chattable) {
	_, err := bot.Send(msg)

	if err != nil {
		log.Println(err.Error())
	}
}

// Message is plugin default message.
func (p Plugin) Message(repo Repo, build Build) []string {
	return []string{fmt.Sprintf("[%s] <%s> (%s)『%s』by %s",
		build.Status,
		build.Link,
		build.Branch,
		build.Message,
		build.Author,
	)}
}