#!/usr/bin/python
# -*- coding: utf-8 -*-
#
# submittodebian - tool to submit patches to Debian's BTS
# Copyright (C) 2007, 2009 Canonical Ltd.
# Author: Soren Hansen <soren@ubuntu.com>,
#         Steve Langasek <slangasek@canonical.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.
#
# 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.
#
# See file /usr/share/common-licenses/GPL for more details.
#
# ##################################################################

import os
import re
import shutil
import sys
from tempfile import mkdtemp

from distro_info import UbuntuDistroInfo

from ubuntutools.config import ubu_email
from ubuntutools.question import YesNoQuestion, EditFile
from ubuntutools.subprocess import call, check_call, Popen, PIPE

try:
    from debian.changelog import Changelog
except ImportError:
    print (u"This utility requires modules from the «python-debian» package, "
           u"which isn't currently installed.")
    sys.exit(1)

if not os.path.exists('/usr/bin/reportbug'):
    print (u"This utility requires the «reportbug» package, which isn't "
           u"currently installed.")
    sys.exit(1)

def get_most_recent_debian_version(changelog):
    for block in changelog:
        version = block.version.full_version
        if not re.search('(ubuntu|build)', version):
            return version

def get_bug_body(changelog):
    entry = next(iter(changelog))
    msg = """In Ubuntu, the attached patch was applied to achieve the following:

## ---------------- REPLACE THIS WITH ACTUAL INFORMATION ---------------------
## Please add all necessary information about why the change needed to go in
## Ubuntu, quote policy, spec or any other background material and why it can
## and should be used in Debian too.  If the patch is composed of multiple
## independent pieces, please send them as separate bug reports.
## ---------------- REPLACE THIS WITH ACTUAL INFORMATION ---------------------

%s

Thanks for considering the patch.
""" % ("\n".join([a for a in entry.changes()]))
    return msg

def gen_debdiff(tmpdir, changelog):
    pkg = changelog.package

    changelog_it = iter(changelog)
    newver = next(changelog_it).version
    oldver = next(changelog_it).version

    debdiff = os.path.join(tmpdir, '%s_%s.debdiff' % (pkg, newver))

    devnull = open('/dev/null', 'w')
    diff_cmd = ['bzr', 'diff', '-r', 'tag:' + str(oldver)]
    if call(diff_cmd, stdout=devnull, stderr=devnull) == 1:
        print "Extracting bzr diff between %s and %s" % (oldver, newver)
    else:
        if oldver.epoch is not None:
            oldver = str(oldver)[str(oldver).index(":")+1:]
        if newver.epoch is not None:
            newver = str(newver)[str(newver).index(":")+1:]

        olddsc = '../%s_%s.dsc' % (pkg, oldver)
        newdsc = '../%s_%s.dsc' % (pkg, newver)

        check_file(olddsc)
        check_file(newdsc)

        print "Generating debdiff between %s and %s" % (oldver, newver)
        diff_cmd = ['debdiff', olddsc, newdsc]

    diff = Popen(diff_cmd, stdout=PIPE)
    debdiff_f = open(debdiff, 'w')
    filterdiff = Popen(['filterdiff', '-x', '*changelog*'],
                       stdin=diff.stdout, stdout=debdiff_f)
    diff.stdout.close()
    filterdiff.wait()
    debdiff_f.close()
    devnull.close()

    return debdiff

def check_file(fname, critical = True):
    if os.path.exists(fname):
        return fname
    else:
        if not critical:
            return False
        print u"Couldn't find «%s».\n" % fname
        sys.exit(1)

def submit_bugreport(body, debdiff, deb_version, changelog):
    devel = UbuntuDistroInfo().devel()

    if os.path.dirname(sys.argv[0]).startswith('/usr/bin'):
        editor_path = '/usr/share/ubuntu-dev-tools'
    else:
        editor_path = os.path.dirname(sys.argv[0])
    env = dict(os.environ.items())
    if 'EDITOR' in env:
        env['UDT_EDIT_WRAPPER_EDITOR'] = env['EDITOR']
    if 'VISUAL' in env:
        env['UDT_EDIT_WRAPPER_VISUAL'] = env['VISUAL']
    env['EDITOR'] = os.path.join(editor_path, 'enforced-editing-wrapper')
    env['VISUAL'] = os.path.join(editor_path, 'enforced-editing-wrapper')
    env['UDT_EDIT_WRAPPER_TEMPLATE_RE'] = (
            '.*REPLACE THIS WITH ACTUAL INFORMATION.*')
    env['UDT_EDIT_WRAPPER_FILE_DESCRIPTION'] = 'bug report'

    cmd = ('reportbug',
           '--no-check-available',
           '--no-check-installed',
           '--pseudo-header', 'User: ubuntu-devel@lists.ubuntu.com',
           '--pseudo-header', 'Usertags: origin-ubuntu %s ubuntu-patch' % devel,
           '--tag', 'patch',
           '--attach', debdiff,
           '--bts', 'debian',
           '--include', body,
           '--package-version', deb_version,
           changelog.package)
    check_call(cmd, env=env)


def check_reportbug_config():
    fn = os.path.expanduser('~/.reportbugrc')
    if os.path.exists(fn):
        return
    email = ubu_email()[1]
    reportbugrc = """# Reportbug configuration generated by submittodebian(1)
# See reportbug.conf(5) for the configuration file format.

# Use Debian's reportbug SMTP Server:
# Note: it's limited to 5 connections per hour, and cannot CC you at submission
# time. See /usr/share/doc/reportbug/README.Users.gz for more details.
smtphost reportbug.debian.org:587
header "X-Debbugs-CC: %s"
no-cc

# Use GMail's SMTP Server:
#smtphost smtp.googlemail.com:587
#smtpuser "<your address>@gmail.com"
#smtptls
""" % email

    with file(fn, 'w') as f:
        f.write(reportbugrc)

    print """\
You have not configured reportbug. Assuming this is the first time you have
used it. Writing a ~/.reportbugrc that will use Debian's mail server, and CC
the bug to you at <%s>

--- Generated ~/.reportbugrc ---
%s
--- End of ~/.reportbugrc ---

If this is not correct, please exit now and edit ~/.reportbugrc or run
reportbug --configure for its configuration wizard.
""" % (email, reportbugrc.strip())

    if YesNoQuestion().ask("Continue submitting this bug", "yes") == "no":
        sys.exit(1)

def main():
    check_reportbug_config()
    changelog_file = (check_file('debian/changelog', critical = False) or
                      check_file('../debian/changelog'))
    changelog = Changelog(file(changelog_file).read())

    deb_version = get_most_recent_debian_version(changelog)
    bug_body = get_bug_body(changelog)

    tmpdir = mkdtemp()
    body = os.path.join(tmpdir, 'bug_body')
    fp = open(body, 'w')
    fp.write(bug_body)
    fp.close()

    debdiff = gen_debdiff(tmpdir, changelog)

    EditFile(debdiff, 'debdiff').edit(optional=True)

    submit_bugreport(body, debdiff, deb_version, changelog)
    os.unlink(body)
    os.unlink(debdiff)
    shutil.rmtree(tmpdir)

if __name__ == '__main__':
    main()
