#!/usr/bin/env python
 import subprocess
 import sys
 
 def cleanUpChroot(chrootPath):
     returnVal,listmountpoints=findmountpoints(chrootPath)

     if not returnVal:
         return False

     sortmountpoints(listmountpoints, chrootPath)

     print listmountpoints

     if not unmountmountpoints(listmountpoints):
         return False

     if not removeAllFilesFromChroot(chrootPath):
         return False

     return True
 
 def removeAllFilesFromChroot(chrootPath):
     cmd="rm -rf "+chrootPath+"/*"
     process = subprocess.Popen("%s" %cmd,shell=True,stdout=subprocess.PIPE,stderr=subprocess.PIPE)
     retval = process.wait()
     if retval!=0:
         print "Unable to remove files from chroot "+chrootPath
         return False
     return True

 def unmountmountpoints(listmountpoints):
     if listmountpoints is None:
         return True
     result=True
     for mountpoint in listmountpoints:
         cmd="umount "+mountpoint
         process = subprocess.Popen("%s" %cmd,shell=True,stdout=subprocess.PIPE,stderr=subprocess.PIPE)
         retval = process.wait()
         if retval!=0:
             result = False
             print "Unable to unmount "+mountpoint
             break
     if not result:
         print "Unable to unmount all mounts. Unable to clean up the chroot"
         return False
     return True

 def findmountpoints(chrootPath):

     if not chrootPath.endswith("/"):
         chrootPath = chrootPath + "/"

     cmd="mount | grep "+chrootPath+" | cut -d' ' -s -f3"

     process = subprocess.Popen("%s" %cmd,shell=True,stdout=subprocess.PIPE,stderr=subprocess.PIPE)
     retval = process.wait()
     if retval!=0:
         print "Unable to find mountpoints in chroot"
         return False,None
     mountpoints=process.communicate()[0]
     mountpoints= mountpoints.replace("\n"," ").strip()
     if mountpoints == "":
         print "No mount points found"
         return True,None
     listmountpoints=mountpoints.split(" ")
     return True,listmountpoints
 
 def sortmountpoints(listmountpoints,chrootPath):
     if listmountpoints is None:
         return True
     sortedmountpoints=listmountpoints
     sorted(sortedmountpoints)
     sortedmountpoints.reverse()
 
 def main():
     if len(sys.argv) < 2:
         print "Usage: ./clean-up-chroot.py <chrootpath>"

         sys.exit(1)
     if not cleanUpChroot(sys.argv[1]):
         sys.exit(1)
     sys.exit(0)

 
 if __name__=="__main__":
     main()