slack-smart-bot 1.9.1 → 1.11.0

data/lib/slack/smart-bot/utils/get_routines.rb

@@ -1,11 +1,31 @@
 class SlackSmartBot
 
   def get_routines(channel = @channel_id)
-    if File.exist?("#{config.path}/routines/routines_#{channel}.rb")
+    require 'yaml'
+    routines_file = "#{config.path}/routines/routines_#{channel}.yaml"
+
+    if File.exist?("#{config.path}/routines/routines_#{channel}.rb") #backwards compatible
       file_conf = IO.readlines("#{config.path}/routines/routines_#{channel}.rb").join
-      unless file_conf.to_s() == ""
+      if file_conf.to_s() == ""
+        @routines = {}
+      else
         @routines = eval(file_conf)
       end
+      File.open(routines_file, 'w') {|file| file.write(@routines.to_yaml) }
+      File.delete("#{config.path}/routines/routines_#{channel}.rb")
+    end
+
+    if File.exist?(routines_file)      
+      routines = @routines
+      10.times do
+        routines = YAML.load(File.read(routines_file))
+        if routines.is_a?(Hash)
+          break
+        else
+          sleep (0.1*(rand(2)+1))
+        end
+      end
+      @routines = routines unless routines.is_a?(FalseClass)
     end
   end
 end

data/lib/slack/smart-bot/utils/get_shares.rb

@@ -0,0 +1,12 @@
+class SlackSmartBot
+
+  def get_shares()
+    channel = @channels_name[@channel_id]
+    if File.exist?("#{config.path}/shares/#{channel}.csv")
+      "#{config.path}/shares/#{channel}.csv"
+      t = CSV.table("#{config.path}/shares/#{channel}.csv", headers: ['share_id', 'user_deleted', 'user_created', 'date', 'time', 'type', 'to_channel', 'condition'])
+      t.delete_if {|row| row[:user_deleted] != ''}
+      @shares[channel] = t
+    end
+  end
+end

data/lib/slack/smart-bot/utils/get_teams.rb

@@ -0,0 +1,22 @@
+class SlackSmartBot
+  def get_teams
+    @teams ||= {}
+    teams_file = config.file_path.gsub(".rb", "_teams.yaml")
+    if File.exist?(teams_file)
+      if !defined?(@datetime_teams_file) or @datetime_teams_file != File.mtime(teams_file)
+        require 'yaml'
+        teams = @teams
+        10.times do
+          teams = YAML.load(File.read(teams_file))
+          if teams.is_a?(Hash)
+            break
+          else
+            sleep (0.1*(rand(2)+1))
+          end
+        end
+        @teams = teams unless teams.is_a?(FalseClass)
+        @datetime_teams_file = File.mtime(teams_file)
+      end
+    end
+  end
+end

data/lib/slack/smart-bot/utils/has_access.rb

@@ -0,0 +1,28 @@
+class SlackSmartBot
+  def has_access?(method, user = nil)
+    user = Thread.current[:user] if user.nil?
+    if config[:allow_access].key?(method) and !config[:allow_access][method].include?(user.name) and !config[:allow_access][method].include?(user.id) and
+       (!user.key?(:enterprise_user) or (user.key?(:enterprise_user) and !config[:allow_access][method].include?(user[:enterprise_user].id)))
+      respond "You don't have access to use this command, please contact an Admin to be able to use it: <@#{config.admins.join(">, <@")}>"
+      return false
+    else
+      if Thread.current[:typem] == :on_call
+        channel = Thread.current[:dchannel]
+      elsif Thread.current[:using_channel].to_s == ""
+        channel = Thread.current[:dest]
+      else
+        channel = Thread.current[:using_channel]
+      end
+      if !@access_channels.key?(channel) or !@access_channels[channel].key?(method.to_s) or @access_channels[channel][method.to_s].include?(user.name)
+        return true
+      else
+        if @admins_channels.key?(channel) and !@admins_channels[channel].empty?
+            respond "You don't have access to use this command, please contact an Admin to be able to use it: <@#{@admins_channels[channel].join(">, <@")}>"
+        else
+            respond "You don't have access to use this command, please contact an Admin to be able to use it."
+        end        
+        return false
+      end
+    end
+  end
+end

data/lib/slack/smart-bot/utils/is_admin.rb

@@ -0,0 +1,27 @@
+class SlackSmartBot
+    def is_admin?(from=nil)
+        if from.nil?
+            user = Thread.current[:user]
+            from = user.name
+        end
+        
+        if (Thread.current[:dchannel].to_s!='' and !@channels_creator.key?(Thread.current[:dchannel])) or 
+            (Thread.current[:dest].to_s!='' and Thread.current[:dest][0]!='D' and !@channels_creator.key?(Thread.current[:dest])) or 
+            (Thread.current[:using_channel].to_s!='' and !@channels_creator.key?(:using_channel))
+            get_channels_name_and_id() 
+        end
+
+        if config.masters.include?(from) or 
+           config.admins.include?(from) or 
+           (Thread.current[:typem] == :on_call and @admins_channels.key?(Thread.current[:dchannel]) and @admins_channels[Thread.current[:dchannel]].include?(from)) or
+           (Thread.current[:using_channel].to_s == '' and @admins_channels.key?(Thread.current[:dest]) and @admins_channels[Thread.current[:dest]].include?(from)) or
+           (@admins_channels.key?(Thread.current[:using_channel]) and @admins_channels[Thread.current[:using_channel]].include?(from)) or 
+           (Thread.current[:using_channel].to_s=='' and @channels_creator.key?(Thread.current[:dest]) and from == @channels_creator[Thread.current[:dest]]) or 
+           (Thread.current[:typem] == :on_call  and @channels_creator.key?(Thread.current[:dchannel]) and from == @channels_creator[Thread.current[:dchannel]]) or 
+           (@channels_creator.key?(Thread.current[:using_channel]) and from == @channels_creator[Thread.current[:using_channel]])
+            return true
+        else
+            return false
+        end
+    end
+end

data/lib/slack/smart-bot/utils/save_stats.rb

@@ -1,45 +1,50 @@
 class SlackSmartBot
-
-    def save_stats(method, data: {})
-        if config.stats
-            begin
-              require 'csv'
-              if !File.exist?("#{config.stats_path}.#{Time.now.strftime("%Y-%m")}.log")
-                CSV.open("#{config.stats_path}.#{Time.now.strftime("%Y-%m")}.log", 'wb') do |csv|
-                    csv << ['date','bot_channel', 'bot_channel_id', 'dest_channel', 'dest_channel_id', 'type_message', 'user_name', 'user_id', 'text', 'command', 'files']
-                end
-              end
-              if data.empty?
-                data = {
-                  dest: Thread.current[:dest],
-                  typem: Thread.current[:typem],
-                  user: Thread.current[:user],
-                  files: Thread.current[:files?],
-                  command: Thread.current[:command],
-                  routine: Thread.current[:routine]
-                }
-              end
-              if method.to_s == 'ruby_code' and data.files
-                command_txt = 'ruby'
-              else
-                command_txt = data.command
-              end
-
-              if data.routine
-                user_name = "routine/#{data.user.name}"
-                user_id = "routine/#{data.user.id}"
-              else
-                user_name = data.user.name
-                user_id = data.user.id
-              end
-              CSV.open("#{config.stats_path}.#{Time.now.strftime("%Y-%m")}.log", "a+") do |csv|
-                csv << [Time.now, config.channel, @channel_id, @channels_name[data.dest], data.dest, data.typem, user_name, user_id, command_txt, method, data.files]
-              end
-            rescue Exception => exception
-              @logger.fatal "There was a problem on the stats"
-              @logger.fatal exception
+  def save_stats(method, data: {})
+    if has_access?(method, Thread.current[:user])
+      if config.stats
+        begin
+          require "csv"
+          if !File.exist?("#{config.stats_path}.#{Time.now.strftime("%Y-%m")}.log")
+            CSV.open("#{config.stats_path}.#{Time.now.strftime("%Y-%m")}.log", "wb") do |csv|
+              csv << ["date", "bot_channel", "bot_channel_id", "dest_channel", "dest_channel_id", "type_message", "user_name", "user_id", "text", "command", "files"]
             end
+          end
+          if data.empty?
+            data = {
+              dest: Thread.current[:dest],
+              typem: Thread.current[:typem],
+              user: Thread.current[:user],
+              files: Thread.current[:files?],
+              command: Thread.current[:command],
+              routine: Thread.current[:routine],
+            }
+          end
+          if method.to_s == "ruby_code" and data.files
+            command_txt = "ruby"
+          else
+            command_txt = data.command
+          end
+          command_txt.gsub!(/```.+```/m, "```CODE```")
+          command_txt = "#{command_txt[0..99]}..." if command_txt.size > 100
+
+          if data.routine
+            user_name = "routine/#{data.user.name}"
+            user_id = "routine/#{data.user.id}"
+          else
+            user_name = data.user.name
+            user_id = data.user.id
+          end
+          CSV.open("#{config.stats_path}.#{Time.now.strftime("%Y-%m")}.log", "a+") do |csv|
+            csv << [Time.now, config.channel, @channel_id, @channels_name[data.dest], data.dest, data.typem, user_name, user_id, command_txt, method, data.files]
+          end
+        rescue Exception => exception
+          @logger.fatal "There was a problem on the stats"
+          @logger.fatal exception
         end
+      end
+    else
+      sleep 0.2
+      Thread.exit
     end
-
-end  
+  end
+end

data/lib/slack/smart-bot/utils/save_status.rb

@@ -0,0 +1,67 @@
+class SlackSmartBot
+
+    def save_status(status, status_id, message)
+      require 'csv'
+      Dir.mkdir("#{config.path}/status") unless Dir.exist?("#{config.path}/status")
+
+      CSV.open("#{config.path}/status/#{config.channel}_status.csv", "a+") do |csv|
+        csv << [Time.now.strftime("%Y/%m/%d"), Time.now.strftime("%H:%M:%S"), status, status_id, message]
+      end
+      if defined?(@channels_list) #wait until the 'client' started
+        channel_info = @channels_list.select { |c| c.id == @channel_id}[-1]
+        if channel_info.nil? or channel_info.is_private
+          channel_link = "##{config.channel}"
+        else
+          channel_link = "<##{@channel_id}|#{config.channel}>"
+        end
+      else
+        channel_link = "##{config.channel}"
+      end
+
+      if status_id == :disconnected
+        Thread.new do
+          sleep 50
+          @logger.info "check disconnection 50 scs later #{@last_notified_status_id}"
+          unless @last_notified_status_id == :connected
+            respond ":red_circle: The *SmartBot* on *#{channel_link}* is down. An admin will take a look. <@#{config.admins.join(">, <@")}>", config.status_channel
+          end
+        end
+      end
+      if @channels_id.is_a?(Hash) and @channels_id.keys.include?(config.status_channel)
+        is_back = false
+        m = ''
+        if (Time.now-@last_status_change) > 20 or !defined?(@last_notified_status_id)
+          if status_id == :connected
+            if defined?(@last_notified_status_id)
+              m = ":exclamation: :large_green_circle: The *SmartBot* on *#{channel_link}* was not available for #{(Time.now-@last_status_change).round(0)} secs. *Now it is up and running again.*" 
+            else
+              m = ":large_green_circle: The *SmartBot* on *#{channel_link}* is up and running again." 
+            end
+          end
+        end
+        if status_id == :paused
+          m = ":red_circle: #{message} *#{channel_link}*"
+        elsif status_id == :started
+          m = ":large_green_circle: #{message} *#{channel_link}*"            
+        elsif status_id == :killed or status_id == :exited
+          m = ":red_circle: #{message}"
+        elsif config.on_master_bot and status_id == :maintenance_on
+          if message.to_s == "Sorry I'm on maintenance so I cannot attend your request."
+            m = ":red_circle: The *SmartBot* is on maintenance so not possible to attend any request."
+          else
+            m = ":red_circle: #{message}"
+          end
+        elsif config.on_master_bot and status_id == :maintenance_off
+          m = ":large_green_circle: The *SmartBot* is up and running again."
+        end
+        @last_status_change = Time.now
+        @last_notified_status_id = status_id
+        unless m == ''
+          respond m, config.status_channel
+        end
+      end
+
+
+    end
+
+end  

data/lib/slack/smart-bot/utils/update_access_channels.rb

@@ -0,0 +1,8 @@
+class SlackSmartBot
+
+  def update_access_channels()
+    file = File.open("#{config.path}/rules/#{@channel_id}/access_channels.rb", "w")
+    file.write (@access_channels.inspect)
+    file.close
+  end
+end

data/lib/slack/smart-bot/utils/update_admins_channels.rb

@@ -0,0 +1,8 @@
+class SlackSmartBot
+
+  def update_admins_channels()
+    file = File.open("#{config.path}/rules/#{@channel_id}/admins_channels.rb", "w")
+    file.write (@admins_channels.inspect)
+    file.close
+  end
+end

data/lib/slack/smart-bot/utils/update_bots_file.rb

@@ -1,11 +1,32 @@
 class SlackSmartBot
   def update_bots_file
-    file = File.open(config.file_path.gsub(".rb", "_bots.rb"), "w")
-    bots_created = @bots_created.dup
-    bots_created.each { |k, v| 
-      v[:thread] = "" 
-    }
-    file.write bots_created.inspect
-    file.close
+    bots_file = config.file_path.gsub(".rb", "_bots.yaml")
+
+    if File.exist?(config.file_path.gsub(".rb", "_bots.rb")) #backwards compatible
+      file_conf = IO.readlines(config.file_path.gsub(".rb", "_bots.rb")).join
+      if file_conf.to_s() == ""
+        @bots_created = {}
+      else
+        @bots_created = eval(file_conf)
+      end      
+      File.open(bots_file, 'w') {|file| 
+        file.flock(File::LOCK_EX)
+        file.write(@bots_created.to_yaml) 
+        file.flock(File::LOCK_UN)
+      }
+      File.delete(config.file_path.gsub(".rb", "_bots.rb"))
+    else
+      #not possible to use @bots_created.deep_copy since one of the fields contains a thread
+      bots_created = {}
+      @bots_created.each do |k,v|
+        bots_created[k] = v.dup
+        bots_created[k][:thread] = ''
+      end
+      File.open(bots_file, 'w') {|file|
+        file.flock(File::LOCK_EX)
+        file.write(bots_created.to_yaml) 
+        file.flock(File::LOCK_UN)
+      }
+    end
   end
 end

data/lib/slack/smart-bot/utils/update_repls.rb

@@ -1,8 +1,11 @@
 class SlackSmartBot
-
   def update_repls(channel = @channel_id)
-    file = File.open("#{config.path}/repl/repls_#{channel}.rb", "w")
-    file.write (@repls.inspect)
-    file.close
+    require 'yaml'
+    repl_file = "#{config.path}/repl/repls_#{channel}.yaml"
+    File.open(repl_file, 'w') {|file|
+      file.flock(File::LOCK_EX)
+      file.write(@repls.to_yaml) 
+      file.flock(File::LOCK_UN)
+    }
   end
 end

data/lib/slack/smart-bot/utils/update_routines.rb

@@ -1,8 +1,11 @@
 class SlackSmartBot
 
   def update_routines(channel = @channel_id)
+
+    require 'yaml'
+    routines_file = "#{config.path}/routines/routines_#{channel}.yaml"
+
     routines = {}
-    file = File.open("#{config.path}/routines/routines_#{channel}.rb", "w")
     @routines.each do |k,v|
       routines[k]={}
       v.each do |kk,vv|
@@ -10,7 +13,10 @@ class SlackSmartBot
         routines[k][kk][:thread]=""
       end
     end
-    file.write (routines.inspect)
-    file.close
+    File.open(routines_file, 'w') {|file|
+      file.flock(File::LOCK_EX)
+      file.write(routines.to_yaml) 
+      file.flock(File::LOCK_UN)
+    }
   end
 end

data/lib/slack/smart-bot/utils/update_shortcuts_file.rb

@@ -1,13 +1,20 @@
 class SlackSmartBot
   def update_shortcuts_file
-    file = File.open("#{config.path}/shortcuts/#{config.shortcuts_file}", "w")
-    file.write @shortcuts.inspect
-    file.close
+    require 'yaml'
+    sc_file = "#{config.path}/shortcuts/#{config.shortcuts_file}"
+    File.open(sc_file, 'w') {|file|
+      file.flock(File::LOCK_EX)
+      file.write(@shortcuts.to_yaml) 
+      file.flock(File::LOCK_UN)
+    }
 
     if config.on_master_bot
-      file = File.open("#{config.path}/shortcuts/shortcuts_global.rb", "w")
-      file.write @shortcuts_global.inspect
-      file.close
+      sc_file = "#{config.path}/shortcuts/shortcuts_global.yaml"
+      File.open(sc_file, 'w') {|file|
+        file.flock(File::LOCK_EX)
+        file.write(@shortcuts_global.to_yaml) 
+        file.flock(File::LOCK_UN)
+      }
     end
   end
 end

data/lib/slack/smart-bot/utils/update_teams.rb

@@ -0,0 +1,16 @@
+class SlackSmartBot
+  def update_teams(team=nil)
+    require 'yaml'
+    unless team.nil?
+      get_teams()
+      @teams.merge!(team)
+    end
+    teams_file = config.file_path.gsub(".rb", "_teams.yaml")
+    File.open(teams_file, 'w') {|file|
+      file.flock(File::LOCK_EX)
+      file.write(@teams.to_yaml) 
+      file.flock(File::LOCK_UN)
+    }
+    @datetime_teams_file = File.mtime(teams_file)
+  end
+end

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

@@ -4,6 +4,7 @@ require_relative 'utils/get_bots_created'
 require_relative 'utils/get_channels_name_and_id'
 require_relative 'utils/get_help'
 require_relative 'utils/get_routines'
+require_relative 'utils/get_shares'
 require_relative 'utils/get_rules_imported'
 require_relative 'utils/remove_hash_keys'
 require_relative 'utils/update_bots_file'
@@ -15,4 +16,14 @@ require_relative 'utils/get_repls'
 require_relative 'utils/update_repls'
 require_relative 'utils/answer'
 require_relative 'utils/answer_delete'
+require_relative 'utils/has_access'
+require_relative 'utils/save_status'
+require_relative 'utils/get_admins_channels'
+require_relative 'utils/update_admins_channels'
+require_relative 'utils/is_admin'
+require_relative 'utils/get_access_channels'
+require_relative 'utils/update_access_channels'
+require_relative 'utils/get_command_ids'
+require_relative 'utils/get_teams'
+require_relative 'utils/update_teams'