Make page search play nice with focus

As for the sort the page search works with threads instead of annotations so the directive should look for either annotation or thread.message in the scope
Page search focuses annotations when it opens them.

Thx @csillag for the help

h/js/controllers.coffee

@@ -963,6 +963,8 @@ class Search
 
       $rootScope.search_annotations = threads
       $scope.threads = threads
+      for thread in threads
+        $rootScope.focus thread.message, true
 
     $scope.$on 'ReRenderPageSearch', refresh
     $scope.$on '$routeUpdate', refresh

h/js/directives.coffee

@@ -230,21 +230,25 @@ thread = ['$rootScope', '$window', ($rootScope, $window) ->
       scope.collapsed = false
 
     scope.$on "focusChange", ->
-      if scope.annotation in $rootScope.focused
+      # XXX: This not needed to be done when the viewer and search will be unified
+      ann = scope.annotation ? scope.thread.message
+      if ann in $rootScope.focused
         scope.collapsed = false
       else
-        unless scope.annotation.references?.length
+        unless ann.references?.length
           scope.collapsed = true
 
     scope.toggleCollapsed = (event) ->
       event.stopPropagation()
       return if (ignoreClick event) or Object.keys(childrenEditing).length
       scope.collapsed = !scope.collapsed
+      # XXX: This not needed to be done when the viewer and search will be unified
+      ann = scope.annotation ? scope.thread.message
       if scope.collapsed
-        $rootScope.unFocus scope.annotation, true
+        $rootScope.unFocus ann, true
       else
-        scope.openDetails scope.annotation
-        $rootScope.focus scope.annotation, true
+        scope.openDetails ann
+        $rootScope.focus ann, true
 
     scope.$on 'toggleEditing', (event) ->
       {$id, editing} = event.targetScope