slack-smart-bot 1.12.9 → 1.13.0

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

@@ -0,0 +1,202 @@
+class SlackSmartBot
+  def see_memos_team(user, type: "all", name: nil, topic: "", add_stats: true, team: nil)
+    save_stats(__method__) if add_stats
+
+    get_teams()
+    type = "all" if type.match?(/all\s+memo/i)
+    message = []
+    if @teams.size > 0
+      if team.nil?
+        @teams.each do |team_name, teamv|
+          if (team_name.to_s == name.to_s) or (name.to_s.gsub("-", "").gsub("_", "") == team_name.to_s)
+            if teamv.key?(:memos) and teamv[:memos].size > 0
+              team = teamv.deep_copy
+            else
+              respond "There are no memos for the team #{name}." unless !add_stats
+            end
+            break
+          end
+        end
+      end
+      if team
+        all_memos = {}
+        assigned_members, unassigned_members, not_on_team_channel, channels_members, all_team_members = get_team_members(team)
+        users_link = (Thread.current[:dest][0] == "D")
+        memos_filtered = []
+        all_topics = []
+        team[:memos].each do |memo|
+          if (type == "all" or type.to_s == memo[:type].to_s or type == "") and (topic == "" or memo[:topic].to_s.downcase == topic.to_s.downcase)
+            memos_filtered << memo
+            all_topics << memo.topic
+          end
+        end
+        all_topics.uniq!
+        all_topics.delete(:no_topic)
+        if memos_filtered.size >= 10 and !add_stats
+          message << "   > *_memos_*"
+          message << "        There are too many memos to show. "
+          message << "        Please use the `see MEMO_TYPE team #{team.name} TOPIC` command."
+          message << "        Available topics: #{all_topics.join(", ")}" if all_topics.size > 0
+          message << "        Examples: `see bugs #{team.name} team`, `see all memos #{team.name} team #{all_topics.sample}`, `see tasks #{team.name} team #{all_topics.sample}`"
+        elsif memos_filtered.size > 0
+          memos_filtered.each do |memo|
+            if memo.privacy.empty? or
+               (memo.privacy == "private" and (all_team_members.include?(user.name) and (users_link or channels_members.include?(Thread.current[:dest])))) or
+               (memo.privacy == "personal" and memo.user == user.name and users_link)
+              if memo.type == "jira" and config.jira.host != ""
+                http = NiceHttp.new(config.jira.host)
+                http.headers.authorization = NiceHttpUtils.basic_authentication(user: config.jira.user, password: config.jira.password)
+                if memo.message.match?(/^\w+\-\d+$/)
+                  resp = http.get("/rest/api/latest/issue/#{memo.message}")
+                  issues = [resp.data.json] if resp.code == 200
+                else
+                  resp = http.get("/rest/api/latest/search/?jql=#{memo.message}")
+                  issues = resp.data.json().issues if resp.code == 200
+                end
+                if resp.code == 200
+                  unless issues.empty?
+                    issues.each do |issue|
+                      jira_memo = { jira: true, github: false, status: "", memo_id: memo.memo_id, topic: memo.topic, privacy: memo.privacy, user: memo.user, date: memo.date, message: "", type: memo.type }
+                      jira_memo.message = issue.fields.summary
+                      jira_memo.user = issue.fields.reporter.name
+                      jira_memo.date = issue.fields.created
+                      if memo.topic == :no_topic and !issue.fields.labels.empty?
+                        jira_memo.topic = issue.fields.labels.sort.join("_").split(" ").join("_")
+                      end
+                      case issue.fields.issuetype.name
+                      when "Story"; jira_memo.type = ":abc:"
+                      when "Bug"; jira_memo.type = ":bug:"
+                      when "Task"; jira_memo.type = ":clock1:"
+                      when "New Feature", "Improvement"; jira_memo.type = ":sunny:"
+                      else jira_memo.type = ":memo:"
+                      end
+                      case issue.fields.status.statusCategory.name
+                      when "Done"; jira_memo.status = ":heavy_check_mark:"
+                      when "To Do"; jira_memo.status = ":new:"
+                      when "In Progress"; jira_memo.status = ":runner:"
+                      else jira_memo.status = ":heavy_minus_sign:"
+                      end
+                      #todo: check if possible to add link to status instead of jira issue
+                      #jira_memo.status = " <#{config.jira.host}/browse/#{issue[:key]}|#{jira_memo.status}> "
+                      jira_memo.status += " <#{config.jira.host}/browse/#{issue[:key]}|#{issue[:key]}>"
+
+                      all_memos[jira_memo.topic] ||= []
+                      all_memos[jira_memo.topic] << jira_memo
+                    end
+                  end
+                end
+                http.close
+              elsif memo.type == "github" and config.github.host != ""
+                http = NiceHttp.new(config.github.host)
+                http.headers.authorization = "token #{config.github.token}"
+                memo.message += "?" unless memo.message.include?("?")
+                memo.message += "&per_page=50"
+
+                resp = http.get("/repos/#{memo.message}")
+                issues = resp.data.json()
+                issues = [issues] unless issues.is_a?(Array)
+                if resp.code == 200
+                  unless issues.empty?
+                    issues.each do |issue|
+                      github_memo = { jira: false, github: true, status: "", memo_id: memo.memo_id, topic: memo.topic, privacy: memo.privacy, user: memo.user, date: memo.date, message: "", type: memo.type }
+                      github_memo.message = issue.title
+                      github_memo.user = issue.user.login
+                      github_memo.date = issue.created_at
+                      if issue.labels.empty?
+                        labels = ""
+                      else
+                        labels = issue.labels.name.sort.join("_").split(" ").join("_")
+                      end
+                      if memo.topic == :no_topic and !issue.labels.empty?
+                        github_memo.topic = labels
+                      end
+                      case labels
+                      when /bug/i; github_memo.type = ":bug:"
+                      when /docum/i; github_memo.type = ":abc:"
+                      when /task/i; github_memo.type = ":clock1:"
+                      when /enhancem/i, /improvement/i; github_memo.type = ":sunny:"
+                      else github_memo.type = ":memo:"
+                      end
+                      if issue.key?(:events_url)
+                        resp_events = http.get(issue.events_url)
+                        events = resp_events.data.json(:event)
+                        issue.state = "in progress" if events.include?("referenced")
+                      end
+                      case issue.state
+                      when "closed"; github_memo.status = ":heavy_check_mark:"
+                      when "open"; github_memo.status = ":new:"
+                      when "in progress"; github_memo.status = ":runner:"
+                      else github_memo.status = ":heavy_minus_sign:"
+                      end
+                      #todo: check if possible to add link to status instead of github issue
+                      github_memo.status += " <#{issue.html_url}|##{issue.number}>"
+
+                      all_memos[github_memo.topic] ||= []
+                      all_memos[github_memo.topic] << github_memo
+                    end
+                  end
+                end
+                http.close
+              else
+                memo.jira = false
+                memo.github = false
+                all_memos[memo.topic] ||= []
+                case memo.type
+                when "memo"; memo.type = ":memo:"
+                when "note"; memo.type = ":abc:"
+                when "bug"; memo.type = ":bug:"
+                when "task"; memo.type = ":clock1:"
+                when "feature"; memo.type = ":sunny:"
+                when "issue"; memo.type = ":hammer:"
+                else memo.type = ":heavy_minus_sign:"
+                end
+                all_memos[memo.topic] << memo
+              end
+            end
+          end
+          if !add_stats
+            message << "   > *_memos_*" unless all_memos.empty?
+          else
+            message << "  > *_#{team.name} team #{type}_*"
+          end
+
+          if all_memos.key?(:no_topic)
+            all_memos[:no_topic].sort_by { |memo| memo[:date] }.each do |memo|
+              case memo.privacy
+              when "private"; priv = " `private`"
+              when "personal"; priv = " `personal`"
+              else priv = ""
+              end
+              message << "        #{memo.type} #{memo.date.gsub("-", "/")[0..9]}:  #{memo.status} #{memo.message} (#{memo.user} #{memo.memo_id})#{priv}"
+            end
+          end
+          all_memos[:no_topic] = []
+          all_memos.each do |topic, mems|
+            unless mems.empty?
+              message << "        _`#{topic}`_:"
+              mems.sort_by { |m| m[:date] }.each do |memo|
+                case memo.privacy
+                when "private"; priv = " `private`"
+                when "personal"; priv = " `personal`"
+                else priv = ""
+                end
+                message << "            #{memo.type} #{memo.date.gsub("-", "/")[0..9]}:  #{memo.status} #{memo.message} (#{memo.user} #{memo.memo_id})#{priv}"
+              end
+            end
+          end
+        else
+          message << "There are no memos #{type} #{topic}." unless !add_stats
+        end
+      else
+        respond "There is no team named #{name}." unless !add_stats
+      end
+      if add_stats
+        respond message.join("\n")
+      else
+        return message
+      end
+    else
+      respond "There are no teams added yet\. Use `add team` command to add a team" unless !add_stats
+    end
+  end
+end

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

@@ -62,39 +62,8 @@ class SlackSmartBot
 
           message << "   > *_members_*"
 
-          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!
-            team_members.uniq!
-            unassigned_members = team_members - assigned_members
-            unassigned_members.delete(config.nick)
-            not_on_team_channel = assigned_members - team_members
-            all_team_members += team_members
-          else
-            unassigned_members = []
-            not_on_team_channel = []
-          end
+          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|
@@ -217,148 +186,7 @@ class SlackSmartBot
             end
 
             unless !team.key?(:memos) or team.memos.empty? or (team_name.to_s == "" and search.to_s == "")
-              all_memos = {}
-              team.memos.each do |memo|
-                if memo.privacy.empty? or
-                   (memo.privacy == "private" and (all_team_members.include?(user.name) and (users_link or channels_members.include?(Thread.current[:dest])))) or
-                   (memo.privacy == "personal" and memo.user == user.name and users_link)
-                  if memo.type == "jira" and config.jira.host != ""
-                    http = NiceHttp.new(config.jira.host)
-                    http.headers.authorization = NiceHttpUtils.basic_authentication(user: config.jira.user, password: config.jira.password)
-                    if memo.message.match?(/^\w+\-\d+$/)
-                      resp = http.get("/rest/api/latest/issue/#{memo.message}")
-                      issues = [resp.data.json] if resp.code == 200
-                    else
-                      resp = http.get("/rest/api/latest/search/?jql=#{memo.message}")
-                      issues = resp.data.json().issues if resp.code == 200
-                    end
-                    if resp.code == 200
-                      unless issues.empty?
-                        issues.each do |issue|
-                          jira_memo = { jira: true, github: false, status: "", memo_id: memo.memo_id, topic: memo.topic, privacy: memo.privacy, user: memo.user, date: memo.date, message: "", type: memo.type }
-                          jira_memo.message = issue.fields.summary
-                          jira_memo.user = issue.fields.reporter.name
-                          jira_memo.date = issue.fields.created
-                          if memo.topic == :no_topic and !issue.fields.labels.empty?
-                            jira_memo.topic = issue.fields.labels.sort.join("_").split(" ").join("_")
-                          end
-                          case issue.fields.issuetype.name
-                          when "Story"; jira_memo.type = ":abc:"
-                          when "Bug"; jira_memo.type = ":bug:"
-                          when "Task"; jira_memo.type = ":clock1:"
-                          when "New Feature", "Improvement"; jira_memo.type = ":sunny:"
-                          else jira_memo.type = ":memo:"
-                          end
-                          case issue.fields.status.statusCategory.name
-                          when "Done"; jira_memo.status = ":heavy_check_mark:"
-                          when "To Do"; jira_memo.status = ":new:"
-                          when "In Progress"; jira_memo.status = ":runner:"
-                          else jira_memo.status = ":heavy_minus_sign:"
-                          end
-                          #todo: check if possible to add link to status instead of jira issue
-                          #jira_memo.status = " <#{config.jira.host}/browse/#{issue[:key]}|#{jira_memo.status}> "
-                          jira_memo.status += " <#{config.jira.host}/browse/#{issue[:key]}|#{issue[:key]}>"
-
-                          all_memos[jira_memo.topic] ||= []
-                          all_memos[jira_memo.topic] << jira_memo
-                        end
-                      end
-                    end
-                    http.close
-                  elsif memo.type == "github" and config.github.host != ""
-                    http = NiceHttp.new(config.github.host)
-                    http.headers.authorization = "token #{config.github.token}"
-                    memo.message+="?" unless memo.message.include?('?')
-                    memo.message+="&per_page=50"
-              
-                    resp = http.get("/repos/#{memo.message}")
-                    issues = resp.data.json()
-                    issues = [issues] unless issues.is_a?(Array)
-                    if resp.code == 200
-                      unless issues.empty?
-                        issues.each do |issue|
-                          github_memo = { jira: false, github: true, status: "", memo_id: memo.memo_id, topic: memo.topic, privacy: memo.privacy, user: memo.user, date: memo.date, message: "", type: memo.type }
-                          github_memo.message = issue.title
-                          github_memo.user = issue.user.login
-                          github_memo.date = issue.created_at
-                          if issue.labels.empty?
-                            labels = ''
-                          else
-                            labels = issue.labels.name.sort.join("_").split(" ").join("_") 
-                          end
-                          if memo.topic == :no_topic and !issue.labels.empty?
-                            github_memo.topic = labels
-                          end
-                          case labels
-                          when /bug/i; github_memo.type = ":bug:"
-                          when /docum/i; github_memo.type = ":abc:"
-                          when /task/i; github_memo.type = ":clock1:"
-                          when /enhancem/i, /improvement/i; github_memo.type = ":sunny:"
-                          else github_memo.type = ":memo:"
-                          end
-                          if issue.key?(:events_url)
-                            resp_events = http.get(issue.events_url)
-                            events = resp_events.data.json(:event)
-                            issue.state = "in progress" if events.include?('referenced')
-                          end
-                          case issue.state
-                          when "closed"; github_memo.status = ":heavy_check_mark:"
-                          when "open"; github_memo.status = ":new:"
-                          when "in progress"; github_memo.status = ":runner:"
-                          else github_memo.status = ":heavy_minus_sign:"
-                          end
-                          #todo: check if possible to add link to status instead of github issue
-                          github_memo.status += " <#{issue.html_url}|##{issue.number}>"
-
-                          all_memos[github_memo.topic] ||= []
-                          all_memos[github_memo.topic] << github_memo
-                        end
-                      end
-                    end
-                    http.close
-                  else
-                    memo.jira = false
-                    memo.github = false
-                    all_memos[memo.topic] ||= []
-                    case memo.type
-                    when "memo"; memo.type = ":memo:"
-                    when "note"; memo.type = ":abc:"
-                    when "bug"; memo.type = ":bug:"
-                    when "task"; memo.type = ":clock1:"
-                    when "feature"; memo.type = ":sunny:"
-                    when "issue"; memo.type = ":hammer:"
-                    else memo.type = ":heavy_minus_sign:"
-                    end
-                    all_memos[memo.topic] << memo
-                  end
-                end
-              end
-              message << "   > *_memos_*" unless all_memos.empty?
-
-              if all_memos.key?(:no_topic)
-                all_memos[:no_topic].sort_by { |memo| memo[:date] }.each do |memo|
-                  case memo.privacy
-                  when "private"; priv = " `private`"
-                  when "personal"; priv = " `personal`"
-                  else priv = ""
-                  end
-                  message << "        #{memo.type} #{memo.date.gsub("-", "/")[0..9]}:  #{memo.status} #{memo.message} (#{memo.user} #{memo.memo_id})#{priv}"
-                end
-              end
-              all_memos[:no_topic] = []
-              all_memos.each do |topic, mems|
-                unless mems.empty?
-                  message << "        _`#{topic}`_:"
-                  mems.sort_by { |m| m[:date] }.each do |memo|
-                    case memo.privacy
-                    when "private"; priv = " `private`"
-                    when "personal"; priv = " `personal`"
-                    else priv = ""
-                    end
-                    message << "            #{memo.type} #{memo.date.gsub("-", "/")[0..9]}:  #{memo.status} #{memo.message} (#{memo.user} #{memo.memo_id})#{priv}"
-                  end
-                end
-              end
+              message += see_memos_team(user, type: "all", add_stats: false, team: team)
             end
 
             unless team.info.empty?

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

@@ -1,10 +1,11 @@
 class SlackSmartBot
-  def see_vacations(user, from_user: '', add_stats: true)
+  def see_vacations(user, dest, from_user: '', add_stats: true, year: '')
     save_stats(__method__) if add_stats
 
     get_vacations()
     
     from_user_name = ''
+    year = Date.today.year if year.to_s == ''
 
     if from_user.empty?
       from_user_name = user.name
@@ -13,8 +14,8 @@ class SlackSmartBot
       user_info = @users.select{|u| u.id == from_user or (u.key?(:enterprise_user) and u.enterprise_user.id == from_user)}[-1]
       from_user_name = user_info.name
     end
-
-    if !@vacations.key?(from_user_name) or @vacations[from_user_name].periods.empty?
+    from_user = '' if from_user_name == user.name
+    if !@vacations.key?(from_user_name) or !@vacations[from_user_name].key?(:periods) or @vacations[from_user_name].periods.empty?
       if from_user.empty?
         respond "You didn't add any time off yet. Use `add vacation from YYYY/MM/DD to YYYY/MM/DD`"
       else
@@ -22,36 +23,55 @@ class SlackSmartBot
       end
     else
       messages = []
-      messages << "*Time off <@#{from_user}>*" if !from_user.empty?
+      messages << "*Time off <@#{from_user}> #{year}*" if !from_user.empty?
+      
+      display_calendar(from_user_name, year) if from_user_name == user.name and dest[0] == 'D'
+
       today = Date.today.strftime("%Y/%m/%d")
       current_added = false
       past_added = false
       @vacations[from_user_name].periods.sort_by { |v| v[:from]}.reverse.each do |vac|
-        if !current_added and vac.to >= today
+        if !current_added and vac.to >= today 
           messages << "*Current and future periods*" 
           current_added = true
         end
-        if !past_added and vac.to < today and from_user.empty?
-          messages << "\n*Past periods*" 
-          past_added = true
+        if !past_added and vac.to < today and from_user.empty? and vac.to[0..3] == year
+          if dest[0]=='D'
+            messages << "\n*Past periods #{year}*" 
+            past_added = true
+          else
+            messages << "To see past periods call me from a DM"
+            break
+          end
         end
         unless !from_user.empty? and vac.to < today
-          if !from_user.empty?
-            icon = ":beach_with_umbrella:"
-          elsif vac.type == 'vacation'
-            icon = ':palm_tree:'
-          elsif vac.type == 'sick'
-            icon = ':face_with_thermometer:'
-          elsif vac.type == 'sick child'
-            icon = ':baby:'
-          end
-          if vac.from == vac.to
-            messages << "     #{icon}    #{vac.from}   ##{vac.vacation_id}"
-          else
-            messages << "     #{icon}    #{vac.from} -> #{vac.to}   ##{vac.vacation_id}"
+          if vac.to[0..3] == year
+            if !from_user.empty?
+              icon = ":beach_with_umbrella:"
+            elsif vac.type == 'vacation'
+              icon = ':palm_tree:'
+            elsif vac.type == 'sick'
+              icon = ':face_with_thermometer:'
+            elsif vac.type == 'sick child'
+              icon = ':baby:'
+            end
+            if vac.from == vac.to
+              messages << "     #{icon}    #{vac.from}   ##{vac.vacation_id}"
+            else
+              messages << "     #{icon}    #{vac.from} -> #{vac.to}   ##{vac.vacation_id}"
+            end
           end
         end
       end
+      if !past_added and !current_added and dest[0]=='D' 
+        if from_user.empty?
+          messages << "No time off added yet for #{year}"
+        else
+          messages << "Not possible to see past periods for another user"
+        end
+      elsif !past_added and dest[0]=='D' and !from_user.empty? and from_user_name != user.name
+        messages << "Not possible to see past periods for another user"
+      end
       respond messages.join("\n")
     end
   end

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

@@ -0,0 +1,21 @@
+class SlackSmartBot
+  def set_public_holidays(country, state, user)
+    save_stats(__method__)
+
+    result = public_holidays(country, state, Date.today.year.to_s, '', '', add_stats: false, publish_results: false)
+    if result == true
+        if state == ""
+            country_region = country
+        else
+            country_region = "#{country}/#{state}"
+        end
+        respond "Public holidays for *#{country_region}* set."
+        get_vacations()
+        @vacations[user.name] ||= {}
+        @vacations[user.name][:public_holidays] = country_region
+        update_vacations()
+    else
+        respond "Be sure the country and state are correct."
+    end
+  end
+end

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

@@ -32,6 +32,7 @@ class SlackSmartBot
         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 != ""