support for more sites

Author: benbalter

dist/script.js

@@ -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")

dist/script.js.map

@@ -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;
 }