David Haukeness
5 years ago
7 changed files with 244 additions and 3 deletions
@ -0,0 +1,62 @@ |
|||||||
|
package main |
||||||
|
|
||||||
|
import ( |
||||||
|
"fmt" |
||||||
|
|
||||||
|
"github.com/kf5grd/keybasebot" |
||||||
|
"samhofi.us/x/keybase/v2" |
||||||
|
"samhofi.us/x/keybase/v2/types/chat1" |
||||||
|
) |
||||||
|
|
||||||
|
func nsfwFilter() keybasebot.Adapter { |
||||||
|
return func(action keybasebot.BotAction) keybasebot.BotAction { |
||||||
|
return func(m chat1.MsgSummary, b *keybasebot.Bot) (bool, error) { |
||||||
|
// if its not a team, nsfw is always allowed
|
||||||
|
if m.Channel.MembersType == keybase.USER { |
||||||
|
b.Logger.Debug("boobs are allowed in %s", m.Channel.Name) |
||||||
|
return action(m, b) |
||||||
|
} |
||||||
|
|
||||||
|
// if the team is in the allowed list, nsfw is allowed
|
||||||
|
if m.Channel.MembersType == keybase.TEAM { |
||||||
|
data, ok := b.Meta["nsfwAllowed"] |
||||||
|
if !ok { |
||||||
|
err := fmt.Errorf("Unable to fetch allowed teams from meta") |
||||||
|
return false, err |
||||||
|
} |
||||||
|
allowedTeams := data.(map[chat1.ConvIDStr]bool) |
||||||
|
if isAllowed := allowedTeams[m.ConvID]; isAllowed { |
||||||
|
return action(m, b) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// if the team is not allowed, fail silently
|
||||||
|
b.Logger.Debug("boobs are not allowed in %s", m.Channel.Name) |
||||||
|
return true, nil |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
func requireOwner() keybasebot.Adapter { |
||||||
|
return func(action keybasebot.BotAction) keybasebot.BotAction { |
||||||
|
return func(m chat1.MsgSummary, b *keybasebot.Bot) (bool, error) { |
||||||
|
b.Logger.Debug("checking owner command auth") |
||||||
|
// if the team is in the allowed list, nsfw is allowed
|
||||||
|
data, ok := b.Meta["owner"] |
||||||
|
if !ok { |
||||||
|
b.Logger.Debug("unable to fetch owner from meta") |
||||||
|
err := fmt.Errorf("owner command used without owner set") |
||||||
|
return false, err |
||||||
|
} |
||||||
|
myOwner := data.(string) |
||||||
|
if m.Sender.Username == myOwner { |
||||||
|
b.Logger.Debug("owner command approved") |
||||||
|
return action(m, b) |
||||||
|
} |
||||||
|
|
||||||
|
// if the team is not allowed, fail silently
|
||||||
|
b.Logger.Info("@%s tried to user owner command without auth.", m.Sender.Username) |
||||||
|
return true, nil |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,33 @@ |
|||||||
|
package cmd |
||||||
|
|
||||||
|
import ( |
||||||
|
"os" |
||||||
|
|
||||||
|
"github.com/kf5grd/keybasebot" |
||||||
|
"github.com/teris-io/shortid" |
||||||
|
"samhofi.us/x/keybase/v2/types/chat1" |
||||||
|
) |
||||||
|
|
||||||
|
/*var BoobsAd = chat1.UserBotCommandInput{ |
||||||
|
Name: "eyebleach", |
||||||
|
Usage: "", |
||||||
|
Description: "Sends some cute critters", |
||||||
|
}*/ |
||||||
|
|
||||||
|
func Boobs(m chat1.MsgSummary, b *keybasebot.Bot) (bool, error) { |
||||||
|
path, desc, err := getRandomRedditMedia(b, "boobs", "top", 100) |
||||||
|
if err != nil { |
||||||
|
eid := shortid.MustGenerate() |
||||||
|
b.Logger.Error("%s: %+v", eid, err) |
||||||
|
b.KB.ReactByConvID(m.ConvID, m.Id, "Error: %s", eid) |
||||||
|
return true, nil |
||||||
|
} |
||||||
|
// upload the image
|
||||||
|
b.KB.UploadToConversation(m.ConvID, desc, path) |
||||||
|
// delete the file
|
||||||
|
err = os.Remove(path) |
||||||
|
if err != nil { |
||||||
|
b.Logger.Error("Unable to remove file %s", path) |
||||||
|
} |
||||||
|
return true, nil |
||||||
|
} |
||||||
@ -0,0 +1,67 @@ |
|||||||
|
package main |
||||||
|
|
||||||
|
import ( |
||||||
|
"github.com/kf5grd/keybasebot" |
||||||
|
"github.com/kf5grd/keybasebot/pkg/kvstore" |
||||||
|
"samhofi.us/x/keybase/v2/types/chat1" |
||||||
|
) |
||||||
|
|
||||||
|
func setNsfw(m chat1.MsgSummary, b *keybasebot.Bot) (bool, error) { |
||||||
|
b.KB.SendMessageByConvID(m.ConvID, "Pong!") |
||||||
|
b.Logger.Info("owner command in convid %s", m.ConvID) |
||||||
|
return true, nil |
||||||
|
} |
||||||
|
|
||||||
|
func addNsfwTeam(b *keybasebot.Bot, c chat1.ConvIDStr) bool { |
||||||
|
allowedTeams := make(map[chat1.ConvIDStr]bool) |
||||||
|
// get the allowed teams
|
||||||
|
data, ok := b.Meta["nsfwAllowed"] |
||||||
|
if !ok { |
||||||
|
// the key doesn't exist yet so go ahead and create it
|
||||||
|
allowedTeams[c] = true |
||||||
|
} else { |
||||||
|
// the key exists and needs to be updated
|
||||||
|
allowedTeams := data.(map[chat1.ConvIDStr]bool) |
||||||
|
allowedTeams[c] = true |
||||||
|
} |
||||||
|
// write the key to meta
|
||||||
|
b.Meta["nsfwAllowed"] = allowedTeams |
||||||
|
// write the key to kvStore if enabled
|
||||||
|
data, ok = b.Meta["kvstore"] |
||||||
|
if ok { |
||||||
|
team := data.(string) |
||||||
|
kvKey := kvstore.New("nsfwAllowed", allowedTeams, -1) |
||||||
|
err := kvstore.Put(b.KB, team, "ssh0le", kvKey) |
||||||
|
if err != nil { |
||||||
|
return false |
||||||
|
} |
||||||
|
} |
||||||
|
return true |
||||||
|
} |
||||||
|
|
||||||
|
func delNsfwTeam(b *keybasebot.Bot, c chat1.ConvIDStr) bool { |
||||||
|
allowedTeams := make(map[chat1.ConvIDStr]bool) |
||||||
|
// get the allowed teams
|
||||||
|
data, ok := b.Meta["nsfwAllowed"] |
||||||
|
if !ok { |
||||||
|
// the key doesn't exist yet so go return
|
||||||
|
return true |
||||||
|
} else { |
||||||
|
// the key exists and needs to be updated
|
||||||
|
allowedTeams := data.(map[chat1.ConvIDStr]bool) |
||||||
|
delete(allowedTeams, c) |
||||||
|
} |
||||||
|
// write the key to meta
|
||||||
|
b.Meta["nsfwAllowed"] = allowedTeams |
||||||
|
// write the key to kvStore if enabled
|
||||||
|
data, ok = b.Meta["kvstore"] |
||||||
|
if ok { |
||||||
|
team := data.(string) |
||||||
|
kvKey := kvstore.New("nsfwAllowed", allowedTeams, -1) |
||||||
|
err := kvstore.Put(b.KB, team, "ssh0le", kvKey) |
||||||
|
if err != nil { |
||||||
|
return false |
||||||
|
} |
||||||
|
} |
||||||
|
return true |
||||||
|
} |
||||||
Loading…
Reference in new issue