/b-gtk/fix-viz

# Copyright (C) 2006 by Szilveszter Farkas (Phanatic)

# Some parts of the code are:

# Copyright (C) 2005, 2006 by Canonical Ltd

# 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

import errno

import os

import bzrlib.errors as errors

from errors import (AlreadyBranchError, BranchExistsWithoutWorkingTree,

                    NonExistingParent, NonExistingRevision, NonExistingSource,

                    RevisionValueError, TargetAlreadyExists)

def init(location):

    from bzrlib.builtins import get_format_type

    import bzrlib.bzrdir as bzrdir

    format = get_format_type('default')

    if not os.path.exists(location):

        os.mkdir(location)

    try:

        existing_bzrdir = bzrdir.BzrDir.open(location)

    except errors.NotBranchError:

        bzrdir.BzrDir.create_branch_convenience(location, format=format)

    else:

        if existing_bzrdir.has_branch():

            if existing_bzrdir.has_workingtree():

                raise AlreadyBranchError(location)

            else:

                raise BranchExistsWithoutWorkingTree(location)

        else:

            existing_bzrdir.create_branch()

            existing_bzrdir.create_workingtree()

def branch(from_location, to_location, revision=None):

    from bzrlib.branch import Branch

    from bzrlib.transport import get_transport

    if revision is None:

        revision = [None]

    elif len(revision) > 1:

        raise RevisionValueError('bzr branch --revision takes exactly 1 revision value')

    try:

        br_from = Branch.open(from_location)

    except OSError, e:

        if e.errno == errno.ENOENT:

            raise NonExistingSource(from_location)

        else:

            raise

    br_from.lock_read()

    try:

        basis_dir = None

        if len(revision) == 1 and revision[0] is not None:

            revision_id = revision[0].in_history(br_from)[1]

        else:

            revision_id = br_from.last_revision()

        to_location = to_location + '/' + os.path.basename(from_location.rstrip("/\\"))

        name = None

        to_transport = get_transport(to_location)

        try:

            to_transport.mkdir('.')

        except errors.FileExists:

            raise TargetAlreadyExists(to_location)

        except errors.NoSuchFile:

            raise NonExistingParent(to_location)

        try:

            dir = br_from.bzrdir.sprout(to_transport.base, revision_id, basis_dir)

            branch = dir.open_branch()

        except errors.NoSuchRevision:

            to_transport.delete_tree('.')

            raise NonExistingRevision(from_location, revision[0])

        if name:

            branch.control_files.put_utf8('branch-name', name)

    finally:

        br_from.unlock()

    return branch.revno()

def checkout():

    """ FIXME - will be implemented later """