# Copyright (C) 2005 Aaron Bentley and Panometrics, Inc.
# <abentley@panoramicfeedback.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.
#
#    You should have received a copy of the GNU General Public License
#    along with this program; if not, write to the Free Software
#    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
"""Assign the root directory for bug tracking"""
import os.path
from libbe import cmdutil, bugdir
__desc__ = __doc__

def execute(args):
    """
    >>> from libbe import utility, rcs
    >>> import os
    >>> dir = utility.Dir()
    >>> try:
    ...     bugdir.BugDir(dir.path)
    ... except bugdir.NoBugDir, e:
    ...     True
    True
    >>> execute([dir.path])
    No revision control detected.
    Directory initialized.
    >>> del(dir)

    >>> dir = utility.Dir()
    >>> os.chdir(dir.path)
    >>> rcs = rcs.installed_rcs()
    >>> rcs.init('.')
    >>> print rcs.name
    Arch
    >>> execute([])
    Using Arch for revision control.
    Directory initialized.
    >>> rcs.cleanup()

    >>> try:
    ...     execute(['.'])
    ... except cmdutil.UserError, e:
    ...     str(e).startswith("Directory already initialized: ")
    True
    >>> execute(['/highly-unlikely-to-exist'])
    Traceback (most recent call last):
    UserError: No such directory: /highly-unlikely-to-exist
    >>> os.chdir('/')
    """
    options, args = get_parser().parse_args(args)
    if len(args) > 1:
        print help()
        raise cmdutil.UserError, "Too many arguments"
    if len(args) == 1:
        basedir = args[0]
    else:
        basedir = "."
    if os.path.exists(basedir) == False:
        pass
        #raise cmdutil.UserError, "No such directory: %s" % basedir
    try:
        bd = bugdir.BugDir(basedir, from_disk=False, sink_to_existing_root=False, assert_new_BugDir=True)
    except bugdir.NoRootEntry:
        raise cmdutil.UserError("No such directory: %s" % basedir)
    except bugdir.AlreadyInitialized:
        raise cmdutil.UserError("Directory already initialized: %s" % basedir)
    bd.save()
    if bd.rcs.name is not "None":
        print "Using %s for revision control." % bd.rcs.name
    else:
        print "No revision control detected."
    print "Directory initialized."

def get_parser():
    parser = cmdutil.CmdOptionParser("be set-root [DIRECTORY]")
    return parser

longhelp="""
This command initializes Bugs Everywhere support for the specified directory
and all its subdirectories.  It will auto-detect any supported revision control
system.  You can use "be set rcs_name" to change the rcs being used.

The directory defaults to your current working directory.

It is usually a good idea to put the Bugs Everywhere root at the source code
root, but you can put it anywhere.  If you run "be set-root" in a subdirectory,
then only bugs created in that subdirectory (and its children) will appear
there.
"""

def help():
    return get_parser().help_str() + longhelp