#!/usr/local/bin/python
#
# find torrents >15 days
#

from sys import *
from os.path import *
from sha import *
import os
from shutil import *
import datetime
import time
from common import *

########################################################################
# delete a dir hierarchy
def deleteDownloadedTorrent(dir):
	if os.path.isfile(dir):
		os.unlink(dir)
	else:
		for root, dirs, files in os.walk(dir):
			for currentDir in dirs:
				deleteDownloadedTorrent(os.path.join( dir, currentDir ) )
			for currentFile in files:
				fullFilePath = os.path.join( dir, currentFile )
				os.remove( fullFilePath )
		os.rmdir( dir )

########################################################################
# find a string in a file
def fileNameExistsInFile(fname,readFile):
	try:
		fn = open( readFile, 'r' )
	except:
		#print 'Cant open %s' % readFile
		# if we can't open torrents.list, dont bother
		return False
	
	for line in fn:
		line = line[:-1]
		if basename(line) == fname:
			fn.close()
			return True
	fn.close()
	return False

########################################################################
# stat()s a dir/file, checks to see if its older than n number of days
def isFileOld(fn, daysOld=15):
	stats = os.stat(fn)
	lastmod = datetime.date.fromtimestamp(stats[8])

	today = datetime.date.today()
	oldTime = today + datetime.timedelta(days=-1 * daysOld)

	return oldTime > lastmod

########################################################################
# find old files
def findOldTorrents(rootdir,daysOld=15):
	oldFiles=[]

	for root, dirs, files in os.walk(rootdir):
		for currentDir in dirs:
			fn = os.path.join(root,currentDir)
			if isFileOld( fn,daysOld ):
				#print '%s' % fn
				oldFiles.append( fn )

		for currentFile in files:
			fn = os.path.join(root, currentFile)
			if isFileOld( fn, daysOld ):
				#print '%s' % fn
				oldFiles.append( fn )
	return oldFiles
########################################################################
# main 
####
if os.getuid() > 0:
	print 'Only root can run this!'
	exit()

if len(argv) > 1:
	oldtime = int(argv[1])
else:
	oldtime=15

oldies = findOldTorrents( EXPIRED_TORRENT_DIR, oldtime )

for old in oldies:
	oldfn = basename(old)
	canDelete = True

	## TODO: consult a list to obtain torrent users
	for user in ['howie', 'adam', 'nvilla']:
		tfile = os.path.join( '/home', user, 'torrents.list')
		
		if fileNameExistsInFile( oldfn, tfile):
			print '%s needs to grab %s' % (user, old)
			canDelete = False

	if canDelete:
		if os.path.exists(old):
			deleteDownloadedTorrent(old)
			print 'Removed: %s' % old