#!/usr/bin/exec ruby

require 'weaver'
require 'trollop'

class Website < Sinatra::Base

	set :static, true 							# set up static file routing
	set :public_folder, Gem.datadir("weaver")	# set up the static dir (with images/js/css inside)
	set :bind, '0.0.0.0'

	def getWeavefile(viewname)

		pathToFile = viewname.split("/").delete_if(&:empty?)

		viewFile = nil

		remainingPath = []

		loop do 
			path = "#{File.join(['source'] + pathToFile)}.weave"
			indexPath = File.join(["source"] + pathToFile + ["index.weave"])

			#puts path
			#puts indexPath

			viewFile = path if File.exists?(path)
			viewFile = indexPath if File.exists?(indexPath)

			break if viewFile or pathToFile.length == 0

			remainingPath << pathToFile.last

			pathToFile = pathToFile.first pathToFile.size-1
		end 

		{ viewFile: viewFile, path: remainingPath }
	end

	get '/*' do

		if params[:splat].first.start_with? "images/"
			return send_file(params[:splat].first, :disposition => 'inline')
		end

		if params[:splat].first.start_with? "js/"
			return send_file(params[:splat].first, :disposition => 'inline')
		end

		if params[:splat].first.start_with? "css/"
			return send_file(params[:splat].first, :disposition => 'inline')
		end


		viewname = params[:splat].first
		result = getWeavefile viewname
		viewFile = result[:viewFile]
		path = File.join(result[:path].reverse)

		if viewFile

			site = Weaver::Weave.new(viewFile)

			if site.pages.has_key?(path)

				level = viewname.split(/\//, 0).length
				site.pages[path].generate(level)

			else
				puts "Not found: #{viewname} in #{viewFile}"
				puts "Path is: #{path}"
				status 404
			end
		else
			puts "Not found: #{viewname} in #{viewFile}"
				puts "Path is: #{path}"
			status 404
		end
	end

end

def build!(args)

	opts = Trollop::options do
		banner <<-EOS
Generate website

Usage:
   weaver build [options]
where [options] are:
		EOS

		opt :root, "The root URL to be used in links (e.g.) https://astrobunny.net", type: :string, default: "/"
	end

	puts "Building site..."

	buildDir = "#{Dir.pwd}/build" 
	FileUtils::rm_rf buildDir
	FileUtils.cp_r(Gem.datadir("weaver"), buildDir)
	if Dir.exist? "images"
		FileUtils.cp_r("images", buildDir)
	end
	if Dir.exist? "js"
		Dir["js/*.*"].each do |x|
			FileUtils.cp_r(x, File.join(buildDir, "js"))
		end
	end
	if Dir.exist? "css"
		Dir["css/*.*"].each do |x|
			FileUtils.cp_r(x, File.join(buildDir, "css"))
		end
	end

	files = Dir["source/**/*.weave"]
	files.each do |file|
		baseDir = file.chomp('.weave').chomp('index').sub(/^source\//, "")

		site = Weaver::Weave.new(file, opts)

		site.pages.each do |localPath, page|
			resultPath = File.join(baseDir, localPath)

			level = resultPath.split(/\//, 0).select{|x| x.length != 0}.length

			puts "Writing: #{file} -> #{resultPath} (Level: #{level})"

			fileName = File.join([buildDir, resultPath, "index.html"])
			FileUtils::mkdir_p File.dirname(fileName)
			File.write(fileName, page.generate(level))
		end
	end

end

def preview!
	puts "Weaver Preview Mode in: #{Dir.pwd}"

	Website.run!
end

def create!(args)

	if args.length != 1
		puts "Usage: weaver create <folder>"
		return
	end

	gemfile = <<-GEMFILE
source 'https://rubygems.org'

gem 'weaver', '~> #{Weaver::VERSION}'
	GEMFILE

	source = <<-SOURCE
topnav_page "", "My first weaver page" do
	header do
		col 12 do
			h1 "Main page"

		end
	end

	row do
		col 12 do
			h2 "Examples"
			p "Some content"
		end
	end
end
	SOURCE

	to_create = "#{args[0]}"
	FileUtils::mkdir_p "#{to_create}"
	FileUtils::mkdir_p "#{to_create}/cache"
	File.write("#{to_create}/cache/this_directory_is_used_by_weaver_as_a_cache","")
	FileUtils::mkdir_p "#{to_create}/source"
	FileUtils::mkdir_p "#{to_create}/images"
	FileUtils::mkdir_p "#{to_create}/js"
	FileUtils::mkdir_p "#{to_create}/css"
	File.write "#{to_create}/Gemfile", gemfile
	File.write "#{to_create}/source/index.weave", source
end

command = ARGV.shift
if command == "build"
	build!(ARGV)
elsif command == "create"
	create!(ARGV)
else
	preview!
end

exit!