require 'time'

module XmlSitemap
  PERIODS = [:none, :always, :hourly, :daily, :weekly, :monthly, :yearly, :never]

  class Item
    attr_reader :path
    attr_reader :updated
    attr_reader :priority
    attr_reader :changefreq
    
    def initialize(opts={})
      @path = opts[:url] if opts.key?(:url)
      @updated = opts[:updated] || Time.now
      @priority = opts[:priority] || 1.0
      @changefreq = opts[:period] || :weekly
    end
  end

  class Map
    attr_reader :domain, :items
    attr_reader :buffer
    attr_reader :created_at
    attr_accessor :index_path
    
    # Creates new Map class for specified domain
    def initialize(domain)
      @domain = domain
      @created_at = Time.now
      @items = []
      yield self if block_given?
    end
    
    # Yields Map class for easier access
    def generate
      raise ArgumentError, 'Block required' unless block_given?
      yield self
    end
    
    # Add new item to sitemap list
    def add(opts)
      @items << XmlSitemap::Item.new(opts)
    end
    
    # Get map items count
    def size
      @items.size
    end
    
    # Render XML
    def render
      return '' if @items.size == 0
      output = []
      output << '<?xml version="1.0" encoding="UTF-8"?>'
      output << '<urlset xmlns="http://www.sitemaps.org/schemas/sitemap/0.9" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://www.sitemaps.org/schemas/sitemap/0.9 http://www.sitemaps.org/schemas/sitemap/0.9/sitemap.xsd">'
      @items.each do |item|
        output << '<url>'
          output << "<loc>http://#{@domain}#{item.path.to_s}</loc>"
          output << "<lastmod>#{item.updated.utc.iso8601}</lastmod>"
          output << "<changefreq>#{item.changefreq.to_s}</changefreq>"
          output << "<priority>#{item.priority.to_s}</priority>"
        output << '</url>'
      end
      output << '</urlset>'
      return output.join("\n")
    end
  end
  
  class Index
    attr_reader :domain, :maps
    
    def initialize(domain)
      @domain = domain
      @maps = []
    end
    
    # Add XmlSitemap::Map item to sitemap index
    def add(map)
      raise 'XmlSitemap::Map object requred!' unless map.kind_of?(Map)
      @maps << {:loc => "http://#{@domain}/#{map.index_path}", :lastmod => map.created_at.utc.iso8601}
    end
    
    # Generate sitemap XML index
    def render
      output = [] ; map_id = 1
      output << '<?xml version="1.0" encoding="UTF-8"?>'
      output << '<sitemapindex xmlns="http://www.sitemaps.org/schemas/sitemap/0.9">'
      @maps.each do |m|
        output << '<sitemap>'
        output << "<loc>#{m[:loc]}</loc>"
        output << "<lastmod>#{m[:lastmod]}</lastmod>"
        output << '</sitemap>'
      end
      output << '</sitemapindex>'
      return output.join("\n")
    end
  end
end