slack-smart-bot 1.14.2 → 1.15.0

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

@@ -5,14 +5,22 @@ class SlackSmartBot
       get_personal_settings()
       @personal_settings.merge!(user_personal_settings)
     end
-    user = Thread.current[:user]
-    personal_settings_file = File.join(config.path, "personal_settings", "ps_#{user.name}.yaml")
+    user = Thread.current[:user].dup
+    team_id = user.team_id
+    team_id_user = Thread.current[:team_id_user]
+
+    unless Dir.exist?("#{config.path}/personal_settings/#{team_id}")
+      Dir.mkdir("#{config.path}/personal_settings/#{team_id}")
+    end
+
+    personal_settings_file = File.join(config.path, "personal_settings/#{team_id}", "ps_#{user.name}.yaml")
 
     File.open(personal_settings_file, 'w') {|file|
       file.flock(File::LOCK_EX)
-      file.write(Utils::Encryption.encrypt(@personal_settings[user.name].to_yaml, config))
+      file.write(Utils::Encryption.encrypt(@personal_settings[team_id_user].to_yaml, config))
       file.flock(File::LOCK_UN)
     }
+    get_personal_settings() #to update the @personal_settings_hash
     @datetime_personal_settings_file[personal_settings_file] = File.mtime(personal_settings_file)
   end
 end

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

@@ -1,8 +1,23 @@
 class SlackSmartBot
 
   def update_rules_imported
-    file = File.open("#{config.path}/rules/rules_imported.rb", "w")
-    file.write @rules_imported.inspect
-    file.close
+    require 'yaml'
+    file_path = "#{config.path}/rules/rules_imported"
+    if File.exist?("#{file_path}.rb") #backwards compatible
+      file_conf = IO.readlines("#{file_path}.rb").join
+      if file_conf.to_s() == ""
+        @rules_imported = {}
+      else
+        @rules_imported = eval(file_conf)
+      end
+      File.open("#{file_path}.yaml", 'w') {|file| file.write(@rules_imported.to_yaml) }
+      File.delete("#{file_path}.rb")
+    end
+
+    File.open("#{file_path}.yaml", 'w') {|file|
+      file.flock(File::LOCK_EX)
+      file.write(@rules_imported.to_yaml) 
+      file.flock(File::LOCK_UN)
+    }
   end
 end

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

@@ -6,11 +6,14 @@ class SlackSmartBot
       @vacations.merge!(vacation)
     end
     user = Thread.current[:user]
-    vacations_file = File.join(config.path, "vacations", "v_#{user.name}.yaml")
+    team_id_user = Thread.current[:team_id_user]
+    #create folder if doesn't exist
+    FileUtils.mkdir_p(File.join(config.path, "vacations/#{user.team_id}")) unless File.exist?(File.join(config.path, "vacations/#{user.team_id}"))
+    vacations_file = File.join(config.path, "vacations/#{user.team_id}", "v_#{user.name}.yaml")
 
     File.open(vacations_file, 'w') {|file|
       file.flock(File::LOCK_EX)
-      file.write(Utils::Encryption.encrypt(@vacations[user.name].to_yaml, config))
+      file.write(Utils::Encryption.encrypt(@vacations[team_id_user].to_yaml, config))
       file.flock(File::LOCK_UN)
     }
     @datetime_vacations_file[vacations_file] = File.mtime(vacations_file)

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

@@ -0,0 +1,276 @@
+class SlackSmartBot
+  def upgrade_to_use_team_ids()
+    team_id = config.team_id
+
+    if Dir.exist?("#{config.path}/rules/") #admins_channels.yaml and access_channels.yaml
+      files_updated = []
+      Dir.glob(config.path + "/rules/**/*").select { |i| i[/admins_channels.yaml$/] }.each do |f|
+        t = YAML.load_file(f)
+        n = {}
+        t.each do |k, v|
+          if v.length > 0
+            # do it only if it is not already a team_id: /^[A-Z0-9]{7,11}_/
+            # if it matches the pattern, it is already with a team_id
+            # if it doesn't match the pattern, it is not with a team_id
+            n[k] = []
+            v.each do |m|
+              if !m.to_s.match?(/^[A-Z0-9]{7,11}_/)
+                n[k] << team_id.to_s + "_" + m
+                files_updated << f unless files_updated.include?(f)
+              else
+                n[k] << m
+              end
+            end
+          else
+            n[k] = v
+          end
+        end
+        File.open(f, "w") { |f| f.write n.to_yaml } if files_updated.include?(f)
+      end
+      @logger.info "Updated admins_channels.yaml files to use team ids" unless files_updated.empty?
+
+      files_updated = []
+      Dir.glob(config.path + "/rules/**/*").select { |i| i[/access_channels.yaml$/] }.each do |f|
+        t = YAML.load_file(f)
+        n = {}
+        t.each do |k, v|
+          n[k] = {}
+          v.each do |k2, v2|
+            if v2.length > 0
+              # do it only if it is not already a team_id: /^[A-Z0-9]{7,11}_/
+              # if it matches the pattern, it is already with a team_id
+              # if it doesn't match the pattern, it is not with a team_id
+              n[k][k2] = []
+              v2.each do |m|
+                if !m.to_s.match?(/^[A-Z0-9]{7,11}_/)
+                  n[k][k2] << team_id.to_s + "_" + m
+                  files_updated << f unless files_updated.include?(f)
+                else
+                  n[k][k2] << m
+                end
+              end
+            else
+              n[k][k2] = v2
+            end
+          end
+        end
+        File.open(f, "w") { |f| f.write n.to_yaml } if files_updated.include?(f)
+
+      end
+      @logger.info "Updated access_channels.yaml files to use team ids" unless files_updated.empty?
+
+      files_updated = []
+
+      Dir.glob(config.path + "/rules/*").select { |i| i[/rules_imported.yaml$/] }.each do |f|
+        t = YAML.load_file(f)
+        n = {}
+        t.each do |k, v|
+          if !k.to_s.match?(/^[A-Z0-9]{7,11}_/)
+            n["#{team_id.to_s}_#{k}"] = v
+            files_updated << f unless files_updated.include?(f)
+          else
+            n[k] = v
+          end
+        end
+        File.open(f, "w") { |f| f.write n.to_yaml } if files_updated.include?(f)
+      end
+      @logger.info "Updated rules_imported.yaml files to use team ids" unless files_updated.empty?
+    end
+
+    if Dir.exist?("#{config.path}/teams/")
+      files_updated = []
+      #todo: do it also for deleted memos files
+      Dir.entries("#{config.path}/teams/").select { |i| i[/\.yaml$/] }.each do |f|
+        t = YAML.load(Utils::Encryption.decrypt(File.read("#{config.path}/teams/#{f}"), config))
+        t[:members].each do |k, v|
+          n = []
+          v.each do |m|
+            if !m.to_s.match?(/^[A-Z0-9]{7,11}_/)
+              n << team_id.to_s + "_" + m
+              files_updated << f unless files_updated.include?(f)
+            else
+              n << m
+            end
+          end
+          t[:members][k] = n
+        end
+
+        # update memos.user and memos.comments.user_name
+        if t.key?(:memos)
+          t[:memos].each do |m|
+            if !m[:user].to_s.match?(/^[A-Z0-9]{7,11}_/)
+              m[:user] = team_id.to_s + "_" + m[:user]
+              files_updated << f unless files_updated.include?(f)
+            end
+            if m.key?(:comments)
+              m[:comments].each do |c|
+                if !c[:user_name].to_s.match?(/^[A-Z0-9]{7,11}_/)
+                  c[:user_name] = team_id.to_s + "_" + c[:user_name]
+                  files_updated << f unless files_updated.include?(f)
+                end
+              end
+            end
+          end
+        end
+
+        if !t[:user].to_s.match?(/^[A-Z0-9]{7,11}_/)
+          t[:user] = team_id.to_s + "_" + t[:user]
+          files_updated << f unless files_updated.include?(f)
+        end
+        if !t[:creator].to_s.match?(/^[A-Z0-9]{7,11}_/)
+          t[:creator] = team_id.to_s + "_" + t[:creator]
+          files_updated << f unless files_updated.include?(f)
+        end
+        if files_updated.include?(f)
+          File.open("#{config.path}/teams/#{f}", "w") { |file|
+            file.flock(File::LOCK_EX)
+            file.write(Utils::Encryption.encrypt(t.to_yaml, config))
+            file.flock(File::LOCK_UN)
+          }
+        end
+      end
+      @logger.info "Updated teams files to use team ids" unless files_updated.empty?
+    end
+
+    #add team_id to the user_name: team_id_user_name, exclude the key :all
+    if Dir.exist?("#{config.path}/shortcuts/")
+      files_updated = []
+      Dir.entries("#{config.path}/shortcuts/").select { |i| i[/\.yaml$/] }.each do |f|
+        t = YAML.load_file("#{config.path}/shortcuts/#{f}")
+        n = {}
+        t.each do |k, v|
+          if k != :all and !k.to_s.match?(/^[A-Z0-9]{7,11}_/)
+            n[team_id.to_s + "_" + k.to_s] = t[k]
+            files_updated << f unless files_updated.include?(f)
+          else
+            n[k] = t[k]
+          end
+        end
+        File.open("#{config.path}/shortcuts/#{f}", "w") { |f| f.write n.to_yaml } if files_updated.include?(f)
+      end
+      @logger.info "Updated shortcuts files to use team ids" unless files_updated.empty?
+    end
+
+    #repl files are yaml
+    if Dir.exist?("#{config.path}/repl/")
+      files_updated = []
+      Dir.entries("#{config.path}/repl/").select { |i| i[/\.yaml$/] }.each do |f|
+        t = YAML.load_file("#{config.path}/repl/#{f}")
+        t.each do |k, v|
+          if v[:creator_team_id].to_s == ""
+            v[:creator_team_id] = team_id
+            files_updated << f unless files_updated.include?(f)
+          end
+        end
+        File.open("#{config.path}/repl/#{f}", "w") { |f| f.write t.to_yaml } if files_updated.include?(f)
+      end
+      @logger.info "Updated repl files to use team ids" unless files_updated.empty?
+    end
+
+    if Dir.exist?("#{config.path}/routines/")
+      files_updated = []
+      Dir.entries("#{config.path}/routines/").select { |i| i[/\.yaml$/] }.each do |f|
+        t = YAML.load_file("#{config.path}/routines/#{f}")
+        t.each do |k, v|
+          v.each do |k2, v2|
+            if v2[:creator_team_id].to_s == ""
+              v2[:creator_team_id] = team_id
+              files_updated << f unless files_updated.include?(f)
+            end
+          end
+        end
+        File.open("#{config.path}/routines/#{f}", "w") { |f| f.write t.to_yaml } if files_updated.include?(f)
+      end
+      @logger.info "Updated routines files to use team ids" unless files_updated.empty?
+    end
+
+    #shares files are csv
+    if Dir.exist?("#{config.path}/shares/")
+      Dir.entries("#{config.path}/shares/").select { |i| i[/\.csv$/] }.each do |f|
+        #verify if the file has the team_id already, in that case the number of columns should be 10
+        t = CSV.table("#{config.path}/shares/#{f}")
+        if t.headers.length == 8
+          t = CSV.table("#{config.path}/shares/#{f}", headers: ["share_id", "user_deleted", "user_created", "date", "time", "type", "to_channel", "condition"])
+          #save it in this order:
+          new_headers = ["share_id", "user_team_id_deleted", "user_deleted", "user_team_id_created", "user_created", "date", "time", "type", "to_channel", "condition"]
+          new_t = []
+          t.each do |m|
+            if m[:user_deleted].to_s == ""
+              user_team_id_deleted = ""
+            else
+              user_team_id_deleted = team_id
+            end
+            new_t << [m[:share_id], user_team_id_deleted, m[:user_deleted], team_id, m[:user_created], m[:date], m[:time], m[:type], m[:to_channel], m[:condition]]
+          end
+          CSV.open("#{config.path}/shares/#{f}", "wb") do |csv|
+            new_t.each do |row|
+              csv << row
+            end
+          end
+          @logger.info "Updated shares to use team ids"
+        end
+      end
+    end
+
+    #announcements
+    if Dir.exist?("#{config.path}/announcements/")
+      Dir.entries("#{config.path}/announcements/").select { |i| i[/\.csv$/] }.each do |f|
+        #verify if the file has the team_id already, in that case the number of columns should be 9
+        t = CSV.table("#{config.path}/announcements/#{f}")
+        if t.headers.length == 7
+          t = CSV.table("#{config.path}/announcements/#{f}", headers: ["message_id", "user_deleted", "user_created", "date", "time", "type", "message"])
+          #save it in this order:
+          new_headers = ["message_id", "user_team_id_deleted", "user_deleted", "user_team_id_created", "user_created", "date", "time", "type", "message"]
+          new_t = []
+          t.each do |m|
+            if m[:user_deleted].to_s == ""
+              user_team_id_deleted = ""
+            else
+              user_team_id_deleted = team_id
+            end
+            new_t << [m[:message_id], user_team_id_deleted, m[:user_deleted], team_id, m[:user_created], m[:date], m[:time], m[:type], m[:message]]
+          end
+          CSV.open("#{config.path}/announcements/#{f}", "wb") do |csv|
+            new_t.each do |row|
+              csv << row
+            end
+          end
+
+          @logger.info "Updated announcements to use team ids"
+        end
+      end
+    end
+
+    if Dir.exist?("#{config.path}/vacations/") and !Dir.exist?(config.path + "/vacations/" + team_id.to_s)
+      FileUtils.mkdir_p(config.path + "/vacations/" + team_id.to_s)
+      files_updated = []
+      Dir.glob(config.path + "/vacations/*").select { |i| i[/\.yaml$/] }.each do |f|
+        FileUtils.mv(f, config.path + "/vacations/" + team_id.to_s)
+        files_updated << f
+      end
+      @logger.info "Updated vacations to use team ids. All moved to #{config.path}/vacations/#{team_id}/" unless files_updated.empty?
+    end
+
+    if Dir.exist?("#{config.path}/openai/") and !Dir.exist?(config.path + "/openai/" + team_id.to_s)
+      FileUtils.mkdir_p(config.path + "/openai/" + team_id.to_s)
+      files_updated = []
+      Dir.glob(config.path + "/openai/*").each do |file_or_dir|
+        next if file_or_dir == config.path + "/openai/" + team_id.to_s
+        FileUtils.mv(file_or_dir, config.path + "/openai/" + team_id.to_s)
+        files_updated << file_or_dir
+      end
+      @logger.info "Updated openai to use team ids. All moved to #{config.path}/openai/#{team_id}/" unless files_updated.empty?
+    end
+
+    if Dir.exist?("#{config.path}/personal_settings/") and !Dir.exist?(config.path + "/personal_settings/" + team_id.to_s)
+      FileUtils.mkdir_p(config.path + "/personal_settings/" + team_id.to_s)
+      files_updated = []
+      Dir.glob(config.path + "/personal_settings/*").each do |file_or_dir|
+        next if file_or_dir == config.path + "/personal_settings/" + team_id.to_s
+        FileUtils.mv(file_or_dir, config.path + "/personal_settings/" + team_id.to_s)
+        files_updated << file_or_dir
+      end
+      @logger.info "Updated personal_settings to use team ids. All moved to #{config.path}/personal_settings/#{team_id}/" unless files_updated.empty?
+    end
+  end
+end

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

@@ -36,4 +36,9 @@ require_relative 'utils/encryption/encryption_get_key_iv'
 require_relative 'utils/encryption/encrypt'
 require_relative 'utils/encryption/decrypt'
 require_relative 'utils/get_team_members'
-
+require_relative 'utils/get_openai_sessions'
+require_relative 'utils/update_openai_sessions'
+require_relative 'utils/local_time'
+require_relative 'utils/get_countries_candelarific'
+require_relative 'utils/upgrade_to_use_team_ids'
+require_relative 'utils/find_user'