[Jekyll] WordPress migration

migrate.rb

#!/usr/bin/env ruby

# Input: WordPress XML export file.
# Outputs: a series of Textile files ready to be included in a Jekyll site,
# and comments.yml which contains all approved comments with metadata which
# can be used for a Disqus import.
# Credits: @ecerulm https://gist.github.com/500506
# Changes from the original gist: http://gist.github.com/268428
# 1. Handles titles containing special characters. Those have to be YAML escaped 
# 2. Use the original permalinks in wordpress.

require 'rubygems'
require 'hpricot'
require 'clothred'
require 'time'
require 'yaml'
require 'fileutils'
require 'uri'

WORDPRESS_XML_FILE_PATH = "#{ENV['PWD']}/wordpress.xml"
OUTPUT_PATH = "#{ENV['PWD']}/_posts"
ORIGINAL_DOMAIN = "http://example.com"

class Post
  attr_accessor :title, :post_date, :created_at, :slug, :post_id, :content, :textile_content
  attr_accessor :hpricot_element, :permalink, :categories

  def initialize(item)
    @hpricot_element = item

    @title = item.search("title").first.inner_text
    @permalink = item.at("link2").inner_text.gsub(/^#{ORIGINAL_DOMAIN}/,'').gsub(/\/$/,"/index.html")
    @permalink = URI.unescape(@permalink)
    puts "permalink #{@permalink}"
    
    @post_date = item.search("wp:post_date").first.inner_text
    @created_at = Date.parse(post_date)

    @slug = item.search("wp:post_name").first.inner_text
    
    @categories = []
    item.search("category").each { |cat| 
      @categories << cat.inner_text
    }
    @categories.uniq!
    
    p @categories

    @post_id = item.at("wp:post_id").inner_text

    @content = item.search("content:encoded").first.inner_text
    text = ClothRed.new(content)
    @textile_content = text.to_textile
  end

  def to_jekyll
    buf = ""
    buf << "---\n"
    buf << "layout: post\n"
    buf << "title: #{title.to_yaml.gsub(/^--- /,'')}" #escape character in title
    buf << "permalink: #{permalink.to_yaml.gsub(/^--- /,'').chomp}\n"
    buf << "post_id: #{post_id}\n"
    buf << "categories: #{categories.to_yaml.gsub(/^--- /,'').chomp}\n"
    buf << "---\n\n"
    buf << textile_content
  end

  def save(root_path)
    File.open("#{root_path}/#{created_at}-#{slug}.textile", "w") { |file| file.write self.to_jekyll }
    self
  end

  def save_comments(path)
    comment_elements = @hpricot_element.search("wp:comment").reject do |c|
      c.search("wp:comment_approved").inner_text != "1"
    end

    File.open("#{path}/comments.yml", "a") do |yaml_file|
      comment_elements.collect { |el| Comment.new(self, el) }.each { |comment| comment.write_to yaml_file }
    end
  end

  class << self
    def parse(element, path)
      return nil unless element.is_a?(Hpricot::Elem)
      post = Post.new(element)
      puts "saving post: #{post.title}"
      post.save(path)
    end
  end
end

class Comment
  attr_accessor :author_name, :author_email, :author_url, :content, :post

  def initialize(post, element)
    @post_id = post.post_id
    @post_title = post.title
    @author_name = element.search("wp:comment_author").first.inner_text
    @author_email = element.search("wp:comment_author_email").first.inner_text
    @author_url = element.search("wp:comment_author_url").first.inner_text
    @content = element.search("wp:comment_content").first.inner_text || ""

    comment_date = element.search("wp:comment_date_gmt").first.inner_text
    @created_at = Time.parse("#{comment_date} GMT")
  end

  def write_to(file)
    file.write self.to_yaml + "\n" unless @content.size == 0
  end
end

# main

file = File.open(WORDPRESS_XML_FILE_PATH,"rb");
contents = file.read
# Hpricot will destroy <link> contents as it expecting a <link href="">
# so we change link to link2 and Hpricot will leave it alone. 
contents.gsub!(/link>/, "link2>") 



doc = Hpricot(contents)

FileUtils.mkdir_p OUTPUT_PATH
File.open("#{OUTPUT_PATH}/comments.yml", "w") { |f| }

(doc / "item").each do |item|
  next unless item.search("wp:status").first.inner_text == "publish"
  post = Post.parse(item, OUTPUT_PATH)
  post.save_comments(OUTPUT_PATH)
end