kybus-bot 0.4.6 → 0.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: '01392aefbe8bf2a7f5be90e5cc0f77fbbcf66ece5ed77c2bd9f01be3fb9b476b'
-  data.tar.gz: 43ec132ea325c8fe9cd6401d6976a80fcd9e57939483b8f79ea48848b0015bc4
+  metadata.gz: 2b9b6a8575a2207a5aeff2a9a890973019460a86d904f2bc5989e4da2269cbff
+  data.tar.gz: 527bbd3abfb670bb76b4181e919124d43c2fc0c4cd353bb6457fc7a36ba34db0
 SHA512:
-  metadata.gz: 63f439ab82c6faf1324133146b5b47bff2a0c9fa8d29c8daab00f322edd6c5b47a639f0bd99150cdd420e9a95e2ddd9f32a62245f9ddd85bd6eba35af643481a
-  data.tar.gz: a8a5cffea95d2831d911577d5754d45ed97361bda2f99ddf42812aa2b4058b800abb8a17944d7d84c8f14eba4f4d734290b5c0c24355082e3bf54340adce3bee
+  metadata.gz: 31fbae47c94d21fe626e3b00a24fae3e22d5928da955e9efc795f76808baab49bed4ef358e229fe7172f4db343dbd8d8a1a62f2a3f5945e8b7416c1738209796
+  data.tar.gz: b23e6febfc1daf445702e2670f181de76bbcd7170426ded2e6a52cf6a2e081092246be1603a275e5e52623916446e3f7b9b9a85d112872a5a1995f064a973fd1

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/version.rb

@@ -2,6 +2,6 @@
 
 module Kybus
   module Bot
-    VERSION = '0.4.6'
+    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.6
+  version: 0.5.0
 platform: ruby
 authors:
 - Gilberto Vargas
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2022-01-10 00:00:00.000000000 Z
+date: 2022-02-16 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: kybus-core