slack-smart-bot 1.14.2 → 1.15.1

data/lib/slack/smart-bot/commands/on_master/admin_master/exit_bot.rb

@@ -10,19 +10,19 @@ class SlackSmartBot
   # helpadmin:    <https://github.com/MarioRuiz/slack-smart-bot#bot-management|more info>
   # helpadmin: command_id: :exit_bot
   # helpadmin:
-  def exit_bot(command, from, dest, display_name, silent: false)
+  def exit_bot(command, user, dest, display_name, silent: false)
     save_stats(__method__)
     if config.on_master_bot
-      if config.masters.include?(from) #admin user
+      if config.team_id_masters.include?("#{user.team_id}_#{user.name}") #master admin user
         if answer.empty?
-          ask("are you sure?", command, from, dest)
+          ask("are you sure?", command, user, dest)
         else
           case answer
           when /yes/i, /yep/i, /sure/i
             react :runner
             @bots_created.each { |key, value|
               value[:thread] = ""
-              send_msg_channel(key, "Bot has been closed by #{from}") unless silent
+              send_msg_channel(key, "Bot has been closed by #{user.name}") unless silent
               save_status :off, :exited, "The admin closed SmartBot on *##{value.channel_name}*"
               sleep 0.5
             }
@@ -30,10 +30,15 @@ class SlackSmartBot
             sleep 0.5
             file = File.open("#{config.path}/config_tmp.status", "w")
             config.exit_bot = true
-            file.write config.inspect
+            @config_log.exit_bot = true
+            file.write @config_log.inspect
             file.close
             @status = :exit
             respond "Game over!", dest
+            @listening[:threads].each do |thread_ts, channel_thread|
+              unreact :running, thread_ts, channel: channel_thread
+              respond "ChatGPT session closed since SmartBot is going to be closed", channel_thread, thread_ts: thread_ts
+            end
             if config.simulate
               sleep 2
               @status = :off
@@ -49,10 +54,10 @@ class SlackSmartBot
               exit!
             end
           when /no/i, /nope/i, /cancel/i
-            answer_delete(from)
+            answer_delete(user)
             respond "Thanks, I'm happy to be alive", dest
           else
-            ask("I don't understand, are you sure do you want me to close? (yes or no)", command, from, dest)
+            ask("I don't understand, are you sure do you want me to close? (yes or no)", command, user, dest)
           end
         end
       else

data/lib/slack/smart-bot/commands/on_master/admin_master/notify_message.rb

@@ -10,10 +10,10 @@ class SlackSmartBot
   # helpmaster:    <https://github.com/MarioRuiz/slack-smart-bot#sending-notifications|more info>
   # helpmaster: command_id: :notify_message
   # helpmaster:
-  def notify_message(dest, from, where, message)
+  def notify_message(dest, user, where, message)
     save_stats(__method__)
     if config.on_master_bot
-      if config.masters.include?(from) #admin user
+      if config.team_id_masters.include?("#{user.team_id}_#{user.name}") #master admin user
         if where.nil? #not all and not channel
           @bots_created.each do |k, v|
             respond message, k

data/lib/slack/smart-bot/commands/on_master/admin_master/publish_announcements.rb

@@ -12,7 +12,7 @@ class SlackSmartBot
   def publish_announcements(user)
     save_stats(__method__)
     if config.on_master_bot
-      if config.masters.include?(user.name) #admin user
+      if config.team_id_masters.include?("#{user.team_id}_#{user.name}") #master admin user
         channels = Dir.entries("#{config.path}/announcements/")
         channels.select! {|i| i[/\.csv$/]}
         channels.each do |channel|

data/lib/slack/smart-bot/commands/on_master/admin_master/set_general_message.rb

@@ -13,10 +13,10 @@ class SlackSmartBot
   # helpmaster:    <https://github.com/MarioRuiz/slack-smart-bot#bot-management|more info>
   # helpmaster: command_id: :set_general_message
   # helpmaster:
-  def set_general_message(from, status, message)
+  def set_general_message(user, status, message)
     save_stats(__method__)
     if config.on_master_bot
-      if config.masters.include?(from) #admin user
+      if config.team_id_masters.include?("#{user.team_id}_#{user.name}") #master admin user
         if status == 'on'
           config.general_message = message
           respond "General message has been set."
@@ -24,11 +24,11 @@ class SlackSmartBot
           config.general_message = ''
           respond "General message won't be displayed anymore."
         end
-        
+        @config_log.general_message = config.general_message
         file = File.open("#{config.path}/config_tmp.status", "w")
-        file.write config.inspect
+        file.write @config_log.inspect
         file.close
-    
+
       else
         respond 'Only master admins on master channel can use this command.'
       end

data/lib/slack/smart-bot/commands/on_master/admin_master/set_maintenance.rb

@@ -16,10 +16,10 @@ class SlackSmartBot
   # helpmaster:    <https://github.com/MarioRuiz/slack-smart-bot#bot-management|more info>
   # helpmaster: command_id: :set_maintenance
   # helpmaster:
-  def set_maintenance(from, status, message)
+  def set_maintenance(user, status, message)
     save_stats(__method__)
     if config.on_master_bot
-      if config.masters.include?(from) #admin user
+      if config.team_id_masters.include?("#{user.team_id}_#{user.name}") #master admin user
         if message == ''
           config.on_maintenance_message = "Sorry I'm on maintenance so I cannot attend your request."
         else
@@ -35,11 +35,11 @@ class SlackSmartBot
           respond "From now on I won't be on maintenance. Everything is back to normal!"
           save_status :on, :maintenance_off, config.on_maintenance_message
         end
-        
+        @config_log.on_maintenance = config.on_maintenance
         file = File.open("#{config.path}/config_tmp.status", "w")
-        file.write config.inspect
+        file.write @config_log.inspect
         file.close
-    
+
       else
         respond 'Only master admins on master channel can use this command.'
       end

data/lib/slack/smart-bot/commands/on_master/create_bot.rb

@@ -58,13 +58,13 @@ class SlackSmartBot
               default_rules = (__FILE__).gsub(/slack\/smart-bot\/commands\/on_master\/create_bot\.rb$/, "slack-smart-bot_rules.rb")
               default_general_rules = (__FILE__).gsub(/slack\/smart-bot\/commands\/on_master\/create_bot\.rb$/, "slack-smart-bot_general_rules.rb")
               default_general_commands = (__FILE__).gsub(/slack\/smart-bot\/commands\/on_master\/create_bot\.rb$/, "slack-smart-bot_general_commands.rb")
-              
+
               File.delete(config.path + rules_file) if File.exist?(config.path + rules_file)
               FileUtils.copy_file(default_rules, config.path + rules_file) unless File.exist?(config.path + rules_file)
               FileUtils.copy_file(default_general_rules, config.path + general_rules_file) unless File.exist?(config.path + general_rules_file)
               FileUtils.copy_file(default_general_commands, config.path + general_commands_file) unless File.exist?(config.path + general_commands_file)
               admin_users = Array.new()
-              creator_info = @users.select{|u| u.id == channel_found.creator or (u.key?(:enterprise_user) and u.enterprise_user.id == channel_found.creator)}[-1]
+              creator_info = find_user(channel_found.creator)
               if creator_info.nil? or creator_info.empty? or creator_info.user.nil?
                 admin_users = [from] + config.masters
               else
@@ -72,7 +72,7 @@ class SlackSmartBot
               end
               admin_users.uniq!
               @logger.info "BOT_SILENT=#{silent} ruby #{config.file_path} \"#{channel}\" \"#{admin_users.join(",")}\" \"#{rules_file}\" on"
-          
+
               if cloud
                 respond "Copy the bot folder to your cloud location and run `ruby #{config.file} \"#{channel}\" \"#{admin_users.join(",")}\" \"#{rules_file}\" on&`", dest
               else

data/lib/slack/smart-bot/commands.rb

@@ -80,11 +80,21 @@ require_relative "commands/general/personal_settings"
 require_relative "commands/general/teams/memos/add_memo_team_comment"
 require_relative "commands/general/teams/memos/see_memo_team"
 require_relative 'commands/general/ai/open_ai/open_ai_chat'
+require_relative 'commands/general/ai/open_ai/open_ai_chat_get_prompts'
+require_relative 'commands/general/ai/open_ai/open_ai_chat_delete_session'
+require_relative 'commands/general/ai/open_ai/open_ai_chat_share_session'
+require_relative 'commands/general/ai/open_ai/open_ai_chat_list_sessions'
+require_relative 'commands/general/ai/open_ai/open_ai_chat_add_collaborator'
+require_relative 'commands/general/ai/open_ai/open_ai_chat_use_model'
+require_relative 'commands/general/ai/open_ai/open_ai_chat_copy_session'
 require_relative 'commands/general/ai/open_ai/open_ai_generate_image'
 require_relative 'commands/general/ai/open_ai/open_ai_variations_image'
 require_relative 'commands/general/ai/open_ai/open_ai_edit_image'
 require_relative 'commands/general/ai/open_ai/open_ai_models'
 require_relative 'commands/general/ai/open_ai/open_ai_whisper'
+require_relative 'commands/general/recap'
+require_relative 'commands/general/summarize'
+require_relative 'commands/general/get_smartbot_readme'
 
 class SlackSmartBot
     include SlackSmartBot::Commands::General::AI::OpenAI

data/lib/slack/smart-bot/config.rb

@@ -0,0 +1,126 @@
+require 'ostruct'
+
+class SlackSmartBot
+
+  class Config < OpenStruct
+    # token [String] (default: ""): The API Slack token.
+    # user_token [String] (default: ""): The Slack User token.
+    # granular_token [String] (default: ""): The Slack granular token.
+    # path [String] (default: "."): Path to the folder where the bot is running
+    # silent [Boolean] (default: false): If true the bot will not send any message to the chat when starting or stopping
+    # testing [Boolean] (default: false): Whether the bot is running in testing mode.
+    # simulate [Boolean] (default: false): Whether the bot is running in simulation mode.
+    # stats [Boolean] (default: false): Whether the bot should collect and log statistics.
+    # allow_access [Hash] (default: {}): The access control settings for the bot.
+    # on_maintenance [Boolean] (default: false): Whether the bot is on maintenance mode.
+    # on_maintenance_message [String] (default: "Sorry I'm on maintenance so I cannot attend your request."): The message to send when the bot is on maintenance mode.
+    # general_message [String] (default: ""): The general message added to be sent on every message the bot sends.
+    # logrtm [Boolean] (default: false): Whether the bot should log all messages received and sent (RTM).
+    # status_channel [String] (default: "smartbot-status"): The channel where the bot will send status messages.
+    # stats_channel [String] (default: "smartbot-stats"): The channel where the bot is allowed to send statistics.
+    # jira [Hash] (default: { host: "", user: "", password: "" }): The settings for Jira.
+    # github [Hash] (default: , host: "https://api.github.com" }): The settings for GitHub.
+    # public_holidays [Hash] (default: { api_key: "", host: "https://calendarific.com", default_calendar: '' }): The settings for public holidays.
+    # encrypt [Boolean] (default: true): Whether the bot should encrypt the data.
+    # encryption [Hash] (default: { key: "", iv: "" }): The settings for encryption. If not provided the bot will generate a new key and iv.
+    # recover_encrypted [Boolean] (default: false): Whether the bot should recover the encrypted data in case now encrypt is set to false but data is still encrypted. This is used for testing purposes.
+    # ai [Hash] (default: { open_ai: { access_token: "", host: "" }): The settings for OpenAI.
+    # ai.open_ai.chat_gpt [Hash] (default: {access_token: "", host: "", model: "gpt-3.5-turbo", smartbot_model: "gpt-3.5-turbo", api_type: :openai, api_version: "", fixed_user: ''}): The settings for OpenAI ChatGPT.
+    # ai.open_ai.dall_e [Hash] (default: {access_token: "", host: "", model: "", api_type: :openai, image_size: "256x256"}): The settings for OpenAI DALL-E.
+    # ai.open_ai.whisper [Hash] (default: {access_token: "", host: "", model: "whisper-1", api_type: :openai}): The settings for OpenAI Whisper.
+    # ai.open_ai.models [Hash] (default: {access_token: "", host: "", url: "", api_type: :openai, api_version: ""}): The settings for OpenAI Models.
+    # file [String] (default: ""): The file to load the bot from.
+    # masters [Array] (default: []): The list of master users.
+    # team_id_masters [Array] (default: []): The list of master team_id + user ids.
+    # master_channel [String] (default: ""): The Smartbot master channel.
+    # channel [String] (default: ""): The Smartbot channel.
+    # status_init [Symbol] (default: :on): The initial status of the bot.
+    # rules_file [String] (default: ""): The file to load the rules from.
+    # admins [Array] (default: []): The list of admin users.
+    # team_id_admins [Array] (default: []): The list of admin team_id + user ids.
+    # start_bots [Boolean] (default: true): Whether the bot should start the bots when starting.
+    # ldap [Hash] (default: { host: "", port: 389, auth: {user: '', password: ''}, treebase: "dc=ds,dc=eng,dc=YOURCOMPANY,dc=com" }): The settings for LDAP. @ldap connection will be created. It will populate the sso_user_name key to 'user' searching by Slack mail specified in profile.
+    # authorizations [Hash] (default: {}): The authorizations for services. for example: { confluence: {host: 'confluence.example.com', authorization: 'Bearer Adjjj3dddfj'}}
+    def initialize(*args)
+      super
+      self.token ||= ""
+      self.user_token ||= ""
+      self.granular_token ||= ""
+
+      self.path ||= "."
+      self.file ||= ""
+      self.rules_file ||= ""
+
+      self.start_bots ||= true
+      self.status_init ||= :on
+      self.silent ||= false
+      self.testing ||= false
+      self.simulate ||= false
+      self.logrtm ||= false
+
+      self.masters ||= []
+      self.team_id_masters ||= []
+      self.admins ||= []
+      self.team_id_admins ||= []
+
+      self.stats ||= false
+      self.allow_access ||= {}
+
+      self.on_maintenance ||= false
+      self.on_maintenance_message ||= "Sorry I'm on maintenance so I cannot attend your request."
+      self.general_message ||= ""
+
+      self.master_channel ||= "" # smartbot master channel
+      self.channel ||= "" # smartbot channel
+      self.status_channel ||= "smartbot-status"
+      self.stats_channel ||= "smartbot-stats"
+
+      self.jira ||= { host: "", user: "", password: "" }
+      self.github ||= { token: "", host: "https://api.github.com" }
+      self.public_holidays ||= { api_key: "", host: "https://calendarific.com", default_calendar: ""}
+
+      self.encrypt ||= true
+      self.encryption ||= { key: "", iv: "" }
+      self.recover_encrypted ||= false
+
+      self.ai ||= { open_ai: { testing: false, access_token: "", host: "", chat_gpt: {}, dall_e: {}, whisper: {}, models: {} } }
+      self.ai[:open_ai][:host] = "https://#{self.ai[:open_ai][:host]}" unless self.ai[:open_ai][:host].empty? || self.ai[:open_ai][:host].start_with?("http")
+      self.ai[:open_ai][:chat_gpt] ||= {}
+      self.ai[:open_ai][:dall_e] ||= {}
+      self.ai[:open_ai][:whisper] ||= {}
+      self.ai[:open_ai][:models] ||= {}
+
+      self.ai[:open_ai][:chat_gpt][:access_token] ||= self.ai[:open_ai][:access_token]
+      self.ai[:open_ai][:dall_e][:access_token] ||= self.ai[:open_ai][:access_token]
+      self.ai[:open_ai][:whisper][:access_token] ||= self.ai[:open_ai][:access_token]
+      self.ai[:open_ai][:models][:access_token] ||= self.ai[:open_ai][:access_token]
+
+      self.ai[:open_ai][:chat_gpt][:host] ||= self.ai[:open_ai][:host]
+      self.ai[:open_ai][:dall_e][:host] ||= self.ai[:open_ai][:host]
+      self.ai[:open_ai][:whisper][:host] ||= self.ai[:open_ai][:host]
+      self.ai[:open_ai][:models][:host] ||= self.ai[:open_ai][:host]
+
+      self.ai[:open_ai][:chat_gpt][:model] ||= "gpt-3.5-turbo"
+      self.ai[:open_ai][:chat_gpt][:smartbot_model] ||= self.ai[:open_ai][:chat_gpt][:model]
+      self.ai[:open_ai][:dall_e][:model] ||= ""
+      self.ai[:open_ai][:whisper][:model] ||= "whisper-1"
+
+      self.ai[:open_ai][:chat_gpt][:api_type] ||= :openai
+      self.ai[:open_ai][:dall_e][:api_type] ||= :openai
+      self.ai[:open_ai][:whisper][:api_type] ||= :openai
+      self.ai[:open_ai][:models][:api_type] ||= self.ai[:open_ai][:chat_gpt][:api_type]
+
+      self.ai[:open_ai][:chat_gpt][:api_version] ||= ""
+      self.ai[:open_ai][:chat_gpt][:fixed_user] ||= ""
+      self.ai[:open_ai][:models][:api_version] ||= self.ai[:open_ai][:chat_gpt][:api_version]
+
+      self.ai[:open_ai][:dall_e][:image_size] ||= "256x256"
+
+      self.ai[:open_ai][:models][:url] ||= ""
+
+      self.ldap ||= { host: "", port: 389, auth: { user: '', password: '' }, treebase: "dc=ds,dc=eng,dc=YOURCOMPANY,dc=com" }
+
+      self.authorizations ||= {}
+    end
+  end
+end

data/lib/slack/smart-bot/listen.rb

@@ -1,46 +1,47 @@
 class SlackSmartBot
   def listen_simulate
     @salutations = [config[:nick], "<@#{config[:nick_id]}>", "bot", "smart", "smartbot", "smart-bot", "smart bot"]
-    @pings = []
+    @pings ||= []
     @last_activity_check = Time.now
     get_bots_created()
     @buffer_complete = [] unless defined?(@buffer_complete)
     b = File.read("#{config.path}/buffer_complete.log", encoding: "UTF-8")
-    result = b.scan(/^\|(\w+)\|(\w+)\|(\w+)\|([^~]+)~~~/m)
+    result = b.scan(/^\|(\w+)\|(\w*)\|(\w+)\|(\w+)\|([^~]+)~~~/m)
     result.delete(nil)
     new_messages = result[@buffer_complete.size..-1]
     unless new_messages.nil? or new_messages.empty?
       @buffer_complete = result
       new_messages.each do |message|
         channel = message[0].strip
-        user = message[1].strip
-        user_name = message[2].strip
-        command = message[3].to_s.strip
+        thread_ts = message[1].strip
+        user = message[2].strip
+        user_name = message[3].strip
+        command = message[4].to_s.strip
         # take in consideration that on simulation we are treating all messages even those that are not populated on real cases like when the message is not populated to the specific bot connection when message is sent with the bot
         @logger.info "treat message: #{message}" if config.testing
         if command.match?(/^\s*\-!!/) or command.match?(/^\s*\-\^/)
           command.scan(/`([^`]+)`/).flatten.each do |cmd|
             if cmd.to_s != ""
               cmd = "^#{cmd}"
-              treat_message({ channel: channel, user: user, text: cmd, user_name: user_name }, false)
+              treat_message({ channel: channel, thread_ts: thread_ts, ts: thread_ts, user: user, text: cmd, user_name: user_name }, false)
             end
           end
         elsif command.match?(/^\s*\-!/)
           command.scan(/`([^`]+)`/).flatten.each do |cmd|
             if cmd.to_s != ""
               cmd = "!#{cmd}"
-              treat_message({ channel: channel, user: user, text: cmd, user_name: user_name }, false)
+              treat_message({ channel: channel, thread_ts: thread_ts, ts: thread_ts, user: user, text: cmd, user_name: user_name }, false)
             end
           end
         else
-          treat_message({ channel: channel, user: user, text: command, user_name: user_name })
+          treat_message({ channel: channel, thread_ts: thread_ts, ts: thread_ts, user: user, text: command, user_name: user_name })
         end
       end
     end
   end
 
   def listen
-    @pings = []
+    @pings ||= []
     @last_activity_check = Time.now
     get_bots_created()
 
@@ -72,7 +73,7 @@ class SlackSmartBot
     started = false
     while restarts < 200 and !started
       begin
-        @logger.info "Bot starting: #{config.inspect}"
+        @logger.info "Bot starting: #{@config_log.inspect}"
         client.start!
       rescue Slack::RealTime::Client::ClientAlreadyStartedError
         @logger.info "ClientAlreadyStarted so we continue with execution"
@@ -84,7 +85,7 @@ class SlackSmartBot
           @logger.info "*" * 50
           @logger.fatal "Rescued on starting: #{e.inspect}"
           @logger.info "Waiting 60 seconds to retry. restarts: #{restarts}"
-          puts "#{Time.now}: Not able to start client. Waiting 60 seconds to retry: #{config.inspect}"
+          puts "#{Time.now}: Not able to start client. Waiting 60 seconds to retry: #{@config_log.inspect}"
           sleep 60
         else
           exit!