git-p4.py 118.6 KB
Newer Older
1 2 3 4
#!/usr/bin/env python
#
# git-p4.py -- A tool for bidirectional operation between a Perforce depot and git.
#
5 6
# Author: Simon Hausmann <simon@lst.de>
# Copyright: 2007 Simon Hausmann <simon@lst.de>
7
#            2007 Trolltech ASA
8 9 10
# License: MIT <http://www.opensource.org/licenses/mit-license.php>
#

11 12
import optparse, sys, os, marshal, subprocess, shelve
import tempfile, getopt, os.path, time, platform
13
import re, shutil
14

15
verbose = False
16

17
# Only labels/tags matching this will be imported/exported
18
defaultLabelRegexp = r'[a-zA-Z0-9_\-.]+$'
19 20 21 22 23 24 25 26

def p4_build_cmd(cmd):
    """Build a suitable p4 command line.

    This consolidates building and returning a p4 command line into one
    location. It means that hooking into the environment, or other configuration
    can be done more easily.
    """
27
    real_cmd = ["p4"]
28 29 30

    user = gitConfig("git-p4.user")
    if len(user) > 0:
31
        real_cmd += ["-u",user]
32 33 34

    password = gitConfig("git-p4.password")
    if len(password) > 0:
35
        real_cmd += ["-P", password]
36 37 38

    port = gitConfig("git-p4.port")
    if len(port) > 0:
39
        real_cmd += ["-p", port]
40 41 42

    host = gitConfig("git-p4.host")
    if len(host) > 0:
43
        real_cmd += ["-H", host]
44 45 46

    client = gitConfig("git-p4.client")
    if len(client) > 0:
47
        real_cmd += ["-c", client]
48

49 50 51 52 53

    if isinstance(cmd,basestring):
        real_cmd = ' '.join(real_cmd) + ' ' + cmd
    else:
        real_cmd += cmd
54 55
    return real_cmd

56
def chdir(dir):
57
    # P4 uses the PWD environment variable rather than getcwd(). Since we're
58 59
    # not using the shell, we have to set it ourselves.  This path could
    # be relative, so go there first, then figure out where we ended up.
60
    os.chdir(dir)
61
    os.environ['PWD'] = os.getcwd()
62

63 64 65 66 67 68 69
def die(msg):
    if verbose:
        raise Exception(msg)
    else:
        sys.stderr.write(msg + "\n")
        sys.exit(1)

70
def write_pipe(c, stdin):
71
    if verbose:
72
        sys.stderr.write('Writing pipe: %s\n' % str(c))
H
Han-Wen Nienhuys 已提交
73

74 75 76 77 78 79 80
    expand = isinstance(c,basestring)
    p = subprocess.Popen(c, stdin=subprocess.PIPE, shell=expand)
    pipe = p.stdin
    val = pipe.write(stdin)
    pipe.close()
    if p.wait():
        die('Command failed: %s' % str(c))
H
Han-Wen Nienhuys 已提交
81 82 83

    return val

84
def p4_write_pipe(c, stdin):
85
    real_cmd = p4_build_cmd(c)
86
    return write_pipe(real_cmd, stdin)
87

88 89
def read_pipe(c, ignore_error=False):
    if verbose:
90
        sys.stderr.write('Reading pipe: %s\n' % str(c))
91

92 93 94
    expand = isinstance(c,basestring)
    p = subprocess.Popen(c, stdout=subprocess.PIPE, shell=expand)
    pipe = p.stdout
H
Han-Wen Nienhuys 已提交
95
    val = pipe.read()
96 97
    if p.wait() and not ignore_error:
        die('Command failed: %s' % str(c))
H
Han-Wen Nienhuys 已提交
98 99 100

    return val

101 102 103
def p4_read_pipe(c, ignore_error=False):
    real_cmd = p4_build_cmd(c)
    return read_pipe(real_cmd, ignore_error)
H
Han-Wen Nienhuys 已提交
104

H
cleanup  
Han-Wen Nienhuys 已提交
105
def read_pipe_lines(c):
106
    if verbose:
107 108 109 110 111
        sys.stderr.write('Reading pipe: %s\n' % str(c))

    expand = isinstance(c, basestring)
    p = subprocess.Popen(c, stdout=subprocess.PIPE, shell=expand)
    pipe = p.stdout
H
Han-Wen Nienhuys 已提交
112
    val = pipe.readlines()
113 114
    if pipe.close() or p.wait():
        die('Command failed: %s' % str(c))
H
Han-Wen Nienhuys 已提交
115 116

    return val
117

118 119
def p4_read_pipe_lines(c):
    """Specifically invoke p4 on the command supplied. """
A
Anand Kumria 已提交
120
    real_cmd = p4_build_cmd(c)
121 122
    return read_pipe_lines(real_cmd)

123 124 125 126 127 128 129 130 131
def p4_has_command(cmd):
    """Ask p4 for help on this command.  If it returns an error, the
       command does not exist in this version of p4."""
    real_cmd = p4_build_cmd(["help", cmd])
    p = subprocess.Popen(real_cmd, stdout=subprocess.PIPE,
                                   stderr=subprocess.PIPE)
    p.communicate()
    return p.returncode == 0

132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150
def p4_has_move_command():
    """See if the move command exists, that it supports -k, and that
       it has not been administratively disabled.  The arguments
       must be correct, but the filenames do not have to exist.  Use
       ones with wildcards so even if they exist, it will fail."""

    if not p4_has_command("move"):
        return False
    cmd = p4_build_cmd(["move", "-k", "@from", "@to"])
    p = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    (out, err) = p.communicate()
    # return code will be 1 in either case
    if err.find("Invalid option") >= 0:
        return False
    if err.find("disabled") >= 0:
        return False
    # assume it failed because @... was invalid changelist
    return True

H
Han-Wen Nienhuys 已提交
151
def system(cmd):
152
    expand = isinstance(cmd,basestring)
153
    if verbose:
154 155
        sys.stderr.write("executing %s\n" % str(cmd))
    subprocess.check_call(cmd, shell=expand)
H
Han-Wen Nienhuys 已提交
156

157 158
def p4_system(cmd):
    """Specifically invoke p4 as the system command. """
A
Anand Kumria 已提交
159
    real_cmd = p4_build_cmd(cmd)
160 161 162 163
    expand = isinstance(real_cmd, basestring)
    subprocess.check_call(real_cmd, shell=expand)

def p4_integrate(src, dest):
164
    p4_system(["integrate", "-Dt", wildcard_encode(src), wildcard_encode(dest)])
165

166
def p4_sync(f, *options):
167
    p4_system(["sync"] + list(options) + [wildcard_encode(f)])
168 169

def p4_add(f):
170 171 172 173 174
    # forcibly add file names with wildcards
    if wildcard_present(f):
        p4_system(["add", "-f", f])
    else:
        p4_system(["add", f])
175 176

def p4_delete(f):
177
    p4_system(["delete", wildcard_encode(f)])
178 179

def p4_edit(f):
180
    p4_system(["edit", wildcard_encode(f)])
181 182

def p4_revert(f):
183
    p4_system(["revert", wildcard_encode(f)])
184

185 186
def p4_reopen(type, f):
    p4_system(["reopen", "-t", type, wildcard_encode(f)])
187

188 189 190
def p4_move(src, dest):
    p4_system(["move", "-k", wildcard_encode(src), wildcard_encode(dest)])

P
Pete Wyckoff 已提交
191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213
def p4_describe(change):
    """Make sure it returns a valid result by checking for
       the presence of field "time".  Return a dict of the
       results."""

    ds = p4CmdList(["describe", "-s", str(change)])
    if len(ds) != 1:
        die("p4 describe -s %d did not return 1 result: %s" % (change, str(ds)))

    d = ds[0]

    if "p4ExitCode" in d:
        die("p4 describe -s %d exited with %d: %s" % (change, d["p4ExitCode"],
                                                      str(d)))
    if "code" in d:
        if d["code"] == "error":
            die("p4 describe -s %d returned error code: %s" % (change, str(d)))

    if "time" not in d:
        die("p4 describe -s %d returned no \"time\": %s" % (change, str(d)))

    return d

P
Pete Wyckoff 已提交
214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247
#
# Canonicalize the p4 type and return a tuple of the
# base type, plus any modifiers.  See "p4 help filetypes"
# for a list and explanation.
#
def split_p4_type(p4type):

    p4_filetypes_historical = {
        "ctempobj": "binary+Sw",
        "ctext": "text+C",
        "cxtext": "text+Cx",
        "ktext": "text+k",
        "kxtext": "text+kx",
        "ltext": "text+F",
        "tempobj": "binary+FSw",
        "ubinary": "binary+F",
        "uresource": "resource+F",
        "uxbinary": "binary+Fx",
        "xbinary": "binary+x",
        "xltext": "text+Fx",
        "xtempobj": "binary+Swx",
        "xtext": "text+x",
        "xunicode": "unicode+x",
        "xutf16": "utf16+x",
    }
    if p4type in p4_filetypes_historical:
        p4type = p4_filetypes_historical[p4type]
    mods = ""
    s = p4type.split("+")
    base = s[0]
    mods = ""
    if len(s) > 1:
        mods = s[1]
    return (base, mods)
D
David Brown 已提交
248

249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271
#
# return the raw p4 type of a file (text, text+ko, etc)
#
def p4_type(file):
    results = p4CmdList(["fstat", "-T", "headType", file])
    return results[0]['headType']

#
# Given a type base and modifier, return a regexp matching
# the keywords that can be expanded in the file
#
def p4_keywords_regexp_for_type(base, type_mods):
    if base in ("text", "unicode", "binary"):
        kwords = None
        if "ko" in type_mods:
            kwords = 'Id|Header'
        elif "k" in type_mods:
            kwords = 'Id|Header|Author|Date|DateTime|Change|File|Revision'
        else:
            return None
        pattern = r"""
            \$              # Starts with a dollar, followed by...
            (%s)            # one of the keywords, followed by...
272
            (:[^$\n]+)?     # possibly an old expansion, followed by...
273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289
            \$              # another dollar
            """ % kwords
        return pattern
    else:
        return None

#
# Given a file, return a regexp matching the possible
# RCS keywords that will be expanded, or None for files
# with kw expansion turned off.
#
def p4_keywords_regexp_for_file(file):
    if not os.path.exists(file):
        return None
    else:
        (type_base, type_mods) = split_p4_type(p4_type(file))
        return p4_keywords_regexp_for_type(type_base, type_mods)
D
David Brown 已提交
290

291 292 293 294 295 296 297 298 299 300 301 302 303
def setP4ExecBit(file, mode):
    # Reopens an already open file and changes the execute bit to match
    # the execute bit setting in the passed in mode.

    p4Type = "+x"

    if not isModeExec(mode):
        p4Type = getP4OpenedType(file)
        p4Type = re.sub('^([cku]?)x(.*)', '\\1\\2', p4Type)
        p4Type = re.sub('(.*?\+.*?)x(.*?)', '\\1\\2', p4Type)
        if p4Type[-1] == "+":
            p4Type = p4Type[0:-1]

304
    p4_reopen(p4Type, file)
305 306 307 308

def getP4OpenedType(file):
    # Returns the perforce file type for the given file.

309
    result = p4_read_pipe(["opened", wildcard_encode(file)])
310
    match = re.match(".*\((.+)\)\r?$", result)
311 312 313
    if match:
        return match.group(1)
    else:
314
        die("Could not determine file type for %s (result: '%s')" % (file, result))
315

316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335
# Return the set of all p4 labels
def getP4Labels(depotPaths):
    labels = set()
    if isinstance(depotPaths,basestring):
        depotPaths = [depotPaths]

    for l in p4CmdList(["labels"] + ["%s..." % p for p in depotPaths]):
        label = l['label']
        labels.add(label)

    return labels

# Return the set of all git tags
def getGitTags():
    gitTags = set()
    for line in read_pipe_lines(["git", "tag"]):
        tag = line.strip()
        gitTags.add(tag)
    return gitTags

336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375
def diffTreePattern():
    # This is a simple generator for the diff tree regex pattern. This could be
    # a class variable if this and parseDiffTreeEntry were a part of a class.
    pattern = re.compile(':(\d+) (\d+) (\w+) (\w+) ([A-Z])(\d+)?\t(.*?)((\t(.*))|$)')
    while True:
        yield pattern

def parseDiffTreeEntry(entry):
    """Parses a single diff tree entry into its component elements.

    See git-diff-tree(1) manpage for details about the format of the diff
    output. This method returns a dictionary with the following elements:

    src_mode - The mode of the source file
    dst_mode - The mode of the destination file
    src_sha1 - The sha1 for the source file
    dst_sha1 - The sha1 fr the destination file
    status - The one letter status of the diff (i.e. 'A', 'M', 'D', etc)
    status_score - The score for the status (applicable for 'C' and 'R'
                   statuses). This is None if there is no score.
    src - The path for the source file.
    dst - The path for the destination file. This is only present for
          copy or renames. If it is not present, this is None.

    If the pattern is not matched, None is returned."""

    match = diffTreePattern().next().match(entry)
    if match:
        return {
            'src_mode': match.group(1),
            'dst_mode': match.group(2),
            'src_sha1': match.group(3),
            'dst_sha1': match.group(4),
            'status': match.group(5),
            'status_score': match.group(6),
            'src': match.group(7),
            'dst': match.group(10)
        }
    return None

376 377 378 379 380 381 382 383
def isModeExec(mode):
    # Returns True if the given git mode represents an executable file,
    # otherwise False.
    return mode[-3:] == "755"

def isModeExecChanged(src_mode, dst_mode):
    return isModeExec(src_mode) != isModeExec(dst_mode)

384
def p4CmdList(cmd, stdin=None, stdin_mode='w+b', cb=None):
385 386 387 388 389 390 391 392 393

    if isinstance(cmd,basestring):
        cmd = "-G " + cmd
        expand = True
    else:
        cmd = ["-G"] + cmd
        expand = False

    cmd = p4_build_cmd(cmd)
H
Han-Wen Nienhuys 已提交
394
    if verbose:
395
        sys.stderr.write("Opening pipe: %s\n" % str(cmd))
S
Scott Lamb 已提交
396 397 398 399 400 401 402

    # Use a temporary file to avoid deadlocks without
    # subprocess.communicate(), which would put another copy
    # of stdout into memory.
    stdin_file = None
    if stdin is not None:
        stdin_file = tempfile.TemporaryFile(prefix='p4-stdin', mode=stdin_mode)
403 404 405 406 407
        if isinstance(stdin,basestring):
            stdin_file.write(stdin)
        else:
            for i in stdin:
                stdin_file.write(i + '\n')
S
Scott Lamb 已提交
408 409 410
        stdin_file.flush()
        stdin_file.seek(0)

411 412
    p4 = subprocess.Popen(cmd,
                          shell=expand,
S
Scott Lamb 已提交
413 414
                          stdin=stdin_file,
                          stdout=subprocess.PIPE)
415 416 417 418

    result = []
    try:
        while True:
S
Scott Lamb 已提交
419
            entry = marshal.load(p4.stdout)
420 421 422 423
            if cb is not None:
                cb(entry)
            else:
                result.append(entry)
424 425
    except EOFError:
        pass
S
Scott Lamb 已提交
426 427
    exitCode = p4.wait()
    if exitCode != 0:
428 429 430
        entry = {}
        entry["p4ExitCode"] = exitCode
        result.append(entry)
431 432 433 434 435 436 437 438 439 440

    return result

def p4Cmd(cmd):
    list = p4CmdList(cmd)
    result = {}
    for entry in list:
        result.update(entry)
    return result;

441 442 443
def p4Where(depotPath):
    if not depotPath.endswith("/"):
        depotPath += "/"
444
    depotPath = depotPath + "..."
445
    outputList = p4CmdList(["where", depotPath])
446 447
    output = None
    for entry in outputList:
448 449 450 451 452 453 454 455 456 457
        if "depotFile" in entry:
            if entry["depotFile"] == depotPath:
                output = entry
                break
        elif "data" in entry:
            data = entry.get("data")
            space = data.find(" ")
            if data[:space] == depotPath:
                output = entry
                break
458 459
    if output == None:
        return ""
460 461
    if output["code"] == "error":
        return ""
462 463 464 465 466 467 468 469 470 471 472 473
    clientPath = ""
    if "path" in output:
        clientPath = output.get("path")
    elif "data" in output:
        data = output.get("data")
        lastSpace = data.rfind(" ")
        clientPath = data[lastSpace + 1:]

    if clientPath.endswith("..."):
        clientPath = clientPath[:-3]
    return clientPath

474
def currentGitBranch():
475
    return read_pipe("git name-rev HEAD").split(" ")[1].strip()
476

477
def isValidGitDir(path):
H
Han-Wen Nienhuys 已提交
478 479
    if (os.path.exists(path + "/HEAD")
        and os.path.exists(path + "/refs") and os.path.exists(path + "/objects")):
480 481 482
        return True;
    return False

483
def parseRevision(ref):
484
    return read_pipe("git rev-parse %s" % ref).strip()
485

486 487 488 489 490
def branchExists(ref):
    rev = read_pipe(["git", "rev-parse", "-q", "--verify", ref],
                     ignore_error=True)
    return len(rev) > 0

491 492
def extractLogMessageFromGitCommit(commit):
    logMessage = ""
H
Han-Wen Nienhuys 已提交
493 494

    ## fixme: title is first line of commit, not 1st paragraph.
495
    foundTitle = False
H
Han-Wen Nienhuys 已提交
496
    for log in read_pipe_lines("git cat-file commit %s" % commit):
497 498
       if not foundTitle:
           if len(log) == 1:
S
Simon Hausmann 已提交
499
               foundTitle = True
500 501 502 503 504
           continue

       logMessage += log
    return logMessage

H
Han-Wen Nienhuys 已提交
505
def extractSettingsGitLog(log):
506 507 508
    values = {}
    for line in log.split("\n"):
        line = line.strip()
509 510 511 512 513 514 515 516 517 518 519 520 521 522
        m = re.search (r"^ *\[git-p4: (.*)\]$", line)
        if not m:
            continue

        assignments = m.group(1).split (':')
        for a in assignments:
            vals = a.split ('=')
            key = vals[0].strip()
            val = ('='.join (vals[1:])).strip()
            if val.endswith ('\"') and val.startswith('"'):
                val = val[1:-1]

            values[key] = val

523 524 525
    paths = values.get("depot-paths")
    if not paths:
        paths = values.get("depot-path")
526 527
    if paths:
        values['depot-paths'] = paths.split(',')
H
Han-Wen Nienhuys 已提交
528
    return values
529

530
def gitBranchExists(branch):
H
Han-Wen Nienhuys 已提交
531 532
    proc = subprocess.Popen(["git", "rev-parse", branch],
                            stderr=subprocess.PIPE, stdout=subprocess.PIPE);
533
    return proc.wait() == 0;
534

535
_gitConfig = {}
536
def gitConfig(key, args = None): # set args to "--bool", for instance
537
    if not _gitConfig.has_key(key):
538 539 540 541 542
        argsFilter = ""
        if args != None:
            argsFilter = "%s " % args
        cmd = "git config %s%s" % (argsFilter, key)
        _gitConfig[key] = read_pipe(cmd, ignore_error=True).strip()
543
    return _gitConfig[key]
544

545 546 547 548 549
def gitConfigList(key):
    if not _gitConfig.has_key(key):
        _gitConfig[key] = read_pipe("git config --get-all %s" % key, ignore_error=True).strip().split(os.linesep)
    return _gitConfig[key]

550 551 552 553 554 555 556
def p4BranchesInGit(branchesAreInRemotes=True):
    """Find all the branches whose names start with "p4/", looking
       in remotes or heads as specified by the argument.  Return
       a dictionary of { branch: revision } for each one found.
       The branch names are the short names, without any
       "p4/" prefix."""

557 558 559 560
    branches = {}

    cmdline = "git rev-parse --symbolic "
    if branchesAreInRemotes:
561
        cmdline += "--remotes"
562
    else:
563
        cmdline += "--branches"
564 565 566 567

    for line in read_pipe_lines(cmdline):
        line = line.strip()

568 569 570 571 572
        # only import to p4/
        if not line.startswith('p4/'):
            continue
        # special symbolic ref to p4/master
        if line == "p4/HEAD":
573 574
            continue

575 576
        # strip off p4/ prefix
        branch = line[len("p4/"):]
577 578

        branches[branch] = parseRevision(line)
579

580 581
    return branches

582
def findUpstreamBranchPoint(head = "HEAD"):
583 584 585 586 587 588 589 590 591 592 593
    branches = p4BranchesInGit()
    # map from depot-path to branch name
    branchByDepotPath = {}
    for branch in branches.keys():
        tip = branches[branch]
        log = extractLogMessageFromGitCommit(tip)
        settings = extractSettingsGitLog(log)
        if settings.has_key("depot-paths"):
            paths = ",".join(settings["depot-paths"])
            branchByDepotPath[paths] = "remotes/p4/" + branch

594 595 596
    settings = None
    parent = 0
    while parent < 65535:
597
        commit = head + "~%s" % parent
598 599
        log = extractLogMessageFromGitCommit(commit)
        settings = extractSettingsGitLog(log)
600 601 602 603
        if settings.has_key("depot-paths"):
            paths = ",".join(settings["depot-paths"])
            if branchByDepotPath.has_key(paths):
                return [branchByDepotPath[paths], settings]
604

605
        parent = parent + 1
606

607
    return ["", settings]
608

609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658
def createOrUpdateBranchesFromOrigin(localRefPrefix = "refs/remotes/p4/", silent=True):
    if not silent:
        print ("Creating/updating branch(es) in %s based on origin branch(es)"
               % localRefPrefix)

    originPrefix = "origin/p4/"

    for line in read_pipe_lines("git rev-parse --symbolic --remotes"):
        line = line.strip()
        if (not line.startswith(originPrefix)) or line.endswith("HEAD"):
            continue

        headName = line[len(originPrefix):]
        remoteHead = localRefPrefix + headName
        originHead = line

        original = extractSettingsGitLog(extractLogMessageFromGitCommit(originHead))
        if (not original.has_key('depot-paths')
            or not original.has_key('change')):
            continue

        update = False
        if not gitBranchExists(remoteHead):
            if verbose:
                print "creating %s" % remoteHead
            update = True
        else:
            settings = extractSettingsGitLog(extractLogMessageFromGitCommit(remoteHead))
            if settings.has_key('change') > 0:
                if settings['depot-paths'] == original['depot-paths']:
                    originP4Change = int(original['change'])
                    p4Change = int(settings['change'])
                    if originP4Change > p4Change:
                        print ("%s (%s) is newer than %s (%s). "
                               "Updating p4 branch from origin."
                               % (originHead, originP4Change,
                                  remoteHead, p4Change))
                        update = True
                else:
                    print ("Ignoring: %s was imported from %s while "
                           "%s was imported from %s"
                           % (originHead, ','.join(original['depot-paths']),
                              remoteHead, ','.join(settings['depot-paths'])))

        if update:
            system("git update-ref %s %s" % (remoteHead, originHead))

def originP4BranchesExist():
        return gitBranchExists("origin") or gitBranchExists("origin/p4") or gitBranchExists("origin/p4/master")

659 660
def p4ChangesForPaths(depotPaths, changeRange):
    assert depotPaths
661 662 663 664
    cmd = ['changes']
    for p in depotPaths:
        cmd += ["%s...%s" % (p, changeRange)]
    output = p4_read_pipe_lines(cmd)
665

666
    changes = {}
667
    for line in output:
668 669
        changeNum = int(line.split(" ")[1])
        changes[changeNum] = True
670

671 672 673
    changelist = changes.keys()
    changelist.sort()
    return changelist
674

675 676 677 678 679 680 681 682 683 684 685 686 687
def p4PathStartsWith(path, prefix):
    # This method tries to remedy a potential mixed-case issue:
    #
    # If UserA adds  //depot/DirA/file1
    # and UserB adds //depot/dira/file2
    #
    # we may or may not have a problem. If you have core.ignorecase=true,
    # we treat DirA and dira as the same directory
    ignorecase = gitConfig("core.ignorecase", "--bool") == "true"
    if ignorecase:
        return path.lower().startswith(prefix.lower())
    return path.startswith(prefix)

688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727
def getClientSpec():
    """Look at the p4 client spec, create a View() object that contains
       all the mappings, and return it."""

    specList = p4CmdList("client -o")
    if len(specList) != 1:
        die('Output from "client -o" is %d lines, expecting 1' %
            len(specList))

    # dictionary of all client parameters
    entry = specList[0]

    # just the keys that start with "View"
    view_keys = [ k for k in entry.keys() if k.startswith("View") ]

    # hold this new View
    view = View()

    # append the lines, in order, to the view
    for view_num in range(len(view_keys)):
        k = "View%d" % view_num
        if k not in view_keys:
            die("Expected view key %s missing" % k)
        view.append(entry[k])

    return view

def getClientRoot():
    """Grab the client directory."""

    output = p4CmdList("client -o")
    if len(output) != 1:
        die('Output from "client -o" is %d lines, expecting 1' % len(output))

    entry = output[0]
    if "Root" not in entry:
        die('Client has no "Root"')

    return entry["Root"]

728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755
#
# P4 wildcards are not allowed in filenames.  P4 complains
# if you simply add them, but you can force it with "-f", in
# which case it translates them into %xx encoding internally.
#
def wildcard_decode(path):
    # Search for and fix just these four characters.  Do % last so
    # that fixing it does not inadvertently create new %-escapes.
    # Cannot have * in a filename in windows; untested as to
    # what p4 would do in such a case.
    if not platform.system() == "Windows":
        path = path.replace("%2A", "*")
    path = path.replace("%23", "#") \
               .replace("%40", "@") \
               .replace("%25", "%")
    return path

def wildcard_encode(path):
    # do % first to avoid double-encoding the %s introduced here
    path = path.replace("%", "%25") \
               .replace("*", "%2A") \
               .replace("#", "%23") \
               .replace("@", "%40")
    return path

def wildcard_present(path):
    return path.translate(None, "*#@%") != path

756 757 758
class Command:
    def __init__(self):
        self.usage = "usage: %prog [options]"
759
        self.needsGit = True
760
        self.verbose = False
761

762 763 764
class P4UserMap:
    def __init__(self):
        self.userMapFromPerforceServer = False
765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784
        self.myP4UserId = None

    def p4UserId(self):
        if self.myP4UserId:
            return self.myP4UserId

        results = p4CmdList("user -o")
        for r in results:
            if r.has_key('User'):
                self.myP4UserId = r['User']
                return r['User']
        die("Could not find your p4 user id")

    def p4UserIsMe(self, p4User):
        # return True if the given p4 user is actually me
        me = self.p4UserId()
        if not p4User or p4User != me:
            return False
        else:
            return True
785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822

    def getUserCacheFilename(self):
        home = os.environ.get("HOME", os.environ.get("USERPROFILE"))
        return home + "/.gitp4-usercache.txt"

    def getUserMapFromPerforceServer(self):
        if self.userMapFromPerforceServer:
            return
        self.users = {}
        self.emails = {}

        for output in p4CmdList("users"):
            if not output.has_key("User"):
                continue
            self.users[output["User"]] = output["FullName"] + " <" + output["Email"] + ">"
            self.emails[output["Email"]] = output["User"]


        s = ''
        for (key, val) in self.users.items():
            s += "%s\t%s\n" % (key.expandtabs(1), val.expandtabs(1))

        open(self.getUserCacheFilename(), "wb").write(s)
        self.userMapFromPerforceServer = True

    def loadUserMapFromCache(self):
        self.users = {}
        self.userMapFromPerforceServer = False
        try:
            cache = open(self.getUserCacheFilename(), "rb")
            lines = cache.readlines()
            cache.close()
            for line in lines:
                entry = line.strip().split("\t")
                self.users[entry[0]] = entry[1]
        except IOError:
            self.getUserMapFromPerforceServer()

823
class P4Debug(Command):
824
    def __init__(self):
825
        Command.__init__(self)
826
        self.options = []
827
        self.description = "A tool to debug the output of p4 -G."
828
        self.needsGit = False
829 830

    def run(self, args):
831
        j = 0
832
        for output in p4CmdList(args):
833 834
            print 'Element: %d' % j
            j += 1
835
            print output
836
        return True
837

838 839 840 841
class P4RollBack(Command):
    def __init__(self):
        Command.__init__(self)
        self.options = [
842
            optparse.make_option("--local", dest="rollbackLocalBranches", action="store_true")
843 844
        ]
        self.description = "A tool to debug the multi-branch import. Don't use :)"
845
        self.rollbackLocalBranches = False
846 847 848 849 850

    def run(self, args):
        if len(args) != 1:
            return False
        maxChange = int(args[0])
851

852
        if "p4ExitCode" in p4Cmd("changes -m 1"):
853 854
            die("Problems executing p4");

855 856
        if self.rollbackLocalBranches:
            refPrefix = "refs/heads/"
H
Han-Wen Nienhuys 已提交
857
            lines = read_pipe_lines("git rev-parse --symbolic --branches")
858 859
        else:
            refPrefix = "refs/remotes/"
H
Han-Wen Nienhuys 已提交
860
            lines = read_pipe_lines("git rev-parse --symbolic --remotes")
861 862 863

        for line in lines:
            if self.rollbackLocalBranches or (line.startswith("p4/") and line != "p4/HEAD\n"):
864 865
                line = line.strip()
                ref = refPrefix + line
866
                log = extractLogMessageFromGitCommit(ref)
H
Han-Wen Nienhuys 已提交
867 868 869 870 871
                settings = extractSettingsGitLog(log)

                depotPaths = settings['depot-paths']
                change = settings['change']

872
                changed = False
873

874 875
                if len(p4Cmd("changes -m 1 "  + ' '.join (['%s...@%s' % (p, maxChange)
                                                           for p in depotPaths]))) == 0:
876 877 878 879
                    print "Branch %s did not exist at change %s, deleting." % (ref, maxChange)
                    system("git update-ref -d %s `git rev-parse %s`" % (ref, ref))
                    continue

H
Han-Wen Nienhuys 已提交
880
                while change and int(change) > maxChange:
881
                    changed = True
882 883
                    if self.verbose:
                        print "%s is at %s ; rewinding towards %s" % (ref, change, maxChange)
884 885
                    system("git update-ref %s \"%s^\"" % (ref, ref))
                    log = extractLogMessageFromGitCommit(ref)
H
Han-Wen Nienhuys 已提交
886 887 888 889 890
                    settings =  extractSettingsGitLog(log)


                    depotPaths = settings['depot-paths']
                    change = settings['change']
891 892

                if changed:
893
                    print "%s rewound to %s" % (ref, change)
894 895 896

        return True

897
class P4Submit(Command, P4UserMap):
898 899 900

    conflict_behavior_choices = ("ask", "skip", "quit")

901
    def __init__(self):
902
        Command.__init__(self)
903
        P4UserMap.__init__(self)
904 905
        self.options = [
                optparse.make_option("--origin", dest="origin"),
906
                optparse.make_option("-M", dest="detectRenames", action="store_true"),
907 908
                # preserve the user, requires relevant p4 permissions
                optparse.make_option("--preserve-user", dest="preserveUser", action="store_true"),
909
                optparse.make_option("--export-labels", dest="exportLabels", action="store_true"),
910
                optparse.make_option("--dry-run", "-n", dest="dry_run", action="store_true"),
911
                optparse.make_option("--prepare-p4-only", dest="prepare_p4_only", action="store_true"),
912 913
                optparse.make_option("--conflict", dest="conflict_behavior",
                                     choices=self.conflict_behavior_choices)
914 915
        ]
        self.description = "Submit changes from git to the perforce depot."
916
        self.usage += " [name of git branch to submit into perforce depot]"
917
        self.origin = ""
918
        self.detectRenames = False
919
        self.preserveUser = gitConfig("git-p4.preserveUser").lower() == "true"
920
        self.dry_run = False
921
        self.prepare_p4_only = False
922
        self.conflict_behavior = None
923
        self.isWindows = (platform.system() == "Windows")
924
        self.exportLabels = False
925
        self.p4HasMoveCommand = p4_has_move_command()
926 927 928 929 930

    def check(self):
        if len(p4CmdList("opened ...")) > 0:
            die("You have files opened with perforce! Close them before starting the sync.")

931 932 933 934 935 936 937 938 939 940 941 942 943 944 945 946 947 948 949 950 951 952 953 954 955 956 957 958
    def separate_jobs_from_description(self, message):
        """Extract and return a possible Jobs field in the commit
           message.  It goes into a separate section in the p4 change
           specification.

           A jobs line starts with "Jobs:" and looks like a new field
           in a form.  Values are white-space separated on the same
           line or on following lines that start with a tab.

           This does not parse and extract the full git commit message
           like a p4 form.  It just sees the Jobs: line as a marker
           to pass everything from then on directly into the p4 form,
           but outside the description section.

           Return a tuple (stripped log message, jobs string)."""

        m = re.search(r'^Jobs:', message, re.MULTILINE)
        if m is None:
            return (message, None)

        jobtext = message[m.start():]
        stripped_message = message[:m.start()].rstrip()
        return (stripped_message, jobtext)

    def prepareLogMessage(self, template, message, jobs):
        """Edits the template returned from "p4 change -o" to insert
           the message in the Description field, and the jobs text in
           the Jobs field."""
959 960
        result = ""

961 962
        inDescriptionSection = False

963 964 965 966 967
        for line in template.split("\n"):
            if line.startswith("#"):
                result += line + "\n"
                continue

968
            if inDescriptionSection:
969
                if line.startswith("Files:") or line.startswith("Jobs:"):
970
                    inDescriptionSection = False
971 972 973
                    # insert Jobs section
                    if jobs:
                        result += jobs + "\n"
974 975 976 977 978 979 980 981 982 983
                else:
                    continue
            else:
                if line.startswith("Description:"):
                    inDescriptionSection = True
                    line += "\n"
                    for messageLine in message.split("\n"):
                        line += "\t" + messageLine + "\n"

            result += line + "\n"
984 985 986

        return result

987 988 989 990 991 992 993 994 995 996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009
    def patchRCSKeywords(self, file, pattern):
        # Attempt to zap the RCS keywords in a p4 controlled file matching the given pattern
        (handle, outFileName) = tempfile.mkstemp(dir='.')
        try:
            outFile = os.fdopen(handle, "w+")
            inFile = open(file, "r")
            regexp = re.compile(pattern, re.VERBOSE)
            for line in inFile.readlines():
                line = regexp.sub(r'$\1$', line)
                outFile.write(line)
            inFile.close()
            outFile.close()
            # Forcibly overwrite the original file
            os.unlink(file)
            shutil.move(outFileName, file)
        except:
            # cleanup our temporary file
            os.unlink(outFileName)
            print "Failed to strip RCS keywords in %s" % file
            raise

        print "Patched up RCS keywords in %s" % file

1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043
    def p4UserForCommit(self,id):
        # Return the tuple (perforce user,git email) for a given git commit id
        self.getUserMapFromPerforceServer()
        gitEmail = read_pipe("git log --max-count=1 --format='%%ae' %s" % id)
        gitEmail = gitEmail.strip()
        if not self.emails.has_key(gitEmail):
            return (None,gitEmail)
        else:
            return (self.emails[gitEmail],gitEmail)

    def checkValidP4Users(self,commits):
        # check if any git authors cannot be mapped to p4 users
        for id in commits:
            (user,email) = self.p4UserForCommit(id)
            if not user:
                msg = "Cannot find p4 user for email %s in commit %s." % (email, id)
                if gitConfig('git-p4.allowMissingP4Users').lower() == "true":
                    print "%s" % msg
                else:
                    die("Error: %s\nSet git-p4.allowMissingP4Users to true to allow this." % msg)

    def lastP4Changelist(self):
        # Get back the last changelist number submitted in this client spec. This
        # then gets used to patch up the username in the change. If the same
        # client spec is being used by multiple processes then this might go
        # wrong.
        results = p4CmdList("client -o")        # find the current client
        client = None
        for r in results:
            if r.has_key('Client'):
                client = r['Client']
                break
        if not client:
            die("could not get client spec")
1044
        results = p4CmdList(["changes", "-c", client, "-m", "1"])
1045 1046 1047 1048 1049 1050 1051 1052
        for r in results:
            if r.has_key('change'):
                return r['change']
        die("Could not get changelist number for last submit - cannot patch up user details")

    def modifyChangelistUser(self, changelist, newUser):
        # fixup the user field of a changelist after it has been submitted.
        changes = p4CmdList("change -o %s" % changelist)
1053 1054 1055 1056 1057 1058 1059 1060 1061
        if len(changes) != 1:
            die("Bad output from p4 change modifying %s to user %s" %
                (changelist, newUser))

        c = changes[0]
        if c['User'] == newUser: return   # nothing to do
        c['User'] = newUser
        input = marshal.dumps(c)

1062 1063 1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074
        result = p4CmdList("change -f -i", stdin=input)
        for r in result:
            if r.has_key('code'):
                if r['code'] == 'error':
                    die("Could not modify user field of changelist %s to %s:%s" % (changelist, newUser, r['data']))
            if r.has_key('data'):
                print("Updated user field for changelist %s to %s" % (changelist, newUser))
                return
        die("Could not modify user field of changelist %s to %s" % (changelist, newUser))

    def canChangeChangelists(self):
        # check to see if we have p4 admin or super-user permissions, either of
        # which are required to modify changelists.
1075
        results = p4CmdList(["protects", self.depotPath])
1076 1077 1078 1079 1080 1081 1082 1083
        for r in results:
            if r.has_key('perm'):
                if r['perm'] == 'admin':
                    return 1
                if r['perm'] == 'super':
                    return 1
        return 0

1084
    def prepareSubmitTemplate(self):
1085 1086 1087 1088 1089 1090 1091
        """Run "p4 change -o" to grab a change specification template.
           This does not use "p4 -G", as it is nice to keep the submission
           template in original order, since a human might edit it.

           Remove lines in the Files section that show changes to files
           outside the depot path we're committing into."""

1092 1093
        template = ""
        inFilesSection = False
1094
        for line in p4_read_pipe_lines(['change', '-o']):
1095 1096
            if line.endswith("\r\n"):
                line = line[:-2] + "\n"
1097 1098 1099 1100 1101 1102 1103
            if inFilesSection:
                if line.startswith("\t"):
                    # path starts and ends with a tab
                    path = line[1:]
                    lastTab = path.rfind("\t")
                    if lastTab != -1:
                        path = path[:lastTab]
1104
                        if not p4PathStartsWith(path, self.depotPath):
1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115
                            continue
                else:
                    inFilesSection = False
            else:
                if line.startswith("Files:"):
                    inFilesSection = True

            template += line

        return template

P
Pete Wyckoff 已提交
1116 1117 1118 1119 1120 1121 1122 1123 1124 1125 1126 1127 1128
    def edit_template(self, template_file):
        """Invoke the editor to let the user change the submission
           message.  Return true if okay to continue with the submit."""

        # if configured to skip the editing part, just submit
        if gitConfig("git-p4.skipSubmitEdit") == "true":
            return True

        # look at the modification time, to check later if the user saved
        # the file
        mtime = os.stat(template_file).st_mtime

        # invoke the editor
1129
        if os.environ.has_key("P4EDITOR") and (os.environ.get("P4EDITOR") != ""):
P
Pete Wyckoff 已提交
1130 1131 1132 1133 1134 1135 1136 1137 1138 1139
            editor = os.environ.get("P4EDITOR")
        else:
            editor = read_pipe("git var GIT_EDITOR").strip()
        system(editor + " " + template_file)

        # If the file was not saved, prompt to see if this patch should
        # be skipped.  But skip this verification step if configured so.
        if gitConfig("git-p4.skipSubmitEditCheck") == "true":
            return True

1140 1141 1142 1143 1144 1145 1146 1147 1148 1149
        # modification time updated means user saved the file
        if os.stat(template_file).st_mtime > mtime:
            return True

        while True:
            response = raw_input("Submit template unchanged. Submit anyway? [y]es, [n]o (skip this patch) ")
            if response == 'y':
                return True
            if response == 'n':
                return False
P
Pete Wyckoff 已提交
1150

1151
    def applyCommit(self, id):
1152 1153 1154 1155
        """Apply one commit, return True if it succeeded."""

        print "Applying", read_pipe(["git", "show", "-s",
                                     "--format=format:%h %s", id])
1156

1157
        (p4User, gitEmail) = self.p4UserForCommit(id)
1158

1159
        diff = read_pipe_lines("git diff-tree -r %s \"%s^\" \"%s\"" % (self.diffOpts, id, id))
1160 1161
        filesToAdd = set()
        filesToDelete = set()
1162
        editedFiles = set()
1163
        pureRenameCopy = set()
1164
        filesToChangeExecBit = {}
1165

1166
        for line in diff:
1167 1168 1169
            diff = parseDiffTreeEntry(line)
            modifier = diff['status']
            path = diff['src']
1170
            if modifier == "M":
1171
                p4_edit(path)
1172 1173
                if isModeExecChanged(diff['src_mode'], diff['dst_mode']):
                    filesToChangeExecBit[path] = diff['dst_mode']
1174
                editedFiles.add(path)
1175 1176
            elif modifier == "A":
                filesToAdd.add(path)
1177
                filesToChangeExecBit[path] = diff['dst_mode']
1178 1179 1180 1181 1182 1183
                if path in filesToDelete:
                    filesToDelete.remove(path)
            elif modifier == "D":
                filesToDelete.add(path)
                if path in filesToAdd:
                    filesToAdd.remove(path)
1184 1185
            elif modifier == "C":
                src, dest = diff['src'], diff['dst']
1186
                p4_integrate(src, dest)
1187
                pureRenameCopy.add(dest)
1188
                if diff['src_sha1'] != diff['dst_sha1']:
1189
                    p4_edit(dest)
1190
                    pureRenameCopy.discard(dest)
1191
                if isModeExecChanged(diff['src_mode'], diff['dst_mode']):
1192
                    p4_edit(dest)
1193
                    pureRenameCopy.discard(dest)
1194 1195 1196
                    filesToChangeExecBit[dest] = diff['dst_mode']
                os.unlink(dest)
                editedFiles.add(dest)
1197
            elif modifier == "R":
1198
                src, dest = diff['src'], diff['dst']
1199 1200 1201
                if self.p4HasMoveCommand:
                    p4_edit(src)        # src must be open before move
                    p4_move(src, dest)  # opens for (move/delete, move/add)
1202
                else:
1203 1204 1205 1206 1207
                    p4_integrate(src, dest)
                    if diff['src_sha1'] != diff['dst_sha1']:
                        p4_edit(dest)
                    else:
                        pureRenameCopy.add(dest)
1208
                if isModeExecChanged(diff['src_mode'], diff['dst_mode']):
1209 1210
                    if not self.p4HasMoveCommand:
                        p4_edit(dest)   # with move: already open, writable
1211
                    filesToChangeExecBit[dest] = diff['dst_mode']
1212 1213 1214
                if not self.p4HasMoveCommand:
                    os.unlink(dest)
                    filesToDelete.add(src)
1215
                editedFiles.add(dest)
1216 1217 1218
            else:
                die("unknown modifier %s for %s" % (modifier, path))

1219
        diffcmd = "git format-patch -k --stdout \"%s^\"..\"%s\"" % (id, id)
1220
        patchcmd = diffcmd + " | git apply "
1221 1222
        tryPatchCmd = patchcmd + "--check -"
        applyPatchCmd = patchcmd + "--check --apply -"
1223
        patch_succeeded = True
1224

1225
        if os.system(tryPatchCmd) != 0:
1226 1227
            fixed_rcs_keywords = False
            patch_succeeded = False
1228
            print "Unfortunately applying the change failed!"
1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261

            # Patch failed, maybe it's just RCS keyword woes. Look through
            # the patch to see if that's possible.
            if gitConfig("git-p4.attemptRCSCleanup","--bool") == "true":
                file = None
                pattern = None
                kwfiles = {}
                for file in editedFiles | filesToDelete:
                    # did this file's delta contain RCS keywords?
                    pattern = p4_keywords_regexp_for_file(file)

                    if pattern:
                        # this file is a possibility...look for RCS keywords.
                        regexp = re.compile(pattern, re.VERBOSE)
                        for line in read_pipe_lines(["git", "diff", "%s^..%s" % (id, id), file]):
                            if regexp.search(line):
                                if verbose:
                                    print "got keyword match on %s in %s in %s" % (pattern, line, file)
                                kwfiles[file] = pattern
                                break

                for file in kwfiles:
                    if verbose:
                        print "zapping %s with %s" % (line,pattern)
                    self.patchRCSKeywords(file, kwfiles[file])
                    fixed_rcs_keywords = True

            if fixed_rcs_keywords:
                print "Retrying the patch with RCS keywords cleaned up"
                if os.system(tryPatchCmd) == 0:
                    patch_succeeded = True

        if not patch_succeeded:
1262 1263 1264
            for f in editedFiles:
                p4_revert(f)
            return False
1265

1266 1267 1268
        #
        # Apply the patch for real, and do add/delete/+x handling.
        #
1269
        system(applyPatchCmd)
1270 1271

        for f in filesToAdd:
1272
            p4_add(f)
1273
        for f in filesToDelete:
1274 1275
            p4_revert(f)
            p4_delete(f)
1276

1277 1278 1279 1280 1281
        # Set/clear executable bits
        for f in filesToChangeExecBit.keys():
            mode = filesToChangeExecBit[f]
            setP4ExecBit(f, mode)

1282 1283 1284 1285
        #
        # Build p4 change description, starting with the contents
        # of the git commit message.
        #
1286 1287
        logMessage = extractLogMessageFromGitCommit(id)
        logMessage = logMessage.strip()
1288
        (logMessage, jobs) = self.separate_jobs_from_description(logMessage)
1289

1290
        template = self.prepareSubmitTemplate()
1291
        submitTemplate = self.prepareLogMessage(template, logMessage, jobs)
1292

1293
        if self.preserveUser:
1294
           submitTemplate += "\n######## Actual user %s, modified after commit\n" % p4User
1295

1296 1297 1298 1299
        if self.checkAuthorship and not self.p4UserIsMe(p4User):
            submitTemplate += "######## git author %s does not match your p4 account.\n" % gitEmail
            submitTemplate += "######## Use option --preserve-user to modify authorship.\n"
            submitTemplate += "######## Variable git-p4.skipUserNameCheck hides this message.\n"
1300

1301 1302 1303
        separatorLine = "######## everything below this line is just the diff #######\n"

        # diff
1304 1305 1306 1307 1308 1309 1310
        if os.environ.has_key("P4DIFF"):
            del(os.environ["P4DIFF"])
        diff = ""
        for editedFile in editedFiles:
            diff += p4_read_pipe(['diff', '-du',
                                  wildcard_encode(editedFile)])

1311
        # new file diff
1312 1313 1314 1315 1316 1317 1318 1319 1320 1321
        newdiff = ""
        for newFile in filesToAdd:
            newdiff += "==== new file ====\n"
            newdiff += "--- /dev/null\n"
            newdiff += "+++ %s\n" % newFile
            f = open(newFile, "r")
            for line in f.readlines():
                newdiff += "+" + line
            f.close()

1322
        # change description file: submitTemplate, separatorLine, diff, newdiff
1323 1324 1325 1326 1327 1328 1329 1330 1331
        (handle, fileName) = tempfile.mkstemp()
        tmpFile = os.fdopen(handle, "w+")
        if self.isWindows:
            submitTemplate = submitTemplate.replace("\n", "\r\n")
            separatorLine = separatorLine.replace("\n", "\r\n")
            newdiff = newdiff.replace("\n", "\r\n")
        tmpFile.write(submitTemplate + separatorLine + diff + newdiff)
        tmpFile.close()

1332 1333 1334 1335 1336 1337 1338 1339 1340 1341 1342 1343 1344 1345 1346 1347 1348 1349 1350 1351 1352 1353 1354 1355 1356 1357 1358 1359 1360 1361 1362 1363 1364 1365 1366
        if self.prepare_p4_only:
            #
            # Leave the p4 tree prepared, and the submit template around
            # and let the user decide what to do next
            #
            print
            print "P4 workspace prepared for submission."
            print "To submit or revert, go to client workspace"
            print "  " + self.clientPath
            print
            print "To submit, use \"p4 submit\" to write a new description,"
            print "or \"p4 submit -i %s\" to use the one prepared by" \
                  " \"git p4\"." % fileName
            print "You can delete the file \"%s\" when finished." % fileName

            if self.preserveUser and p4User and not self.p4UserIsMe(p4User):
                print "To preserve change ownership by user %s, you must\n" \
                      "do \"p4 change -f <change>\" after submitting and\n" \
                      "edit the User field."
            if pureRenameCopy:
                print "After submitting, renamed files must be re-synced."
                print "Invoke \"p4 sync -f\" on each of these files:"
                for f in pureRenameCopy:
                    print "  " + f

            print
            print "To revert the changes, use \"p4 revert ...\", and delete"
            print "the submit template file \"%s\"" % fileName
            if filesToAdd:
                print "Since the commit adds new files, they must be deleted:"
                for f in filesToAdd:
                    print "  " + f
            print
            return True

1367 1368 1369
        #
        # Let the user edit the change description, then submit it.
        #
1370 1371
        if self.edit_template(fileName):
            # read the edited message and submit
1372
            ret = True
1373 1374
            tmpFile = open(fileName, "rb")
            message = tmpFile.read()
1375
            tmpFile.close()
1376 1377 1378 1379
            submitTemplate = message[:message.index(separatorLine)]
            if self.isWindows:
                submitTemplate = submitTemplate.replace("\r\n", "\n")
            p4_write_pipe(['submit', '-i'], submitTemplate)
1380

1381 1382 1383 1384 1385 1386 1387 1388 1389 1390 1391 1392
            if self.preserveUser:
                if p4User:
                    # Get last changelist number. Cannot easily get it from
                    # the submit command output as the output is
                    # unmarshalled.
                    changelist = self.lastP4Changelist()
                    self.modifyChangelistUser(changelist, p4User)

            # The rename/copy happened by applying a patch that created a
            # new file.  This leaves it writable, which confuses p4.
            for f in pureRenameCopy:
                p4_sync(f, "-f")
1393

1394
        else:
1395
            # skip this patch
1396
            ret = False
1397 1398 1399 1400 1401 1402
            print "Submission cancelled, undoing p4 changes."
            for f in editedFiles:
                p4_revert(f)
            for f in filesToAdd:
                p4_revert(f)
                os.remove(f)
1403 1404
            for f in filesToDelete:
                p4_revert(f)
1405 1406

        os.remove(fileName)
1407
        return ret
1408

1409 1410 1411
    # Export git tags as p4 labels. Create a p4 label and then tag
    # with that.
    def exportGitTags(self, gitTags):
1412 1413 1414 1415
        validLabelRegexp = gitConfig("git-p4.labelExportRegexp")
        if len(validLabelRegexp) == 0:
            validLabelRegexp = defaultLabelRegexp
        m = re.compile(validLabelRegexp)
1416 1417 1418 1419 1420

        for name in gitTags:

            if not m.match(name):
                if verbose:
1421
                    print "tag %s does not match regexp %s" % (name, validLabelRegexp)
1422 1423 1424
                continue

            # Get the p4 commit this corresponds to
1425 1426
            logMessage = extractLogMessageFromGitCommit(name)
            values = extractSettingsGitLog(logMessage)
1427

1428
            if not values.has_key('change'):
1429 1430 1431 1432
                # a tag pointing to something not sent to p4; ignore
                if verbose:
                    print "git tag %s does not give a p4 commit" % name
                continue
1433 1434
            else:
                changelist = values['change']
1435 1436 1437 1438 1439 1440 1441 1442 1443 1444 1445 1446 1447 1448 1449 1450 1451 1452 1453 1454 1455 1456 1457 1458 1459 1460 1461 1462 1463 1464

            # Get the tag details.
            inHeader = True
            isAnnotated = False
            body = []
            for l in read_pipe_lines(["git", "cat-file", "-p", name]):
                l = l.strip()
                if inHeader:
                    if re.match(r'tag\s+', l):
                        isAnnotated = True
                    elif re.match(r'\s*$', l):
                        inHeader = False
                        continue
                else:
                    body.append(l)

            if not isAnnotated:
                body = ["lightweight tag imported by git p4\n"]

            # Create the label - use the same view as the client spec we are using
            clientSpec = getClientSpec()

            labelTemplate  = "Label: %s\n" % name
            labelTemplate += "Description:\n"
            for b in body:
                labelTemplate += "\t" + b + "\n"
            labelTemplate += "View:\n"
            for mapping in clientSpec.mappings:
                labelTemplate += "\t%s\n" % mapping.depot_side.path

1465 1466
            if self.dry_run:
                print "Would create p4 label %s for tag" % name
1467 1468 1469
            elif self.prepare_p4_only:
                print "Not creating p4 label %s for tag due to option" \
                      " --prepare-p4-only" % name
1470 1471
            else:
                p4_write_pipe(["label", "-i"], labelTemplate)
1472

1473 1474 1475
                # Use the label
                p4_system(["tag", "-l", name] +
                          ["%s@%s" % (mapping.depot_side.path, changelist) for mapping in clientSpec.mappings])
1476

1477 1478
                if verbose:
                    print "created p4 label for tag %s" % name
1479

1480
    def run(self, args):
1481 1482
        if len(args) == 0:
            self.master = currentGitBranch()
1483
            if len(self.master) == 0 or not gitBranchExists("refs/heads/%s" % self.master):
1484 1485 1486
                die("Detecting current git branch failed!")
        elif len(args) == 1:
            self.master = args[0]
1487 1488
            if not branchExists(self.master):
                die("Branch %s does not exist" % self.master)
1489 1490 1491
        else:
            return False

J
Jing Xue 已提交
1492 1493 1494 1495
        allowSubmit = gitConfig("git-p4.allowSubmit")
        if len(allowSubmit) > 0 and not self.master in allowSubmit.split(","):
            die("%s is not in git-p4.allowSubmit" % self.master)

1496
        [upstream, settings] = findUpstreamBranchPoint()
1497
        self.depotPath = settings['depot-paths'][0]
1498 1499
        if len(self.origin) == 0:
            self.origin = upstream
1500

1501 1502 1503 1504
        if self.preserveUser:
            if not self.canChangeChangelists():
                die("Cannot preserve user names without p4 super-user or admin permissions")

1505 1506 1507 1508 1509 1510 1511 1512 1513 1514
        # if not set from the command line, try the config file
        if self.conflict_behavior is None:
            val = gitConfig("git-p4.conflict")
            if val:
                if val not in self.conflict_behavior_choices:
                    die("Invalid value '%s' for config git-p4.conflict" % val)
            else:
                val = "ask"
            self.conflict_behavior = val

1515 1516
        if self.verbose:
            print "Origin branch is " + self.origin
1517

1518
        if len(self.depotPath) == 0:
1519 1520 1521
            print "Internal error: cannot locate perforce depot path from existing branches"
            sys.exit(128)

1522 1523 1524 1525 1526
        self.useClientSpec = False
        if gitConfig("git-p4.useclientspec", "--bool") == "true":
            self.useClientSpec = True
        if self.useClientSpec:
            self.clientSpecDirs = getClientSpec()
1527

1528 1529 1530 1531 1532
        if self.useClientSpec:
            # all files are relative to the client spec
            self.clientPath = getClientRoot()
        else:
            self.clientPath = p4Where(self.depotPath)
1533

1534 1535
        if self.clientPath == "":
            die("Error: Cannot locate perforce checkout of %s in client view" % self.depotPath)
1536

1537
        print "Perforce checkout for depot path %s located at %s" % (self.depotPath, self.clientPath)
1538
        self.oldWorkingDirectory = os.getcwd()
1539

1540
        # ensure the clientPath exists
1541
        new_client_dir = False
1542
        if not os.path.exists(self.clientPath):
1543
            new_client_dir = True
1544 1545
            os.makedirs(self.clientPath)

1546
        chdir(self.clientPath)
1547 1548
        if self.dry_run:
            print "Would synchronize p4 checkout in %s" % self.clientPath
1549
        else:
1550 1551 1552 1553 1554 1555
            print "Synchronizing p4 checkout..."
            if new_client_dir:
                # old one was destroyed, and maybe nobody told p4
                p4_sync("...", "-f")
            else:
                p4_sync("...")
1556 1557
        self.check()

S
Simon Hausmann 已提交
1558 1559 1560 1561
        commits = []
        for line in read_pipe_lines("git rev-list --no-merges %s..%s" % (self.origin, self.master)):
            commits.append(line.strip())
        commits.reverse()
1562

1563 1564 1565 1566 1567
        if self.preserveUser or (gitConfig("git-p4.skipUserNameCheck") == "true"):
            self.checkAuthorship = False
        else:
            self.checkAuthorship = True

1568 1569 1570
        if self.preserveUser:
            self.checkValidP4Users(commits)

1571 1572 1573 1574 1575 1576 1577 1578 1579 1580 1581 1582 1583 1584 1585 1586 1587 1588 1589 1590 1591 1592 1593 1594 1595 1596 1597 1598 1599 1600 1601
        #
        # Build up a set of options to be passed to diff when
        # submitting each commit to p4.
        #
        if self.detectRenames:
            # command-line -M arg
            self.diffOpts = "-M"
        else:
            # If not explicitly set check the config variable
            detectRenames = gitConfig("git-p4.detectRenames")

            if detectRenames.lower() == "false" or detectRenames == "":
                self.diffOpts = ""
            elif detectRenames.lower() == "true":
                self.diffOpts = "-M"
            else:
                self.diffOpts = "-M%s" % detectRenames

        # no command-line arg for -C or --find-copies-harder, just
        # config variables
        detectCopies = gitConfig("git-p4.detectCopies")
        if detectCopies.lower() == "false" or detectCopies == "":
            pass
        elif detectCopies.lower() == "true":
            self.diffOpts += " -C"
        else:
            self.diffOpts += " -C%s" % detectCopies

        if gitConfig("git-p4.detectCopiesHarder", "--bool") == "true":
            self.diffOpts += " --find-copies-harder"

1602 1603 1604 1605
        #
        # Apply the commits, one at a time.  On failure, ask if should
        # continue to try the rest of the patches, or quit.
        #
1606 1607
        if self.dry_run:
            print "Would apply"
1608
        applied = []
1609 1610
        last = len(commits) - 1
        for i, commit in enumerate(commits):
1611 1612 1613 1614 1615 1616
            if self.dry_run:
                print " ", read_pipe(["git", "show", "-s",
                                      "--format=format:%h %s", commit])
                ok = True
            else:
                ok = self.applyCommit(commit)
1617 1618
            if ok:
                applied.append(commit)
1619
            else:
1620 1621 1622 1623
                if self.prepare_p4_only and i < last:
                    print "Processing only the first commit due to option" \
                          " --prepare-p4-only"
                    break
1624 1625 1626
                if i < last:
                    quit = False
                    while True:
1627 1628 1629 1630 1631 1632 1633 1634 1635 1636 1637 1638 1639 1640 1641
                        # prompt for what to do, or use the option/variable
                        if self.conflict_behavior == "ask":
                            print "What do you want to do?"
                            response = raw_input("[s]kip this commit but apply"
                                                 " the rest, or [q]uit? ")
                            if not response:
                                continue
                        elif self.conflict_behavior == "skip":
                            response = "s"
                        elif self.conflict_behavior == "quit":
                            response = "q"
                        else:
                            die("Unknown conflict_behavior '%s'" %
                                self.conflict_behavior)

1642 1643 1644 1645 1646 1647 1648 1649 1650
                        if response[0] == "s":
                            print "Skipping this commit, but applying the rest"
                            break
                        if response[0] == "q":
                            print "Quitting"
                            quit = True
                            break
                    if quit:
                        break
1651

1652
        chdir(self.oldWorkingDirectory)
1653

1654 1655
        if self.dry_run:
            pass
1656 1657
        elif self.prepare_p4_only:
            pass
1658
        elif len(commits) == len(applied):
1659
            print "All commits applied!"
1660

S
Simon Hausmann 已提交
1661 1662
            sync = P4Sync()
            sync.run([])
1663

S
Simon Hausmann 已提交
1664 1665
            rebase = P4Rebase()
            rebase.rebase()
1666

1667 1668 1669 1670 1671 1672 1673 1674 1675 1676 1677 1678 1679 1680
        else:
            if len(applied) == 0:
                print "No commits applied."
            else:
                print "Applied only the commits marked with '*':"
                for c in commits:
                    if c in applied:
                        star = "*"
                    else:
                        star = " "
                    print star, read_pipe(["git", "show", "-s",
                                           "--format=format:%h %s",  c])
                print "You will have to do 'git p4 sync' and rebase."

1681
        if gitConfig("git-p4.exportLabels", "--bool") == "true":
1682
            self.exportLabels = True
1683 1684 1685 1686 1687 1688 1689 1690

        if self.exportLabels:
            p4Labels = getP4Labels(self.depotPath)
            gitTags = getGitTags()

            missingGitTags = gitTags - p4Labels
            self.exportGitTags(missingGitTags)

1691 1692 1693 1694
        # exit with error unless everything applied perfecly
        if len(commits) != len(applied):
                sys.exit(1)

1695 1696
        return True

P
Pete Wyckoff 已提交
1697 1698 1699 1700 1701 1702 1703 1704 1705 1706 1707 1708 1709 1710 1711 1712 1713 1714 1715 1716 1717 1718 1719 1720 1721 1722 1723 1724 1725 1726 1727 1728 1729 1730 1731 1732 1733 1734
class View(object):
    """Represent a p4 view ("p4 help views"), and map files in a
       repo according to the view."""

    class Path(object):
        """A depot or client path, possibly containing wildcards.
           The only one supported is ... at the end, currently.
           Initialize with the full path, with //depot or //client."""

        def __init__(self, path, is_depot):
            self.path = path
            self.is_depot = is_depot
            self.find_wildcards()
            # remember the prefix bit, useful for relative mappings
            m = re.match("(//[^/]+/)", self.path)
            if not m:
                die("Path %s does not start with //prefix/" % self.path)
            prefix = m.group(1)
            if not self.is_depot:
                # strip //client/ on client paths
                self.path = self.path[len(prefix):]

        def find_wildcards(self):
            """Make sure wildcards are valid, and set up internal
               variables."""

            self.ends_triple_dot = False
            # There are three wildcards allowed in p4 views
            # (see "p4 help views").  This code knows how to
            # handle "..." (only at the end), but cannot deal with
            # "%%n" or "*".  Only check the depot_side, as p4 should
            # validate that the client_side matches too.
            if re.search(r'%%[1-9]', self.path):
                die("Can't handle %%n wildcards in view: %s" % self.path)
            if self.path.find("*") >= 0:
                die("Can't handle * wildcards in view: %s" % self.path)
            triple_dot_index = self.path.find("...")
            if triple_dot_index >= 0:
1735 1736
                if triple_dot_index != len(self.path) - 3:
                    die("Can handle only single ... wildcard, at end: %s" %
P
Pete Wyckoff 已提交
1737 1738 1739 1740 1741 1742 1743 1744 1745 1746 1747 1748 1749 1750 1751 1752 1753 1754 1755 1756 1757 1758 1759 1760 1761 1762 1763 1764 1765 1766 1767 1768 1769 1770 1771 1772 1773 1774 1775 1776 1777 1778 1779 1780 1781 1782 1783 1784 1785 1786 1787 1788 1789 1790
                        self.path)
                self.ends_triple_dot = True

        def ensure_compatible(self, other_path):
            """Make sure the wildcards agree."""
            if self.ends_triple_dot != other_path.ends_triple_dot:
                 die("Both paths must end with ... if either does;\n" +
                     "paths: %s %s" % (self.path, other_path.path))

        def match_wildcards(self, test_path):
            """See if this test_path matches us, and fill in the value
               of the wildcards if so.  Returns a tuple of
               (True|False, wildcards[]).  For now, only the ... at end
               is supported, so at most one wildcard."""
            if self.ends_triple_dot:
                dotless = self.path[:-3]
                if test_path.startswith(dotless):
                    wildcard = test_path[len(dotless):]
                    return (True, [ wildcard ])
            else:
                if test_path == self.path:
                    return (True, [])
            return (False, [])

        def match(self, test_path):
            """Just return if it matches; don't bother with the wildcards."""
            b, _ = self.match_wildcards(test_path)
            return b

        def fill_in_wildcards(self, wildcards):
            """Return the relative path, with the wildcards filled in
               if there are any."""
            if self.ends_triple_dot:
                return self.path[:-3] + wildcards[0]
            else:
                return self.path

    class Mapping(object):
        def __init__(self, depot_side, client_side, overlay, exclude):
            # depot_side is without the trailing /... if it had one
            self.depot_side = View.Path(depot_side, is_depot=True)
            self.client_side = View.Path(client_side, is_depot=False)
            self.overlay = overlay  # started with "+"
            self.exclude = exclude  # started with "-"
            assert not (self.overlay and self.exclude)
            self.depot_side.ensure_compatible(self.client_side)

        def __str__(self):
            c = " "
            if self.overlay:
                c = "+"
            if self.exclude:
                c = "-"
            return "View.Mapping: %s%s -> %s" % \
P
Pete Wyckoff 已提交
1791
                   (c, self.depot_side.path, self.client_side.path)
P
Pete Wyckoff 已提交
1792 1793 1794 1795 1796 1797 1798 1799 1800 1801 1802 1803 1804 1805 1806 1807 1808 1809 1810 1811 1812 1813 1814 1815 1816 1817 1818 1819 1820 1821 1822 1823 1824 1825 1826 1827 1828 1829 1830 1831 1832 1833 1834 1835 1836 1837 1838 1839 1840 1841 1842 1843 1844 1845 1846 1847 1848 1849 1850 1851 1852 1853 1854 1855 1856 1857 1858 1859 1860 1861 1862 1863 1864 1865 1866 1867 1868 1869 1870 1871 1872 1873 1874 1875 1876 1877 1878 1879 1880 1881 1882 1883 1884 1885 1886 1887 1888 1889 1890

        def map_depot_to_client(self, depot_path):
            """Calculate the client path if using this mapping on the
               given depot path; does not consider the effect of other
               mappings in a view.  Even excluded mappings are returned."""
            matches, wildcards = self.depot_side.match_wildcards(depot_path)
            if not matches:
                return ""
            client_path = self.client_side.fill_in_wildcards(wildcards)
            return client_path

    #
    # View methods
    #
    def __init__(self):
        self.mappings = []

    def append(self, view_line):
        """Parse a view line, splitting it into depot and client
           sides.  Append to self.mappings, preserving order."""

        # Split the view line into exactly two words.  P4 enforces
        # structure on these lines that simplifies this quite a bit.
        #
        # Either or both words may be double-quoted.
        # Single quotes do not matter.
        # Double-quote marks cannot occur inside the words.
        # A + or - prefix is also inside the quotes.
        # There are no quotes unless they contain a space.
        # The line is already white-space stripped.
        # The two words are separated by a single space.
        #
        if view_line[0] == '"':
            # First word is double quoted.  Find its end.
            close_quote_index = view_line.find('"', 1)
            if close_quote_index <= 0:
                die("No first-word closing quote found: %s" % view_line)
            depot_side = view_line[1:close_quote_index]
            # skip closing quote and space
            rhs_index = close_quote_index + 1 + 1
        else:
            space_index = view_line.find(" ")
            if space_index <= 0:
                die("No word-splitting space found: %s" % view_line)
            depot_side = view_line[0:space_index]
            rhs_index = space_index + 1

        if view_line[rhs_index] == '"':
            # Second word is double quoted.  Make sure there is a
            # double quote at the end too.
            if not view_line.endswith('"'):
                die("View line with rhs quote should end with one: %s" %
                    view_line)
            # skip the quotes
            client_side = view_line[rhs_index+1:-1]
        else:
            client_side = view_line[rhs_index:]

        # prefix + means overlay on previous mapping
        overlay = False
        if depot_side.startswith("+"):
            overlay = True
            depot_side = depot_side[1:]

        # prefix - means exclude this path
        exclude = False
        if depot_side.startswith("-"):
            exclude = True
            depot_side = depot_side[1:]

        m = View.Mapping(depot_side, client_side, overlay, exclude)
        self.mappings.append(m)

    def map_in_client(self, depot_path):
        """Return the relative location in the client where this
           depot file should live.  Returns "" if the file should
           not be mapped in the client."""

        paths_filled = []
        client_path = ""

        # look at later entries first
        for m in self.mappings[::-1]:

            # see where will this path end up in the client
            p = m.map_depot_to_client(depot_path)

            if p == "":
                # Depot path does not belong in client.  Must remember
                # this, as previous items should not cause files to
                # exist in this path either.  Remember that the list is
                # being walked from the end, which has higher precedence.
                # Overlap mappings do not exclude previous mappings.
                if not m.overlay:
                    paths_filled.append(m.client_side)

            else:
                # This mapping matched; no need to search any further.
                # But, the mapping could be rejected if the client path
P
Pete Wyckoff 已提交
1891 1892
                # has already been claimed by an earlier mapping (i.e.
                # one later in the list, which we are walking backwards).
P
Pete Wyckoff 已提交
1893 1894 1895 1896 1897 1898 1899 1900 1901 1902 1903 1904 1905 1906 1907 1908 1909
                already_mapped_in_client = False
                for f in paths_filled:
                    # this is View.Path.match
                    if f.match(p):
                        already_mapped_in_client = True
                        break
                if not already_mapped_in_client:
                    # Include this file, unless it is from a line that
                    # explicitly said to exclude it.
                    if not m.exclude:
                        client_path = p

                # a match, even if rejected, always stops the search
                break

        return client_path

1910
class P4Sync(Command, P4UserMap):
1911 1912
    delete_actions = ( "delete", "move/delete", "purge" )

1913 1914
    def __init__(self):
        Command.__init__(self)
1915
        P4UserMap.__init__(self)
1916 1917 1918 1919 1920
        self.options = [
                optparse.make_option("--branch", dest="branch"),
                optparse.make_option("--detect-branches", dest="detectBranches", action="store_true"),
                optparse.make_option("--changesfile", dest="changesFile"),
                optparse.make_option("--silent", dest="silent", action="store_true"),
1921
                optparse.make_option("--detect-labels", dest="detectLabels", action="store_true"),
1922
                optparse.make_option("--import-labels", dest="importLabels", action="store_true"),
1923 1924
                optparse.make_option("--import-local", dest="importIntoRemotes", action="store_false",
                                     help="Import into refs/heads/ , not refs/remotes"),
1925
                optparse.make_option("--max-changes", dest="maxChanges"),
1926
                optparse.make_option("--keep-path", dest="keepRepoPath", action='store_true',
1927 1928 1929
                                     help="Keep entire BRANCH/DIR/SUBDIR prefix during import"),
                optparse.make_option("--use-client-spec", dest="useClientSpec", action='store_true',
                                     help="Only sync files that are included in the Perforce Client Spec")
1930 1931 1932 1933 1934 1935 1936 1937 1938 1939 1940
        ]
        self.description = """Imports from Perforce into a git repository.\n
    example:
    //depot/my/project/ -- to import the current head
    //depot/my/project/@all -- to import everything
    //depot/my/project/@1,6 -- to import only from revision 1 to 6

    (a ... is not needed in the path p4 specification, it's added implicitly)"""

        self.usage += " //depot/path[@revRange]"
        self.silent = False
1941 1942
        self.createdBranches = set()
        self.committedChanges = set()
1943
        self.branch = ""
1944
        self.detectBranches = False
1945
        self.detectLabels = False
1946
        self.importLabels = False
1947
        self.changesFile = ""
1948
        self.syncWithOrigin = True
1949
        self.importIntoRemotes = True
1950
        self.maxChanges = ""
1951
        self.isWindows = (platform.system() == "Windows")
1952
        self.keepRepoPath = False
1953
        self.depotPaths = None
1954
        self.p4BranchesInGit = []
T
Tommy Thorn 已提交
1955
        self.cloneExclude = []
1956
        self.useClientSpec = False
1957
        self.useClientSpec_from_options = False
P
Pete Wyckoff 已提交
1958
        self.clientSpecDirs = None
1959 1960
        self.tempBranches = []
        self.tempBranchLocation = "git-p4-tmp"
1961

1962 1963 1964
        if gitConfig("git-p4.syncFromOrigin") == "false":
            self.syncWithOrigin = False

1965 1966 1967 1968 1969 1970 1971 1972
    # Force a checkpoint in fast-import and wait for it to finish
    def checkpoint(self):
        self.gitStream.write("checkpoint\n\n")
        self.gitStream.write("progress checkpoint\n\n")
        out = self.gitOutput.readline()
        if self.verbose:
            print "checkpoint finished: " + out

1973
    def extractFilesFromCommit(self, commit):
T
Tommy Thorn 已提交
1974 1975
        self.cloneExclude = [re.sub(r"\.\.\.$", "", path)
                             for path in self.cloneExclude]
1976 1977 1978 1979
        files = []
        fnum = 0
        while commit.has_key("depotFile%s" % fnum):
            path =  commit["depotFile%s" % fnum]
1980

T
Tommy Thorn 已提交
1981
            if [p for p in self.cloneExclude
1982
                if p4PathStartsWith(path, p)]:
T
Tommy Thorn 已提交
1983 1984 1985
                found = False
            else:
                found = [p for p in self.depotPaths
1986
                         if p4PathStartsWith(path, p)]
1987
            if not found:
1988 1989 1990 1991 1992 1993 1994 1995 1996 1997 1998 1999
                fnum = fnum + 1
                continue

            file = {}
            file["path"] = path
            file["rev"] = commit["rev%s" % fnum]
            file["action"] = commit["action%s" % fnum]
            file["type"] = commit["type%s" % fnum]
            files.append(file)
            fnum = fnum + 1
        return files

2000
    def stripRepoPath(self, path, prefixes):
2001 2002 2003 2004 2005
        """When streaming files, this is called to map a p4 depot path
           to where it should go in git.  The prefixes are either
           self.depotPaths, or self.branchPrefixes in the case of
           branch detection."""

2006
        if self.useClientSpec:
2007 2008
            # branch detection moves files up a level (the branch name)
            # from what client spec interpretation gives
2009
            path = self.clientSpecDirs.map_in_client(path)
2010 2011 2012 2013 2014 2015 2016 2017 2018 2019 2020 2021
            if self.detectBranches:
                for b in self.knownBranches:
                    if path.startswith(b + "/"):
                        path = path[len(b)+1:]

        elif self.keepRepoPath:
            # Preserve everything in relative path name except leading
            # //depot/; just look at first prefix as they all should
            # be in the same depot.
            depot = re.sub("^(//[^/]+/).*", r'\1', prefixes[0])
            if p4PathStartsWith(path, depot):
                path = path[len(depot):]
2022

2023 2024 2025 2026
        else:
            for p in prefixes:
                if p4PathStartsWith(path, p):
                    path = path[len(p):]
2027
                    break
2028

2029
        path = wildcard_decode(path)
2030
        return path
H
Han-Wen Nienhuys 已提交
2031

2032
    def splitFilesIntoBranches(self, commit):
2033 2034 2035
        """Look at each depotFile in the commit to figure out to what
           branch it belongs."""

2036
        branches = {}
2037 2038 2039
        fnum = 0
        while commit.has_key("depotFile%s" % fnum):
            path =  commit["depotFile%s" % fnum]
2040
            found = [p for p in self.depotPaths
2041
                     if p4PathStartsWith(path, p)]
2042
            if not found:
2043 2044 2045 2046 2047 2048 2049 2050 2051 2052
                fnum = fnum + 1
                continue

            file = {}
            file["path"] = path
            file["rev"] = commit["rev%s" % fnum]
            file["action"] = commit["action%s" % fnum]
            file["type"] = commit["type%s" % fnum]
            fnum = fnum + 1

2053 2054 2055 2056 2057 2058
            # start with the full relative path where this file would
            # go in a p4 client
            if self.useClientSpec:
                relPath = self.clientSpecDirs.map_in_client(path)
            else:
                relPath = self.stripRepoPath(path, self.depotPaths)
2059

2060
            for branch in self.knownBranches.keys():
2061 2062
                # add a trailing slash so that a commit into qt/4.2foo
                # doesn't end up in qt/4.2, e.g.
H
Han-Wen Nienhuys 已提交
2063
                if relPath.startswith(branch + "/"):
2064 2065
                    if branch not in branches:
                        branches[branch] = []
2066
                    branches[branch].append(file)
2067
                    break
2068 2069 2070

        return branches

2071 2072 2073 2074 2075 2076 2077 2078
    # output one file from the P4 stream
    # - helper for streamP4Files

    def streamOneP4File(self, file, contents):
        relPath = self.stripRepoPath(file['depotFile'], self.branchPrefixes)
        if verbose:
            sys.stderr.write("%s\n" % relPath)

P
Pete Wyckoff 已提交
2079 2080 2081 2082 2083 2084 2085 2086
        (type_base, type_mods) = split_p4_type(file["type"])

        git_mode = "100644"
        if "x" in type_mods:
            git_mode = "100755"
        if type_base == "symlink":
            git_mode = "120000"
            # p4 print on a symlink contains "target\n"; remove the newline
2087 2088
            data = ''.join(contents)
            contents = [data[:-1]]
2089

P
Pete Wyckoff 已提交
2090
        if type_base == "utf16":
2091 2092 2093 2094 2095
            # p4 delivers different text in the python output to -G
            # than it does when using "print -o", or normal p4 client
            # operations.  utf16 is converted to ascii or utf8, perhaps.
            # But ascii text saved as -t utf16 is completely mangled.
            # Invoke print -o to get the real contents.
2096
            text = p4_read_pipe(['print', '-q', '-o', '-', file['depotFile']])
2097 2098
            contents = [ text ]

P
Pete Wyckoff 已提交
2099 2100 2101 2102 2103 2104 2105 2106 2107 2108 2109 2110 2111
        if type_base == "apple":
            # Apple filetype files will be streamed as a concatenation of
            # its appledouble header and the contents.  This is useless
            # on both macs and non-macs.  If using "print -q -o xx", it
            # will create "xx" with the data, and "%xx" with the header.
            # This is also not very useful.
            #
            # Ideally, someday, this script can learn how to generate
            # appledouble files directly and import those to git, but
            # non-mac machines can never find a use for apple filetype.
            print "\nIgnoring apple filetype file %s" % file['depotFile']
            return

P
Pete Wyckoff 已提交
2112 2113 2114
        # Perhaps windows wants unicode, utf16 newlines translated too;
        # but this is not doing it.
        if self.isWindows and type_base == "text":
2115 2116 2117 2118 2119 2120
            mangled = []
            for data in contents:
                data = data.replace("\r\n", "\n")
                mangled.append(data)
            contents = mangled

2121 2122
        # Note that we do not try to de-mangle keywords on utf16 files,
        # even though in theory somebody may want that.
2123 2124 2125 2126 2127 2128
        pattern = p4_keywords_regexp_for_type(type_base, type_mods)
        if pattern:
            regexp = re.compile(pattern, re.VERBOSE)
            text = ''.join(contents)
            text = regexp.sub(r'$\1$', text)
            contents = [ text ]
2129

P
Pete Wyckoff 已提交
2130
        self.gitStream.write("M %s inline %s\n" % (git_mode, relPath))
2131 2132 2133 2134 2135 2136 2137 2138 2139 2140 2141 2142 2143 2144 2145 2146 2147 2148 2149 2150

        # total length...
        length = 0
        for d in contents:
            length = length + len(d)

        self.gitStream.write("data %d\n" % length)
        for d in contents:
            self.gitStream.write(d)
        self.gitStream.write("\n")

    def streamOneP4Deletion(self, file):
        relPath = self.stripRepoPath(file['path'], self.branchPrefixes)
        if verbose:
            sys.stderr.write("delete %s\n" % relPath)
        self.gitStream.write("D %s\n" % relPath)

    # handle another chunk of streaming data
    def streamP4FilesCb(self, marshalled):

2151 2152 2153 2154 2155 2156 2157 2158 2159 2160 2161 2162 2163 2164 2165 2166 2167 2168 2169 2170 2171 2172 2173
        # catch p4 errors and complain
        err = None
        if "code" in marshalled:
            if marshalled["code"] == "error":
                if "data" in marshalled:
                    err = marshalled["data"].rstrip()
        if err:
            f = None
            if self.stream_have_file_info:
                if "depotFile" in self.stream_file:
                    f = self.stream_file["depotFile"]
            # force a failure in fast-import, else an empty
            # commit will be made
            self.gitStream.write("\n")
            self.gitStream.write("die-now\n")
            self.gitStream.close()
            # ignore errors, but make sure it exits first
            self.importProcess.wait()
            if f:
                die("Error from p4 print for %s: %s" % (f, err))
            else:
                die("Error from p4 print: %s" % err)

2174 2175 2176 2177 2178 2179
        if marshalled.has_key('depotFile') and self.stream_have_file_info:
            # start of a new file - output the old one first
            self.streamOneP4File(self.stream_file, self.stream_contents)
            self.stream_file = {}
            self.stream_contents = []
            self.stream_have_file_info = False
2180

2181 2182 2183 2184 2185 2186 2187
        # pick up the new file information... for the
        # 'data' field we need to append to our array
        for k in marshalled.keys():
            if k == 'data':
                self.stream_contents.append(marshalled['data'])
            else:
                self.stream_file[k] = marshalled[k]
2188

2189
        self.stream_have_file_info = True
2190 2191 2192

    # Stream directly from "p4 files" into "git fast-import"
    def streamP4Files(self, files):
2193 2194
        filesForCommit = []
        filesToRead = []
2195
        filesToDelete = []
2196

2197
        for f in files:
P
Pete Wyckoff 已提交
2198 2199 2200 2201 2202
            # if using a client spec, only add the files that have
            # a path in the client
            if self.clientSpecDirs:
                if self.clientSpecDirs.map_in_client(f['path']) == "":
                    continue
2203

P
Pete Wyckoff 已提交
2204 2205 2206 2207 2208
            filesForCommit.append(f)
            if f['action'] in self.delete_actions:
                filesToDelete.append(f)
            else:
                filesToRead.append(f)
H
Han-Wen Nienhuys 已提交
2209

2210 2211 2212
        # deleted files...
        for f in filesToDelete:
            self.streamOneP4Deletion(f)
2213

2214 2215 2216 2217
        if len(filesToRead) > 0:
            self.stream_file = {}
            self.stream_contents = []
            self.stream_have_file_info = False
2218

2219 2220 2221
            # curry self argument
            def streamP4FilesCbSelf(entry):
                self.streamP4FilesCb(entry)
H
Han-Wen Nienhuys 已提交
2222

2223 2224 2225 2226 2227
            fileArgs = ['%s#%s' % (f['path'], f['rev']) for f in filesToRead]

            p4CmdList(["-x", "-", "print"],
                      stdin=fileArgs,
                      cb=streamP4FilesCbSelf)
2228

2229 2230 2231
            # do the last chunk
            if self.stream_file.has_key('depotFile'):
                self.streamOneP4File(self.stream_file, self.stream_contents)
H
Han-Wen Nienhuys 已提交
2232

2233 2234 2235 2236 2237 2238
    def make_email(self, userid):
        if userid in self.users:
            return self.users[userid]
        else:
            return "%s <a@b>" % userid

2239 2240 2241 2242 2243 2244 2245 2246 2247 2248 2249 2250 2251 2252 2253 2254 2255 2256 2257 2258 2259 2260 2261 2262 2263 2264 2265 2266 2267 2268 2269 2270
    # Stream a p4 tag
    def streamTag(self, gitStream, labelName, labelDetails, commit, epoch):
        if verbose:
            print "writing tag %s for commit %s" % (labelName, commit)
        gitStream.write("tag %s\n" % labelName)
        gitStream.write("from %s\n" % commit)

        if labelDetails.has_key('Owner'):
            owner = labelDetails["Owner"]
        else:
            owner = None

        # Try to use the owner of the p4 label, or failing that,
        # the current p4 user id.
        if owner:
            email = self.make_email(owner)
        else:
            email = self.make_email(self.p4UserId())
        tagger = "%s %s %s" % (email, epoch, self.tz)

        gitStream.write("tagger %s\n" % tagger)

        print "labelDetails=",labelDetails
        if labelDetails.has_key('Description'):
            description = labelDetails['Description']
        else:
            description = 'Label from git p4'

        gitStream.write("data %d\n" % len(description))
        gitStream.write(description)
        gitStream.write("\n")

2271
    def commit(self, details, files, branch, parent = ""):
2272 2273 2274
        epoch = details["time"]
        author = details["user"]

2275 2276 2277
        if self.verbose:
            print "commit into %s" % branch

2278 2279 2280 2281
        # start with reading files; if that fails, we should not
        # create a commit.
        new_files = []
        for f in files:
2282
            if [p for p in self.branchPrefixes if p4PathStartsWith(f['path'], p)]:
2283 2284
                new_files.append (f)
            else:
2285
                sys.stderr.write("Ignoring file outside of prefix: %s\n" % f['path'])
2286

2287
        self.gitStream.write("commit %s\n" % branch)
H
Han-Wen Nienhuys 已提交
2288
#        gitStream.write("mark :%s\n" % details["change"])
2289 2290
        self.committedChanges.add(int(details["change"]))
        committer = ""
2291 2292
        if author not in self.users:
            self.getUserMapFromPerforceServer()
2293
        committer = "%s %s %s" % (self.make_email(author), epoch, self.tz)
2294 2295 2296 2297 2298

        self.gitStream.write("committer %s\n" % committer)

        self.gitStream.write("data <<EOT\n")
        self.gitStream.write(details["desc"])
2299 2300
        self.gitStream.write("\n[git-p4: depot-paths = \"%s\": change = %s" %
                             (','.join(self.branchPrefixes), details["change"]))
2301 2302 2303
        if len(details['options']) > 0:
            self.gitStream.write(": options = %s" % details['options'])
        self.gitStream.write("]\nEOT\n\n")
2304 2305

        if len(parent) > 0:
2306 2307
            if self.verbose:
                print "parent %s" % parent
2308 2309
            self.gitStream.write("from %s\n" % parent)

2310
        self.streamP4Files(new_files)
2311 2312
        self.gitStream.write("\n")

2313 2314
        change = int(details["change"])

2315
        if self.labels.has_key(change):
2316 2317 2318
            label = self.labels[change]
            labelDetails = label[0]
            labelRevisions = label[1]
2319 2320
            if self.verbose:
                print "Change %s is labelled %s" % (change, labelDetails)
2321

2322
            files = p4CmdList(["files"] + ["%s...@%s" % (p, change)
2323
                                                for p in self.branchPrefixes])
2324 2325 2326 2327 2328

            if len(files) == len(labelRevisions):

                cleanedFiles = {}
                for info in files:
2329
                    if info["action"] in self.delete_actions:
2330 2331 2332 2333
                        continue
                    cleanedFiles[info["depotFile"]] = info["rev"]

                if cleanedFiles == labelRevisions:
2334
                    self.streamTag(self.gitStream, 'tag_%s' % labelDetails['label'], labelDetails, branch, epoch)
2335 2336

                else:
2337
                    if not self.silent:
2338 2339
                        print ("Tag %s does not match with change %s: files do not match."
                               % (labelDetails["label"], change))
2340 2341

            else:
2342
                if not self.silent:
2343 2344
                    print ("Tag %s does not match with change %s: file count is different."
                           % (labelDetails["label"], change))
2345

2346
    # Build a dictionary of changelists and labels, for "detect-labels" option.
2347 2348 2349
    def getLabels(self):
        self.labels = {}

2350
        l = p4CmdList(["labels"] + ["%s..." % p for p in self.depotPaths])
S
Simon Hausmann 已提交
2351
        if len(l) > 0 and not self.silent:
2352
            print "Finding files belonging to labels in %s" % `self.depotPaths`
S
Simon Hausmann 已提交
2353 2354

        for output in l:
2355 2356 2357
            label = output["label"]
            revisions = {}
            newestChange = 0
2358 2359
            if self.verbose:
                print "Querying files for label %s" % label
2360 2361 2362
            for file in p4CmdList(["files"] +
                                      ["%s...@%s" % (p, label)
                                          for p in self.depotPaths]):
2363 2364 2365 2366 2367
                revisions[file["depotFile"]] = file["rev"]
                change = int(file["change"])
                if change > newestChange:
                    newestChange = change

2368 2369 2370 2371
            self.labels[newestChange] = [output, revisions]

        if self.verbose:
            print "Label changes: %s" % self.labels.keys()
2372

2373 2374 2375 2376 2377 2378 2379 2380 2381
    # Import p4 labels as git tags. A direct mapping does not
    # exist, so assume that if all the files are at the same revision
    # then we can use that, or it's something more complicated we should
    # just ignore.
    def importP4Labels(self, stream, p4Labels):
        if verbose:
            print "import p4 labels: " + ' '.join(p4Labels)

        ignoredP4Labels = gitConfigList("git-p4.ignoredP4Labels")
2382
        validLabelRegexp = gitConfig("git-p4.labelImportRegexp")
2383 2384 2385 2386 2387 2388 2389 2390 2391 2392 2393 2394 2395 2396 2397 2398 2399 2400 2401 2402 2403 2404 2405 2406 2407 2408 2409 2410 2411 2412 2413 2414 2415 2416 2417
        if len(validLabelRegexp) == 0:
            validLabelRegexp = defaultLabelRegexp
        m = re.compile(validLabelRegexp)

        for name in p4Labels:
            commitFound = False

            if not m.match(name):
                if verbose:
                    print "label %s does not match regexp %s" % (name,validLabelRegexp)
                continue

            if name in ignoredP4Labels:
                continue

            labelDetails = p4CmdList(['label', "-o", name])[0]

            # get the most recent changelist for each file in this label
            change = p4Cmd(["changes", "-m", "1"] + ["%s...@%s" % (p, name)
                                for p in self.depotPaths])

            if change.has_key('change'):
                # find the corresponding git commit; take the oldest commit
                changelist = int(change['change'])
                gitCommit = read_pipe(["git", "rev-list", "--max-count=1",
                     "--reverse", ":/\[git-p4:.*change = %d\]" % changelist])
                if len(gitCommit) == 0:
                    print "could not find git commit for changelist %d" % changelist
                else:
                    gitCommit = gitCommit.strip()
                    commitFound = True
                    # Convert from p4 time format
                    try:
                        tmwhen = time.strptime(labelDetails['Update'], "%Y/%m/%d %H:%M:%S")
                    except ValueError:
2418
                        print "Could not convert label time %s" % labelDetails['Update']
2419 2420 2421 2422 2423 2424 2425 2426 2427 2428 2429 2430 2431 2432 2433 2434 2435
                        tmwhen = 1

                    when = int(time.mktime(tmwhen))
                    self.streamTag(stream, name, labelDetails, gitCommit, when)
                    if verbose:
                        print "p4 label %s mapped to git commit %s" % (name, gitCommit)
            else:
                if verbose:
                    print "Label %s has no changelists - possibly deleted?" % name

            if not commitFound:
                # We can't import this label; don't try again as it will get very
                # expensive repeatedly fetching all the files for labels that will
                # never be imported. If the label is moved in the future, the
                # ignore will need to be removed manually.
                system(["git", "config", "--add", "git-p4.ignoredP4Labels", name])

2436 2437
    def guessProjectName(self):
        for p in self.depotPaths:
S
Simon Hausmann 已提交
2438 2439 2440 2441 2442 2443
            if p.endswith("/"):
                p = p[:-1]
            p = p[p.strip().rfind("/") + 1:]
            if not p.endswith("/"):
               p += "/"
            return p
2444

2445
    def getBranchMapping(self):
2446 2447
        lostAndFoundBranches = set()

2448 2449 2450 2451 2452 2453 2454
        user = gitConfig("git-p4.branchUser")
        if len(user) > 0:
            command = "branches -u %s" % user
        else:
            command = "branches"

        for info in p4CmdList(command):
2455
            details = p4Cmd(["branch", "-o", info["branch"]])
2456 2457 2458 2459 2460 2461 2462 2463 2464
            viewIdx = 0
            while details.has_key("View%s" % viewIdx):
                paths = details["View%s" % viewIdx].split(" ")
                viewIdx = viewIdx + 1
                # require standard //depot/foo/... //depot/bar/... mapping
                if len(paths) != 2 or not paths[0].endswith("/...") or not paths[1].endswith("/..."):
                    continue
                source = paths[0]
                destination = paths[1]
2465
                ## HACK
2466
                if p4PathStartsWith(source, self.depotPaths[0]) and p4PathStartsWith(destination, self.depotPaths[0]):
2467 2468
                    source = source[len(self.depotPaths[0]):-4]
                    destination = destination[len(self.depotPaths[0]):-4]
2469

2470 2471 2472 2473 2474 2475
                    if destination in self.knownBranches:
                        if not self.silent:
                            print "p4 branch %s defines a mapping from %s to %s" % (info["branch"], source, destination)
                            print "but there exists another mapping from %s to %s already!" % (self.knownBranches[destination], destination)
                        continue

2476 2477 2478 2479
                    self.knownBranches[destination] = source

                    lostAndFoundBranches.discard(destination)

2480
                    if source not in self.knownBranches:
2481 2482
                        lostAndFoundBranches.add(source)

2483 2484 2485 2486 2487 2488 2489 2490 2491 2492 2493 2494 2495 2496 2497 2498 2499 2500 2501
        # Perforce does not strictly require branches to be defined, so we also
        # check git config for a branch list.
        #
        # Example of branch definition in git config file:
        # [git-p4]
        #   branchList=main:branchA
        #   branchList=main:branchB
        #   branchList=branchA:branchC
        configBranches = gitConfigList("git-p4.branchList")
        for branch in configBranches:
            if branch:
                (source, destination) = branch.split(":")
                self.knownBranches[destination] = source

                lostAndFoundBranches.discard(destination)

                if source not in self.knownBranches:
                    lostAndFoundBranches.add(source)

2502 2503 2504

        for branch in lostAndFoundBranches:
            self.knownBranches[branch] = branch
2505

2506 2507 2508 2509 2510 2511 2512 2513 2514
    def getBranchMappingFromGitBranches(self):
        branches = p4BranchesInGit(self.importIntoRemotes)
        for branch in branches.keys():
            if branch == "master":
                branch = "main"
            else:
                branch = branch[len(self.projectName):]
            self.knownBranches[branch] = branch

H
Han-Wen Nienhuys 已提交
2515 2516 2517 2518 2519 2520 2521 2522 2523 2524
    def updateOptionDict(self, d):
        option_keys = {}
        if self.keepRepoPath:
            option_keys['keepRepoPath'] = 1

        d["options"] = ' '.join(sorted(option_keys.keys()))

    def readOptions(self, d):
        self.keepRepoPath = (d.has_key('options')
                             and ('keepRepoPath' in d['options']))
2525

2526 2527 2528 2529 2530 2531 2532 2533 2534
    def gitRefForBranch(self, branch):
        if branch == "main":
            return self.refPrefix + "master"

        if len(branch) <= 0:
            return branch

        return self.refPrefix + self.projectName + branch

2535 2536 2537 2538 2539 2540 2541 2542 2543 2544 2545 2546 2547 2548 2549 2550 2551 2552 2553 2554 2555 2556 2557 2558 2559 2560 2561 2562 2563 2564 2565 2566 2567 2568 2569 2570 2571 2572 2573 2574 2575 2576 2577 2578 2579 2580 2581 2582 2583 2584 2585
    def gitCommitByP4Change(self, ref, change):
        if self.verbose:
            print "looking in ref " + ref + " for change %s using bisect..." % change

        earliestCommit = ""
        latestCommit = parseRevision(ref)

        while True:
            if self.verbose:
                print "trying: earliest %s latest %s" % (earliestCommit, latestCommit)
            next = read_pipe("git rev-list --bisect %s %s" % (latestCommit, earliestCommit)).strip()
            if len(next) == 0:
                if self.verbose:
                    print "argh"
                return ""
            log = extractLogMessageFromGitCommit(next)
            settings = extractSettingsGitLog(log)
            currentChange = int(settings['change'])
            if self.verbose:
                print "current change %s" % currentChange

            if currentChange == change:
                if self.verbose:
                    print "found %s" % next
                return next

            if currentChange < change:
                earliestCommit = "^%s" % next
            else:
                latestCommit = "%s" % next

        return ""

    def importNewBranch(self, branch, maxChange):
        # make fast-import flush all changes to disk and update the refs using the checkpoint
        # command so that we can try to find the branch parent in the git history
        self.gitStream.write("checkpoint\n\n");
        self.gitStream.flush();
        branchPrefix = self.depotPaths[0] + branch + "/"
        range = "@1,%s" % maxChange
        #print "prefix" + branchPrefix
        changes = p4ChangesForPaths([branchPrefix], range)
        if len(changes) <= 0:
            return False
        firstChange = changes[0]
        #print "first change in branch: %s" % firstChange
        sourceBranch = self.knownBranches[branch]
        sourceDepotPath = self.depotPaths[0] + sourceBranch
        sourceRef = self.gitRefForBranch(sourceBranch)
        #print "source " + sourceBranch

2586
        branchParentChange = int(p4Cmd(["changes", "-m", "1", "%s...@1,%s" % (sourceDepotPath, firstChange)])["change"])
2587 2588 2589 2590 2591 2592 2593 2594 2595
        #print "branch parent: %s" % branchParentChange
        gitParent = self.gitCommitByP4Change(sourceRef, branchParentChange)
        if len(gitParent) > 0:
            self.initialParents[self.gitRefForBranch(branch)] = gitParent
            #print "parent git commit: %s" % gitParent

        self.importChanges(changes)
        return True

2596 2597 2598 2599 2600 2601 2602 2603 2604 2605 2606 2607 2608 2609
    def searchParent(self, parent, branch, target):
        parentFound = False
        for blob in read_pipe_lines(["git", "rev-list", "--reverse", "--no-merges", parent]):
            blob = blob.strip()
            if len(read_pipe(["git", "diff-tree", blob, target])) == 0:
                parentFound = True
                if self.verbose:
                    print "Found parent of %s in commit %s" % (branch, blob)
                break
        if parentFound:
            return blob
        else:
            return None

2610 2611 2612
    def importChanges(self, changes):
        cnt = 1
        for change in changes:
P
Pete Wyckoff 已提交
2613
            description = p4_describe(change)
2614 2615 2616 2617 2618 2619 2620 2621 2622 2623 2624 2625 2626
            self.updateOptionDict(description)

            if not self.silent:
                sys.stdout.write("\rImporting revision %s (%s%%)" % (change, cnt * 100 / len(changes)))
                sys.stdout.flush()
            cnt = cnt + 1

            try:
                if self.detectBranches:
                    branches = self.splitFilesIntoBranches(description)
                    for branch in branches.keys():
                        ## HACK  --hwn
                        branchPrefix = self.depotPaths[0] + branch + "/"
2627
                        self.branchPrefixes = [ branchPrefix ]
2628 2629 2630 2631 2632 2633 2634 2635 2636 2637 2638 2639 2640 2641 2642

                        parent = ""

                        filesForCommit = branches[branch]

                        if self.verbose:
                            print "branch is %s" % branch

                        self.updatedBranches.add(branch)

                        if branch not in self.createdBranches:
                            self.createdBranches.add(branch)
                            parent = self.knownBranches[branch]
                            if parent == branch:
                                parent = ""
2643 2644 2645 2646 2647 2648 2649 2650 2651 2652 2653 2654 2655
                            else:
                                fullBranch = self.projectName + branch
                                if fullBranch not in self.p4BranchesInGit:
                                    if not self.silent:
                                        print("\n    Importing new branch %s" % fullBranch);
                                    if self.importNewBranch(branch, change - 1):
                                        parent = ""
                                        self.p4BranchesInGit.append(fullBranch)
                                    if not self.silent:
                                        print("\n    Resuming with change %s" % change);

                                if self.verbose:
                                    print "parent determined through known branches: %s" % parent
2656

2657 2658
                        branch = self.gitRefForBranch(branch)
                        parent = self.gitRefForBranch(parent)
2659 2660 2661 2662 2663 2664 2665 2666

                        if self.verbose:
                            print "looking for initial parent for %s; current parent is %s" % (branch, parent)

                        if len(parent) == 0 and branch in self.initialParents:
                            parent = self.initialParents[branch]
                            del self.initialParents[branch]

2667 2668 2669 2670 2671
                        blob = None
                        if len(parent) > 0:
                            tempBranch = os.path.join(self.tempBranchLocation, "%d" % (change))
                            if self.verbose:
                                print "Creating temporary branch: " + tempBranch
2672
                            self.commit(description, filesForCommit, tempBranch)
2673 2674 2675 2676
                            self.tempBranches.append(tempBranch)
                            self.checkpoint()
                            blob = self.searchParent(parent, branch, tempBranch)
                        if blob:
2677
                            self.commit(description, filesForCommit, branch, blob)
2678 2679 2680
                        else:
                            if self.verbose:
                                print "Parent of %s not found. Committing into head of %s" % (branch, parent)
2681
                            self.commit(description, filesForCommit, branch, parent)
2682 2683
                else:
                    files = self.extractFilesFromCommit(description)
2684
                    self.commit(description, files, self.branch,
2685 2686 2687 2688 2689 2690
                                self.initialParent)
                    self.initialParent = ""
            except IOError:
                print self.gitError.read()
                sys.exit(1)

2691 2692 2693
    def importHeadRevision(self, revision):
        print "Doing initial import of %s from revision %s into %s" % (' '.join(self.depotPaths), revision, self.branch)

2694 2695
        details = {}
        details["user"] = "git perforce import user"
2696
        details["desc"] = ("Initial import of %s from the state at revision %s\n"
2697 2698 2699 2700 2701
                           % (' '.join(self.depotPaths), revision))
        details["change"] = revision
        newestRevision = 0

        fileCnt = 0
2702 2703 2704
        fileArgs = ["%s...%s" % (p,revision) for p in self.depotPaths]

        for info in p4CmdList(["files"] + fileArgs):
2705

2706
            if 'code' in info and info['code'] == 'error':
2707 2708
                sys.stderr.write("p4 returned an error: %s\n"
                                 % info['data'])
2709 2710 2711 2712
                if info['data'].find("must refer to client") >= 0:
                    sys.stderr.write("This particular p4 error is misleading.\n")
                    sys.stderr.write("Perhaps the depot path was misspelled.\n");
                    sys.stderr.write("Depot path:  %s\n" % " ".join(self.depotPaths))
2713
                sys.exit(1)
2714 2715
            if 'p4ExitCode' in info:
                sys.stderr.write("p4 exitcode: %s\n" % info['p4ExitCode'])
2716 2717 2718 2719 2720 2721 2722
                sys.exit(1)


            change = int(info["change"])
            if change > newestRevision:
                newestRevision = change

2723
            if info["action"] in self.delete_actions:
2724 2725 2726 2727 2728 2729 2730 2731 2732 2733
                # don't increase the file cnt, otherwise details["depotFile123"] will have gaps!
                #fileCnt = fileCnt + 1
                continue

            for prop in ["depotFile", "rev", "action", "type" ]:
                details["%s%s" % (prop, fileCnt)] = info[prop]

            fileCnt = fileCnt + 1

        details["change"] = newestRevision
2734

P
Pete Wyckoff 已提交
2735
        # Use time from top-most change so that all git p4 clones of
2736
        # the same p4 repo have the same commit SHA1s.
P
Pete Wyckoff 已提交
2737 2738
        res = p4_describe(newestRevision)
        details["time"] = res["time"]
2739

2740 2741
        self.updateOptionDict(details)
        try:
2742
            self.commit(details, self.extractFilesFromCommit(details), self.branch)
2743 2744 2745 2746 2747
        except IOError:
            print "IO error with git fast-import. Is your git version recent enough?"
            print self.gitError.read()


2748
    def run(self, args):
2749
        self.depotPaths = []
2750 2751
        self.changeRange = ""
        self.initialParent = ""
2752
        self.previousDepotPaths = []
2753
        self.hasOrigin = False
H
Han-Wen Nienhuys 已提交
2754

2755 2756 2757 2758
        # map from branch depot path to parent branch
        self.knownBranches = {}
        self.initialParents = {}

2759 2760 2761
        if self.importIntoRemotes:
            self.refPrefix = "refs/remotes/p4/"
        else:
2762
            self.refPrefix = "refs/heads/p4/"
2763

2764 2765 2766 2767 2768 2769
        if self.syncWithOrigin:
            self.hasOrigin = originP4BranchesExist()
            if self.hasOrigin:
                if not self.silent:
                    print 'Syncing with origin first, using "git fetch origin"'
                system("git fetch origin")
2770

2771
        if len(self.branch) == 0:
2772
            self.branch = self.refPrefix + "master"
2773
            if gitBranchExists("refs/heads/p4") and self.importIntoRemotes:
2774
                system("git update-ref %s refs/heads/p4" % self.branch)
2775
                system("git branch -D p4")
2776

2777 2778 2779 2780 2781
        # accept either the command-line option, or the configuration variable
        if self.useClientSpec:
            # will use this after clone to set the variable
            self.useClientSpec_from_options = True
        else:
2782 2783 2784
            if gitConfig("git-p4.useclientspec", "--bool") == "true":
                self.useClientSpec = True
        if self.useClientSpec:
2785
            self.clientSpecDirs = getClientSpec()
2786

H
Han-Wen Nienhuys 已提交
2787 2788 2789
        # TODO: should always look at previous commits,
        # merge with previous imports, if possible.
        if args == []:
2790
            if self.hasOrigin:
2791
                createOrUpdateBranchesFromOrigin(self.refPrefix, self.silent)
2792 2793 2794 2795 2796 2797

            # branches holds mapping from branch name to sha1
            branches = p4BranchesInGit(self.importIntoRemotes)
            self.p4BranchesInGit = branches.keys()
            for branch in branches.keys():
                self.initialParents[self.refPrefix + branch] = branches[branch]
2798 2799 2800 2801 2802

            if len(self.p4BranchesInGit) > 1:
                if not self.silent:
                    print "Importing from/into multiple branches"
                self.detectBranches = True
2803

2804 2805 2806 2807 2808
            if self.verbose:
                print "branches: %s" % self.p4BranchesInGit

            p4Change = 0
            for branch in self.p4BranchesInGit:
2809
                logMsg =  extractLogMessageFromGitCommit(self.refPrefix + branch)
H
Han-Wen Nienhuys 已提交
2810 2811

                settings = extractSettingsGitLog(logMsg)
2812

H
Han-Wen Nienhuys 已提交
2813 2814 2815 2816
                self.readOptions(settings)
                if (settings.has_key('depot-paths')
                    and settings.has_key ('change')):
                    change = int(settings['change']) + 1
2817 2818
                    p4Change = max(p4Change, change)

H
Han-Wen Nienhuys 已提交
2819 2820
                    depotPaths = sorted(settings['depot-paths'])
                    if self.previousDepotPaths == []:
2821
                        self.previousDepotPaths = depotPaths
2822
                    else:
2823 2824
                        paths = []
                        for (prev, cur) in zip(self.previousDepotPaths, depotPaths):
2825 2826 2827 2828
                            prev_list = prev.split("/")
                            cur_list = cur.split("/")
                            for i in range(0, min(len(cur_list), len(prev_list))):
                                if cur_list[i] <> prev_list[i]:
2829
                                    i = i - 1
2830 2831
                                    break

2832
                            paths.append ("/".join(cur_list[:i + 1]))
2833 2834

                        self.previousDepotPaths = paths
2835 2836

            if p4Change > 0:
H
Han-Wen Nienhuys 已提交
2837
                self.depotPaths = sorted(self.previousDepotPaths)
2838
                self.changeRange = "@%s,#head" % p4Change
2839 2840
                if not self.detectBranches:
                    self.initialParent = parseRevision(self.branch)
2841
                if not self.silent and not self.detectBranches:
2842
                    print "Performing incremental import into %s git branch" % self.branch
2843

2844 2845 2846 2847
        # accept multiple ref name abbreviations:
        #    refs/foo/bar/branch -> use it exactly
        #    p4/branch -> prepend refs/remotes/ or refs/heads/
        #    branch -> prepend refs/remotes/p4/ or refs/heads/p4/
2848
        if not self.branch.startswith("refs/"):
2849 2850 2851 2852 2853 2854 2855
            if self.importIntoRemotes:
                prepend = "refs/remotes/"
            else:
                prepend = "refs/heads/"
            if not self.branch.startswith("p4/"):
                prepend += "p4/"
            self.branch = prepend + self.branch
2856

2857
        if len(args) == 0 and self.depotPaths:
2858
            if not self.silent:
2859
                print "Depot paths: %s" % ' '.join(self.depotPaths)
2860
        else:
2861
            if self.depotPaths and self.depotPaths != args:
2862
                print ("previous import used depot path %s and now %s was specified. "
2863 2864
                       "This doesn't work!" % (' '.join (self.depotPaths),
                                               ' '.join (args)))
2865
                sys.exit(1)
2866

H
Han-Wen Nienhuys 已提交
2867
            self.depotPaths = sorted(args)
2868

2869
        revision = ""
2870 2871
        self.users = {}

2872 2873 2874 2875 2876 2877 2878 2879 2880 2881 2882
        # Make sure no revision specifiers are used when --changesfile
        # is specified.
        bad_changesfile = False
        if len(self.changesFile) > 0:
            for p in self.depotPaths:
                if p.find("@") >= 0 or p.find("#") >= 0:
                    bad_changesfile = True
                    break
        if bad_changesfile:
            die("Option --changesfile is incompatible with revision specifiers")

2883 2884 2885 2886 2887 2888 2889
        newPaths = []
        for p in self.depotPaths:
            if p.find("@") != -1:
                atIdx = p.index("@")
                self.changeRange = p[atIdx:]
                if self.changeRange == "@all":
                    self.changeRange = ""
H
Han-Wen Nienhuys 已提交
2890
                elif ',' not in self.changeRange:
2891
                    revision = self.changeRange
2892
                    self.changeRange = ""
2893
                p = p[:atIdx]
2894 2895
            elif p.find("#") != -1:
                hashIdx = p.index("#")
2896
                revision = p[hashIdx:]
2897
                p = p[:hashIdx]
2898
            elif self.previousDepotPaths == []:
2899 2900 2901 2902
                # pay attention to changesfile, if given, else import
                # the entire p4 tree at the head revision
                if len(self.changesFile) == 0:
                    revision = "#head"
2903 2904 2905 2906 2907 2908 2909 2910 2911

            p = re.sub ("\.\.\.$", "", p)
            if not p.endswith("/"):
                p += "/"

            newPaths.append(p)

        self.depotPaths = newPaths

2912 2913 2914
        # --detect-branches may change this for each branch
        self.branchPrefixes = self.depotPaths

2915
        self.loadUserMapFromCache()
2916 2917 2918
        self.labels = {}
        if self.detectLabels:
            self.getLabels();
2919

2920
        if self.detectBranches:
2921 2922 2923
            ## FIXME - what's a P4 projectName ?
            self.projectName = self.guessProjectName()

2924 2925 2926 2927
            if self.hasOrigin:
                self.getBranchMappingFromGitBranches()
            else:
                self.getBranchMapping()
2928 2929 2930 2931 2932
            if self.verbose:
                print "p4-git branches: %s" % self.p4BranchesInGit
                print "initial parents: %s" % self.initialParents
            for b in self.p4BranchesInGit:
                if b != "master":
2933 2934

                    ## FIXME
2935 2936
                    b = b[len(self.projectName):]
                self.createdBranches.add(b)
2937

2938
        self.tz = "%+03d%02d" % (- time.timezone / 3600, ((- time.timezone % 3600) / 60))
2939

2940 2941 2942 2943 2944 2945 2946
        self.importProcess = subprocess.Popen(["git", "fast-import"],
                                              stdin=subprocess.PIPE,
                                              stdout=subprocess.PIPE,
                                              stderr=subprocess.PIPE);
        self.gitOutput = self.importProcess.stdout
        self.gitStream = self.importProcess.stdin
        self.gitError = self.importProcess.stderr
2947

2948
        if revision:
2949
            self.importHeadRevision(revision)
2950 2951 2952
        else:
            changes = []

2953
            if len(self.changesFile) > 0:
2954
                output = open(self.changesFile).readlines()
2955
                changeSet = set()
2956 2957 2958 2959 2960 2961 2962 2963
                for line in output:
                    changeSet.add(int(line))

                for change in changeSet:
                    changes.append(change)

                changes.sort()
            else:
P
Pete Wyckoff 已提交
2964 2965
                # catch "git p4 sync" with no new branches, in a repo that
                # does not have any existing p4 branches
2966
                if len(args) == 0 and not self.p4BranchesInGit:
2967
                    die("No remote p4 branches.  Perhaps you never did \"git p4 clone\" in here.");
2968
                if self.verbose:
2969
                    print "Getting p4 changes for %s...%s" % (', '.join(self.depotPaths),
2970
                                                              self.changeRange)
2971
                changes = p4ChangesForPaths(self.depotPaths, self.changeRange)
2972

2973
                if len(self.maxChanges) > 0:
2974
                    changes = changes[:min(int(self.maxChanges), len(changes))]
2975

2976
            if len(changes) == 0:
2977
                if not self.silent:
2978
                    print "No changes to import!"
2979 2980 2981 2982 2983
            else:
                if not self.silent and not self.detectBranches:
                    print "Import destination: %s" % self.branch

                self.updatedBranches = set()
2984

2985
                self.importChanges(changes)
2986

2987 2988 2989 2990 2991 2992 2993
                if not self.silent:
                    print ""
                    if len(self.updatedBranches) > 0:
                        sys.stdout.write("Updated branches: ")
                        for b in self.updatedBranches:
                            sys.stdout.write("%s " % b)
                        sys.stdout.write("\n")
2994

2995
        if gitConfig("git-p4.importLabels", "--bool") == "true":
2996
            self.importLabels = True
2997

2998 2999 3000 3001 3002 3003
        if self.importLabels:
            p4Labels = getP4Labels(self.depotPaths)
            gitTags = getGitTags()

            missingP4Labels = p4Labels - gitTags
            self.importP4Labels(self.gitStream, missingP4Labels)
3004 3005

        self.gitStream.close()
3006
        if self.importProcess.wait() != 0:
3007
            die("fast-import failed: %s" % self.gitError.read())
3008 3009 3010
        self.gitOutput.close()
        self.gitError.close()

3011 3012 3013 3014 3015 3016
        # Cleanup temporary branches created during import
        if self.tempBranches != []:
            for branch in self.tempBranches:
                read_pipe("git update-ref -d %s" % branch)
            os.rmdir(os.path.join(os.environ.get("GIT_DIR", ".git"), self.tempBranchLocation))

3017 3018 3019 3020 3021 3022 3023
        # Create a symbolic ref p4/HEAD pointing to p4/<branch> to allow
        # a convenient shortcut refname "p4".
        if self.importIntoRemotes:
            head_ref = self.refPrefix + "HEAD"
            if not gitBranchExists(head_ref) and gitBranchExists(self.branch):
                system(["git", "symbolic-ref", head_ref, self.branch])

3024 3025
        return True

S
Simon Hausmann 已提交
3026 3027 3028
class P4Rebase(Command):
    def __init__(self):
        Command.__init__(self)
3029 3030 3031 3032
        self.options = [
                optparse.make_option("--import-labels", dest="importLabels", action="store_true"),
        ]
        self.importLabels = False
3033 3034
        self.description = ("Fetches the latest revision from perforce and "
                            + "rebases the current work (branch) against it")
S
Simon Hausmann 已提交
3035 3036 3037

    def run(self, args):
        sync = P4Sync()
3038
        sync.importLabels = self.importLabels
S
Simon Hausmann 已提交
3039
        sync.run([])
3040

3041 3042 3043
        return self.rebase()

    def rebase(self):
3044 3045 3046 3047 3048
        if os.system("git update-index --refresh") != 0:
            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.");
        if len(read_pipe("git diff-index HEAD --")) > 0:
            die("You have uncommited changes. Please commit them before rebasing or stash them away with git stash.");

3049 3050 3051 3052 3053 3054 3055 3056
        [upstream, settings] = findUpstreamBranchPoint()
        if len(upstream) == 0:
            die("Cannot find upstream branchpoint for rebase")

        # the branchpoint may be p4/foo~3, so strip off the parent
        upstream = re.sub("~[0-9]+$", "", upstream)

        print "Rebasing the current branch onto %s" % upstream
3057
        oldHead = read_pipe("git rev-parse HEAD").strip()
3058
        system("git rebase %s" % upstream)
3059
        system("git diff-tree --stat --summary -M %s HEAD" % oldHead)
S
Simon Hausmann 已提交
3060 3061
        return True

3062 3063 3064 3065
class P4Clone(P4Sync):
    def __init__(self):
        P4Sync.__init__(self)
        self.description = "Creates a new git repository and imports from Perforce into it"
H
Han-Wen Nienhuys 已提交
3066
        self.usage = "usage: %prog [options] //depot/path[@revRange]"
T
Tommy Thorn 已提交
3067
        self.options += [
H
Han-Wen Nienhuys 已提交
3068 3069
            optparse.make_option("--destination", dest="cloneDestination",
                                 action='store', default=None,
T
Tommy Thorn 已提交
3070 3071 3072
                                 help="where to leave result of the clone"),
            optparse.make_option("-/", dest="cloneExclude",
                                 action="append", type="string",
P
Pete Wyckoff 已提交
3073 3074 3075
                                 help="exclude depot path"),
            optparse.make_option("--bare", dest="cloneBare",
                                 action="store_true", default=False),
T
Tommy Thorn 已提交
3076
        ]
H
Han-Wen Nienhuys 已提交
3077
        self.cloneDestination = None
3078
        self.needsGit = False
P
Pete Wyckoff 已提交
3079
        self.cloneBare = False
3080

T
Tommy Thorn 已提交
3081 3082 3083 3084 3085 3086
    # This is required for the "append" cloneExclude action
    def ensure_value(self, attr, value):
        if not hasattr(self, attr) or getattr(self, attr) is None:
            setattr(self, attr, value)
        return getattr(self, attr)

H
Han-Wen Nienhuys 已提交
3087 3088 3089 3090 3091
    def defaultDestination(self, args):
        ## TODO: use common prefix of args?
        depotPath = args[0]
        depotDir = re.sub("(@[^@]*)$", "", depotPath)
        depotDir = re.sub("(#[^#]*)$", "", depotDir)
3092
        depotDir = re.sub(r"\.\.\.$", "", depotDir)
H
Han-Wen Nienhuys 已提交
3093 3094 3095
        depotDir = re.sub(r"/$", "", depotDir)
        return os.path.split(depotDir)[1]

3096 3097 3098
    def run(self, args):
        if len(args) < 1:
            return False
H
Han-Wen Nienhuys 已提交
3099 3100 3101 3102

        if self.keepRepoPath and not self.cloneDestination:
            sys.stderr.write("Must specify destination for --keep-path\n")
            sys.exit(1)
3103

3104
        depotPaths = args
3105 3106 3107 3108 3109

        if not self.cloneDestination and len(depotPaths) > 1:
            self.cloneDestination = depotPaths[-1]
            depotPaths = depotPaths[:-1]

T
Tommy Thorn 已提交
3110
        self.cloneExclude = ["/"+p for p in self.cloneExclude]
3111 3112 3113
        for p in depotPaths:
            if not p.startswith("//"):
                return False
3114

H
Han-Wen Nienhuys 已提交
3115
        if not self.cloneDestination:
3116
            self.cloneDestination = self.defaultDestination(args)
3117

3118
        print "Importing from %s into %s" % (', '.join(depotPaths), self.cloneDestination)
P
Pete Wyckoff 已提交
3119

3120 3121
        if not os.path.exists(self.cloneDestination):
            os.makedirs(self.cloneDestination)
3122
        chdir(self.cloneDestination)
P
Pete Wyckoff 已提交
3123 3124 3125 3126 3127 3128

        init_cmd = [ "git", "init" ]
        if self.cloneBare:
            init_cmd.append("--bare")
        subprocess.check_call(init_cmd)

3129
        if not P4Sync.run(self, depotPaths):
3130
            return False
3131 3132 3133 3134 3135 3136 3137 3138 3139

        # create a master branch and check out a work tree
        if gitBranchExists(self.branch):
            system([ "git", "branch", "master", self.branch ])
            if not self.cloneBare:
                system([ "git", "checkout", "-f" ])
        else:
            print 'Not checking out any branch, use ' \
                  '"git checkout -q -b master <branch>"'
3140

3141 3142 3143 3144
        # auto-set this variable if invoked with --use-client-spec
        if self.useClientSpec_from_options:
            system("git config --bool git-p4.useclientspec true")

3145 3146
        return True

3147 3148 3149 3150 3151 3152 3153 3154 3155
class P4Branches(Command):
    def __init__(self):
        Command.__init__(self)
        self.options = [ ]
        self.description = ("Shows the git branches that hold imports and their "
                            + "corresponding perforce depot paths")
        self.verbose = False

    def run(self, args):
3156 3157 3158
        if originP4BranchesExist():
            createOrUpdateBranchesFromOrigin()

3159 3160 3161 3162 3163 3164 3165 3166 3167 3168 3169 3170 3171 3172 3173 3174
        cmdline = "git rev-parse --symbolic "
        cmdline += " --remotes"

        for line in read_pipe_lines(cmdline):
            line = line.strip()

            if not line.startswith('p4/') or line == "p4/HEAD":
                continue
            branch = line

            log = extractLogMessageFromGitCommit("refs/remotes/%s" % branch)
            settings = extractSettingsGitLog(log)

            print "%s <= %s (%s)" % (branch, ",".join(settings["depot-paths"]), settings["change"])
        return True

3175 3176 3177 3178 3179 3180 3181 3182 3183
class HelpFormatter(optparse.IndentedHelpFormatter):
    def __init__(self):
        optparse.IndentedHelpFormatter.__init__(self)

    def format_description(self, description):
        if description:
            return description + "\n"
        else:
            return ""
3184

3185 3186 3187 3188 3189 3190 3191 3192 3193
def printUsage(commands):
    print "usage: %s <command> [options]" % sys.argv[0]
    print ""
    print "valid commands: %s" % ", ".join(commands)
    print ""
    print "Try %s <command> --help for command specific help." % sys.argv[0]
    print ""

commands = {
H
Han-Wen Nienhuys 已提交
3194 3195
    "debug" : P4Debug,
    "submit" : P4Submit,
3196
    "commit" : P4Submit,
H
Han-Wen Nienhuys 已提交
3197 3198 3199
    "sync" : P4Sync,
    "rebase" : P4Rebase,
    "clone" : P4Clone,
3200 3201
    "rollback" : P4RollBack,
    "branches" : P4Branches
3202 3203 3204
}


H
Han-Wen Nienhuys 已提交
3205 3206 3207 3208
def main():
    if len(sys.argv[1:]) == 0:
        printUsage(commands.keys())
        sys.exit(2)
3209

H
Han-Wen Nienhuys 已提交
3210 3211
    cmdName = sys.argv[1]
    try:
H
Han-Wen Nienhuys 已提交
3212 3213
        klass = commands[cmdName]
        cmd = klass()
H
Han-Wen Nienhuys 已提交
3214 3215 3216 3217 3218 3219 3220
    except KeyError:
        print "unknown command %s" % cmdName
        print ""
        printUsage(commands.keys())
        sys.exit(2)

    options = cmd.options
H
Han-Wen Nienhuys 已提交
3221
    cmd.gitdir = os.environ.get("GIT_DIR", None)
H
Han-Wen Nienhuys 已提交
3222 3223 3224

    args = sys.argv[2:]

P
Pete Wyckoff 已提交
3225
    options.append(optparse.make_option("--verbose", "-v", dest="verbose", action="store_true"))
3226 3227
    if cmd.needsGit:
        options.append(optparse.make_option("--git-dir", dest="gitdir"))
H
Han-Wen Nienhuys 已提交
3228

3229 3230 3231 3232
    parser = optparse.OptionParser(cmd.usage.replace("%prog", "%prog " + cmdName),
                                   options,
                                   description = cmd.description,
                                   formatter = HelpFormatter())
H
Han-Wen Nienhuys 已提交
3233

3234
    (cmd, args) = parser.parse_args(sys.argv[2:], cmd);
H
Han-Wen Nienhuys 已提交
3235 3236 3237
    global verbose
    verbose = cmd.verbose
    if cmd.needsGit:
H
Han-Wen Nienhuys 已提交
3238 3239 3240 3241 3242
        if cmd.gitdir == None:
            cmd.gitdir = os.path.abspath(".git")
            if not isValidGitDir(cmd.gitdir):
                cmd.gitdir = read_pipe("git rev-parse --git-dir").strip()
                if os.path.exists(cmd.gitdir):
H
Han-Wen Nienhuys 已提交
3243 3244
                    cdup = read_pipe("git rev-parse --show-cdup").strip()
                    if len(cdup) > 0:
3245
                        chdir(cdup);
3246

H
Han-Wen Nienhuys 已提交
3247 3248 3249
        if not isValidGitDir(cmd.gitdir):
            if isValidGitDir(cmd.gitdir + "/.git"):
                cmd.gitdir += "/.git"
H
Han-Wen Nienhuys 已提交
3250
            else:
H
Han-Wen Nienhuys 已提交
3251
                die("fatal: cannot locate git repository at %s" % cmd.gitdir)
3252

H
Han-Wen Nienhuys 已提交
3253
        os.environ["GIT_DIR"] = cmd.gitdir
3254

H
Han-Wen Nienhuys 已提交
3255 3256
    if not cmd.run(args):
        parser.print_help()
3257
        sys.exit(2)
3258 3259


H
Han-Wen Nienhuys 已提交
3260 3261
if __name__ == '__main__':
    main()