1
0
Fork 1
mirror of https://github.com/wezm/wezm.net.git synced 2024-12-25 21:49:53 +00:00
wezm.net/script/importer/wordpress.rb
2010-06-09 07:31:47 +10:00

171 lines
4.8 KiB
Ruby

require 'nokogiri'
require 'nanoc3'
require 'uri'
module Importer
class Wordpress
def initialize(wordpress_export_path, nanoc_site_path, rewrite_map_path)
@export_file = File.open(wordpress_export_path)
@export = Nokogiri::XML(@export_file)
@site = Nanoc3::Site.new(nanoc_site_path)
@rewrite_map = []
@rewrite_map_path = rewrite_map_path
load_categories
load_tags
end
def load_categories
puts "Loading categories"
@categories = {}
@export.xpath('//rss/channel/wp:category').each do |category|
name = get(category, 'wp:cat_name')
parent = get(category, 'wp:category_parent')
parent = nil if parent.empty?
@categories[name] = {
:slug => get(category, 'wp:category_nicename'),
:name => name,
:parent => parent
}
end
end
def load_tags
puts "Loading tags"
@tags = {}
@export.xpath('//rss/channel/wp:tag').each do |tag|
slug = get(tag, 'wp:tag_slug')
@tags[slug] = {
:slug => slug,
:name => get(tag, 'wp:tag_name'),
}
end
end
def find_topmost_category(category)
return category if category[:parent].nil?
find_topmost_category(@categories[category[:parent]])
end
def run
# Loop over each post
@export.xpath('//rss/channel/item').each do |item|
item_type = item.xpath('wp:post_type').first.text
case item_type
when 'post'
process_post(item)
when 'page'
process_page(item)
when 'attachment'
process_attachment(item)
else
puts "Unknown post type: #{item_type}"
end
end
write_rewrite_map
end
protected
def get(node, xpath)
elem = node.at_xpath(xpath)
elem ? elem.content : nil
end
def process_post(post)
puts "Processing post: #{post.at_css('title').text}"
content = get(post, 'content:encoded')
tags = []
post.css('category[domain=tag]').each do |tag|
if tag['nicename']
tags << tag['nicename']
else
tags << tag.text.downcase
end
end
categories = []
post.css('category[domain=category]').each do |category|
categories << category.text
end
categories.uniq!
if categories.empty?
categories << 'Uncategorized'
end
begin
post_date = Date.strptime(get(post, 'wp:post_date_gmt'), "%Y-%m-%d %H:%M:%S")
rescue ArgumentError
post_date = Date.today
end
attributes = {
:tags => tags.uniq,
:categories => categories,
:permalink => get(post, 'link'),
:status => get(post, 'wp:status'),
:slug => get(post, 'wp:post_name'),
:post_id => get(post, 'wp:post_id').to_i,
:section => find_topmost_category(@categories[categories.first])[:slug],
:title => get(post, 'title'),
:created_at => get(post, 'wp:post_date_gmt'),
}
attributes[:kind] = attributes[:status] == 'publish' ? 'article' : 'draft'
if attributes[:slug].empty?
puts "WARNING: Error post #{attributes[:post_id]} has no slug, generating one"
attributes[:slug] = attributes[:title].downcase.gsub(/[^0-9a-zA-Z]/, '-').gsub(/-{2,}/, '-')
end
if attributes[:status] == 'draft'
path = ['', 'drafts', attributes[:slug], ''].join('/')
else
path = ['', attributes[:section], post_date.year, ("%02d" % post_date.month), attributes[:slug], ''].join('/')
end
add_item(content, attributes, path)
end
def process_page(page)
puts "WARNING: Skipping page: #{page.at_css('title').text}"
end
def process_attachment(attachment)
# puts "Processing attachment"
# Don't need these for now
end
def add_item(content, attributes, identifier)
# content = row['post_content']
# attributes = {
# :title => row['post_title'],
# :published_on => row['post_date_gmt'],
# :modified_on => row['post_modified_gmt'],
# :status => row['post_status'],
# :excerpt => row['post_excerpt']
# }
# identifier = '/posts/' + post_date.year.to_s + '/' + post_date.month.to_s + '/' + post_name + '/'
if attributes[:status] == 'publish'
@rewrite_map << [attributes[:permalink], identifier]
end
@site.data_sources.first.create_item(content, attributes, identifier)
puts "Added item at #{identifier}"
end
def write_rewrite_map
File.open(@rewrite_map_path, 'w') do |f|
@rewrite_map.each do |old_url, new_path|
uri = URI.parse(old_url)
f.puts uri.path + "\t" + new_path
puts uri.path + " => " + new_path
end
end
end
end
end