96 lines
1.8 KiB
Go
96 lines
1.8 KiB
Go
package main
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
|
|
"git.ophivana.moe/cat/rpcfetch"
|
|
)
|
|
|
|
var (
|
|
pidS = strconv.Itoa(os.Getpid())
|
|
hostname string
|
|
launchTime = time.Now().Unix()
|
|
)
|
|
|
|
var substitute = map[string]func(s *applyState) string{
|
|
"pid": func(_ *applyState) string {
|
|
return pidS
|
|
},
|
|
"hostname": func(_ *applyState) string {
|
|
return hostname
|
|
},
|
|
"1min": func(s *applyState) string {
|
|
s.populateLoadavg()
|
|
return s.loadavg[0]
|
|
},
|
|
"5min": func(s *applyState) string {
|
|
s.populateLoadavg()
|
|
return s.loadavg[1]
|
|
},
|
|
"15min": func(s *applyState) string {
|
|
s.populateLoadavg()
|
|
return s.loadavg[2]
|
|
},
|
|
"used": func(s *applyState) string {
|
|
s.populateMem()
|
|
return fmt.Sprintf("%.1f GiB", float64(s.mem[1]-s.mem[0])/(1<<20))
|
|
},
|
|
|
|
"total": func(s *applyState) string {
|
|
s.populateMem()
|
|
return fmt.Sprintf("%.1f GiB", float64(s.mem[1])/(1<<20))
|
|
},
|
|
}
|
|
|
|
type applyState struct {
|
|
loadavg *[3]string
|
|
mem *[2]int
|
|
}
|
|
|
|
func (s *applyState) replace(t string) string {
|
|
for k, f := range substitute {
|
|
t = strings.ReplaceAll(t, "%"+k, f(s))
|
|
}
|
|
return t
|
|
}
|
|
|
|
func apply() {
|
|
confLock.RLock()
|
|
act := *conf.Activity
|
|
s := &applyState{}
|
|
|
|
act.State = s.replace(act.State)
|
|
act.Details = s.replace(act.Details)
|
|
act.Assets = &rpcfetch.ActivityAssets{
|
|
LargeImage: act.Assets.LargeImage,
|
|
LargeText: s.replace(act.Assets.LargeText),
|
|
SmallImage: act.Assets.SmallImage,
|
|
SmallText: s.replace(act.Assets.SmallText),
|
|
}
|
|
confLock.RUnlock()
|
|
|
|
if nonce, err := retry(&act); err != nil {
|
|
log.Fatalf("error setting activity: %s", err)
|
|
} else {
|
|
log.Printf("activity updated with nonce %s", nonce)
|
|
}
|
|
}
|
|
|
|
func retry(act *rpcfetch.Activity) (string, error) {
|
|
try:
|
|
nonce, err := d.SetActivity(act)
|
|
if errors.Is(err, rpcfetch.ErrAgain) {
|
|
log.Println("retrying in 5 seconds...")
|
|
time.Sleep(5 * time.Second)
|
|
goto try
|
|
}
|
|
|
|
return nonce, err
|
|
}
|