#!/usr/bin/env python

# Copyright (c) 2012 Wind River Systems, Inc.
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License version 2 as
# published by the Free Software Foundation.
#
# This program 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 General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA

import os
import sys
import optparse
import re
import commands
import shutil

versions = {}
obsolete_dirs = []
parser = None

def err_quit(msg):
    print msg
    parser.print_usage()
    sys.exit(1)

def parse_version(verstr):
    elems = verstr.split(':')
    epoch = elems[0]
    if len(epoch) == 0:
        return elems[1]
    else:
        return epoch + '_' + elems[1]

def parse_dir(match, pkgabsdir):
    pkg_name = match.group(1)
    pkg_version = match.group(2)
    if pkg_name in versions:
        if pkg_version != versions[pkg_name]:
            obsolete_dirs.append(pkgabsdir)
        return True
    return False

def main():
    global parser
    parser = optparse.OptionParser(
        usage = """%prog

Remove the obsolete packages' build directories in WORKDIR.
This script must be ran under BUILDDIR after source file \"oe-init-build-env\".""")

    options, args = parser.parse_args(sys.argv)

    builddir = commands.getoutput('echo $BUILDDIR')
    if len(builddir) == 0:
        err_quit("Please source file \"oe-init-build-env\" first.\n")

    if os.getcwd() != builddir:
        err_quit("Please run %s under: %s\n" % (os.path.basename(args[0]), builddir))

    print 'Updating bitbake caches...'
    cmd = "bitbake -s"
    (ret, output) = commands.getstatusoutput(cmd)
    if ret != 0:
        print "Execute 'bitbake -s' failed. Can't get packages' versions."
        return 1

    output = output.split('\n')
    index = 0
    while len(output[index]) > 0:
        index += 1
    alllines = output[index+1:]

    for line in alllines:
        # empty again means end of the versions output
        if len(line) == 0:
            break
        line = line.strip()
        line = re.sub('\s+', ' ', line)
        elems = line.split(' ')
        if len(elems) == 2:
            version = parse_version(elems[1])
        else:
            version = parse_version(elems[2])
        versions[elems[0]] = version

    cmd = "bitbake -e | grep ^TMPDIR"
    (ret, output) = commands.getstatusoutput(cmd)
    if ret != 0:
        print "Execute 'bitbke -e' failed. Can't get TMPDIR."
        return 1

    tmpdir = output.split('"')[1]
    workdir = os.path.join(tmpdir, 'work')
    if not os.path.exists(workdir):
        print "WORKDIR %s does NOT exist. Quit." % workdir
        return 1

    for archdir in os.listdir(workdir):
        archdir = os.path.join(workdir, archdir)
        if not os.path.isdir(archdir):
            pass

        for pkgdir in sorted(os.listdir(archdir)):
            pkgabsdir = os.path.join(archdir, pkgdir)
            if not os.path.isdir(pkgabsdir):
                pass

            # parse the package directory names
            # parse native/nativesdk packages first
            match = re.match('(.*?-native.*?)-(.*)', pkgdir)
            if match and parse_dir(match, pkgabsdir):
                continue

            # parse package names which ends with numbers such as 'glib-2.0'
            match = re.match('(.*?-[\.\d]+)-(\d.*)', pkgdir)
            if match and parse_dir(match, pkgabsdir):
                continue

            # other packages
            match = re.match('(.*?)-(\d.*)', pkgdir)
            if match and parse_dir(match, pkgabsdir):
                continue

    for d in obsolete_dirs:
        print "Deleleting %s" % d
        shutil.rmtree(d, True)

    if len(obsolete_dirs):
        print '\nTotal %d items.' % len(obsolete_dirs)
    else:
        print '\nNo obsolete directory found under %s.' % workdir

    return 0

if __name__ == '__main__':
    try:
        ret = main()
    except Exception:
        ret = 2
        import traceback
        traceback.print_exc(3)
    sys.exit(ret)