378 lines
7.3 KiB
Go
378 lines
7.3 KiB
Go
package main
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"log"
|
|
"path/filepath"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/appleboy/drone-facebook/template"
|
|
"gopkg.in/telegram-bot-api.v4"
|
|
)
|
|
|
|
type (
|
|
// Repo information.
|
|
Repo struct {
|
|
Owner string
|
|
Name string
|
|
}
|
|
|
|
// Commit information.
|
|
Commit struct {
|
|
Sha string
|
|
Branch string
|
|
Link string
|
|
Author string
|
|
Email string
|
|
Message string
|
|
}
|
|
|
|
// Build information.
|
|
Build struct {
|
|
Tag string
|
|
Event string
|
|
Number int
|
|
Status string
|
|
Link string
|
|
Started float64
|
|
Finished float64
|
|
PR string
|
|
}
|
|
|
|
// Config for the plugin.
|
|
Config struct {
|
|
Token string
|
|
Debug bool
|
|
MatchEmail bool
|
|
WebPreview bool
|
|
To []string
|
|
Message []string
|
|
Photo []string
|
|
Document []string
|
|
Sticker []string
|
|
Audio []string
|
|
Voice []string
|
|
Location []string
|
|
Video []string
|
|
Venue []string
|
|
Format string
|
|
}
|
|
|
|
// Plugin values.
|
|
Plugin struct {
|
|
Repo Repo
|
|
Commit Commit
|
|
Build Build
|
|
Config Config
|
|
}
|
|
|
|
// Location format
|
|
Location struct {
|
|
Title string
|
|
Address string
|
|
Latitude float64
|
|
Longitude float64
|
|
}
|
|
)
|
|
|
|
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 escapeMarkdown(keys []string) []string {
|
|
var newKeys []string
|
|
|
|
for _, value := range keys {
|
|
value = escapeMarkdownOne(value)
|
|
if len(value) == 0 {
|
|
continue
|
|
}
|
|
newKeys = append(newKeys, value)
|
|
}
|
|
|
|
return newKeys
|
|
}
|
|
|
|
func escapeMarkdownOne(str string) string {
|
|
str = strings.Replace(str, `\_`, `_`, -1)
|
|
str = strings.Replace(str, `_`, `\_`, -1)
|
|
|
|
return str
|
|
}
|
|
|
|
func globList(keys []string) []string {
|
|
var newKeys []string
|
|
|
|
for _, pattern := range keys {
|
|
pattern = strings.Trim(pattern, " ")
|
|
matches, err := filepath.Glob(pattern)
|
|
if err != nil {
|
|
fmt.Printf("Glob error for %q: %s\n", pattern, err)
|
|
continue
|
|
}
|
|
newKeys = append(newKeys, matches...)
|
|
}
|
|
|
|
return newKeys
|
|
}
|
|
|
|
func convertLocation(value string) (Location, bool) {
|
|
var latitude, longitude float64
|
|
var title, address string
|
|
var err error
|
|
values := trimElement(strings.Split(value, ","))
|
|
|
|
if len(values) < 2 {
|
|
return Location{}, true
|
|
}
|
|
|
|
if len(values) > 2 {
|
|
title = values[2]
|
|
}
|
|
|
|
if len(values) > 3 {
|
|
title = values[2]
|
|
address = values[3]
|
|
}
|
|
|
|
latitude, err = strconv.ParseFloat(values[0], 64)
|
|
|
|
if err != nil {
|
|
log.Println(err.Error())
|
|
return Location{}, true
|
|
}
|
|
|
|
longitude, err = strconv.ParseFloat(values[1], 64)
|
|
|
|
if err != nil {
|
|
log.Println(err.Error())
|
|
return Location{}, true
|
|
}
|
|
|
|
return Location{
|
|
Title: title,
|
|
Address: address,
|
|
Latitude: latitude,
|
|
Longitude: longitude,
|
|
}, false
|
|
}
|
|
|
|
func parseTo(to []string, authorEmail string, matchEmail bool) []int64 {
|
|
var emails []int64
|
|
var ids []int64
|
|
attachEmail := true
|
|
|
|
for _, value := range trimElement(to) {
|
|
idArray := trimElement(strings.Split(value, ":"))
|
|
|
|
// check id
|
|
id, err := strconv.ParseInt(idArray[0], 10, 64)
|
|
if err != nil {
|
|
continue
|
|
}
|
|
|
|
// check match author email
|
|
if len(idArray) > 1 {
|
|
if email := idArray[1]; email != authorEmail {
|
|
continue
|
|
}
|
|
|
|
emails = append(emails, id)
|
|
attachEmail = false
|
|
continue
|
|
}
|
|
|
|
ids = append(ids, id)
|
|
}
|
|
|
|
if matchEmail && !attachEmail {
|
|
return emails
|
|
}
|
|
|
|
ids = append(ids, emails...)
|
|
|
|
return ids
|
|
}
|
|
|
|
// Exec executes the plugin.
|
|
func (p Plugin) Exec() error {
|
|
|
|
if len(p.Config.Token) == 0 || len(p.Config.To) == 0 {
|
|
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.Commit, p.Build)
|
|
}
|
|
|
|
bot, err := tgbotapi.NewBotAPI(p.Config.Token)
|
|
|
|
// enable bot debug mode
|
|
if p.Config.Debug {
|
|
bot.Debug = true
|
|
}
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
bot.Debug = p.Config.Debug
|
|
|
|
ids := parseTo(p.Config.To, p.Commit.Email, p.Config.MatchEmail)
|
|
photos := globList(trimElement(p.Config.Photo))
|
|
documents := globList(trimElement(p.Config.Document))
|
|
stickers := globList(trimElement(p.Config.Sticker))
|
|
audios := globList(trimElement(p.Config.Audio))
|
|
voices := globList(trimElement(p.Config.Voice))
|
|
videos := globList(trimElement(p.Config.Video))
|
|
locations := trimElement(p.Config.Location)
|
|
venues := trimElement(p.Config.Venue)
|
|
|
|
message = trimElement(message)
|
|
|
|
if p.Config.Format == "markdown" {
|
|
message = escapeMarkdown(message)
|
|
|
|
p.Commit.Message = escapeMarkdownOne(p.Commit.Message)
|
|
p.Commit.Branch = escapeMarkdownOne(p.Commit.Branch)
|
|
p.Commit.Link = escapeMarkdownOne(p.Commit.Link)
|
|
p.Commit.Author = escapeMarkdownOne(p.Commit.Author)
|
|
p.Commit.Email = escapeMarkdownOne(p.Commit.Email)
|
|
|
|
p.Build.Tag = escapeMarkdownOne(p.Build.Tag)
|
|
p.Build.Link = escapeMarkdownOne(p.Build.Link)
|
|
p.Build.PR = escapeMarkdownOne(p.Build.PR)
|
|
|
|
p.Repo.Owner = escapeMarkdownOne(p.Repo.Owner)
|
|
p.Repo.Name = escapeMarkdownOne(p.Repo.Name)
|
|
}
|
|
|
|
// send message.
|
|
for _, user := range ids {
|
|
for _, value := range message {
|
|
txt, err := template.RenderTrim(value, p)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
msg := tgbotapi.NewMessage(user, txt)
|
|
msg.ParseMode = p.Config.Format
|
|
msg.DisableWebPagePreview = !p.Config.WebPreview
|
|
if err := p.Send(bot, msg); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
for _, value := range photos {
|
|
msg := tgbotapi.NewPhotoUpload(user, value)
|
|
if err := p.Send(bot, msg); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
for _, value := range documents {
|
|
msg := tgbotapi.NewDocumentUpload(user, value)
|
|
if err := p.Send(bot, msg); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
for _, value := range stickers {
|
|
msg := tgbotapi.NewStickerUpload(user, value)
|
|
if err := p.Send(bot, msg); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
for _, value := range audios {
|
|
msg := tgbotapi.NewAudioUpload(user, value)
|
|
msg.Title = "Audio Message."
|
|
if err := p.Send(bot, msg); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
for _, value := range voices {
|
|
msg := tgbotapi.NewVoiceUpload(user, value)
|
|
if err := p.Send(bot, msg); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
for _, value := range videos {
|
|
msg := tgbotapi.NewVideoUpload(user, value)
|
|
msg.Caption = "Video Message"
|
|
if err := p.Send(bot, msg); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
for _, value := range locations {
|
|
location, empty := convertLocation(value)
|
|
|
|
if empty {
|
|
continue
|
|
}
|
|
|
|
msg := tgbotapi.NewLocation(user, location.Latitude, location.Longitude)
|
|
if err := p.Send(bot, msg); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
for _, value := range venues {
|
|
location, empty := convertLocation(value)
|
|
|
|
if empty {
|
|
continue
|
|
}
|
|
|
|
msg := tgbotapi.NewVenue(user, location.Title, location.Address, location.Latitude, location.Longitude)
|
|
if err := p.Send(bot, msg); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Send bot message.
|
|
func (p Plugin) Send(bot *tgbotapi.BotAPI, msg tgbotapi.Chattable) error {
|
|
message, err := bot.Send(msg)
|
|
|
|
if p.Config.Debug {
|
|
log.Println("=====================")
|
|
log.Printf("Response Message: %#v\n", message)
|
|
log.Println("=====================")
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
// Message is plugin default message.
|
|
func (p Plugin) Message(repo Repo, commit Commit, build Build) []string {
|
|
return []string{fmt.Sprintf("[%s] <%s> (%s)『%s』by %s",
|
|
build.Status,
|
|
build.Link,
|
|
commit.Branch,
|
|
commit.Message,
|
|
commit.Author,
|
|
)}
|
|
}
|