Add two new twitch chat handlers in v5.1.0

This commit is contained in:
Alex Vanin 2020-03-18 22:33:33 +03:00
parent d59b1c2d3b
commit 4d7d14e0b5
7 changed files with 400 additions and 3 deletions

View file

@ -1,5 +1,10 @@
# Changelog
## 5.1.0 - 2020-03-18
### Added
- PetCat twitch chat handler
- DailyEmote twitch chat handler
## 5.0.1 - 2020-03-08
### Changed
- Updated `go-twitch-irc` lib to v2.2.2

View file

@ -8,6 +8,7 @@ import (
"galched-bot/modules/discord"
"galched-bot/modules/grace"
"galched-bot/modules/patpet"
"galched-bot/modules/settings"
"galched-bot/modules/subday"
"galched-bot/modules/twitchat"
@ -87,7 +88,8 @@ func main() {
var err error
app := fx.New(
fx.Logger(new(silentPrinter)),
fx.Provide(settings.New, grace.New, discord.New, subday.New, twitchat.New, web.New, youtube.New),
fx.Provide(settings.New, grace.New, discord.New, subday.New,
twitchat.New, web.New, youtube.New, patpet.New),
fx.Invoke(start))
err = app.Start(context.Background())

88
modules/patpet/patpet.go Normal file
View file

@ -0,0 +1,88 @@
package patpet
import (
"encoding/json"
"fmt"
"io/ioutil"
"log"
"os"
"sync"
"galched-bot/modules/settings"
)
type (
Pet struct {
sync.RWMutex
path string
counter int
}
)
func New(s *settings.Settings) (*Pet, error) {
var (
err error
counter int
)
petData, err := ioutil.ReadFile(s.PetDataPath)
if err != nil {
log.Print("pet: cannot read data file", err)
log.Print("pet: creating new counter")
} else {
err = json.Unmarshal(petData, &counter)
if err != nil {
counter = 0
log.Print("pet: cannot unmarshal data file", err)
log.Print("pet: creating new counter")
} else {
log.Print("pet: using previously saved counter")
}
}
return &Pet{
RWMutex: sync.RWMutex{},
path: s.PetDataPath,
counter: counter,
}, nil
}
func (p *Pet) Pet() int {
p.Lock()
defer p.Unlock()
p.counter++
return p.counter
}
func (p *Pet) Counter() int {
p.RUnlock()
defer p.RUnlock()
return p.counter
}
func (p *Pet) Dump() {
p.RLock()
defer p.RUnlock()
data, err := json.Marshal(p.counter)
if err != nil {
log.Print("pet: cannot marshal counter", err)
return
}
file, err := os.Create(p.path)
if err != nil {
log.Print("pet: cannot open counter file", err)
return
}
_, err = fmt.Fprintf(file, string(data))
if err != nil {
log.Print("pet: cannot write to counter file")
}
err = file.Close()
if err != nil {
log.Print("pet: cannot close counter file")
}
log.Print("pet: counter dumped to file:", p.counter)
}

View file

@ -8,12 +8,13 @@ import (
)
const (
version = "5.0.1"
version = "5.1.0"
twitchUser = "galchedbot"
twitchIRCRoom = "galched"
discordTokenPath = "./tokens/.discordtoken"
twitchTokenPath = "./tokens/.twitchtoken"
subdayDataPath = "./backups/subday"
petDataPath = "./backups/pets"
youtubeTokenPath = "./tokens/.youtubetoken"
webLoginsPath = "./tokens/.weblogins"
@ -43,6 +44,7 @@ type (
TwitchToken string
YoutubeToken string
SubdayDataPath string
PetDataPath string
PermittedRoles []string
DiscordVoiceChannel string
Songs []SongInfo
@ -85,6 +87,7 @@ func New() (*Settings, error) {
TwitchUser: twitchUser,
TwitchIRCRoom: twitchIRCRoom,
SubdayDataPath: subdayDataPath,
PetDataPath: petDataPath,
DiscordVoiceChannel: "301793085522706432",
PermittedRoles: []string{subRole1, subRole2, galchedRole, smorcRole},
Songs: []SongInfo{

View file

@ -0,0 +1,257 @@
package twitchat
import (
"fmt"
"hash/fnv"
"math/rand"
"time"
"github.com/gempir/go-twitch-irc/v2"
)
const emoteMsg = "!emote"
var (
emotes = []string{
"4Head",
"ANELE",
"ArgieB8",
"ArsonNoSexy",
"AsexualPride",
"AsianGlow",
"BCWarrior",
"BOP",
"BabyRage",
"BatChest",
"BegWan",
"BibleThump",
"BigBrother",
"BigPhish",
"BisexualPride",
"BlargNaut",
"BlessRNG",
"BloodTrail",
"BrainSlug",
"BrokeBack",
"BuddhaBar",
"CarlSmile",
"ChefFrank",
"CoolCat",
"CoolStoryBob",
"CorgiDerp",
"CrreamAwk",
"CurseLit",
"DAESuppy",
"DBstyle",
"DansGame",
"DarkMode",
"DatSheffy",
"DendiFace",
"DogFace",
"DoritosChip",
"DrinkPurple",
"DxCat",
"EarthDay",
"EleGiggle",
"EntropyWins",
"FBBlock",
"FBCatch",
"FBChallenge",
"FBPass",
"FBPenalty",
"FBRun",
"FBSpiral",
"FBtouchdown",
"FUNgineer",
"FailFish",
"FrankerZ",
"FreakinStinkin",
"FutureMan",
"GayPride",
"GenderFluidPride",
"GingerPower",
"GivePLZ",
"GrammarKing",
"GreenTeam",
"GunRun",
"HSCheers",
"HSWP",
"HassaanChop",
"HassanChop",
"HeyGuys",
"HolidayCookie",
"HolidayLog",
"HolidayOrnament",
"HolidayPresent",
"HolidaySanta",
"HolidayTree",
"HotPokket",
"HumbleLife",
"IntersexPride",
"InuyoFace",
"ItsBoshyTime",
"JKanStyle",
"Jebaited",
"JonCarnage",
"KAPOW",
"Kappa",
"KappaClaus",
"KappaPride",
"KappaRoss",
"KappaWealth",
"Kappu",
"Keepo",
"KevinTurtle",
"Kippa",
"KomodoHype",
"KonCha",
"Kreygasm",
"LUL",
"LesbianPride",
"MVGame",
"Mau5",
"MaxLOL",
"MercyWing1",
"MercyWing2",
"MikeHogu",
"MingLee",
"MorphinTime",
"MrDestructoid",
"NinjaGrumpy",
"NomNom",
"NonBinaryPride",
"NotATK",
"NotLikeThis",
"OSFrog",
"OhMyDog",
"OneHand",
"OpieOP",
"OptimizePrime",
"PJSalt",
"PJSugar",
"PMSTwin",
"PRChase",
"PanicVis",
"PansexualPride",
"PartyHat",
"PartyTime",
"PeoplesChamp",
"PermaSmug",
"PicoMause",
"PinkMercy",
"PipeHype",
"PixelBob",
"PogChamp",
"Poooound",
"PopCorn",
"PorscheWIN",
"PowerUpL",
"PowerUpR",
"PraiseIt",
"PrimeMe",
"PunOko",
"PunchTrees",
"PurpleStar",
"RaccAttack",
"RalpherZ",
"RedCoat",
"RedTeam",
"ResidentSleeper",
"RitzMitz",
"RlyTho",
"RuleFive",
"SMOrc",
"SSSsss",
"SabaPing",
"SeemsGood",
"SeriousSloth",
"ShadyLulu",
"ShazBotstix",
"SingsMic",
"SingsNote",
"SmoocherZ",
"SoBayed",
"SoonerLater",
"Squid1",
"Squid2",
"Squid3",
"Squid4",
"StinkyCheese",
"StoneLightning",
"StrawBeary",
"SuperVinlin",
"SwiftRage",
"TBAngel",
"TF2John",
"TPFufun",
"TPcrunchyroll",
"TTours",
"TakeNRG",
"TearGlove",
"TehePelo",
"ThankEgg",
"TheIlluminati",
"TheRinger",
"TheTarFu",
"TheThing",
"ThunBeast",
"TinyFace",
"TombRaid",
"TooSpicy",
"TransgenderPride",
"TriHard",
"TwitchLit",
"TwitchRPG",
"TwitchSings",
"TwitchUnity",
"TwitchVotes",
"UWot",
"UnSane",
"UncleNox",
"VoHiYo",
"VoteNay",
"VoteYea",
"WTRuck",
"WholeWheat",
"WutFace",
"YouDontSay",
"YouWHY",
"bleedPurple",
"cmonBruh",
"copyThis",
"duDudu",
"imGlitch",
"mcaT",
"panicBasket",
"pastaThat",
"riPepperonis",
"twitchRaid",
}
)
type (
dailyEmote struct{}
)
func DailyEmote() *dailyEmote {
return new(dailyEmote)
}
func (h *dailyEmote) IsValid(m *twitch.PrivateMessage) bool {
return (m.Tags["msg-id"] == "highlighted-message") && m.Message == emoteMsg
}
func (h *dailyEmote) Handle(m *twitch.PrivateMessage, r Responser) {
data := time.Now().Format("2006-01-02") + m.User.DisplayName
rng := rand.New(rand.NewSource(hashSeed(data)))
emote := emotes[rng.Intn(len(emotes))]
msg := fmt.Sprintf("@%s твой эмоут дня: %s", m.User.DisplayName, emote)
r.Say(m.Channel, msg)
}
func hashSeed(s string) int64 {
h := fnv.New64()
h.Write([]byte(s))
return int64(h.Sum64())
}

View file

@ -0,0 +1,39 @@
package twitchat
import (
"fmt"
"strings"
"galched-bot/modules/patpet"
"github.com/gempir/go-twitch-irc/v2"
)
const (
petMsg1 = "!погладь"
petMsg2 = "!гладь"
petMsg3 = "!погладить"
)
type (
petCat struct {
cat *patpet.Pet
}
)
func PetCat(pet *patpet.Pet) *petCat {
return &petCat{
cat: pet,
}
}
func (h *petCat) IsValid(m *twitch.PrivateMessage) bool {
return (m.Tags["msg-id"] == "highlighted-message") && (strings.HasPrefix(m.Message, petMsg1) ||
strings.HasPrefix(m.Message, petMsg2) ||
strings.HasPrefix(m.Message, petMsg3))
}
func (h *petCat) Handle(m *twitch.PrivateMessage, r Responser) {
msg := fmt.Sprintf("Котэ поглажен уже %d раз(а) InuyoFace", h.cat.Pet())
r.Say(m.Channel, msg)
h.cat.Dump()
}

View file

@ -1,6 +1,7 @@
package twitchat
import (
"galched-bot/modules/patpet"
"galched-bot/modules/settings"
"galched-bot/modules/youtube"
@ -15,13 +16,15 @@ type (
}
)
func New(s *settings.Settings, r *youtube.Requester) (*TwitchIRC, error) {
func New(s *settings.Settings, r *youtube.Requester, pet *patpet.Pet) (*TwitchIRC, error) {
var irc = new(TwitchIRC)
irc.username = s.TwitchUser
irc.handlers = append(irc.handlers, DupHandler())
irc.handlers = append(irc.handlers, DailyEmote())
irc.handlers = append(irc.handlers, SongRequest(r))
irc.handlers = append(irc.handlers, PetCat(pet))
// irc.handlers = append(irc.handlers, LogCheck())
irc.chat = twitch.NewClient(s.TwitchUser, s.TwitchToken)