#!/usr/bin/python3
#
# Collect information about a package installation/upgrade failure.
#
# Copyright (c) 2007 - 2009 Canonical Ltd.
# Author: Martin Pitt <martin.pitt@ubuntu.com>
#
# This program is free software; you can redistribute it and/or modify it
# under the terms of the GNU General Public License as published by the
# Free Software Foundation; either version 2 of the License, or (at your
# option) any later version.  See http://www.gnu.org/copyleft/gpl.html for
# the full text of the license.

import os
import os.path
import optparse
import sys

import apport
import apport.fileutils
import apport.packaging


def mkattrname(path):
    '''Convert a file path to a problem report attribute name.'''

    name = ''
    for dir in path.split(os.sep):
        if dir:
            name += ''.join([c for c in dir[0].upper() + dir[1:] if c.isalnum()])
    return name

#
# main
#

# parse command line arguments
optparser = optparse.OptionParser('%prog [options]')
optparser.add_option('-p', '--package', help='Specify the package name which failed to upgrade (mandatory)')
optparser.add_option('-l', '--log', action='append', dest='logs',
                     help='Append given log file, or, if it is a directory, all files in it (can be specified multiple times)')
optparser.add_option('-t', '--tags', help='Add the following tags to the bug report (comma separated)')
options = optparser.parse_args()[0]

if not options.package:
    apport.fatal('You need to specify a package name with --package')
    sys.exit(1)

# create report
pr = apport.Report('Package')
try:
    version = apport.packaging.get_version(options.package)
except ValueError:
    # package not installed
    version = None
pr['Package'] = '%s %s' % (options.package, version or '(not installed)')
pr['SourcePackage'] = apport.packaging.get_source(options.package)
pr['ErrorMessage'] = (sys.stdin, False)

if options.tags:
    tags = options.tags.replace(',', '')
    pr['Tags'] = tags

for l in (options.logs or []):
    if os.path.isfile(l):
        pr[mkattrname(l)] = (l,)
    elif os.path.isdir(l):
        for f in os.listdir(l):
            path = os.path.join(l, f)
            if os.path.isfile(path):
                pr[mkattrname(path)] = (path,)

# write report
try:
    with apport.fileutils.make_report_file(pr) as f:
        pr.write(f)
except (IOError, OSError) as e:
    apport.fatal('Cannot create report: ' + str(e))