index.py

# Matthew Coppola
# server code for raspberry pi 0.1
# implementation for pro tools session repository
import os, time, shutil, zipfile, datetime, subprocess, threading
from bottle import route, run, request, redirect, FlupFCGIServer, static_file
from HTMLwriter import HTMLwriter
from util import zipdir, formatLoc, unzip, unzipReplace, sendEmail, noteCountForSongs, notesForFile, timeSinceNow, niceDate


class ZipDir ( threading.Thread ):
	def __init__(self, account, loc, emailTo):
              threading.Thread.__init__(self)
              self.account = account
              self.loc = loc
              self.emailTo = emailTo

	def run (self):
		subprocess.call('site/scripts/zip.sh ' + self.account + ' ' + self.loc, shell=True)
		subject = '%s has been Zipped!' % self.loc
		body = self.loc + ' has been zipped.  You can download the zip file from within the %s directory' % self.loc
		sendEmail('mcoppola832@gmail.com', self.emailTo, subject, body)
		#redirect ('/account/' + self.account + '/' + self.loc)

# --------------------------------------------------------------------------------------------//
# HOME FUNCTIONS------------------------------------------------------------------------------//

@route('/', method='GET')
def index():
	return html.head + html.loginForm + html.linksFooter + html.foot

@route('/login', method='GET')
def login():
	redirect('/')

@route('/login', method='POST')
def do_login():
	global user, loggedIn, users, passwords, access
	user = request.forms.get('login')
	password = request.forms.get('password')
	count=0
	for u in users:
		if u == user:
			if passwords[count] == password:
				loggedIn = str(user)
				access = True
				redirect('/home/' + user)
			else:
				return html.head + html.index + html.loginForm + html.linksFooter + '<br><p style="text-align: center">Wrong password.  Try again.</p>'
		else:
			count+=1
	return html.head + html.index + html.loginForm + html.linksFooter + '<br><p style="text-align: center">Username not found.</p>'


@route('/logout')
def logout():
	global loggedIn, loc
	loggedIn = ''
	loc = ''
	redirect('/')

@route('/about')
def about():
	return html.about

# STATIC FILE RETURN ----------------------------------------/
@route('/static/<filename>')
def server_static(filename):
    return static_file(filename, root='static/')

@route('/file/<file>')
def returnFile(file):
	return open(file, 'r')

# RESUME ----------------------------------------------------/
@route('/resume')
def resume():
	return open('resume/index.html', 'r')

@route('/resume/<loc>/<filename:path>', method='GET')
def send_static(loc, filename):
	return static_file(filename, root= './resume/' + loc)

# VitaminME -------------------------------------------------/
@route('/vm')
def vitaminMe():
	out = subprocess.Popen('php vm/public_html/index.php', shell=True, stdout=subprocess.PIPE)
	return out.stdout.read()

@route('/vm/<loc>/<filename:path>')
def send_staticVM(loc, filename):
	return static_file(filename, root= './vm/public_html' + loc)

# USER HOME -------------------------------------------------/
@route('/home/<user>')
def userHome(user):
	# display groups that the user belongs to
	checkLogin(user)
	global groups
	fl = open('site/html_gen.txt', 'w')
	fl.write(html.head + html.userHomeHeader(user, '', user) + html.accountListHeader)
	# write all groups that user belongs to
	for g in groups[user]:
		done = False
		commtsBehind = open('site/' + user + '/commits_behind.txt', 'r').read().splitlines()
		for l in commtsBehind:
			if l.startswith(g):
				if(len(l.split(',')) > 2):
					commits = int(len(l.split(',')) - 1)/2
					fl.write('<a href="/account/' + g + '" class="list-group-item li_pt"><b>' + g + '<span class="badge pull-right">%d</span></b></a> \n' % commits)
					done = True
					break
		if not done:
			fl.write('<a href="/account/' + g + '" class="list-group-item li_pt"><b>' + g + '</b></a> \n')
	fl.write(html.accountListFooter)
	fl.write(html.notesListHeader)
	# write the latest notification in each group
	for g in groups[user]:
		with open('site/' + g + '/log.txt', 'r') as logFile:
			log = logFile.readline().split(',')
			for field in log:
				field.strip('\n')
			if (len(log) > 1):
				fl.write(html.notificationLatest(user, log, timeSinceNow(log[0])))
	fl.write(html.notesListFooter + html.homeFooter + html.foot)
	fl.close()
	txt = open('site/html_gen.txt', 'r')
	return txt

# -----------------------------------------------------------/
@route('/master')
def about():
	return html.addUser


def printGenDirectory(account, loc):
	global user
	fl = open('site/html_gen.txt', 'w')
	fl.write(html.head + html.proToolsProjectHeader(account, formatLoc(loc), user))
	if (loc != ''):
		loca = loc + '/'
	else:
		loca = loc
	# write all songs
	fl.write(html.songListHeader(account))
	# get notification count
	commitsBehind = noteCountForSongs(user, account)
	for top, dirs, files in os.walk('data/' + account + '/' + loc ):
		count = 0
		for f in dirs:
			if (commitsBehind[f] > 0):
				commits = commitsBehind[f]
				fl.write('<a href="/account/' + account + '/' + loca + f + '" class="list-group-item li_song"><b>' + f + '<span class="badge pull-right">%d</span></b></font></a> \n' % commits)
			else:
				fl.write('<a href="/account/' + account + '/' + loca + f + '" class="list-group-item li_song"><b>' + f + '</b></font></a> \n')
		# for f in files:
		# 	count = count + 1
		# 	size = os.path.getsize('data/' + account + '/' + loc + '/' + f)
		# 	date = time.ctime(os.path.getmtime('data/' + account + '/' + loc + '/' + f))
		# 	fl.write('<li>' + '<a href="/' + account + '/down/' + loca + str(count) + '">'
		# 		+ f + '</a>' + ' | '
		# 		+ str(size) + ' bytes' +  ' | '
		# 		+ str(date) + '</li>' + '\n')
		break
	fl.write(html.songListFooter + html.notesListHeader)

	# write all notifications
	with open('site/' + user + '/notifications.txt', 'r') as logFile:
		logLines = logFile.readlines()
	for l in logLines:
		log = l.split(',')
		if (len(log) > 0):
			if (log[1] == account):
				fl.write(html.notification(user, log, timeSinceNow(log[0])))
	fl.write(html.notesListFooter + html.linksListHeader(account))

	# write all links
	with open('site/' + account + '/links.txt', 'r') as f:
		links = f.readlines()
	index = 0;
	for l in links:
		# url,title
		link = l.split(',')
		fl.write(html.linkItem(user, account, link, index))
		index+=1
	fl.write(html.linksListFooter + '</div><!-- /row -->')
	fl.write(html.foot)
	fl.close()
	txt = open('site/html_gen.txt', 'r')
	return txt

def printProToolsDirectory(account, loc):
	global user
	fl = open('site/html_gen.txt', 'w')
	fl.write(html.head + html.proToolsSessionHeader(account, formatLoc(loc), user))
	if (loc != ''):
		loca = loc + '/'
	else:
		loca = loc
	for top, dirs, files in os.walk('data/' + account + '/' + loc ):
		count = 0
		fl.write('<ul style="margin-left:-44px">')
		for f in dirs:
			fl.write(html.folderItem(account, loca, f))
		fl.write('</ul> <ol>')
		for f in files:
			count = count + 1
			size = os.path.getsize('data/' + account + '/' + loc + '/' + f)
			date = time.ctime(os.path.getmtime('data/' + account + '/' + loc + '/' + f))

			if f.endswith('.ptx'):
				if (notesForFile(user, account, loc, f)):
					fl.write('<a href="/'+ account + '/down/' + loca + str(count) + '"style="margin-left:-38px"><i style="color:#F2522E" class="glyphicon glyphicon-flag"></i>&nbsp' + '<font style="margin-left:4px" color="#66CCFF">'
						+ f + '</a></font>' + ' | '
						+ str(size/1024/1024) + ' MB' +  ' | '
						+ str(date) + '<br>' + '\n')
				else:
					fl.write('<a href="/'+ account + '/down/' + loca + str(count) + '" style="margin-left:-15px">' + '<font color="#66CCFF">'
						+ f + '</a></font>' + ' | '
						+ str(size/1024/1024) + ' MB' +  ' | '
						+ str(date) + '<br>' + '\n')
			elif f.endswith('.ptf'):
				if (notesForFile(user, account, loc, f)):
					fl.write('<a href="/'+ account + '/down/' + loca + str(count) + '"style="margin-left:-38px"><i style="color:#F2522E" class="glyphicon glyphicon-flag"></i>&nbsp' + '<font style="margin-left:4px" color="#6699FF">'
						+ f + '</a></font>' + ' | '
						+ str(size/1024/1024) + ' MB' +  ' | '
						+ str(date) + '<br>' + '\n')
				else:
					fl.write('<a href="/'+ account + '/down/' + loca + str(count) + '" style="margin-left:-15px">' + '<font color="#6699FF">'
						+ f + '</a></font>' + ' | '
						+ str(size/1024/1024) + ' MB' +  ' | '
						+ str(date) + '<br>' + '\n')
			elif f.endswith('.zip'):
				fl.write('<a href="/'+ account + '/down/' + loca + str(count) + '" style="margin-left:-15px">' + '<font color="#00CC99">'
					+ f + '</a></font>' + ' | '
					+ str(size/1024/1024) + ' MB' +  ' | '
					+ str(date) + '<br>' + '\n')
			else:
				fl.write('<a href="/'+ account + '/down/' + loca + str(count) + '" style="margin-left:-15px">' + '<font color="#999">'
					+ f + '</a></font>' + ' | '
					+ str(size/1024/1024) + ' MB' +  ' | '
					+ str(date) + '<br>' + '\n')
		fl.write('</ol>')
		break

	# write session history
	fl.write('</div>' + html.sessionHistoryHeader)
	bandLog = open('site/%s/log.txt' % account, 'r').read().splitlines()
	songLog = []
	commitsBehind = noteCountForSongs(user, account)[loc]
	count = 0
	for line in bandLog:
		fields = line.split(',')
		if (fields[2] == loc):
			songLog.append(fields)
	for note in songLog:
		if (count < commitsBehind):
			fl.write(html.sessionHistoryItemBehind(user, note, niceDate(note[0])))
			count += 1
		else:
			fl.write(html.sessionHistoryItem(user, note, niceDate(note[0])))
	fl.write(html.sessionHistoryFooter + html.foot)
	fl.close()
	txt = open('site/html_gen.txt', 'r')
	return txt

# PRO TOOLS FUNCTIONS --------------------------------------------------/

@route('/replace/<account>/<loc>')
def replace(account, loc):
	global user
	checkLogin(user)
	return html.uploadSongZip(account, loc)

@route('/replace/<account>/<loc>', method='POST')
def doReplace(account, loc):
	global user
	checkLogin(user)
	checkAccess()
	upload = request.files.get('upload')
	if(upload.filename and upload.filename.endswith('.zip')):
		fn = os.path.basename(upload.filename)
		open('data/' + account + '/' + fn, 'w').write(upload.file.read())
		print 'file %s was upload' % fn
		#move directory to _previous
		try:
			shutil.move('data/' + account + '/' + loc + '/', 'data/' + account + '/_previous/')
		except:
			shutil.rmtree('data/' + account + '/_previous/' + loc + '/')
			shutil.move('data/' + account + '/' + loc + '/', 'data/' + account + '/_previous/' + loc + '/')
		#zip directory 
		(dirName, fileName) = fn.split('.')
		if not os.path.exists('data/' + account + '/' + dirName):
			os.mkdir('data/' + account + '/' + dirName)
		unzip('data/' + account + '/' + fn, 'data/' + account + '/' + dirName)
		os.remove('data/' + account + '/' + fn)
		redirect('/account/' + account + '/' + dirName)
	else:
		return 'error, directory was not replaced'	


@route('/mkzip_prompt/<account>/<loc>', method='GET')
def makeZipPrompt(account, loc):
	global user
	checkLogin(user)
	checkAccess()
	return html.mkzip_prompt(account, loc)

@route('/mkzip_prompt/<account>/<loc>', method='POST')
def domakeZipPrompt(account, loc):
	email = request.forms.get('email')
	redirect ('/mkzip/'+account+'/'+loc+'/'+email)

@route('/mkzip/<account>/<loc>/<emailTo>')
def makeZip(account, loc, emailTo):
	global user
	checkLogin(user)
	checkAccess()
	#make zip of directory
	ziplog = open('ziplog', 'wb')
	ziplog.close()
	z = ZipDir(account, loc, emailTo)
	z.start()
	#redirect ('/account/' + account + '/' + loc)
	redirect('/mkziplog')

@route('/mkziplog')
def mkZiplog():
	return html.mkziplog()
	

@route('/addPTX/<account>/<loc>')
def addPTX(account, loc):
	global user
	checkLogin(user)
	return html.uploadPTX(account, loc)

@route('/addPTX/<account>/<loc>', method='POST')
def doAddPTX(account, loc):
	global user
	checkLogin(user)
	checkAccess()
	upload = request.files.get('upload')
	message = request.forms.get('message').strip(',')
	if upload.filename:
		print upload.filename
		head, fn = os.path.split(upload.filename)
		if not fn.endswith('.ptx') and not fn.endswith('.ptf'):
			return 'error, please choose a ptx or ptf file'
		for top, dirs, files in os.walk('data/' + account + '/' + loc ):
			for f in files:
				if (f.lower() == fn.lower()):
					#move existing session file
					split = fn.split('.')
					newName = split[0] + "_" + datetime.datetime.now().strftime("%Y-%m-%d %H:%M") + split[1]
					shutil.move('data/' + account + '/' + loc + '/' + fn, 'data/' + account + '/' + loc + '/Session File Backups/' + newName)
					#write new session file
					open('data/' + account + "/" + loc + '/' + fn, 'wb').write(upload.file.read())
					logger('addPTX', loc, account, message, fn)
					redirect('/account/' + account + '/' + loc)
					break
			open('data/' + account + "/" + loc + '/' + fn, 'wb').write(upload.file.read())
			logger('addPTX', loc, account, message, fn)
			redirect('/account/' + account + '/' + loc)
	return 'error, no file was uploaded'


@route('/addSong/<account>', method='POST')
def doAddSong(account):
	global user
	checkLogin(user)
	checkAccess()
	upload = request.files.get('upload')
	name = request.files.get('')
	if upload.filename:
		head, fn = os.path.split(upload.filename)
		if not fn.endswith('.zip'):
			return 'please choose a .zip file'
		open('data/' + account + '/' + fn, 'wb').write(upload.file.read())
		(dirName, fileName) = fn.split('.')
		if not os.path.exists('data/' + account + '/' + dirName):
			os.mkdir('data/' + account + '/' + dirName)
		unzip('data/' + account + '/' + fn, 'data/' + account + '/' + dirName)
		os.remove('data/' + account + '/' + fn)
		logger('addSong', fn.split('.')[0], account)
		redirect('/account/' + account)

@route('/addAudio/<account>/<loc>')
def addAudio(account, loc):
	return 'comming soon'

# GENERIC FUNCTIONS ----------------------------------------------------/

@route('/upload_gen_folder/<account>', method='GET')
def genUpload(account):
	global user
	checkLogin(user)
	return html.uploadGen(account, '')

@route('/upload_gen_folder/<account>', method='POST')
def doGenUpload(account):
	global user
	checkLogin(user)
	upload = request.files.get('upload')
	if upload.filename:
		fn = os.path.basename(upload.filename)
		open('data/' + account + '/' + fn, 'wb').write(upload.file.read())
		redirect('/account/' + account)

@route('/upload_gen_folder/<account>/<loc>', method='GET')
def genUpload(account, loc = ''):
	global user
	checkLogin(user)
	return html.uploadGen(account, loc)

@route('/upload_gen_folder/<account>/<loc>', method='POST')
def doGenUpload(account, loc = ''):
	global user
	checkLogin(user)
	upload = request.files.get('upload')
	if upload.filename:
		fn = os.path.basename(upload.filename)
		open('data/' + account + '/' + loc + '/' + fn, 'wb').write(upload.file.read())
		redirect('/account/' + account + '/' + loc)

#download html link for files in top directory
@route('/<account>/down/<num:int>')
def download(account, num):
	global user
	checkLogin(user)
	checkAccess()
	num = int(num) - 1
	files = [f for f in os.listdir('data/' + account) 
		if os.path.isfile('data/' + account + '/' + f)]
	fn = files[num]
	return static_file(fn, root=('data/' + account), download=fn)

#download html link for files in current directory
@route('/<account>/down/<loc>/<num:int>')
def download(account, loc, num):
	global user
	checkLogin(user)
	checkAccess()
	num = int(num) - 1
	files = [f for f in os.listdir('data/' + account + '/' + loc) 
		if os.path.isfile('data/' + account + '/' + loc + '/' + f)]
	fn = files[num]
	notes = open('site/' + user + '/' + 'commits_behind.txt', 'r+').readlines()
	newLine = ''
	for i in range(len(notes)):
		if (notes[i].startswith(account)):
			newLine = notes[i]
			fields = notes[i].split(',')
			j = 0
			while (j < len(fields) - 1):
				if (fields[j].strip('\n') == fn):
					fields.pop(j)
					fields.pop(j - 1)
					newLine = ''
					for f in fields:
						if (len(f.strip('\n')) > 0):
							newLine += f.strip('\n') + ','
					j = j - 2
				j += 1
			notes[i] = newLine + '\n'
			break
	f = open('site/' + user + '/' + 'commits_behind.txt', 'w')
	for note in notes:
		f.write(note)
	f.close()
	return static_file(fn, root=('data/' + account + '/' + loc), download=fn)

#hack FIX 
@route('/<account>/down/<loc>/<loc2>/<num:int>')
def downloadHACK(account, loc, loc2, num):
	global user
	checkLogin(user)
	checkAccess()
	loc = loc+'/'+loc2
	num = int(num) - 1
	files = [f for f in os.listdir('data/' + account + '/' + loc) 
		if os.path.isfile('data/' + account + '/' + loc + '/'+ f)]
	fn = files[num]
	return static_file(fn, root=('data/' + account + '/' + loc), download=fn)

@route('/ptkeeper_demo/<num:int>')
def ptKeeperDemo(num):
	if num == 8320620:
		global loggedIn, user, access
		loggedIn = 'mc'
		user = 'mc'
		access = False
		redirect('/account/museyroom')
	else:
		redirect('/noaccess')

#keep generic at bottom of file
@route('/account/<account>', method='GET')
def accountGeneric(account):
	global user
	checkLogin(user)
	settings = open('site/' + account + '/settings.txt', 'r').read().splitlines()
	if (settings[0] == 'p'):
		return printGenDirectory(account, '')
	elif (settings[0] == 'a'):
		return printGenDirectory(account, '')

@route('/account/<account>/<dir>', method='GET')
def accountGenericLoc(account, dir):
	global user
	checkLogin(user)
	settings = open('site/' + account + '/settings.txt', 'r').read().splitlines()
	if (settings[0] == 'p'):
		return printProToolsDirectory(account, dir)
	elif (settings[0] == 'a'):
		return printGenDirectory(account, dir)


@route('/account/<account>/<dir>/<loc>', method='GET')
def accountGenericLocHack(account, dir, loc):
	global user
	checkLogin(user)
	settings = open('site/' + account + '/settings.txt', 'r').read().splitlines()
	if (settings[0] == 'p'):
		return printProToolsDirectory(account, str(dir + '/' + loc))
	elif (settings[0] == 'a'):
		return printGenDirectory(account, str(dir + '/' + loc))

@route('/account/<account>/<dir>/<loc>/<loc2>', method='GET')
def accountGenericLoc2Hack(account, dir, loc, loc2):
	global user
	checkLogin(user)
	settings = open('site/' + account + '/settings.txt', 'r').read().splitlines()
	if (settings[0] == 'p'):
		return printProToolsDirectory(account, str(dir + '/' + loc + '/' + loc2))
	elif (settings[0] == 'a'):
		return printGenDirectory(account, str(dir + '/' + loc + '/' + loc2))

# CHANGE PASSWORD ------------------------------------------------/
@route('/pass/<user>', method='GET')
def changePassword(user):
	return 'not available'

# ADD LINK ------------------------------------------------/
# @route('/addlink/<account>', method='GET')
# def addLink(account):
# 	global user
# 	checkLogin(user)
# 	return html.addLinkForm(account)

@route('/addlink/<account>', method='POST')
def doAddLink(account):
	global user
	checkLogin(user)
	title = request.forms.get('title')
	url = request.forms.get('url')
	if url.startswith('https://'):
		url = url[7:]
	file = open("site/%s/links.txt" % account, "r+")
	links = file.read().splitlines()
	links.append('https://%s,%s' %(url, title))
	file.seek(0)
	for l in links:
		file.write(l+'\n')
	file.truncate
	file.close()
	redirect('/account/%s' % account)

@route('/removeLink/<account>/<i:int>')
def removeLink(account, i):
	global user
	checkLogin(user)
	file = open("site/%s/links.txt" % account, "r+")
	links = file.read().splitlines()
	file.seek(0)
	file.truncate()
	print links
	links.pop(i)
	print links
	for l in links:
		file.write(l+'\n')
	
	file.close()
	redirect('/account/%s' % account)

@route('/removeNotification/<date>/<previous>')
def removeNotification(date, previous):
	global user
	checkLogin(user)
	file = open("site/%s/notifications.txt" % user, "r+")
	notes = file.read().splitlines()
	for i in range(len(notes) - 1):
		if notes[i].startswith(date):
			notes.pop(i)
			break
	file.close()
	file = open("site/%s/notifications.txt" % user, "wb")
	for note in notes:
		file.write(note + '\n')
	file.close()
	redirect('/account/%s' % previous)


@route('/noaccess')
def noAccess():
	return html.noAccess

def checkLogin(user = 'null'):
	global loggedIn, groups
	verified = False
	# if D: 
	# 	return
	if (loggedIn == user):
		verified = True
	if (not verified):
		redirect('/login')

def checkAccess():
	global access
	if not access:
		redirect('/noaccess')

def logger(action, loc, account, message='', filename=''):
	global user, groups
	log = open('site/' + account + '/log.txt').read()
	newLog = str(datetime.datetime.now().strftime("%Y-%m-%d-%H-%M-%S")) + ',' +  account + ',' + loc +  ',' + user + ',' + logActions[action] + ',' + message + '\n'
	log = str(newLog) + str(log)
	open('site/' + account + '/log.txt', 'wb').write(log)
	for u in groups.keys():
		# write notification for everyone including self
		for group in groups[u]:
			# check if user is collaborator on group
			if (group == account):
				notes = open('site/' + u + '/notifications.txt', 'r+').readlines()
				newNotes = open('site/' + u + '/notifications.txt', 'w')
				newNotes.write(newLog)
				for oldNote in notes:
					newNotes.write(oldNote)
				newNotes.close()
				break
		# write commit behind for everyone else
		print 'u = ' + u + ' user = ' + user
		if (u != user):
			for group in groups[u]:
				# check if user is collaborator on group
				print 'group = ' + group + ' account = ' + account
				if (group == account):
					try:
						with open('site/' + u + '/commits_behind.txt'):
							commitsBehind = open('site/' + u + '/commits_behind.txt', 'r+').readlines()
							newFile = open('site/' + u + '/commits_behind.txt', 'wb')
							for l in commitsBehind:
								if l.startswith(account):
									l = l.strip('\n') + loc + ',' + filename + ',\n'
								print l
								newFile.write(l)
							newFile.close()
							break
					except IOError:
					   print 'Error looking for %s : commits_behind.txt' %u
						

#locActions (action: description)
logActions = {'addPTX': 'ptx commit', 'addPTF': 'ptf commit', 'addSong': 'new song'}
#groups (user: [groups])
groups = {'null':[], 'ben':['museyroom'], 'mc':['wellboys','museyroom', 'caddy', 'drunken_bear'], 'david':['drunken_bear', 'caddy'], 'owen':['drunken_bear', 'wellboys'], 'caddy':['caddy']}
users = open('site/users.txt', 'r').read().splitlines()
passwords = open('site/passwords.txt', 'r').read().splitlines()
html = HTMLwriter()
user = 'null'
access = True
password = ''
loggedIn = ''
#on pi server=FlupFCGIServer
run(host='127.0.0.1', port=8080, server=FlupFCGIServer)