mirror of
https://git.phreedom.club/localhost_frssoft/mastodon-group-bot.git
synced 2024-12-27 03:29:48 +00:00
file distribution
This commit is contained in:
fade
parent
3c96ba5095
commit
4c046e3870
94
bot.go
Normal file
94
bot.go
Normal file
@ -0,0 +1,94 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"log"
|
||||||
|
"regexp"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/mattn/go-mastodon"
|
||||||
|
)
|
||||||
|
|
||||||
|
func run_bot(Conf Config) {
|
||||||
|
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, _ := c.GetAccountCurrentUser(ctx)
|
||||||
|
followers, _ := c.GetAccountFollowers(ctx, my_account.ID, &mastodon.Pagination{Limit: 60})
|
||||||
|
|
||||||
|
// 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" {
|
||||||
|
var message = fmt.Sprintf("%s @%s", Conf.WelcomeMessage, notif.Account.Acct)
|
||||||
|
postToot(message, "public")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Read message
|
||||||
|
if notif.Type == "mention" {
|
||||||
|
for i := 0; i < len(followers); i++ { // Follow check
|
||||||
|
if notif.Status.Account.Acct == string(followers[i].Acct) {
|
||||||
|
if notif.Status.Visibility == "public" { // Reblog toot
|
||||||
|
c.Reblog(ctx, notif.Status.ID)
|
||||||
|
} else if notif.Status.Visibility == "direct" { // Admin commands
|
||||||
|
for y := 0; y < len(Conf.Admins); y++ {
|
||||||
|
if notif.Status.Account.Acct == Conf.Admins[y] {
|
||||||
|
text := notif.Status.Content
|
||||||
|
recmd := regexp.MustCompile(`<.*?> `)
|
||||||
|
command := recmd.ReplaceAllString(text, "")
|
||||||
|
args := strings.Split(command, " ")
|
||||||
|
|
||||||
|
if len(args) == 2 {
|
||||||
|
switch args[0] {
|
||||||
|
case "unboost":
|
||||||
|
c.Unreblog(ctx, mastodon.ID((args[1])))
|
||||||
|
case "delete":
|
||||||
|
c.DeleteStatus(ctx, mastodon.ID(args[1]))
|
||||||
|
case "block":
|
||||||
|
c.AccountBlock(ctx, mastodon.ID(args[1]))
|
||||||
|
case "unblock":
|
||||||
|
c.AccountUnblock(ctx, mastodon.ID(args[1]))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} 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")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
32
config.go
Normal file
32
config.go
Normal file
@ -0,0 +1,32 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"flag"
|
||||||
|
"log"
|
||||||
|
"os"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Config struct {
|
||||||
|
Server string `json:"Server"`
|
||||||
|
ClientID string `json:"ClientID"`
|
||||||
|
ClientSecret string `json:"ClientSecret"`
|
||||||
|
AccessToken string `json:"AccessToken"`
|
||||||
|
WelcomeMessage string `json:"WelcomeMessage"`
|
||||||
|
Admins []string `json:"Admins"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func read_conf() Config {
|
||||||
|
ConfPath := flag.String("config", "config.json", "Path to config")
|
||||||
|
flag.Parse()
|
||||||
|
|
||||||
|
data, err := os.ReadFile(*ConfPath)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var Conf Config
|
||||||
|
json.Unmarshal(data, &Conf)
|
||||||
|
|
||||||
|
return Conf
|
||||||
|
}
|
||||||
116
main.go
116
main.go
@ -1,119 +1,5 @@
|
|||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
"encoding/json"
|
|
||||||
"flag"
|
|
||||||
"fmt"
|
|
||||||
"log"
|
|
||||||
"os"
|
|
||||||
"regexp"
|
|
||||||
"strings"
|
|
||||||
|
|
||||||
"github.com/mattn/go-mastodon"
|
|
||||||
)
|
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
// Parse args
|
run_bot(read_conf())
|
||||||
ConfPath := flag.String("config", "config.json", "Path to config")
|
|
||||||
flag.Parse()
|
|
||||||
|
|
||||||
// Reading config
|
|
||||||
type Config struct {
|
|
||||||
Server string `json:"Server"`
|
|
||||||
ClientID string `json:"ClientID"`
|
|
||||||
ClientSecret string `json:"ClientSecret"`
|
|
||||||
AccessToken string `json:"AccessToken"`
|
|
||||||
WelcomeMessage string `json:"WelcomeMessage"`
|
|
||||||
Admins []string `json:"Admins"`
|
|
||||||
}
|
|
||||||
|
|
||||||
data, err := os.ReadFile(*ConfPath)
|
|
||||||
if err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
var Conf Config
|
|
||||||
json.Unmarshal(data, &Conf)
|
|
||||||
|
|
||||||
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, _ := c.GetAccountCurrentUser(ctx)
|
|
||||||
followers, _ := c.GetAccountFollowers(ctx, my_account.ID, &mastodon.Pagination{Limit: 60})
|
|
||||||
|
|
||||||
// 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 subscriber
|
|
||||||
if notif.Type == "follow" {
|
|
||||||
var message = fmt.Sprintf("%s @%s", Conf.WelcomeMessage, notif.Account.Acct)
|
|
||||||
postToot(message, "public")
|
|
||||||
}
|
|
||||||
|
|
||||||
// Read message
|
|
||||||
if notif.Type == "mention" {
|
|
||||||
for i := 0; i < len(followers); i++ { // Follow check
|
|
||||||
if notif.Status.Account.Acct == string(followers[i].Acct) {
|
|
||||||
if notif.Status.Visibility == "public" { // Reblog toot
|
|
||||||
c.Reblog(ctx, notif.Status.ID)
|
|
||||||
} else if notif.Status.Visibility == "direct" { // Admin commands
|
|
||||||
for y := 0; y < len(Conf.Admins); y++ {
|
|
||||||
if notif.Status.Account.Acct == Conf.Admins[y] {
|
|
||||||
text := notif.Status.Content
|
|
||||||
recmd := regexp.MustCompile(`<.*?> `)
|
|
||||||
command := recmd.ReplaceAllString(text, "")
|
|
||||||
args := strings.Split(command, " ")
|
|
||||||
|
|
||||||
if len(args) == 2 {
|
|
||||||
switch args[0] {
|
|
||||||
case "unboost":
|
|
||||||
c.Unreblog(ctx, mastodon.ID((args[1])))
|
|
||||||
case "delete":
|
|
||||||
c.DeleteStatus(ctx, mastodon.ID(args[1]))
|
|
||||||
case "block":
|
|
||||||
c.AccountBlock(ctx, mastodon.ID(args[1]))
|
|
||||||
case "unblock":
|
|
||||||
c.AccountUnblock(ctx, mastodon.ID(args[1]))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} 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")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user