kybus-bot 0.4.4 → 0.5.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 770ceb23f2b040686b9337adb8a5687c2b28dba0c076eb46fdb821ba15bcb938
-  data.tar.gz: 2e0a3b827a4390f00d03b6717994b9df5b2bbc284a76cf154e4bc189d8279f9e
+  metadata.gz: fcb4039ffd0fa38cadb57ba4a6a0813d5c105590677f1e6f6a7ef85fa6ee2365
+  data.tar.gz: abf1c8941afd74565c25e012b3eec8e94e17d1e0a42a83b2bc20be1eceac0192
 SHA512:
-  metadata.gz: '08c94625e9e4b288bc3f07bd827fffba932af381af77b8a142825dd254229c6a66f0c31c00444b7056f782cff2e123263bb636708a099e8102754ad9fe431933'
-  data.tar.gz: d1a311b425287b4de3e6ead3409b4bb1183dccc6614bb66758168f120144e91dc5bf523378862e098b924fa7bc0cffd344a32477fa0f0abe546835d83cf07b31
+  metadata.gz: f1e09cd5437c55dbfc6a38df6c3f8ff1383f6f2974538e3ec2ebd380ea38db0e12cd9ce8ba82b87de15bd1200fef0d610ab1e81ee34c644e5fd064f564a31bd7
+  data.tar.gz: 159c13cb384b3168647e572544cdb96098a4d11b8ae4ed7c1143ba562fd0c52eccdf9295729f97e1614ef44d020681bd9615dc064522664114a3c211d9319566

data/lib/kybus/bot/adapters/debug.rb

@@ -1,5 +1,8 @@
 # frozen_string_literal: true
 
+require_relative 'base'
+require_relative '../message'
+
 module Kybus
   module Bot
     # :nodoc: #
@@ -8,9 +11,12 @@ module Kybus
       # Wraps a debugging message inside a class.
       class DebugMessage < Kybus::Bot::Message
         # It receives a string with the raw text and the id of the channel
-        def initialize(text, channel)
+        attr_reader :attachment
+
+        def initialize(text, channel, attachment = nil)
           @text = text
           @channel = channel
+          @attachment = attachment
         end
 
         # Returns the channel id
@@ -22,6 +28,14 @@ module Kybus
         def raw_message
           @text
         end
+
+        def user
+          channel_id
+        end
+
+        def has_attachment?
+          !!attachment
+        end
       end
 
       # This class simulates a message chat with a user.
@@ -52,7 +66,7 @@ module Kybus
           DebugMessage.new(@pending_messages.shift, @name)
         end
 
-        def send_data(message)
+        def send_data(message, attachment)
           return unless @echo
 
           puts "Sending message to channel: #{@name}"
@@ -62,8 +76,8 @@ module Kybus
         attr_writer :echo
 
         # receives the answer from the bot
-        def answer(message)
-          send_data(message)
+        def answer(message, attachment = nil)
+          send_data(message, attachment)
           @state = :open
         end
       end
@@ -112,8 +126,8 @@ module Kybus
         end
 
         # interface for sending messages
-        def send_message(channel_name, contents)
-          channel(channel_name).answer(contents)
+        def send_message(channel_name, contents, attachment = nil)
+          channel(channel_name).answer(contents, attachment)
         end
 
         # interface for sending video

data/lib/kybus/bot/adapters/telegram.rb

@@ -36,11 +36,61 @@ module Kybus
           @message.chat.type == 'private'
         end
 
+        def has_attachment?
+          !!@message.document
+        end
+
+        def attachment
+          @message.document
+        end
+
         def user
           @message.from.id
         end
       end
 
+      class TelegramFile
+        extend Kybus::DRY::ResourceInjector
+        attr_reader :id
+        def initialize(message)
+          if message.is_a?(String)
+            @id = message
+          elsif message.is_a?(Hash)
+            @id = message['id'] || message[:id]
+          elsif message.is_a?(TelegramFile)
+            @id = message.id
+          else
+            @id = message.file_id
+          end
+        end
+
+        def to_h
+          {
+            provide: 'telegram',
+            id: @id
+          }
+        end
+
+        def cli
+          @cli ||= TelegramFile.resource(:cli)
+        end
+
+        def meta
+          @meta ||= cli.api.get_file(file_id: @id)
+        end
+
+        def original_name
+          meta.dig('result', 'file_name')
+        end
+
+        def download
+          token = cli.api.token
+          file_path = meta.dig('result', 'file_path')
+          path = "https://api.telegram.org/file/bot#{token}/#{file_path}"
+          Faraday.get(path).body
+        end
+      end
+
       ##
       # This adapter is intended to be used on unit tests and development.
       class Telegram
@@ -53,6 +103,7 @@ module Kybus
         def initialize(configs)
           @config = configs
           @client = ::Telegram::Bot::Client.new(@config['token'])
+          TelegramFile.register(:cli, @client)
         end
 
         # Interface for receiving message
@@ -99,6 +150,16 @@ module Kybus
           file = Faraday::UploadIO.new(image_url, 'image/jpeg')
           @client.api.send_photo(chat_id: channel_name, photo: file)
         end
+
+        # interface for sending document
+        def send_document(channel_name, image_url)
+          file = Faraday::UploadIO.new(image_url, 'application/octect-stream')
+          @client.api.send_document(chat_id: channel_name, document: file)
+        end
+
+        def file_builder(file)
+          TelegramFile.new(file)
+        end
       end
 
       register('telegram', Telegram)

data/lib/kybus/bot/base.rb

@@ -41,6 +41,10 @@ module Kybus
         provider.send_audio(channel || current_channel, content)
       end
 
+      def send_document(content, channel = nil)
+        provider.send_document(channel || current_channel, content)
+      end
+
       # Configurations needed:
       # - pool_size: number of threads created in execution
       # - provider: a configuration for a thread provider.
@@ -100,6 +104,7 @@ module Kybus
           self.command = message.raw_message
         else
           add_param(message.raw_message)
+          add_file(message.attachment) if message.has_attachment?
         end
         if command_ready?
           run_command!
@@ -145,6 +150,25 @@ module Kybus
         current_params
       end
 
+      def add_file(file)
+        return if @state[:requested_param].nil?
+
+        log_debug('Received new file',
+                  param: @state[:requested_param].to_sym,
+                  file: file.to_h)
+
+        files[@state[:requested_param].to_sym] = provider.file_builder(file)
+        @state[:params][("_#{@state[:requested_param]}_filename").to_sym] = file.file_name
+      end
+
+      def files
+        @state[:files] ||= {}
+      end
+
+      def file(name)
+        (file = files[name]) && provider.file_builder(file)
+      end
+
       def mention(name)
         provider.mention(name)
       end
@@ -177,7 +201,8 @@ module Kybus
         log_debug('Message set as command', command: cmd)
 
         @state[:cmd] = cmd.split(' ').first
-        @state[:params] = {}
+        @state[:params] = { }
+        @state[:files] = { }
       end
 
       # validates which is the following parameter required
@@ -212,18 +237,27 @@ module Kybus
       # Private implementation for load message
       def load_state(channel)
         data = @factory.get(channel)
-        data[:params] = JSON.parse(data[:params], symbolize_names: true)
+        data[:params] = JSON.parse(data[:params] || '{}', symbolize_names: true)
+        data[:files] = JSON.parse(data[:files] || '{}', symbolize_names: true)
         data
       rescue Kybus::Storage::Exceptions::ObjectNotFound
         @factory.create(channel_id: channel, params: {}.to_json)
       end
 
+      def parse_state!
+      end
+
       # Saves the state into storage
       def save_state!
-        json = @state[:params]
-        @state[:params] = json.to_json
+        backup = @state.clone
+        %i[params files].each do |param|
+          @state[param] = @state[param].to_json
+        end
+
         @state.store
-        @state[:params] = json
+        %i[params files].each do |param|
+          @state[param] = backup[param]
+        end
       end
 
       def session

data/lib/kybus/bot/migrator.rb

@@ -11,6 +11,7 @@ module Kybus
             String :channel_id
             String :user
             String :params, text: true
+            String :files, text: true
             String :cmd
             String :requested_param
           end

data/lib/kybus/bot/test.rb

@@ -0,0 +1,32 @@
+require_relative 'base'
+require_relative 'adapters/debug'
+
+module Kybus
+  module Bot
+    class Base
+      CONFIG = {
+        'name' => 'test',
+        'state_repository' => {
+          'name' => 'json',
+          'storage' => 'storage'
+        },
+        'pool_size' => 1,
+        'provider' => {
+          'name' => 'debug',
+          'echo' => true,
+          'channels' => { 'testing' => [] }
+        }
+      }.freeze
+
+      def self.make_test_bot
+        new(CONFIG)
+      end
+
+      def receives(msg, attachments = nil)
+        msg = ::Kybus::Bot::Adapter::DebugMessage.new(msg, 'testing', attachments)
+        @last_message = msg
+        process_message(msg)
+      end
+    end
+  end
+end

data/lib/kybus/bot/version.rb

@@ -2,6 +2,6 @@
 
 module Kybus
   module Bot
-    VERSION = '0.4.4'
+    VERSION = '0.5.1'
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: kybus-bot
 version: !ruby/object:Gem::Version
-  version: 0.4.4
+  version: 0.5.1
 platform: ruby
 authors:
 - Gilberto Vargas
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2021-12-24 00:00:00.000000000 Z
+date: 2022-02-16 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: kybus-core
@@ -222,6 +222,7 @@ files:
 - lib/kybus/bot/command_definition.rb
 - lib/kybus/bot/message.rb
 - lib/kybus/bot/migrator.rb
+- lib/kybus/bot/test.rb
 - lib/kybus/bot/testing.rb
 - lib/kybus/bot/version.rb
 homepage: https://github.com/tachomex/kybus