require "middleman/xmlvalidator/version"
require "schema"

module Middleman
  module Xmlvalidator
    class << self
		def registered(app)
			app.after_build do |builder|
				puts "", "Validating with NokoGiri", ""

				Dir.glob("build/**/*BingSiteAuth.xml").each do |full_path|
					puts "  validating".blue + "  #{full_path}....." + (ValidateNokoGiri.valid(full_path, 'BingSiteAuth.xsd') == true ? "COMPLETE".green : "ERRORS FOUND".red)
					ValidateNokoGiri.validate(full_path, 'BingSiteAuth.xsd').each do |error|
						puts "     " + error.message
					end
				end

				Dir.glob("build/**/*crossdomain.xml").each do |full_path|
					puts "  validating".blue + "  #{full_path}....." + (ValidateNokoGiri.valid(full_path, 'crossdomain.xsd') == true ? "COMPLETE".green : "ERRORS FOUND".red)
					ValidateNokoGiri.validate(full_path, 'crossdomain.xsd').each do |error|
						puts "     " + error.message
					end
				end

				Dir.glob("build/**/*Sitemap.xml").each do |full_path|
					puts "  validating".blue + "  #{full_path}....." + (ValidateNokoGiri.valid(full_path, 'Sitemap3.xsd') == true ? "COMPLETE".green : "ERRORS FOUND".red)
					ValidateNokoGiri.validate(full_path, 'Sitemap3.xsd').each do |error|
						puts "     " + error.message
					end
				end

				Dir.glob("build/**/*.rss").each do |full_path|
					puts "  validating".blue + "  #{full_path}....." + (ValidateNokoGiri.valid(full_path, 'RSSSchema.xsd') == true ? "COMPLETE".green : "ERRORS FOUND".red)
					ValidateNokoGiri.validate(full_path, 'RSSSchema.xsd').each do |error|
						puts "     " + error.message
					end
				end

				Dir.glob("build/**/*.xml").each do |full_path|
					if not (full_path.match(/^.*BingSiteAuth.xml$/) || full_path.match(/^.*crossdomain.xml$/) || full_path.match(/^.*Sitemap.xml$/))
						puts "  validating".blue + "  #{full_path}....." + (ValidateNokoGiri.valid(full_path, 'XMLSchema.xsd') == true ? "COMPLETE".green : "ERRORS FOUND".red)
						ValidateNokoGiri.validate(full_path, 'XMLSchema.xsd').each do |error|
							puts "     " + error.message
						end
					end
				end

				puts "", "Validation with NokoGiri " + "Complete".green, ""
			end
		end
		alias :included :registered
	end

	def self.validate(document_path, schema_path)
		schema = Nokogiri::XML::Schema(File.read("lib/schema/" + schema_path))
		document = Nokogiri::XML(File.read(document_path))
		schema.validate(document)
	end

	def self.valid(document_path, schema_path)
		schema = Nokogiri::XML::Schema(File.read("lib/schema/" + schema_path))
		document = Nokogiri::XML(File.read(document_path))
		schema.valid?(document)
	end
  end
end

::Middleman::Extensions.register(:validate_xml, Middleman::Xmlvalidator)