#!/usr/bin/python

# Copyright (C) Mark Eichin.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the 3-clause BSD terms are met.
# (License subject to change.)

"""Hiveminder Braindump tag lister

Fetches all of your tags, in the (currently) cheapest possible way.
(At the moment that involves fetching all of your tasks...)
"""

__revision__ = "$Id: hive-expensive-tags.py,v 1.3 2008/01/05 03:33:14 eichin Exp $"
__version__ = "%s/%s" % (__revision__.split()[3], __revision__.split()[2])

import os
import urllib
import optparse
import sys
import re

try:
    import cElementTree as etree
except ImportError:
    try:
        import elementtree.ElementTree as etree
    except ImportError:
        import xml.etree.ElementTree as etree


# TODO: stolen from non-importable hive-braindump, fix that up
def my_sid(known_sid=[]):
    """fetch my sid from ~/.hiveminder"""

    if not known_sid:
        hive_file = os.path.expanduser("~/.hiveminder")
        for line in file(hive_file):
            if line.startswith("sid:"):
                known_sid.append(line.strip().split(":", 1)[-1].strip())
                break
        else:
            raise Exception("no sid found!")
    return known_sid[0]

def my_email(known_email=[]):
    """fetch my email from ~/.hiveminder"""

    if not known_email:
        hive_file = os.path.expanduser("~/.hiveminder")
        for line in file(hive_file):
            if line.startswith("email:"):
                known_email.append(line.strip().split(":", 1)[-1].strip())
                break
        else:
            raise Exception("no email found!")
    return known_email[0]


def fetch_all_tags():
    """get back a tree of everything"""
    task_search = "http://hiveminder.com/=/action/BTDT.Action.TaskSearch.xml"

    uo = urllib.FancyURLopener()
    uo.addheader("Cookie", "JIFTY_SID_HIVEMINDER=%s" % my_sid())

    u = uo.open(task_search, data=urllib.urlencode({
        # we *do* need the owner...
        "owner": my_email(),
        }))

    xmlresponse = u.read()
    try:
        response = etree.fromstring(xmlresponse)
    except Exception, xerr:     # xml.parsers.expat.ExpatError
        print "got", xerr, "parsing", xmlresponse
        raise

    tags = set()
    for tags_element in response.findall("content/tasks/tags"):
        # TODO: filter out completed/deleted tasks, maybe?
        if tags_element.text:
            for tag in re.findall('"(.*?)"', tags_element.text):
                tags.add(tag)

    return tags

def elisp_quote(s):
    """quote appropriately for appearing inline as an elisp strings"""
    s.replace('"', r'\"')
    return '"' + s + '"'

if __name__ == "__main__":
    sys.exit("DON'T RUN THIS.  Use hive-cheap-tags.py instead.")
    parser = optparse.OptionParser(usage=__doc__, version="%%prog %s" % __version__)
    parser.add_option("--zero", "-0", "--print0", action="store_true")
    parser.add_option("--elisp", action="store_true")
    options, args = parser.parse_args()
    if args:
        parser.print_help()
        sys.exit(1)

    tags = fetch_all_tags()

    if options.elisp:
        print "(setq hiveminder-tags '("
    for tag in sorted(tags):
        if options.zero:
            sys.stdout.write(tag)
            sys.stdout.write('\0')
        elif options.elisp:
            print elisp_quote(tag)
        else:
            print tag
    if options.elisp:
        print "))"