Import XML of Wordpress to Octopress

wp-xml-octopress-import.rb

# -*- coding: utf-8 -*-
require 'fileutils'
require 'date'
require 'yaml'
require 'uri'
require 'rexml/document'
include REXML

doc = Document.new File.new(ARGV[0])

FileUtils.mkdir_p "_posts"

doc.elements.each("rss/channel/item[wp:status = 'publish' and wp:post_type = 'post']") do |e|
  post = e.elements
  slug = post['wp:post_name'].text
  #slug = post['title'].text
  date = DateTime.parse(post['wp:post_date'].text)
  name = "%02d-%02d-%02d-%s.markdown" % [date.year, date.month, date.day, slug]

  content = post['content:encoded'].text

  content = content.gsub(/<code>(.*?)<\/code>/, '`\1`')

  ## 追加
  content = content.gsub(/<pre lang="([^"]*)">(.*?)<\/pre>/m, '<div class="bogus-wrapper"><notextile><figure class="code"><figcaption><span>lang:\1 </span></figcaption><div class="highlight"><table><tr><td class="gutter"><pre class="line-numbers"><span class="line-number">1</span></pre></td><td class="code"><pre><code class=''><span class="line">\2</span></code></pre></td></tr></table></div></figure></notextile></div>')

  (1..3).each do |i|
  content = content.gsub(/<h#{i}>([^<]*)<\/h#{i}>/, ('#'*i) + ' \1')
  end

  puts "Converting: #{name}"

  # data = {
  #    'layout' => 'blog_post',
  #    'title' => post['title'].text,
  #    'excerpt' => post['excerpt:encoded'].text,
  #    'wordpress_id' => post['wp:post_id'].text,
  #    'wordpress_url' => post['guid'].text
  # }.delete_if { |k,v| v.nil? || v == ''}.to_yaml

  File.open("_posts/#{name}", "w") do |f|
  f.puts "---"
  #f.puts data
  f.puts "layout: post"
  f.puts "comment: true"
  f.puts "title: \"#{post['title'].text}\""
  f.puts "---"
  f.puts content
  end

end