#!/usr/bin/env python
# ex:ts=4:sw=4:sts=4:et
# -*- tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*-
#
# Copyright (C) 2003, 2004  Chris Larson
# Copyright (C) 2003, 2004  Phil Blundell
# Copyright (C) 2003 - 2005 Michael 'Mickey' Lauer
# Copyright (C) 2005        Holger Hans Peter Freyther
# Copyright (C) 2005        ROAD GmbH
# Copyright (C) 2006        Richard Purdie
#
# 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.,
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.

import sys, os, getopt, glob, copy, os.path, re, time
sys.path.insert(0,os.path.join(os.path.dirname(os.path.dirname(sys.argv[0])), 'lib'))
import bb
from bb import utils, data, parse, event, cache, providers
from sets import Set
import itertools, optparse

parsespin = itertools.cycle( r'|/-\\' )

__version__ = "1.6.6"

#============================================================================#
# BBParsingStatus
#============================================================================#
class BBParsingStatus:
    """
    The initial idea for this status class is to use the data when it is
    already loaded instead of loading it from various place over and over
    again.
    """

    def __init__(self):
        self.providers   = {}
        self.rproviders = {}
        self.packages = {}
        self.packages_dynamic = {}
        self.bbfile_priority = {}
        self.bbfile_config_priorities = []
        self.ignored_dependencies = None
        self.possible_world = []
        self.world_target = Set()
        self.pkg_pn = {}
        self.pkg_fn = {}
        self.pkg_pvpr = {}
        self.pkg_dp = {}
        self.pn_provides = {}
        self.all_depends = Set()
        self.build_all = {}
        self.deps = {}
        self.rundeps = {}
        self.runrecs = {}
        self.task_queues = {}
        self.task_deps = {}
        self.stamp = {}
        self.preferred = {}

    def handle_bb_data(self, file_name, bb_cache, cached):
        """
        We will fill the dictionaries with the stuff we
        need for building the tree more fast
        """

        pn       = bb_cache.getVar('PN', file_name, True)
        pv       = bb_cache.getVar('PV', file_name, True)
        pr       = bb_cache.getVar('PR', file_name, True)
        dp       = int(bb_cache.getVar('DEFAULT_PREFERENCE', file_name, True) or "0")
        provides  = Set([pn] + (bb_cache.getVar("PROVIDES", file_name, True) or "").split())
        depends   =  bb.utils.explode_deps(bb_cache.getVar("DEPENDS", file_name, True) or "")
        packages  = (bb_cache.getVar('PACKAGES', file_name, True) or "").split()
        packages_dynamic = (bb_cache.getVar('PACKAGES_DYNAMIC', file_name, True) or "").split()
        rprovides = (bb_cache.getVar("RPROVIDES", file_name, True) or "").split()

        self.task_queues[file_name] = bb_cache.getVar("_task_graph", file_name, True)
        self.task_deps[file_name] = bb_cache.getVar("_task_deps", file_name, True)

        # build PackageName to FileName lookup table
        if pn not in self.pkg_pn:
            self.pkg_pn[pn] = []
        self.pkg_pn[pn].append(file_name)

        self.build_all[file_name] = int(bb_cache.getVar('BUILD_ALL_DEPS', file_name, True) or "0")
        self.stamp[file_name] = bb_cache.getVar('STAMP', file_name, True)

        # build FileName to PackageName lookup table
        self.pkg_fn[file_name] = pn
        self.pkg_pvpr[file_name] = (pv,pr)
        self.pkg_dp[file_name] = dp

        # Build forward and reverse provider hashes
        # Forward: virtual -> [filenames]
        # Reverse: PN -> [virtuals]
        if pn not in self.pn_provides:
            self.pn_provides[pn] = Set()
        self.pn_provides[pn] |= provides

        for provide in provides:
            if provide not in self.providers:
                self.providers[provide] = []
            self.providers[provide].append(file_name)

        self.deps[file_name] = Set()
        for dep in depends:
            self.all_depends.add(dep)
            self.deps[file_name].add(dep)

        # Build reverse hash for PACKAGES, so runtime dependencies 
        # can be be resolved (RDEPENDS, RRECOMMENDS etc.)
        for package in packages:
            if not package in self.packages:
                self.packages[package] = []
            self.packages[package].append(file_name)
            rprovides += (bb_cache.getVar("RPROVIDES_%s" % package, file_name, 1) or "").split() 

        for package in packages_dynamic:
            if not package in self.packages_dynamic:
                self.packages_dynamic[package] = []
            self.packages_dynamic[package].append(file_name)

        for rprovide in rprovides:
            if not rprovide in self.rproviders:
                self.rproviders[rprovide] = []
            self.rproviders[rprovide].append(file_name)

        # Build hash of runtime depends and rececommends

        def add_dep(deplist, deps):
            for dep in deps:
                if not dep in deplist:
                    deplist[dep] = ""

        if not file_name in self.rundeps:
            self.rundeps[file_name] = {}
        if not file_name in self.runrecs:
            self.runrecs[file_name] = {}

        for package in packages + [pn]:
            if not package in self.rundeps[file_name]:
                self.rundeps[file_name][package] = {}
            if not package in self.runrecs[file_name]:
                self.runrecs[file_name][package] = {}

            add_dep(self.rundeps[file_name][package], bb.utils.explode_deps(bb_cache.getVar('RDEPENDS', file_name, True) or ""))
            add_dep(self.runrecs[file_name][package], bb.utils.explode_deps(bb_cache.getVar('RRECOMMENDS', file_name, True) or ""))
            add_dep(self.rundeps[file_name][package], bb.utils.explode_deps(bb_cache.getVar("RDEPENDS_%s" % package, file_name, True) or ""))
            add_dep(self.runrecs[file_name][package], bb.utils.explode_deps(bb_cache.getVar("RRECOMMENDS_%s" % package, file_name, True) or ""))

        # Collect files we may need for possible world-dep
        # calculations
        if not bb_cache.getVar('BROKEN', file_name, True) and not bb_cache.getVar('EXCLUDE_FROM_WORLD', file_name, True):
            self.possible_world.append(file_name)


#============================================================================#
# BBStatistics
#============================================================================#
class BBStatistics:
    """
    Manage build statistics for one run
    """
    def __init__(self ):
        self.attempt = 0
        self.success = 0
        self.fail = 0
        self.deps = 0

    def show( self ):
        print "Build statistics:"
        print "  Attempted builds: %d" % self.attempt
        if self.fail:
            print "  Failed builds: %d" % self.fail
        if self.deps:
            print "  Dependencies not satisfied: %d" % self.deps
        if self.fail or self.deps: return 1
        else: return 0


#============================================================================#
# BBOptions
#============================================================================#
class BBConfiguration( object ):
    """
    Manages build options and configurations for one run
    """
    def __init__( self, options ):
        for key, val in options.__dict__.items():
            setattr( self, key, val )

#============================================================================#
# BBCooker
#============================================================================#
class BBCooker:
    """
    Manages one bitbake build run
    """

    ParsingStatus = BBParsingStatus     # make it visible from the shell
    Statistics = BBStatistics           # make it visible from the shell

    def __init__( self ):
        self.build_cache_fail = []
        self.build_cache = []
        self.rbuild_cache = []
        self.building_list = []
        self.build_path = []
        self.consider_msgs_cache = []
        self.stats = BBStatistics()
        self.status = None

        self.cache = None
        self.bb_cache = None

    def tryBuildPackage( self, fn, item, the_data ):
        """Build one package"""
        bb.event.fire(bb.event.PkgStarted(item, the_data))
        try:
            self.stats.attempt += 1
            if self.configuration.force:
                bb.data.setVarFlag('do_%s' % self.configuration.cmd, 'force', 1, the_data)
            if not self.configuration.dry_run:
                bb.build.exec_task('do_%s' % self.configuration.cmd, the_data)
            bb.event.fire(bb.event.PkgSucceeded(item, the_data))
            self.build_cache.append(fn)
            return True
        except bb.build.FuncFailed:
            self.stats.fail += 1
            bb.msg.error(bb.msg.domain.Build, "task stack execution failed")
            bb.event.fire(bb.event.PkgFailed(item, the_data))
            self.build_cache_fail.append(fn)
            raise
        except bb.build.EventException, e:
            self.stats.fail += 1
            event = e.args[1]
            bb.msg.error(bb.msg.domain.Build, "%s event exception, aborting" % bb.event.getName(event))
            bb.event.fire(bb.event.PkgFailed(item, the_data))
            self.build_cache_fail.append(fn)
            raise

    def tryBuild( self, fn, virtual , buildAllDeps , build_depends = []):
        """
        Build a provider and its dependencies. 
        build_depends is a list of previous build dependencies (not runtime)
        If build_depends is empty, we're dealing with a runtime depends
        """

        the_data = self.bb_cache.loadDataFull(fn, self)

        # Only follow all (runtime) dependencies if doing a build
        if not buildAllDeps and self.configuration.cmd is "build":
            buildAllDeps = self.status.build_all[fn]

        # Error on build time dependency loops
        if build_depends and build_depends.count(fn) > 1:
            bb.msg.error(bb.msg.domain.Depends, "%s depends on itself (eventually)" % fn)
            bb.msg.error(bb.msg.domain.Depends, "upwards chain is: %s" % (" -> ".join(self.build_path)))
            return False

        # See if this is a runtime dependency we've already built
        # Or a build dependency being handled in a different build chain
        if fn in self.building_list:
            return self.addRunDeps(fn, virtual , buildAllDeps)

        item = self.status.pkg_fn[fn]

        self.building_list.append(fn)

        pathstr = "%s (%s)" % (item, virtual)
        self.build_path.append(pathstr)

        depends_list = bb.utils.explode_deps(bb.data.getVar('DEPENDS', the_data, True) or "")

        bb.msg.note(2, bb.msg.domain.Depends, "current path: %s" % (" -> ".join(self.build_path)))
        bb.msg.note(2, bb.msg.domain.Depends, "dependencies for %s are: %s" % (item, " ".join(depends_list)))

        try:
            failed = False

            depcmd = self.configuration.cmd
            bbdepcmd = bb.data.getVarFlag('do_%s' % self.configuration.cmd, 'bbdepcmd', the_data)
            if bbdepcmd is not None:
                if bbdepcmd == "":
                    depcmd = None
                else:
                    depcmd = bbdepcmd

            if depcmd:
                oldcmd = self.configuration.cmd
                self.configuration.cmd = depcmd

            for dependency in depends_list:
                if dependency in self.status.ignored_dependencies:
                    continue
                if not depcmd:
                    continue
                if self.buildProvider( dependency , buildAllDeps , build_depends ) == 0:
                    bb.msg.error(bb.msg.domain.Depends, "dependency %s (for %s) not satisfied" % (dependency,item))
                    failed = True
                    if self.configuration.abort:
                        break

            if depcmd:
                self.configuration.cmd = oldcmd

            if failed:
                self.stats.deps += 1
                return False

            if not self.addRunDeps(fn, virtual , buildAllDeps):
                return False

            if bb.build.stamp_is_current('do_%s' % self.configuration.cmd, the_data):
                self.build_cache.append(fn)
                return True

            return self.tryBuildPackage( fn, item, the_data )

        finally:
            self.building_list.remove(fn)
            self.build_path.remove(pathstr)


    def showVersions( self ):
        pkg_pn = self.status.pkg_pn
        preferred_versions = {}
        latest_versions = {}

        # Sort by priority
        for pn in pkg_pn.keys():
            (last_ver,last_file,pref_ver,pref_file) = bb.providers.findBestProvider(pn, self.configuration.data, self.status)
            preferred_versions[pn] = (pref_ver, pref_file)
            latest_versions[pn] = (last_ver, last_file)

        pkg_list = pkg_pn.keys()
        pkg_list.sort()

        for p in pkg_list:
            pref = preferred_versions[p]
            latest = latest_versions[p]

            if pref != latest:
                prefstr = pref[0][0] + "-" + pref[0][1]
            else:
                prefstr = ""

            print "%-30s %20s %20s" % (p, latest[0][0] + "-" + latest[0][1],
                                        prefstr)


    def showEnvironment( self ):
        """Show the outer or per-package environment"""
        if self.configuration.buildfile:
            self.cb = None
            self.bb_cache = bb.cache.init(self)
            try:
                self.configuration.data = self.bb_cache.loadDataFull(self.configuration.buildfile, self)
            except IOError, e:
                bb.msg.fatal(bb.msg.domain.Parsing, "Unable to read %s: %s" % ( self.configuration.buildfile, e ))
            except Exception, e:
                bb.msg.fatal(bb.msg.domain.Parsing, "%s" % e)
        # emit variables and shell functions
        try:
            data.update_data( self.configuration.data )
            data.emit_env(sys.__stdout__, self.configuration.data, True)
        except Exception, e:
            bb.msg.fatal(bb.msg.domain.Parsing, "%s" % e)
        # emit the metadata which isnt valid shell
        for e in self.configuration.data.keys():
            if data.getVarFlag( e, 'python', self.configuration.data ):
                sys.__stdout__.write("\npython %s () {\n%s}\n" % (e, data.getVar(e, self.configuration.data, 1)))

    def generateDotGraph( self, pkgs_to_build, ignore_deps ):
        """
        Generate two graphs one for the DEPENDS and RDEPENDS. The current
        implementation creates crappy graphs ;)

        pkgs_to_build A list of packages that needs to be built
        ignore_deps   A list of names where processing of dependencies
                      should be stopped. e.g. dependencies that get
        """

        def myFilterProvider( providers, item):
            """
            Take a list of providers and filter according to environment
            variables. In contrast to filterProviders we do not discriminate
            and take PREFERRED_PROVIDER into account.
            """
            eligible = []
            preferred_versions = {}

            # Collate providers by PN
            pkg_pn = {}
            for p in providers:
                pn = self.status.pkg_fn[p]
                if pn not in pkg_pn:
                    pkg_pn[pn] = []
                pkg_pn[pn].append(p)

            bb.msg.debug(1, bb.msg.domain.Provider, "providers for %s are: %s" % (item, pkg_pn.keys()))

            for pn in pkg_pn.keys():
                preferred_versions[pn] = bb.providers.findBestProvider(pn, self.configuration.data, self.status, pkg_pn)[2:4]
                eligible.append(preferred_versions[pn][1])

            for p in eligible:
                if p in self.build_cache_fail:
                    bb.msg.debug(1, bb.msg.domain.Provider, "rejecting already-failed %s" % p)
                    eligible.remove(p)

            if len(eligible) == 0:
                bb.msg.error(bb.msg.domain.Provider, "no eligible providers for %s" % item)
                return 0

            prefervar = bb.data.getVar('PREFERRED_PROVIDER_%s' % item, self.configuration.data, 1)

            # try the preferred provider first
            if prefervar:
                for p in eligible:
                    if prefervar == self.status.pkg_fn[p]:
                        bb.msg.note(1, bb.msg.domain.Provider, "Selecting PREFERRED_PROVIDER %s" % prefervar)
                        eligible.remove(p)
                        eligible = [p] + eligible

            return eligible




        # try to avoid adding the same rdepends over an over again
        seen_depends  = []
        seen_rdepends = []


        def add_depends(package_list):
            """
            Add all depends of all packages from this list
            """
            for package in package_list:
                if package in seen_depends or package in ignore_deps:
                    continue

                seen_depends.append( package )
                if not package in self.status.providers:
                    """
                    We have not seen this name -> error in
                    dependency handling
                    """
                    bb.note( "ERROR with provider: %(package)s" % vars() )
                    print >> depends_file, '"%(package)s" -> ERROR' % vars()
                    continue

                # get all providers for this package
                providers = self.status.providers[package]

                # now let us find the bestProvider for it
                fn = myFilterProvider(providers, package)[0]

                depends  = bb.utils.explode_deps(self.bb_cache.getVar('DEPENDS', fn, True) or "")
                version  = self.bb_cache.getVar('PV', fn, True ) + '-' + self.bb_cache.getVar('PR', fn, True)
                add_depends ( depends )

                # now create the node
                print >> depends_file, '"%(package)s" [label="%(package)s\\n%(version)s"]' % vars()

                depends = filter( (lambda x: x not in ignore_deps), depends )
                for depend in depends:
                    print >> depends_file, '"%(package)s" -> "%(depend)s"' % vars()


        def add_all_depends( the_depends, the_rdepends ):
            """
            Add both DEPENDS and RDEPENDS. RDEPENDS will get dashed
            lines
            """
            package_list = the_depends + the_rdepends
            for package in package_list:
                if package in seen_rdepends or package in ignore_deps:
                    continue

                seen_rdepends.append( package )

                # Let us find out if the package is a DEPENDS or RDEPENDS
                # and we will set 'providers' with the avilable providers
                # for the package.
                if package in the_depends:
                    if not package in self.status.providers:
                        bb.note( "ERROR with provider: %(package)s" % vars() )
                        print >> alldepends_file, '"%(package)s" -> ERROR' % vars()
                        continue

                    providers = self.status.providers[package]
                elif package in the_rdepends:
                    if len(bb.providers.getRuntimeProviders(self.status, package)) == 0:
                        bb.note( "ERROR with rprovider: %(package)s" % vars() )
                        print >> alldepends_file, '"%(package)s" -> ERROR [style="dashed"]' % vars()
                        continue

                    providers = bb.providers.getRuntimeProviders(self.status, package)
                else:
                    # something went wrong...
                    print "Complete ERROR! %s" % package
                    continue

                # now let us find the bestProvider for it
                fn = myFilterProvider(providers, package)[0]

                # Now we have a filename let us get the depends and RDEPENDS of it
                depends  = bb.utils.explode_deps(self.bb_cache.getVar('DEPENDS', fn, True) or "")
                if fn in self.status.rundeps and package in self.status.rundeps[fn]:
                    rdepends= self.status.rundeps[fn][package].keys()
                else:
                    rdepends = []
                version  = self.bb_cache.getVar('PV', fn, True ) + '-' + self.bb_cache.getVar('PR', fn, True)

                # handle all the depends and rdepends of package
                add_all_depends ( depends, rdepends )

                # now create the node using package name
                print >> alldepends_file, '"%(package)s" [label="%(package)s\\n%(version)s"]' % vars()

                # remove the stuff we want to ignore and add the edges
                depends = filter( (lambda x: x not in ignore_deps), depends )
                rdepends = filter( (lambda x: x not in ignore_deps), rdepends )
                for depend in depends:
                    print >> alldepends_file, '"%(package)s" -> "%(depend)s"' % vars()
                for depend in rdepends:
                    print >> alldepends_file, '"%(package)s" -> "%(depend)s" [style=dashed]' % vars()


        # Add depends now
        depends_file = file('depends.dot', 'w' )
        print >> depends_file, "digraph depends {"
        add_depends( pkgs_to_build )
        print >> depends_file,  "}"

        # Add all depends now
        alldepends_file = file('alldepends.dot', 'w' )
        print >> alldepends_file, "digraph alldepends {"
        add_all_depends( pkgs_to_build, [] )
        print >> alldepends_file, "}"

    def buildProvider( self, item , buildAllDeps , build_depends = [] ):
        """
        Build something to provide a named build requirement
        (takes item names from DEPENDS namespace)
        """

        fn = None
        discriminated = False

        if not item in self.status.providers:
            bb.msg.error(bb.msg.domain.Depends, "Nothing provides dependency %s" % item)
            bb.event.fire(bb.event.NoProvider(item,self.configuration.data))
            return 0

        all_p = self.status.providers[item]

        for p in all_p:
            if p in self.build_cache:
                bb.msg.debug(1, bb.msg.domain.Provider, "already built %s in this run" % p)
                return 1

        eligible = bb.providers.filterProviders(all_p, item, self.configuration.data, self.status, self.build_cache_fail)

        if not eligible:
            return 0

        prefervar = bb.data.getVar('PREFERRED_PROVIDER_%s' % item, self.configuration.data, 1)
        if prefervar:
            self.status.preferred[item] = prefervar

        if item in self.status.preferred:
            for p in eligible:
                pn = self.status.pkg_fn[p]
                if self.status.preferred[item] == pn:
                    bb.msg.note(2, bb.msg.domain.Provider, "selecting %s to satisfy %s due to PREFERRED_PROVIDERS" % (pn, item))
                    eligible.remove(p)
                    eligible = [p] + eligible
                    discriminated = True
                    break

        if len(eligible) > 1 and discriminated == False:
            if item not in self.consider_msgs_cache:
                providers_list = []
                for fn in eligible:
                    providers_list.append(self.status.pkg_fn[fn])
                bb.msg.note(1, bb.msg.domain.Provider, "multiple providers are available (%s);" % ", ".join(providers_list))
                bb.msg.note(1, bb.msg.domain.Provider, "consider defining PREFERRED_PROVIDER_%s" % item)
                bb.event.fire(bb.event.MultipleProviders(item,providers_list,self.configuration.data))
            self.consider_msgs_cache.append(item)


        # run through the list until we find one that we can build
        for fn in eligible:
            bb.msg.debug(2, bb.msg.domain.Provider, "selecting %s to satisfy %s" % (fn, item))
            if self.tryBuild(fn, item, buildAllDeps, build_depends + [fn]):
                return 1

        bb.msg.note(1, bb.msg.domain.Provider, "no buildable providers for %s" % item)
        bb.event.fire(bb.event.NoProvider(item,self.configuration.data))
        return 0

    def buildRProvider( self, item , buildAllDeps ):
        """
        Build something to provide a named runtime requirement
        (takes item names from RDEPENDS/PACKAGES namespace)
        """

        fn = None
        all_p = []
        discriminated = False

        if not buildAllDeps:
            return True

        all_p = bb.providers.getRuntimeProviders(self.status, item)

        if not all_p:
            bb.msg.error(bb.msg.domain.Provider, "Nothing provides runtime dependency %s" % (item))
            bb.event.fire(bb.event.NoProvider(item,self.configuration.data,runtime=True))
            return False

        for p in all_p:
            if p in self.rbuild_cache:
                bb.msg.debug(2, bb.msg.domain.Provider, "Already built %s providing runtime %s" % (p,item))
                return True
            if p in self.build_cache:
                bb.msg.debug(2, bb.msg.domain.Provider, "Already built %s but adding any further RDEPENDS for %s" % (p, item))
                return self.addRunDeps(p, item , buildAllDeps)

        eligible = bb.providers.filterProviders(all_p, item, self.configuration.data, self.status, self.build_cache_fail)
        if not eligible:
            return 0

        preferred = []
        for p in eligible:
            pn = self.status.pkg_fn[p]
            provides = self.status.pn_provides[pn]
            for provide in provides:
                prefervar = bb.data.getVar('PREFERRED_PROVIDER_%s' % provide, self.configuration.data, 1)
                if prefervar == pn:
                    bb.msg.note(2, bb.msg.domain.Provider, "selecting %s to satisfy runtime %s due to PREFERRED_PROVIDERS" % (pn, item))
                    eligible.remove(p)
                    eligible = [p] + eligible
                    preferred.append(p)

        if len(eligible) > 1 and len(preferred) == 0:
            if item not in self.consider_msgs_cache:
                providers_list = []
                for fn in eligible:
                    providers_list.append(self.status.pkg_fn[fn])
                bb.msg.note(1, bb.msg.domain.Provider, "multiple providers are available (%s);" % ", ".join(providers_list))
                bb.msg.note(1, bb.msg.domain.Provider, "consider defining a PREFERRED_PROVIDER to match runtime %s" % item)
                bb.event.fire(bb.event.MultipleProviders(item,providers_list,self.configuration.data,runtime=True))
            self.consider_msgs_cache.append(item)

        if len(preferred) > 1:
            if item not in self.consider_msgs_cache:
                providers_list = []
                for fn in preferred:
                    providers_list.append(self.status.pkg_fn[fn])
                bb.msg.note(1, bb.msg.domain.Provider, "multiple preferred providers are available (%s);" % ", ".join(providers_list))
                bb.msg.note(1, bb.msg.domain.Provider, "consider defining only one PREFERRED_PROVIDER to match runtime %s" % item)
                bb.event.fire(bb.event.MultipleProviders(item,providers_list,self.configuration.data,runtime=True))
            self.consider_msgs_cache.append(item)

        # run through the list until we find one that we can build
        for fn in eligible:
            bb.msg.debug(2, bb.msg.domain.Provider, "selecting %s to satisfy runtime %s" % (fn, item))
            if self.tryBuild(fn, item, buildAllDeps):
                return True

        bb.msg.error(bb.msg.domain.Provider, "No buildable providers for runtime %s" % item)
        bb.event.fire(bb.event.NoProvider(item,self.configuration.data))
        return False

    def addRunDeps(self , fn, item , buildAllDeps):
        """
        Add any runtime dependencies of runtime item provided by fn 
        as long as item has't previously been processed by this function.
        """

        if item in self.rbuild_cache:
            return True

        if not buildAllDeps:
            return True

        rdepends = []
        self.rbuild_cache.append(item)

        if fn in self.status.rundeps and item in self.status.rundeps[fn]:
            rdepends += self.status.rundeps[fn][item].keys()
        if fn in self.status.runrecs and item in self.status.runrecs[fn]:
            rdepends += self.status.runrecs[fn][item].keys()

        bb.msg.debug(2, bb.msg.domain.Provider, "Additional runtime dependencies for %s are: %s" % (item, " ".join(rdepends)))

        for rdepend in rdepends:
            if rdepend in self.status.ignored_dependencies:
                continue
            if not self.buildRProvider(rdepend, buildAllDeps):
                return False
        return True

    def buildDepgraph( self ):
        all_depends = self.status.all_depends
        pn_provides = self.status.pn_provides

        localdata = data.createCopy(self.configuration.data)
        bb.data.update_data(localdata)

        def calc_bbfile_priority(filename):
            for (regex, pri) in self.status.bbfile_config_priorities:
                if regex.match(filename):
                    return pri
            return 0

        # Handle PREFERRED_PROVIDERS
        for p in (bb.data.getVar('PREFERRED_PROVIDERS', localdata, 1) or "").split():
            (providee, provider) = p.split(':')
            if providee in self.status.preferred and self.status.preferred[providee] != provider:
                bb.msg.error(bb.msg.domain.Provider, "conflicting preferences for %s: both %s and %s specified" % (providee, provider, self.status.preferred[providee]))
            self.status.preferred[providee] = provider

        # Calculate priorities for each file
        for p in self.status.pkg_fn.keys():
            self.status.bbfile_priority[p] = calc_bbfile_priority(p)

    def buildWorldTargetList(self):
        """
         Build package list for "bitbake world"
        """
        all_depends = self.status.all_depends
        pn_provides = self.status.pn_provides
        bb.msg.debug(1, bb.msg.domain.Parsing, "collating packages for \"world\"")
        for f in self.status.possible_world:
            terminal = True
            pn = self.status.pkg_fn[f]

            for p in pn_provides[pn]:
                if p.startswith('virtual/'):
                    bb.msg.debug(2, bb.msg.domain.Parsing, "World build skipping %s due to %s provider starting with virtual/" % (f, p))
                    terminal = False
                    break
                for pf in self.status.providers[p]:
                    if self.status.pkg_fn[pf] != pn:
                        bb.msg.debug(2, bb.msg.domain.Parsing, "World build skipping %s due to both us and %s providing %s" % (f, pf, p))
                        terminal = False
                        break
            if terminal:
                self.status.world_target.add(pn)

            # drop reference count now
            self.status.possible_world = None
            self.status.all_depends    = None

    def myProgressCallback( self, x, y, f, bb_cache, from_cache ):
        # feed the status with new input

        self.status.handle_bb_data(f, bb_cache, from_cache)

        if os.isatty(sys.stdout.fileno()):
            sys.stdout.write("\rNOTE: Handling BitBake files: %s (%04d/%04d) [%2d %%]" % ( parsespin.next(), x, y, x*100/y ) )
            sys.stdout.flush()
        else:
            if x == 1:
                sys.stdout.write("Parsing .bb files, please wait...")
                sys.stdout.flush()
            if x == y:
                sys.stdout.write("done.")
                sys.stdout.flush()

    def interactiveMode( self ):
        """Drop off into a shell"""
        try:
            from bb import shell
        except ImportError, details:
            bb.msg.fatal(bb.msg.domain.Parsing, "Sorry, shell not available (%s)" % details )
        else:
            bb.data.update_data( self.configuration.data )
            shell.start( self )
            sys.exit( 0 )

    def parseConfigurationFile( self, afile ):
        try:
            self.configuration.data = bb.parse.handle( afile, self.configuration.data )

            # Add the handlers we inherited by INHERIT
            # we need to do this manually as it is not guranteed
            # we will pick up these classes... as we only INHERIT
            # on .inc and .bb files but not on .conf
            data = bb.data.createCopy( self.configuration.data )
            inherits  = ["base"] + (bb.data.getVar('INHERIT', data, True ) or "").split()
            for inherit in inherits:
                data = bb.parse.handle( os.path.join('classes', '%s.bbclass' % inherit ), data, True )

            # FIXME: This assumes that we included at least one .inc file
            for var in bb.data.keys(data):
                if bb.data.getVarFlag(var, 'handler', data):
                    bb.event.register(var,bb.data.getVar(var, data))

        except IOError:
            bb.msg.fatal(bb.msg.domain.Parsing, "Unable to open %s" % afile )
        except bb.parse.ParseError, details:
            bb.msg.fatal(bb.msg.domain.Parsing, "Unable to parse %s (%s)" % (afile, details) )

    def handleCollections( self, collections ):
        """Handle collections"""
        if collections:
            collection_list = collections.split()
            for c in collection_list:
                regex = bb.data.getVar("BBFILE_PATTERN_%s" % c, self.configuration.data, 1)
                if regex == None:
                    bb.msg.error(bb.msg.domain.Parsing, "BBFILE_PATTERN_%s not defined" % c)
                    continue
                priority = bb.data.getVar("BBFILE_PRIORITY_%s" % c, self.configuration.data, 1)
                if priority == None:
                    bb.msg.error(bb.msg.domain.Parsing, "BBFILE_PRIORITY_%s not defined" % c)
                    continue
                try:
                    cre = re.compile(regex)
                except re.error:
                    bb.msg.error(bb.msg.domain.Parsing, "BBFILE_PATTERN_%s \"%s\" is not a valid regular expression" % (c, regex))
                    continue
                try:
                    pri = int(priority)
                    self.status.bbfile_config_priorities.append((cre, pri))
                except ValueError:
                    bb.msg.error(bb.msg.domain.Parsing, "invalid value for BBFILE_PRIORITY_%s: \"%s\"" % (c, priority))


    def cook( self, configuration, args ):
        """
        We are building stuff here. We do the building
        from here. By default we try to execute task
        build.
        """

        self.configuration = configuration

        if self.configuration.verbose:
            bb.msg.set_verbose(True)

        if not self.configuration.cmd:
            self.configuration.cmd = "build"

        if self.configuration.debug:
            bb.msg.set_debug_level(self.configuration.debug)

        self.configuration.data = bb.data.init()

        for f in self.configuration.file:
            self.parseConfigurationFile( f )

        self.parseConfigurationFile( os.path.join( "conf", "bitbake.conf" ) )


        #
        # Special updated configuration we use for firing events
        #
        self.configuration.event_data = bb.data.createCopy(self.configuration.data)
        bb.data.update_data(self.configuration.event_data)

        if self.configuration.show_environment:
            self.showEnvironment()
            sys.exit( 0 )

        # inject custom variables
        if not bb.data.getVar("BUILDNAME", self.configuration.data):
            bb.data.setVar("BUILDNAME", os.popen('date +%Y%m%d%H%M').readline().strip(), self.configuration.data)
        bb.data.setVar("BUILDSTART", time.strftime('%m/%d/%Y %H:%M:%S',time.gmtime()),self.configuration.data)

        buildname = bb.data.getVar("BUILDNAME", self.configuration.data)

        if self.configuration.interactive:
            self.interactiveMode()

        if self.configuration.buildfile is not None:
            bf = os.path.abspath( self.configuration.buildfile )
            try:
                bbfile_data = bb.parse.handle(bf, self.configuration.data)
            except IOError:
                bb.msg.fatal(bb.msg.domain.Parsing, "Unable to open %s" % bf)

            item = bb.data.getVar('PN', bbfile_data, 1)
            try:
                self.tryBuildPackage( bf, item, bbfile_data )
            except bb.build.EventException:
                bb.msg.error(bb.msg.domain.Build,  "Build of '%s' failed" % item )

            sys.exit( self.stats.show() )

        # initialise the parsing status now we know we will need deps
        self.status = BBParsingStatus()

        ignore = bb.data.getVar("ASSUME_PROVIDED", self.configuration.data, 1) or ""
        self.status.ignored_dependencies = Set( ignore.split() )

        self.handleCollections( bb.data.getVar("BBFILE_COLLECTIONS", self.configuration.data, 1) )

        pkgs_to_build = None
        if args:
            if not pkgs_to_build:
                pkgs_to_build = []
            pkgs_to_build.extend(args)
        if not pkgs_to_build:
                bbpkgs = bb.data.getVar('BBPKGS', self.configuration.data, 1)
                if bbpkgs:
                        pkgs_to_build = bbpkgs.split()
        if not pkgs_to_build and not self.configuration.show_versions \
                             and not self.configuration.interactive \
                             and not self.configuration.show_environment:
                print "Nothing to do.  Use 'bitbake world' to build everything, or run 'bitbake --help'"
                print "for usage information."
                sys.exit(0)

        # Import Psyco if available and not disabled
        if not self.configuration.disable_psyco:
            try:
                import psyco
            except ImportError:
                bb.msg.note(1, bb.msg.domain.Collection, "Psyco JIT Compiler (http://psyco.sf.net) not available. Install it to increase performance.")
            else:
                psyco.bind( self.collect_bbfiles )
        else:
            bb.msg.note(1, bb.msg.domain.Collection, "You have disabled Psyco. This decreases performance.")

        try:
            bb.msg.debug(1, bb.msg.domain.Collection, "collecting .bb files")
            self.collect_bbfiles( self.myProgressCallback )
            bb.msg.debug(1, bb.msg.domain.Collection, "parsing complete")
            print
            if self.configuration.parse_only:
                bb.msg.note(1, bb.msg.domain.Collection, "Requested parsing .bb files only.  Exiting.")
                return


            self.buildDepgraph()

            if self.configuration.show_versions:
                self.showVersions()
                sys.exit( 0 )
            if 'world' in pkgs_to_build:
                self.buildWorldTargetList()
                pkgs_to_build.remove('world')
                for t in self.status.world_target:
                    pkgs_to_build.append(t)

            if self.configuration.dot_graph:
                self.generateDotGraph( pkgs_to_build, self.configuration.ignored_dot_deps )
                sys.exit( 0 )


            bb.event.fire(bb.event.BuildStarted(buildname, pkgs_to_build, self.configuration.event_data))

            failures = 0
            for k in pkgs_to_build:
                failed = False
                try:
                    if self.buildProvider( k , False ) == 0:
                        # already diagnosed
                        failed = True
                except bb.build.EventException:
                    bb.msg.error(bb.msg.domain.Build, "Build of " + k + " failed")
                    failed = True

                if failed:
                    failures += failures
                    if self.configuration.abort:
                        sys.exit(1)

            bb.event.fire(bb.event.BuildCompleted(buildname, pkgs_to_build, self.configuration.event_data, failures))

            sys.exit( self.stats.show() )

        except KeyboardInterrupt:
            bb.msg.note(1, bb.msg.domain.Collection, "KeyboardInterrupt - Build not completed.")
            sys.exit(1)

    def get_bbfiles( self, path = os.getcwd() ):
        """Get list of default .bb files by reading out the current directory"""
        contents = os.listdir(path)
        bbfiles = []
        for f in contents:
            (root, ext) = os.path.splitext(f)
            if ext == ".bb":
                bbfiles.append(os.path.abspath(os.path.join(os.getcwd(),f)))
        return bbfiles

    def find_bbfiles( self, path ):
        """Find all the .bb files in a directory (uses find)"""
        findcmd = 'find ' + path + ' -name *.bb | grep -v SCCS/'
        try:
            finddata = os.popen(findcmd)
        except OSError:
            return []
        return finddata.readlines()

    def collect_bbfiles( self, progressCallback ):
        """Collect all available .bb build files"""
        self.cb = progressCallback
        parsed, cached, skipped, masked = 0, 0, 0, 0
        self.bb_cache = bb.cache.init(self)

        files = (data.getVar( "BBFILES", self.configuration.data, 1 ) or "").split()
        data.setVar("BBFILES", " ".join(files), self.configuration.data)

        if not len(files):
            files = self.get_bbfiles()

        if not len(files):
            bb.msg.error(bb.msg.domain.Collection, "no files to build.")

        newfiles = []
        for f in files:
            if os.path.isdir(f):
                dirfiles = self.find_bbfiles(f)
                if dirfiles:
                    newfiles += dirfiles
                    continue
            newfiles += glob.glob(f) or [ f ]

        bbmask = bb.data.getVar('BBMASK', self.configuration.data, 1) or ""
        try:
            bbmask_compiled = re.compile(bbmask)
        except sre_constants.error:
            bb.msg.fatal(bb.msg.domain.Collection, "BBMASK is not a valid regular expression.")

        for i in xrange( len( newfiles ) ):
            f = newfiles[i]
            if bbmask and bbmask_compiled.search(f):
                bb.msg.debug(1, bb.msg.domain.Collection, "bbmake: skipping %s" % f, f)
                masked += 1
                continue
            bb.msg.debug(1, bb.msg.domain.Collection, "bbmake: parsing %s" % f, f)

            # read a file's metadata
            try:
                fromCache, skip = self.bb_cache.loadData(f, self)
                if skip:
                    skipped += 1
                    bb.msg.debug(2, bb.msg.domain.Collection, "Skipping %s" % f, f)
                    self.bb_cache.skip(f)
                    continue
                elif fromCache: cached += 1
                else: parsed += 1
                deps = None

                # allow metadata files to add items to BBFILES
                #data.update_data(self.pkgdata[f])
                addbbfiles = self.bb_cache.getVar('BBFILES', f, False) or None
                if addbbfiles:
                    for aof in addbbfiles.split():
                        if not files.count(aof):
                            if not os.path.isabs(aof):
                                aof = os.path.join(os.path.dirname(f),aof)
                            files.append(aof)

                # now inform the caller
                if self.cb is not None:
                    self.cb( i + 1, len( newfiles ), f, self.bb_cache, fromCache )

            except IOError, e:
                self.bb_cache.remove(f)
                bb.msg.error(bb.msg.domain.Collection, "opening %s: %s" % (f, e), f)
                pass
            except KeyboardInterrupt:
                self.bb_cache.sync()
                raise
            except Exception, e:
                self.bb_cache.remove(f)
                bb.msg.error(bb.msg.domain.Collection, "%s while parsing %s" % (e, f), f)
            except:
                self.bb_cache.remove(f)
                raise

        if self.cb is not None:
            print "\r" # need newline after Handling Bitbake files message
            bb.msg.note(1, bb.msg.domain.Collection, "Parsing finished. %d cached, %d parsed, %d skipped, %d masked." % ( cached, parsed, skipped, masked ))

        self.bb_cache.sync()

#============================================================================#
# main
#============================================================================#

def main():
    parser = optparse.OptionParser( version = "BitBake Build Tool Core version %s, %%prog version %s" % ( bb.__version__, __version__ ),
    usage = """%prog [options] [package ...]

Executes the specified task (default is 'build') for a given set of BitBake files.
It expects that BBFILES is defined, which is a space seperated list of files to
be executed.  BBFILES does support wildcards.
Default BBFILES are the .bb files in the current directory.""" )

    parser.add_option( "-b", "--buildfile", help = "execute the task against this .bb file, rather than a package from BBFILES.",
               action = "store", dest = "buildfile", default = None )

    parser.add_option( "-k", "--continue", help = "continue as much as possible after an error. While the target that failed, and those that depend on it, cannot be remade, the other dependencies of these targets can be processed all the same.",
               action = "store_false", dest = "abort", default = True )

    parser.add_option( "-f", "--force", help = "force run of specified cmd, regardless of stamp status",
               action = "store_true", dest = "force", default = False )

    parser.add_option( "-i", "--interactive", help = "drop into the interactive mode also called the BitBake shell.",
               action = "store_true", dest = "interactive", default = False )

    parser.add_option( "-c", "--cmd", help = "Specify task to execute. Note that this only executes the specified task for the providee and the packages it depends on, i.e. 'compile' does not implicitly call stage for the dependencies (IOW: use only if you know what you are doing). Depending on the base.bbclass a listtaks tasks is defined and will show available tasks",
               action = "store", dest = "cmd", default = "build" )

    parser.add_option( "-r", "--read", help = "read the specified file before bitbake.conf",
               action = "append", dest = "file", default = [] )

    parser.add_option( "-v", "--verbose", help = "output more chit-chat to the terminal",
               action = "store_true", dest = "verbose", default = False )

    parser.add_option( "-D", "--debug", help = "Increase the debug level. You can specify this more than once.",
               action = "count", dest="debug", default = 0)

    parser.add_option( "-n", "--dry-run", help = "don't execute, just go through the motions",
               action = "store_true", dest = "dry_run", default = False )

    parser.add_option( "-p", "--parse-only", help = "quit after parsing the BB files (developers only)",
               action = "store_true", dest = "parse_only", default = False )

    parser.add_option( "-d", "--disable-psyco", help = "disable using the psyco just-in-time compiler (not recommended)",
               action = "store_true", dest = "disable_psyco", default = False )

    parser.add_option( "-s", "--show-versions", help = "show current and preferred versions of all packages",
               action = "store_true", dest = "show_versions", default = False )

    parser.add_option( "-e", "--environment", help = "show the global or per-package environment (this is what used to be bbread)",
               action = "store_true", dest = "show_environment", default = False )

    parser.add_option( "-g", "--graphviz", help = "emit the dependency trees of the specified packages in the dot syntax",
                action = "store_true", dest = "dot_graph", default = False )
    parser.add_option( "-I", "--ignore-deps", help = """Stop processing at the given list of dependencies when generating dependency graphs. This can help to make the graph more appealing""",
                action = "append", dest = "ignored_dot_deps", default = [] )


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

    cooker = BBCooker()
    cooker.cook( BBConfiguration( options ), args[1:] )



if __name__ == "__main__":
    main()
