blob: 449cb73fa5d4ed5aa61ba2b285f82d4b10abdae7 (
plain)
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
|
require 'jekyll/post'
module Jekyll
module RelatedPostsByCategory
# 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)
return [] unless posts.size > 1
highest_freq = tag_freq(posts).values.max
related_scores = Hash.new(0)
posts.each do |post|
post.categories.each do |tag|
if self.categories.include?(tag) && post != self
cat_freq = tag_freq(posts)[tag]
related_scores[post] += (1+highest_freq-cat_freq)
end
end
end
sort_related_posts(related_scores)
end
# 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.categories.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
class Post
include RelatedPostsByCategory
end
end
|
