package main import ( "context" "crypto/sha512" "fmt" "regexp" "strings" "github.com/mattn/go-mastodon" ) var ( c = mastodon.NewClient(&mastodon.Config{ Server: Conf.Server, ClientID: Conf.ClientID, ClientSecret: Conf.ClientSecret, AccessToken: Conf.AccessToken, }) ctx = context.Background() my_account, _ = c.GetAccountCurrentUser(ctx) ) func RunBot() { events, err := c.StreamingUser(ctx) if err != nil { ErrorLogger.Println("Streaming") } followers, err := c.GetAccountFollowers(ctx, my_account.ID, &mastodon.Pagination{Limit: 60}) if err != nil { ErrorLogger.Println("Fetch followers") } // Run bot for { notifEvent, ok := (<-events).(*mastodon.NotificationEvent) if !ok { continue } notif := notifEvent.Notification // New follower if notif.Type == "follow" { acct := notif.Status.Account.Acct if !exist_in_database(acct) { // Add to db and post welcome message InfoLogger.Printf("%s followed", acct) add_to_db(acct) InfoLogger.Printf("%s added to database", acct) message := fmt.Sprintf("%s @%s", Conf.WelcomeMessage, acct) _, err := postToot(message, "public") if err != nil { ErrorLogger.Println("Post welcome message") } InfoLogger.Printf("%s was welcomed", acct) } } // Read message if notif.Type == "mention" { acct := notif.Status.Account.Acct content := notif.Status.Content tooturl := notif.Status.URL for i := range followers { if acct == string(followers[i].Acct) { // Follow check if notif.Status.Visibility == "public" { // Reblog toot if notif.Status.InReplyToID == nil { // Not boost replies // Duplicate protection content_hash := sha512.New() content_hash.Write([]byte(content)) hash := fmt.Sprintf("%x", content_hash.Sum(nil)) if !check_msg_hash(hash) { save_msg_hash(hash) InfoLogger.Printf("Hash of %s added to database", tooturl) } else { WarnLogger.Printf("%s is a duplicate and not boosted", tooturl) break } // Add to db if needed if !exist_in_database(acct) { add_to_db(acct) InfoLogger.Printf("%s added to database", acct) } // Message order if check_order(acct) < Conf.Order_limit { if check_ticket(acct) > 0 { // Message limit take_ticket(acct) InfoLogger.Printf("Ticket of %s was taken", acct) count_order(acct) InfoLogger.Printf("Order of %s was counted", acct) c.Reblog(ctx, notif.Status.ID) InfoLogger.Printf("Toot %s of %s was rebloged", tooturl, acct) } else { WarnLogger.Printf("%s haven't tickets", acct) } } else { WarnLogger.Printf("%s order limit", acct) } } else { WarnLogger.Printf("%s is reply and not boosted", tooturl) } } else if notif.Status.Visibility == "direct" { // Admin commands for y := 0; y < len(Conf.Admins); y++ { if acct == Conf.Admins[y] { recmd := regexp.MustCompile(`<[^>]+>`) command := recmd.ReplaceAllString(content, "") args := strings.Split(command, " ") if len(args) == 3 { mID := mastodon.ID((args[2])) switch args[1] { case "unboost": c.Unreblog(ctx, mID) WarnLogger.Printf("%s was unrebloged", mID) case "delete": c.DeleteStatus(ctx, mID) WarnLogger.Printf("%s was deleted", mID) } } } else { continue } } } else { WarnLogger.Printf("%s is not public toot and not boosted", tooturl) break } } if i == len(followers)-1 { // Notify user if got_notice(acct) == 0 { if !exist_in_database(acct) { add_to_db(acct) InfoLogger.Printf("%s added to database", acct) } message := fmt.Sprintf("@%s %s", acct, Conf.NotFollowedMessage) _, err := postToot(message, "direct") if err != nil { ErrorLogger.Printf("Notify %s", acct) } InfoLogger.Printf("%s has been notified", acct) mark_notice(acct) InfoLogger.Printf("%s marked notification in database", acct) } } } } } }