import requests
import json
import os
import time
import sys
sys.path.insert(1, os.path.expanduser("~")+'/shell')
import gvs_cf





data = {'grant_type':"client_credentials",
		'resource':"https://graph.microsoft.com",
		'client_id':gvs_cf.onedrive_client_id,
		'client_secret':gvs_cf.onedrive_client_secret}
URL = gvs_cf.onedrive_generate_token_url
r = requests.post(url = URL, data = data)
j = json.loads(r.text)


TOKEN = j["access_token"]

onedrivefolder=sys.argv[2]
filepath=sys.argv[1]
uploadfilename=sys.argv[3]



URL = gvs_cf.onedrive_folder_url+onedrivefolder+"/"
headers={'Authorization': "Bearer " + TOKEN}
r = requests.get(URL, headers=headers)
	
def upload_file(filename,headers):

		upload_session = requests.post(URL+"/"+filename+":/createUploadSession", headers=headers).json()


		file_path = os.path.join(root,filename)

		with open(file_path, 'rb') as f:
			total_file_size = os.path.getsize(file_path)
			chunk_size = 327680
			chunk_number = total_file_size//chunk_size
			chunk_leftover = total_file_size - chunk_size * chunk_number
			i = 0
			while True:
				chunk_data = f.read(chunk_size)
				start_index = i*chunk_size
				end_index = start_index + chunk_size
				#If end of file, break
				if not chunk_data:
					break
				if i == chunk_number:
					end_index = start_index + chunk_leftover
				header = {'Content-Length':'{}'.format(chunk_size),'Content-Range':'bytes {}-{}/{}'.format(start_index, end_index-1, total_file_size)}
  				chunk_data_upload = requests.put(upload_session['uploadUrl'], data=chunk_data, headers=header)
				#print(chunk_data_upload)
				#print(chunk_data_upload.json())
				i = i + 1
		# os.remove(os.path.join(root, filename))

directory=filepath


for root, dirs, files in os.walk(directory):
	i=0
	# print(uploadfilename)

	for filename in files:

		if filename != uploadfilename:
					continue
		size=os.stat(directory+"/"+filename).st_size/1024

		if size <= 4096:
			filepath = os.path.join(root,filename)
			#print("Uploading "+filename+"....")
			fileHandle = open(filepath, 'rb')
			r = requests.put(URL+"/"+filename+":/content", data=fileHandle, headers=headers)
			fileHandle.close()
			# if r.status_code == 200 or r.status_code == 201:
			# 	os.remove(os.path.join(root, filename))
		else:
			upload_file(filename,headers)
		time.sleep(1)
		break
print("done")