2022-08-22 14:46:49 +03:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
2022-09-01 14:31:27 +03:00
|
|
|
"crypto/sha512"
|
2022-08-22 14:46:49 +03:00
|
|
|
"fmt"
|
|
|
|
"regexp"
|
|
|
|
"strings"
|
|
|
|
|
|
|
|
"github.com/mattn/go-mastodon"
|
|
|
|
)
|
|
|
|
|
2022-09-08 15:26:25 +03:00
|
|
|
var (
|
|
|
|
c = mastodon.NewClient(&mastodon.Config{
|
2022-08-22 14:46:49 +03:00
|
|
|
Server: Conf.Server,
|
|
|
|
ClientID: Conf.ClientID,
|
|
|
|
ClientSecret: Conf.ClientSecret,
|
|
|
|
AccessToken: Conf.AccessToken,
|
|
|
|
})
|
|
|
|
|
2022-09-08 15:26:25 +03:00
|
|
|
ctx = context.Background()
|
|
|
|
|
|
|
|
my_account, _ = c.GetAccountCurrentUser(ctx)
|
|
|
|
)
|
|
|
|
|
|
|
|
func RunBot() {
|
2022-08-22 14:46:49 +03:00
|
|
|
events, err := c.StreamingUser(ctx)
|
|
|
|
if err != nil {
|
2022-08-28 19:37:43 +03:00
|
|
|
ErrorLogger.Println("Streaming")
|
2022-08-22 14:46:49 +03:00
|
|
|
}
|
|
|
|
|
2022-08-26 13:56:21 +03:00
|
|
|
followers, err := c.GetAccountFollowers(ctx, my_account.ID, &mastodon.Pagination{Limit: 60})
|
|
|
|
if err != nil {
|
2022-08-28 19:37:43 +03:00
|
|
|
ErrorLogger.Println("Fetch followers")
|
2022-08-26 13:56:21 +03:00
|
|
|
}
|
2022-08-22 14:46:49 +03:00
|
|
|
|
|
|
|
// Run bot
|
|
|
|
for {
|
|
|
|
notifEvent, ok := (<-events).(*mastodon.NotificationEvent)
|
|
|
|
if !ok {
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
|
|
|
|
notif := notifEvent.Notification
|
2022-09-08 15:26:25 +03:00
|
|
|
ntype := notif.Type
|
|
|
|
acct := notif.Status.Account.Acct
|
|
|
|
content := notif.Status.Content
|
|
|
|
tooturl := notif.Status.URL
|
2022-08-22 14:46:49 +03:00
|
|
|
|
2022-09-08 15:26:25 +03:00
|
|
|
/*postToot := func(toot string, vis string) (*mastodon.Status, error) {
|
2022-08-22 14:46:49 +03:00
|
|
|
conToot := mastodon.Toot{
|
|
|
|
Status: toot,
|
|
|
|
Visibility: vis,
|
|
|
|
}
|
2022-09-08 15:26:25 +03:00
|
|
|
status, err := c.PostStatus(ctx, &conToot)
|
|
|
|
return status, err
|
|
|
|
}*/
|
2022-08-22 14:46:49 +03:00
|
|
|
|
|
|
|
// New follower
|
2022-09-08 15:26:25 +03:00
|
|
|
if ntype == "follow" {
|
|
|
|
if !exist_in_database(acct) { // Add to db and post welcome message
|
2022-08-28 19:37:43 +03:00
|
|
|
InfoLogger.Printf("%s followed", acct)
|
|
|
|
|
2022-09-01 14:31:27 +03:00
|
|
|
add_to_db(acct)
|
2022-08-28 19:37:43 +03:00
|
|
|
InfoLogger.Printf("%s added to database", acct)
|
|
|
|
|
2022-09-06 12:31:05 +03:00
|
|
|
message := fmt.Sprintf("%s @%s", Conf.WelcomeMessage, acct)
|
2022-09-08 15:26:25 +03:00
|
|
|
_, err := postToot(message, "public")
|
2022-08-28 19:37:43 +03:00
|
|
|
if err != nil {
|
|
|
|
ErrorLogger.Println("Post welcome message")
|
|
|
|
}
|
|
|
|
InfoLogger.Printf("%s was welcomed", acct)
|
2022-08-25 22:23:44 +03:00
|
|
|
}
|
2022-08-22 14:46:49 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
// Read message
|
2022-09-08 15:26:25 +03:00
|
|
|
if ntype == "mention" {
|
2022-08-25 22:23:44 +03:00
|
|
|
for i := 0; i < len(followers); i++ {
|
|
|
|
if acct == string(followers[i].Acct) { // Follow check
|
2022-08-22 14:46:49 +03:00
|
|
|
if notif.Status.Visibility == "public" { // Reblog toot
|
2022-08-25 22:23:44 +03:00
|
|
|
if notif.Status.InReplyToID == nil { // Not boost replies
|
2022-09-01 14:31:27 +03:00
|
|
|
// 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
|
2022-09-08 15:26:25 +03:00
|
|
|
if !exist_in_database(acct) {
|
2022-09-01 14:31:27 +03:00
|
|
|
add_to_db(acct)
|
2022-08-28 19:37:43 +03:00
|
|
|
InfoLogger.Printf("%s added to database", acct)
|
2022-08-25 22:23:44 +03:00
|
|
|
}
|
2022-09-01 14:31:27 +03:00
|
|
|
|
2022-09-06 12:31:05 +03:00
|
|
|
// 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)
|
|
|
|
}
|
2022-08-28 19:37:43 +03:00
|
|
|
} else {
|
2022-09-06 12:31:05 +03:00
|
|
|
WarnLogger.Printf("%s order limit", acct)
|
2022-08-25 22:23:44 +03:00
|
|
|
}
|
2022-08-28 19:37:43 +03:00
|
|
|
} else {
|
2022-09-01 14:31:27 +03:00
|
|
|
WarnLogger.Printf("%s is reply and not boosted", tooturl)
|
2022-08-22 21:05:40 +03:00
|
|
|
}
|
2022-08-22 14:46:49 +03:00
|
|
|
} else if notif.Status.Visibility == "direct" { // Admin commands
|
|
|
|
for y := 0; y < len(Conf.Admins); y++ {
|
2022-08-25 22:23:44 +03:00
|
|
|
if acct == Conf.Admins[y] {
|
2022-09-04 14:17:01 +03:00
|
|
|
recmd := regexp.MustCompile(`<[^>]+>`)
|
2022-09-01 14:31:27 +03:00
|
|
|
command := recmd.ReplaceAllString(content, "")
|
2022-08-22 14:46:49 +03:00
|
|
|
args := strings.Split(command, " ")
|
2022-09-04 14:17:01 +03:00
|
|
|
mID := mastodon.ID((args[2]))
|
2022-08-22 14:46:49 +03:00
|
|
|
|
2022-09-04 14:17:01 +03:00
|
|
|
if len(args) == 3 {
|
|
|
|
switch args[1] {
|
2022-08-22 14:46:49 +03:00
|
|
|
case "unboost":
|
2022-08-25 22:23:44 +03:00
|
|
|
c.Unreblog(ctx, mID)
|
2022-09-06 12:31:05 +03:00
|
|
|
WarnLogger.Printf("%s was unrebloged", mID)
|
2022-08-22 14:46:49 +03:00
|
|
|
case "delete":
|
2022-08-25 22:23:44 +03:00
|
|
|
c.DeleteStatus(ctx, mID)
|
2022-09-06 12:31:05 +03:00
|
|
|
WarnLogger.Printf("%s was deleted", mID)
|
2022-08-22 14:46:49 +03:00
|
|
|
}
|
|
|
|
}
|
2022-08-28 19:37:43 +03:00
|
|
|
} else {
|
2022-09-03 22:16:17 +03:00
|
|
|
continue
|
2022-08-22 14:46:49 +03:00
|
|
|
}
|
|
|
|
}
|
2022-08-28 19:37:43 +03:00
|
|
|
} else {
|
2022-09-01 14:31:27 +03:00
|
|
|
WarnLogger.Printf("%s is not public toot and not boosted", tooturl)
|
2022-08-28 19:37:43 +03:00
|
|
|
break
|
2022-08-22 14:46:49 +03:00
|
|
|
}
|
|
|
|
}
|
2022-09-08 15:26:25 +03:00
|
|
|
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)
|
|
|
|
}
|
|
|
|
}
|
2022-08-22 14:46:49 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|