mirror of
https://github.com/pinpox/gitea-matrix-bot
synced 2024-11-26 08:43:47 +01:00
144 lines
3.2 KiB
Go
144 lines
3.2 KiB
Go
package main
|
|
|
|
import (
|
|
"crypto/rand"
|
|
"fmt"
|
|
"github.com/binaryplease/matrix-bot"
|
|
)
|
|
|
|
//GiteaBot is the main struct to hold the bot
|
|
type GiteaBot struct {
|
|
*matrixbot.MatrixBot
|
|
Subscriptions map[string][]string
|
|
}
|
|
|
|
//SendMessageToRooms sends a message to all roomes that have subscribed to the repo
|
|
func (gb *GiteaBot) SendMessageToRooms(repo, message string) {
|
|
for _, v := range gb.Subscriptions[repo] {
|
|
_, err = gb.Client.SendText(v, message)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func contains(s []string, e string) bool {
|
|
for _, a := range s {
|
|
if a == e {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
//NewGiteaBot creates a new bot form user credentials
|
|
func NewGiteaBot(user, pass string) *GiteaBot {
|
|
|
|
subs := make(map[string][]string)
|
|
|
|
bot, err := matrixbot.NewMatrixBot(user, pass)
|
|
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
gbot := &GiteaBot{
|
|
bot,
|
|
subs,
|
|
}
|
|
|
|
bot.RegisterCommand("!sub", 0, gbot.handleCommandAddSub)
|
|
bot.RegisterCommand("!unsub", 0, gbot.handleCommandRemoveSub)
|
|
bot.RegisterCommand("!listsubs", 0, gbot.handleCommandListSubs)
|
|
bot.RegisterCommand("!help", 0, gbot.handleCommandHelp)
|
|
bot.RegisterCommand("!secret", 0, gbot.handleCommandSecret)
|
|
|
|
return gbot
|
|
|
|
}
|
|
|
|
func tokenGenerator() string {
|
|
b := make([]byte, 4)
|
|
rand.Read(b)
|
|
return fmt.Sprintf("%x", b)
|
|
}
|
|
|
|
func (gb *GiteaBot) handleCommandSecret(message, room, sender string) {
|
|
//TODO
|
|
gb.SendToRoom(room, "NOT implemented yet")
|
|
}
|
|
|
|
func (gb *GiteaBot) handleCommandListSubs(message, room, sender string) {
|
|
|
|
repos := ""
|
|
|
|
for k, repo := range gb.Subscriptions {
|
|
for _, subscriber := range repo {
|
|
if subscriber == room {
|
|
repos = repos + "\n -" + k
|
|
}
|
|
}
|
|
}
|
|
|
|
if repos == "" {
|
|
gb.SendToRoom(room, "This room has not subscribed to any repositorys.")
|
|
} else {
|
|
msg := "This room has is subscribed to the following repositorys:" + repos
|
|
gb.SendToRoom(room, msg)
|
|
}
|
|
}
|
|
|
|
func (gb *GiteaBot) handleCommandAddSub(message string, room, sender string) {
|
|
|
|
repo := message[5:]
|
|
if !contains(gb.Subscriptions[repo], room) {
|
|
gb.Subscriptions[repo] = append(gb.Subscriptions[repo], room)
|
|
gb.SendToRoom(room, "Subscribed to: "+repo)
|
|
} else {
|
|
gb.SendToRoom(room, "This room has already subscribed to: "+repo)
|
|
}
|
|
}
|
|
|
|
func (gb *GiteaBot) handleCommandRemoveSub(message, room, sender string) {
|
|
|
|
repo := message[7:]
|
|
fmt.Println(gb.Subscriptions[repo])
|
|
|
|
if contains(gb.Subscriptions[repo], room) {
|
|
|
|
var tmp []string
|
|
|
|
for _, v := range gb.Subscriptions[repo] {
|
|
if v != room {
|
|
fmt.Println("readding '" + v + "'" + "because it is not equal to '" + room + "'")
|
|
tmp = append(tmp, v)
|
|
} else {
|
|
gb.SendToRoom(room, "Un-subscribed from: "+repo)
|
|
}
|
|
}
|
|
gb.Subscriptions[repo] = tmp
|
|
} else {
|
|
gb.SendToRoom(room, "This room has not subscribed to: "+repo)
|
|
}
|
|
|
|
}
|
|
|
|
func (gb *GiteaBot) handleCommandHelp(message, room, sender string) {
|
|
helpMsg := `
|
|
|
|
I'm your friendly Gitea Bot!
|
|
|
|
You can invite me to any matrix room to get updates on subscribed gitea repositorys.
|
|
The following commands are avaitible:
|
|
|
|
!sub user/repo Subscribe to a repository
|
|
!unsub user/repo Remove subscription to a repository
|
|
!listsubs List the room's subscriptions
|
|
!help Display this message
|
|
|
|
Some of the commands might require admin powers!
|
|
|
|
`
|
|
gb.SendToRoom(room, helpMsg)
|
|
}
|