-
Notifications
You must be signed in to change notification settings - Fork 0
/
related_posts.rb
67 lines (59 loc) · 1.56 KB
/
related_posts.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
require 'jekyll/document'
module BetterRelatedPosts
# Used to remove #related_posts so that it can be overridden
def self.included(klass)
klass.class_eval do
remove_method :related_posts
end
end
# Calculate related posts.
#
# Returns [<Post>]
def related_posts
posts = site.posts.docs
return [] unless posts.size > 1
highest_freq = Jekyll::Document.tag_freq(posts).values.max
related_scores = Hash.new(0)
posts.each do |post|
post.data["tags"].each do |tag|
if self.data["tags"].include?(tag) && post != self
cat_freq = Jekyll::Document.tag_freq(posts)[tag]
related_scores[post] += (1+highest_freq-cat_freq)
end
end
end
Jekyll::Document.sort_related_posts(related_scores)
end
module ClassMethods
# Calculate the frequency of each tag.
#
# Returns {tag => freq, tag => freq, ...}
def tag_freq(posts)
return @tag_freq if @tag_freq
@tag_freq = Hash.new(0)
posts.each do |post|
post.data["tags"].each {|tag| @tag_freq[tag] += 1}
end
@tag_freq
end
# Sort the related posts in order of their score and date
# and return just the posts
def sort_related_posts(related_scores)
related_scores.sort do |a,b|
if a[1] < b[1]
1
elsif a[1] > b[1]
-1
else
b[0].date <=> a[0].date
end
end.collect {|post,freq| post}
end
end
end
module Jekyll
class Document
include BetterRelatedPosts
extend BetterRelatedPosts::ClassMethods
end
end