/brz/remove-bazaar

    S_ISGITLINK,

import re

    lru_cache,

    BzrError,

    TreeReference,

    mode_is_executable,

    squash_revision,

    warn_unusual_mode,

    GitRepository,

MAX_INV_CACHE_SIZE = 50 * 1024 * 1024

 

 

def import_git_blob(texts, mapping, path, hexsha, base_inv, base_ie, parent_id,

    if base_ie is None:

            base_sha = shagitmap.lookup_blob(file_id, base_ie.revision)

    if base_sha == hexsha and base_ie.kind == ie.kind:

        if ie.executable == base_ie.executable:

            ie.revision = base_ie.revision

        else:

            blob = lookup_object(hexsha)

            ie.revision = None

        if pie.text_sha1 == ie.text_sha1 and pie.executable == ie.executable and pie.symlink_target == ie.symlink_target:

        if ie.kind == 'symlink':

            data = ''

        else: 

            data = blob.data

        texts.insert_record_stream([FulltextContentFactory((file_id, ie.revision), tuple(parent_keys), ie.text_sha1, data)])

    invdelta = []

    if base_ie is not None:

        if base_ie.kind == "directory":

            invdelta.extend(remove_disappeared_children(old_path, base_ie.children, []))

    invdelta.append((old_path, path, file_id, ie))

    return (invdelta, shamap)

 

 

class SubmodulesRequireSubtrees(BzrError):

    _fmt = """The repository you are fetching from contains submodules. To continue, upgrade your Bazaar repository to a format that supports nested trees, such as 'development-subtree'."""

    internal = False

 

 

def import_git_submodule(texts, mapping, path, hexsha, base_inv, base_ie,

    parent_id, revision_id, parent_invs, shagitmap, lookup_object):

    file_id = mapping.generate_file_id(path)

    ie = TreeReference(file_id, urlutils.basename(path.decode("utf-8")),

        parent_id)

    ie.revision = revision_id

    if base_ie is None:

        oldpath = None

    else:

        oldpath = path

        if (base_ie.kind == ie.kind and

            base_ie.reference_revision == ie.reference_revision):

            ie.revision = base_ie.revision

    ie.reference_revision = mapping.revision_id_foreign_to_bzr(hexsha)

    texts.insert_record_stream([FulltextContentFactory((file_id, ie.revision), (), None, "")])

    invdelta = [(oldpath, path, file_id, ie)]

    return invdelta, {}, {}

 

 

def remove_disappeared_children(path, base_children, existing_children):

    ret = []

    deletable = [(osutils.pathjoin(path, k), v) for k,v in base_children.iteritems() if k not in existing_children]

    while deletable:

        (path, ie) = deletable.pop()

        ret.append((path, None, ie.file_id, None))

        if ie.kind == "directory":

            for name, child_ie in ie.children.iteritems():

                deletable.append((osutils.pathjoin(path, name), child_ie))

    return ret

 

 

def import_git_tree(texts, mapping, path, hexsha, base_inv, base_ie, parent_id,

    revision_id, parent_invs, shagitmap, lookup_object, allow_submodules=False):

    ie = InventoryDirectory(file_id, urlutils.basename(path.decode("utf-8")),

    if base_ie is None:

        texts.insert_record_stream([FulltextContentFactory((file_id, ie.revision), (), None, "")])

            base_sha = shagitmap.lookup_tree(file_id, base_inv.revision_id)

        if base_ie.kind != "directory":

            ie.revision = revision_id

            texts.insert_record_stream([FulltextContentFactory((ie.file_id, ie.revision), (), None, "")])

            invdelta.append((base_inv.id2path(ie.file_id), path, ie.file_id, ie))

    if base_ie is not None and base_ie.kind == "directory":

        base_children = base_ie.children

    else:

        base_children = {}

            subinvdelta, grandchildmodes, subshamap = import_git_tree(

                    texts, mapping, child_path, child_hexsha, base_inv,

                    base_children.get(basename), file_id, revision_id,

                    parent_invs, shagitmap, lookup_object,

                    allow_submodules=allow_submodules)

            invdelta.extend(subinvdelta)

            child_modes.update(grandchildmodes)

            shamap.extend(subshamap)

        elif S_ISGITLINK(mode): # submodule

            if not allow_submodules:

                raise SubmodulesRequireSubtrees()

            subinvdelta, grandchildmodes, subshamap = import_git_submodule(

                    texts, mapping, child_path, child_hexsha, base_inv, base_children.get(basename),

                    file_id, revision_id, parent_invs, shagitmap, lookup_object)

            subinvdelta, subshamap = import_git_blob(texts, mapping,

                    child_path, child_hexsha, base_inv, base_children.get(basename), file_id,

                    revision_id, parent_invs, shagitmap, lookup_object,

                    mode_is_executable(mode), stat.S_ISLNK(mode))

    if base_ie is not None and base_ie.kind == "directory":

        invdelta.extend(remove_disappeared_children(base_inv.id2path(file_id),

            base_children, existing_children))

def approx_inv_size(inv):

    # Very rough estimate, 1k per inventory entry

    return len(inv) * 1024

 

 

def import_git_commit(repo, mapping, head, lookup_object,

                      target_git_object_retriever, parent_invs_cache):

    o = lookup_object(head)

    rev = mapping.import_commit(o)

    # We have to do this here, since we have to walk the tree and

    # we need to make sure to import the blobs / trees with the right

    # path; this may involve adding them more than once.

    parent_invs = []

    for parent_id in rev.parent_ids:

        try:

            parent_invs.append(parent_invs_cache[parent_id])

        except KeyError:

            parent_inv = repo.get_inventory(parent_id)

            parent_invs.append(parent_inv)

            parent_invs_cache[parent_id] = parent_inv

    if parent_invs == []:

        base_inv = Inventory(root_id=None)

        base_ie = None

    else:

        base_inv = parent_invs[0]

        base_ie = base_inv.root

    inv_delta, unusual_modes, shamap = import_git_tree(repo.texts,

            mapping, "", o.tree, base_inv, base_ie, None, rev.revision_id,

            parent_invs, target_git_object_retriever._idmap, lookup_object,

            allow_submodules=getattr(repo._format, "supports_tree_reference", False))

    target_git_object_retriever._idmap.add_entries(shamap)

    if unusual_modes != {}:

        for path, mode in unusual_modes.iteritems():

            warn_unusual_mode(rev.foreign_revid, path, mode)

        mapping.import_unusual_file_modes(rev, unusual_modes)

    try:

        basis_id = rev.parent_ids[0]

    except IndexError:

        basis_id = NULL_REVISION

        base_inv = None

    rev.inventory_sha1, inv = repo.add_inventory_by_delta(basis_id,

              inv_delta, rev.revision_id, rev.parent_ids,

              base_inv)

    parent_invs_cache[rev.revision_id] = inv

    repo.add_revision(rev.revision_id, rev)

    if "verify" in debug.debug_flags:

        new_unusual_modes = mapping.export_unusual_file_modes(rev)

        if new_unusual_modes != unusual_modes:

            raise AssertionError("unusual modes don't match: %r != %r" % (unusual_modes, new_unusual_modes))

        objs = inventory_to_tree_and_blobs(inv, repo.texts, mapping, unusual_modes)

        for sha1, newobj, path in objs:

            assert path is not None

            oldobj = tree_lookup_path(lookup_object, o.tree, path)

            if oldobj != newobj:

                raise AssertionError("%r != %r in %s" % (oldobj, newobj, path))

 

 

def import_git_objects(repo, mapping, object_iter, target_git_object_retriever,

    :param repo: Target Bazaar repository

    target_git_object_retriever._idmap.start_write_group() # FIXME: try/finally

    def lookup_object(sha):

        try:

            return object_iter[sha]

        except KeyError:

            return target_git_object_retriever[sha]

    parent_invs_cache = lru_cache.LRUSizeCache(compute_size=approx_inv_size,

                                               max_size=MAX_INV_CACHE_SIZE)

            o = lookup_object(head)

            trace.mutter('missing head %s', head)

            squash_revision(repo, rev)

            graph.append((o.id, o.parents))

            target_git_object_retriever._idmap.add_entry(o.id, "commit",

        checked.add(o.id)

    del checked

    batch_size = 100

    revision_ids = topo_sort(graph)

    pack_hints = []

    for offset in range(0, len(revision_ids), batch_size):

        repo.start_write_group()

            for i, head in enumerate(revision_ids[offset:offset+batch_size]):

                if pb is not None:

                    pb.update("fetching revisions", offset+i, len(revision_ids))

                import_git_commit(repo, mapping, head, lookup_object,

                                  target_git_object_retriever,

                                  parent_invs_cache)

        except:

            repo.abort_write_group()

            raise

        else:

            hint = repo.commit_write_group()

            if hint is not None:

                pack_hints.extend(hint)

    target_git_object_retriever._idmap.commit_write_group()

    return pack_hints

    """Base InterRepository that copies revisions from a Git into a non-Git

    def fetch_refs(self, revision_id=None, pb=None, find_ghosts=False,

                ret = [mapping.revision_id_bzr_to_foreign(revid)[0] for revid in interesting_heads if revid not in (None, NULL_REVISION)]

        pack_hint = self.fetch_objects(determine_wants, mapping, pb)

        if pack_hint is not None and self.target._format.pack_compresses:

            self.target.pack(hint=pack_hint)

        if interesting_heads is not None:

            present_interesting_heads = self.target.has_revisions(interesting_heads)

            missing_interesting_heads = set(interesting_heads) - present_interesting_heads

            if missing_interesting_heads:

                raise AssertionError("Missing interesting heads: %r" % missing_interesting_heads)

_GIT_PROGRESS_RE = re.compile(r"(.*?): +(\d+)% \((\d+)/(\d+)\)")

def report_git_progress(pb, text):

    text = text.rstrip("\r\n")

    g = _GIT_PROGRESS_RE.match(text)

    if g is not None:

        (text, pct, current, total) = g.groups()

        pb.update(text, int(current), int(total))

    else:

        pb.update(text, 0, 0)

 

    """InterRepository that copies revisions from a remote Git into a non-Git

    def get_target_heads(self):

        # FIXME: This should be more efficient

        all_revs = self.target.all_revision_ids()

        parent_map = self.target.get_parent_map(all_revs)

        all_parents = set()

        map(all_parents.update, parent_map.itervalues())

        return set(all_revs) - all_parents

 

            report_git_progress(pb, text)

        store = BazaarObjectStore(self.target, mapping)

        self.target.lock_write()

            heads = self.get_target_heads()

            graph_walker = store.get_graph_walker(

                    [store._lookup_revision_sha1(head) for head in heads])

            recorded_wants = []

 

            def record_determine_wants(heads):

                wants = determine_wants(heads)

                recorded_wants.extend(wants)

                return wants

 

            create_pb = None

            if pb is None:

                create_pb = pb = ui.ui_factory.nested_progress_bar()

                objects_iter = self.source.fetch_objects(

                            record_determine_wants, graph_walker,

                            store.get_raw, progress)

                return import_git_objects(self.target, mapping,

                    objects_iter, store, recorded_wants, pb)

                if create_pb:

                    create_pb.finished()

            self.target.unlock()

        return (isinstance(source, RemoteGitRepository) and

    """InterRepository that copies revisions from a local Git into a non-Git

                return import_git_objects(self.target, mapping,

                    self.source._git.object_store, target_git_object_retriever,

                    wants, pb)

        return (isinstance(source, LocalGitRepository) and

    def fetch_objects(self, determine_wants, mapping, pb=None):

        def progress(text):

            trace.note("git: %s", text)

        graphwalker = self.target._git.get_graph_walker()

        if (isinstance(self.source, LocalGitRepository) and

            isinstance(self.target, LocalGitRepository)):

            return self.source._git.fetch(self.target._git, determine_wants,

                progress)

        elif (isinstance(self.source, LocalGitRepository) and

              isinstance(self.target, RemoteGitRepository)):

            raise NotImplementedError

        elif (isinstance(self.source, RemoteGitRepository) and

              isinstance(self.target, LocalGitRepository)):

            f, commit = self.target._git.object_store.add_thin_pack()

            try:

                refs = self.source._git.fetch_pack(determine_wants,

                    graphwalker, f.write, progress)

                commit()

                return refs

            except:

                f.close()

                raise

        else:

            raise AssertionError

 

    def fetch_refs(self, revision_id=None, pb=None, find_ghosts=False,

        return self.fetch_objects(determine_wants, mapping)

        return (isinstance(source, GitRepository) and