Alesia Yakutsionak - 1

Alesia Yakutsionak/1/Gemfile

@@ -0,0 +1,5 @@
+source 'https://rubygems.org'
+
+gem 'nokogiri'
+gem 'redis'
+gem 'telegram-bot-ruby'

Alesia Yakutsionak/1/README.md

@@ -0,0 +1,8 @@
+To test the program:
+
+-run "bundle install"
+-run "redis-server" to start redis
+-run "ruby run_parser.rb"
+-put Your_Telegram_Bot_Token in BOT_TOKEN
+-run "run_bot.rb"
+-write command (/wordplay) to bot in Telegram

Alesia Yakutsionak/1/bot.rb

@@ -0,0 +1,40 @@
+require 'telegram/bot'
+require 'redis'
+
+class Bot
+  INVALID_COMMAND_TEXT = "I didn't understand you.".freeze
+  BOT_TOKEN = 'Your_Telegram_Bot_Token'.freeze
+
+  def initialize
+    @redis = Redis.new
+    @puns_count = @redis.keys('pun_*').count
+  end
+
+  def run
+    Telegram::Bot::Client.run(BOT_TOKEN) do |bot|
+      bot.listen do |message|
+        reply(bot, message)
+      end
+    end
+  end
+
+  private
+
+  def reply(bot, message)
+    case message.text
+    when /^hey|hi|hello$/i
+      bot.api.sendMessage(chat_id: message.chat.id, text: "Hey, #{message.from.first_name}")
+    when '/wordplay' then
+      random_pun = select_random_pun
+      wordplay_text = "Random wordplay [#{random_pun[:number]}]:\n#{random_pun[:text]}"
+      bot.api.sendMessage(chat_id: message.chat.id, text: wordplay_text)
+    else
+      bot.api.sendMessage(chat_id: message.chat.id, text: INVALID_COMMAND_TEXT)
+    end
+  end
+
+  def select_random_pun
+    i = rand(0..@puns_count - 1)
+    { number: i, text: @redis.get("pun_#{i}") }
+  end
+end

Alesia Yakutsionak/1/pun_parser.rb

@@ -0,0 +1,52 @@
+require 'rubygems'
+require 'nokogiri'
+require 'open-uri'
+require 'redis'
+
+class PunParser
+  BASE_URL = 'https://onelinefun.com'.freeze
+
+  def run
+    loop do
+      puts "Parsing #{@link}"
+      page = parse_page
+      @link = find_next_page_link(page)
+      break unless @link
+    end
+    save_puns_to_redis
+  end
+
+  def initialize
+    @link = '/puns/'
+    @redis = Redis.new
+    @puns = []
+  end
+
+  private
+
+  def parse_page
+    url = BASE_URL + @link
+    page = Nokogiri::HTML(URI.open(url))
+    page.css('article div p').each do |p|
+      pun = p.text
+      @puns << pun
+    end
+    page
+  end
+
+  def find_next_page_link(page)
+    a_next = page.css('article div.p > a')
+    @link = if a_next && a_next.text =~ /^next/i
+              a_next.attr('href').value
+            else
+              nil
+            end
+    @link
+  end
+
+  def save_puns_to_redis
+    @puns.each_with_index do |pun, i|
+      @redis.set("pun_#{i}", pun)
+    end
+  end
+end

Alesia Yakutsionak/1/run_bot.rb

@@ -0,0 +1,3 @@
+require_relative 'bot.rb'
+
+Bot.new.run

Alesia Yakutsionak/1/run_parser.rb

@@ -0,0 +1,3 @@
+require_relative 'pun_parser.rb'
+
+PunParser.new.run