# encoding: utf-8

# License: refer to LICENSE file

require 'haml'
require 'pp' if $debug
require 'sinatra'
require 'sinatra/base'
require 'sinatra/reloader' if development? and Gem::Specification::find_all_by_name('sinatra-reloader').any?
#require 'rack/contrib/try_static'
require 'rbconfig'

#
# Kamishibai
# A manga reading webapp
#

module Kamishibai
	class Webserver < Sinatra::Base
		# shutdown hook, save database and bookmarks when exiting
		at_exit do
			$db.save
			$db.save_bookmarks
		end

		# smaller, quicker web server
		set :server, 'thin' unless RUBY_PLATFORM == 'java'

		if $debug
			set :environment, :development
		else
			set :environment, :production
		end

		configure :development do
			enable :logging
			if Gem::Specification::find_all_by_name('sinatra-reloader').any?
				register Sinatra::Reloader

				also_reload 'kamishibai/functions'
				also_reload 'kamishibai/patches'
				also_reload 'kamishibai/book'
				also_reload 'kamishibai/database'

				# reload webserver plug-ins
				Dir.glob( settings.root + '/../**/webserver_*.rb' ) { |f|
					also_reload f.gsub('.rb','')
				}
			end

			# enable caching for public directory
			set :static_cache_control, [:public, :max_age => 1]
		end

		configure :production do
			disable :logging
			disable :raise_errors

			# enable caching for public directory
			set :static_cache_control, [:public, :max_age => 1] # 300
		end

		configure do
			# listen to all interface
			set :bind, $settings.bind

			# setup port
			set :port, $settings.port


			# enable session support
			enable :sessions
			use Rack::Session::Pool, :expire_after => 60*60*24*365

			# enable http gzip
			use Rack::Deflater

			# setup template location
			set :views, File.expand_path( settings.root + '/../../views' )

			# setup public location
			set :public_folder, File.expand_path(settings.views + '/../public')

			# setup 2nd public location for vendor libraries
			# use Rack::TryStatic,
			# 	:root => File.expand_path(settings.views + '/../public/vendor'),
			# 	:urls => %w[/], try: ['.html', 'index.html', '/index.html']


			# authentication
			use Rack::Auth::Basic, "Restricted Area" do |username, password|
				[username, password] == [$settings.username, $settings.password]
			end

			# register mime type for static file
			mime_type :jpeg, 'image/jpeg'
			mime_type :png, 'image/png'
			mime_type :gif, 'image/gif'
			mime_type :css, 'text/css'
			mime_type :javascript, 'application/javascript'
		end

		# setup instance variable
		before do
			# global instance var
		end

		# helper functions
		helpers do
			# precheck the input from url
			def input_check( bookcode, page )
				page = page.to_i

				if $db.has_bookcode?( bookcode )
					@book = $db.get_book( bookcode )
				else
					not_found "No such book code. #{ bookcode }"
				end

				unless @book.pages
					not_found "Book contain no images. #{ bookcode } #{ @book.fullpath }"
				end

				if page < 1 or page > @book.pages
					not_found "No such page. #{ page } #{ bookcode } #{ @book.fullpath }"
				end

				if ! FileTest.exists?( @book.fullpath )
					not_found "File don't exists. #{ bookcode } #{ @book.fullpath }"
				end

				if ! FileTest.file?( @book.fullpath )
					not_found "Not a file. #{ bookcode } #{ @book.fullpath }"
				end
			end

			# regular expression from POST keyword search
			def pregex
				keyword = request['keyword'].untaint
				keyword = keyword.gsub(' ','.+')
				return Regexp.new( keyword, Regexp::IGNORECASE )
			end
		end


		# detect screen dimension
		post '/screen' do
			session[:width] = params[:width].to_i
			session[:height] = params[:height].to_i

			puts "screen dimension. #{session[:width]} x #{session[:height]}" if $debug
		end

		# redirect index page
		get '/' do
			redirect '/browse/'
			# if request.user_agent =~ /(android|tablet|iphone|ipad)/i
			# 	redirect '/browse_tablet/'
			# else
			# 	redirect '/browse/'
			# end
		end

		get '/statistics' do
			haml :statistics, :layout => false
		end

		# browse page with folder/file navigation
		get '/browse/' do
			haml :browse, :layout => false
		end

		get '/bookinfo/*' do |bookcode|
			input_check( bookcode, 1)

			content_type :javascript

			page = @book.page ? @book.page : 'null'

"var book = {
	bookcode: '#{@book.bookcode}',
	basename: '#{File.basename( @book.fullpath ).gsub(/\.cbz$/i,'').escape_html}',
	title:    '#{@book.title.to_s.escape_html}',
	author:   '#{@book.author.to_s.escape_html}',
	page:     #{page},
	pages:    #{@book.pages}
};
"
		end

		get '/setbookmark/*/*' do |bookcode, page|
			page = page.to_i
			input_check( bookcode, page )

			$db.set_bookmark(bookcode, page)

			"bookmarked"
		end

		def self.get_or_post(path, opts={}, &block)
			get(path, opts, &block)
			post(path, opts, &block)
		end

		# reader page
		get '/reader/' do
			cache_control :public, :must_revalidate, :max_age => 1

			haml :reader, :layout => false
		end

		# list sources
		get '/list_sources' do
			content_type :javascript

			%Q{
sources = [
#{$settings.srcs.collect{ |x| "  \"#{x}\"" }.join(",\n")}
]
}
		end

		# directory browse page
		get_or_post '/lists_dir' do
			content_type :text
			path = File.expand_path( request['dir'].untaint )
			order_by = request['order_by'].untaint

			unless FileTest.exists?(path)
				errmsg = "Error. No such path. #{path}"
				puts errmsg if $debug
				# not_found errmsg

				return errmsg
			end

			unless File.stat(path).readable_real?
				errmsg = "Error. Not readable path. #{path}"
				puts errmsg if $debug
				# not_found errmsg

				return errmsg
			end

			# check and add new books, existing books will not be added
			$db.add_books( [ path ], false)
			# refresh db, make sure book filepath is valid
			$db.refresh_bookcodes


			html = "<ul id=\"ul-lists\" class=\"ul-lists\">\n"

			html << "\t<li class=\"directory collapsed updir\"><a href=\"#dir=#{File.dirname(path)}\" rel=\"#{File.dirname(path)}/\"><img src=\"/images/folder-mini-up.png\" /><span>..</span></a></li>\n"



			# poulate the lists with files(books) and directory
			lists = []
			Dir.glob(path.escape_glob + '/*').entries.sort.each { |fp|
				next unless File.stat(fp).readable_real?

				dir = File.dirname(fp)
				f =   File.basename(fp)
				ext = File.extname(fp)[1..-1]

				next unless pregex.match( f ) # skip unless file/dir name match the searchbox's keyword

				if FileTest.directory?(fp) and File.stat(fp).executable_real?
					# a directory
					lists << fp

				elsif ext == 'cbz' and File.stat(fp).readable_real?
					# a file
					
					bookcode = $db.get_bookcode_byfilename( fp )
					unless bookcode
						puts "ERROR: bookcode #{bookcode} not found! skipping to next book..."
						next
					end

					book = $db.get_book( bookcode )

					lists << book
				end
			}

			# sort by order
			lists2 = {}
			case order_by
				when 'size'
					lists.each { |x|

						if x.class == Book
							lists2[x] = x.size
						else
							# x == full path of file or directory
							lists2[x] = 0
						end
					}

					lists2 = lists2.sort { |a, b|	b[1] <=> a[1] }

				when 'date'
					lists.each { |x|
						if x.class == Book
							lists2[x] = x.mtime
						else
							# x == full path of file or directory
							lists2[x] = File.stat(x).mtime.to_i
						end
					}
					lists2 = lists2.sort { |a, b| b[1] <=> a[1] }

				else
					# by name, default
					lists.each { |x|
						if x.class == Book
							lists2[x] = File.basename( x.fullpath )
						else
							# x == full path of file or directory
							lists2[x] = File.basename( x )
						end
					}
					lists2 = lists2.sort { |a, b| a[1] <=> b[1] }
			end

			# create html from lists(2)
			lists2.each { |item, dat|

				if item.class == String
					# a directory

					li_classes = ['directory', 'collapsed']

					if item == 'Trash'
						el_id = 'id="trash"'
						li_classes << 'trash'
						if Dir.glob(item.escape_glob + '/*.cbz').entries.length > 0
							icon = 'trash-full-mini.png'
						else
							icon = 'trash-empty-mini.png'
						end
					else
						icon = 'folder-mini.png'
					end

					html << "\t<li class=\"" + li_classes.join(' ') + "\" #{el_id}><a href=\"#dir=#{item}\" rel=\"#{item}/\"><img src=\"/images/" + icon + "\" /><span>#{File.basename(item)}</span></a></li>\n"

				elsif item.class == Book
					# a file

					book = item
					bookcode = book.bookcode
					fp = book.fullpath
					f = File.basename(fp)
					ext = File.extname(fp)[1..-1]

					img = "<img class='lazy fadeIn fadeIn-1s fadeIn-Delay-Xs' data-original='/thumbnail/#{bookcode}' alt='Loading...' />"

					if book.page
						# prepare the value for the visual read progress
						page  = book.page
						pages = book.pages

						bn = (1.0*page/pages)*100
						pc = bn - bn % 10

						# read percentage css class
						if pc > 0
							readstate = 'read' + pc.to_i.to_s
						else
							readstate = 'read5'
						end

						href = '/tablet#book=' + bookcode + '&page=' + page.to_s
					else
						readstate = 'read0'
						href = '/tablet#book=' + bookcode
					end

					html << "\t<li class=\"file ext_#{ext}\"><a href=\"#{href}\" bookcode=\"#{bookcode}\" rel=\"#{fp}\">#{img}<span class=\"#{readstate}\">#{f.escape_html}</span><span class=\"badge badge-info bookpages\">#{book.pages}</span></a></li>\n"
				end
			}

			return html
		end

		# directory browse page
		post '/jqueryFileTree' do
			content_type :text
			path = request['dir'].untaint

			html = "<ul class=\"jqueryFileTree\" style=\"display: none;\">\n"

			if FileTest.exists?(File.expand_path(path))

				# chdir() to user requested dir (root + "/" + dir)
				Dir.chdir(File.expand_path(path).untaint);

				#loop through all directories
				Dir.glob("*") { |x|
					next unless File.directory?(x.untaint)
					next unless File.stat(x.untaint).readable_real? and File.stat(x.untaint).executable_real?

					fp = path + x
					html << "\t<li class=\"directory collapsed\"><a href=\"#\" rel=\"#{fp}/\" onclick=\"selected_dir('#{fp}');\">#{x.escape_html}</a></li>\n";
				}

			end

			html << "</ul>\n"

			return html
		end

		# cbz thumbnail
		get '/thumbnail/*' do |bookcode|
			cache_control :public, :must_revalidate, :max_age => 3600

			# check and setup the variable
			input_check( bookcode, 1 )

			content_type :jpeg

			# generate thumbnail
			image = mk_thumb( @book.fullpath )

			return image
		end

		# cbz file loader
		get '/cbz/*/*' do |bookcode, page|
			# # fake delay
			# if $imgNum
			# 	$imgNum += 1
			# else
			# 	$imgNum = 1
			# end
			# pp $imgNum

			page = page.to_i
			cache_control :public, :must_revalidate, :max_age => 3600

			input_check( bookcode, page )

			# width/height
			quality = $settings.default_image_quality
			width = session[:width].to_i
			height = session[:height].to_i
			# hard set to hd
			# width  = 1080
			# height = 1920

			# fix/prevent invalid input
			quality = 0 if quality < 0
			quality = 100 if quality > 100
			width = 0 if width < 0 # 0 will be treated as do not resize
			height = 0 if height < 0

			# image type, image data
			image = open_cbz( @book.fullpath, page)

			# set content type, png/jpeg/png/gif/etc
			itype = image_type( image )
			content_type itype

			# # fake delay
			# Thread.new {
			# 	sleep 7
			# 	$imgNum = 0
			# 	sleep 7
			# 	$imgNum = 0
			# }
			# sleep 1.5 * $imgNum
			# $imgNum -= 1

			if width > 0 and height > 0 and $settings.image_resize
				# resize image
				img_resize( image, width, height, { :quality => quality })
			else
				# give raw
				image
			end
		end

		get '/list_db' do
			html = "<html><body>\n"
			p $db.books.length
			$db.books.each { |bookcode, book|
				html << "<pre>#{bookcode}  #{book.page.to_s.rjust(3)}/#{book.pages}  #{book.fullpath}</pre>\n"
			}
			html << "</body></html>"
			html
		end
	end
end


# load webserver plug-ins
Dir.glob( settings.root + '/../**/webserver_*.rb' ) { |f|
	require f.gsub('.rb','')
}