quickrenamecls.py

#!/usr/bin/python

# Textually rename classes

# This is an alternative to ApplySrg2Source using a classes srg,
# for when your computer can't handle it

# Requires some manual fixing afterwards, but its MUCH faster

import sys, os, re, subprocess

def filenameRenames(renames, srcdir):
    pushd = os.getcwd()
    os.chdir(srcdir + "main/java/")  # have to be in root for git 
    if not os.path.exists("net/minecraft/src"): os.mkdir("net/minecraft/src/")
    for before in sorted(renames.keys()):
        after = renames[before]

        if before == after: continue

        cmd = ("git", "mv", before + ".java", after + ".java")
        print " ".join(cmd)
        subprocess.call(cmd)
    os.chdir(pushd)


# Rename classes in files, by text
def textualRename(renames, filenames):
    for filename in filenames:
        print filename
        data = file(filename, "r").read()

        tempPrefix = "cbtmp_"

        for before, after in renames.iteritems():
            data = re.sub(r"\b" + re.escape(lastComponent(before)) + r"\b", tempPrefix + lastComponent(after), data)
            data = re.sub(r"\b" + "net.minecraft.server" + r"\b", "net.minecraft.src", data)

        # To avoid renaming unintentionally transitive renames, we substitute a temporary prefix, do all
        # of the renames, then remove the prefix. This makes the rename operation "atomic".
        data = data.replace(tempPrefix, "")

        file(filename, "w").write(data)

def getJavaSource(srcdir):
    paths = []
    for root, dirs, files in os.walk(srcdir):
        for fn in files:
            if not fn.endswith(".java"): continue

            path = os.path.join(root, fn)
            paths.append(path)
    return paths


# Get class renames map
def getRenames(filename):
    renames = {}
    f = file(filename)
    ren = []
    for line in f.readlines():
        line = line.strip()
        tokens = line.strip().split()
        if tokens[0] != "CL:": continue
        args = tokens[1:]

        inName, outName= args

        renames[inName] = outName

    return renames


def lastComponent(fullName):
    return fullName.split("/")[-1]

def main():
    if len(sys.argv) != 3:
        print "Usage: %s cb2mcp-only-classes.srg ../CraftBukkit/src/" % (sys.argv[0],)
        raise SystemExit

    renames = getRenames(sys.argv[1])
    filenames = getJavaSource(sys.argv[2])

    print "Renaming class references..."
    textualRename(renames, filenames)
    print "Commit to git now, then press enter once committed to continue"
    raw_input()
    print "Renaming files"
    filenameRenames(renames, sys.argv[2])
    print "Rename complete! Commit to git, then fix any errors (org.bukkit.Potion vs net.minecraft.src.Potion, etc.)"
    print "Note, you must now use git-log --follow to view history of any of the renames files"

if __name__ == "__main__":
    main()