#!/usr/bin/env python

# -*- coding: utf-8; -*-

import os

import sys

import optparse

from subprocess import Popen, PIPE

import urllib

import re

import string

from datetime import date

import unicodedata

import smtplib

from email.mime.multipart import MIMEMultipart

from email.mime.text import MIMEText

# export PYTHONPATH="/path/to/your/webkit//WebKitTools/Scripts/"

env = os.environ;

sys.path.append("%s/dev/webkit/land/WebKitTools/Scripts/" % env["HOME"])

if env.has_key("WEBKITDIR"):

    sys.path.append("%s/WebKitTools/Scripts/" % env["WEBKITDIR"])

sys.path.append("%s/WebKitTools/Scripts/" % os.getcwd())

# XXX fscking python2 ascii/utf8 handling

# http://stackoverflow.com/questions/492483/setting-the-correct-encoding-when-piping-stdout-in-python

import sys

import codecs

sys.stdout = codecs.getwriter('utf8')(sys.stdout)

from webkitpy.common.checkout.changelog import parse_bug_id

from webkitpy.common.net.bugzilla import Bugzilla

from webkitpy.thirdparty.BeautifulSoup import BeautifulSoup, SoupStrainer

from webkitpy.common.system.user import User

def die(msg):

    sys.stderr.write(msg + "\n")

    sys.exit(1)

def normalizeBranchName(branch):

    return branch.translate(string.maketrans("._/", "---")) # XXX

def run(cmd, ignore_error=False):

    return Popen(cmd, stdout=PIPE).communicate()[0]

def escapeHtml(s):

    s = s.replace("<", "&lt;")

    s = s.replace(">", ">")

    s = s.replace("\"", """)

    return s

def lastTag():

    branch = run(["git", "symbolic-ref", "HEAD"])[11:-1]

    base = run(["git", "merge-base", "HEAD", "master"]).strip()

    tags = run(["git", "tag", "--contains", base, "-l", "%s*" % branch])

    if tags:

        return tags.strip().splitlines()[-1]

    else:

        return base

def extractShortChangeLog(sha1):

    log = run(["git", "log", "-1", sha1]).splitlines()

    # merge commit

    if log[1].startswith("Merge: "):

        return log[5].strip()

    for line in log[4:]:

        line = line.strip()

        if len(line) == 0:

            continue

        if re.search("20..-..-.. .*<.*>", line):

            continue

        if re.search("reviewed by |unreviewed|patch by", line, re.IGNORECASE):

            continue

        if re.search("^http.?://bugs.webkit", line):

            break;

        if re.search("^\*|^\(", line, re.IGNORECASE):

            break;

        return line

    # just in case

    return log[4].strip()

class MailSender(object):

    def __init__(self):

        env = os.environ

        if env.has_key("WEBKIT_EMAILSENDER"):

            self.src = env["WEBKIT_EMAILSENDER"]

            # accept "Foobar <foo@bar.com>" or "foo@bar.com"

            self.sender = self.src.split("<")[-1:][0].strip(">")

        else:

            self.src = "WebKit Reports <no-reply@webkit.org>"

            self.sender = "no-reply@webkit.org"

    def send(self, dest, subject, htmlMessage, textMessage):

        msg = MIMEMultipart('alternative')

        msg["Subject"] = subject

        msg["From"] = self.src

        msg["To"] = dest

        # python 2.x unicode handling is a mess and the email modules fail to handle unicode messages

        htmlAscii = unicodedata.normalize('NFKD', htmlMessage).encode('ascii', 'ignore')

        textAscii = unicodedata.normalize('NFKD', textMessage).encode('ascii', 'ignore')

        part1 = MIMEText(textAscii, 'plain')

        part2 = MIMEText(htmlAscii, 'html')

        msg.attach(part1)

        msg.attach(part2)

        try:

            smtp = smtplib.SMTP("localhost")

            smtp.sendmail(self.sender, dest.split(","), msg.as_string())

            smtp.quit()

            print "Email has been sent"

        except smtplib.SMTPException:

            print "Error: unable to send email"

class HtmlReleaseNotes(list):

    def _(self, text):

        return escapeHtml(text).decode("utf-8")

    def append(self, text):

        if text == "":

            list.append(self, u"<br>")

        else:

            list.append(self, u"<p>%s</p>" % self._(text))

    def startList(self, title):

        list.append(self, u"<p><strong>%s</strong></p>" % self._(title))

        list.append(self, u"<ul>")

    def endList(self):

        list.append(self, u"</ul>")

    def appendBug(self, id, title):

        list.append(self, u"<li><a href=\"https://webkit.org/b/%s\">#%s</a>: %s" % (id, id, self._(title)))

    def appendCommit(self, commit):

        title = extractShortChangeLog(commit)

        if not title.startswith("Merge branch "):

            list.append(self, u"<li><a href=\"http://gitorious.org/webkit/qtwebkit/commit/%s\">%s</a>: %s" % (commit[:7], commit[:7], self._(title)))

class WikiReleaseNotes(list):

    def _(self, text):

        text = text.replace("^", "{{{^}}}")

        return text.decode("utf-8")

    def append(self, text):

        list.append(self, u"%s" % self._(text))

        list.append(self, u"")

    def startList(self, title):

        list.append(self, u"=== %s ===" % self._(title))

    def endList(self):

        list.append(self, u"")

    def appendBug(self, id, title):

        list.append(self, u" * [https://webkit.org/b/%s #%s]: %s" % (id, id, self._(title)))

    def appendCommit(self, commit):

        title = extractShortChangeLog(commit)

        if not title.startswith("Merge branch "):

            list.append(self, u" * [http://gitorious.org/webkit/qtwebkit/commit/%s %s]: %s" % (commit[:7], commit[:7], self._(title)))

# XXX I'm not proud of this, but I'm recycling old code that just works :P

class ReleaseNotes():

    def __init__(self):

        self.__html = HtmlReleaseNotes()

        self.__wiki = WikiReleaseNotes()

    def html(self):

        return u"\n".join(self.__html)

    def wiki(self):

        return u"\n".join(self.__wiki)

    def append(self, text):

        self.__html.append(text)

        self.__wiki.append(text)

    def startList(self, title):

        self.__html.startList(title)

        self.__wiki.startList(title)

    def endList(self):

        self.__html.endList()

        self.__wiki.endList()

    def appendBug(self, id, title):

        self.__html.appendBug(id, title)

        self.__wiki.appendBug(id, title)

    def appendCommit(self, commit):

        self.__html.appendCommit(commit)

        self.__wiki.appendCommit(commit)

def main():

    op = optparse.OptionParser(usage="usage: %prog [options] [base commit]",

                               epilog="If no [base commit] is given, all commits since the branch creation (the fork point) will be considered")

    op.add_option("", "--since-last-tag", default=False, action="store_true", help="use last tag (or the branch fork point) as the base commit")

    op.add_option("", "--up-to", type="string", default="HEAD", action="store", help="last commit considered (default: HEAD)")

    op.add_option("", "--html", default=False, action="store_true", help="output in HTML format (default is wiki formatting)")

    op.add_option("", "--commits", default=False, action="store_true", help="also list commits made to the branch")

    op.add_option("", "--email", type="string", action="store", help="send an e-mail to EMAIL (comma separated list). Designed to be used weekly")

    op.add_option("", "--announce", default=False, action="store_true", help="send an e-mail to the QtWebKit announce mailing list")

    options, args = op.parse_args()

    if options.since_last_tag:

        base = lastTag()

    elif args:

        base = args[0]

    else:

        base = run(["git", "merge-base", "HEAD", "master"]).strip()

    top = options.up_to

    commits = run(["git", "rev-list", "--reverse", base + ".." + top]).strip().splitlines()

    if not commits or not commits[0]:

        print "No commits found. Try an older revision (tag)"

        sys.exit(1)

    # (Strip off refs/heads/ and newline)

    branchName = run(["git", "symbolic-ref", "HEAD"])[11:-1]

    bugzilla = Bugzilla(dryrun = False)

    # Some bugs are not detected by the script and are handled by hand

    # They can be stored inside git-config as below:

    # [extraBugs]

    #     <normalized branch name> = <bugId>-<commit sha1>

    #     <normalized branch name> = <bugId>-<commit sha1>

    #     ...

    extraBugs = run(["git", "config", "--get-all", "extraBugs.%s" % normalizeBranchName(branchName)]).strip().splitlines()

    bugs = set()

    for bug in extraBugs:

        id = bug.partition("-")[0]

        sha1 = bug.partition("-")[2]

        if sha1 in commits:

            print >> sys.stderr, "Warning: bug #%s is from 'git config', (%s)" % (id, sha1)

            bugs.add(int(id))

    for commit in commits:

        log = run(["git", "cat-file", "commit", commit])

        if re.search("\nThis reverts commit ", log):

            continue

        bug_id = parse_bug_id(log)

        if bug_id:

            bugs.add(int(bug_id))

            continue

    bugs = list(bugs)

    bugs.sort()

    print "%s bugs and %s commits from %s to %s" % (len(bugs), len(commits), base, top)

    notes = ReleaseNotes()

    dateTime = date.today()

    currentWeek = dateTime.isocalendar()[1]

    previousWeek = currentWeek - 1

    notes.append("Release Notes for %s-week%02d - Changes since %s" % (branchName, previousWeek, base))

    notes.append("%s bugs, %s commits" % (len(bugs), len(commits)))

    notes.append("")

    notes.startList("Bugs fixed / Tasks done:")

    for bug_id in bugs:

        try:

            bug = bugzilla.fetch_bug_dictionary(bug_id)

            title = escapeHtml(bug["title"])

            notes.appendBug(bug_id, title)

        except KeyboardInterrupt:

            raise

        except:

            notes.appendBug(bug_id, "(restricted bug)")

    notes.endList()

    if options.commits:

        notes.startList("Commits added or cherry-picked:")

        for commit in commits:

            print commit

            notes.appendCommit(commit)

        notes.endList()

    mail = MailSender()

    textBody = notes.wiki()

    htmlBody = notes.html()

    if options.email:

        subject = "[announce] %s-week%02d status report - changes since %s" % (branchName, previousWeek, base)

        mail.send(options.email, subject, htmlBody, textBody)

    if options.announce:

        subject = "%s-week%02d status report - changes since %s" % (branchName, previousWeek, base)

        response = User.prompt("Really send e-mail to announcement mailing list? (y/n) ")

        if response != "y":

            print "announcement e-mail not sent!"

        else:

            mail.send("qtwebkit-announce@qt.nokia.com", subject, htmlBody, textBody)

    if options.html:

        print htmlBody

    else:

        print "---------------"

        print textBody

        print "---------------"

if __name__ == "__main__":

    try:

       main()

    except KeyboardInterrupt:

        sys.exit(1)

# vim:et tw=0 ts=4 sw=4: