contrib / fast-import / git-p4on commit git-p4: fix test for unsupported P4 Client Views (b10048d)
   1#!/usr/bin/env python
   2#
   3# git-p4.py -- A tool for bidirectional operation between a Perforce depot and git.
   4#
   5# Author: Simon Hausmann <simon@lst.de>
   6# Copyright: 2007 Simon Hausmann <simon@lst.de>
   7#            2007 Trolltech ASA
   8# License: MIT <http://www.opensource.org/licenses/mit-license.php>
   9#
  10
  11import optparse, sys, os, marshal, subprocess, shelve
  12import tempfile, getopt, os.path, time, platform
  13import re
  14
  15verbose = False
  16
  17
  18def p4_build_cmd(cmd):
  19    """Build a suitable p4 command line.
  20
  21    This consolidates building and returning a p4 command line into one
  22    location. It means that hooking into the environment, or other configuration
  23    can be done more easily.
  24    """
  25    real_cmd = ["p4"]
  26
  27    user = gitConfig("git-p4.user")
  28    if len(user) > 0:
  29        real_cmd += ["-u",user]
  30
  31    password = gitConfig("git-p4.password")
  32    if len(password) > 0:
  33        real_cmd += ["-P", password]
  34
  35    port = gitConfig("git-p4.port")
  36    if len(port) > 0:
  37        real_cmd += ["-p", port]
  38
  39    host = gitConfig("git-p4.host")
  40    if len(host) > 0:
  41        real_cmd += ["-h", host]
  42
  43    client = gitConfig("git-p4.client")
  44    if len(client) > 0:
  45        real_cmd += ["-c", client]
  46
  47
  48    if isinstance(cmd,basestring):
  49        real_cmd = ' '.join(real_cmd) + ' ' + cmd
  50    else:
  51        real_cmd += cmd
  52    return real_cmd
  53
  54def chdir(dir):
  55    # P4 uses the PWD environment variable rather than getcwd(). Since we're
  56    # not using the shell, we have to set it ourselves.  This path could
  57    # be relative, so go there first, then figure out where we ended up.
  58    os.chdir(dir)
  59    os.environ['PWD'] = os.getcwd()
  60
  61def die(msg):
  62    if verbose:
  63        raise Exception(msg)
  64    else:
  65        sys.stderr.write(msg + "\n")
  66        sys.exit(1)
  67
  68def write_pipe(c, stdin):
  69    if verbose:
  70        sys.stderr.write('Writing pipe: %s\n' % str(c))
  71
  72    expand = isinstance(c,basestring)
  73    p = subprocess.Popen(c, stdin=subprocess.PIPE, shell=expand)
  74    pipe = p.stdin
  75    val = pipe.write(stdin)
  76    pipe.close()
  77    if p.wait():
  78        die('Command failed: %s' % str(c))
  79
  80    return val
  81
  82def p4_write_pipe(c, stdin):
  83    real_cmd = p4_build_cmd(c)
  84    return write_pipe(real_cmd, stdin)
  85
  86def read_pipe(c, ignore_error=False):
  87    if verbose:
  88        sys.stderr.write('Reading pipe: %s\n' % str(c))
  89
  90    expand = isinstance(c,basestring)
  91    p = subprocess.Popen(c, stdout=subprocess.PIPE, shell=expand)
  92    pipe = p.stdout
  93    val = pipe.read()
  94    if p.wait() and not ignore_error:
  95        die('Command failed: %s' % str(c))
  96
  97    return val
  98
  99def p4_read_pipe(c, ignore_error=False):
 100    real_cmd = p4_build_cmd(c)
 101    return read_pipe(real_cmd, ignore_error)
 102
 103def read_pipe_lines(c):
 104    if verbose:
 105        sys.stderr.write('Reading pipe: %s\n' % str(c))
 106
 107    expand = isinstance(c, basestring)
 108    p = subprocess.Popen(c, stdout=subprocess.PIPE, shell=expand)
 109    pipe = p.stdout
 110    val = pipe.readlines()
 111    if pipe.close() or p.wait():
 112        die('Command failed: %s' % str(c))
 113
 114    return val
 115
 116def p4_read_pipe_lines(c):
 117    """Specifically invoke p4 on the command supplied. """
 118    real_cmd = p4_build_cmd(c)
 119    return read_pipe_lines(real_cmd)
 120
 121def system(cmd):
 122    expand = isinstance(cmd,basestring)
 123    if verbose:
 124        sys.stderr.write("executing %s\n" % str(cmd))
 125    subprocess.check_call(cmd, shell=expand)
 126
 127def p4_system(cmd):
 128    """Specifically invoke p4 as the system command. """
 129    real_cmd = p4_build_cmd(cmd)
 130    expand = isinstance(real_cmd, basestring)
 131    subprocess.check_call(real_cmd, shell=expand)
 132
 133def p4_integrate(src, dest):
 134    p4_system(["integrate", "-Dt", src, dest])
 135
 136def p4_sync(path):
 137    p4_system(["sync", path])
 138
 139def p4_add(f):
 140    p4_system(["add", f])
 141
 142def p4_delete(f):
 143    p4_system(["delete", f])
 144
 145def p4_edit(f):
 146    p4_system(["edit", f])
 147
 148def p4_revert(f):
 149    p4_system(["revert", f])
 150
 151def p4_reopen(type, file):
 152    p4_system(["reopen", "-t", type, file])
 153
 154#
 155# Canonicalize the p4 type and return a tuple of the
 156# base type, plus any modifiers.  See "p4 help filetypes"
 157# for a list and explanation.
 158#
 159def split_p4_type(p4type):
 160
 161    p4_filetypes_historical = {
 162        "ctempobj": "binary+Sw",
 163        "ctext": "text+C",
 164        "cxtext": "text+Cx",
 165        "ktext": "text+k",
 166        "kxtext": "text+kx",
 167        "ltext": "text+F",
 168        "tempobj": "binary+FSw",
 169        "ubinary": "binary+F",
 170        "uresource": "resource+F",
 171        "uxbinary": "binary+Fx",
 172        "xbinary": "binary+x",
 173        "xltext": "text+Fx",
 174        "xtempobj": "binary+Swx",
 175        "xtext": "text+x",
 176        "xunicode": "unicode+x",
 177        "xutf16": "utf16+x",
 178    }
 179    if p4type in p4_filetypes_historical:
 180        p4type = p4_filetypes_historical[p4type]
 181    mods = ""
 182    s = p4type.split("+")
 183    base = s[0]
 184    mods = ""
 185    if len(s) > 1:
 186        mods = s[1]
 187    return (base, mods)
 188
 189
 190def setP4ExecBit(file, mode):
 191    # Reopens an already open file and changes the execute bit to match
 192    # the execute bit setting in the passed in mode.
 193
 194    p4Type = "+x"
 195
 196    if not isModeExec(mode):
 197        p4Type = getP4OpenedType(file)
 198        p4Type = re.sub('^([cku]?)x(.*)', '\\1\\2', p4Type)
 199        p4Type = re.sub('(.*?\+.*?)x(.*?)', '\\1\\2', p4Type)
 200        if p4Type[-1] == "+":
 201            p4Type = p4Type[0:-1]
 202
 203    p4_reopen(p4Type, file)
 204
 205def getP4OpenedType(file):
 206    # Returns the perforce file type for the given file.
 207
 208    result = p4_read_pipe(["opened", file])
 209    match = re.match(".*\((.+)\)\r?$", result)
 210    if match:
 211        return match.group(1)
 212    else:
 213        die("Could not determine file type for %s (result: '%s')" % (file, result))
 214
 215def diffTreePattern():
 216    # This is a simple generator for the diff tree regex pattern. This could be
 217    # a class variable if this and parseDiffTreeEntry were a part of a class.
 218    pattern = re.compile(':(\d+) (\d+) (\w+) (\w+) ([A-Z])(\d+)?\t(.*?)((\t(.*))|$)')
 219    while True:
 220        yield pattern
 221
 222def parseDiffTreeEntry(entry):
 223    """Parses a single diff tree entry into its component elements.
 224
 225    See git-diff-tree(1) manpage for details about the format of the diff
 226    output. This method returns a dictionary with the following elements:
 227
 228    src_mode - The mode of the source file
 229    dst_mode - The mode of the destination file
 230    src_sha1 - The sha1 for the source file
 231    dst_sha1 - The sha1 fr the destination file
 232    status - The one letter status of the diff (i.e. 'A', 'M', 'D', etc)
 233    status_score - The score for the status (applicable for 'C' and 'R'
 234                   statuses). This is None if there is no score.
 235    src - The path for the source file.
 236    dst - The path for the destination file. This is only present for
 237          copy or renames. If it is not present, this is None.
 238
 239    If the pattern is not matched, None is returned."""
 240
 241    match = diffTreePattern().next().match(entry)
 242    if match:
 243        return {
 244            'src_mode': match.group(1),
 245            'dst_mode': match.group(2),
 246            'src_sha1': match.group(3),
 247            'dst_sha1': match.group(4),
 248            'status': match.group(5),
 249            'status_score': match.group(6),
 250            'src': match.group(7),
 251            'dst': match.group(10)
 252        }
 253    return None
 254
 255def isModeExec(mode):
 256    # Returns True if the given git mode represents an executable file,
 257    # otherwise False.
 258    return mode[-3:] == "755"
 259
 260def isModeExecChanged(src_mode, dst_mode):
 261    return isModeExec(src_mode) != isModeExec(dst_mode)
 262
 263def p4CmdList(cmd, stdin=None, stdin_mode='w+b', cb=None):
 264
 265    if isinstance(cmd,basestring):
 266        cmd = "-G " + cmd
 267        expand = True
 268    else:
 269        cmd = ["-G"] + cmd
 270        expand = False
 271
 272    cmd = p4_build_cmd(cmd)
 273    if verbose:
 274        sys.stderr.write("Opening pipe: %s\n" % str(cmd))
 275
 276    # Use a temporary file to avoid deadlocks without
 277    # subprocess.communicate(), which would put another copy
 278    # of stdout into memory.
 279    stdin_file = None
 280    if stdin is not None:
 281        stdin_file = tempfile.TemporaryFile(prefix='p4-stdin', mode=stdin_mode)
 282        if isinstance(stdin,basestring):
 283            stdin_file.write(stdin)
 284        else:
 285            for i in stdin:
 286                stdin_file.write(i + '\n')
 287        stdin_file.flush()
 288        stdin_file.seek(0)
 289
 290    p4 = subprocess.Popen(cmd,
 291                          shell=expand,
 292                          stdin=stdin_file,
 293                          stdout=subprocess.PIPE)
 294
 295    result = []
 296    try:
 297        while True:
 298            entry = marshal.load(p4.stdout)
 299            if cb is not None:
 300                cb(entry)
 301            else:
 302                result.append(entry)
 303    except EOFError:
 304        pass
 305    exitCode = p4.wait()
 306    if exitCode != 0:
 307        entry = {}
 308        entry["p4ExitCode"] = exitCode
 309        result.append(entry)
 310
 311    return result
 312
 313def p4Cmd(cmd):
 314    list = p4CmdList(cmd)
 315    result = {}
 316    for entry in list:
 317        result.update(entry)
 318    return result;
 319
 320def p4Where(depotPath):
 321    if not depotPath.endswith("/"):
 322        depotPath += "/"
 323    depotPath = depotPath + "..."
 324    outputList = p4CmdList(["where", depotPath])
 325    output = None
 326    for entry in outputList:
 327        if "depotFile" in entry:
 328            if entry["depotFile"] == depotPath:
 329                output = entry
 330                break
 331        elif "data" in entry:
 332            data = entry.get("data")
 333            space = data.find(" ")
 334            if data[:space] == depotPath:
 335                output = entry
 336                break
 337    if output == None:
 338        return ""
 339    if output["code"] == "error":
 340        return ""
 341    clientPath = ""
 342    if "path" in output:
 343        clientPath = output.get("path")
 344    elif "data" in output:
 345        data = output.get("data")
 346        lastSpace = data.rfind(" ")
 347        clientPath = data[lastSpace + 1:]
 348
 349    if clientPath.endswith("..."):
 350        clientPath = clientPath[:-3]
 351    return clientPath
 352
 353def currentGitBranch():
 354    return read_pipe("git name-rev HEAD").split(" ")[1].strip()
 355
 356def isValidGitDir(path):
 357    if (os.path.exists(path + "/HEAD")
 358        and os.path.exists(path + "/refs") and os.path.exists(path + "/objects")):
 359        return True;
 360    return False
 361
 362def parseRevision(ref):
 363    return read_pipe("git rev-parse %s" % ref).strip()
 364
 365def branchExists(ref):
 366    rev = read_pipe(["git", "rev-parse", "-q", "--verify", ref],
 367                     ignore_error=True)
 368    return len(rev) > 0
 369
 370def extractLogMessageFromGitCommit(commit):
 371    logMessage = ""
 372
 373    ## fixme: title is first line of commit, not 1st paragraph.
 374    foundTitle = False
 375    for log in read_pipe_lines("git cat-file commit %s" % commit):
 376       if not foundTitle:
 377           if len(log) == 1:
 378               foundTitle = True
 379           continue
 380
 381       logMessage += log
 382    return logMessage
 383
 384def extractSettingsGitLog(log):
 385    values = {}
 386    for line in log.split("\n"):
 387        line = line.strip()
 388        m = re.search (r"^ *\[git-p4: (.*)\]$", line)
 389        if not m:
 390            continue
 391
 392        assignments = m.group(1).split (':')
 393        for a in assignments:
 394            vals = a.split ('=')
 395            key = vals[0].strip()
 396            val = ('='.join (vals[1:])).strip()
 397            if val.endswith ('\"') and val.startswith('"'):
 398                val = val[1:-1]
 399
 400            values[key] = val
 401
 402    paths = values.get("depot-paths")
 403    if not paths:
 404        paths = values.get("depot-path")
 405    if paths:
 406        values['depot-paths'] = paths.split(',')
 407    return values
 408
 409def gitBranchExists(branch):
 410    proc = subprocess.Popen(["git", "rev-parse", branch],
 411                            stderr=subprocess.PIPE, stdout=subprocess.PIPE);
 412    return proc.wait() == 0;
 413
 414_gitConfig = {}
 415def gitConfig(key, args = None): # set args to "--bool", for instance
 416    if not _gitConfig.has_key(key):
 417        argsFilter = ""
 418        if args != None:
 419            argsFilter = "%s " % args
 420        cmd = "git config %s%s" % (argsFilter, key)
 421        _gitConfig[key] = read_pipe(cmd, ignore_error=True).strip()
 422    return _gitConfig[key]
 423
 424def gitConfigList(key):
 425    if not _gitConfig.has_key(key):
 426        _gitConfig[key] = read_pipe("git config --get-all %s" % key, ignore_error=True).strip().split(os.linesep)
 427    return _gitConfig[key]
 428
 429def p4BranchesInGit(branchesAreInRemotes = True):
 430    branches = {}
 431
 432    cmdline = "git rev-parse --symbolic "
 433    if branchesAreInRemotes:
 434        cmdline += " --remotes"
 435    else:
 436        cmdline += " --branches"
 437
 438    for line in read_pipe_lines(cmdline):
 439        line = line.strip()
 440
 441        ## only import to p4/
 442        if not line.startswith('p4/') or line == "p4/HEAD":
 443            continue
 444        branch = line
 445
 446        # strip off p4
 447        branch = re.sub ("^p4/", "", line)
 448
 449        branches[branch] = parseRevision(line)
 450    return branches
 451
 452def findUpstreamBranchPoint(head = "HEAD"):
 453    branches = p4BranchesInGit()
 454    # map from depot-path to branch name
 455    branchByDepotPath = {}
 456    for branch in branches.keys():
 457        tip = branches[branch]
 458        log = extractLogMessageFromGitCommit(tip)
 459        settings = extractSettingsGitLog(log)
 460        if settings.has_key("depot-paths"):
 461            paths = ",".join(settings["depot-paths"])
 462            branchByDepotPath[paths] = "remotes/p4/" + branch
 463
 464    settings = None
 465    parent = 0
 466    while parent < 65535:
 467        commit = head + "~%s" % parent
 468        log = extractLogMessageFromGitCommit(commit)
 469        settings = extractSettingsGitLog(log)
 470        if settings.has_key("depot-paths"):
 471            paths = ",".join(settings["depot-paths"])
 472            if branchByDepotPath.has_key(paths):
 473                return [branchByDepotPath[paths], settings]
 474
 475        parent = parent + 1
 476
 477    return ["", settings]
 478
 479def createOrUpdateBranchesFromOrigin(localRefPrefix = "refs/remotes/p4/", silent=True):
 480    if not silent:
 481        print ("Creating/updating branch(es) in %s based on origin branch(es)"
 482               % localRefPrefix)
 483
 484    originPrefix = "origin/p4/"
 485
 486    for line in read_pipe_lines("git rev-parse --symbolic --remotes"):
 487        line = line.strip()
 488        if (not line.startswith(originPrefix)) or line.endswith("HEAD"):
 489            continue
 490
 491        headName = line[len(originPrefix):]
 492        remoteHead = localRefPrefix + headName
 493        originHead = line
 494
 495        original = extractSettingsGitLog(extractLogMessageFromGitCommit(originHead))
 496        if (not original.has_key('depot-paths')
 497            or not original.has_key('change')):
 498            continue
 499
 500        update = False
 501        if not gitBranchExists(remoteHead):
 502            if verbose:
 503                print "creating %s" % remoteHead
 504            update = True
 505        else:
 506            settings = extractSettingsGitLog(extractLogMessageFromGitCommit(remoteHead))
 507            if settings.has_key('change') > 0:
 508                if settings['depot-paths'] == original['depot-paths']:
 509                    originP4Change = int(original['change'])
 510                    p4Change = int(settings['change'])
 511                    if originP4Change > p4Change:
 512                        print ("%s (%s) is newer than %s (%s). "
 513                               "Updating p4 branch from origin."
 514                               % (originHead, originP4Change,
 515                                  remoteHead, p4Change))
 516                        update = True
 517                else:
 518                    print ("Ignoring: %s was imported from %s while "
 519                           "%s was imported from %s"
 520                           % (originHead, ','.join(original['depot-paths']),
 521                              remoteHead, ','.join(settings['depot-paths'])))
 522
 523        if update:
 524            system("git update-ref %s %s" % (remoteHead, originHead))
 525
 526def originP4BranchesExist():
 527        return gitBranchExists("origin") or gitBranchExists("origin/p4") or gitBranchExists("origin/p4/master")
 528
 529def p4ChangesForPaths(depotPaths, changeRange):
 530    assert depotPaths
 531    cmd = ['changes']
 532    for p in depotPaths:
 533        cmd += ["%s...%s" % (p, changeRange)]
 534    output = p4_read_pipe_lines(cmd)
 535
 536    changes = {}
 537    for line in output:
 538        changeNum = int(line.split(" ")[1])
 539        changes[changeNum] = True
 540
 541    changelist = changes.keys()
 542    changelist.sort()
 543    return changelist
 544
 545def p4PathStartsWith(path, prefix):
 546    # This method tries to remedy a potential mixed-case issue:
 547    #
 548    # If UserA adds  //depot/DirA/file1
 549    # and UserB adds //depot/dira/file2
 550    #
 551    # we may or may not have a problem. If you have core.ignorecase=true,
 552    # we treat DirA and dira as the same directory
 553    ignorecase = gitConfig("core.ignorecase", "--bool") == "true"
 554    if ignorecase:
 555        return path.lower().startswith(prefix.lower())
 556    return path.startswith(prefix)
 557
 558class Command:
 559    def __init__(self):
 560        self.usage = "usage: %prog [options]"
 561        self.needsGit = True
 562
 563class P4UserMap:
 564    def __init__(self):
 565        self.userMapFromPerforceServer = False
 566
 567    def getUserCacheFilename(self):
 568        home = os.environ.get("HOME", os.environ.get("USERPROFILE"))
 569        return home + "/.gitp4-usercache.txt"
 570
 571    def getUserMapFromPerforceServer(self):
 572        if self.userMapFromPerforceServer:
 573            return
 574        self.users = {}
 575        self.emails = {}
 576
 577        for output in p4CmdList("users"):
 578            if not output.has_key("User"):
 579                continue
 580            self.users[output["User"]] = output["FullName"] + " <" + output["Email"] + ">"
 581            self.emails[output["Email"]] = output["User"]
 582
 583
 584        s = ''
 585        for (key, val) in self.users.items():
 586            s += "%s\t%s\n" % (key.expandtabs(1), val.expandtabs(1))
 587
 588        open(self.getUserCacheFilename(), "wb").write(s)
 589        self.userMapFromPerforceServer = True
 590
 591    def loadUserMapFromCache(self):
 592        self.users = {}
 593        self.userMapFromPerforceServer = False
 594        try:
 595            cache = open(self.getUserCacheFilename(), "rb")
 596            lines = cache.readlines()
 597            cache.close()
 598            for line in lines:
 599                entry = line.strip().split("\t")
 600                self.users[entry[0]] = entry[1]
 601        except IOError:
 602            self.getUserMapFromPerforceServer()
 603
 604class P4Debug(Command):
 605    def __init__(self):
 606        Command.__init__(self)
 607        self.options = [
 608            optparse.make_option("--verbose", dest="verbose", action="store_true",
 609                                 default=False),
 610            ]
 611        self.description = "A tool to debug the output of p4 -G."
 612        self.needsGit = False
 613        self.verbose = False
 614
 615    def run(self, args):
 616        j = 0
 617        for output in p4CmdList(args):
 618            print 'Element: %d' % j
 619            j += 1
 620            print output
 621        return True
 622
 623class P4RollBack(Command):
 624    def __init__(self):
 625        Command.__init__(self)
 626        self.options = [
 627            optparse.make_option("--verbose", dest="verbose", action="store_true"),
 628            optparse.make_option("--local", dest="rollbackLocalBranches", action="store_true")
 629        ]
 630        self.description = "A tool to debug the multi-branch import. Don't use :)"
 631        self.verbose = False
 632        self.rollbackLocalBranches = False
 633
 634    def run(self, args):
 635        if len(args) != 1:
 636            return False
 637        maxChange = int(args[0])
 638
 639        if "p4ExitCode" in p4Cmd("changes -m 1"):
 640            die("Problems executing p4");
 641
 642        if self.rollbackLocalBranches:
 643            refPrefix = "refs/heads/"
 644            lines = read_pipe_lines("git rev-parse --symbolic --branches")
 645        else:
 646            refPrefix = "refs/remotes/"
 647            lines = read_pipe_lines("git rev-parse --symbolic --remotes")
 648
 649        for line in lines:
 650            if self.rollbackLocalBranches or (line.startswith("p4/") and line != "p4/HEAD\n"):
 651                line = line.strip()
 652                ref = refPrefix + line
 653                log = extractLogMessageFromGitCommit(ref)
 654                settings = extractSettingsGitLog(log)
 655
 656                depotPaths = settings['depot-paths']
 657                change = settings['change']
 658
 659                changed = False
 660
 661                if len(p4Cmd("changes -m 1 "  + ' '.join (['%s...@%s' % (p, maxChange)
 662                                                           for p in depotPaths]))) == 0:
 663                    print "Branch %s did not exist at change %s, deleting." % (ref, maxChange)
 664                    system("git update-ref -d %s `git rev-parse %s`" % (ref, ref))
 665                    continue
 666
 667                while change and int(change) > maxChange:
 668                    changed = True
 669                    if self.verbose:
 670                        print "%s is at %s ; rewinding towards %s" % (ref, change, maxChange)
 671                    system("git update-ref %s \"%s^\"" % (ref, ref))
 672                    log = extractLogMessageFromGitCommit(ref)
 673                    settings =  extractSettingsGitLog(log)
 674
 675
 676                    depotPaths = settings['depot-paths']
 677                    change = settings['change']
 678
 679                if changed:
 680                    print "%s rewound to %s" % (ref, change)
 681
 682        return True
 683
 684class P4Submit(Command, P4UserMap):
 685    def __init__(self):
 686        Command.__init__(self)
 687        P4UserMap.__init__(self)
 688        self.options = [
 689                optparse.make_option("--verbose", dest="verbose", action="store_true"),
 690                optparse.make_option("--origin", dest="origin"),
 691                optparse.make_option("-M", dest="detectRenames", action="store_true"),
 692                # preserve the user, requires relevant p4 permissions
 693                optparse.make_option("--preserve-user", dest="preserveUser", action="store_true"),
 694        ]
 695        self.description = "Submit changes from git to the perforce depot."
 696        self.usage += " [name of git branch to submit into perforce depot]"
 697        self.interactive = True
 698        self.origin = ""
 699        self.detectRenames = False
 700        self.verbose = False
 701        self.preserveUser = gitConfig("git-p4.preserveUser").lower() == "true"
 702        self.isWindows = (platform.system() == "Windows")
 703        self.myP4UserId = None
 704
 705    def check(self):
 706        if len(p4CmdList("opened ...")) > 0:
 707            die("You have files opened with perforce! Close them before starting the sync.")
 708
 709    # replaces everything between 'Description:' and the next P4 submit template field with the
 710    # commit message
 711    def prepareLogMessage(self, template, message):
 712        result = ""
 713
 714        inDescriptionSection = False
 715
 716        for line in template.split("\n"):
 717            if line.startswith("#"):
 718                result += line + "\n"
 719                continue
 720
 721            if inDescriptionSection:
 722                if line.startswith("Files:") or line.startswith("Jobs:"):
 723                    inDescriptionSection = False
 724                else:
 725                    continue
 726            else:
 727                if line.startswith("Description:"):
 728                    inDescriptionSection = True
 729                    line += "\n"
 730                    for messageLine in message.split("\n"):
 731                        line += "\t" + messageLine + "\n"
 732
 733            result += line + "\n"
 734
 735        return result
 736
 737    def p4UserForCommit(self,id):
 738        # Return the tuple (perforce user,git email) for a given git commit id
 739        self.getUserMapFromPerforceServer()
 740        gitEmail = read_pipe("git log --max-count=1 --format='%%ae' %s" % id)
 741        gitEmail = gitEmail.strip()
 742        if not self.emails.has_key(gitEmail):
 743            return (None,gitEmail)
 744        else:
 745            return (self.emails[gitEmail],gitEmail)
 746
 747    def checkValidP4Users(self,commits):
 748        # check if any git authors cannot be mapped to p4 users
 749        for id in commits:
 750            (user,email) = self.p4UserForCommit(id)
 751            if not user:
 752                msg = "Cannot find p4 user for email %s in commit %s." % (email, id)
 753                if gitConfig('git-p4.allowMissingP4Users').lower() == "true":
 754                    print "%s" % msg
 755                else:
 756                    die("Error: %s\nSet git-p4.allowMissingP4Users to true to allow this." % msg)
 757
 758    def lastP4Changelist(self):
 759        # Get back the last changelist number submitted in this client spec. This
 760        # then gets used to patch up the username in the change. If the same
 761        # client spec is being used by multiple processes then this might go
 762        # wrong.
 763        results = p4CmdList("client -o")        # find the current client
 764        client = None
 765        for r in results:
 766            if r.has_key('Client'):
 767                client = r['Client']
 768                break
 769        if not client:
 770            die("could not get client spec")
 771        results = p4CmdList(["changes", "-c", client, "-m", "1"])
 772        for r in results:
 773            if r.has_key('change'):
 774                return r['change']
 775        die("Could not get changelist number for last submit - cannot patch up user details")
 776
 777    def modifyChangelistUser(self, changelist, newUser):
 778        # fixup the user field of a changelist after it has been submitted.
 779        changes = p4CmdList("change -o %s" % changelist)
 780        if len(changes) != 1:
 781            die("Bad output from p4 change modifying %s to user %s" %
 782                (changelist, newUser))
 783
 784        c = changes[0]
 785        if c['User'] == newUser: return   # nothing to do
 786        c['User'] = newUser
 787        input = marshal.dumps(c)
 788
 789        result = p4CmdList("change -f -i", stdin=input)
 790        for r in result:
 791            if r.has_key('code'):
 792                if r['code'] == 'error':
 793                    die("Could not modify user field of changelist %s to %s:%s" % (changelist, newUser, r['data']))
 794            if r.has_key('data'):
 795                print("Updated user field for changelist %s to %s" % (changelist, newUser))
 796                return
 797        die("Could not modify user field of changelist %s to %s" % (changelist, newUser))
 798
 799    def canChangeChangelists(self):
 800        # check to see if we have p4 admin or super-user permissions, either of
 801        # which are required to modify changelists.
 802        results = p4CmdList("protects %s" % self.depotPath)
 803        for r in results:
 804            if r.has_key('perm'):
 805                if r['perm'] == 'admin':
 806                    return 1
 807                if r['perm'] == 'super':
 808                    return 1
 809        return 0
 810
 811    def p4UserId(self):
 812        if self.myP4UserId:
 813            return self.myP4UserId
 814
 815        results = p4CmdList("user -o")
 816        for r in results:
 817            if r.has_key('User'):
 818                self.myP4UserId = r['User']
 819                return r['User']
 820        die("Could not find your p4 user id")
 821
 822    def p4UserIsMe(self, p4User):
 823        # return True if the given p4 user is actually me
 824        me = self.p4UserId()
 825        if not p4User or p4User != me:
 826            return False
 827        else:
 828            return True
 829
 830    def prepareSubmitTemplate(self):
 831        # remove lines in the Files section that show changes to files outside the depot path we're committing into
 832        template = ""
 833        inFilesSection = False
 834        for line in p4_read_pipe_lines(['change', '-o']):
 835            if line.endswith("\r\n"):
 836                line = line[:-2] + "\n"
 837            if inFilesSection:
 838                if line.startswith("\t"):
 839                    # path starts and ends with a tab
 840                    path = line[1:]
 841                    lastTab = path.rfind("\t")
 842                    if lastTab != -1:
 843                        path = path[:lastTab]
 844                        if not p4PathStartsWith(path, self.depotPath):
 845                            continue
 846                else:
 847                    inFilesSection = False
 848            else:
 849                if line.startswith("Files:"):
 850                    inFilesSection = True
 851
 852            template += line
 853
 854        return template
 855
 856    def edit_template(self, template_file):
 857        """Invoke the editor to let the user change the submission
 858           message.  Return true if okay to continue with the submit."""
 859
 860        # if configured to skip the editing part, just submit
 861        if gitConfig("git-p4.skipSubmitEdit") == "true":
 862            return True
 863
 864        # look at the modification time, to check later if the user saved
 865        # the file
 866        mtime = os.stat(template_file).st_mtime
 867
 868        # invoke the editor
 869        if os.environ.has_key("P4EDITOR"):
 870            editor = os.environ.get("P4EDITOR")
 871        else:
 872            editor = read_pipe("git var GIT_EDITOR").strip()
 873        system(editor + " " + template_file)
 874
 875        # If the file was not saved, prompt to see if this patch should
 876        # be skipped.  But skip this verification step if configured so.
 877        if gitConfig("git-p4.skipSubmitEditCheck") == "true":
 878            return True
 879
 880        # modification time updated means user saved the file
 881        if os.stat(template_file).st_mtime > mtime:
 882            return True
 883
 884        while True:
 885            response = raw_input("Submit template unchanged. Submit anyway? [y]es, [n]o (skip this patch) ")
 886            if response == 'y':
 887                return True
 888            if response == 'n':
 889                return False
 890
 891    def applyCommit(self, id):
 892        print "Applying %s" % (read_pipe("git log --max-count=1 --pretty=oneline %s" % id))
 893
 894        (p4User, gitEmail) = self.p4UserForCommit(id)
 895
 896        if not self.detectRenames:
 897            # If not explicitly set check the config variable
 898            self.detectRenames = gitConfig("git-p4.detectRenames")
 899
 900        if self.detectRenames.lower() == "false" or self.detectRenames == "":
 901            diffOpts = ""
 902        elif self.detectRenames.lower() == "true":
 903            diffOpts = "-M"
 904        else:
 905            diffOpts = "-M%s" % self.detectRenames
 906
 907        detectCopies = gitConfig("git-p4.detectCopies")
 908        if detectCopies.lower() == "true":
 909            diffOpts += " -C"
 910        elif detectCopies != "" and detectCopies.lower() != "false":
 911            diffOpts += " -C%s" % detectCopies
 912
 913        if gitConfig("git-p4.detectCopiesHarder", "--bool") == "true":
 914            diffOpts += " --find-copies-harder"
 915
 916        diff = read_pipe_lines("git diff-tree -r %s \"%s^\" \"%s\"" % (diffOpts, id, id))
 917        filesToAdd = set()
 918        filesToDelete = set()
 919        editedFiles = set()
 920        filesToChangeExecBit = {}
 921        for line in diff:
 922            diff = parseDiffTreeEntry(line)
 923            modifier = diff['status']
 924            path = diff['src']
 925            if modifier == "M":
 926                p4_edit(path)
 927                if isModeExecChanged(diff['src_mode'], diff['dst_mode']):
 928                    filesToChangeExecBit[path] = diff['dst_mode']
 929                editedFiles.add(path)
 930            elif modifier == "A":
 931                filesToAdd.add(path)
 932                filesToChangeExecBit[path] = diff['dst_mode']
 933                if path in filesToDelete:
 934                    filesToDelete.remove(path)
 935            elif modifier == "D":
 936                filesToDelete.add(path)
 937                if path in filesToAdd:
 938                    filesToAdd.remove(path)
 939            elif modifier == "C":
 940                src, dest = diff['src'], diff['dst']
 941                p4_integrate(src, dest)
 942                if diff['src_sha1'] != diff['dst_sha1']:
 943                    p4_edit(dest)
 944                if isModeExecChanged(diff['src_mode'], diff['dst_mode']):
 945                    p4_edit(dest)
 946                    filesToChangeExecBit[dest] = diff['dst_mode']
 947                os.unlink(dest)
 948                editedFiles.add(dest)
 949            elif modifier == "R":
 950                src, dest = diff['src'], diff['dst']
 951                p4_integrate(src, dest)
 952                if diff['src_sha1'] != diff['dst_sha1']:
 953                    p4_edit(dest)
 954                if isModeExecChanged(diff['src_mode'], diff['dst_mode']):
 955                    p4_edit(dest)
 956                    filesToChangeExecBit[dest] = diff['dst_mode']
 957                os.unlink(dest)
 958                editedFiles.add(dest)
 959                filesToDelete.add(src)
 960            else:
 961                die("unknown modifier %s for %s" % (modifier, path))
 962
 963        diffcmd = "git format-patch -k --stdout \"%s^\"..\"%s\"" % (id, id)
 964        patchcmd = diffcmd + " | git apply "
 965        tryPatchCmd = patchcmd + "--check -"
 966        applyPatchCmd = patchcmd + "--check --apply -"
 967
 968        if os.system(tryPatchCmd) != 0:
 969            print "Unfortunately applying the change failed!"
 970            print "What do you want to do?"
 971            response = "x"
 972            while response != "s" and response != "a" and response != "w":
 973                response = raw_input("[s]kip this patch / [a]pply the patch forcibly "
 974                                     "and with .rej files / [w]rite the patch to a file (patch.txt) ")
 975            if response == "s":
 976                print "Skipping! Good luck with the next patches..."
 977                for f in editedFiles:
 978                    p4_revert(f)
 979                for f in filesToAdd:
 980                    os.remove(f)
 981                return
 982            elif response == "a":
 983                os.system(applyPatchCmd)
 984                if len(filesToAdd) > 0:
 985                    print "You may also want to call p4 add on the following files:"
 986                    print " ".join(filesToAdd)
 987                if len(filesToDelete):
 988                    print "The following files should be scheduled for deletion with p4 delete:"
 989                    print " ".join(filesToDelete)
 990                die("Please resolve and submit the conflict manually and "
 991                    + "continue afterwards with git-p4 submit --continue")
 992            elif response == "w":
 993                system(diffcmd + " > patch.txt")
 994                print "Patch saved to patch.txt in %s !" % self.clientPath
 995                die("Please resolve and submit the conflict manually and "
 996                    "continue afterwards with git-p4 submit --continue")
 997
 998        system(applyPatchCmd)
 999
1000        for f in filesToAdd:
1001            p4_add(f)
1002        for f in filesToDelete:
1003            p4_revert(f)
1004            p4_delete(f)
1005
1006        # Set/clear executable bits
1007        for f in filesToChangeExecBit.keys():
1008            mode = filesToChangeExecBit[f]
1009            setP4ExecBit(f, mode)
1010
1011        logMessage = extractLogMessageFromGitCommit(id)
1012        logMessage = logMessage.strip()
1013
1014        template = self.prepareSubmitTemplate()
1015
1016        if self.interactive:
1017            submitTemplate = self.prepareLogMessage(template, logMessage)
1018
1019            if self.preserveUser:
1020               submitTemplate = submitTemplate + ("\n######## Actual user %s, modified after commit\n" % p4User)
1021
1022            if os.environ.has_key("P4DIFF"):
1023                del(os.environ["P4DIFF"])
1024            diff = ""
1025            for editedFile in editedFiles:
1026                diff += p4_read_pipe(['diff', '-du', editedFile])
1027
1028            newdiff = ""
1029            for newFile in filesToAdd:
1030                newdiff += "==== new file ====\n"
1031                newdiff += "--- /dev/null\n"
1032                newdiff += "+++ %s\n" % newFile
1033                f = open(newFile, "r")
1034                for line in f.readlines():
1035                    newdiff += "+" + line
1036                f.close()
1037
1038            if self.checkAuthorship and not self.p4UserIsMe(p4User):
1039                submitTemplate += "######## git author %s does not match your p4 account.\n" % gitEmail
1040                submitTemplate += "######## Use git-p4 option --preserve-user to modify authorship\n"
1041                submitTemplate += "######## Use git-p4 config git-p4.skipUserNameCheck hides this message.\n"
1042
1043            separatorLine = "######## everything below this line is just the diff #######\n"
1044
1045            (handle, fileName) = tempfile.mkstemp()
1046            tmpFile = os.fdopen(handle, "w+")
1047            if self.isWindows:
1048                submitTemplate = submitTemplate.replace("\n", "\r\n")
1049                separatorLine = separatorLine.replace("\n", "\r\n")
1050                newdiff = newdiff.replace("\n", "\r\n")
1051            tmpFile.write(submitTemplate + separatorLine + diff + newdiff)
1052            tmpFile.close()
1053
1054            if self.edit_template(fileName):
1055                # read the edited message and submit
1056                tmpFile = open(fileName, "rb")
1057                message = tmpFile.read()
1058                tmpFile.close()
1059                submitTemplate = message[:message.index(separatorLine)]
1060                if self.isWindows:
1061                    submitTemplate = submitTemplate.replace("\r\n", "\n")
1062                p4_write_pipe(['submit', '-i'], submitTemplate)
1063
1064                if self.preserveUser:
1065                    if p4User:
1066                        # Get last changelist number. Cannot easily get it from
1067                        # the submit command output as the output is
1068                        # unmarshalled.
1069                        changelist = self.lastP4Changelist()
1070                        self.modifyChangelistUser(changelist, p4User)
1071            else:
1072                # skip this patch
1073                print "Submission cancelled, undoing p4 changes."
1074                for f in editedFiles:
1075                    p4_revert(f)
1076                for f in filesToAdd:
1077                    p4_revert(f)
1078                    os.remove(f)
1079
1080            os.remove(fileName)
1081        else:
1082            fileName = "submit.txt"
1083            file = open(fileName, "w+")
1084            file.write(self.prepareLogMessage(template, logMessage))
1085            file.close()
1086            print ("Perforce submit template written as %s. "
1087                   + "Please review/edit and then use p4 submit -i < %s to submit directly!"
1088                   % (fileName, fileName))
1089
1090    def run(self, args):
1091        if len(args) == 0:
1092            self.master = currentGitBranch()
1093            if len(self.master) == 0 or not gitBranchExists("refs/heads/%s" % self.master):
1094                die("Detecting current git branch failed!")
1095        elif len(args) == 1:
1096            self.master = args[0]
1097            if not branchExists(self.master):
1098                die("Branch %s does not exist" % self.master)
1099        else:
1100            return False
1101
1102        allowSubmit = gitConfig("git-p4.allowSubmit")
1103        if len(allowSubmit) > 0 and not self.master in allowSubmit.split(","):
1104            die("%s is not in git-p4.allowSubmit" % self.master)
1105
1106        [upstream, settings] = findUpstreamBranchPoint()
1107        self.depotPath = settings['depot-paths'][0]
1108        if len(self.origin) == 0:
1109            self.origin = upstream
1110
1111        if self.preserveUser:
1112            if not self.canChangeChangelists():
1113                die("Cannot preserve user names without p4 super-user or admin permissions")
1114
1115        if self.verbose:
1116            print "Origin branch is " + self.origin
1117
1118        if len(self.depotPath) == 0:
1119            print "Internal error: cannot locate perforce depot path from existing branches"
1120            sys.exit(128)
1121
1122        self.clientPath = p4Where(self.depotPath)
1123
1124        if len(self.clientPath) == 0:
1125            print "Error: Cannot locate perforce checkout of %s in client view" % self.depotPath
1126            sys.exit(128)
1127
1128        print "Perforce checkout for depot path %s located at %s" % (self.depotPath, self.clientPath)
1129        self.oldWorkingDirectory = os.getcwd()
1130
1131        # ensure the clientPath exists
1132        if not os.path.exists(self.clientPath):
1133            os.makedirs(self.clientPath)
1134
1135        chdir(self.clientPath)
1136        print "Synchronizing p4 checkout..."
1137        p4_sync("...")
1138        self.check()
1139
1140        commits = []
1141        for line in read_pipe_lines("git rev-list --no-merges %s..%s" % (self.origin, self.master)):
1142            commits.append(line.strip())
1143        commits.reverse()
1144
1145        if self.preserveUser or (gitConfig("git-p4.skipUserNameCheck") == "true"):
1146            self.checkAuthorship = False
1147        else:
1148            self.checkAuthorship = True
1149
1150        if self.preserveUser:
1151            self.checkValidP4Users(commits)
1152
1153        while len(commits) > 0:
1154            commit = commits[0]
1155            commits = commits[1:]
1156            self.applyCommit(commit)
1157            if not self.interactive:
1158                break
1159
1160        if len(commits) == 0:
1161            print "All changes applied!"
1162            chdir(self.oldWorkingDirectory)
1163
1164            sync = P4Sync()
1165            sync.run([])
1166
1167            rebase = P4Rebase()
1168            rebase.rebase()
1169
1170        return True
1171
1172class P4Sync(Command, P4UserMap):
1173    delete_actions = ( "delete", "move/delete", "purge" )
1174
1175    def __init__(self):
1176        Command.__init__(self)
1177        P4UserMap.__init__(self)
1178        self.options = [
1179                optparse.make_option("--branch", dest="branch"),
1180                optparse.make_option("--detect-branches", dest="detectBranches", action="store_true"),
1181                optparse.make_option("--changesfile", dest="changesFile"),
1182                optparse.make_option("--silent", dest="silent", action="store_true"),
1183                optparse.make_option("--detect-labels", dest="detectLabels", action="store_true"),
1184                optparse.make_option("--verbose", dest="verbose", action="store_true"),
1185                optparse.make_option("--import-local", dest="importIntoRemotes", action="store_false",
1186                                     help="Import into refs/heads/ , not refs/remotes"),
1187                optparse.make_option("--max-changes", dest="maxChanges"),
1188                optparse.make_option("--keep-path", dest="keepRepoPath", action='store_true',
1189                                     help="Keep entire BRANCH/DIR/SUBDIR prefix during import"),
1190                optparse.make_option("--use-client-spec", dest="useClientSpec", action='store_true',
1191                                     help="Only sync files that are included in the Perforce Client Spec")
1192        ]
1193        self.description = """Imports from Perforce into a git repository.\n
1194    example:
1195    //depot/my/project/ -- to import the current head
1196    //depot/my/project/@all -- to import everything
1197    //depot/my/project/@1,6 -- to import only from revision 1 to 6
1198
1199    (a ... is not needed in the path p4 specification, it's added implicitly)"""
1200
1201        self.usage += " //depot/path[@revRange]"
1202        self.silent = False
1203        self.createdBranches = set()
1204        self.committedChanges = set()
1205        self.branch = ""
1206        self.detectBranches = False
1207        self.detectLabels = False
1208        self.changesFile = ""
1209        self.syncWithOrigin = True
1210        self.verbose = False
1211        self.importIntoRemotes = True
1212        self.maxChanges = ""
1213        self.isWindows = (platform.system() == "Windows")
1214        self.keepRepoPath = False
1215        self.depotPaths = None
1216        self.p4BranchesInGit = []
1217        self.cloneExclude = []
1218        self.useClientSpec = False
1219        self.clientSpecDirs = []
1220
1221        if gitConfig("git-p4.syncFromOrigin") == "false":
1222            self.syncWithOrigin = False
1223
1224    #
1225    # P4 wildcards are not allowed in filenames.  P4 complains
1226    # if you simply add them, but you can force it with "-f", in
1227    # which case it translates them into %xx encoding internally.
1228    # Search for and fix just these four characters.  Do % last so
1229    # that fixing it does not inadvertently create new %-escapes.
1230    #
1231    def wildcard_decode(self, path):
1232        # Cannot have * in a filename in windows; untested as to
1233        # what p4 would do in such a case.
1234        if not self.isWindows:
1235            path = path.replace("%2A", "*")
1236        path = path.replace("%23", "#") \
1237                   .replace("%40", "@") \
1238                   .replace("%25", "%")
1239        return path
1240
1241    def extractFilesFromCommit(self, commit):
1242        self.cloneExclude = [re.sub(r"\.\.\.$", "", path)
1243                             for path in self.cloneExclude]
1244        files = []
1245        fnum = 0
1246        while commit.has_key("depotFile%s" % fnum):
1247            path =  commit["depotFile%s" % fnum]
1248
1249            if [p for p in self.cloneExclude
1250                if p4PathStartsWith(path, p)]:
1251                found = False
1252            else:
1253                found = [p for p in self.depotPaths
1254                         if p4PathStartsWith(path, p)]
1255            if not found:
1256                fnum = fnum + 1
1257                continue
1258
1259            file = {}
1260            file["path"] = path
1261            file["rev"] = commit["rev%s" % fnum]
1262            file["action"] = commit["action%s" % fnum]
1263            file["type"] = commit["type%s" % fnum]
1264            files.append(file)
1265            fnum = fnum + 1
1266        return files
1267
1268    def stripRepoPath(self, path, prefixes):
1269        if self.useClientSpec:
1270
1271            # if using the client spec, we use the output directory
1272            # specified in the client.  For example, a view
1273            #   //depot/foo/branch/... //client/branch/foo/...
1274            # will end up putting all foo/branch files into
1275            #  branch/foo/
1276            for val in self.clientSpecDirs:
1277                if path.startswith(val[0]):
1278                    # replace the depot path with the client path
1279                    path = path.replace(val[0], val[1][1])
1280                    # now strip out the client (//client/...)
1281                    path = re.sub("^(//[^/]+/)", '', path)
1282                    # the rest is all path
1283                    return path
1284
1285        if self.keepRepoPath:
1286            prefixes = [re.sub("^(//[^/]+/).*", r'\1', prefixes[0])]
1287
1288        for p in prefixes:
1289            if p4PathStartsWith(path, p):
1290                path = path[len(p):]
1291
1292        return path
1293
1294    def splitFilesIntoBranches(self, commit):
1295        branches = {}
1296        fnum = 0
1297        while commit.has_key("depotFile%s" % fnum):
1298            path =  commit["depotFile%s" % fnum]
1299            found = [p for p in self.depotPaths
1300                     if p4PathStartsWith(path, p)]
1301            if not found:
1302                fnum = fnum + 1
1303                continue
1304
1305            file = {}
1306            file["path"] = path
1307            file["rev"] = commit["rev%s" % fnum]
1308            file["action"] = commit["action%s" % fnum]
1309            file["type"] = commit["type%s" % fnum]
1310            fnum = fnum + 1
1311
1312            relPath = self.stripRepoPath(path, self.depotPaths)
1313
1314            for branch in self.knownBranches.keys():
1315
1316                # add a trailing slash so that a commit into qt/4.2foo doesn't end up in qt/4.2
1317                if relPath.startswith(branch + "/"):
1318                    if branch not in branches:
1319                        branches[branch] = []
1320                    branches[branch].append(file)
1321                    break
1322
1323        return branches
1324
1325    # output one file from the P4 stream
1326    # - helper for streamP4Files
1327
1328    def streamOneP4File(self, file, contents):
1329        relPath = self.stripRepoPath(file['depotFile'], self.branchPrefixes)
1330        relPath = self.wildcard_decode(relPath)
1331        if verbose:
1332            sys.stderr.write("%s\n" % relPath)
1333
1334        (type_base, type_mods) = split_p4_type(file["type"])
1335
1336        git_mode = "100644"
1337        if "x" in type_mods:
1338            git_mode = "100755"
1339        if type_base == "symlink":
1340            git_mode = "120000"
1341            # p4 print on a symlink contains "target\n"; remove the newline
1342            data = ''.join(contents)
1343            contents = [data[:-1]]
1344
1345        if type_base == "utf16":
1346            # p4 delivers different text in the python output to -G
1347            # than it does when using "print -o", or normal p4 client
1348            # operations.  utf16 is converted to ascii or utf8, perhaps.
1349            # But ascii text saved as -t utf16 is completely mangled.
1350            # Invoke print -o to get the real contents.
1351            text = p4_read_pipe(['print', '-q', '-o', '-', file['depotFile']])
1352            contents = [ text ]
1353
1354        if type_base == "apple":
1355            # Apple filetype files will be streamed as a concatenation of
1356            # its appledouble header and the contents.  This is useless
1357            # on both macs and non-macs.  If using "print -q -o xx", it
1358            # will create "xx" with the data, and "%xx" with the header.
1359            # This is also not very useful.
1360            #
1361            # Ideally, someday, this script can learn how to generate
1362            # appledouble files directly and import those to git, but
1363            # non-mac machines can never find a use for apple filetype.
1364            print "\nIgnoring apple filetype file %s" % file['depotFile']
1365            return
1366
1367        # Perhaps windows wants unicode, utf16 newlines translated too;
1368        # but this is not doing it.
1369        if self.isWindows and type_base == "text":
1370            mangled = []
1371            for data in contents:
1372                data = data.replace("\r\n", "\n")
1373                mangled.append(data)
1374            contents = mangled
1375
1376        # Note that we do not try to de-mangle keywords on utf16 files,
1377        # even though in theory somebody may want that.
1378        if type_base in ("text", "unicode", "binary"):
1379            if "ko" in type_mods:
1380                text = ''.join(contents)
1381                text = re.sub(r'\$(Id|Header):[^$]*\$', r'$\1$', text)
1382                contents = [ text ]
1383            elif "k" in type_mods:
1384                text = ''.join(contents)
1385                text = re.sub(r'\$(Id|Header|Author|Date|DateTime|Change|File|Revision):[^$]*\$', r'$\1$', text)
1386                contents = [ text ]
1387
1388        self.gitStream.write("M %s inline %s\n" % (git_mode, relPath))
1389
1390        # total length...
1391        length = 0
1392        for d in contents:
1393            length = length + len(d)
1394
1395        self.gitStream.write("data %d\n" % length)
1396        for d in contents:
1397            self.gitStream.write(d)
1398        self.gitStream.write("\n")
1399
1400    def streamOneP4Deletion(self, file):
1401        relPath = self.stripRepoPath(file['path'], self.branchPrefixes)
1402        if verbose:
1403            sys.stderr.write("delete %s\n" % relPath)
1404        self.gitStream.write("D %s\n" % relPath)
1405
1406    # handle another chunk of streaming data
1407    def streamP4FilesCb(self, marshalled):
1408
1409        if marshalled.has_key('depotFile') and self.stream_have_file_info:
1410            # start of a new file - output the old one first
1411            self.streamOneP4File(self.stream_file, self.stream_contents)
1412            self.stream_file = {}
1413            self.stream_contents = []
1414            self.stream_have_file_info = False
1415
1416        # pick up the new file information... for the
1417        # 'data' field we need to append to our array
1418        for k in marshalled.keys():
1419            if k == 'data':
1420                self.stream_contents.append(marshalled['data'])
1421            else:
1422                self.stream_file[k] = marshalled[k]
1423
1424        self.stream_have_file_info = True
1425
1426    # Stream directly from "p4 files" into "git fast-import"
1427    def streamP4Files(self, files):
1428        filesForCommit = []
1429        filesToRead = []
1430        filesToDelete = []
1431
1432        for f in files:
1433            includeFile = True
1434            for val in self.clientSpecDirs:
1435                if f['path'].startswith(val[0]):
1436                    if val[1][0] <= 0:
1437                        includeFile = False
1438                    break
1439
1440            if includeFile:
1441                filesForCommit.append(f)
1442                if f['action'] in self.delete_actions:
1443                    filesToDelete.append(f)
1444                else:
1445                    filesToRead.append(f)
1446
1447        # deleted files...
1448        for f in filesToDelete:
1449            self.streamOneP4Deletion(f)
1450
1451        if len(filesToRead) > 0:
1452            self.stream_file = {}
1453            self.stream_contents = []
1454            self.stream_have_file_info = False
1455
1456            # curry self argument
1457            def streamP4FilesCbSelf(entry):
1458                self.streamP4FilesCb(entry)
1459
1460            fileArgs = ['%s#%s' % (f['path'], f['rev']) for f in filesToRead]
1461
1462            p4CmdList(["-x", "-", "print"],
1463                      stdin=fileArgs,
1464                      cb=streamP4FilesCbSelf)
1465
1466            # do the last chunk
1467            if self.stream_file.has_key('depotFile'):
1468                self.streamOneP4File(self.stream_file, self.stream_contents)
1469
1470    def commit(self, details, files, branch, branchPrefixes, parent = ""):
1471        epoch = details["time"]
1472        author = details["user"]
1473        self.branchPrefixes = branchPrefixes
1474
1475        if self.verbose:
1476            print "commit into %s" % branch
1477
1478        # start with reading files; if that fails, we should not
1479        # create a commit.
1480        new_files = []
1481        for f in files:
1482            if [p for p in branchPrefixes if p4PathStartsWith(f['path'], p)]:
1483                new_files.append (f)
1484            else:
1485                sys.stderr.write("Ignoring file outside of prefix: %s\n" % f['path'])
1486
1487        self.gitStream.write("commit %s\n" % branch)
1488#        gitStream.write("mark :%s\n" % details["change"])
1489        self.committedChanges.add(int(details["change"]))
1490        committer = ""
1491        if author not in self.users:
1492            self.getUserMapFromPerforceServer()
1493        if author in self.users:
1494            committer = "%s %s %s" % (self.users[author], epoch, self.tz)
1495        else:
1496            committer = "%s <a@b> %s %s" % (author, epoch, self.tz)
1497
1498        self.gitStream.write("committer %s\n" % committer)
1499
1500        self.gitStream.write("data <<EOT\n")
1501        self.gitStream.write(details["desc"])
1502        self.gitStream.write("\n[git-p4: depot-paths = \"%s\": change = %s"
1503                             % (','.join (branchPrefixes), details["change"]))
1504        if len(details['options']) > 0:
1505            self.gitStream.write(": options = %s" % details['options'])
1506        self.gitStream.write("]\nEOT\n\n")
1507
1508        if len(parent) > 0:
1509            if self.verbose:
1510                print "parent %s" % parent
1511            self.gitStream.write("from %s\n" % parent)
1512
1513        self.streamP4Files(new_files)
1514        self.gitStream.write("\n")
1515
1516        change = int(details["change"])
1517
1518        if self.labels.has_key(change):
1519            label = self.labels[change]
1520            labelDetails = label[0]
1521            labelRevisions = label[1]
1522            if self.verbose:
1523                print "Change %s is labelled %s" % (change, labelDetails)
1524
1525            files = p4CmdList(["files"] + ["%s...@%s" % (p, change)
1526                                                    for p in branchPrefixes])
1527
1528            if len(files) == len(labelRevisions):
1529
1530                cleanedFiles = {}
1531                for info in files:
1532                    if info["action"] in self.delete_actions:
1533                        continue
1534                    cleanedFiles[info["depotFile"]] = info["rev"]
1535
1536                if cleanedFiles == labelRevisions:
1537                    self.gitStream.write("tag tag_%s\n" % labelDetails["label"])
1538                    self.gitStream.write("from %s\n" % branch)
1539
1540                    owner = labelDetails["Owner"]
1541                    tagger = ""
1542                    if author in self.users:
1543                        tagger = "%s %s %s" % (self.users[owner], epoch, self.tz)
1544                    else:
1545                        tagger = "%s <a@b> %s %s" % (owner, epoch, self.tz)
1546                    self.gitStream.write("tagger %s\n" % tagger)
1547                    self.gitStream.write("data <<EOT\n")
1548                    self.gitStream.write(labelDetails["Description"])
1549                    self.gitStream.write("EOT\n\n")
1550
1551                else:
1552                    if not self.silent:
1553                        print ("Tag %s does not match with change %s: files do not match."
1554                               % (labelDetails["label"], change))
1555
1556            else:
1557                if not self.silent:
1558                    print ("Tag %s does not match with change %s: file count is different."
1559                           % (labelDetails["label"], change))
1560
1561    def getLabels(self):
1562        self.labels = {}
1563
1564        l = p4CmdList("labels %s..." % ' '.join (self.depotPaths))
1565        if len(l) > 0 and not self.silent:
1566            print "Finding files belonging to labels in %s" % `self.depotPaths`
1567
1568        for output in l:
1569            label = output["label"]
1570            revisions = {}
1571            newestChange = 0
1572            if self.verbose:
1573                print "Querying files for label %s" % label
1574            for file in p4CmdList(["files"] +
1575                                      ["%s...@%s" % (p, label)
1576                                          for p in self.depotPaths]):
1577                revisions[file["depotFile"]] = file["rev"]
1578                change = int(file["change"])
1579                if change > newestChange:
1580                    newestChange = change
1581
1582            self.labels[newestChange] = [output, revisions]
1583
1584        if self.verbose:
1585            print "Label changes: %s" % self.labels.keys()
1586
1587    def guessProjectName(self):
1588        for p in self.depotPaths:
1589            if p.endswith("/"):
1590                p = p[:-1]
1591            p = p[p.strip().rfind("/") + 1:]
1592            if not p.endswith("/"):
1593               p += "/"
1594            return p
1595
1596    def getBranchMapping(self):
1597        lostAndFoundBranches = set()
1598
1599        user = gitConfig("git-p4.branchUser")
1600        if len(user) > 0:
1601            command = "branches -u %s" % user
1602        else:
1603            command = "branches"
1604
1605        for info in p4CmdList(command):
1606            details = p4Cmd("branch -o %s" % info["branch"])
1607            viewIdx = 0
1608            while details.has_key("View%s" % viewIdx):
1609                paths = details["View%s" % viewIdx].split(" ")
1610                viewIdx = viewIdx + 1
1611                # require standard //depot/foo/... //depot/bar/... mapping
1612                if len(paths) != 2 or not paths[0].endswith("/...") or not paths[1].endswith("/..."):
1613                    continue
1614                source = paths[0]
1615                destination = paths[1]
1616                ## HACK
1617                if p4PathStartsWith(source, self.depotPaths[0]) and p4PathStartsWith(destination, self.depotPaths[0]):
1618                    source = source[len(self.depotPaths[0]):-4]
1619                    destination = destination[len(self.depotPaths[0]):-4]
1620
1621                    if destination in self.knownBranches:
1622                        if not self.silent:
1623                            print "p4 branch %s defines a mapping from %s to %s" % (info["branch"], source, destination)
1624                            print "but there exists another mapping from %s to %s already!" % (self.knownBranches[destination], destination)
1625                        continue
1626
1627                    self.knownBranches[destination] = source
1628
1629                    lostAndFoundBranches.discard(destination)
1630
1631                    if source not in self.knownBranches:
1632                        lostAndFoundBranches.add(source)
1633
1634        # Perforce does not strictly require branches to be defined, so we also
1635        # check git config for a branch list.
1636        #
1637        # Example of branch definition in git config file:
1638        # [git-p4]
1639        #   branchList=main:branchA
1640        #   branchList=main:branchB
1641        #   branchList=branchA:branchC
1642        configBranches = gitConfigList("git-p4.branchList")
1643        for branch in configBranches:
1644            if branch:
1645                (source, destination) = branch.split(":")
1646                self.knownBranches[destination] = source
1647
1648                lostAndFoundBranches.discard(destination)
1649
1650                if source not in self.knownBranches:
1651                    lostAndFoundBranches.add(source)
1652
1653
1654        for branch in lostAndFoundBranches:
1655            self.knownBranches[branch] = branch
1656
1657    def getBranchMappingFromGitBranches(self):
1658        branches = p4BranchesInGit(self.importIntoRemotes)
1659        for branch in branches.keys():
1660            if branch == "master":
1661                branch = "main"
1662            else:
1663                branch = branch[len(self.projectName):]
1664            self.knownBranches[branch] = branch
1665
1666    def listExistingP4GitBranches(self):
1667        # branches holds mapping from name to commit
1668        branches = p4BranchesInGit(self.importIntoRemotes)
1669        self.p4BranchesInGit = branches.keys()
1670        for branch in branches.keys():
1671            self.initialParents[self.refPrefix + branch] = branches[branch]
1672
1673    def updateOptionDict(self, d):
1674        option_keys = {}
1675        if self.keepRepoPath:
1676            option_keys['keepRepoPath'] = 1
1677
1678        d["options"] = ' '.join(sorted(option_keys.keys()))
1679
1680    def readOptions(self, d):
1681        self.keepRepoPath = (d.has_key('options')
1682                             and ('keepRepoPath' in d['options']))
1683
1684    def gitRefForBranch(self, branch):
1685        if branch == "main":
1686            return self.refPrefix + "master"
1687
1688        if len(branch) <= 0:
1689            return branch
1690
1691        return self.refPrefix + self.projectName + branch
1692
1693    def gitCommitByP4Change(self, ref, change):
1694        if self.verbose:
1695            print "looking in ref " + ref + " for change %s using bisect..." % change
1696
1697        earliestCommit = ""
1698        latestCommit = parseRevision(ref)
1699
1700        while True:
1701            if self.verbose:
1702                print "trying: earliest %s latest %s" % (earliestCommit, latestCommit)
1703            next = read_pipe("git rev-list --bisect %s %s" % (latestCommit, earliestCommit)).strip()
1704            if len(next) == 0:
1705                if self.verbose:
1706                    print "argh"
1707                return ""
1708            log = extractLogMessageFromGitCommit(next)
1709            settings = extractSettingsGitLog(log)
1710            currentChange = int(settings['change'])
1711            if self.verbose:
1712                print "current change %s" % currentChange
1713
1714            if currentChange == change:
1715                if self.verbose:
1716                    print "found %s" % next
1717                return next
1718
1719            if currentChange < change:
1720                earliestCommit = "^%s" % next
1721            else:
1722                latestCommit = "%s" % next
1723
1724        return ""
1725
1726    def importNewBranch(self, branch, maxChange):
1727        # make fast-import flush all changes to disk and update the refs using the checkpoint
1728        # command so that we can try to find the branch parent in the git history
1729        self.gitStream.write("checkpoint\n\n");
1730        self.gitStream.flush();
1731        branchPrefix = self.depotPaths[0] + branch + "/"
1732        range = "@1,%s" % maxChange
1733        #print "prefix" + branchPrefix
1734        changes = p4ChangesForPaths([branchPrefix], range)
1735        if len(changes) <= 0:
1736            return False
1737        firstChange = changes[0]
1738        #print "first change in branch: %s" % firstChange
1739        sourceBranch = self.knownBranches[branch]
1740        sourceDepotPath = self.depotPaths[0] + sourceBranch
1741        sourceRef = self.gitRefForBranch(sourceBranch)
1742        #print "source " + sourceBranch
1743
1744        branchParentChange = int(p4Cmd("changes -m 1 %s...@1,%s" % (sourceDepotPath, firstChange))["change"])
1745        #print "branch parent: %s" % branchParentChange
1746        gitParent = self.gitCommitByP4Change(sourceRef, branchParentChange)
1747        if len(gitParent) > 0:
1748            self.initialParents[self.gitRefForBranch(branch)] = gitParent
1749            #print "parent git commit: %s" % gitParent
1750
1751        self.importChanges(changes)
1752        return True
1753
1754    def importChanges(self, changes):
1755        cnt = 1
1756        for change in changes:
1757            description = p4Cmd("describe %s" % change)
1758            self.updateOptionDict(description)
1759
1760            if not self.silent:
1761                sys.stdout.write("\rImporting revision %s (%s%%)" % (change, cnt * 100 / len(changes)))
1762                sys.stdout.flush()
1763            cnt = cnt + 1
1764
1765            try:
1766                if self.detectBranches:
1767                    branches = self.splitFilesIntoBranches(description)
1768                    for branch in branches.keys():
1769                        ## HACK  --hwn
1770                        branchPrefix = self.depotPaths[0] + branch + "/"
1771
1772                        parent = ""
1773
1774                        filesForCommit = branches[branch]
1775
1776                        if self.verbose:
1777                            print "branch is %s" % branch
1778
1779                        self.updatedBranches.add(branch)
1780
1781                        if branch not in self.createdBranches:
1782                            self.createdBranches.add(branch)
1783                            parent = self.knownBranches[branch]
1784                            if parent == branch:
1785                                parent = ""
1786                            else:
1787                                fullBranch = self.projectName + branch
1788                                if fullBranch not in self.p4BranchesInGit:
1789                                    if not self.silent:
1790                                        print("\n    Importing new branch %s" % fullBranch);
1791                                    if self.importNewBranch(branch, change - 1):
1792                                        parent = ""
1793                                        self.p4BranchesInGit.append(fullBranch)
1794                                    if not self.silent:
1795                                        print("\n    Resuming with change %s" % change);
1796
1797                                if self.verbose:
1798                                    print "parent determined through known branches: %s" % parent
1799
1800                        branch = self.gitRefForBranch(branch)
1801                        parent = self.gitRefForBranch(parent)
1802
1803                        if self.verbose:
1804                            print "looking for initial parent for %s; current parent is %s" % (branch, parent)
1805
1806                        if len(parent) == 0 and branch in self.initialParents:
1807                            parent = self.initialParents[branch]
1808                            del self.initialParents[branch]
1809
1810                        self.commit(description, filesForCommit, branch, [branchPrefix], parent)
1811                else:
1812                    files = self.extractFilesFromCommit(description)
1813                    self.commit(description, files, self.branch, self.depotPaths,
1814                                self.initialParent)
1815                    self.initialParent = ""
1816            except IOError:
1817                print self.gitError.read()
1818                sys.exit(1)
1819
1820    def importHeadRevision(self, revision):
1821        print "Doing initial import of %s from revision %s into %s" % (' '.join(self.depotPaths), revision, self.branch)
1822
1823        details = {}
1824        details["user"] = "git perforce import user"
1825        details["desc"] = ("Initial import of %s from the state at revision %s\n"
1826                           % (' '.join(self.depotPaths), revision))
1827        details["change"] = revision
1828        newestRevision = 0
1829
1830        fileCnt = 0
1831        fileArgs = ["%s...%s" % (p,revision) for p in self.depotPaths]
1832
1833        for info in p4CmdList(["files"] + fileArgs):
1834
1835            if 'code' in info and info['code'] == 'error':
1836                sys.stderr.write("p4 returned an error: %s\n"
1837                                 % info['data'])
1838                if info['data'].find("must refer to client") >= 0:
1839                    sys.stderr.write("This particular p4 error is misleading.\n")
1840                    sys.stderr.write("Perhaps the depot path was misspelled.\n");
1841                    sys.stderr.write("Depot path:  %s\n" % " ".join(self.depotPaths))
1842                sys.exit(1)
1843            if 'p4ExitCode' in info:
1844                sys.stderr.write("p4 exitcode: %s\n" % info['p4ExitCode'])
1845                sys.exit(1)
1846
1847
1848            change = int(info["change"])
1849            if change > newestRevision:
1850                newestRevision = change
1851
1852            if info["action"] in self.delete_actions:
1853                # don't increase the file cnt, otherwise details["depotFile123"] will have gaps!
1854                #fileCnt = fileCnt + 1
1855                continue
1856
1857            for prop in ["depotFile", "rev", "action", "type" ]:
1858                details["%s%s" % (prop, fileCnt)] = info[prop]
1859
1860            fileCnt = fileCnt + 1
1861
1862        details["change"] = newestRevision
1863
1864        # Use time from top-most change so that all git-p4 clones of
1865        # the same p4 repo have the same commit SHA1s.
1866        res = p4CmdList("describe -s %d" % newestRevision)
1867        newestTime = None
1868        for r in res:
1869            if r.has_key('time'):
1870                newestTime = int(r['time'])
1871        if newestTime is None:
1872            die("\"describe -s\" on newest change %d did not give a time")
1873        details["time"] = newestTime
1874
1875        self.updateOptionDict(details)
1876        try:
1877            self.commit(details, self.extractFilesFromCommit(details), self.branch, self.depotPaths)
1878        except IOError:
1879            print "IO error with git fast-import. Is your git version recent enough?"
1880            print self.gitError.read()
1881
1882
1883    def getClientSpec(self):
1884        specList = p4CmdList( "client -o" )
1885        temp = {}
1886        for entry in specList:
1887            for k,v in entry.iteritems():
1888                if k.startswith("View"):
1889
1890                    # p4 has these %%1 to %%9 arguments in specs to
1891                    # reorder paths; which we can't handle (yet :)
1892                    if re.search('%%\d', v) != None:
1893                        print "Sorry, can't handle %%n arguments in client specs"
1894                        sys.exit(1)
1895                    if re.search('\*', v) != None:
1896                        print "Sorry, can't handle * mappings in client specs"
1897                        sys.exit(1)
1898
1899                    if v.startswith('"'):
1900                        start = 1
1901                    else:
1902                        start = 0
1903                    index = v.find("...")
1904
1905                    # save the "client view"; i.e the RHS of the view
1906                    # line that tells the client where to put the
1907                    # files for this view.
1908                    cv = v[index+3:].strip() # +3 to remove previous '...'
1909
1910                    # if the client view doesn't end with a
1911                    # ... wildcard, then we're going to mess up the
1912                    # output directory, so fail gracefully.
1913                    if not cv.endswith('...'):
1914                        print 'Sorry, client view in "%s" needs to end with wildcard' % (k)
1915                        sys.exit(1)
1916                    cv=cv[:-3]
1917
1918                    # now save the view; +index means included, -index
1919                    # means it should be filtered out.
1920                    v = v[start:index]
1921                    if v.startswith("-"):
1922                        v = v[1:]
1923                        include = -len(v)
1924                    else:
1925                        include = len(v)
1926
1927                    temp[v] = (include, cv)
1928
1929        self.clientSpecDirs = temp.items()
1930        self.clientSpecDirs.sort( lambda x, y: abs( y[1][0] ) - abs( x[1][0] ) )
1931
1932    def run(self, args):
1933        self.depotPaths = []
1934        self.changeRange = ""
1935        self.initialParent = ""
1936        self.previousDepotPaths = []
1937
1938        # map from branch depot path to parent branch
1939        self.knownBranches = {}
1940        self.initialParents = {}
1941        self.hasOrigin = originP4BranchesExist()
1942        if not self.syncWithOrigin:
1943            self.hasOrigin = False
1944
1945        if self.importIntoRemotes:
1946            self.refPrefix = "refs/remotes/p4/"
1947        else:
1948            self.refPrefix = "refs/heads/p4/"
1949
1950        if self.syncWithOrigin and self.hasOrigin:
1951            if not self.silent:
1952                print "Syncing with origin first by calling git fetch origin"
1953            system("git fetch origin")
1954
1955        if len(self.branch) == 0:
1956            self.branch = self.refPrefix + "master"
1957            if gitBranchExists("refs/heads/p4") and self.importIntoRemotes:
1958                system("git update-ref %s refs/heads/p4" % self.branch)
1959                system("git branch -D p4");
1960            # create it /after/ importing, when master exists
1961            if not gitBranchExists(self.refPrefix + "HEAD") and self.importIntoRemotes and gitBranchExists(self.branch):
1962                system("git symbolic-ref %sHEAD %s" % (self.refPrefix, self.branch))
1963
1964        if not self.useClientSpec:
1965            if gitConfig("git-p4.useclientspec", "--bool") == "true":
1966                self.useClientSpec = True
1967        if self.useClientSpec:
1968            self.getClientSpec()
1969
1970        # TODO: should always look at previous commits,
1971        # merge with previous imports, if possible.
1972        if args == []:
1973            if self.hasOrigin:
1974                createOrUpdateBranchesFromOrigin(self.refPrefix, self.silent)
1975            self.listExistingP4GitBranches()
1976
1977            if len(self.p4BranchesInGit) > 1:
1978                if not self.silent:
1979                    print "Importing from/into multiple branches"
1980                self.detectBranches = True
1981
1982            if self.verbose:
1983                print "branches: %s" % self.p4BranchesInGit
1984
1985            p4Change = 0
1986            for branch in self.p4BranchesInGit:
1987                logMsg =  extractLogMessageFromGitCommit(self.refPrefix + branch)
1988
1989                settings = extractSettingsGitLog(logMsg)
1990
1991                self.readOptions(settings)
1992                if (settings.has_key('depot-paths')
1993                    and settings.has_key ('change')):
1994                    change = int(settings['change']) + 1
1995                    p4Change = max(p4Change, change)
1996
1997                    depotPaths = sorted(settings['depot-paths'])
1998                    if self.previousDepotPaths == []:
1999                        self.previousDepotPaths = depotPaths
2000                    else:
2001                        paths = []
2002                        for (prev, cur) in zip(self.previousDepotPaths, depotPaths):
2003                            prev_list = prev.split("/")
2004                            cur_list = cur.split("/")
2005                            for i in range(0, min(len(cur_list), len(prev_list))):
2006                                if cur_list[i] <> prev_list[i]:
2007                                    i = i - 1
2008                                    break
2009
2010                            paths.append ("/".join(cur_list[:i + 1]))
2011
2012                        self.previousDepotPaths = paths
2013
2014            if p4Change > 0:
2015                self.depotPaths = sorted(self.previousDepotPaths)
2016                self.changeRange = "@%s,#head" % p4Change
2017                if not self.detectBranches:
2018                    self.initialParent = parseRevision(self.branch)
2019                if not self.silent and not self.detectBranches:
2020                    print "Performing incremental import into %s git branch" % self.branch
2021
2022        if not self.branch.startswith("refs/"):
2023            self.branch = "refs/heads/" + self.branch
2024
2025        if len(args) == 0 and self.depotPaths:
2026            if not self.silent:
2027                print "Depot paths: %s" % ' '.join(self.depotPaths)
2028        else:
2029            if self.depotPaths and self.depotPaths != args:
2030                print ("previous import used depot path %s and now %s was specified. "
2031                       "This doesn't work!" % (' '.join (self.depotPaths),
2032                                               ' '.join (args)))
2033                sys.exit(1)
2034
2035            self.depotPaths = sorted(args)
2036
2037        revision = ""
2038        self.users = {}
2039
2040        # Make sure no revision specifiers are used when --changesfile
2041        # is specified.
2042        bad_changesfile = False
2043        if len(self.changesFile) > 0:
2044            for p in self.depotPaths:
2045                if p.find("@") >= 0 or p.find("#") >= 0:
2046                    bad_changesfile = True
2047                    break
2048        if bad_changesfile:
2049            die("Option --changesfile is incompatible with revision specifiers")
2050
2051        newPaths = []
2052        for p in self.depotPaths:
2053            if p.find("@") != -1:
2054                atIdx = p.index("@")
2055                self.changeRange = p[atIdx:]
2056                if self.changeRange == "@all":
2057                    self.changeRange = ""
2058                elif ',' not in self.changeRange:
2059                    revision = self.changeRange
2060                    self.changeRange = ""
2061                p = p[:atIdx]
2062            elif p.find("#") != -1:
2063                hashIdx = p.index("#")
2064                revision = p[hashIdx:]
2065                p = p[:hashIdx]
2066            elif self.previousDepotPaths == []:
2067                # pay attention to changesfile, if given, else import
2068                # the entire p4 tree at the head revision
2069                if len(self.changesFile) == 0:
2070                    revision = "#head"
2071
2072            p = re.sub ("\.\.\.$", "", p)
2073            if not p.endswith("/"):
2074                p += "/"
2075
2076            newPaths.append(p)
2077
2078        self.depotPaths = newPaths
2079
2080
2081        self.loadUserMapFromCache()
2082        self.labels = {}
2083        if self.detectLabels:
2084            self.getLabels();
2085
2086        if self.detectBranches:
2087            ## FIXME - what's a P4 projectName ?
2088            self.projectName = self.guessProjectName()
2089
2090            if self.hasOrigin:
2091                self.getBranchMappingFromGitBranches()
2092            else:
2093                self.getBranchMapping()
2094            if self.verbose:
2095                print "p4-git branches: %s" % self.p4BranchesInGit
2096                print "initial parents: %s" % self.initialParents
2097            for b in self.p4BranchesInGit:
2098                if b != "master":
2099
2100                    ## FIXME
2101                    b = b[len(self.projectName):]
2102                self.createdBranches.add(b)
2103
2104        self.tz = "%+03d%02d" % (- time.timezone / 3600, ((- time.timezone % 3600) / 60))
2105
2106        importProcess = subprocess.Popen(["git", "fast-import"],
2107                                         stdin=subprocess.PIPE, stdout=subprocess.PIPE,
2108                                         stderr=subprocess.PIPE);
2109        self.gitOutput = importProcess.stdout
2110        self.gitStream = importProcess.stdin
2111        self.gitError = importProcess.stderr
2112
2113        if revision:
2114            self.importHeadRevision(revision)
2115        else:
2116            changes = []
2117
2118            if len(self.changesFile) > 0:
2119                output = open(self.changesFile).readlines()
2120                changeSet = set()
2121                for line in output:
2122                    changeSet.add(int(line))
2123
2124                for change in changeSet:
2125                    changes.append(change)
2126
2127                changes.sort()
2128            else:
2129                # catch "git-p4 sync" with no new branches, in a repo that
2130                # does not have any existing git-p4 branches
2131                if len(args) == 0 and not self.p4BranchesInGit:
2132                    die("No remote p4 branches.  Perhaps you never did \"git p4 clone\" in here.");
2133                if self.verbose:
2134                    print "Getting p4 changes for %s...%s" % (', '.join(self.depotPaths),
2135                                                              self.changeRange)
2136                changes = p4ChangesForPaths(self.depotPaths, self.changeRange)
2137
2138                if len(self.maxChanges) > 0:
2139                    changes = changes[:min(int(self.maxChanges), len(changes))]
2140
2141            if len(changes) == 0:
2142                if not self.silent:
2143                    print "No changes to import!"
2144                return True
2145
2146            if not self.silent and not self.detectBranches:
2147                print "Import destination: %s" % self.branch
2148
2149            self.updatedBranches = set()
2150
2151            self.importChanges(changes)
2152
2153            if not self.silent:
2154                print ""
2155                if len(self.updatedBranches) > 0:
2156                    sys.stdout.write("Updated branches: ")
2157                    for b in self.updatedBranches:
2158                        sys.stdout.write("%s " % b)
2159                    sys.stdout.write("\n")
2160
2161        self.gitStream.close()
2162        if importProcess.wait() != 0:
2163            die("fast-import failed: %s" % self.gitError.read())
2164        self.gitOutput.close()
2165        self.gitError.close()
2166
2167        return True
2168
2169class P4Rebase(Command):
2170    def __init__(self):
2171        Command.__init__(self)
2172        self.options = [ ]
2173        self.description = ("Fetches the latest revision from perforce and "
2174                            + "rebases the current work (branch) against it")
2175        self.verbose = False
2176
2177    def run(self, args):
2178        sync = P4Sync()
2179        sync.run([])
2180
2181        return self.rebase()
2182
2183    def rebase(self):
2184        if os.system("git update-index --refresh") != 0:
2185            die("Some files in your working directory are modified and different than what is in your index. You can use git update-index <filename> to bring the index up-to-date or stash away all your changes with git stash.");
2186        if len(read_pipe("git diff-index HEAD --")) > 0:
2187            die("You have uncommited changes. Please commit them before rebasing or stash them away with git stash.");
2188
2189        [upstream, settings] = findUpstreamBranchPoint()
2190        if len(upstream) == 0:
2191            die("Cannot find upstream branchpoint for rebase")
2192
2193        # the branchpoint may be p4/foo~3, so strip off the parent
2194        upstream = re.sub("~[0-9]+$", "", upstream)
2195
2196        print "Rebasing the current branch onto %s" % upstream
2197        oldHead = read_pipe("git rev-parse HEAD").strip()
2198        system("git rebase %s" % upstream)
2199        system("git diff-tree --stat --summary -M %s HEAD" % oldHead)
2200        return True
2201
2202class P4Clone(P4Sync):
2203    def __init__(self):
2204        P4Sync.__init__(self)
2205        self.description = "Creates a new git repository and imports from Perforce into it"
2206        self.usage = "usage: %prog [options] //depot/path[@revRange]"
2207        self.options += [
2208            optparse.make_option("--destination", dest="cloneDestination",
2209                                 action='store', default=None,
2210                                 help="where to leave result of the clone"),
2211            optparse.make_option("-/", dest="cloneExclude",
2212                                 action="append", type="string",
2213                                 help="exclude depot path"),
2214            optparse.make_option("--bare", dest="cloneBare",
2215                                 action="store_true", default=False),
2216        ]
2217        self.cloneDestination = None
2218        self.needsGit = False
2219        self.cloneBare = False
2220
2221    # This is required for the "append" cloneExclude action
2222    def ensure_value(self, attr, value):
2223        if not hasattr(self, attr) or getattr(self, attr) is None:
2224            setattr(self, attr, value)
2225        return getattr(self, attr)
2226
2227    def defaultDestination(self, args):
2228        ## TODO: use common prefix of args?
2229        depotPath = args[0]
2230        depotDir = re.sub("(@[^@]*)$", "", depotPath)
2231        depotDir = re.sub("(#[^#]*)$", "", depotDir)
2232        depotDir = re.sub(r"\.\.\.$", "", depotDir)
2233        depotDir = re.sub(r"/$", "", depotDir)
2234        return os.path.split(depotDir)[1]
2235
2236    def run(self, args):
2237        if len(args) < 1:
2238            return False
2239
2240        if self.keepRepoPath and not self.cloneDestination:
2241            sys.stderr.write("Must specify destination for --keep-path\n")
2242            sys.exit(1)
2243
2244        depotPaths = args
2245
2246        if not self.cloneDestination and len(depotPaths) > 1:
2247            self.cloneDestination = depotPaths[-1]
2248            depotPaths = depotPaths[:-1]
2249
2250        self.cloneExclude = ["/"+p for p in self.cloneExclude]
2251        for p in depotPaths:
2252            if not p.startswith("//"):
2253                return False
2254
2255        if not self.cloneDestination:
2256            self.cloneDestination = self.defaultDestination(args)
2257
2258        print "Importing from %s into %s" % (', '.join(depotPaths), self.cloneDestination)
2259
2260        if not os.path.exists(self.cloneDestination):
2261            os.makedirs(self.cloneDestination)
2262        chdir(self.cloneDestination)
2263
2264        init_cmd = [ "git", "init" ]
2265        if self.cloneBare:
2266            init_cmd.append("--bare")
2267        subprocess.check_call(init_cmd)
2268
2269        if not P4Sync.run(self, depotPaths):
2270            return False
2271        if self.branch != "master":
2272            if self.importIntoRemotes:
2273                masterbranch = "refs/remotes/p4/master"
2274            else:
2275                masterbranch = "refs/heads/p4/master"
2276            if gitBranchExists(masterbranch):
2277                system("git branch master %s" % masterbranch)
2278                if not self.cloneBare:
2279                    system("git checkout -f")
2280            else:
2281                print "Could not detect main branch. No checkout/master branch created."
2282
2283        return True
2284
2285class P4Branches(Command):
2286    def __init__(self):
2287        Command.__init__(self)
2288        self.options = [ ]
2289        self.description = ("Shows the git branches that hold imports and their "
2290                            + "corresponding perforce depot paths")
2291        self.verbose = False
2292
2293    def run(self, args):
2294        if originP4BranchesExist():
2295            createOrUpdateBranchesFromOrigin()
2296
2297        cmdline = "git rev-parse --symbolic "
2298        cmdline += " --remotes"
2299
2300        for line in read_pipe_lines(cmdline):
2301            line = line.strip()
2302
2303            if not line.startswith('p4/') or line == "p4/HEAD":
2304                continue
2305            branch = line
2306
2307            log = extractLogMessageFromGitCommit("refs/remotes/%s" % branch)
2308            settings = extractSettingsGitLog(log)
2309
2310            print "%s <= %s (%s)" % (branch, ",".join(settings["depot-paths"]), settings["change"])
2311        return True
2312
2313class HelpFormatter(optparse.IndentedHelpFormatter):
2314    def __init__(self):
2315        optparse.IndentedHelpFormatter.__init__(self)
2316
2317    def format_description(self, description):
2318        if description:
2319            return description + "\n"
2320        else:
2321            return ""
2322
2323def printUsage(commands):
2324    print "usage: %s <command> [options]" % sys.argv[0]
2325    print ""
2326    print "valid commands: %s" % ", ".join(commands)
2327    print ""
2328    print "Try %s <command> --help for command specific help." % sys.argv[0]
2329    print ""
2330
2331commands = {
2332    "debug" : P4Debug,
2333    "submit" : P4Submit,
2334    "commit" : P4Submit,
2335    "sync" : P4Sync,
2336    "rebase" : P4Rebase,
2337    "clone" : P4Clone,
2338    "rollback" : P4RollBack,
2339    "branches" : P4Branches
2340}
2341
2342
2343def main():
2344    if len(sys.argv[1:]) == 0:
2345        printUsage(commands.keys())
2346        sys.exit(2)
2347
2348    cmd = ""
2349    cmdName = sys.argv[1]
2350    try:
2351        klass = commands[cmdName]
2352        cmd = klass()
2353    except KeyError:
2354        print "unknown command %s" % cmdName
2355        print ""
2356        printUsage(commands.keys())
2357        sys.exit(2)
2358
2359    options = cmd.options
2360    cmd.gitdir = os.environ.get("GIT_DIR", None)
2361
2362    args = sys.argv[2:]
2363
2364    if len(options) > 0:
2365        if cmd.needsGit:
2366            options.append(optparse.make_option("--git-dir", dest="gitdir"))
2367
2368        parser = optparse.OptionParser(cmd.usage.replace("%prog", "%prog " + cmdName),
2369                                       options,
2370                                       description = cmd.description,
2371                                       formatter = HelpFormatter())
2372
2373        (cmd, args) = parser.parse_args(sys.argv[2:], cmd);
2374    global verbose
2375    verbose = cmd.verbose
2376    if cmd.needsGit:
2377        if cmd.gitdir == None:
2378            cmd.gitdir = os.path.abspath(".git")
2379            if not isValidGitDir(cmd.gitdir):
2380                cmd.gitdir = read_pipe("git rev-parse --git-dir").strip()
2381                if os.path.exists(cmd.gitdir):
2382                    cdup = read_pipe("git rev-parse --show-cdup").strip()
2383                    if len(cdup) > 0:
2384                        chdir(cdup);
2385
2386        if not isValidGitDir(cmd.gitdir):
2387            if isValidGitDir(cmd.gitdir + "/.git"):
2388                cmd.gitdir += "/.git"
2389            else:
2390                die("fatal: cannot locate git repository at %s" % cmd.gitdir)
2391
2392        os.environ["GIT_DIR"] = cmd.gitdir
2393
2394    if not cmd.run(args):
2395        parser.print_help()
2396        sys.exit(2)
2397
2398
2399if __name__ == '__main__':
2400    main()