mastodon-group-bot/bot.go

package main

import (
	"context"
	"fmt"
	"log"
	"regexp"
	"strings"

	"github.com/mattn/go-mastodon"
)

func run_bot(Conf Config, DB string) {
	c := mastodon.NewClient(&mastodon.Config{
		Server:       Conf.Server,
		ClientID:     Conf.ClientID,
		ClientSecret: Conf.ClientSecret,
		AccessToken:  Conf.AccessToken,
	})

	ctx := context.Background()
	events, err := c.StreamingUser(ctx)
	if err != nil {
		log.Fatal(err)
	}

	my_account, err := c.GetAccountCurrentUser(ctx)
	if err != nil {
		log.Fatal(err)
	}
	followers, err := c.GetAccountFollowers(ctx, my_account.ID, &mastodon.Pagination{Limit: 60})
	if err != nil {
		log.Fatal(err)
	}

	// Run bot
	for {
		notifEvent, ok := (<-events).(*mastodon.NotificationEvent)
		if !ok {
			continue
		}

		notif := notifEvent.Notification

		// Posting function
		postToot := func(toot string, vis string) error {
			conToot := mastodon.Toot{
				Status:     toot,
				Visibility: vis,
			}
			_, err := c.PostStatus(ctx, &conToot)
			return err
		}

		// New follower
		if notif.Type == "follow" {
			acct := notif.Account.Acct
			if !followed(acct, DB) { // Add to db and post welcome message
				add_to_db(acct, Conf.Max_toots, DB)
				var message = fmt.Sprintf("%s @%s", Conf.WelcomeMessage, acct)
				postToot(message, "public")
			}
		}

		// Read message
		if notif.Type == "mention" {
			acct := notif.Status.Account.Acct
			for i := 0; i < len(followers); i++ {
				if acct == string(followers[i].Acct) { // Follow check
					if notif.Status.Visibility == "public" { // Reblog toot
						if notif.Status.InReplyToID == nil { // Not boost replies
							if !followed(acct, DB) { // Add to db if needed
								add_to_db(acct, Conf.Max_toots, DB)
							}
							if check_ticket(acct, Conf.Max_toots, Conf.Toots_interval, DB) > 0 { // Limit
								take_ticket(acct, DB)
								c.Reblog(ctx, notif.Status.ID)
							}
						}
					} else if notif.Status.Visibility == "direct" { // Admin commands
						for y := 0; y < len(Conf.Admins); y++ {
							if acct == Conf.Admins[y] {
								text := notif.Status.Content
								recmd := regexp.MustCompile(`<.*?> `)
								command := recmd.ReplaceAllString(text, "")
								args := strings.Split(command, " ")
								mID := mastodon.ID((args[1]))

								if len(args) == 2 {
									switch args[0] {
									case "unboost":
										c.Unreblog(ctx, mID)
									case "delete":
										c.DeleteStatus(ctx, mID)
									case "block":
										c.AccountBlock(ctx, mID)
									case "unblock":
										c.AccountUnblock(ctx, mID)
									}
								}
							} else {
								var message = fmt.Sprintf("@%s%s", notif.Account.Acct, ", you are not admin!")
								postToot(message, "direct")
							}
						}
					}
				} else {
					var message = fmt.Sprintf("@%s%s", notif.Account.Acct, ", you are not subscribed!")
					postToot(message, "direct")
				}
			}
		}
	}
}