/brz/remove-bazaar

from bzrlib.lazy_import import lazy_import

from itertools import izip

from StringIO import StringIO

import sys

import time

lazy_import(globals(), """

from bzrlib import commands

from bzrlib.tuned_gzip import GzipFile

from bzrlib.workingtree import WorkingTree

from bzrlib.tests import TestUtil

from bzrlib import urlutils

from bzrlib.plugins.multiparent.multiparent import MultiVersionedFile

""")

class cmd_mp_regen(commands.Command):

    """Generate a multiparent versionedfile"""

    takes_args = ['file?']

    takes_options = [commands.Option('sync-snapshots',

                                     help='Snapshots follow source'),

                     commands.Option('snapshot-interval', type=int,

                                     help='take snapshots every x revisions'),

                     commands.Option('lsprof-timed', help='Use lsprof'),

                     commands.Option('dump',

                                     help='dump pseudo-knit to stdout'),

                     commands.Option('extract', help='test extract time'),

                     commands.Option('single', help='use a single parent'),

                     commands.Option('verify', help='verify added texts'),

                    ]

    hidden = True

    def run(self, file=None, sync_snapshots=False, snapshot_interval=26,

            lsprof_timed=False, dump=False, extract=False, single=False,

            verify=False):

        if file is None:

            wt, path = WorkingTree.open_containing('.')

            file_weave = wt.branch.repository.get_inventory_weave()

        else:

            wt, path = WorkingTree.open_containing(file)

            file_id = wt.path2id(path)

            bt = wt.branch.repository.revision_tree(wt.last_revision())

            file_weave = bt.get_weave(file_id)

        url = file_weave.transport.abspath(file_weave.filename)

        print >> sys.stderr, 'Importing: %s' % \

            urlutils.local_path_from_url(url)

        if sync_snapshots:

            print >> sys.stderr, 'Snapshots follow input'

        else:

            print >> sys.stderr, 'Snapshot interval: %d' % snapshot_interval

        vf = MultiVersionedFile(snapshot_interval)

        snapshots = set(r for r in file_weave.versions() if

                        file_weave._index.get_method(r) == 'fulltext')

        if sync_snapshots:

            to_sync = snapshots

        else:

            to_sync = vf.select_snapshots(file_weave)

        print >> sys.stderr, "%d fulltexts" % len(snapshots)

        print >> sys.stderr, "%d snapshots" % len(to_sync)

        vf.import_versionedfile(file_weave, to_sync, single_parent=single,

                                verify=verify)

        vf.clear_cache()

        if False:

            for revision_id in file_weave.get_ancestry(

                [bt.inventory[file_id].revision]):

                if vf.get_line_list([revision_id])[0] != \

                    file_weave.get_lines(revision_id):

                    open(revision_id + '.old', 'wb').writelines(

                        file_weave.get_lines(revision_id))

                    open(revision_id + '.new', 'wb').writelines(

                        vf.get_line_list(revision_id)[0])

        if extract:

            revisions = file_weave.versions()[-1:]

            if lsprof_timed:

                from bzrlib.lsprof import profile

                ret, stats = profile(vf.get_line_list, revisions)

                stats.sort()

                stats.pprint()

            start = time.clock()

            print >> sys.stderr, revisions[0]

            for x in range(1000):

                vf.clear_cache()

                vf.get_line_list(revisions)

            print >> sys.stderr, time.clock() - start

            start = time.clock()

            for x in range(1000):

                file_weave.get_line_list(revisions)

            print >> sys.stderr, time.clock() - start

        if dump:

            revisions = file_weave.versions()

            for revision, diff in vf._diffs.iteritems():

                sio = StringIO()

                data_file = GzipFile(None, mode='wb', fileobj=sio)

                print >> data_file, 'version %s' % revision

                data_file.writelines(diff.to_patch())

                data_file.close()

                sys.stdout.write(sio.getvalue())

commands.register_command(cmd_mp_regen)

def test_suite():

    from bzrlib.plugins.multiparent import test_multiparent

    return TestUtil.TestLoader().loadTestsFromModule(test_multiparent)