#

# Copyright (c) 2006-2009 Nokia Corporation and/or its subsidiary(-ies).

# All rights reserved.

# This component and the accompanying materials are made available

# under the terms of the License "Eclipse Public License v1.0"

# which accompanies this distribution, and is available

# at the URL "http://www.eclipse.org/legal/epl-v10.html".

#

# Initial Contributors:

# Nokia Corporation - initial contribution.

#

# Contributors:

#

# Description: 

# raptor linux distribution creation module

# This module allow to crate raptor linux distribution archive (tar.gz) e.g. sbs_linux_dist.tar.gz

# Script extect the following command-line parameters:

# 1. Archive name

# 2. List of files/directories to include

# By default windows specific and source files are excluded.

# example including bin and python subdir:

# sbs_dist.py sbs_linux_dist.tar.gz bin python

#

import os

import re

import tarfile

import sys

import dos2unix

#------------------------------------------------------------------------------

# Create tar.gz archive including given files (fileName list and tarinfo list)

#------------------------------------------------------------------------------

def createTarGZ(tarName, fileList):

    tar = tarfile.open(tarName, "w|gz")

    for name in fileList:

        tar.add(name)

    return tar

#------------------------------------------------------------------------------

# Lists files in each of given directories

#------------------------------------------------------------------------------

def listFilesInDirs(paths):

    fileList = []

    for path in paths:

        fileList.extend(listFiles(path))

    return fileList

#------------------------------------------------------------------------------

# Lists files in given directory

#------------------------------------------------------------------------------

def listFiles(path):

    fileList = []

    for root, dirs, files in os.walk(path):

        for index in range(len(files)):

            fileList.append(root + "/" + files[index])

    return fileList

#------------------------------------------------------------------------------

# Excludes files matching "pattern" from given files list

#------------------------------------------------------------------------------

def excludeFiles(fileList, pattern):

    filteredFileList = []

    regExp = re.compile(pattern)

    for fileName in fileList:

        if not(regExp.match(fileName)):

           filteredFileList.append(fileName)

    return filteredFileList

#------------------------------------------------------------------------------

# Groups given paths as files or directories

#------------------------------------------------------------------------------

def groupFilesAndDirs(filesAndDirs):

    files = []

    dirs = []

    for name in filesAndDirs:

        if os.path.isdir(name):

             dirs.append(name)

        else:

             if os.path.isfile(name):

                files.append(name)

             else:

                  print "Warning: Neither a file nor a directory! Ignoring parameter - " + name

    return (files,dirs)

#------------------------------------------------------------------------------

# Prepares regular expression to exclude unnecessary files

#------------------------------------------------------------------------------

def prepareExcludedFilesRegExp():

    pathPrefixRegExp = ".*[\\\/]"

    filesRegExp = "((sbs)|(.*\.bat)|(.*\.pyc)|(.*\.cmd)|(.*\.exe)|(.*\.dll)|(sbs_dist.py)"

    filesRegExp = filesRegExp + "|(dos2unix.py)|(raptor_py2exe_setup.py)|(make)|(bash)|(bashbug))+"

    return "^" + pathPrefixRegExp + filesRegExp + "$"

#------------------------------------------------------------------------------

# Includes all files in fileList in given tar with altered executable permision (+X) for all

#------------------------------------------------------------------------------

def includeAsExecutable(tar, fileList):

    for f in fileList:

        tarinfo = tar.gettarinfo(f)

        # OR with 73 (001 001 001) - +X for all

        tarinfo.mode = tarinfo.mode | 73

        tar.addfile(tarinfo,file(f, "rb"))

#------------------------------------------------------------------------------

# Validate script parameters

#------------------------------------------------------------------------------

def validateParameters(tarFileName, filesToInclude):

    if not(len(tarFileName) > 0):

       print "Error: No archive name given."

       sys.exit()

    if not(len(filesToInclude) > 0):

       print "Error: No files/directories names to include in archive given."

       sys.exit()

tarFileName = sys.argv[1]

# files and directories

filesAndDirsToInclude = sys.argv[2:]

validateParameters(tarFileName, filesAndDirsToInclude)

(filesToInclude,dirsToInclude) = groupFilesAndDirs(filesAndDirsToInclude)

fileList = listFilesInDirs(dirsToInclude)

fileList.extend(filesToInclude)

filteredFileList = excludeFiles(fileList, prepareExcludedFilesRegExp())

dos2unix.dos2unix("bin/sbs")

tar = createTarGZ(tarFileName, filteredFileList)

fileToBeExecutableList = ["bin/sbs", "linux-i386/bin/make", "linux-i386/bin/bash", "linux-i386/bin/bashbug",

			  "bin/sbs_descramble"]

includeAsExecutable(tar, fileToBeExecutableList)

tar.close()