require 'rubygems'
require 'nokogiri'
require 'mechanize'

WWW::Mechanize.html_parser = Nokogiri::HTML

# TODO: Tests (mock out expected DropBox output)
# TODO: Directory object, File object

class DropBox
	#before_filter :login_filter, :except => [:login]
	
	def initialize(email, password, folder_namespace = "")
		@email = email
		@password = password
		@agent = WWW::Mechanize.new
		@folder_namespace = folder_namespace.gsub(/^\//,"")
		@logged_in = false
	end
	
	def login
		page = @agent.get('https://www.dropbox.com/login')
		login_form = page.forms.detect { |f| f.action == "/login" }
		login_form.login_email = @email
		login_form.login_password = @password
		
		home_page = @agent.submit(login_form)
		# todo check if we are logged in! (ie search for email and "Log out"
		@logged_in = true
		@token = home_page.at('//script[contains(text(), "TOKEN")]').content.match("TOKEN: '(.*)',$")[1]
		
		# check if we have our namespace
		
		home_page
	end
	
	def login_filter
		login unless @logged_in
	end
		
	def list(path = "/")
		index(path)
	end
	
	def index(path = "/")
 		login_filter
		path = namespace_path(path)
		
		list = @agent.post("/browse2#{path}?ajax=yes", {"d"=> 1, "t" => @token })
		
		listing = list.search('div.browse-file-box-details').collect do |file|
			details = {}
			details['name'] = file.at('div.details-filename a').content.strip
			details['url']  = file.at('div.details-filename a')["href"]
			#details['size'] = file.at('div.details-size a').try(:content).try(:strip)
			#details['modified'] = file.at('div.details-modified a').try(:content).try(:strip)
			
			if match_data = details['url'].match(/^\/browse_plain(.*)$/)
				details['directory'] = true
				details['path'] = normalize_namespace(match_data[1])
			else
				details['directory'] = false
				details['path'] = normalize_namespace(details['url'][33..-1])
			end
			
			details
		end
		
		return listing
	end
	
	def list_history(path)
		login_filter
		
		path = namespace_path(path)

		history = @agent.get("/revisions#{path}")
		listing = history.search("table.filebrowser > tr").select{|r| r.search("td").count > 1 }.collect do |r|
			
			# warning, this is very brittle!
			details = {}
			details["version"] = r.search("td a").first.content.strip
			details["url"] = r.search("td a").first["href"]
			details["size"] = r.search("td").last.content.strip
			details["modified"] = r.search("td")[2].content.strip
			details["version_id"] = details["url"].match(/^.*sjid=([\d]*)$/)[1]
			details['path'] = normalize_namespace(details['url'][33..-1])
			
			details
		end
		
		return listing
	end

	def get(path)
		show(path)
	end
	
	def show(path)
		# change to before filter
		login_filter
		
		path = namespace_path(path)
		
		#https://dl-web.dropbox.com/get/testing.txt?w=0ff80d5d&sjid=125987568
		@agent.get("https://dl-web.dropbox.com/get/#{path}").content
	end
		
	def create_directory(new_path, destination = "/" )
		# change to before filter
		login unless @logged_in
		destination = namespace_path(destination)
		@agent.post("/cmd/new#{destination}",{"to_path"=>new_path, "folder"=>"yes", "t" => @token })
		# check status code
	end
	
	def create(file, destination = "/")
		# change to before filter
		if @logged_in
			home_page = @agent.get('https://www.dropbox.com/home')
		else
			home_page = login
		end
		
		destination = namespace_path(destination)

		upload_form = home_page.forms.detect{ |f| f.action == "https://dl-web.dropbox.com/upload" }
		upload_form.dest = "/#{@folder_namespace}/#{destination}"
		upload_form.file_uploads.first.file_name = file if file
		
		@agent.submit(upload_form)
	end

	def update(file, destination = "/")
		create(file, destination)
	end
	
	def rename(file, destination)
		login_filter
		file = namespace_path(file)
		destination = namespace_path(destination)
		@agent.post("/cmd/rename#{file}", {"to_path"=> destination, "t" => @token })		
	end
	
	def destroy(paths)
		login_filter
		paths = [paths].flatten
		paths = paths.collect { |path| namespace_path(path) }
		@agent.post("/cmd/delete", {"files"=> paths, "t" => @token })
	end

	def purge(paths)
		login_filter
		paths = [paths].flatten
		paths = paths.collect { |path| namespace_path(path) }
		@agent.post("/cmd/purge", {"files"=> paths, "t" => @token })
	end

	private
	def namespace_path(path)
		# remove the start slash if we have one
		path.gsub(/^\//,"")
		if @folder_namespace.empty?
			"/#{path}"
		else
			"/#{@folder_namespace}/#{path}"			
		end
	end
	
	def normalize_namespace(file)
		file.gsub(/^\/#{@folder_namespace}/,"")
	end

end