BigW Consortium Gitlab

note_message.rb 2.25 KB
Newer Older
1 2 3
class SlackService
  class NoteMessage < BaseMessage
    attr_reader :message
4
    attr_reader :user_name
5 6 7 8 9 10 11 12
    attr_reader :project_name
    attr_reader :project_link
    attr_reader :note
    attr_reader :note_url
    attr_reader :title

    def initialize(params)
      params = HashWithIndifferentAccess.new(params)
13
      @user_name = params[:user][:name]
14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47
      @project_name = params[:project_name]
      @project_url = params[:project_url]

      obj_attr = params[:object_attributes]
      obj_attr = HashWithIndifferentAccess.new(obj_attr)
      @note = obj_attr[:note]
      @note_url = obj_attr[:url]
      noteable_type = obj_attr[:noteable_type]

      case noteable_type
      when "Commit"
        create_commit_note(HashWithIndifferentAccess.new(params[:commit]))
      when "Issue"
        create_issue_note(HashWithIndifferentAccess.new(params[:issue]))
      when "MergeRequest"
        create_merge_note(HashWithIndifferentAccess.new(params[:merge_request]))
      when "Snippet"
        create_snippet_note(HashWithIndifferentAccess.new(params[:snippet]))
      end
    end

    def attachments
      description_message
    end

    private

    def format_title(title)
      title.lines.first.chomp
    end

    def create_commit_note(commit)
      commit_sha = commit[:id]
      commit_sha = Commit.truncate_sha(commit_sha)
48 49 50
      commented_on_message(
        "[commit #{commit_sha}](#{@note_url})",
        format_title(commit[:message]))
51 52 53
    end

    def create_issue_note(issue)
54 55 56
      commented_on_message(
        "[issue ##{issue[:iid]}](#{@note_url})",
        format_title(issue[:title]))
57 58 59
    end

    def create_merge_note(merge_request)
60 61 62
      commented_on_message(
        "[merge request ##{merge_request[:iid]}](#{@note_url})",
        format_title(merge_request[:title]))
63 64 65
    end

    def create_snippet_note(snippet)
66 67 68
      commented_on_message(
        "[snippet ##{snippet[:id]}](#{@note_url})",
        format_title(snippet[:title]))
69 70 71 72 73 74 75 76 77
    end

    def description_message
      [{ text: format(@note), color: attachment_color }]
    end

    def project_link
      "[#{@project_name}](#{@project_url})"
    end
78 79 80 81

    def commented_on_message(target_link, title)
      @message = "#{@user_name} commented on #{target_link} in #{project_link}: *#{title}*"
    end
82 83
  end
end