diff svncommand.py @ 0:f2636cfed115

Initial import of hgsubversion into a public repository.
author Augie Fackler <durin42@gmail.com>
date Tue, 30 Sep 2008 11:42:52 -0500
parents
children 1a5bb173170b
line wrap: on
line diff
new file mode 100644
--- /dev/null
+++ b/svncommand.py
@@ -0,0 +1,155 @@
+import os
+import pickle
+import stat
+
+from mercurial import hg
+from mercurial import node
+
+import svnwrap
+import hg_delta_editor
+import util
+from util import register_subcommand, svn_subcommands
+# dirty trick to force demandimport to run my decorator anyway.
+from utility_commands import print_wc_url
+from fetch_command import fetch_revisions
+from push_cmd import commit_from_rev
+# shut up, pyflakes, we must import those
+__x = [print_wc_url, fetch_revisions, commit_from_rev, ]
+
+mode755 = (stat.S_IXUSR | stat.S_IXGRP| stat.S_IXOTH | stat.S_IRUSR |
+           stat.S_IRGRP| stat.S_IROTH | stat.S_IWUSR)
+mode644 = (stat.S_IRUSR | stat.S_IRGRP| stat.S_IROTH | stat.S_IWUSR)
+
+
+def svncmd(ui, repo, subcommand, *args, **opts):
+    if subcommand not in svn_subcommands:
+        candidates = []
+        for c in svn_subcommands:
+            if c.startswith(subcommand):
+                candidates.append(c)
+        if len(candidates) == 1:
+            subcommand = candidates[0]
+    path = os.path.dirname(repo.path)
+    try:
+        opts['svn_url'] = open(os.path.join(repo.path, 'svn', 'url')).read()
+        return svn_subcommands[subcommand](ui, args=args,
+                                           hg_repo_path=path,
+                                           repo=repo,
+                                           **opts)
+    except TypeError, e:
+        print e
+        print 'Bad arguments for subcommand %s' % subcommand
+    except KeyError, e:
+        print 'Unknown subcommand %s' % subcommand
+
+@register_subcommand('help')
+def help_command(ui, args=None, **opts):
+    """Get help on the subsubcommands.
+    """
+    if args and args[0] in svn_subcommands:
+        print svn_subcommands[args[0]].__doc__.strip()
+        return
+    print 'Valid commands:', ' '.join(sorted(svn_subcommands.keys()))
+
+@register_subcommand('gentags')
+def generate_hg_tags(ui, hg_repo_path, **opts):
+    """Save tags to .hg/localtags
+    """
+    hg_editor = hg_delta_editor.HgChangeReceiver(hg_repo_path, ui_=ui)
+    f = open(hg_editor.tag_info_file)
+    tag_info = pickle.load(f)
+    f = open(os.path.join(hg_repo_path, '.hg', 'localtags'), 'w')
+    for tag, source in tag_info.iteritems():
+        source_ha = hg_editor.get_parent_revision(source[1]+1, source[0])
+        f.write('%s tag:%s\n' % (node.hex(source_ha), tag))
+
+@register_subcommand('up')
+def update(ui, args, repo, clean=False, **opts):
+    """Update to a specified Subversion revision number.
+    """
+    assert len(args) == 1
+    rev = int(args[0])
+    path = os.path.join(repo.path, 'svn', 'rev_map')
+    answers = []
+    for k,v in pickle.load(open(path)).iteritems():
+        if k[0] == rev:
+            answers.append((v, k[1]))
+    if len(answers) == 1:
+        if clean:
+            return hg.clean(repo, answers[0][0])
+        return hg.update(repo, answers[0][0])
+    elif len(answers) == 0:
+        ui.status('Revision %s did not produce an hg revision.\n' % rev)
+        return 1
+    else:
+        ui.status('Non-ambiguous revision!\n')
+        ui.status('\n'.join(['%s on %s' % (node.hex(a[0]), a[1]) for a in
+                             answers]+['']))
+    return 1
+
+
+@register_subcommand('verify_revision')
+def verify_revision(ui, args, repo, force=False, **opts):
+    """Verify a single converted revision.
+    Note: This wipes your working copy and then exports the corresponding
+    Subversion into your working copy to verify. Use with caution.
+    """
+    assert len(args) == 1
+    if not force:
+        assert repo.status(ignored=True,
+                           unknown=True) == ([], [], [], [], [], [], [])
+    rev = int(args[0])
+    wc_path = os.path.dirname(repo.path)
+    svn_url = open(os.path.join(repo.path, 'svn', 'url')).read()
+    svn = svnwrap.SubversionRepo(svn_url)
+    util.wipe_all_files(wc_path)
+    if update(ui, args, repo, clean=True) == 0:
+        util.wipe_all_files(wc_path)
+        br = repo.dirstate.branch()
+        if br == 'default':
+            br = None
+        if br:
+            diff_path = 'branches/%s' % br
+        else:
+            diff_path = 'trunk'
+        svn.fetch_all_files_to_dir(diff_path, rev, wc_path)
+        stat = repo.status(unknown=True)
+        ignored = [s for s in stat[4]
+                   if '/.svn/'  not in s and not s.startswith('.svn/')]
+        stat = stat[0:4]
+        if stat != ([], [], [], [],) or ignored != []:
+            ui.status('Something is wrong with this revision.\n')
+            return 2
+        else:
+            ui.status('OK.\n')
+            return 0
+    return 1
+
+@register_subcommand('verify_all_revisions')
+def verify_all_revisions(ui, args, repo, **opts):
+    """Verify all the converted revisions, optionally starting at a revision.
+
+    Note: This is *extremely* abusive of the Subversion server. It exports every
+    revision of the code one revision at a time.
+    """
+    assert repo.status(ignored=True,
+                       unknown=True) == ([], [], [], [], [], [], [])
+    start_rev = 0
+    args = list(args)
+    if args:
+        start_rev = int(args.pop(0))
+    revmap_f = open(os.path.join(repo.path, 'svn', 'rev_map'))
+    revmap = pickle.load(revmap_f)
+    revs = sorted(revmap.keys())
+    for revnum, br in revs:
+        if revnum < start_rev:
+            continue
+        res = verify_revision(ui, [revnum], repo, force=True)
+        if res == 0:
+            print revnum, 'verfied'
+        elif res == 1:
+            print revnum, 'skipped'
+        else:
+            print revnum, 'failed'
+            return 1
+    return 0