rbender 0.3.0 → 0.4.92

data/lib/rbender/config_handler.rb

@@ -0,0 +1,35 @@
+require 'yaml'
+
+module RBender
+	module ConfigHandler
+		@@config_path = 'config.yaml'
+		CONFIG_NAME = 'config.yaml'
+
+
+		def self.underscore(str)
+			str.gsub(/::/, '/').
+					gsub(/([A-Z]+)([A-Z][a-z])/, '\1_\2').
+					gsub(/([a-z\d])([A-Z])/, '\1_\2').
+					tr("-", "_").
+					downcase
+		end
+
+		def self.settings
+			begin
+				@@settings = YAML.load(File.read(@@config_path))
+			rescue
+				raise "Config file doesn't exists!"
+			end
+
+			def @@settings.save
+				File.write(@@config_path, @@settings.to_yaml)
+			end
+
+			@@settings
+		end
+
+		def self.config_path=(path)
+			@@config_path = "#{path}/#{CONFIG_NAME}"
+		end
+	end
+end

data/lib/rbender/keyboard.rb

@@ -1,143 +1,115 @@
 class RBender::Keyboard
 
-  attr_accessor :markup,
-                :localizations,
-                :actions,
-                :response,
-                :switchers,
-                :markup_tg,
-                :markup_final,
-                :switch_groups,
-                :session
-
-  def initialize(response)
-    @response = response
-    @actions = {}
-    @markup = []
-    @localizations = {}
-    @switchers = {}
-    @markup_tg = nil
-    @markup_final = {}
-    @switch_groups = {}
-    @button_switch_group = {}
-    @one_time = false
-    @session = nil
-    @hide_on_action = false
-    @resize = false
-    @force_invoke = false
-  end
-
-  def hide_on_action
-    @hide_on_action = true
-  end
-
-  def force_invoked?
-    @force_invoke
-  end
-
-  # Adds button to the keyboard
-  #
-  # Example:
-  # button :btn_test, "Test button" do
-  #     some_actions
-  # end
-  #
-  # button :btn_with_loc, {ru: "Кнопка", en: "Button"} do
-  #     some_actions
-  # end
-
-  def button(id, localizations, *icons, &action)
-    @actions[id] = action
-    @localizations[id] = localizations
-  end
-
-  # Checks keyboard one time or not
-  def one_time?
-    @one_time
-  end
-
-  # makes a keyboard one time
-  def one_time
-    @one_time = true
-  end
-
-  # Add a line to markup
-  def add_line (*buttons)
-    @markup += [buttons]
-  end
-
-  # Set response when keyboard is invoked
-  def set_response(new_response)
-    @response = new_response
-  end
-
-  # Adds switcher to buttons
-  def switcher(id, *switches)
-    @switchers[id] = switches
-  end
-
-  # Adds switch group
-  def switch_group(group_name, *buttons)
-    @switch_groups[group_name] = buttons
-  end
-
-  # Resize telegram buttons
-  def resize
-    @resize = true
-  end
-
-  # Adds button switch group
-  def button_switch_group(button_id)
-    @switch_groups.each do |group, buttons|
-      if buttons.include? button_id
-        return group
-      end
-    end
-    nil
-  end
-
-  # Method to initialize a keyboard
-  #
-  def _build(session)
-    #TODO: add localization support
-    markup = []
-    @markup.each do |line|
-      buf_line = []
-      line.each do |button_id|
-        btn_id = button_id # used for replacing group mask
-        if @switch_groups.member? button_id # if it's switch group mask
-          button_num = 0
-          if session[:_keyboard_switch_groups].member? btn_id
-            button_num = session[:_keyboard_switch_groups][btn_id]
-          else
-            session[:_keyboard_switch_groups][btn_id] = 0
-          end
-          button = @localizations[@switch_groups[button_id][button_num]].dup
-          btn_id = @switch_groups[btn_id][button_num]
-        else # if it's normal button
-          button = @localizations[btn_id].dup
-        end
-
-        if @switchers.member? btn_id # Code for switcher
-          switch_num = 0
-          if session[:_keyboard_switchers].member? btn_id # If switcher inside session
-            switch_num = session[:_keyboard_switchers][btn_id]
-          else
-            session[:_keyboard_switchers][btn_id] = 0 # Switcher init
-          end
-          switcher = @switchers[btn_id][switch_num]
-          button << " " << switcher unless switcher.empty?
-        end
-
-        buf_line << button
-        @markup_final[btn_id] = button
-      end
-      markup << buf_line
-    end
-    # puts @keyboard.markup_final
-    @markup_tg = Telegram::Bot::Types::ReplyKeyboardMarkup
-                     .new(keyboard: markup,
-                          one_time_keyboard: one_time?,
-                          resize_keyboard: @resize)
-  end
+	attr_accessor :markup,
+								:localizations,
+								:actions,
+								:response,
+								:switchers,
+								:markup_tg,
+								:markup_final,
+								:switch_groups,
+								:session
+
+	def initialize
+		@actions             = {}
+		@markup              = []
+		@localizations       = {}
+		@markup_tg           = nil
+		@markup_final        = {}
+		@button_switch_group = {}
+		@one_time            = false
+		@session             = nil
+		@hide_on_action      = false
+		@resize              = false
+		@force_invoke        = false
+	end
+
+	def hide_on_action
+		@hide_on_action = true
+	end
+
+	def force_invoked?
+		@force_invoke
+	end
+
+	# Adds button to the keyboard
+	#
+	# Example:
+	# button :btn_test, "Test button" do
+	#     some_actions
+	# end
+	#
+	# button :btn_with_loc, {ru: "Кнопка", en: "Button"} do
+	#     some_actions
+	# end
+
+	def button(id, localizations, &action)
+		@actions[id]       = action
+		@localizations[id] = localizations
+	end
+
+	# Checks keyboard one time or not
+	def one_time?
+		@one_time
+	end
+
+	# makes a keyboard one time
+	def one_time
+		@one_time = true
+	end
+
+	# Add a line to markup
+	def add_line (*buttons)
+		@markup += [buttons]
+	end
+
+	# Set response when keyboard is invoked
+	def set_response(new_response)
+		@response = new_response
+	end
+
+	# Adds switcher to buttons
+	def switcher(id, *switches)
+		@switchers[id] = switches
+	end
+
+	# Adds switch group
+	def switch_group(group_name, *buttons)
+		@switch_groups[group_name] = buttons
+	end
+
+	# Resize telegram buttons
+	def resize
+		@resize = true
+	end
+
+	# Adds button switch group
+	def button_switch_group(button_id)
+		@switch_groups.each do |group, buttons|
+			if buttons.include? button_id
+				return group
+			end
+		end
+		nil
+	end
+
+	# Method to initialize a keyboard
+	#
+	def build(session)
+		markup = []
+		@markup.each do |line|
+			buf_line = []
+			line.each do |button_id|
+				button = @localizations[button_id].dup
+
+				buf_line << button
+				@markup_final[btn_id] = button
+			end
+			markup << buf_line
+		end
+		@markup_tg = Telegram::Bot::Types::ReplyKeyboardMarkup
+										 .new(keyboard:          markup,
+													one_time_keyboard: one_time?,
+													resize_keyboard:   @resize)
+	end
 end

data/lib/rbender/keyboard_inline.rb

@@ -45,13 +45,13 @@ class RBender::KeyboardInline
     @markup += [buttons]
   end
 
-  def _invoke
+  def invoke
     instance_eval(&@keyboard_block)
   end
 
-  def _build
+  def build
     @markup = []
-    _invoke
+    invoke()
 
     buttons = {}
     buttons.merge! @buttons_callback

data/lib/rbender/methods.rb

@@ -0,0 +1,129 @@
+class RBender::Methods
+
+	def initialize(message, api, session)
+		@message = message
+		@api = api
+		@session = session
+	end
+
+	#--------------
+	# User methods
+	#--------------
+
+	# Set message user gets while keyboard has invoked
+	def set_response(new_response)
+		@keyboard.set_response(new_response)
+	end
+
+	# Returns session hash
+	def session
+		@session
+	end
+
+	# Returns message object
+	def message
+		@message
+	end
+
+	# Returns Inline keyboard object by name
+	def inline_markup(name)
+		raise "Keyboard #{name} doesn't exists!" unless @inline_keyboards.member? name
+		keyboard = @inline_keyboards[name]
+		keyboard.instance_eval(&@helpers_block) unless @helpers_block.nil?
+		keyboard.build
+		keyboard.markup_tg
+	end
+
+	def switch(state_to)
+		@session[:state_stack].push(@session[:state])
+		@session[:state] = state_to
+	end
+
+	def switch_prev
+		@session[:state] = @session[:state_stack].pop
+	end
+
+
+	#--------------
+	# API METHODS
+	#--------------
+	# Hides inline keyboard
+	# Must be called from any inline keyboard state
+	def hide_inline
+		edit_message_reply_markup
+	end
+
+	# Hides keyboard's markup.
+	def hide_keyboard
+
+	end
+
+	#
+	# @param text [String] string
+	#
+	def answer_callback_query(text: nil,
+														show_alert: nil)
+		begin
+			@api.answer_callback_query callback_query_id: @message.id,
+																 text:              text,
+																 show_alert:        show_alert
+		rescue
+		end
+	end
+
+	def send_message(text:,
+									 chat_id: @message.from.id,
+									 parse_mode: nil,
+									 disable_web_page_preview: nil,
+									 disable_notification: nil,
+									 reply_to_message_id: nil,
+									 reply_markup: nil)
+
+		if text.strip.empty?
+			raise "A text can't be empty or consists of space symbols only"
+		end
+		@api.send_message chat_id:                  chat_id,
+											text:                     text,
+											disable_web_page_preview: disable_web_page_preview,
+											disable_notification:     disable_notification,
+											reply_to_message_id:      reply_to_message_id,
+											parse_mode:               parse_mode,
+											reply_markup:             reply_markup
+	end
+
+	def edit_message_text(inline_message_id: nil,
+												text:,
+												message_id: @message.message.message_id,
+												parse_mode: nil,
+												disable_web_page_preview: nil,
+												reply_markup: nil)
+		begin
+			@api.edit_message_text chat_id:                  @message.from.id,
+														 message_id:               message_id,
+														 text:                     text,
+														 inline_message_id:        inline_message_id,
+														 parse_mode:               parse_mode,
+														 disable_web_page_preview: disable_web_page_preview,
+														 reply_markup:             reply_markup
+		rescue
+		end
+	end
+
+	def edit_message_reply_markup(chat_id: @message.from.id,
+																message_id: @message.message.message_id,
+																inline_message_id: nil,
+																reply_markup: nil)
+		begin
+			@api.edit_message_reply_markup chat_id:           chat_id,
+																		 message_id:        message_id,
+																		 inline_message_id: inline_message_id,
+																		 reply_markup:      reply_markup
+		rescue
+		end
+	end
+
+	def get_file(file_id:)
+		@api.get_file file_id: file_id
+	end
+end
+

data/lib/rbender/mongo_client.rb

@@ -10,15 +10,13 @@ class RBender::MongoClient
     instance.mongo_client
   end
 
-  def self.setup(bot_name, mongo_server_ip, mongo_server_port)
-    instance.setup(bot_name, mongo_server_ip, mongo_server_port)
+  def self.setup(bot_name, mongo_string)
+    instance.setup(bot_name, mongo_string)
   end
 
-  def setup(bot_name, mongo_server_ip, mongo_server_port)
+  def setup(bot_name, mongo_string)
     @bot_name = bot_name
-    @mongo_server_ip = mongo_server_ip
-    @mongo_server_port = mongo_server_port
-    @mongo_client = Mongo::Client.new(["#{mongo_server_ip}:#{mongo_server_port}"], :database => bot_name)
+    @mongo_client = Mongo::Client.new(mongo_string, :database => bot_name)
 
     Mongo::Logger.logger.level = ::Logger::FATAL
   end