#!/usr/bin/python
# -*- coding: utf-8 -*-
#
# Copyright (C) 2007 Canonical Ltd.
# Created by Daniel Holbach <daniel.holbach@ubuntu.com>
# Modified by Jonathan Patrick Davies <jpds@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; version 3.
# 
# 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-3 for more details.
#
# ##################################################################
#
# This script is used to checkout or branch all the Bazaar branches
# in a Launchpad team.

import os
import re
import subprocess
import sys
import urllib2
from optparse import OptionParser

def main():
    usage = "Usage: %prog [-d <directory>] -t <team> [-o <operation>]"
    usage += "\nUsage: %prog <team>"
    optParser = OptionParser(usage)
    optsParsed = 0

    # Our options.
    optParser.add_option("-d", "--directory", action = "store_true",
            dest = "directory", help = "Directory to download branches to.")
    optParser.add_option("-t", "--team", action = "store_true",
            dest = "lpteam", help = "Launchpad team to download branches from.")
    optParser.add_option("-o", "--operation", action = "store_true",
            dest = "operation", help = "Whether to branch or checkout the " \
                "Bazaar branches. May be either 'branch' or 'checkout'.")

    (options, args) = optParser.parse_args()

    # Parse our options.
    # No flags, and no team name specified.
    if not options.lpteam and not args:
        print >> sys.stderr, "No team has been specified."
        optParser.print_help()
        sys.exit(1)
    
    if args:
        team = args[0]

    # Dictionary settings.
    if options.directory:
        try:
            directory = args[optsParsed]
            optsParsed += 1
        except IndexError:
            print >> sys.stderr, "The '-d' option requires an argument."
            optParser.print_help()
            sys.exit(1)

        if not os.path.isdir(directory): # Check that it is a directory.
            print >> sys.stderr, "%s is not a valid directory." % directory
            optParser.print_help()
            sys.exit(1)
        else:
            directory = os.path.abspath(args[0])
    else:
        # Otherwise use our current directory.
        directory = os.getcwd()

    # Launchpad team setting.
    if options.lpteam:
        try:
            team = args[optsParsed]
            optsParsed += 1
        except IndexError:
            print >> sys.stderr, "The '-t' option requires an argument."
            optParser.print_help()
            sys.exit(1)

    # Type of Bazaar operation to perform.
    if options.operation:
        try: 
            operation_type = args[optsParsed]
            optsParsed += 1
        except IndexError:
            print >> sys.stderr, "The '-o' option requires an argument."
            optParser.print_help()
            sys.exit(1)

        # Got an argument, check if it is valid.
        if operation_type.lower() not in ("branch", "checkout"):
            print >> sys.stderr, "Invalid operation '%s' for '-o' flag." % \
                operation_type
            optParser.print_help()
            sys.exit(1)
    else:
        operation_type = "branch"

    pwd = os.getcwd()
    os.chdir(directory)

    # Try to open the teams code page.
    try:
        sock = urllib2.urlopen("https://code.launchpad.net/~%s" % team)
    except urllib2.HTTPError:
        print >> sys.stderr, "The page https://code.launchpad.net/~%s does " \
            "not exist." % team
        print >> sys.stderr, "Perhaps invalid team name?"
        sys.exit(1)

    try:
        os.makedirs(team)
    except:
        pass
    os.chdir(team)
    
    branch_list_page = sock.read()
    sock.close()
    
    branch_page_urls_regex = r'.*<a href="/(~%s/.*)".*' % team
    branch_page_urls = re.findall(branch_page_urls_regex, branch_list_page)
    
    print "Downloading all branches for the team '%s'. This may take some " \
        "time." % team
    
    for url in branch_page_urls:
        sock = urllib2.urlopen("https://code.launchpad.net/%s" % url)
        branch_page = sock.read()
        sock.close()
	
        branch_url_regex = r'<tt>bzr branch lp:~(.*)</tt>'
        branch_url = re.findall(branch_url_regex, branch_page)
        print "Downloading branch: lp:~%s." % branch_url[0]

        if branch_url[0]:
            product = branch_url[0].split("/")[-2]
            branch_nick = branch_url[0].split("/")[-1]
       
  	    if not os.path.exists(product):
  	        os.makedirs(product)
           	os.chdir(product)
       	       	    
        if not os.path.exists(branch_nick):
            subprocess.call(["bzr", operation_type, "lp:~%s" % branch_url[0]])
        else:
            os.chdir(branch_nick)
            subprocess.call(["bzr", "merge", "--pull", "--remember"])
            os.chdir(os.path.join(directory, team))
            
    os.chdir(pwd)
    sys.exit(0)
    
if __name__ == "__main__":
    main()