idanoo/go-mastodon-stats
1
0
Fork 0
mirror of https://github.com/idanoo/go-mastodon-stats synced 2025-07-01 22:02:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Test?

Browse source
This commit is contained in:
idanoo 2023-01-30 12:30:11 +13:00
parent 97caf4bc37
commit d927f1c5a1
Signed by: idanoo
GPG key ID: 387387CDBC02F132
5 changed files with 46 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

6
.env.example
View file

@ -15,5 +15,7 @@ MATRIX_WEBHOOK_API_KEY=""
MATRIX_WEBHOOK_CHANNEL=""
MASTODON_INSTANCE_URL=""
MASTODON_USERNAME=""
MASTODON_PASSWORD=""
MASTODON_CLIENT_ID=""
MASTODON_CLIENT_SECRET=""
MASTODON_CLIENT_USERNAME=""
MASTODON_CLIENT_PASSWORD=""

14
cmd/gomastodonstats/main.go
View file

@ -51,12 +51,16 @@ func main() {
// Load mastodon data if exists
gms.MASTODON_INSTANCE_URL = os.Getenv("MASTODON_INSTANCE_URL")
gms.MASTODON_USERNAME = os.Getenv("MASTODON_USERNAME")
gms.MASTODON_PASSWORD = os.Getenv("MASTODON_PASSWORD")
gms.MASTODON_CLIENT_ID = os.Getenv("MASTODON_CLIENT_ID")
gms.MASTODON_CLIENT_SECRET = os.Getenv("MASTODON_CLIENT_SECRET")
gms.MASTODON_CLIENT_USERNAME = os.Getenv("MASTODON_CLIENT_USERNAME")
gms.MASTODON_CLIENT_PASSWORD = os.Getenv("MASTODON_CLIENT_PASSWORD")
if gms.MASTODON_INSTANCE_URL == "" ||
gms.MASTODON_USERNAME == "" ||
gms.MASTODON_PASSWORD == "" {
log.Println("MASTODON_INSTANCE_URL info incompelete. Skipping")
gms.MASTODON_CLIENT_ID == "" ||
gms.MASTODON_CLIENT_SECRET == "" ||
gms.MASTODON_CLIENT_USERNAME == "" ||
gms.MASTODON_CLIENT_PASSWORD == "" {
log.Println("MASTODON_INSTANCE info incompelete. Skipping")
// Set URL empty so we can check this later on
gms.MASTODON_INSTANCE_URL = ""

11
internal/gomastodonstats/consts.go
View file

@ -14,10 +14,13 @@ var (
MATRIX_WEBHOOK_API_KEY string
MATRIX_WEBHOOK_CHANNEL string
MASTODON_INSTANCE_URL string
MASTODON_USERNAME string
MASTODON_PASSWORD string
MASTODON_CLIENT_NAME = "go-mastodon-stats"
MASTODON_CLIENT_NAME = "go-mastodon-stats"
MASTODON_INSTANCE_URL string
MASTODON_CLIENT_ID string
MASTODON_CLIENT_SECRET string
MASTODON_CLIENT_USERNAME string
MASTODON_CLIENT_PASSWORD string
// UserCount metric name
METRICNAME_USERCOUNT = "userCount"

33
internal/gomastodonstats/mastodon.go
View file

@ -4,10 +4,29 @@ import (
"context"
"fmt"
"log"
"os"
"github.com/mattn/go-mastodon"
)
func doTheRegisterThing() {
srv, err := mastodon.RegisterApp(context.Background(), &mastodon.AppConfig{
Server: fmt.Sprintf("https://%s", MASTODON_INSTANCE_URL),
ClientName: MASTODON_CLIENT_NAME,
Scopes: "read write follow",
RedirectURIs: "urn:ietf:wg:oauth:2.0:oob",
Website: "https://github.com/mattn/go-mastodon",
})
if err != nil {
log.Fatal("Couldn't register the app. Error: %v\n\nExiting...\n", err)
}
log.Println(
"Pls open this link: %s\n\n" + srv.AuthURI +
"When completeAdd this to your config under MASTODON_TOKEN and rerun the app")
os.Exit(0)
}
func postToMastodon(metrics []metric) {
if MASTODON_INSTANCE_URL == "" {
log.Printf("Skipping posting to mastodon. Missing configuration")
@ -16,24 +35,22 @@ func postToMastodon(metrics []metric) {
c := mastodon.NewClient(&mastodon.Config{
Server: fmt.Sprintf("https://%s", MASTODON_INSTANCE_URL),
ClientID: MASTODON_CLIENT_NAME,
ClientSecret: "random-secret-that-doesnt-really-matter-lol",
ClientID: MASTODON_CLIENT_ID,
ClientSecret: MASTODON_CLIENT_SECRET,
})
err := c.Authenticate(context.Background(), MASTODON_USERNAME, MASTODON_PASSWORD)
err := c.Authenticate(context.Background(), MASTODON_CLIENT_USERNAME, MASTODON_CLIENT_PASSWORD)
if err != nil {
log.Println("Invalid mastodon credentials?")
log.Print(err)
return
log.Fatal(err)
}
// Build status
startOfDay := getStartofDay()
msg := fmt.Sprintf("Stats for %s\n\n", startOfDay.String())
msg := fmt.Sprintf("User counts for %s :laserkiwi:\n\n", startOfDay.String())
for _, m := range metrics {
msg = msg + getPrintableString(m) + "\n"
}
msg = msg + "\n\n" + "#Stats"
msg = msg + "\n\n" + "#DailyStats"
toot := &mastodon.Toot{
Status: msg,