file distribution

This commit is contained in:
fade 2022-08-22 07:46:49 -04:00
parent 3c96ba5095
commit 4c046e3870
3 changed files with 127 additions and 115 deletions

94
bot.go Normal file
View 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
View 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
View File

@ -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")
}
}
}
}
} }