#!/usr/bin/env python
# Box Upload
# This code accepts arguments to upload files.

# WARNING!
# This code does not check SSL certificate validity!
# TODO: Implement certificate validation.

# Get the API key by making a Box application here: http://box.com/developers/services/edit/
api_key = 'replace-with-api-key'

import os
import sys
import urllib
import urllib2
import MultipartPostHandler, cookielib # MultipartPostHandler can be found at: http://pypi.python.org/pypi/MultipartPostHandler2
from xml.dom import minidom

def authenticate(api_key):
	xml_ticket = minidom.parseString(urllib.urlopen('https://www.box.com/api/1.0/rest?action=get_ticket&api_key='+api_key).read()) # This grabs the XML authentication ticket that we'll be using to generate the auth_token.
	status = xml_ticket.firstChild.childNodes[0].firstChild.data # This gets the ticket status to make sure we actually retrieved a valid ticket.
	if status == 'get_ticket_ok': # Did we get a valid ticket? If so, continue. If not, the user needs to check their API key.
		ticket = xml_ticket.firstChild.childNodes[1].firstChild.data
		print 'Please go to https://www.box.com/api/1.0/auth/'+ticket+' and follow the instructions there. Then, press ENTER or RETURN to continue.'
		raw_input()
		xml_auth_token = minidom.parseString(urllib.urlopen('https://www.box.net/api/1.0/rest?action=get_auth_token&api_key='+api_key+'&ticket='+ticket).read()) # This is the auth token in XML format.
		auth_status = xml_auth_token.firstChild.childNodes[0].firstChild.data # Reads the authentication status into "auth_status"
		if auth_status == 'get_auth_token_ok':
			auth_token = xml_auth_token.firstChild.childNodes[1].firstChild.data
			token_path = os.path.expanduser('~/.box_auth_token')
			auth_token_file = open(token_path,'w')
			auth_token_file.write(auth_token)
			auth_token_file.close()
			print 'The authentication token has been saved in the file "'+token_path+'". If you remove this file, you will need to reauthenticate.'
			print ''
			print 'Please run "boxupload" again to initiate your upload.'
			return
		elif auth_status == 'not_logged_in':
			print "I'm sorry, but you haven't authorized me to connect to your account. Please run \"boxupload\" again to restart the authentication process."
		elif auth_status == 'get_auth_token_error':
			print "I'm sorry, but there was an error in retrieving your authentication token. Please run \"boxupload\" again to restart the authentication process."
		else:
			print "Error: "+auth_status+". Try reauthenticating."
	else:
		print 'Error retrieving ticket! Check your API key.'

def get_account_info(api_key,auth_token):
	xml_account_info = minidom.parseString(urllib.urlopen('https://www.box.net/api/1.0/rest?action=get_account_info&api_key='+api_key+'&auth_token='+auth_token).read())
	status = xml_account_info.firstChild.childNodes[0].firstChild.data
	if status == 'get_account_info_ok':
		login = str(xml_account_info.firstChild.childNodes[1].childNodes[0].firstChild.data)
		email = str(xml_account_info.firstChild.childNodes[1].childNodes[1].firstChild.data)
		access_id = int(xml_account_info.firstChild.childNodes[1].childNodes[2].firstChild.data)
		user_id = int(xml_account_info.firstChild.childNodes[1].childNodes[3].firstChild.data)
		space_amount = int(xml_account_info.firstChild.childNodes[1].childNodes[4].firstChild.data) # Total capacity of the account in bytes
		space_used = int(xml_account_info.firstChild.childNodes[1].childNodes[5].firstChild.data) # Data usage of the account in bytes
		max_upload_size = int(xml_account_info.firstChild.childNodes[1].childNodes[6].firstChild.data) # Maximum file upload size in bytes
		account_info = [login, email, access_id, user_id, space_amount, space_used, max_upload_size] # Turn the data into a list for easy access
		return account_info
	elif status == 'not_logged_in':
		print "I'm sorry, your authentication token is invalid. Try removing your \"auth_token\" file and reauthenticating."
	elif status == 'Wrong input':
		print "I'm sorry, the API key you are using is mistyped."

def upload(file,auth_token):
	if not os.path.exists(file):
		print 'ERROR: File \"'+file+'\" was not found!'
		return
	url = 'https://upload.box.net/api/1.0/upload/'+auth_token+'/0'
	cookies = cookielib.CookieJar()
	opener = urllib2.build_opener(urllib2.HTTPCookieProcessor(cookies), MultipartPostHandler.MultipartPostHandler)
	params = { "file" : open(file, "rb") }
	xml_response = minidom.parseString(opener.open(url, params).read())
	upload_status = xml_response.firstChild.childNodes[0].firstChild.data
	if upload_status == 'upload_ok':
		print "Upload successful!"
	else:
		print "Upload failure!"

try: # Check for and read auth_token file
	token_path = os.path.expanduser('~/.box_auth_token')
	auth_token_file = open(token_path,'r')
	auth_token = auth_token_file.read()
	pass
except IOError as e:
	authenticate(api_key) # If there's no auth_token, initiate authentication.
	sys.exit(1)

if len(sys.argv) < 2:
    print "usage: "+sys.argv[0]+" file_1 [file_2] [file_3] ... [file_n]"
    sys.exit(1)

# account_info = get_account_info(api_key,auth_token)

''' # Uncomment for debugging.
print 'Auth token: ' + auth_token
print 'Max upload size: ' + str((account_info[6])/1024.0/1024) + ' MB'
print 'Total capacity: ' + str((account_info[4])/1024.0/1024) + ' MB | Space used: ' + str((account_info[5])/1024.0/1024) + ' MB | Space remaining: ' + str((account_info[4]-account_info[5])/1024.0/1024) + ' MB | Percent free: ' + str(100.0*(account_info[4]-account_info[5])/account_info[4]) + '%'
'''

arguments = sys.argv
arguments.pop(0)

for argv in arguments:
	upload(argv,auth_token)