#!/usr/bin/python
#
# Copyright (C) 2011, Stefano Rivera <stefanor@ubuntu.com>
#
# Permission to use, copy, modify, and/or distribute this software for any
# purpose with or without fee is hereby granted, provided that the above
# copyright notice and this permission notice appear in all copies.
#
# THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
# WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
# MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
# ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
# WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
# ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
# OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.

import gzip
import json
import optparse
import os
import time
import urllib

from ubuntutools.lp.lpapicache import Distribution

DATA_URL = ('http://people.ubuntuwire.org/~stefanor/ubuntu-image-contents/'
            'image_contents.json.gz')


def load_index():
    cachedir = os.path.expanduser('~/.cache/ubuntu-dev-tools')
    fn = os.path.join(cachedir, 'image_contents.json.gz')

    if (not os.path.isfile(fn)
            or time.time() - os.path.getmtime(fn) > 60 * 60 * 24):
        if not os.path.isdir(cachedir):
            os.makedirs(cachedir)
        urllib.urlretrieve(DATA_URL, fn)

    with gzip.open(fn, 'r') as f:
        return json.load(f)


def main():
    parser = optparse.OptionParser('%prog [options] package...')
    parser.add_option('-b', '--binary',
                      default=False, action='store_true',
                      help="Binary packages are being specified, "
                           "not source packages (fast)")
    options, args = parser.parse_args()

    if len(args) < 1:
        parser.error("At least one package must be specified")

    index = load_index()
    if options.binary:
        for binary in args:
            if binary in index:
                for flavor, type_ in index[binary]:
                    print "%s is present on %s %s" % (binary, flavor, type_)
            else:
                print "%s is not present on any current images" % binary
    else:
        archive = Distribution('ubuntu').getArchive()
        for source in args:
            spph = archive.getSourcePackage(source)
            binaries = frozenset(bpph.getPackageName()
                                 for bpph in spph.getBinaries())
            present = False
            for binary in binaries:
                if binary in index:
                    present = True
                    for flavor, type_ in index[binary]:
                        print ("%s: %s is present on %s %s"
                               % (source, binary, flavor, type_))
            if not present:
                print ("No binaries from %s are present on any current images"
                       % source)


if __name__ == '__main__':
    main()