support for more sites

benbalter · Jan 31, 2015 · c54522c · c54522c
1 parent c02598f
commit c54522c
Show file tree

Hide file tree

Showing 6 changed files with 17 additions and 16 deletions.
diff --git a/dist/script.js b/dist/script.js
diff --git a/dist/script.js.map b/dist/script.js.map
diff --git a/dist/style.css b/dist/style.css
diff --git a/dist/style.css.map b/dist/style.css.map
diff --git a/src/script.coffee b/src/script.coffee
@@ -4,7 +4,7 @@ class GitHubMentionHighlighter
     mentions = []
     for mention in $(".user-mention, .member-mention")
       $mention = $(mention)
-      mentions.push $mention if $mention.text() == "@#{@username}"
+      mentions.push $mention if $mention.text() == "@#{@username()}"
     mentions
 
   teamMentions: ->
@@ -22,15 +22,16 @@ class GitHubMentionHighlighter
           cache: true
           success: (data) =>
             members = data["members"]
-      mentions.push $mention if $.inArray(@username, members) != -1
+      mentions.push $mention if $.inArray(@username(), members) != -1
     mentions
 
   mentions: ->
     $.merge @userMentions(), @teamMentions()
 
-  constructor: ->
-    @username = $(".supportocat a, #user-links .name").text().trim()
+  username: ->
+    @_username ||= $(".supportocat a, #user-links .name, .header-right .logged-in a").text().trim().replace("@", "")
 
+  constructor: ->
     for $mention in @mentions()
       $mention.addClass("highlight")
       $mention.parents(".timeline-comment, .timeline-entry").addClass("highlight")

diff --git a/src/style.scss b/src/style.scss
@@ -1,7 +1,9 @@
 .timeline-comment:not(.current-user) .user-mention.highlight,
 .timeline-comment:not(.current-user) .team-mention.highlight,
 .timeline-entry .member-mention.highlight,
-.timeline-entry .team-mention.highlight  {
+.timeline-entry .team-mention.highlight,
+.markdown-body .user-mention.highlight,
+.markdown-body .user-mention.highlight {
   background: yellow !important;
 }