require 'active_support/core_ext/hash/keys' require 'active_support/core_ext/hash/transform_values' require 'active_support/core_ext/hash/indifferent_access' module Telegram module Bot module ConfigMethods # Overwrite config. attr_writer :bots_config # Keep this setting here, so we can avoid loading Bot::UpdatesPoller # when polling is disabled. attr_writer :bot_poller_mode # It just tells routes helpers whether to add routed bots to # Bot::UpdatesPoller, so their config will be available by bot key in # Bot::UpdatesPoller.start. # # It's enabled by default in Rails dev environment and `rake telegram:bot:poller` # task. Use `BOT_POLLER_MODE=true` envvar to set it manually. def bot_poller_mode? return @bot_poller_mode if defined?(@bot_poller_mode) @bot_poller_mode = ENV.fetch('BOT_POLLER_MODE') do Rails.env.development? if defined?(Rails.env) end end # Hash of bots made with bots_config. def bots @bots ||= bots_config.each_with_object({}) do |(id, config), h| h[id] = Client.wrap(config, id: id) end end # Default bot. def bot @bot ||= bots.fetch(:default) do raise 'Default bot is not configured.' \ ' Add :default to bots_config' \ ' or use telegram.bot/telegram.bots.default section in secrets.yml.' end end # Hash of botan clients made from #bots. def botans @botans ||= bots.transform_values(&:botan) end # Returns config for .bots method. By default uses `telegram['bots']` section # from `secrets.yml` merging `telegram['bot']` at `:default` key. # # Can be overwritten with .bots_config= def bots_config @bots_config ||= if defined?(Rails.application) secrets = Rails.application.secrets. fetch(:telegram, {}).with_indifferent_access secrets.fetch(:bots, {}).symbolize_keys.tap do |config| default = secrets[:bot] config[:default] = default if default end else {} end end # Resets all cached bots and their configs. def reset_bots @bots = nil @bot = nil @bots_config = nil @botans = nil end end end end