lfa-wordpress-import/lib/wordpress/post.rb

module Refinery
  module WordPress
    class Post < Page
      def tags
        # xml dump has "post_tag" for wordpress 3.1 and "tag" for 3.0
        path = if node.xpath("category[@domain='post_tag']").count > 0
          "category[@domain='post_tag']"
        else
          "category[@domain='tag']"
        end

        node.xpath(path).collect do |tag_node| 
          Tag.new(tag_node.text)
        end
      end

      def tag_list
        tags.collect(&:name).join(',')
      end

      def categories
        node.xpath("category[@domain='category']").collect do |cat|
          Category.new(cat.text)
        end
      end

      def comments
        node.xpath("wp:comment").collect do |comment_node|
          Comment.new(comment_node)
        end
      end

      def to_refinery
        user = ::User.find_by_username(creator) || ::User.first
        raise "Referenced User doesn't exist! Make sure the authors are imported first." \
          unless user
        
        begin
          unless draft?
            #p "creating post " + title + " Draft status: " + draft?.to_s
            
            post = ::BlogPost.new :title => title, :body => content_formatted, 
              :draft => draft?, :published_at => post_date, :created_at => post_date, 
              :author => user, :tag_list => tag_list
            post.save!

            ::BlogPost.transaction do
              categories.each do |category|
                post.categories << category.to_refinery
              end

              comments.each do |comment|
                comment = comment.to_refinery
                comment.post = post
                comment.save
              end
            end
          end
        rescue ActiveRecord::RecordInvalid 
          # if the title has already been taken (WP allows duplicates here,
          # refinery doesn't) append the post_id to it, making it unique
          post.title = "#{title}-#{post_id}"
          post.save
        end

        post
      end
    end
  end
end