require "jekyll"
require "jekyll-archives"

module Jekyll

  class MoreArchives < Jekyll::Generator
    safe true

    def generate(site)
      @site = site
      @posts = site.posts
      @archives = []

      read
      @site.pages.concat(@archives)
      
      # don't overwrite the category/tag etc archives already done by jekyll-archives
      @site.config["archives"] = @site.config["archives"] + @archives
    end

    def read
      years.each do |year, y_posts|
        # call jekyll-archives Archive class
        @archives << Jekyll::Archives::Archive.new(@site, { :year => year }, "year", y_posts)
      end
    end

    def years
      date_attr_hash(@posts.docs, "%Y")
    end

    private

    def date_attr_hash(posts, id)
      hash = Hash.new { |hsh, key| hsh[key] = [] }
      posts.each {
        |post|
      
        # do the normal date thing
        hash[post.date.strftime(id)] << post
      
        # now see if we have a second date to show the post in
        if defined?(post["original_date"]) and not post["original_date"].nil?
        
          date = post["original_date"]
          # strftime doesn't like the dates I've written with HH:MM added, so parse them here
          if not date.is_a?(Date)
            date = Date.parse(date)
          end
          
          hash[date.strftime(id)] << post
        end
      }
      
      hash.each_value {
        # add uniq! to make sure each year still only has one copy of the post
        |posts| posts.sort!.reverse!.uniq!
      }
      
      hash
    end
  end

end