inserting order on results

Author: raphox

README.md

@@ -10,7 +10,7 @@
 Run the below code:
 
 ```
-docker-compose run web
+docker-compose up web
 ```
 
 ## Setup

app/models/user_neo4j.rb

@@ -54,8 +54,10 @@ def friends_by_search(term)
       where('ANY(title IN l.titles WHERE toLower(title) CONTAINS toLower({title})) OR
         ANY(subtitle IN l.subtitles WHERE toLower(subtitle) CONTAINS toLower({title}))').
       where('l <> user').
-      params({ title: term })
+      params({ title: term }).
+      to_a
 
+    # TODO: Use only one query to get users and shortest paths
     friends.each do |friend|
       friends_path[friend.uuid] = Neo4j::ActiveBase.current_session.
         query("MATCH (r:UserNeo4j { uuid: '#{uuid}' }),
@@ -64,6 +66,17 @@ def friends_by_search(term)
           RETURN p", limit: 1).first.p.nodes
     end
 
+    # TODO: Use only queries to generate order
+    friends.sort_by! do |item|
+      total = 0
+
+      total += (friends_path[item.uuid].length - 2) * 10
+      total += item.titles.any? { titles.include?(term) } ? -10 : 0
+      total += item.subtitles.any? { |subtitle| subtitle.include?(term) } ? -5 : 0
+
+      total
+    end
+
     [friends, friends_path]
   end
 

lib/website_scrapper.rb

@@ -13,7 +13,7 @@ def self.titles(url)
 
     {
       titles: nokogiri.search('h1').map { |item| item.text.squish },
-      subtitles: nokogiri.search('h2').map { |item| item.text.squish },
+      subtitles: nokogiri.search('h2, h3').map { |item| item.text.squish },
     }
   end
 end