slack-smart-bot 1.13.2 → 1.14.0

data/lib/slack/smart-bot/commands/general/teams/see_teams.rb

@@ -0,0 +1,236 @@
+class SlackSmartBot
+  module Commands
+    module General
+      module Teams
+        def see_teams(user, team_name, search = "", add_stats: true)
+          save_stats(__method__) if add_stats
+
+          get_teams()
+          teams = @teams.deep_copy
+          if teams.empty?
+            respond "There are no teams added yet. Use `add team` command to add a team."
+          elsif team_name.to_s != "" and !teams.key?(team_name.to_sym) and (teams.keys.select { |t| (t.to_s.gsub("-", "").gsub("_", "") == team_name.to_s) }).empty?
+            respond "It seems like the team *#{team_name}* doesn't exist.\nRelated commands `add team TEAM_NAME PROPERTIES`, `see team TEAM_NAME`, `see teams`"
+          else
+            users_link = (Thread.current[:dest][0] == "D")
+            filter = (search != "")
+            react :runner
+            @users = get_users() if add_stats
+
+            messages = []
+            search_members = []
+            search_channels = []
+            search_info = []
+            if filter
+              search.split(" ").each do |s|
+                if s.match(/<@(\w+)>/i)
+                  m = $1
+                  user_info = @users.select { |u| u.id.downcase == m.downcase or (u.key?(:enterprise_user) and u.enterprise_user.id.downcase == m.downcase) }[-1]
+                  search_members << user_info.name unless user_info.nil?
+                elsif s.match(/<#(\w+)\|[^>]*>/i)
+                  c = $1.upcase
+                  search_channels << @channels_name[c] if @channels_name.key?(c)
+                else
+                  search_info << s
+                end
+              end
+            end
+            if team_name.to_s == "" and search.to_s == ""
+              dest = :on_thread
+              messages.unshift("Since there are many lines returned the results are returned on a thread by default.")
+            else
+              dest = Thread.current[:dest]
+            end
+
+            teams.each do |name, team|
+              filter ? add = false : add = true
+              if team_name.to_s == "" or (team_name.to_s == name.to_s) or (name.to_s.gsub("-", "").gsub("_", "") == team_name.to_s)
+                message = []
+                message << "*#{name.capitalize}*"
+
+                if filter and search_info.size > 0
+                  all_info = true
+                  search_info.each do |s|
+                    if (team.members.keys.find { |e| /#{s}/i =~ e })
+                      add = true
+                      break
+                    end
+                    if !name.match?(/#{s}/i)
+                      all_info = false
+                      break
+                    end
+                  end
+                  add = true if all_info
+                end
+
+                message << "   > *_members_*"
+
+                assigned_members, unassigned_members, not_on_team_channel, channels_members, all_team_members = get_team_members(team)
+
+                unless unassigned_members.empty?
+                  um = unassigned_members.dup
+                  um.each do |m|
+                    user_info = @users.select { |u| u.name == m or (u.key?(:enterprise_user) and u.enterprise_user.name == m) }[-1]
+                    unless user_info.nil? or user_info.profile.title.to_s == ""
+                      team.members[user_info.profile.title.to_snake_case] ||= []
+                      team.members[user_info.profile.title.to_snake_case] << m
+                      unassigned_members.delete(m)
+                    end
+                  end
+                  unless unassigned_members.empty?
+                    team.members["unassigned"] ||= []
+                    team.members["unassigned"] += unassigned_members
+                    team.members["unassigned"].sort!
+                  end
+                end
+                unless not_on_team_channel.empty?
+                  team.members["not on members channel"] = not_on_team_channel
+                  team.members["not on members channel"].sort!
+                end
+                add = true if (team.members.values.flatten & search_members).size > 0
+                add = true if (team.channels.values.flatten & search_channels).size > 0
+                if filter and search_info.size > 0
+                  all_info = true
+                  search_info.each do |s|
+                    if (team.members.keys.find { |e| /#{s}/i =~ e })
+                      add = true
+                      break
+                    end
+                    if !team.info.match?(/#{s}/i)
+                      all_info = false
+                      break
+                    end
+                  end
+                  add = true if all_info
+                end
+
+                if add
+                  if team_name.to_s != ""
+                    team.members.each do |type, members|
+                      message << "        _`#{type}`_:  "
+                      members.each do |member|
+                        types = [":palm_tree:", ":spiral_calendar_pad:", ":face_with_thermometer:", ":baby:"]
+                        member_info = @users.select { |u| u.name == member }[-1]
+                        if !member_info.nil? and !member_info.deleted
+                          member_id = member_info.id
+                          info = get_user_info(member_id)
+                          emoji = info.user.profile.status_emoji
+                          if types.include?(emoji)
+                            status = emoji
+                          else
+                            active = (get_presence(member_id).presence.to_s == "active")
+                            if active
+                              user_info = @users.select { |u| u.id == member_id or (u.key?(:enterprise_user) and u.enterprise_user.name == member_id) }[-1]
+                              if (user_info.tz_offset - user.tz_offset).abs <= (4 * 3600)
+                                status = ":large_green_circle:"
+                              else
+                                status = ":large_yellow_circle:"
+                              end
+                            else
+                              status = ":white_circle:"
+                            end
+                          end
+                        else
+                          status = ":exclamation:"
+                        end
+                        unless status == ":exclamation:"
+                          if users_link
+                            message[-1] += "  #{status}<@#{member}>, "
+                          else
+                            user_info = @users.select { |u| u.name == member or (u.key?(:enterprise_user) and u.enterprise_user.name == member) }[-1]
+                            unless user_info.nil?
+                              if user_info.profile.display_name == ""
+                                name = user_info.name
+                              else
+                                name = user_info.profile.display_name
+                              end
+                              message[-1] += "  #{status} #{name}, "
+                            end
+                          end
+                        end
+                      end
+                      message[-1].chop!
+                      message[-1].chop!
+                    end
+                  else
+                    team.members.each do |type, members|
+                      if users_link
+                        message << "        _`#{type}`_:  <@#{members.join(">,  <@")}>"
+                      else
+                        membersn = []
+                        members.each do |m|
+                          user_info = @users.select { |u| u.name == m or (u.key?(:enterprise_user) and u.enterprise_user.name == m) }[-1]
+                          unless user_info.nil? or user_info.deleted
+                            if user_info.profile.display_name == ""
+                              name = user_info.name
+                            else
+                              name = user_info.profile.display_name
+                            end
+                            membersn << name
+                          end
+                        end
+                        message << "        _`#{type}`_:  #{membersn.join("  /  ")}"
+                      end
+                    end
+                  end
+                end
+
+                if add
+                  message << "   > *_channels_*"
+                  team.channels.each do |type, channels|
+                    channel_ids = []
+                    channels.each do |ch|
+                      channel_info = @channels_list.select { |c| c.name.to_s.downcase == ch.to_s.downcase }[-1]
+                      if @channels_id.key?(ch) and (!channel_info.is_private or (channel_info.is_private and (team.members.values + [team.creator]).flatten.include?(user.name)))
+                        channel_ids << @channels_id[ch]
+                      end
+                    end
+                    message << "        _`#{type}`_:  <##{channel_ids.join("> <#")}>" unless channel_ids.empty?
+                  end
+
+                  unless !team.key?(:memos) or team.memos.empty? or (team_name.to_s == "" and search.to_s == "")
+                    message += see_memos_team(user, type: "all", add_stats: false, team: team)
+                  end
+
+                  unless team.info.empty?
+                    team.info.split("\n").each do |m|
+                      message << ">#{m}"
+                    end
+                    message << "> "
+                    message << "> "
+                  end
+                  messages << message.join("\n")
+                end
+              end
+            end
+            unreact :runner
+            if messages.empty?
+              if filter
+                respond "It seems like we didn't find any team with the criteria supplied. Call `see teams` for a full list of teams."
+              else
+                respond "It seems like there are no teams added.\nUse `add team TEAM_NAME PROPERTIES` to add one. Call `bot help add team` for extended info."
+              end
+            else
+              if team_name.to_s != ""
+                message = "\n\n:palm_tree: On vacation / "
+                message += ":spiral_calendar_pad: In a meeting / "
+                message += ":face_with_thermometer: :baby: Sick leave / "
+                message += ":white_circle: Away / "
+                message += ":large_yellow_circle: Available in remote timezone / "
+                message += ":large_green_circle: Available"
+                messages[-1] << message
+                messages[-1] << "\n:information_source: Remote Time zone is >4h away from your current (#{user.tz_label})"
+              end
+              messages.each do |msg|
+                respond msg, dest, unfurl_links: false, unfurl_media: false
+              end
+              unless team_name.to_s.empty?
+                see_vacations_team(user, team_name, Date.today.strftime("%Y/%m/%d"), add_stats: false)
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/slack/smart-bot/commands/general/teams/see_vacations_team.rb

@@ -0,0 +1,183 @@
+class SlackSmartBot
+  module Commands
+    module General
+      module Teams
+        def see_vacations_team(user, team_name, date, add_stats: true)
+          save_stats(__method__) if add_stats
+
+          get_teams()
+          teams = @teams.deep_copy
+          if teams.empty?
+            respond "There are no teams added yet. Use `add team` command to add a team."
+          elsif team_name.to_s != "" and !teams.key?(team_name.to_sym) and (teams.keys.select { |t| (t.to_s.gsub("-", "").gsub("_", "") == team_name.to_s) }).empty?
+            respond "It seems like the team *#{team_name}* doesn't exist.\nRelated commands `add team TEAM_NAME PROPERTIES`, `see team TEAM_NAME`, `see teams`"
+          else
+            teams.each do |name, team|
+              if team_name == name.to_s or (name.to_s.gsub("-", "").gsub("_", "") == team_name.to_s)
+                team_name = name.to_s
+                break
+              end
+            end
+            date.gsub!("-", "/")
+            get_vacations()
+            team = teams[team_name.to_sym]
+            assigned_members = team.members.values.flatten
+            assigned_members.uniq!
+            assigned_members.dup.each do |m|
+              user_info = @users.select { |u| u.id == m or (u.key?(:enterprise_user) and u.enterprise_user.id == m) or u.name == m or (u.key?(:enterprise_user) and u.enterprise_user.name == m) }[-1]
+              assigned_members.delete(m) if user_info.nil? or user_info.deleted
+            end
+
+            channels_members = []
+            all_team_members = assigned_members.dup
+            if team.channels.key?("members")
+              team_members = []
+              team.channels["members"].each do |ch|
+                get_channels_name_and_id() unless @channels_id.key?(ch)
+                tm = get_channel_members(@channels_id[ch])
+                if tm.nil?
+                  respond ":exclamation: Add the Smart Bot to *##{ch}* channel to be able to get the list of members.", dest
+                else
+                  channels_members << @channels_id[ch]
+                  tm.each do |m|
+                    user_info = @users.select { |u| u.id == m or (u.key?(:enterprise_user) and u.enterprise_user.id == m) }[-1]
+                    team_members << user_info.name unless user_info.is_app_user or user_info.is_bot
+                  end
+                end
+              end
+              team_members.flatten!
+              all_team_members += team_members
+              all_team_members.uniq!
+            end
+            unless all_team_members.empty?
+              blocks_header =
+                {
+                  "type": "context",
+                  elements: [
+                    {
+                      type: "mrkdwn",
+                      text: "*Time Off #{team_name} team* from #{date} ",
+                    },
+                  ],
+                }
+
+              from = Date.parse(date, "%Y/%m/%d")
+              blocks = []
+              if config[:public_holidays].key?(:default_calendar)
+                defaulted_country_region = config[:public_holidays][:default_calendar].downcase
+              else
+                defaulted_country_region = ""
+              end
+              all_team_members.each do |m|
+                @users = get_users() if @users.empty?
+                info = @users.select { |u| u.id == m or (u.key?(:enterprise_user) and u.enterprise_user.id == m) or u.name == m or (u.key?(:enterprise_user) and u.enterprise_user.name == m) }[-1]
+                unless info.nil?
+                  country_region = ""
+                  if @vacations.key?(m) and @vacations[m][:public_holidays].to_s != ""
+                    country_region = @vacations[m][:public_holidays].downcase
+                  elsif config[:public_holidays].key?(:default_calendar) and country_region.empty?
+                    country_region = defaulted_country_region
+                  end
+
+                  info = get_user_info(info.id)
+                  if @vacations.key?(m)
+                    v = ""
+                    (from..(from + 20)).each do |d|
+                      v += "#{d.strftime("%d")} " if d.wday == 1 or d == from
+                      on_vacation = false
+                      @vacations[m].periods ||= []
+                      @vacations[m].periods.each do |p|
+                        if p.from <= d.strftime("%Y/%m/%d") and p.to >= d.strftime("%Y/%m/%d")
+                          if d.wday == 0 or d.wday == 6
+                            v += ":large_orange_square: "
+                          else
+                            v += ":large_red_square: "
+                          end
+                          on_vacation = true
+                          break
+                        end
+                      end
+                      unless on_vacation
+                        if country_region != "" and (!@public_holidays.key?(country_region) or !@public_holidays[country_region].key?(d.year.to_s))
+                          country, location = country_region.split("/")
+                          public_holidays(country.to_s, location.to_s, d.year.to_s, "", "", add_stats: false, publish_results: false)
+                        end
+                        if @public_holidays.key?(country_region) and @public_holidays[country_region].key?(d.year.to_s)
+                          phd = @public_holidays[country_region][d.year.to_s].date.iso
+                        else
+                          phd = []
+                        end
+                        date_text = d.strftime("%Y-%m-%d")
+                        if phd.include?(date_text)
+                          v += ":large_red_square: "
+                        elsif d.wday == 0 or d.wday == 6
+                          v += ":large_yellow_square: "
+                        else
+                          v += ":white_square: "
+                        end
+                      end
+                    end
+                  else
+                    v = ""
+                    (from..(from + 20)).each do |d|
+                      if country_region != "" and (!@public_holidays.key?(country_region) or !@public_holidays[country_region].key?(d.year.to_s))
+                        country, location = country_region.split("/")
+                        public_holidays(country.to_s, location.to_s, d.year.to_s, "", "", add_stats: false, publish_results: false)
+                      end
+                      if @public_holidays.key?(country_region) and @public_holidays[country_region].key?(d.year.to_s)
+                        phd = @public_holidays[country_region][d.year.to_s].date.iso
+                      else
+                        phd = []
+                      end
+                      if d.wday == 1 or d == from
+                        v += "#{d.strftime("%d")} "
+                      end
+                      date_text = d.strftime("%Y-%m-%d")
+                      if phd.include?(date_text)
+                        v += ":large_red_square: "
+                      elsif d.wday == 0 or d.wday == 6
+                        v += ":large_yellow_square: "
+                      else
+                        v += ":white_square: "
+                      end
+                    end
+                  end
+
+                  blocks << {
+                    type: "context",
+                    elements: [
+                      {
+                        type: "image",
+                        image_url: info.user.profile.image_24,
+                        alt_text: info.user.name,
+                      },
+                      {
+                            type: "plain_text",
+                            text: v,
+                          },
+                    ],
+                  }
+                end
+              end
+              first = true
+              blocks.each_slice(10).each do |b|
+                if first
+                  b.unshift(blocks_header)
+                  first = false
+                end
+                respond blocks: b
+              end
+              message = ""
+              if !defaulted_country_region.empty?
+                message = "Defaulted public holidays calendar: #{defaulted_country_region}\n"
+              end
+              message += "To change your public holidays calendar, use the command `set public holidays to COUNTRY/STATE`. "
+              message += "\nExamples: `set public holidays to Iceland`, `set public holidays to Spain/Madrid`"
+              respond message
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/slack/smart-bot/commands/general/teams/update_team.rb

@@ -0,0 +1,137 @@
+class SlackSmartBot
+  module Commands
+    module General
+      module Teams
+        def update_team(user, team_name, new_name: "", new_info: "", delete_opts: "", add_opts: "")
+          save_stats(__method__)
+          get_teams()
+          if @teams.key?(team_name.to_sym)
+            assigned_members = @teams[team_name.to_sym].members.values.flatten
+            assigned_members.uniq!
+            all_team_members = assigned_members.dup
+            team_members = []
+            if @teams[team_name.to_sym].channels.key?("members")
+              @teams[team_name.to_sym].channels["members"].each do |ch|
+                get_channels_name_and_id() unless @channels_id.key?(ch)
+                tm = get_channel_members(@channels_id[ch])
+                tm.each do |m|
+                  user_info = @users.select { |u| u.id == m or (u.key?(:enterprise_user) and u.enterprise_user.id == m) }[-1]
+                  team_members << user_info.name unless user_info.is_app_user or user_info.is_bot
+                end
+              end
+            end
+            team_members.flatten!
+            team_members.uniq!
+            all_team_members += team_members
+            all_team_members.uniq!
+          end
+          if !@teams.key?(team_name.to_sym)
+            respond "It seems like the team *#{team_name}* doesn't exist.\nRelated commands `add team TEAM_NAME PROPERTIES`, `see team TEAM_NAME`, `see teams`"
+          elsif !(all_team_members + [@teams[team_name.to_sym].creator] + config.masters).flatten.include?(user.name)
+            respond "You have to be a member of the team, the creator or a Master admin to be able to update this team."
+          else
+            wrong = false
+            if new_name != ""
+              team = @teams[team_name.to_sym].deep_copy
+              @teams[new_name.to_sym] = team
+              @teams.delete(team_name.to_sym)
+              File.delete(File.join(config.path, "teams", "t_#{team_name}.yaml"))
+              message = "The *#{team_name}* team has been renamed #{new_name}."
+              team_name = new_name
+            elsif new_info != ""
+              @teams[team_name.to_sym].info = new_info
+            elsif delete_opts != ""
+              last_type = nil
+              delete_opts.split(" ").each do |opt|
+                if opt.match?(/^\s*$/)
+                  #blank
+                elsif opt.match?(/^[\w\-]+$/i)
+                  last_type = opt
+                elsif opt.match(/<?@(\w+)>?/i) #accepts also @username for the case the user has been deactivated
+                  member_id = $1
+                  member_info = @users.select { |u| u.id == member_id or u.name == member_id or (u.key?(:enterprise_user) and u.enterprise_user.id == member_id) }[-1]
+                  if last_type.nil?
+                    @teams[team_name.to_sym].members.each do |type, members|
+                      @teams[team_name.to_sym].members[type].delete(member_info.name)
+                    end
+                  else
+                    @teams[team_name.to_sym].members[last_type] ||= []
+                    @teams[team_name.to_sym].members[last_type].delete(member_info.name)
+                  end
+                elsif opt.match(/<#(\w+)\|[^>]*>/i)
+                  channel_id = $1
+                  get_channels_name_and_id() unless @channels_name.keys.include?(channel_id)
+                  channel = @channels_name[channel_id]
+                  if last_type.nil?
+                    @teams[team_name.to_sym].channels.each do |type, channels|
+                      @teams[team_name.to_sym].channels[type].delete(channel)
+                    end
+                  else
+                    @teams[team_name.to_sym].channels[last_type] ||= []
+                    @teams[team_name.to_sym].channels[last_type].delete(channel)
+                  end
+                else
+                  respond "It seems like the members or channel list is not correct. Please double check."
+                  wrong = true
+                  break
+                end
+              end
+              tmembers = @teams[team_name.to_sym].members.deep_copy
+              tmembers.each do |type, members|
+                @teams[team_name.to_sym].members.delete(type) if members.empty?
+              end
+              tchannels = @teams[team_name.to_sym].channels.deep_copy
+              tchannels.each do |type, channels|
+                @teams[team_name.to_sym].channels.delete(type) if channels.empty?
+              end
+            elsif add_opts != ""
+              last_type = nil
+              add_opts.split(" ").each do |opt|
+                if opt.match?(/^\s*$/)
+                  #blank
+                elsif opt.match?(/^[\w\-]+$/i)
+                  last_type = opt
+                elsif opt.match(/<@(\w+)>/i)
+                  member_id = $1
+                  last_type = "no_type" if last_type.nil?
+                  member_info = @users.select { |u| u.id == member_id or (u.key?(:enterprise_user) and u.enterprise_user.id == member_id) }[-1]
+                  @teams[team_name.to_sym].members[last_type] ||= []
+                  @teams[team_name.to_sym].members[last_type] << member_info.name
+                  @teams[team_name.to_sym].members[last_type].uniq!
+                elsif opt.match(/<#(\w+)\|[^>]*>/i)
+                  channel_id = $1
+                  get_channels_name_and_id() unless @channels_name.keys.include?(channel_id)
+                  channel = @channels_name[channel_id]
+                  @teams[team_name.to_sym].channels[last_type] ||= []
+                  @teams[team_name.to_sym].channels[last_type] << channel
+                  @teams[team_name.to_sym].channels[last_type].uniq!
+                else
+                  respond "It seems like the members or channel list is not correct. Please double check."
+                  wrong = true
+                  break
+                end
+              end
+              tmembers = @teams[team_name.to_sym].members.deep_copy
+              tmembers.each do |type, members|
+                @teams[team_name.to_sym].members.delete(type) if members.empty?
+              end
+              tchannels = @teams[team_name.to_sym].channels.deep_copy
+              tchannels.each do |type, channels|
+                @teams[team_name.to_sym].channels.delete(type) if channels.empty?
+              end
+            end
+            unless wrong
+              message ||= "The *#{team_name}* team has been updated."
+              @teams[team_name.to_sym].status = :updated
+              @teams[team_name.to_sym].user = user.name
+              @teams[team_name.to_sym].date = Time.now.strftime("%Y-%m-%dT%H:%M:%S.000Z")[0..18]
+              update_teams()
+            end
+            respond message
+            see_teams(user, team_name, add_stats: false) unless wrong
+          end
+        end
+      end
+    end
+  end
+end