artemis.py
author Mirko Friedenhagen <mirko.friedenhagen@1und1.de>
Tue, 08 Jan 2008 23:45:23 +0100
changeset 16 49645303d045
parent 15 e5b03298394e
child 17 f70d7e98eb21
permissions -rw-r--r--
Spaces instead of tabs

# Author: Dmitriy Morozov <hg@foxcub.org>, 2007

"""A very simple and lightweight issue tracker for Mercurial."""

from mercurial import hg, util
from mercurial.i18n import _
import os, time, random, mailbox, glob, socket, ConfigParser


state = {'new': 'new', 'fixed': 'fixed'}
state['default'] = state['new']
issues_dir = ".issues"
filter_prefix = ".filter"
date_format = '%a, %d %b %Y %H:%M:%S'


def ilist(ui, repo, **opts):
    """List issues associated with the project"""

    # Process options
    show_all = opts['all']
    properties = []
    match_date, date_match = False, lambda x: True
    if opts['date']:
        match_date, date_match = True, util.matchdate(opts['date'])

    # Find issues
    issues_path = os.path.join(repo.root, issues_dir)
    if not os.path.exists(issues_path): return

    issues = glob.glob(os.path.join(issues_path, '*'))
   
    # Process filter
    if opts['filter']:
        filters = glob.glob(os.path.join(issues_path, filter_prefix + '*'))
        config = ConfigParser.SafeConfigParser()
        config.read(filters)
        if not config.has_section(opts['filter']):
            ui.warning('No filter %s defined\n', opts['filter'])
        else:
            properties += config.items(opts['filter'])
   
    _get_properties(opts['property'])
   
    for issue in issues:
        mbox = mailbox.mbox(issue)
        property_match = True
        for property,value in properties:
            property_match = property_match and (mbox[0][property] == value)
        if not show_all and (not properties or not property_match) and (properties or mbox[0]['State'].upper() == state['fixed'].upper()): continue


        if match_date and not date_match(util.parsedate(mbox[0]['date'])[0]): continue
        ui.write("%s (%d) [%s]: %s\n" % (issue[len(issues_path)+1:], # +1 for trailing /
                                         len(mbox)-1,                 # number of replies (-1 for self)
                                         mbox[0]['State'],
                                         mbox[0]['Subject']))
   

def iadd(ui, repo, id = None, comment = 0):
    """Adds a new issue, or comment to an existing issue ID or its comment COMMENT"""
   
    comment = int(comment)

    # First, make sure issues have a directory
    issues_path = os.path.join(repo.root, issues_dir)
    if not os.path.exists(issues_path): os.mkdir(issues_path)

    if id:
        issue_fn, issue_id = _find_issue(ui, repo, id)
        if not issue_fn:
            ui.warn('No such issue\n')
            return
   
    user = ui.username()

    default_issue_text  =         "From: %s\nDate: %s\n" % (user, util.datestr(format = date_format))
    if not id:
        default_issue_text +=     "State: %s\n" % state['default']
    default_issue_text +=        "Subject: brief description\n\n"
    default_issue_text +=         "Detailed description."

    issue = ui.edit(default_issue_text, user)
    if issue.strip() == '':
        ui.warn('Empty issue, ignoring\n')
        return
    if issue.strip() == default_issue_text:
        ui.warn('Unchanged issue text, ignoring\n')
        return

    # Create the message
    msg = mailbox.mboxMessage(issue)
    msg.set_from('artemis', True)
   
    # Pick random filename
    if not id:
        issue_fn = issues_path
        while os.path.exists(issue_fn):
            issue_id = _random_id()
            issue_fn = os.path.join(issues_path, issue_id)
    # else: issue_fn already set

    # Add message to the mailbox
    mbox = mailbox.mbox(issue_fn)
    if id and comment not in mbox:
        ui.warn('No such comment number in mailbox, commenting on the issue itself\n')
    if not id:
        msg.add_header('Message-Id', "<%s-0-artemis@%s>" % (issue_id, socket.gethostname()))
    else:
        msg.add_header('Message-Id', "<%s-%s-artemis@%s>" % (issue_id, _random_id(), socket.gethostname()))
        msg.add_header('References', mbox[(comment < len(mbox) and comment) or 0]['Message-Id'])
        msg.add_header('In-Reply-To', mbox[(comment < len(mbox) and comment) or 0]['Message-Id'])
    mbox.add(msg)
    mbox.close()

    # If adding issue, add the new mailbox to the repository
    if not id:
        repo.add([issue_fn[(len(repo.root)+1):]])            # +1 for the trailing /
        ui.status('Added new issue %s\n' % issue_id)


def ishow(ui, repo, id, comment = 0, **opts):
    """Shows issue ID, or possibly its comment COMMENT"""
   
    comment = int(comment)
    issue, id = _find_issue(ui, repo, id)
    if not issue: return
    mbox = mailbox.mbox(issue)

    if opts['all']:
        ui.write('='*70 + '\n')
        for i in xrange(len(mbox)):
            _write_message(ui, mbox[i], i)
            ui.write('-'*70 + '\n')
        return

    _show_mbox(ui, mbox, comment)


def iupdate(ui, repo, id, **opts):
    """Update properties of issue ID"""

    issue, id = _find_issue(ui, repo, id)
    if not issue: return

    properties = _get_properties(opts['property'])
   
    # Read the issue
    mbox = mailbox.mbox(issue)
    msg = mbox[0]

    # Fix the properties
    properties_text = ''
    for property, value in properties:
        msg.replace_header(property, value)
        properties_text += '%s=%s\n' % (property, value)
    mbox[0] = msg

    # Write down a comment about updated properties
    if properties and not opts['no_property_comment']:
        user = ui.username()
        properties_text  =     "From: %s\nDate: %s\nSubject: properties changes (%s)\n\n%s" % \
                            (user, util.datestr(format = date_format),
                             _pretty_list(list(set([property for property, value in properties]))),
                             properties_text)
        msg = mailbox.mboxMessage(properties_text)
        msg.add_header('Message-Id', "<%s-%s-artemis@%s>" % (id, _random_id(), socket.gethostname()))
        msg.add_header('References', mbox[0]['Message-Id'])
        msg.add_header('In-Reply-To', mbox[0]['Message-Id'])
        msg.set_from('artemis', True)
        mbox.add(msg)
    mbox.flush()

    # Show updated message
    _show_mbox(ui, mbox, 0)


def _find_issue(ui, repo, id):
    issues_path = os.path.join(repo.root, issues_dir)
    if not os.path.exists(issues_path): return False

    issues = glob.glob(os.path.join(issues_path, id + '*'))

    if len(issues) == 0:
        return False, 0
    elif len(issues) > 1:
        ui.status("Multiple choices:\n")
        for i in issues: ui.status('  ', i[len(issues_path)+1:], '\n')
        return False, 0
   
    return issues[0], issues[0][len(issues_path)+1:]

def _get_properties(property_list):
    return [p.split('=') for p in property_list]
   
def _write_message(ui, message, index = 0):
    if index: ui.write("Comment: %d\n" % index)
    if ui.verbose:
        ui.write(message.as_string().strip() + '\n')
    else:
        if 'From' in message: ui.write('From: %s\n' % message['From'])
        if 'Date' in message: ui.write('Date: %s\n' % message['Date'])
        if 'Subject' in message: ui.write('Subject: %s\n' % message['Subject'])
        if 'State' in message: ui.write('State: %s\n' % message['State'])
        ui.write('\n' + message.get_payload().strip() + '\n')

def _show_mbox(ui, mbox, comment):
    # Output the issue (or comment)
    if comment >= len(mbox):
        comment = 0
        ui.warn('Comment out of range, showing the issue itself\n')
    msg = mbox[comment]
    ui.write('='*70 + '\n')
    if comment:
        ui.write('Subject: %s\n' % mbox[0]['Subject'])
        ui.write('State: %s\n' % mbox[0]['State'])
        ui.write('-'*70 + '\n')
    _write_message(ui, msg, comment)
    ui.write('-'*70 + '\n')

    # Read the mailbox into the messages and children dictionaries
    messages = {}
    children = {}
    for i in xrange(len(mbox)):
        m = mbox[i]
        messages[m['Message-Id']] = (i,m)
        children.setdefault(m['In-Reply-To'], []).append(m['Message-Id'])
    children[None] = []                # Safeguard against infinte loop on empty Message-Id

    # Iterate over children
    id = msg['Message-Id']
    id_stack = (id in children and map(lambda x: (x, 1), reversed(children[id]))) or []
    if not id_stack: return
    ui.write('Comments:\n')
    while id_stack:
        id,offset = id_stack.pop()
        id_stack += (id in children and map(lambda x: (x, offset+1), reversed(children[id]))) or []
        index, msg = messages[id]
        ui.write('  '*offset + ('%d: ' % index) + msg['Subject'] + '\n')
    ui.write('-'*70 + '\n')

def _pretty_list(lst):
    s = ''
    for i in lst:
        s += i + ', '
    return s[:-2]

def _random_id():
    return "%x" % random.randint(2**63, 2**64-1)


cmdtable = {
    'ilist':    (ilist,
                 [('a', 'all', False,
                   'list all issues (by default only those with state new)'),
                  ('p', 'property', [],
                   'list issues with specific field values (e.g., -p state=fixed)'),
                  ('d', 'date', '', 'restrict to issues matching the date (e.g., -d ">12/28/2007)"'),
                  ('f', 'filter', '', 'restrict to pre-defined filter (in %s/%s*)' % (issues_dir, filter_prefix))],
                 _('hg ilist [OPTIONS]')),
    'iadd':       (iadd, 
                 [],
                 _('hg iadd [ID] [COMMENT]')),
    'ishow':      (ishow,
                 [('a', 'all', None, 'list all comments')],
                 _('hg ishow [OPTIONS] ID [COMMENT]')),
    'iupdate':    (iupdate,
                 [('p', 'property', [],
                   'update properties (e.g., -p state=fixed)'),
                  ('n', 'no-property-comment', None,
                   'do not add a comment about changed properties')],
                 _('hg iupdate [OPTIONS] ID'))
}