#! /usr/bin/env python
#*******************************************************************************
# ALMA - Atacama Large Millimiter Array
# Copyright (c) European Southern Observatory, 2013 
# 
# This library is free software; you can redistribute it and/or
# modify it under the terms of the GNU Lesser General Public
# License as published by the Free Software Foundation; either
# version 2.1 of the License, or (at your option) any later version.
# 
# This library is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# Lesser General Public License for more details.
# 
# You should have received a copy of the GNU Lesser General Public
# License along with this library; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307  USA
#
#
# who       when      what
# --------  --------  ----------------------------------------------
# acaproni  2013-06-28  created
#
import os
import sys
import fnmatch
from datetime import datetime
from subprocess import call

# The package of ExtProd to include in the tar
acsExtProdPackages = (
                      "ant",
                      "maven",
                      "Eclipse",
                      "Eclipse4",
                      "JacORB",
                      "mico",
                      "Python",
                      "TAO",
                      "tcltk",
                      "boost")

def getAcsVersionFromACSROOT():
    '''
    Get the ACS version out of ACSROOT.
    
    Raise a exception in case of error
    '''
    if os.environ.has_key('ACSROOT'):
        acsRoot=os.environ['ACSROOT']
    else:
        raise Exception("ACSROOT not defined!")
    # Assumes ACSROOT to be something like /alma/ACS-12.0/ACSSW
    strs=acsRoot.split("/")
    if len(strs)!=4:
        raise Exception("Malformed ACSROOT: "+acsRoot)
    temp=strs[2].replace(".","_")
    return temp

def getAcsExtProdSrcFolder():
    """
    Get the folder where all the ExtProd packages reside
    
    It is relative to the root i.e. something like alma/ACS-12.0/
    """
    if os.environ.has_key('ACSROOT'):
        acsRoot=os.environ['ACSROOT']
    else:
        raise Exception("ACSROOT not defined!")
    temp=acsRoot.replace("ACSSW", "")
    temp=temp.replace("/", "",1)
    return temp

def removeTempFiles(pattern):
    """
    Remove temporary files generated during the building to save 
    disk space and time generating the tar.
        
    @param pattern: all the files in $ACSROOT
        whose name matches with the pattern will be deleted
    """
    if os.environ.has_key('ACSROOT'):
        acsRoot=os.environ['ACSROOT']
    else:
        raise Exception("ACSROOT not defined!")
    print "Removing useless files from",acsRoot,"..."
    totFiles=totRemoved=0
    for root, dirs, files in os.walk(acsRoot):
        for file in files:
            totFiles=totFiles+1
            if fnmatch.fnmatch(os.path.join(root, file), pattern):
                totRemoved=totRemoved+1
                os.remove(os.path.join(root, file))
                
    print "Checked",totFiles,"and removed",totRemoved
    

def getDate():
    '''
    Return the date to be set in the name of the tar
    '''
    now=datetime.today()
    return "{0}{1:02}{2:02}".format(now.year,now.month,now.day)

def getArchitecture():
    '''
    Return the architecture 32/64 bit
    '''
    un=os.uname()
    return un[4]

def getDistribution():
    '''
    Get the distribution (RHEL, SL and so on)
    by parsing /etc/redhat-release
    
    @return: a string describing the distribution like RH6.5
             or UNKNOWN if /etc/redhat-release does not exist or is unreadable.
             
    '''
    ret="UNKNOWN"
    fname="/etc/redhat-release"
    with open(fname) as f:
        content = f.readlines()
    distr=str(content)
    if distr.find("Red Hat Enterprise Linux")>=0:
        temp="RH"
    elif distr.find("Scientific Linux")>=0:
        temp="SL"
    else:
        return ret
    parts=distr.split(" ")
    version=parts[len(parts)-2]
    return temp+version

if __name__=="__main__":
    acs=getAcsVersionFromACSROOT()
    date=getDate()
    arch=getArchitecture()
    dist=getDistribution()
    currentFolder=os.getcwd()
    srcFolder=getAcsExtProdSrcFolder()
    print "Creating ExtProd tar for", acs,"in",currentFolder
    tarName=acs+"-ExtProd-"+date+"-"+arch+"-"+dist+".tar.gz"
    print "Tar name",tarName
    tarNameFullPath=currentFolder+"/"+tarName
    
    # Go to root /
    os.chdir("/")
    print "Moved into",os.getcwd()
    
    # Remove object files from ACSROOT
    #
    # This is commeted out because we can exclude such files with a param of tar
    #removeTempFiles("*.o")
    
    # build the command to pass to call
  
    cmd=[]
    cmd.append("tar")
    cmd.append("cpzf")
    cmd.append(tarNameFullPath)
    cmd.append("--exclude")
    cmd.append("*.o")
    for pkg in acsExtProdPackages:
        cmd.append(srcFolder+pkg)
    print "Running tar....",
    call(cmd)
    
    # Back to original folder
    os.chdir(currentFolder)
    print "Done"