kybus-bot 0.4.3 → 0.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7d6ea56c2bd47e58087f433d504c984c2055958b8b5cbd6fde2b438ffbcc77f9
-  data.tar.gz: e256785b5022eb71f9f8943ab8e54dae9696b6ffe97431f443b23630c9dfad81
+  metadata.gz: 2b9b6a8575a2207a5aeff2a9a890973019460a86d904f2bc5989e4da2269cbff
+  data.tar.gz: 527bbd3abfb670bb76b4181e919124d43c2fc0c4cd353bb6457fc7a36ba34db0
 SHA512:
-  metadata.gz: 1ef852ff7b0d785be08dac681cb10185ddde7b4362e343d7e188e93a3e2bf143ba48f126c301f1cfff8b7e6069f172de1b92057ad9be258fcb86a5b0880908e0
-  data.tar.gz: 7dc0c97b640a932295c9d5b13454a865d9eb6384a95ba613de292b4f5c0d28ef1953617f1d3af7a4ccd4839d48a97ca27aa31b48c2c37e53c7854e4d87a19994
+  metadata.gz: 31fbae47c94d21fe626e3b00a24fae3e22d5928da955e9efc795f76808baab49bed4ef358e229fe7172f4db343dbd8d8a1a62f2a3f5945e8b7416c1738209796
+  data.tar.gz: b23e6febfc1daf445702e2670f181de76bbcd7170426ded2e6a52cf6a2e081092246be1603a275e5e52623916446e3f7b9b9a85d112872a5a1995f064a973fd1

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,10 @@ 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)
+        def initialize(text, channel, attachments = {})
           @text = text
           @channel = channel
+          @attachments = attachments
         end
 
         # Returns the channel id
@@ -22,6 +26,10 @@ module Kybus
         def raw_message
           @text
         end
+
+        def user
+          channel_id
+        end
       end
 
       # This class simulates a message chat with a user.

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,10 +150,33 @@ 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
 
+      def registered_commands
+        @commands.registered_commands
+      end
+
       # Loads command from state
       def current_command_object
         command = @state[:cmd]
@@ -173,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
@@ -208,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/command_definition.rb

@@ -36,6 +36,10 @@ module Kybus
         @commands[name] = Command.new(params, block)
       end
 
+      def registered_commands
+        @commands.keys
+      end
+
       # Returns a command with the name
       def [](name)
         @commands[name]

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 = {})
+        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.3'
+    VERSION = '0.5.0'
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: kybus-bot
 version: !ruby/object:Gem::Version
-  version: 0.4.3
+  version: 0.5.0
 platform: ruby
 authors:
 - Gilberto Vargas
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2021-12-23 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