From 5d68781df11bb8ca5aae83b81414fe515cfb5646 Mon Sep 17 00:00:00 2001 From: Tearzz Date: Thu, 19 Apr 2018 15:26:24 +0200 Subject: [PATCH] Updated TwitchBot.rb Removed command handling. (Moved to CommandsHandler.rb) Changed console commands. --- TwitchBot.rb | 144 ++++++++++++++------------------------------------- 1 file changed, 40 insertions(+), 104 deletions(-) diff --git a/TwitchBot.rb b/TwitchBot.rb index 725ecb1..b0f9dde 100644 --- a/TwitchBot.rb +++ b/TwitchBot.rb @@ -8,12 +8,14 @@ class String end # Requied packages / modules -require 'socket' -require 'logger' -require 'open-uri' +require('socket') +require('logger') # Create logger -File.delete("log.txt") # Clear previous log +if File.file?("log.txt") + File.delete("log.txt") # Clear previous log +end + log = Logger.new("log.txt", formatter: proc {|severity, datetime, progname, msg| "#{datetime}: #{msg}\n"}) @@ -21,27 +23,21 @@ log = Logger.new("log.txt", formatter: proc {|severity, datetime, progname, msg| load "credentials.txt" log.info("Loading \"credentials.txt\"") +require_relative('CommandsHandler') # File for handling commands +log.info("Loading \"CommandsHandler.rb\"") + # -------- IGNORE -------- # OAUTH.downcase! BOTNAME.downcase! CHANNEL.downcase!.gsub!("#", "") -# -------- IGNORE -------- # # Save "Preparing to connect" to "log.txt" log.info("Preparing to connect") # Variables socket = TCPSocket.new('irc.chat.twitch.tv', 6667) -send = "PRIVMSG ##{CHANNEL} :" # shortcut for sending messages -running = true -content = nil -message_count = 0 -message_limit = Time.now.to_i -# Commands -commands = ["!examplecommand","!ping"] # Add commands here -api_commands = ["!followed","!uptime"] # API commands here -admin_commands = ["!disconnect"] # Channel owner commands here +running = true # Authorization Login socket.puts("PASS #{OAUTH}") # Send the password(oauth) to Twitch @@ -49,117 +45,57 @@ socket.puts("NICK #{BOTNAME}") # Send the botname to Twitch socket.puts("JOIN ##{CHANNEL}") # Send the channel to Twitch # Save "Connected!" to "log.txt -log.info("Joining #{CHANNEL.capitalize} as #{BOTNAME.capitalize} using OAUTH Token: #{OAUTH[6,OAUTH.length-12]}" + "*"*12) -# This will add "Joining Somechannel as Bot using OAUTH Token: OAUTH:SomeTokenHereWithHid************" to log.txt +log.info("Joining #{CHANNEL.capitalize} as #{BOTNAME.capitalize} using OAUTH Token: #{OAUTH[6,OAUTH.length-16]}" + "*"*16) Thread.abort_on_exception = true # Loop (Background Thread) for recieving Twitch chat data Thread.start do - socket.puts(send + "Connected!") # Send "Connected!" to the Twitch channel - puts "#{BOTNAME} Joined ##{CHANNEL}" # Connection Status - puts "You should be fully connected now" # Connection Status + socket.puts("PRIVMSG ##{CHANNEL} :Connected! PogChamp") # Send "Connected!" to the Twitch channel + log.info("Connected to chat!") + + puts "#{BOTNAME} Joined ##{CHANNEL}" # Connection Status + puts "You should be fully connected now" # Connection Status puts "" puts "Type \"clear\" to clear terminal" + puts "Type \"project \" to write to project file" + puts "Type \"disconnect\" to disconnect" puts "" while (running) do ready = IO.select([socket]) ready[0].each do |s| line = s.gets - # Respond to Twitch IRC "PING" Message - if line.index("PING") == 0 - line.strip! - socket.puts("PONG :tmi.twitch.tv\r\n") - log.info("[IRC Message]: " + line) - log.info("[IRC Response]: PONG :tmi.twitch.tv") - puts("-".bold.red*line.length) - puts "[Twitch] ".bold.cyan + "IRC: ".bold.yellow + line.bold.green - puts("-".bold.red*line.length) + if line =~ /^/ + response, log_msg = HandleCommands(line) end - match = line.match(/^:(.+)!(.+)PRIVMSG ##{CHANNEL} :(.+)$/) - message = match && match[3] - if message =~ /^/ - message = message.strip - user = match[1] # Get username - - # Twitch message limit - (Max 100 messages in 30 secs - Applies to mods and above) - # Avoid global ban - if Time.now.to_i - message_limit > 30 # If more than 30 seconds has passed - message_count = 0 # Reset "message_count" - end - if message_count == 0 # If "message_count" is 0 - message_limit = Time.now.to_i # Start counting to 30 again - end - message_count = message_count + 1 + + if response + socket.puts(response) + response = nil end - # // COMMANDS // COMMANDS // COMMANDS - if message != nil - msg = message.downcase - if admin_commands.include?(msg) and user == CHANNEL # ADMIN COMMANDS -> user == CHANNEL (OWNER) - if msg.include?("!disconnect") - socket.puts(send + "Disconnecting") # Disconnect from the channel - socket.puts("PART ##{CHANNEL}") # Disconnect from the channel - Disconnect() - log.info("[Command] #{user}: #{message}") - end - user = user.capitalize # Capitalize first letter (Cuz I'm that kind of person) - elsif commands.include?(msg) and message_count < 80 - puts "[Command] ".bold.cyan + "#{user}: ".bold + "#{message}".bold.cyan - if api_commands.include?(msg) # API commands - if msg.include?("uptime") - file = open("https://decapi.me/twitch/uptime?channel=#{CHANNEL}") # API for uptime - content = "#{CHANNEL} has been live for: " + file.read - elsif msg.include?("followed") - file = open("https://decapi.me/twitch/followage/#{CHANNEL}/#{user}") # API for followed - content = file.read - if content == "Follow not found" - content = "#{user} is not following #{CHANNEL}" - else - content = "#{user} has been following #{CHANNEL} for " + content - end - end - puts "[Response] ".bold.red + "Cortexio: ".bold + "API: ".bold.yellow + "\"#{content}\"".bold.red - else - file = open "Responses/" + msg.gsub!("!","") + ".txt" # open matching file - content = file.read - file.close - puts "[Response] ".bold.red + "Cortexio: ".bold + "File: ".bold.yellow + "\"#{msg}.txt\"".bold.red - end - file.close - log.info("[Command] #{user}: #{message}") - else - puts "[Message] ".bold.green + "#{user}: ".bold + "#{message}".bold.green - log.info("[Message] #{user}: #{message}") - if message[0] == "!" # Unrecognized command - content = "Unrecognized command: \"#{message}\" - Type !commands to see a list of available commands." - end - end - # Response handling - if content != nil - content.gsub!("USER", "@#{user}") # Take a look at "ping.txt" to see example usage - content.gsub!("CHANNEL", "#{CHANNEL}") - if content.include? "COMMANDS" - content.gsub!("COMMANDS", "#{commands}") - content.gsub!("\"", "") - content.gsub!("[","") - content.gsub!("]","") - end - socket.puts(send + content) # Send response if any - content = nil # Too avoid multiple messages with the same response - end + if log_msg + log.info(log_msg) + log_msg = nil end end end end -def Disconnect() # End script - running = false - exit -end - # Loop to keep bot going while (running) do - gets.chomp + input = gets.chomp + if input == "clear" + Clear() + + elsif input.index("project") == 0 + Project(input) + + elsif input == "disconnect" + socket.puts("PRIVMSG ##{CHANNEL} : Disconnected BibleThump") + socket.puts("PART ##{CHANNEL}\r\n") + running = false + exit + end end