require 'thor/shell/color'

module Alchemy
	class Seeder

		class << self

			# This seed builds the necessary page structure for alchemy in your database.
			# Run the alchemy:db:seed rake task to seed your database.
			def seed!
				desc "Seeding your database"
				errors = []
				notices = []
				
				default_language = Alchemy::Config.get(:default_language)
				
				lang = Alchemy::Language.find_or_initialize_by_code(
					:name => default_language['name'],
					:code => default_language['code'],
					:frontpage_name => default_language['frontpage_name'],
					:page_layout => default_language['page_layout'],
					:public => true,
					:default => true
				)
				if lang.new_record?
					if lang.save
						log "Created language #{lang.name}."
					else
						errors << "Errors while creating language #{lang.name}: #{lang.errors.full_messages}"
					end
				else
					notices << "Language #{lang.name} was already present."
				end
				
				root = Alchemy::Page.find_or_initialize_by_name(
					:name => 'Root',
					:do_not_sweep => true
				)
				if root.new_record?
					if root.save
						log "Created page #{root.name}."
					else
						errors << "Errors while creating page #{root.name}: #{root.errors.full_messages}"
					end
				else
					notices << "Page #{root.name} was already present."
				end
				
				if errors.blank?
					log "Successfully seeded your database!" if notices.blank?
					notices.each do |note|
						log(note, :skip)
					end
				else
					log("Some pages could not be created:", :error)
					errors.each do |error|
						log(error, :error)
					end
				end
			end

		private

			def color(name)
				case name
				when :green
					Thor::Shell::Color::GREEN
				when :red
					Thor::Shell::Color::RED
				when :yellow
					Thor::Shell::Color::YELLOW
				when :black
					Thor::Shell::Color::BLACK
				when :clear
					Thor::Shell::Color::CLEAR
				else
					""
				end
			end

			def log(message, type=nil)
				case type
				when :skip
					puts "#{color(:yellow)}== Skipping! #{message}#{color(:clear)}"
				when :error
					puts "#{color(:red)}!! ERROR: #{message}#{color(:clear)}"
				else
					puts "#{color(:green)}== #{message}#{color(:clear)}"
				end
			end

			def desc(message)
				puts "\n#{message}"
				puts "#{'-' * message.length}\n"
			end

		end

	end
end