git-p4.py 120.2 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 583 584 585 586 587 588 589 590 591 592
def branch_exists(branch):
    """Make sure that the given ref name really exists."""

    cmd = [ "git", "rev-parse", "--symbolic", "--verify", branch ]
    p = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    out, _ = p.communicate()
    if p.returncode:
        return False
    # expect exactly one line of output: the branch name
    return out.rstrip() == branch

593
def findUpstreamBranchPoint(head = "HEAD"):
594 595 596 597 598 599 600 601 602 603 604
    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

605 606 607
    settings = None
    parent = 0
    while parent < 65535:
608
        commit = head + "~%s" % parent
609 610
        log = extractLogMessageFromGitCommit(commit)
        settings = extractSettingsGitLog(log)
611 612 613 614
        if settings.has_key("depot-paths"):
            paths = ",".join(settings["depot-paths"])
            if branchByDepotPath.has_key(paths):
                return [branchByDepotPath[paths], settings]
615

616
        parent = parent + 1
617

618
    return ["", settings]
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 659 660 661 662 663 664 665 666 667 668 669
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")

670 671
def p4ChangesForPaths(depotPaths, changeRange):
    assert depotPaths
672 673 674 675
    cmd = ['changes']
    for p in depotPaths:
        cmd += ["%s...%s" % (p, changeRange)]
    output = p4_read_pipe_lines(cmd)
676

677
    changes = {}
678
    for line in output:
679 680
        changeNum = int(line.split(" ")[1])
        changes[changeNum] = True
681

682 683 684
    changelist = changes.keys()
    changelist.sort()
    return changelist
685

686 687 688 689 690 691 692 693 694 695 696 697 698
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)

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 728 729 730 731 732 733 734 735 736 737 738
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"]

739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766
#
# 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

767 768 769
class Command:
    def __init__(self):
        self.usage = "usage: %prog [options]"
770
        self.needsGit = True
771
        self.verbose = False
772

773 774 775
class P4UserMap:
    def __init__(self):
        self.userMapFromPerforceServer = False
776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794 795
        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
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 823 824 825 826 827 828 829 830 831 832 833

    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()

834
class P4Debug(Command):
835
    def __init__(self):
836
        Command.__init__(self)
837
        self.options = []
838
        self.description = "A tool to debug the output of p4 -G."
839
        self.needsGit = False
840 841

    def run(self, args):
842
        j = 0
843
        for output in p4CmdList(args):
844 845
            print 'Element: %d' % j
            j += 1
846
            print output
847
        return True
848

849 850 851 852
class P4RollBack(Command):
    def __init__(self):
        Command.__init__(self)
        self.options = [
853
            optparse.make_option("--local", dest="rollbackLocalBranches", action="store_true")
854 855
        ]
        self.description = "A tool to debug the multi-branch import. Don't use :)"
856
        self.rollbackLocalBranches = False
857 858 859 860 861

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

863
        if "p4ExitCode" in p4Cmd("changes -m 1"):
864 865
            die("Problems executing p4");

866 867
        if self.rollbackLocalBranches:
            refPrefix = "refs/heads/"
H
Han-Wen Nienhuys 已提交
868
            lines = read_pipe_lines("git rev-parse --symbolic --branches")
869 870
        else:
            refPrefix = "refs/remotes/"
H
Han-Wen Nienhuys 已提交
871
            lines = read_pipe_lines("git rev-parse --symbolic --remotes")
872 873 874

        for line in lines:
            if self.rollbackLocalBranches or (line.startswith("p4/") and line != "p4/HEAD\n"):
875 876
                line = line.strip()
                ref = refPrefix + line
877
                log = extractLogMessageFromGitCommit(ref)
H
Han-Wen Nienhuys 已提交
878 879 880 881 882
                settings = extractSettingsGitLog(log)

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

883
                changed = False
884

885 886
                if len(p4Cmd("changes -m 1 "  + ' '.join (['%s...@%s' % (p, maxChange)
                                                           for p in depotPaths]))) == 0:
887 888 889 890
                    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 已提交
891
                while change and int(change) > maxChange:
892
                    changed = True
893 894
                    if self.verbose:
                        print "%s is at %s ; rewinding towards %s" % (ref, change, maxChange)
895 896
                    system("git update-ref %s \"%s^\"" % (ref, ref))
                    log = extractLogMessageFromGitCommit(ref)
H
Han-Wen Nienhuys 已提交
897 898 899 900 901
                    settings =  extractSettingsGitLog(log)


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

                if changed:
904
                    print "%s rewound to %s" % (ref, change)
905 906 907

        return True

908
class P4Submit(Command, P4UserMap):
909 910 911

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

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

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

942 943 944 945 946 947 948 949 950 951 952 953 954 955 956 957 958 959 960 961 962 963 964 965 966 967 968 969
    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."""
970 971
        result = ""

972 973
        inDescriptionSection = False

974 975 976 977 978
        for line in template.split("\n"):
            if line.startswith("#"):
                result += line + "\n"
                continue

979
            if inDescriptionSection:
980
                if line.startswith("Files:") or line.startswith("Jobs:"):
981
                    inDescriptionSection = False
982 983 984
                    # insert Jobs section
                    if jobs:
                        result += jobs + "\n"
985 986 987 988 989 990 991 992 993 994
                else:
                    continue
            else:
                if line.startswith("Description:"):
                    inDescriptionSection = True
                    line += "\n"
                    for messageLine in message.split("\n"):
                        line += "\t" + messageLine + "\n"

            result += line + "\n"
995 996 997

        return result

998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020
    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

1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054
    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")
1055
        results = p4CmdList(["changes", "-c", client, "-m", "1"])
1056 1057 1058 1059 1060 1061 1062 1063
        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)
1064 1065 1066 1067 1068 1069 1070 1071 1072
        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)

1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085
        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.
1086
        results = p4CmdList(["protects", self.depotPath])
1087 1088 1089 1090 1091 1092 1093 1094
        for r in results:
            if r.has_key('perm'):
                if r['perm'] == 'admin':
                    return 1
                if r['perm'] == 'super':
                    return 1
        return 0

1095
    def prepareSubmitTemplate(self):
1096 1097 1098 1099 1100 1101 1102
        """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."""

1103 1104
        template = ""
        inFilesSection = False
1105
        for line in p4_read_pipe_lines(['change', '-o']):
1106 1107
            if line.endswith("\r\n"):
                line = line[:-2] + "\n"
1108 1109 1110 1111 1112 1113 1114
            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]
1115
                        if not p4PathStartsWith(path, self.depotPath):
1116 1117 1118 1119 1120 1121 1122 1123 1124 1125 1126
                            continue
                else:
                    inFilesSection = False
            else:
                if line.startswith("Files:"):
                    inFilesSection = True

            template += line

        return template

P
Pete Wyckoff 已提交
1127 1128 1129 1130 1131 1132 1133 1134 1135 1136 1137 1138 1139
    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
1140
        if os.environ.has_key("P4EDITOR") and (os.environ.get("P4EDITOR") != ""):
P
Pete Wyckoff 已提交
1141 1142 1143 1144 1145 1146 1147 1148 1149 1150
            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

1151 1152 1153 1154 1155 1156 1157 1158 1159 1160
        # 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 已提交
1161

1162
    def applyCommit(self, id):
1163 1164 1165 1166
        """Apply one commit, return True if it succeeded."""

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

1168
        (p4User, gitEmail) = self.p4UserForCommit(id)
1169

1170
        diff = read_pipe_lines("git diff-tree -r %s \"%s^\" \"%s\"" % (self.diffOpts, id, id))
1171 1172
        filesToAdd = set()
        filesToDelete = set()
1173
        editedFiles = set()
1174
        pureRenameCopy = set()
1175
        filesToChangeExecBit = {}
1176

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

1230
        diffcmd = "git format-patch -k --stdout \"%s^\"..\"%s\"" % (id, id)
1231
        patchcmd = diffcmd + " | git apply "
1232 1233
        tryPatchCmd = patchcmd + "--check -"
        applyPatchCmd = patchcmd + "--check --apply -"
1234
        patch_succeeded = True
1235

1236
        if os.system(tryPatchCmd) != 0:
1237 1238
            fixed_rcs_keywords = False
            patch_succeeded = False
1239
            print "Unfortunately applying the change failed!"
1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266 1267 1268 1269 1270 1271 1272

            # 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:
1273 1274 1275
            for f in editedFiles:
                p4_revert(f)
            return False
1276

1277 1278 1279
        #
        # Apply the patch for real, and do add/delete/+x handling.
        #
1280
        system(applyPatchCmd)
1281 1282

        for f in filesToAdd:
1283
            p4_add(f)
1284
        for f in filesToDelete:
1285 1286
            p4_revert(f)
            p4_delete(f)
1287

1288 1289 1290 1291 1292
        # Set/clear executable bits
        for f in filesToChangeExecBit.keys():
            mode = filesToChangeExecBit[f]
            setP4ExecBit(f, mode)

1293 1294 1295 1296
        #
        # Build p4 change description, starting with the contents
        # of the git commit message.
        #
1297 1298
        logMessage = extractLogMessageFromGitCommit(id)
        logMessage = logMessage.strip()
1299
        (logMessage, jobs) = self.separate_jobs_from_description(logMessage)
1300

1301
        template = self.prepareSubmitTemplate()
1302
        submitTemplate = self.prepareLogMessage(template, logMessage, jobs)
1303

1304
        if self.preserveUser:
1305
           submitTemplate += "\n######## Actual user %s, modified after commit\n" % p4User
1306

1307 1308 1309 1310
        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"
1311

1312 1313 1314
        separatorLine = "######## everything below this line is just the diff #######\n"

        # diff
1315 1316 1317 1318 1319 1320 1321
        if os.environ.has_key("P4DIFF"):
            del(os.environ["P4DIFF"])
        diff = ""
        for editedFile in editedFiles:
            diff += p4_read_pipe(['diff', '-du',
                                  wildcard_encode(editedFile)])

1322
        # new file diff
1323 1324 1325 1326 1327 1328 1329 1330 1331 1332
        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()

1333
        # change description file: submitTemplate, separatorLine, diff, newdiff
1334 1335 1336 1337 1338 1339 1340 1341 1342
        (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()

1343 1344 1345 1346 1347 1348 1349 1350 1351 1352 1353 1354 1355 1356 1357 1358 1359 1360 1361 1362 1363 1364 1365 1366 1367 1368 1369 1370 1371 1372 1373 1374 1375 1376 1377
        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

1378 1379 1380
        #
        # Let the user edit the change description, then submit it.
        #
1381 1382
        if self.edit_template(fileName):
            # read the edited message and submit
1383
            ret = True
1384 1385
            tmpFile = open(fileName, "rb")
            message = tmpFile.read()
1386
            tmpFile.close()
1387 1388 1389 1390
            submitTemplate = message[:message.index(separatorLine)]
            if self.isWindows:
                submitTemplate = submitTemplate.replace("\r\n", "\n")
            p4_write_pipe(['submit', '-i'], submitTemplate)
1391

1392 1393 1394 1395 1396 1397 1398 1399 1400 1401 1402 1403
            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")
1404

1405
        else:
1406
            # skip this patch
1407
            ret = False
1408 1409 1410 1411 1412 1413
            print "Submission cancelled, undoing p4 changes."
            for f in editedFiles:
                p4_revert(f)
            for f in filesToAdd:
                p4_revert(f)
                os.remove(f)
1414 1415
            for f in filesToDelete:
                p4_revert(f)
1416 1417

        os.remove(fileName)
1418
        return ret
1419

1420 1421 1422
    # Export git tags as p4 labels. Create a p4 label and then tag
    # with that.
    def exportGitTags(self, gitTags):
1423 1424 1425 1426
        validLabelRegexp = gitConfig("git-p4.labelExportRegexp")
        if len(validLabelRegexp) == 0:
            validLabelRegexp = defaultLabelRegexp
        m = re.compile(validLabelRegexp)
1427 1428 1429 1430 1431

        for name in gitTags:

            if not m.match(name):
                if verbose:
1432
                    print "tag %s does not match regexp %s" % (name, validLabelRegexp)
1433 1434 1435
                continue

            # Get the p4 commit this corresponds to
1436 1437
            logMessage = extractLogMessageFromGitCommit(name)
            values = extractSettingsGitLog(logMessage)
1438

1439
            if not values.has_key('change'):
1440 1441 1442 1443
                # a tag pointing to something not sent to p4; ignore
                if verbose:
                    print "git tag %s does not give a p4 commit" % name
                continue
1444 1445
            else:
                changelist = values['change']
1446 1447 1448 1449 1450 1451 1452 1453 1454 1455 1456 1457 1458 1459 1460 1461 1462 1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475

            # 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

1476 1477
            if self.dry_run:
                print "Would create p4 label %s for tag" % name
1478 1479 1480
            elif self.prepare_p4_only:
                print "Not creating p4 label %s for tag due to option" \
                      " --prepare-p4-only" % name
1481 1482
            else:
                p4_write_pipe(["label", "-i"], labelTemplate)
1483

1484 1485 1486
                # Use the label
                p4_system(["tag", "-l", name] +
                          ["%s@%s" % (mapping.depot_side.path, changelist) for mapping in clientSpec.mappings])
1487

1488 1489
                if verbose:
                    print "created p4 label for tag %s" % name
1490

1491
    def run(self, args):
1492 1493
        if len(args) == 0:
            self.master = currentGitBranch()
1494
            if len(self.master) == 0 or not gitBranchExists("refs/heads/%s" % self.master):
1495 1496 1497
                die("Detecting current git branch failed!")
        elif len(args) == 1:
            self.master = args[0]
1498 1499
            if not branchExists(self.master):
                die("Branch %s does not exist" % self.master)
1500 1501 1502
        else:
            return False

J
Jing Xue 已提交
1503 1504 1505 1506
        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)

1507
        [upstream, settings] = findUpstreamBranchPoint()
1508
        self.depotPath = settings['depot-paths'][0]
1509 1510
        if len(self.origin) == 0:
            self.origin = upstream
1511

1512 1513 1514 1515
        if self.preserveUser:
            if not self.canChangeChangelists():
                die("Cannot preserve user names without p4 super-user or admin permissions")

1516 1517 1518 1519 1520 1521 1522 1523 1524 1525
        # 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

1526 1527
        if self.verbose:
            print "Origin branch is " + self.origin
1528

1529
        if len(self.depotPath) == 0:
1530 1531 1532
            print "Internal error: cannot locate perforce depot path from existing branches"
            sys.exit(128)

1533 1534 1535 1536 1537
        self.useClientSpec = False
        if gitConfig("git-p4.useclientspec", "--bool") == "true":
            self.useClientSpec = True
        if self.useClientSpec:
            self.clientSpecDirs = getClientSpec()
1538

1539 1540 1541 1542 1543
        if self.useClientSpec:
            # all files are relative to the client spec
            self.clientPath = getClientRoot()
        else:
            self.clientPath = p4Where(self.depotPath)
1544

1545 1546
        if self.clientPath == "":
            die("Error: Cannot locate perforce checkout of %s in client view" % self.depotPath)
1547

1548
        print "Perforce checkout for depot path %s located at %s" % (self.depotPath, self.clientPath)
1549
        self.oldWorkingDirectory = os.getcwd()
1550

1551
        # ensure the clientPath exists
1552
        new_client_dir = False
1553
        if not os.path.exists(self.clientPath):
1554
            new_client_dir = True
1555 1556
            os.makedirs(self.clientPath)

1557
        chdir(self.clientPath)
1558 1559
        if self.dry_run:
            print "Would synchronize p4 checkout in %s" % self.clientPath
1560
        else:
1561 1562 1563 1564 1565 1566
            print "Synchronizing p4 checkout..."
            if new_client_dir:
                # old one was destroyed, and maybe nobody told p4
                p4_sync("...", "-f")
            else:
                p4_sync("...")
1567 1568
        self.check()

S
Simon Hausmann 已提交
1569 1570 1571 1572
        commits = []
        for line in read_pipe_lines("git rev-list --no-merges %s..%s" % (self.origin, self.master)):
            commits.append(line.strip())
        commits.reverse()
1573

1574 1575 1576 1577 1578
        if self.preserveUser or (gitConfig("git-p4.skipUserNameCheck") == "true"):
            self.checkAuthorship = False
        else:
            self.checkAuthorship = True

1579 1580 1581
        if self.preserveUser:
            self.checkValidP4Users(commits)

1582 1583 1584 1585 1586 1587 1588 1589 1590 1591 1592 1593 1594 1595 1596 1597 1598 1599 1600 1601 1602 1603 1604 1605 1606 1607 1608 1609 1610 1611 1612
        #
        # 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"

1613 1614 1615 1616
        #
        # Apply the commits, one at a time.  On failure, ask if should
        # continue to try the rest of the patches, or quit.
        #
1617 1618
        if self.dry_run:
            print "Would apply"
1619
        applied = []
1620 1621
        last = len(commits) - 1
        for i, commit in enumerate(commits):
1622 1623 1624 1625 1626 1627
            if self.dry_run:
                print " ", read_pipe(["git", "show", "-s",
                                      "--format=format:%h %s", commit])
                ok = True
            else:
                ok = self.applyCommit(commit)
1628 1629
            if ok:
                applied.append(commit)
1630
            else:
1631 1632 1633 1634
                if self.prepare_p4_only and i < last:
                    print "Processing only the first commit due to option" \
                          " --prepare-p4-only"
                    break
1635 1636 1637
                if i < last:
                    quit = False
                    while True:
1638 1639 1640 1641 1642 1643 1644 1645 1646 1647 1648 1649 1650 1651 1652
                        # 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)

1653 1654 1655 1656 1657 1658 1659 1660 1661
                        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
1662

1663
        chdir(self.oldWorkingDirectory)
1664

1665 1666
        if self.dry_run:
            pass
1667 1668
        elif self.prepare_p4_only:
            pass
1669
        elif len(commits) == len(applied):
1670
            print "All commits applied!"
1671

S
Simon Hausmann 已提交
1672 1673
            sync = P4Sync()
            sync.run([])
1674

S
Simon Hausmann 已提交
1675 1676
            rebase = P4Rebase()
            rebase.rebase()
1677

1678 1679 1680 1681 1682 1683 1684 1685 1686 1687 1688 1689 1690 1691
        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."

1692
        if gitConfig("git-p4.exportLabels", "--bool") == "true":
1693
            self.exportLabels = True
1694 1695 1696 1697 1698 1699 1700 1701

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

            missingGitTags = gitTags - p4Labels
            self.exportGitTags(missingGitTags)

1702 1703 1704 1705
        # exit with error unless everything applied perfecly
        if len(commits) != len(applied):
                sys.exit(1)

1706 1707
        return True

P
Pete Wyckoff 已提交
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 1735 1736 1737 1738 1739 1740 1741 1742 1743 1744 1745
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:
1746 1747
                if triple_dot_index != len(self.path) - 3:
                    die("Can handle only single ... wildcard, at end: %s" %
P
Pete Wyckoff 已提交
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 1791 1792 1793 1794 1795 1796 1797 1798 1799 1800 1801
                        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 已提交
1802
                   (c, self.depot_side.path, self.client_side.path)
P
Pete Wyckoff 已提交
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 1891 1892 1893 1894 1895 1896 1897 1898 1899 1900 1901

        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 已提交
1902 1903
                # has already been claimed by an earlier mapping (i.e.
                # one later in the list, which we are walking backwards).
P
Pete Wyckoff 已提交
1904 1905 1906 1907 1908 1909 1910 1911 1912 1913 1914 1915 1916 1917 1918 1919 1920
                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

1921
class P4Sync(Command, P4UserMap):
1922 1923
    delete_actions = ( "delete", "move/delete", "purge" )

1924 1925
    def __init__(self):
        Command.__init__(self)
1926
        P4UserMap.__init__(self)
1927 1928 1929 1930 1931
        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"),
1932
                optparse.make_option("--detect-labels", dest="detectLabels", action="store_true"),
1933
                optparse.make_option("--import-labels", dest="importLabels", action="store_true"),
1934 1935
                optparse.make_option("--import-local", dest="importIntoRemotes", action="store_false",
                                     help="Import into refs/heads/ , not refs/remotes"),
1936
                optparse.make_option("--max-changes", dest="maxChanges"),
1937
                optparse.make_option("--keep-path", dest="keepRepoPath", action='store_true',
1938 1939 1940
                                     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")
1941 1942 1943 1944 1945 1946 1947 1948 1949 1950 1951
        ]
        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
1952 1953
        self.createdBranches = set()
        self.committedChanges = set()
1954
        self.branch = ""
1955
        self.detectBranches = False
1956
        self.detectLabels = False
1957
        self.importLabels = False
1958
        self.changesFile = ""
1959
        self.syncWithOrigin = True
1960
        self.importIntoRemotes = True
1961
        self.maxChanges = ""
1962
        self.isWindows = (platform.system() == "Windows")
1963
        self.keepRepoPath = False
1964
        self.depotPaths = None
1965
        self.p4BranchesInGit = []
T
Tommy Thorn 已提交
1966
        self.cloneExclude = []
1967
        self.useClientSpec = False
1968
        self.useClientSpec_from_options = False
P
Pete Wyckoff 已提交
1969
        self.clientSpecDirs = None
1970 1971
        self.tempBranches = []
        self.tempBranchLocation = "git-p4-tmp"
1972

1973 1974 1975
        if gitConfig("git-p4.syncFromOrigin") == "false":
            self.syncWithOrigin = False

1976 1977 1978 1979 1980 1981 1982 1983
    # 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

1984
    def extractFilesFromCommit(self, commit):
T
Tommy Thorn 已提交
1985 1986
        self.cloneExclude = [re.sub(r"\.\.\.$", "", path)
                             for path in self.cloneExclude]
1987 1988 1989 1990
        files = []
        fnum = 0
        while commit.has_key("depotFile%s" % fnum):
            path =  commit["depotFile%s" % fnum]
1991

T
Tommy Thorn 已提交
1992
            if [p for p in self.cloneExclude
1993
                if p4PathStartsWith(path, p)]:
T
Tommy Thorn 已提交
1994 1995 1996
                found = False
            else:
                found = [p for p in self.depotPaths
1997
                         if p4PathStartsWith(path, p)]
1998
            if not found:
1999 2000 2001 2002 2003 2004 2005 2006 2007 2008 2009 2010
                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

2011
    def stripRepoPath(self, path, prefixes):
2012 2013 2014 2015 2016
        """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."""

2017
        if self.useClientSpec:
2018 2019
            # branch detection moves files up a level (the branch name)
            # from what client spec interpretation gives
2020
            path = self.clientSpecDirs.map_in_client(path)
2021 2022 2023 2024 2025 2026 2027 2028 2029 2030 2031 2032
            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):]
2033

2034 2035 2036 2037
        else:
            for p in prefixes:
                if p4PathStartsWith(path, p):
                    path = path[len(p):]
2038
                    break
2039

2040
        path = wildcard_decode(path)
2041
        return path
H
Han-Wen Nienhuys 已提交
2042

2043
    def splitFilesIntoBranches(self, commit):
2044 2045 2046
        """Look at each depotFile in the commit to figure out to what
           branch it belongs."""

2047
        branches = {}
2048 2049 2050
        fnum = 0
        while commit.has_key("depotFile%s" % fnum):
            path =  commit["depotFile%s" % fnum]
2051
            found = [p for p in self.depotPaths
2052
                     if p4PathStartsWith(path, p)]
2053
            if not found:
2054 2055 2056 2057 2058 2059 2060 2061 2062 2063
                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

2064 2065 2066 2067 2068 2069
            # 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)
2070

2071
            for branch in self.knownBranches.keys():
2072 2073
                # 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 已提交
2074
                if relPath.startswith(branch + "/"):
2075 2076
                    if branch not in branches:
                        branches[branch] = []
2077
                    branches[branch].append(file)
2078
                    break
2079 2080 2081

        return branches

2082 2083 2084 2085 2086 2087 2088 2089
    # 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 已提交
2090 2091 2092 2093 2094 2095 2096 2097
        (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
2098 2099
            data = ''.join(contents)
            contents = [data[:-1]]
2100

P
Pete Wyckoff 已提交
2101
        if type_base == "utf16":
2102 2103 2104 2105 2106
            # 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.
2107
            text = p4_read_pipe(['print', '-q', '-o', '-', file['depotFile']])
2108 2109
            contents = [ text ]

P
Pete Wyckoff 已提交
2110 2111 2112 2113 2114 2115 2116 2117 2118 2119 2120 2121 2122
        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 已提交
2123 2124 2125
        # Perhaps windows wants unicode, utf16 newlines translated too;
        # but this is not doing it.
        if self.isWindows and type_base == "text":
2126 2127 2128 2129 2130 2131
            mangled = []
            for data in contents:
                data = data.replace("\r\n", "\n")
                mangled.append(data)
            contents = mangled

2132 2133
        # Note that we do not try to de-mangle keywords on utf16 files,
        # even though in theory somebody may want that.
2134 2135 2136 2137 2138 2139
        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 ]
2140

P
Pete Wyckoff 已提交
2141
        self.gitStream.write("M %s inline %s\n" % (git_mode, relPath))
2142 2143 2144 2145 2146 2147 2148 2149 2150 2151 2152 2153 2154 2155 2156 2157 2158 2159 2160 2161

        # 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):

2162 2163 2164 2165 2166 2167 2168 2169 2170 2171 2172 2173 2174 2175 2176 2177 2178 2179 2180 2181 2182 2183 2184
        # 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)

2185 2186 2187 2188 2189 2190
        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
2191

2192 2193 2194 2195 2196 2197 2198
        # 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]
2199

2200
        self.stream_have_file_info = True
2201 2202 2203

    # Stream directly from "p4 files" into "git fast-import"
    def streamP4Files(self, files):
2204 2205
        filesForCommit = []
        filesToRead = []
2206
        filesToDelete = []
2207

2208
        for f in files:
P
Pete Wyckoff 已提交
2209 2210 2211 2212 2213
            # 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
2214

P
Pete Wyckoff 已提交
2215 2216 2217 2218 2219
            filesForCommit.append(f)
            if f['action'] in self.delete_actions:
                filesToDelete.append(f)
            else:
                filesToRead.append(f)
H
Han-Wen Nienhuys 已提交
2220

2221 2222 2223
        # deleted files...
        for f in filesToDelete:
            self.streamOneP4Deletion(f)
2224

2225 2226 2227 2228
        if len(filesToRead) > 0:
            self.stream_file = {}
            self.stream_contents = []
            self.stream_have_file_info = False
2229

2230 2231 2232
            # curry self argument
            def streamP4FilesCbSelf(entry):
                self.streamP4FilesCb(entry)
H
Han-Wen Nienhuys 已提交
2233

2234 2235 2236 2237 2238
            fileArgs = ['%s#%s' % (f['path'], f['rev']) for f in filesToRead]

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

2240 2241 2242
            # do the last chunk
            if self.stream_file.has_key('depotFile'):
                self.streamOneP4File(self.stream_file, self.stream_contents)
H
Han-Wen Nienhuys 已提交
2243

2244 2245 2246 2247 2248 2249
    def make_email(self, userid):
        if userid in self.users:
            return self.users[userid]
        else:
            return "%s <a@b>" % userid

2250 2251 2252 2253 2254 2255 2256 2257 2258 2259 2260 2261 2262 2263 2264 2265 2266 2267 2268 2269 2270 2271 2272 2273 2274 2275 2276 2277 2278 2279 2280 2281
    # 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")

2282
    def commit(self, details, files, branch, parent = ""):
2283 2284 2285
        epoch = details["time"]
        author = details["user"]

2286 2287 2288
        if self.verbose:
            print "commit into %s" % branch

2289 2290 2291 2292
        # start with reading files; if that fails, we should not
        # create a commit.
        new_files = []
        for f in files:
2293
            if [p for p in self.branchPrefixes if p4PathStartsWith(f['path'], p)]:
2294 2295
                new_files.append (f)
            else:
2296
                sys.stderr.write("Ignoring file outside of prefix: %s\n" % f['path'])
2297

2298
        self.gitStream.write("commit %s\n" % branch)
H
Han-Wen Nienhuys 已提交
2299
#        gitStream.write("mark :%s\n" % details["change"])
2300 2301
        self.committedChanges.add(int(details["change"]))
        committer = ""
2302 2303
        if author not in self.users:
            self.getUserMapFromPerforceServer()
2304
        committer = "%s %s %s" % (self.make_email(author), epoch, self.tz)
2305 2306 2307 2308 2309

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

        self.gitStream.write("data <<EOT\n")
        self.gitStream.write(details["desc"])
2310 2311
        self.gitStream.write("\n[git-p4: depot-paths = \"%s\": change = %s" %
                             (','.join(self.branchPrefixes), details["change"]))
2312 2313 2314
        if len(details['options']) > 0:
            self.gitStream.write(": options = %s" % details['options'])
        self.gitStream.write("]\nEOT\n\n")
2315 2316

        if len(parent) > 0:
2317 2318
            if self.verbose:
                print "parent %s" % parent
2319 2320
            self.gitStream.write("from %s\n" % parent)

2321
        self.streamP4Files(new_files)
2322 2323
        self.gitStream.write("\n")

2324 2325
        change = int(details["change"])

2326
        if self.labels.has_key(change):
2327 2328 2329
            label = self.labels[change]
            labelDetails = label[0]
            labelRevisions = label[1]
2330 2331
            if self.verbose:
                print "Change %s is labelled %s" % (change, labelDetails)
2332

2333
            files = p4CmdList(["files"] + ["%s...@%s" % (p, change)
2334
                                                for p in self.branchPrefixes])
2335 2336 2337 2338 2339

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

                cleanedFiles = {}
                for info in files:
2340
                    if info["action"] in self.delete_actions:
2341 2342 2343 2344
                        continue
                    cleanedFiles[info["depotFile"]] = info["rev"]

                if cleanedFiles == labelRevisions:
2345
                    self.streamTag(self.gitStream, 'tag_%s' % labelDetails['label'], labelDetails, branch, epoch)
2346 2347

                else:
2348
                    if not self.silent:
2349 2350
                        print ("Tag %s does not match with change %s: files do not match."
                               % (labelDetails["label"], change))
2351 2352

            else:
2353
                if not self.silent:
2354 2355
                    print ("Tag %s does not match with change %s: file count is different."
                           % (labelDetails["label"], change))
2356

2357
    # Build a dictionary of changelists and labels, for "detect-labels" option.
2358 2359 2360
    def getLabels(self):
        self.labels = {}

2361
        l = p4CmdList(["labels"] + ["%s..." % p for p in self.depotPaths])
S
Simon Hausmann 已提交
2362
        if len(l) > 0 and not self.silent:
2363
            print "Finding files belonging to labels in %s" % `self.depotPaths`
S
Simon Hausmann 已提交
2364 2365

        for output in l:
2366 2367 2368
            label = output["label"]
            revisions = {}
            newestChange = 0
2369 2370
            if self.verbose:
                print "Querying files for label %s" % label
2371 2372 2373
            for file in p4CmdList(["files"] +
                                      ["%s...@%s" % (p, label)
                                          for p in self.depotPaths]):
2374 2375 2376 2377 2378
                revisions[file["depotFile"]] = file["rev"]
                change = int(file["change"])
                if change > newestChange:
                    newestChange = change

2379 2380 2381 2382
            self.labels[newestChange] = [output, revisions]

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

2384 2385 2386 2387 2388 2389 2390 2391 2392
    # 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")
2393
        validLabelRegexp = gitConfig("git-p4.labelImportRegexp")
2394 2395 2396 2397 2398 2399 2400 2401 2402 2403 2404 2405 2406 2407 2408 2409 2410 2411 2412 2413 2414 2415 2416 2417 2418 2419 2420 2421 2422 2423 2424 2425 2426 2427 2428
        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:
2429
                        print "Could not convert label time %s" % labelDetails['Update']
2430 2431 2432 2433 2434 2435 2436 2437 2438 2439 2440 2441 2442 2443 2444 2445 2446
                        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])

2447 2448
    def guessProjectName(self):
        for p in self.depotPaths:
S
Simon Hausmann 已提交
2449 2450 2451 2452 2453 2454
            if p.endswith("/"):
                p = p[:-1]
            p = p[p.strip().rfind("/") + 1:]
            if not p.endswith("/"):
               p += "/"
            return p
2455

2456
    def getBranchMapping(self):
2457 2458
        lostAndFoundBranches = set()

2459 2460 2461 2462 2463 2464 2465
        user = gitConfig("git-p4.branchUser")
        if len(user) > 0:
            command = "branches -u %s" % user
        else:
            command = "branches"

        for info in p4CmdList(command):
2466
            details = p4Cmd(["branch", "-o", info["branch"]])
2467 2468 2469 2470 2471 2472 2473 2474 2475
            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]
2476
                ## HACK
2477
                if p4PathStartsWith(source, self.depotPaths[0]) and p4PathStartsWith(destination, self.depotPaths[0]):
2478 2479
                    source = source[len(self.depotPaths[0]):-4]
                    destination = destination[len(self.depotPaths[0]):-4]
2480

2481 2482 2483 2484 2485 2486
                    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

2487 2488 2489 2490
                    self.knownBranches[destination] = source

                    lostAndFoundBranches.discard(destination)

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

2494 2495 2496 2497 2498 2499 2500 2501 2502 2503 2504 2505 2506 2507 2508 2509 2510 2511 2512
        # 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)

2513 2514 2515

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

2517 2518 2519 2520 2521 2522 2523 2524 2525
    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 已提交
2526 2527 2528 2529 2530 2531 2532 2533 2534 2535
    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']))
2536

2537 2538 2539 2540 2541 2542 2543 2544 2545
    def gitRefForBranch(self, branch):
        if branch == "main":
            return self.refPrefix + "master"

        if len(branch) <= 0:
            return branch

        return self.refPrefix + self.projectName + branch

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 2586 2587 2588 2589 2590 2591 2592 2593 2594 2595 2596
    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

2597
        branchParentChange = int(p4Cmd(["changes", "-m", "1", "%s...@1,%s" % (sourceDepotPath, firstChange)])["change"])
2598 2599 2600 2601 2602 2603 2604 2605 2606
        #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

2607 2608 2609 2610 2611 2612 2613 2614 2615 2616 2617 2618 2619 2620
    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

2621 2622 2623
    def importChanges(self, changes):
        cnt = 1
        for change in changes:
P
Pete Wyckoff 已提交
2624
            description = p4_describe(change)
2625 2626 2627 2628 2629 2630 2631 2632 2633 2634 2635 2636 2637
            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 + "/"
2638
                        self.branchPrefixes = [ branchPrefix ]
2639 2640 2641 2642 2643 2644 2645 2646 2647 2648 2649 2650 2651 2652 2653

                        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 = ""
2654 2655 2656 2657 2658 2659 2660 2661 2662 2663 2664 2665 2666
                            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
2667

2668 2669
                        branch = self.gitRefForBranch(branch)
                        parent = self.gitRefForBranch(parent)
2670 2671 2672 2673 2674 2675 2676 2677

                        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]

2678 2679 2680 2681 2682
                        blob = None
                        if len(parent) > 0:
                            tempBranch = os.path.join(self.tempBranchLocation, "%d" % (change))
                            if self.verbose:
                                print "Creating temporary branch: " + tempBranch
2683
                            self.commit(description, filesForCommit, tempBranch)
2684 2685 2686 2687
                            self.tempBranches.append(tempBranch)
                            self.checkpoint()
                            blob = self.searchParent(parent, branch, tempBranch)
                        if blob:
2688
                            self.commit(description, filesForCommit, branch, blob)
2689 2690 2691
                        else:
                            if self.verbose:
                                print "Parent of %s not found. Committing into head of %s" % (branch, parent)
2692
                            self.commit(description, filesForCommit, branch, parent)
2693 2694
                else:
                    files = self.extractFilesFromCommit(description)
2695
                    self.commit(description, files, self.branch,
2696
                                self.initialParent)
2697
                    # only needed once, to connect to the previous commit
2698 2699 2700 2701 2702
                    self.initialParent = ""
            except IOError:
                print self.gitError.read()
                sys.exit(1)

2703 2704 2705
    def importHeadRevision(self, revision):
        print "Doing initial import of %s from revision %s into %s" % (' '.join(self.depotPaths), revision, self.branch)

2706 2707
        details = {}
        details["user"] = "git perforce import user"
2708
        details["desc"] = ("Initial import of %s from the state at revision %s\n"
2709 2710 2711 2712 2713
                           % (' '.join(self.depotPaths), revision))
        details["change"] = revision
        newestRevision = 0

        fileCnt = 0
2714 2715 2716
        fileArgs = ["%s...%s" % (p,revision) for p in self.depotPaths]

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

2718
            if 'code' in info and info['code'] == 'error':
2719 2720
                sys.stderr.write("p4 returned an error: %s\n"
                                 % info['data'])
2721 2722 2723 2724
                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))
2725
                sys.exit(1)
2726 2727
            if 'p4ExitCode' in info:
                sys.stderr.write("p4 exitcode: %s\n" % info['p4ExitCode'])
2728 2729 2730 2731 2732 2733 2734
                sys.exit(1)


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

2735
            if info["action"] in self.delete_actions:
2736 2737 2738 2739 2740 2741 2742 2743 2744 2745
                # 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
2746

P
Pete Wyckoff 已提交
2747
        # Use time from top-most change so that all git p4 clones of
2748
        # the same p4 repo have the same commit SHA1s.
P
Pete Wyckoff 已提交
2749 2750
        res = p4_describe(newestRevision)
        details["time"] = res["time"]
2751

2752 2753
        self.updateOptionDict(details)
        try:
2754
            self.commit(details, self.extractFilesFromCommit(details), self.branch)
2755 2756 2757 2758 2759
        except IOError:
            print "IO error with git fast-import. Is your git version recent enough?"
            print self.gitError.read()


2760
    def run(self, args):
2761
        self.depotPaths = []
2762
        self.changeRange = ""
2763
        self.previousDepotPaths = []
2764
        self.hasOrigin = False
H
Han-Wen Nienhuys 已提交
2765

2766 2767 2768 2769
        # map from branch depot path to parent branch
        self.knownBranches = {}
        self.initialParents = {}

2770 2771 2772
        if self.importIntoRemotes:
            self.refPrefix = "refs/remotes/p4/"
        else:
2773
            self.refPrefix = "refs/heads/p4/"
2774

2775 2776 2777 2778 2779 2780
        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")
2781

2782
        branch_arg_given = bool(self.branch)
2783
        if len(self.branch) == 0:
2784
            self.branch = self.refPrefix + "master"
2785
            if gitBranchExists("refs/heads/p4") and self.importIntoRemotes:
2786
                system("git update-ref %s refs/heads/p4" % self.branch)
2787
                system("git branch -D p4")
2788

2789 2790 2791 2792 2793
        # 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:
2794 2795 2796
            if gitConfig("git-p4.useclientspec", "--bool") == "true":
                self.useClientSpec = True
        if self.useClientSpec:
2797
            self.clientSpecDirs = getClientSpec()
2798

H
Han-Wen Nienhuys 已提交
2799 2800 2801
        # TODO: should always look at previous commits,
        # merge with previous imports, if possible.
        if args == []:
2802
            if self.hasOrigin:
2803
                createOrUpdateBranchesFromOrigin(self.refPrefix, self.silent)
2804 2805 2806

            # branches holds mapping from branch name to sha1
            branches = p4BranchesInGit(self.importIntoRemotes)
2807 2808 2809 2810 2811 2812 2813 2814

            # restrict to just this one, disabling detect-branches
            if branch_arg_given:
                short = self.branch.split("/")[-1]
                if short in branches:
                    self.p4BranchesInGit = [ short ]
            else:
                self.p4BranchesInGit = branches.keys()
2815 2816 2817 2818 2819

            if len(self.p4BranchesInGit) > 1:
                if not self.silent:
                    print "Importing from/into multiple branches"
                self.detectBranches = True
2820 2821 2822
                for branch in branches.keys():
                    self.initialParents[self.refPrefix + branch] = \
                        branches[branch]
2823

2824 2825 2826 2827 2828
            if self.verbose:
                print "branches: %s" % self.p4BranchesInGit

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

                settings = extractSettingsGitLog(logMsg)
2832

H
Han-Wen Nienhuys 已提交
2833 2834 2835 2836
                self.readOptions(settings)
                if (settings.has_key('depot-paths')
                    and settings.has_key ('change')):
                    change = int(settings['change']) + 1
2837 2838
                    p4Change = max(p4Change, change)

H
Han-Wen Nienhuys 已提交
2839 2840
                    depotPaths = sorted(settings['depot-paths'])
                    if self.previousDepotPaths == []:
2841
                        self.previousDepotPaths = depotPaths
2842
                    else:
2843 2844
                        paths = []
                        for (prev, cur) in zip(self.previousDepotPaths, depotPaths):
2845 2846 2847 2848
                            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]:
2849
                                    i = i - 1
2850 2851
                                    break

2852
                            paths.append ("/".join(cur_list[:i + 1]))
2853 2854

                        self.previousDepotPaths = paths
2855 2856

            if p4Change > 0:
H
Han-Wen Nienhuys 已提交
2857
                self.depotPaths = sorted(self.previousDepotPaths)
2858
                self.changeRange = "@%s,#head" % p4Change
2859
                if not self.silent and not self.detectBranches:
2860
                    print "Performing incremental import into %s git branch" % self.branch
2861

2862 2863 2864 2865
        # 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/
2866
        if not self.branch.startswith("refs/"):
2867 2868 2869 2870 2871 2872 2873
            if self.importIntoRemotes:
                prepend = "refs/remotes/"
            else:
                prepend = "refs/heads/"
            if not self.branch.startswith("p4/"):
                prepend += "p4/"
            self.branch = prepend + self.branch
2874

2875
        if len(args) == 0 and self.depotPaths:
2876
            if not self.silent:
2877
                print "Depot paths: %s" % ' '.join(self.depotPaths)
2878
        else:
2879
            if self.depotPaths and self.depotPaths != args:
2880
                print ("previous import used depot path %s and now %s was specified. "
2881 2882
                       "This doesn't work!" % (' '.join (self.depotPaths),
                                               ' '.join (args)))
2883
                sys.exit(1)
2884

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

2887
        revision = ""
2888 2889
        self.users = {}

2890 2891 2892 2893 2894 2895 2896 2897 2898 2899 2900
        # 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")

2901 2902 2903 2904 2905 2906 2907
        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 已提交
2908
                elif ',' not in self.changeRange:
2909
                    revision = self.changeRange
2910
                    self.changeRange = ""
2911
                p = p[:atIdx]
2912 2913
            elif p.find("#") != -1:
                hashIdx = p.index("#")
2914
                revision = p[hashIdx:]
2915
                p = p[:hashIdx]
2916
            elif self.previousDepotPaths == []:
2917 2918 2919 2920
                # pay attention to changesfile, if given, else import
                # the entire p4 tree at the head revision
                if len(self.changesFile) == 0:
                    revision = "#head"
2921 2922 2923 2924 2925 2926 2927 2928 2929

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

            newPaths.append(p)

        self.depotPaths = newPaths

2930 2931 2932
        # --detect-branches may change this for each branch
        self.branchPrefixes = self.depotPaths

2933
        self.loadUserMapFromCache()
2934 2935 2936
        self.labels = {}
        if self.detectLabels:
            self.getLabels();
2937

2938
        if self.detectBranches:
2939 2940 2941
            ## FIXME - what's a P4 projectName ?
            self.projectName = self.guessProjectName()

2942 2943 2944 2945
            if self.hasOrigin:
                self.getBranchMappingFromGitBranches()
            else:
                self.getBranchMapping()
2946 2947 2948 2949 2950
            if self.verbose:
                print "p4-git branches: %s" % self.p4BranchesInGit
                print "initial parents: %s" % self.initialParents
            for b in self.p4BranchesInGit:
                if b != "master":
2951 2952

                    ## FIXME
2953 2954
                    b = b[len(self.projectName):]
                self.createdBranches.add(b)
2955

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

2958 2959 2960 2961 2962 2963 2964
        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
2965

2966
        if revision:
2967
            self.importHeadRevision(revision)
2968 2969 2970
        else:
            changes = []

2971
            if len(self.changesFile) > 0:
2972
                output = open(self.changesFile).readlines()
2973
                changeSet = set()
2974 2975 2976 2977 2978 2979 2980 2981
                for line in output:
                    changeSet.add(int(line))

                for change in changeSet:
                    changes.append(change)

                changes.sort()
            else:
P
Pete Wyckoff 已提交
2982 2983
                # catch "git p4 sync" with no new branches, in a repo that
                # does not have any existing p4 branches
2984 2985 2986 2987 2988 2989 2990 2991 2992 2993 2994 2995 2996 2997 2998
                if len(args) == 0:
                    if not self.p4BranchesInGit:
                        die("No remote p4 branches.  Perhaps you never did \"git p4 clone\" in here.")

                    # The default branch is master, unless --branch is used to
                    # specify something else.  Make sure it exists, or complain
                    # nicely about how to use --branch.
                    if not self.detectBranches:
                        if not branch_exists(self.branch):
                            if branch_arg_given:
                                die("Error: branch %s does not exist." % self.branch)
                            else:
                                die("Error: no branch %s; perhaps specify one with --branch." %
                                    self.branch)

2999
                if self.verbose:
3000
                    print "Getting p4 changes for %s...%s" % (', '.join(self.depotPaths),
3001
                                                              self.changeRange)
3002
                changes = p4ChangesForPaths(self.depotPaths, self.changeRange)
3003

3004
                if len(self.maxChanges) > 0:
3005
                    changes = changes[:min(int(self.maxChanges), len(changes))]
3006

3007
            if len(changes) == 0:
3008
                if not self.silent:
3009
                    print "No changes to import!"
3010 3011 3012 3013 3014
            else:
                if not self.silent and not self.detectBranches:
                    print "Import destination: %s" % self.branch

                self.updatedBranches = set()
3015

3016 3017 3018 3019 3020 3021 3022 3023
                if not self.detectBranches:
                    if args:
                        # start a new branch
                        self.initialParent = ""
                    else:
                        # build on a previous revision
                        self.initialParent = parseRevision(self.branch)

3024
                self.importChanges(changes)
3025

3026 3027 3028 3029 3030 3031 3032
                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")
3033

3034
        if gitConfig("git-p4.importLabels", "--bool") == "true":
3035
            self.importLabels = True
3036

3037 3038 3039 3040 3041 3042
        if self.importLabels:
            p4Labels = getP4Labels(self.depotPaths)
            gitTags = getGitTags()

            missingP4Labels = p4Labels - gitTags
            self.importP4Labels(self.gitStream, missingP4Labels)
3043 3044

        self.gitStream.close()
3045
        if self.importProcess.wait() != 0:
3046
            die("fast-import failed: %s" % self.gitError.read())
3047 3048 3049
        self.gitOutput.close()
        self.gitError.close()

3050 3051 3052 3053 3054 3055
        # 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))

3056 3057 3058 3059 3060 3061 3062
        # 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])

3063 3064
        return True

S
Simon Hausmann 已提交
3065 3066 3067
class P4Rebase(Command):
    def __init__(self):
        Command.__init__(self)
3068 3069 3070 3071
        self.options = [
                optparse.make_option("--import-labels", dest="importLabels", action="store_true"),
        ]
        self.importLabels = False
3072 3073
        self.description = ("Fetches the latest revision from perforce and "
                            + "rebases the current work (branch) against it")
S
Simon Hausmann 已提交
3074 3075 3076

    def run(self, args):
        sync = P4Sync()
3077
        sync.importLabels = self.importLabels
S
Simon Hausmann 已提交
3078
        sync.run([])
3079

3080 3081 3082
        return self.rebase()

    def rebase(self):
3083 3084 3085 3086 3087
        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.");

3088 3089 3090 3091 3092 3093 3094 3095
        [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
3096
        oldHead = read_pipe("git rev-parse HEAD").strip()
3097
        system("git rebase %s" % upstream)
3098
        system("git diff-tree --stat --summary -M %s HEAD" % oldHead)
S
Simon Hausmann 已提交
3099 3100
        return True

3101 3102 3103 3104
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 已提交
3105
        self.usage = "usage: %prog [options] //depot/path[@revRange]"
T
Tommy Thorn 已提交
3106
        self.options += [
H
Han-Wen Nienhuys 已提交
3107 3108
            optparse.make_option("--destination", dest="cloneDestination",
                                 action='store', default=None,
T
Tommy Thorn 已提交
3109 3110 3111
                                 help="where to leave result of the clone"),
            optparse.make_option("-/", dest="cloneExclude",
                                 action="append", type="string",
P
Pete Wyckoff 已提交
3112 3113 3114
                                 help="exclude depot path"),
            optparse.make_option("--bare", dest="cloneBare",
                                 action="store_true", default=False),
T
Tommy Thorn 已提交
3115
        ]
H
Han-Wen Nienhuys 已提交
3116
        self.cloneDestination = None
3117
        self.needsGit = False
P
Pete Wyckoff 已提交
3118
        self.cloneBare = False
3119

T
Tommy Thorn 已提交
3120 3121 3122 3123 3124 3125
    # 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 已提交
3126 3127 3128 3129 3130
    def defaultDestination(self, args):
        ## TODO: use common prefix of args?
        depotPath = args[0]
        depotDir = re.sub("(@[^@]*)$", "", depotPath)
        depotDir = re.sub("(#[^#]*)$", "", depotDir)
3131
        depotDir = re.sub(r"\.\.\.$", "", depotDir)
H
Han-Wen Nienhuys 已提交
3132 3133 3134
        depotDir = re.sub(r"/$", "", depotDir)
        return os.path.split(depotDir)[1]

3135 3136 3137
    def run(self, args):
        if len(args) < 1:
            return False
H
Han-Wen Nienhuys 已提交
3138 3139 3140 3141

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

3143
        depotPaths = args
3144 3145 3146 3147 3148

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

T
Tommy Thorn 已提交
3149
        self.cloneExclude = ["/"+p for p in self.cloneExclude]
3150 3151 3152
        for p in depotPaths:
            if not p.startswith("//"):
                return False
3153

H
Han-Wen Nienhuys 已提交
3154
        if not self.cloneDestination:
3155
            self.cloneDestination = self.defaultDestination(args)
3156

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

3159 3160
        if not os.path.exists(self.cloneDestination):
            os.makedirs(self.cloneDestination)
3161
        chdir(self.cloneDestination)
P
Pete Wyckoff 已提交
3162 3163 3164 3165 3166 3167

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

3168
        if not P4Sync.run(self, depotPaths):
3169
            return False
3170 3171 3172 3173 3174 3175 3176 3177 3178

        # 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>"'
3179

3180 3181 3182 3183
        # auto-set this variable if invoked with --use-client-spec
        if self.useClientSpec_from_options:
            system("git config --bool git-p4.useclientspec true")

3184 3185
        return True

3186 3187 3188 3189 3190 3191 3192 3193 3194
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):
3195 3196 3197
        if originP4BranchesExist():
            createOrUpdateBranchesFromOrigin()

3198 3199 3200 3201 3202 3203 3204 3205 3206 3207 3208 3209 3210 3211 3212 3213
        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

3214 3215 3216 3217 3218 3219 3220 3221 3222
class HelpFormatter(optparse.IndentedHelpFormatter):
    def __init__(self):
        optparse.IndentedHelpFormatter.__init__(self)

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

3224 3225 3226 3227 3228 3229 3230 3231 3232
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 已提交
3233 3234
    "debug" : P4Debug,
    "submit" : P4Submit,
3235
    "commit" : P4Submit,
H
Han-Wen Nienhuys 已提交
3236 3237 3238
    "sync" : P4Sync,
    "rebase" : P4Rebase,
    "clone" : P4Clone,
3239 3240
    "rollback" : P4RollBack,
    "branches" : P4Branches
3241 3242 3243
}


H
Han-Wen Nienhuys 已提交
3244 3245 3246 3247
def main():
    if len(sys.argv[1:]) == 0:
        printUsage(commands.keys())
        sys.exit(2)
3248

H
Han-Wen Nienhuys 已提交
3249 3250
    cmdName = sys.argv[1]
    try:
H
Han-Wen Nienhuys 已提交
3251 3252
        klass = commands[cmdName]
        cmd = klass()
H
Han-Wen Nienhuys 已提交
3253 3254 3255 3256 3257 3258 3259
    except KeyError:
        print "unknown command %s" % cmdName
        print ""
        printUsage(commands.keys())
        sys.exit(2)

    options = cmd.options
H
Han-Wen Nienhuys 已提交
3260
    cmd.gitdir = os.environ.get("GIT_DIR", None)
H
Han-Wen Nienhuys 已提交
3261 3262 3263

    args = sys.argv[2:]

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

3268 3269 3270 3271
    parser = optparse.OptionParser(cmd.usage.replace("%prog", "%prog " + cmdName),
                                   options,
                                   description = cmd.description,
                                   formatter = HelpFormatter())
H
Han-Wen Nienhuys 已提交
3272

3273
    (cmd, args) = parser.parse_args(sys.argv[2:], cmd);
H
Han-Wen Nienhuys 已提交
3274 3275 3276
    global verbose
    verbose = cmd.verbose
    if cmd.needsGit:
H
Han-Wen Nienhuys 已提交
3277 3278 3279 3280 3281
        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 已提交
3282 3283
                    cdup = read_pipe("git rev-parse --show-cdup").strip()
                    if len(cdup) > 0:
3284
                        chdir(cdup);
3285

H
Han-Wen Nienhuys 已提交
3286 3287 3288
        if not isValidGitDir(cmd.gitdir):
            if isValidGitDir(cmd.gitdir + "/.git"):
                cmd.gitdir += "/.git"
H
Han-Wen Nienhuys 已提交
3289
            else:
H
Han-Wen Nienhuys 已提交
3290
                die("fatal: cannot locate git repository at %s" % cmd.gitdir)
3291

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

H
Han-Wen Nienhuys 已提交
3294 3295
    if not cmd.run(args):
        parser.print_help()
3296
        sys.exit(2)
3297 3298


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