Permalink
Switch branches/tags
Nothing to show
Find file Copy path
Fetching contributors…
Cannot retrieve contributors at this time
66 lines (58 sloc) 1.49 KB
require 'jekyll/post'
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)
return [] unless posts.size > 1
highest_freq = Jekyll::Post.tag_freq(posts).values.max
related_scores = Hash.new(0)
posts.each do |post|
post.tags.each do |tag|
if self.tags.include?(tag) && post != self
cat_freq = Jekyll::Post.tag_freq(posts)[tag]
related_scores[post] += (1+highest_freq-cat_freq)
end
end
end
Jekyll::Post.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.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 Post
include BetterRelatedPosts
extend BetterRelatedPosts::ClassMethods
end
end