|
|
|
@@ -10,8 +10,6 @@ import ( |
|
|
|
"math/rand" |
|
|
|
"net" |
|
|
|
"net/textproto" |
|
|
|
"os" |
|
|
|
"path/filepath" |
|
|
|
"regexp" |
|
|
|
"strings" |
|
|
|
"time" |
|
|
|
@@ -19,188 +17,6 @@ import ( |
|
|
|
rgb "github.com/foresthoffman/rgblog" |
|
|
|
) |
|
|
|
|
|
|
|
var karaokards = [...]string{ |
|
|
|
"Chorus contains a name", |
|
|
|
"Refers to an animal", |
|
|
|
"Aggressive", |
|
|
|
"Contains instructions", |
|
|
|
"Refers to relationships", |
|
|
|
"Folk", |
|
|
|
"Novelty", |
|
|
|
"Pre-sixties", |
|
|
|
"1960s", |
|
|
|
"Ballad", |
|
|
|
"Musical", |
|
|
|
"Heartbroken", |
|
|
|
"Contains made-up words", |
|
|
|
"Refers to a colour", |
|
|
|
"Chorus contains me, mine or my", |
|
|
|
"Chorus contains want, need or have", |
|
|
|
"Chorus contains how, when or why", |
|
|
|
"Refers to money", |
|
|
|
"Chorus contains love, like or hate", |
|
|
|
"Chorus contains man, woman or everybody", |
|
|
|
"Anthemic", |
|
|
|
"Refers to explosives", |
|
|
|
"Chorus contains a number", |
|
|
|
"1970s", |
|
|
|
"2000s", |
|
|
|
"Love Song", |
|
|
|
"Country-pop", |
|
|
|
"Grunge", |
|
|
|
"Indie", |
|
|
|
"This is a morality tale", |
|
|
|
"I've never sung this before", |
|
|
|
"They'd beat me in a fight", |
|
|
|
"My mortal enemy", |
|
|
|
"Their name starts with the same letter as mine", |
|
|
|
"This is NOT my era", |
|
|
|
"Chorus contains oh, ooh or baby", |
|
|
|
"Samples another song", |
|
|
|
"Chorus contains don't, won't or can't", |
|
|
|
"Euphoric", |
|
|
|
"Title contains day, night or tomorrow", |
|
|
|
"Chorus contains boy, girl or child", |
|
|
|
"Refers to space", |
|
|
|
"Soul", |
|
|
|
"R&B", |
|
|
|
"Dance", |
|
|
|
"Electronie", |
|
|
|
"Pop", |
|
|
|
"1990s", |
|
|
|
"Rock", |
|
|
|
"Title contains brackets", |
|
|
|
"Refers to religion", |
|
|
|
"Refers to music", |
|
|
|
"Chorus contains this, that or there", |
|
|
|
"Chorus contains up, down or over", |
|
|
|
"Beautiful", |
|
|
|
"Mean about someone", |
|
|
|
"Refers to weather", |
|
|
|
"Chorus contains you, your or you're", |
|
|
|
"Gloomy", |
|
|
|
"Contains questions", |
|
|
|
"Refers to death", |
|
|
|
"Refers to sleep", |
|
|
|
"Chorus contains heart, head or soul", |
|
|
|
"Rock", |
|
|
|
"Pop", |
|
|
|
"Country", |
|
|
|
"Punk", |
|
|
|
"Rap", |
|
|
|
"Christmas", |
|
|
|
"Motown", |
|
|
|
"This is their Second-best song", |
|
|
|
"I shouldn't know this song. But I do!", |
|
|
|
"I don't need the screen!", |
|
|
|
"I'm too old for this song", |
|
|
|
"The story of my life", |
|
|
|
"I love this song SO much", |
|
|
|
"They're twice my age!", |
|
|
|
"Chorus contains I, I’m or I’ve", |
|
|
|
"Title is one word long", |
|
|
|
"Soundtrack", |
|
|
|
"Pop", |
|
|
|
"Is a metaphor", |
|
|
|
"Title is at least five words long", |
|
|
|
"Chorus contains move, stay or go", |
|
|
|
"Refers to a place", |
|
|
|
"R&B", |
|
|
|
"Metal", |
|
|
|
"Good workout music", |
|
|
|
"Requires audience participation", |
|
|
|
"Power Ballad", |
|
|
|
"1980s", |
|
|
|
"Rock", |
|
|
|
"Rap", |
|
|
|
"Pop-punk", |
|
|
|
"Alternative", |
|
|
|
"Soul", |
|
|
|
"My secret shame", |
|
|
|
"This gets me a bit emotional", |
|
|
|
"Out of my range", |
|
|
|
"This person is bad at their job", |
|
|
|
"They look like someone here", |
|
|
|
"I don't know the verse", |
|
|
|
"They're half my age!", |
|
|
|
"Play this at my funeral", |
|
|
|
"This is NOT my genre", |
|
|
|
"I hate this song so much", |
|
|
|
"Girl band", |
|
|
|
"Male solo", |
|
|
|
"Artist begins with A", |
|
|
|
"Just one word", |
|
|
|
"Mixed gender band", |
|
|
|
"Artist begins with C", |
|
|
|
"Artist begins with G", |
|
|
|
"Two people", |
|
|
|
"TV contestant", |
|
|
|
"European", |
|
|
|
"Artist begins with P", |
|
|
|
"Artist begins with M", |
|
|
|
"In a famous family", |
|
|
|
"Male-fronted band", |
|
|
|
"Australian", |
|
|
|
"One-hit wonder", |
|
|
|
"Female-fronted band", |
|
|
|
"Artist begins with T", |
|
|
|
"Rock", |
|
|
|
"Indie-rock", |
|
|
|
"R&B", |
|
|
|
"Latin", |
|
|
|
"Disco", |
|
|
|
"Britpop", |
|
|
|
"2010s", |
|
|
|
"I was a teenager!", |
|
|
|
"The music video is so good", |
|
|
|
"I’m younger than this song", |
|
|
|
"First dance at my imaginary wedding", |
|
|
|
"I'm worried about them", |
|
|
|
"This person is the best dancer", |
|
|
|
"I know the dance", |
|
|
|
"Mostly shouting", |
|
|
|
"They're not my gender", |
|
|
|
"I wish we were married", |
|
|
|
"I played this song too often", |
|
|
|
"They are so influential", |
|
|
|
"My favourite song of theirs", |
|
|
|
"Perfect montage music", |
|
|
|
"Artist uses their surname", |
|
|
|
"Famous partner", |
|
|
|
"Artist begins with E", |
|
|
|
"Troubled artist", |
|
|
|
"Actor", |
|
|
|
"Artist begins with F", |
|
|
|
"Solo artist", |
|
|
|
"Artist begins with R", |
|
|
|
"Hellraiser", |
|
|
|
"Artist begins with ‘The’", |
|
|
|
"10+ year career", |
|
|
|
"Boy band", |
|
|
|
"Female solo", |
|
|
|
"British", |
|
|
|
"Inappropriately clothed", |
|
|
|
"Award winners", |
|
|
|
"Artist begins with S", |
|
|
|
"Artist begins with W", |
|
|
|
"Asian", |
|
|
|
"They split up :(", |
|
|
|
"North American", |
|
|
|
"Pop", |
|
|
|
"Goth or Emo", |
|
|
|
"This is a bit creepy, frankly", |
|
|
|
"I’ve seen them in real life", |
|
|
|
"A beautiful love story", |
|
|
|
"OK, this is just ridiculous", |
|
|
|
"Artist begins with B", |
|
|
|
"Artist begins with D", |
|
|
|
"They’re dead :(", |
|
|
|
"Artist begins with L", |
|
|
|
"Amazing hair", |
|
|
|
"Artist begins with J"} |
|
|
|
|
|
|
|
var selectablePrompts []string |
|
|
|
|
|
|
|
const PSTFormat = "Jan 2 15:04:05 PST" |
|
|
|
|
|
|
|
// Regex for parsing PRIVMSG strings. |
|
|
|
@@ -234,36 +50,37 @@ type KardBot struct { |
|
|
|
PrivatePath string |
|
|
|
Server string |
|
|
|
startTime time.Time |
|
|
|
Prompts []string |
|
|
|
} |
|
|
|
|
|
|
|
// Connects the bot to the Twitch IRC server. The bot will continue to try to connect until it |
|
|
|
// succeeds or is forcefully shutdown. |
|
|
|
func (bb *kardBot) Connect() { |
|
|
|
func (bb *KardBot) Connect() { |
|
|
|
var err error |
|
|
|
rgb.YPrintf("[%s] Connecting to %s...\n", timeStamp(), bb.Server) |
|
|
|
rgb.YPrintf("[%s] Connecting to %s...\n", TimeStamp(), bb.Server) |
|
|
|
|
|
|
|
// makes connection to Twitch IRC server |
|
|
|
bb.conn, err = net.Dial("tcp", bb.Server+":"+bb.Port) |
|
|
|
if nil != err { |
|
|
|
rgb.YPrintf("[%s] Cannot connect to %s, retrying.\n", timeStamp(), bb.Server) |
|
|
|
rgb.YPrintf("[%s] Cannot connect to %s, retrying.\n", TimeStamp(), bb.Server) |
|
|
|
bb.Connect() |
|
|
|
return |
|
|
|
} |
|
|
|
rgb.YPrintf("[%s] Connected to %s!\n", timeStamp(), bb.Server) |
|
|
|
rgb.YPrintf("[%s] Connected to %s!\n", TimeStamp(), bb.Server) |
|
|
|
bb.startTime = time.Now() |
|
|
|
} |
|
|
|
|
|
|
|
// Officially disconnects the bot from the Twitch IRC server. |
|
|
|
func (bb *kardBot) Disconnect() { |
|
|
|
func (bb *KardBot) Disconnect() { |
|
|
|
bb.conn.Close() |
|
|
|
upTime := time.Now().Sub(bb.startTime).Seconds() |
|
|
|
rgb.YPrintf("[%s] Closed connection from %s! | Live for: %fs\n", timeStamp(), bb.Server, upTime) |
|
|
|
rgb.YPrintf("[%s] Closed connection from %s! | Live for: %fs\n", TimeStamp(), bb.Server, upTime) |
|
|
|
} |
|
|
|
|
|
|
|
// Listens for and logs messages from chat. Responds to commands from the channel owner. The bot |
|
|
|
// continues until it gets disconnected, told to shutdown, or forcefully shutdown. |
|
|
|
func (bb *kardBot) HandleChat() error { |
|
|
|
rgb.YPrintf("[%s] Watching #%s...\n", timeStamp(), bb.Channel) |
|
|
|
func (bb *KardBot) HandleChat() error { |
|
|
|
rgb.YPrintf("[%s] Watching #%s...\n", TimeStamp(), bb.Channel) |
|
|
|
|
|
|
|
// reads from connection |
|
|
|
tp := textproto.NewReader(bufio.NewReader(bb.conn)) |
|
|
|
@@ -280,7 +97,7 @@ func (bb *kardBot) HandleChat() error { |
|
|
|
} |
|
|
|
|
|
|
|
// logs the response from the IRC server |
|
|
|
rgb.YPrintf("[%s] %s\n", timeStamp(), line) |
|
|
|
rgb.YPrintf("[%s] %s\n", TimeStamp(), line) |
|
|
|
|
|
|
|
if "PING :tmi.twitch.tv" == line { |
|
|
|
|
|
|
|
@@ -298,7 +115,7 @@ func (bb *kardBot) HandleChat() error { |
|
|
|
switch msgType { |
|
|
|
case "PRIVMSG": |
|
|
|
msg := matches[3] |
|
|
|
rgb.GPrintf("[%s] %s: %s\n", timeStamp(), userName, msg) |
|
|
|
rgb.GPrintf("[%s] %s: %s\n", TimeStamp(), userName, msg) |
|
|
|
|
|
|
|
// parse commands from user message |
|
|
|
cmdMatches := CmdRegex.FindStringSubmatch(msg) |
|
|
|
@@ -307,9 +124,9 @@ func (bb *kardBot) HandleChat() error { |
|
|
|
|
|
|
|
switch cmd { |
|
|
|
case "card": |
|
|
|
rgb.CPrintf("[%s] Card asked for!\n", timeStamp()) |
|
|
|
rgb.CPrintf("[%s] Card asked for!\n", TimeStamp()) |
|
|
|
|
|
|
|
bb.Say("Your prompt is : " + selectablePrompts[rand.Intn(len(selectablePrompts))]) |
|
|
|
bb.Say("Your prompt is : " + bb.Prompts[rand.Intn(len(bb.Prompts))]) |
|
|
|
} |
|
|
|
|
|
|
|
// channel-owner specific commands |
|
|
|
@@ -318,7 +135,7 @@ func (bb *kardBot) HandleChat() error { |
|
|
|
case "tbdown": |
|
|
|
rgb.CPrintf( |
|
|
|
"[%s] Shutdown command received. Shutting down now...\n", |
|
|
|
timeStamp(), |
|
|
|
TimeStamp(), |
|
|
|
) |
|
|
|
|
|
|
|
bb.Disconnect() |
|
|
|
@@ -338,17 +155,17 @@ func (bb *kardBot) HandleChat() error { |
|
|
|
} |
|
|
|
|
|
|
|
// Makes the bot join its pre-specified channel. |
|
|
|
func (bb *kardBot) JoinChannel() { |
|
|
|
rgb.YPrintf("[%s] Joining #%s...\n", timeStamp(), bb.Channel) |
|
|
|
func (bb *KardBot) JoinChannel() { |
|
|
|
rgb.YPrintf("[%s] Joining #%s...\n", TimeStamp(), bb.Channel) |
|
|
|
bb.conn.Write([]byte("PASS " + bb.Credentials.Password + "\r\n")) |
|
|
|
bb.conn.Write([]byte("NICK " + bb.Name + "\r\n")) |
|
|
|
bb.conn.Write([]byte("JOIN #" + bb.Channel + "\r\n")) |
|
|
|
|
|
|
|
rgb.YPrintf("[%s] Joined #%s as @%s!\n", timeStamp(), bb.Channel, bb.Name) |
|
|
|
rgb.YPrintf("[%s] Joined #%s as @%s!\n", TimeStamp(), bb.Channel, bb.Name) |
|
|
|
} |
|
|
|
|
|
|
|
// Reads from the private credentials file and stores the data in the bot's Credentials field. |
|
|
|
func (bb *kardBot) ReadCredentials() error { |
|
|
|
func (bb *KardBot) ReadCredentials() error { |
|
|
|
|
|
|
|
// reads from the file |
|
|
|
credFile, err := ioutil.ReadFile(bb.PrivatePath) |
|
|
|
@@ -368,7 +185,7 @@ func (bb *kardBot) ReadCredentials() error { |
|
|
|
} |
|
|
|
|
|
|
|
// Makes the bot send a message to the chat channel. |
|
|
|
func (bb *kardBot) Say(msg string) error { |
|
|
|
func (bb *KardBot) Say(msg string) error { |
|
|
|
if "" == msg { |
|
|
|
return errors.New("BasicBot.Say: msg was empty.") |
|
|
|
} |
|
|
|
@@ -388,7 +205,7 @@ func (bb *kardBot) Say(msg string) error { |
|
|
|
// Starts a loop where the bot will attempt to connect to the Twitch IRC server, then connect to the |
|
|
|
// pre-specified channel, and then handle the chat. It will attempt to reconnect until it is told to |
|
|
|
// shut down, or is forcefully shutdown. |
|
|
|
func (bb *kardBot) Start() { |
|
|
|
func (bb *KardBot) Start() { |
|
|
|
err := bb.ReadCredentials() |
|
|
|
if nil != err { |
|
|
|
fmt.Println(err) |
|
|
|
@@ -412,38 +229,10 @@ func (bb *kardBot) Start() { |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
func timeStamp() string { |
|
|
|
return TimeStamp(PSTFormat) |
|
|
|
func TimeStamp() string { |
|
|
|
return TimeStampFmt(PSTFormat) |
|
|
|
} |
|
|
|
|
|
|
|
func TimeStamp(format string) string { |
|
|
|
func TimeStampFmt(format string) string { |
|
|
|
return time.Now().Format(format) |
|
|
|
} |
|
|
|
|
|
|
|
type customStringsStruct struct { |
|
|
|
Strings []string `json:"strings,omitempty"` |
|
|
|
} |
|
|
|
|
|
|
|
var customStrings customStringsStruct |
|
|
|
|
|
|
|
func readBonusStrings() []string { |
|
|
|
|
|
|
|
ex, err := os.Executable() |
|
|
|
if err != nil { |
|
|
|
return []string{} |
|
|
|
fmt.Println("Could not read `strings.json`, will only have builtin prompts. File reading error", err) |
|
|
|
} |
|
|
|
exPath := filepath.Dir(ex) |
|
|
|
data, err := ioutil.ReadFile(exPath + "/strings.json") |
|
|
|
if err != nil { |
|
|
|
fmt.Println("Could not read `strings.json`, will only have builtin prompts. File reading error", err) |
|
|
|
return []string{} |
|
|
|
} |
|
|
|
err = json.Unmarshal(data, &customStrings) |
|
|
|
if err != nil { |
|
|
|
fmt.Println("Could not unmarshal `strings.json`, will only have builtin prompts. Unmarshal error", err) |
|
|
|
return []string{} |
|
|
|
} |
|
|
|
fmt.Println("Read ", len(customStrings.Strings), " prompts from `strings.json`") |
|
|
|
return customStrings.Strings |
|
|
|
} |
