3 # git-p4.py -- A tool for bidirectional operation between a Perforce depot and git.
5 # Author: Simon Hausmann <simon@lst.de>
6 # Copyright: 2007 Simon Hausmann <simon@lst.de>
8 # License: MIT <http://www.opensource.org/licenses/mit-license.php>
11 import optparse
, sys
, os
, marshal
, subprocess
, shelve
12 import tempfile
, getopt
, os
.path
, time
, platform
17 # Only labels/tags matching this will be imported/exported
18 defaultLabelRegexp
= r
'[a-zA-Z0-9_\-.]+$'
20 def p4_build_cmd(cmd
):
21 """Build a suitable p4 command line.
23 This consolidates building and returning a p4 command line into one
24 location. It means that hooking into the environment, or other configuration
25 can be done more easily.
29 user
= gitConfig("git-p4.user")
31 real_cmd
+= ["-u",user
]
33 password
= gitConfig("git-p4.password")
35 real_cmd
+= ["-P", password
]
37 port
= gitConfig("git-p4.port")
39 real_cmd
+= ["-p", port
]
41 host
= gitConfig("git-p4.host")
43 real_cmd
+= ["-H", host
]
45 client
= gitConfig("git-p4.client")
47 real_cmd
+= ["-c", client
]
50 if isinstance(cmd
,basestring
):
51 real_cmd
= ' '.join(real_cmd
) + ' ' + cmd
57 # P4 uses the PWD environment variable rather than getcwd(). Since we're
58 # not using the shell, we have to set it ourselves. This path could
59 # be relative, so go there first, then figure out where we ended up.
61 os
.environ
['PWD'] = os
.getcwd()
67 sys
.stderr
.write(msg
+ "\n")
70 def write_pipe(c
, stdin
):
72 sys
.stderr
.write('Writing pipe: %s\n' % str(c
))
74 expand
= isinstance(c
,basestring
)
75 p
= subprocess
.Popen(c
, stdin
=subprocess
.PIPE
, shell
=expand
)
77 val
= pipe
.write(stdin
)
80 die('Command failed: %s' % str(c
))
84 def p4_write_pipe(c
, stdin
):
85 real_cmd
= p4_build_cmd(c
)
86 return write_pipe(real_cmd
, stdin
)
88 def read_pipe(c
, ignore_error
=False):
90 sys
.stderr
.write('Reading pipe: %s\n' % str(c
))
92 expand
= isinstance(c
,basestring
)
93 p
= subprocess
.Popen(c
, stdout
=subprocess
.PIPE
, shell
=expand
)
96 if p
.wait() and not ignore_error
:
97 die('Command failed: %s' % str(c
))
101 def p4_read_pipe(c
, ignore_error
=False):
102 real_cmd
= p4_build_cmd(c
)
103 return read_pipe(real_cmd
, ignore_error
)
105 def read_pipe_lines(c
):
107 sys
.stderr
.write('Reading pipe: %s\n' % str(c
))
109 expand
= isinstance(c
, basestring
)
110 p
= subprocess
.Popen(c
, stdout
=subprocess
.PIPE
, shell
=expand
)
112 val
= pipe
.readlines()
113 if pipe
.close() or p
.wait():
114 die('Command failed: %s' % str(c
))
118 def p4_read_pipe_lines(c
):
119 """Specifically invoke p4 on the command supplied. """
120 real_cmd
= p4_build_cmd(c
)
121 return read_pipe_lines(real_cmd
)
124 expand
= isinstance(cmd
,basestring
)
126 sys
.stderr
.write("executing %s\n" % str(cmd
))
127 subprocess
.check_call(cmd
, shell
=expand
)
130 """Specifically invoke p4 as the system command. """
131 real_cmd
= p4_build_cmd(cmd
)
132 expand
= isinstance(real_cmd
, basestring
)
133 subprocess
.check_call(real_cmd
, shell
=expand
)
135 def p4_integrate(src
, dest
):
136 p4_system(["integrate", "-Dt", src
, dest
])
139 p4_system(["sync", path
])
142 p4_system(["add", f
])
145 p4_system(["delete", f
])
148 p4_system(["edit", f
])
151 p4_system(["revert", f
])
153 def p4_reopen(type, file):
154 p4_system(["reopen", "-t", type, file])
157 # Canonicalize the p4 type and return a tuple of the
158 # base type, plus any modifiers. See "p4 help filetypes"
159 # for a list and explanation.
161 def split_p4_type(p4type
):
163 p4_filetypes_historical
= {
164 "ctempobj": "binary+Sw",
170 "tempobj": "binary+FSw",
171 "ubinary": "binary+F",
172 "uresource": "resource+F",
173 "uxbinary": "binary+Fx",
174 "xbinary": "binary+x",
176 "xtempobj": "binary+Swx",
178 "xunicode": "unicode+x",
181 if p4type
in p4_filetypes_historical
:
182 p4type
= p4_filetypes_historical
[p4type
]
184 s
= p4type
.split("+")
192 # return the raw p4 type of a file (text, text+ko, etc)
195 results
= p4CmdList(["fstat", "-T", "headType", file])
196 return results
[0]['headType']
199 # Given a type base and modifier, return a regexp matching
200 # the keywords that can be expanded in the file
202 def p4_keywords_regexp_for_type(base
, type_mods
):
203 if base
in ("text", "unicode", "binary"):
205 if "ko" in type_mods
:
207 elif "k" in type_mods
:
208 kwords
= 'Id|Header|Author|Date|DateTime|Change|File|Revision'
212 \$ # Starts with a dollar, followed by...
213 (%s) # one of the keywords, followed by...
214 (:[^$]+)? # possibly an old expansion, followed by...
222 # Given a file, return a regexp matching the possible
223 # RCS keywords that will be expanded, or None for files
224 # with kw expansion turned off.
226 def p4_keywords_regexp_for_file(file):
227 if not os
.path
.exists(file):
230 (type_base
, type_mods
) = split_p4_type(p4_type(file))
231 return p4_keywords_regexp_for_type(type_base
, type_mods
)
233 def setP4ExecBit(file, mode
):
234 # Reopens an already open file and changes the execute bit to match
235 # the execute bit setting in the passed in mode.
239 if not isModeExec(mode
):
240 p4Type
= getP4OpenedType(file)
241 p4Type
= re
.sub('^([cku]?)x(.*)', '\\1\\2', p4Type
)
242 p4Type
= re
.sub('(.*?\+.*?)x(.*?)', '\\1\\2', p4Type
)
243 if p4Type
[-1] == "+":
244 p4Type
= p4Type
[0:-1]
246 p4_reopen(p4Type
, file)
248 def getP4OpenedType(file):
249 # Returns the perforce file type for the given file.
251 result
= p4_read_pipe(["opened", file])
252 match
= re
.match(".*\((.+)\)\r?$", result
)
254 return match
.group(1)
256 die("Could not determine file type for %s (result: '%s')" % (file, result
))
258 # Return the set of all p4 labels
259 def getP4Labels(depotPaths
):
261 if isinstance(depotPaths
,basestring
):
262 depotPaths
= [depotPaths
]
264 for l
in p4CmdList(["labels"] + ["%s..." % p
for p
in depotPaths
]):
270 # Return the set of all git tags
273 for line
in read_pipe_lines(["git", "tag"]):
278 def diffTreePattern():
279 # This is a simple generator for the diff tree regex pattern. This could be
280 # a class variable if this and parseDiffTreeEntry were a part of a class.
281 pattern
= re
.compile(':(\d+) (\d+) (\w+) (\w+) ([A-Z])(\d+)?\t(.*?)((\t(.*))|$)')
285 def parseDiffTreeEntry(entry
):
286 """Parses a single diff tree entry into its component elements.
288 See git-diff-tree(1) manpage for details about the format of the diff
289 output. This method returns a dictionary with the following elements:
291 src_mode - The mode of the source file
292 dst_mode - The mode of the destination file
293 src_sha1 - The sha1 for the source file
294 dst_sha1 - The sha1 fr the destination file
295 status - The one letter status of the diff (i.e. 'A', 'M', 'D', etc)
296 status_score - The score for the status (applicable for 'C' and 'R'
297 statuses). This is None if there is no score.
298 src - The path for the source file.
299 dst - The path for the destination file. This is only present for
300 copy or renames. If it is not present, this is None.
302 If the pattern is not matched, None is returned."""
304 match
= diffTreePattern().next().match(entry
)
307 'src_mode': match
.group(1),
308 'dst_mode': match
.group(2),
309 'src_sha1': match
.group(3),
310 'dst_sha1': match
.group(4),
311 'status': match
.group(5),
312 'status_score': match
.group(6),
313 'src': match
.group(7),
314 'dst': match
.group(10)
318 def isModeExec(mode
):
319 # Returns True if the given git mode represents an executable file,
321 return mode
[-3:] == "755"
323 def isModeExecChanged(src_mode
, dst_mode
):
324 return isModeExec(src_mode
) != isModeExec(dst_mode
)
326 def p4CmdList(cmd
, stdin
=None, stdin_mode
='w+b', cb
=None):
328 if isinstance(cmd
,basestring
):
335 cmd
= p4_build_cmd(cmd
)
337 sys
.stderr
.write("Opening pipe: %s\n" % str(cmd
))
339 # Use a temporary file to avoid deadlocks without
340 # subprocess.communicate(), which would put another copy
341 # of stdout into memory.
343 if stdin
is not None:
344 stdin_file
= tempfile
.TemporaryFile(prefix
='p4-stdin', mode
=stdin_mode
)
345 if isinstance(stdin
,basestring
):
346 stdin_file
.write(stdin
)
349 stdin_file
.write(i
+ '\n')
353 p4
= subprocess
.Popen(cmd
,
356 stdout
=subprocess
.PIPE
)
361 entry
= marshal
.load(p4
.stdout
)
371 entry
["p4ExitCode"] = exitCode
377 list = p4CmdList(cmd
)
383 def p4Where(depotPath
):
384 if not depotPath
.endswith("/"):
386 depotPath
= depotPath
+ "..."
387 outputList
= p4CmdList(["where", depotPath
])
389 for entry
in outputList
:
390 if "depotFile" in entry
:
391 if entry
["depotFile"] == depotPath
:
394 elif "data" in entry
:
395 data
= entry
.get("data")
396 space
= data
.find(" ")
397 if data
[:space
] == depotPath
:
402 if output
["code"] == "error":
406 clientPath
= output
.get("path")
407 elif "data" in output
:
408 data
= output
.get("data")
409 lastSpace
= data
.rfind(" ")
410 clientPath
= data
[lastSpace
+ 1:]
412 if clientPath
.endswith("..."):
413 clientPath
= clientPath
[:-3]
416 def currentGitBranch():
417 return read_pipe("git name-rev HEAD").split(" ")[1].strip()
419 def isValidGitDir(path
):
420 if (os
.path
.exists(path
+ "/HEAD")
421 and os
.path
.exists(path
+ "/refs") and os
.path
.exists(path
+ "/objects")):
425 def parseRevision(ref
):
426 return read_pipe("git rev-parse %s" % ref
).strip()
428 def branchExists(ref
):
429 rev
= read_pipe(["git", "rev-parse", "-q", "--verify", ref
],
433 def extractLogMessageFromGitCommit(commit
):
436 ## fixme: title is first line of commit, not 1st paragraph.
438 for log
in read_pipe_lines("git cat-file commit %s" % commit
):
447 def extractSettingsGitLog(log
):
449 for line
in log
.split("\n"):
451 m
= re
.search (r
"^ *\[git-p4: (.*)\]$", line
)
455 assignments
= m
.group(1).split (':')
456 for a
in assignments
:
458 key
= vals
[0].strip()
459 val
= ('='.join (vals
[1:])).strip()
460 if val
.endswith ('\"') and val
.startswith('"'):
465 paths
= values
.get("depot-paths")
467 paths
= values
.get("depot-path")
469 values
['depot-paths'] = paths
.split(',')
472 def gitBranchExists(branch
):
473 proc
= subprocess
.Popen(["git", "rev-parse", branch
],
474 stderr
=subprocess
.PIPE
, stdout
=subprocess
.PIPE
);
475 return proc
.wait() == 0;
478 def gitConfig(key
, args
= None): # set args to "--bool", for instance
479 if not _gitConfig
.has_key(key
):
482 argsFilter
= "%s " % args
483 cmd
= "git config %s%s" % (argsFilter
, key
)
484 _gitConfig
[key
] = read_pipe(cmd
, ignore_error
=True).strip()
485 return _gitConfig
[key
]
487 def gitConfigList(key
):
488 if not _gitConfig
.has_key(key
):
489 _gitConfig
[key
] = read_pipe("git config --get-all %s" % key
, ignore_error
=True).strip().split(os
.linesep
)
490 return _gitConfig
[key
]
492 def p4BranchesInGit(branchesAreInRemotes
= True):
495 cmdline
= "git rev-parse --symbolic "
496 if branchesAreInRemotes
:
497 cmdline
+= " --remotes"
499 cmdline
+= " --branches"
501 for line
in read_pipe_lines(cmdline
):
504 ## only import to p4/
505 if not line
.startswith('p4/') or line
== "p4/HEAD":
510 branch
= re
.sub ("^p4/", "", line
)
512 branches
[branch
] = parseRevision(line
)
515 def findUpstreamBranchPoint(head
= "HEAD"):
516 branches
= p4BranchesInGit()
517 # map from depot-path to branch name
518 branchByDepotPath
= {}
519 for branch
in branches
.keys():
520 tip
= branches
[branch
]
521 log
= extractLogMessageFromGitCommit(tip
)
522 settings
= extractSettingsGitLog(log
)
523 if settings
.has_key("depot-paths"):
524 paths
= ",".join(settings
["depot-paths"])
525 branchByDepotPath
[paths
] = "remotes/p4/" + branch
529 while parent
< 65535:
530 commit
= head
+ "~%s" % parent
531 log
= extractLogMessageFromGitCommit(commit
)
532 settings
= extractSettingsGitLog(log
)
533 if settings
.has_key("depot-paths"):
534 paths
= ",".join(settings
["depot-paths"])
535 if branchByDepotPath
.has_key(paths
):
536 return [branchByDepotPath
[paths
], settings
]
540 return ["", settings
]
542 def createOrUpdateBranchesFromOrigin(localRefPrefix
= "refs/remotes/p4/", silent
=True):
544 print ("Creating/updating branch(es) in %s based on origin branch(es)"
547 originPrefix
= "origin/p4/"
549 for line
in read_pipe_lines("git rev-parse --symbolic --remotes"):
551 if (not line
.startswith(originPrefix
)) or line
.endswith("HEAD"):
554 headName
= line
[len(originPrefix
):]
555 remoteHead
= localRefPrefix
+ headName
558 original
= extractSettingsGitLog(extractLogMessageFromGitCommit(originHead
))
559 if (not original
.has_key('depot-paths')
560 or not original
.has_key('change')):
564 if not gitBranchExists(remoteHead
):
566 print "creating %s" % remoteHead
569 settings
= extractSettingsGitLog(extractLogMessageFromGitCommit(remoteHead
))
570 if settings
.has_key('change') > 0:
571 if settings
['depot-paths'] == original
['depot-paths']:
572 originP4Change
= int(original
['change'])
573 p4Change
= int(settings
['change'])
574 if originP4Change
> p4Change
:
575 print ("%s (%s) is newer than %s (%s). "
576 "Updating p4 branch from origin."
577 % (originHead
, originP4Change
,
578 remoteHead
, p4Change
))
581 print ("Ignoring: %s was imported from %s while "
582 "%s was imported from %s"
583 % (originHead
, ','.join(original
['depot-paths']),
584 remoteHead
, ','.join(settings
['depot-paths'])))
587 system("git update-ref %s %s" % (remoteHead
, originHead
))
589 def originP4BranchesExist():
590 return gitBranchExists("origin") or gitBranchExists("origin/p4") or gitBranchExists("origin/p4/master")
592 def p4ChangesForPaths(depotPaths
, changeRange
):
596 cmd
+= ["%s...%s" % (p
, changeRange
)]
597 output
= p4_read_pipe_lines(cmd
)
601 changeNum
= int(line
.split(" ")[1])
602 changes
[changeNum
] = True
604 changelist
= changes
.keys()
608 def p4PathStartsWith(path
, prefix
):
609 # This method tries to remedy a potential mixed-case issue:
611 # If UserA adds //depot/DirA/file1
612 # and UserB adds //depot/dira/file2
614 # we may or may not have a problem. If you have core.ignorecase=true,
615 # we treat DirA and dira as the same directory
616 ignorecase
= gitConfig("core.ignorecase", "--bool") == "true"
618 return path
.lower().startswith(prefix
.lower())
619 return path
.startswith(prefix
)
622 """Look at the p4 client spec, create a View() object that contains
623 all the mappings, and return it."""
625 specList
= p4CmdList("client -o")
626 if len(specList
) != 1:
627 die('Output from "client -o" is %d lines, expecting 1' %
630 # dictionary of all client parameters
633 # just the keys that start with "View"
634 view_keys
= [ k
for k
in entry
.keys() if k
.startswith("View") ]
639 # append the lines, in order, to the view
640 for view_num
in range(len(view_keys
)):
641 k
= "View%d" % view_num
642 if k
not in view_keys
:
643 die("Expected view key %s missing" % k
)
644 view
.append(entry
[k
])
649 """Grab the client directory."""
651 output
= p4CmdList("client -o")
653 die('Output from "client -o" is %d lines, expecting 1' % len(output
))
656 if "Root" not in entry
:
657 die('Client has no "Root"')
663 self
.usage
= "usage: %prog [options]"
669 self
.userMapFromPerforceServer
= False
670 self
.myP4UserId
= None
674 return self
.myP4UserId
676 results
= p4CmdList("user -o")
678 if r
.has_key('User'):
679 self
.myP4UserId
= r
['User']
681 die("Could not find your p4 user id")
683 def p4UserIsMe(self
, p4User
):
684 # return True if the given p4 user is actually me
686 if not p4User
or p4User
!= me
:
691 def getUserCacheFilename(self
):
692 home
= os
.environ
.get("HOME", os
.environ
.get("USERPROFILE"))
693 return home
+ "/.gitp4-usercache.txt"
695 def getUserMapFromPerforceServer(self
):
696 if self
.userMapFromPerforceServer
:
701 for output
in p4CmdList("users"):
702 if not output
.has_key("User"):
704 self
.users
[output
["User"]] = output
["FullName"] + " <" + output
["Email"] + ">"
705 self
.emails
[output
["Email"]] = output
["User"]
709 for (key
, val
) in self
.users
.items():
710 s
+= "%s\t%s\n" % (key
.expandtabs(1), val
.expandtabs(1))
712 open(self
.getUserCacheFilename(), "wb").write(s
)
713 self
.userMapFromPerforceServer
= True
715 def loadUserMapFromCache(self
):
717 self
.userMapFromPerforceServer
= False
719 cache
= open(self
.getUserCacheFilename(), "rb")
720 lines
= cache
.readlines()
723 entry
= line
.strip().split("\t")
724 self
.users
[entry
[0]] = entry
[1]
726 self
.getUserMapFromPerforceServer()
728 class P4Debug(Command
):
730 Command
.__init
__(self
)
732 self
.description
= "A tool to debug the output of p4 -G."
733 self
.needsGit
= False
737 for output
in p4CmdList(args
):
738 print 'Element: %d' % j
743 class P4RollBack(Command
):
745 Command
.__init
__(self
)
747 optparse
.make_option("--local", dest
="rollbackLocalBranches", action
="store_true")
749 self
.description
= "A tool to debug the multi-branch import. Don't use :)"
750 self
.rollbackLocalBranches
= False
755 maxChange
= int(args
[0])
757 if "p4ExitCode" in p4Cmd("changes -m 1"):
758 die("Problems executing p4");
760 if self
.rollbackLocalBranches
:
761 refPrefix
= "refs/heads/"
762 lines
= read_pipe_lines("git rev-parse --symbolic --branches")
764 refPrefix
= "refs/remotes/"
765 lines
= read_pipe_lines("git rev-parse --symbolic --remotes")
768 if self
.rollbackLocalBranches
or (line
.startswith("p4/") and line
!= "p4/HEAD\n"):
770 ref
= refPrefix
+ line
771 log
= extractLogMessageFromGitCommit(ref
)
772 settings
= extractSettingsGitLog(log
)
774 depotPaths
= settings
['depot-paths']
775 change
= settings
['change']
779 if len(p4Cmd("changes -m 1 " + ' '.join (['%s...@%s' % (p
, maxChange
)
780 for p
in depotPaths
]))) == 0:
781 print "Branch %s did not exist at change %s, deleting." % (ref
, maxChange
)
782 system("git update-ref -d %s `git rev-parse %s`" % (ref
, ref
))
785 while change
and int(change
) > maxChange
:
788 print "%s is at %s ; rewinding towards %s" % (ref
, change
, maxChange
)
789 system("git update-ref %s \"%s^\"" % (ref
, ref
))
790 log
= extractLogMessageFromGitCommit(ref
)
791 settings
= extractSettingsGitLog(log
)
794 depotPaths
= settings
['depot-paths']
795 change
= settings
['change']
798 print "%s rewound to %s" % (ref
, change
)
802 class P4Submit(Command
, P4UserMap
):
804 Command
.__init
__(self
)
805 P4UserMap
.__init
__(self
)
807 optparse
.make_option("--origin", dest
="origin"),
808 optparse
.make_option("-M", dest
="detectRenames", action
="store_true"),
809 # preserve the user, requires relevant p4 permissions
810 optparse
.make_option("--preserve-user", dest
="preserveUser", action
="store_true"),
811 optparse
.make_option("--export-labels", dest
="exportLabels", action
="store_true"),
813 self
.description
= "Submit changes from git to the perforce depot."
814 self
.usage
+= " [name of git branch to submit into perforce depot]"
815 self
.interactive
= True
817 self
.detectRenames
= False
818 self
.preserveUser
= gitConfig("git-p4.preserveUser").lower() == "true"
819 self
.isWindows
= (platform
.system() == "Windows")
820 self
.exportLabels
= False
823 if len(p4CmdList("opened ...")) > 0:
824 die("You have files opened with perforce! Close them before starting the sync.")
826 # replaces everything between 'Description:' and the next P4 submit template field with the
828 def prepareLogMessage(self
, template
, message
):
831 inDescriptionSection
= False
833 for line
in template
.split("\n"):
834 if line
.startswith("#"):
835 result
+= line
+ "\n"
838 if inDescriptionSection
:
839 if line
.startswith("Files:") or line
.startswith("Jobs:"):
840 inDescriptionSection
= False
844 if line
.startswith("Description:"):
845 inDescriptionSection
= True
847 for messageLine
in message
.split("\n"):
848 line
+= "\t" + messageLine
+ "\n"
850 result
+= line
+ "\n"
854 def patchRCSKeywords(self
, file, pattern
):
855 # Attempt to zap the RCS keywords in a p4 controlled file matching the given pattern
856 (handle
, outFileName
) = tempfile
.mkstemp(dir='.')
858 outFile
= os
.fdopen(handle
, "w+")
859 inFile
= open(file, "r")
860 regexp
= re
.compile(pattern
, re
.VERBOSE
)
861 for line
in inFile
.readlines():
862 line
= regexp
.sub(r
'$\1$', line
)
866 # Forcibly overwrite the original file
868 shutil
.move(outFileName
, file)
870 # cleanup our temporary file
871 os
.unlink(outFileName
)
872 print "Failed to strip RCS keywords in %s" % file
875 print "Patched up RCS keywords in %s" % file
877 def p4UserForCommit(self
,id):
878 # Return the tuple (perforce user,git email) for a given git commit id
879 self
.getUserMapFromPerforceServer()
880 gitEmail
= read_pipe("git log --max-count=1 --format='%%ae' %s" % id)
881 gitEmail
= gitEmail
.strip()
882 if not self
.emails
.has_key(gitEmail
):
883 return (None,gitEmail
)
885 return (self
.emails
[gitEmail
],gitEmail
)
887 def checkValidP4Users(self
,commits
):
888 # check if any git authors cannot be mapped to p4 users
890 (user
,email
) = self
.p4UserForCommit(id)
892 msg
= "Cannot find p4 user for email %s in commit %s." % (email
, id)
893 if gitConfig('git-p4.allowMissingP4Users').lower() == "true":
896 die("Error: %s\nSet git-p4.allowMissingP4Users to true to allow this." % msg
)
898 def lastP4Changelist(self
):
899 # Get back the last changelist number submitted in this client spec. This
900 # then gets used to patch up the username in the change. If the same
901 # client spec is being used by multiple processes then this might go
903 results
= p4CmdList("client -o") # find the current client
906 if r
.has_key('Client'):
910 die("could not get client spec")
911 results
= p4CmdList(["changes", "-c", client
, "-m", "1"])
913 if r
.has_key('change'):
915 die("Could not get changelist number for last submit - cannot patch up user details")
917 def modifyChangelistUser(self
, changelist
, newUser
):
918 # fixup the user field of a changelist after it has been submitted.
919 changes
= p4CmdList("change -o %s" % changelist
)
920 if len(changes
) != 1:
921 die("Bad output from p4 change modifying %s to user %s" %
922 (changelist
, newUser
))
925 if c
['User'] == newUser
: return # nothing to do
927 input = marshal
.dumps(c
)
929 result
= p4CmdList("change -f -i", stdin
=input)
931 if r
.has_key('code'):
932 if r
['code'] == 'error':
933 die("Could not modify user field of changelist %s to %s:%s" % (changelist
, newUser
, r
['data']))
934 if r
.has_key('data'):
935 print("Updated user field for changelist %s to %s" % (changelist
, newUser
))
937 die("Could not modify user field of changelist %s to %s" % (changelist
, newUser
))
939 def canChangeChangelists(self
):
940 # check to see if we have p4 admin or super-user permissions, either of
941 # which are required to modify changelists.
942 results
= p4CmdList(["protects", self
.depotPath
])
944 if r
.has_key('perm'):
945 if r
['perm'] == 'admin':
947 if r
['perm'] == 'super':
951 def prepareSubmitTemplate(self
):
952 # remove lines in the Files section that show changes to files outside the depot path we're committing into
954 inFilesSection
= False
955 for line
in p4_read_pipe_lines(['change', '-o']):
956 if line
.endswith("\r\n"):
957 line
= line
[:-2] + "\n"
959 if line
.startswith("\t"):
960 # path starts and ends with a tab
962 lastTab
= path
.rfind("\t")
964 path
= path
[:lastTab
]
965 if not p4PathStartsWith(path
, self
.depotPath
):
968 inFilesSection
= False
970 if line
.startswith("Files:"):
971 inFilesSection
= True
977 def edit_template(self
, template_file
):
978 """Invoke the editor to let the user change the submission
979 message. Return true if okay to continue with the submit."""
981 # if configured to skip the editing part, just submit
982 if gitConfig("git-p4.skipSubmitEdit") == "true":
985 # look at the modification time, to check later if the user saved
987 mtime
= os
.stat(template_file
).st_mtime
990 if os
.environ
.has_key("P4EDITOR") and (os
.environ
.get("P4EDITOR") != ""):
991 editor
= os
.environ
.get("P4EDITOR")
993 editor
= read_pipe("git var GIT_EDITOR").strip()
994 system(editor
+ " " + template_file
)
996 # If the file was not saved, prompt to see if this patch should
997 # be skipped. But skip this verification step if configured so.
998 if gitConfig("git-p4.skipSubmitEditCheck") == "true":
1001 # modification time updated means user saved the file
1002 if os
.stat(template_file
).st_mtime
> mtime
:
1006 response
= raw_input("Submit template unchanged. Submit anyway? [y]es, [n]o (skip this patch) ")
1012 def applyCommit(self
, id):
1013 print "Applying %s" % (read_pipe("git log --max-count=1 --pretty=oneline %s" % id))
1015 (p4User
, gitEmail
) = self
.p4UserForCommit(id)
1017 if not self
.detectRenames
:
1018 # If not explicitly set check the config variable
1019 self
.detectRenames
= gitConfig("git-p4.detectRenames")
1021 if self
.detectRenames
.lower() == "false" or self
.detectRenames
== "":
1023 elif self
.detectRenames
.lower() == "true":
1026 diffOpts
= "-M%s" % self
.detectRenames
1028 detectCopies
= gitConfig("git-p4.detectCopies")
1029 if detectCopies
.lower() == "true":
1031 elif detectCopies
!= "" and detectCopies
.lower() != "false":
1032 diffOpts
+= " -C%s" % detectCopies
1034 if gitConfig("git-p4.detectCopiesHarder", "--bool") == "true":
1035 diffOpts
+= " --find-copies-harder"
1037 diff
= read_pipe_lines("git diff-tree -r %s \"%s^\" \"%s\"" % (diffOpts
, id, id))
1039 filesToDelete
= set()
1041 filesToChangeExecBit
= {}
1044 diff
= parseDiffTreeEntry(line
)
1045 modifier
= diff
['status']
1049 if isModeExecChanged(diff
['src_mode'], diff
['dst_mode']):
1050 filesToChangeExecBit
[path
] = diff
['dst_mode']
1051 editedFiles
.add(path
)
1052 elif modifier
== "A":
1053 filesToAdd
.add(path
)
1054 filesToChangeExecBit
[path
] = diff
['dst_mode']
1055 if path
in filesToDelete
:
1056 filesToDelete
.remove(path
)
1057 elif modifier
== "D":
1058 filesToDelete
.add(path
)
1059 if path
in filesToAdd
:
1060 filesToAdd
.remove(path
)
1061 elif modifier
== "C":
1062 src
, dest
= diff
['src'], diff
['dst']
1063 p4_integrate(src
, dest
)
1064 if diff
['src_sha1'] != diff
['dst_sha1']:
1066 if isModeExecChanged(diff
['src_mode'], diff
['dst_mode']):
1068 filesToChangeExecBit
[dest
] = diff
['dst_mode']
1070 editedFiles
.add(dest
)
1071 elif modifier
== "R":
1072 src
, dest
= diff
['src'], diff
['dst']
1073 p4_integrate(src
, dest
)
1074 if diff
['src_sha1'] != diff
['dst_sha1']:
1076 if isModeExecChanged(diff
['src_mode'], diff
['dst_mode']):
1078 filesToChangeExecBit
[dest
] = diff
['dst_mode']
1080 editedFiles
.add(dest
)
1081 filesToDelete
.add(src
)
1083 die("unknown modifier %s for %s" % (modifier
, path
))
1085 diffcmd
= "git format-patch -k --stdout \"%s^\"..\"%s\"" % (id, id)
1086 patchcmd
= diffcmd
+ " | git apply "
1087 tryPatchCmd
= patchcmd
+ "--check -"
1088 applyPatchCmd
= patchcmd
+ "--check --apply -"
1089 patch_succeeded
= True
1091 if os
.system(tryPatchCmd
) != 0:
1092 fixed_rcs_keywords
= False
1093 patch_succeeded
= False
1094 print "Unfortunately applying the change failed!"
1096 # Patch failed, maybe it's just RCS keyword woes. Look through
1097 # the patch to see if that's possible.
1098 if gitConfig("git-p4.attemptRCSCleanup","--bool") == "true":
1102 for file in editedFiles | filesToDelete
:
1103 # did this file's delta contain RCS keywords?
1104 pattern
= p4_keywords_regexp_for_file(file)
1107 # this file is a possibility...look for RCS keywords.
1108 regexp
= re
.compile(pattern
, re
.VERBOSE
)
1109 for line
in read_pipe_lines(["git", "diff", "%s^..%s" % (id, id), file]):
1110 if regexp
.search(line
):
1112 print "got keyword match on %s in %s in %s" % (pattern
, line
, file)
1113 kwfiles
[file] = pattern
1116 for file in kwfiles
:
1118 print "zapping %s with %s" % (line
,pattern
)
1119 self
.patchRCSKeywords(file, kwfiles
[file])
1120 fixed_rcs_keywords
= True
1122 if fixed_rcs_keywords
:
1123 print "Retrying the patch with RCS keywords cleaned up"
1124 if os
.system(tryPatchCmd
) == 0:
1125 patch_succeeded
= True
1127 if not patch_succeeded
:
1128 print "What do you want to do?"
1130 while response
!= "s" and response
!= "a" and response
!= "w":
1131 response
= raw_input("[s]kip this patch / [a]pply the patch forcibly "
1132 "and with .rej files / [w]rite the patch to a file (patch.txt) ")
1134 print "Skipping! Good luck with the next patches..."
1135 for f
in editedFiles
:
1137 for f
in filesToAdd
:
1140 elif response
== "a":
1141 os
.system(applyPatchCmd
)
1142 if len(filesToAdd
) > 0:
1143 print "You may also want to call p4 add on the following files:"
1144 print " ".join(filesToAdd
)
1145 if len(filesToDelete
):
1146 print "The following files should be scheduled for deletion with p4 delete:"
1147 print " ".join(filesToDelete
)
1148 die("Please resolve and submit the conflict manually and "
1149 + "continue afterwards with git p4 submit --continue")
1150 elif response
== "w":
1151 system(diffcmd
+ " > patch.txt")
1152 print "Patch saved to patch.txt in %s !" % self
.clientPath
1153 die("Please resolve and submit the conflict manually and "
1154 "continue afterwards with git p4 submit --continue")
1156 system(applyPatchCmd
)
1158 for f
in filesToAdd
:
1160 for f
in filesToDelete
:
1164 # Set/clear executable bits
1165 for f
in filesToChangeExecBit
.keys():
1166 mode
= filesToChangeExecBit
[f
]
1167 setP4ExecBit(f
, mode
)
1169 logMessage
= extractLogMessageFromGitCommit(id)
1170 logMessage
= logMessage
.strip()
1172 template
= self
.prepareSubmitTemplate()
1174 if self
.interactive
:
1175 submitTemplate
= self
.prepareLogMessage(template
, logMessage
)
1177 if self
.preserveUser
:
1178 submitTemplate
= submitTemplate
+ ("\n######## Actual user %s, modified after commit\n" % p4User
)
1180 if os
.environ
.has_key("P4DIFF"):
1181 del(os
.environ
["P4DIFF"])
1183 for editedFile
in editedFiles
:
1184 diff
+= p4_read_pipe(['diff', '-du', editedFile
])
1187 for newFile
in filesToAdd
:
1188 newdiff
+= "==== new file ====\n"
1189 newdiff
+= "--- /dev/null\n"
1190 newdiff
+= "+++ %s\n" % newFile
1191 f
= open(newFile
, "r")
1192 for line
in f
.readlines():
1193 newdiff
+= "+" + line
1196 if self
.checkAuthorship
and not self
.p4UserIsMe(p4User
):
1197 submitTemplate
+= "######## git author %s does not match your p4 account.\n" % gitEmail
1198 submitTemplate
+= "######## Use option --preserve-user to modify authorship.\n"
1199 submitTemplate
+= "######## Variable git-p4.skipUserNameCheck hides this message.\n"
1201 separatorLine
= "######## everything below this line is just the diff #######\n"
1203 (handle
, fileName
) = tempfile
.mkstemp()
1204 tmpFile
= os
.fdopen(handle
, "w+")
1206 submitTemplate
= submitTemplate
.replace("\n", "\r\n")
1207 separatorLine
= separatorLine
.replace("\n", "\r\n")
1208 newdiff
= newdiff
.replace("\n", "\r\n")
1209 tmpFile
.write(submitTemplate
+ separatorLine
+ diff
+ newdiff
)
1212 if self
.edit_template(fileName
):
1213 # read the edited message and submit
1214 tmpFile
= open(fileName
, "rb")
1215 message
= tmpFile
.read()
1217 submitTemplate
= message
[:message
.index(separatorLine
)]
1219 submitTemplate
= submitTemplate
.replace("\r\n", "\n")
1220 p4_write_pipe(['submit', '-i'], submitTemplate
)
1222 if self
.preserveUser
:
1224 # Get last changelist number. Cannot easily get it from
1225 # the submit command output as the output is
1227 changelist
= self
.lastP4Changelist()
1228 self
.modifyChangelistUser(changelist
, p4User
)
1231 print "Submission cancelled, undoing p4 changes."
1232 for f
in editedFiles
:
1234 for f
in filesToAdd
:
1240 fileName
= "submit.txt"
1241 file = open(fileName
, "w+")
1242 file.write(self
.prepareLogMessage(template
, logMessage
))
1244 print ("Perforce submit template written as %s. "
1245 + "Please review/edit and then use p4 submit -i < %s to submit directly!"
1246 % (fileName
, fileName
))
1248 # Export git tags as p4 labels. Create a p4 label and then tag
1250 def exportGitTags(self
, gitTags
):
1251 validLabelRegexp
= gitConfig("git-p4.labelExportRegexp")
1252 if len(validLabelRegexp
) == 0:
1253 validLabelRegexp
= defaultLabelRegexp
1254 m
= re
.compile(validLabelRegexp
)
1256 for name
in gitTags
:
1258 if not m
.match(name
):
1260 print "tag %s does not match regexp %s" % (name
, validTagRegexp
)
1263 # Get the p4 commit this corresponds to
1264 logMessage
= extractLogMessageFromGitCommit(name
)
1265 values
= extractSettingsGitLog(logMessage
)
1267 if not values
.has_key('change'):
1268 # a tag pointing to something not sent to p4; ignore
1270 print "git tag %s does not give a p4 commit" % name
1273 changelist
= values
['change']
1275 # Get the tag details.
1279 for l
in read_pipe_lines(["git", "cat-file", "-p", name
]):
1282 if re
.match(r
'tag\s+', l
):
1284 elif re
.match(r
'\s*$', l
):
1291 body
= ["lightweight tag imported by git p4\n"]
1293 # Create the label - use the same view as the client spec we are using
1294 clientSpec
= getClientSpec()
1296 labelTemplate
= "Label: %s\n" % name
1297 labelTemplate
+= "Description:\n"
1299 labelTemplate
+= "\t" + b
+ "\n"
1300 labelTemplate
+= "View:\n"
1301 for mapping
in clientSpec
.mappings
:
1302 labelTemplate
+= "\t%s\n" % mapping
.depot_side
.path
1304 p4_write_pipe(["label", "-i"], labelTemplate
)
1307 p4_system(["tag", "-l", name
] +
1308 ["%s@%s" % (mapping
.depot_side
.path
, changelist
) for mapping
in clientSpec
.mappings
])
1311 print "created p4 label for tag %s" % name
1313 def run(self
, args
):
1315 self
.master
= currentGitBranch()
1316 if len(self
.master
) == 0 or not gitBranchExists("refs/heads/%s" % self
.master
):
1317 die("Detecting current git branch failed!")
1318 elif len(args
) == 1:
1319 self
.master
= args
[0]
1320 if not branchExists(self
.master
):
1321 die("Branch %s does not exist" % self
.master
)
1325 allowSubmit
= gitConfig("git-p4.allowSubmit")
1326 if len(allowSubmit
) > 0 and not self
.master
in allowSubmit
.split(","):
1327 die("%s is not in git-p4.allowSubmit" % self
.master
)
1329 [upstream
, settings
] = findUpstreamBranchPoint()
1330 self
.depotPath
= settings
['depot-paths'][0]
1331 if len(self
.origin
) == 0:
1332 self
.origin
= upstream
1334 if self
.preserveUser
:
1335 if not self
.canChangeChangelists():
1336 die("Cannot preserve user names without p4 super-user or admin permissions")
1339 print "Origin branch is " + self
.origin
1341 if len(self
.depotPath
) == 0:
1342 print "Internal error: cannot locate perforce depot path from existing branches"
1345 self
.useClientSpec
= False
1346 if gitConfig("git-p4.useclientspec", "--bool") == "true":
1347 self
.useClientSpec
= True
1348 if self
.useClientSpec
:
1349 self
.clientSpecDirs
= getClientSpec()
1351 if self
.useClientSpec
:
1352 # all files are relative to the client spec
1353 self
.clientPath
= getClientRoot()
1355 self
.clientPath
= p4Where(self
.depotPath
)
1357 if self
.clientPath
== "":
1358 die("Error: Cannot locate perforce checkout of %s in client view" % self
.depotPath
)
1360 print "Perforce checkout for depot path %s located at %s" % (self
.depotPath
, self
.clientPath
)
1361 self
.oldWorkingDirectory
= os
.getcwd()
1363 # ensure the clientPath exists
1364 if not os
.path
.exists(self
.clientPath
):
1365 os
.makedirs(self
.clientPath
)
1367 chdir(self
.clientPath
)
1368 print "Synchronizing p4 checkout..."
1373 for line
in read_pipe_lines("git rev-list --no-merges %s..%s" % (self
.origin
, self
.master
)):
1374 commits
.append(line
.strip())
1377 if self
.preserveUser
or (gitConfig("git-p4.skipUserNameCheck") == "true"):
1378 self
.checkAuthorship
= False
1380 self
.checkAuthorship
= True
1382 if self
.preserveUser
:
1383 self
.checkValidP4Users(commits
)
1385 while len(commits
) > 0:
1387 commits
= commits
[1:]
1388 self
.applyCommit(commit
)
1389 if not self
.interactive
:
1392 if len(commits
) == 0:
1393 print "All changes applied!"
1394 chdir(self
.oldWorkingDirectory
)
1402 if gitConfig("git-p4.exportLabels", "--bool") == "true":
1403 self
.exportLabels
= true
1405 if self
.exportLabels
:
1406 p4Labels
= getP4Labels(self
.depotPath
)
1407 gitTags
= getGitTags()
1409 missingGitTags
= gitTags
- p4Labels
1410 self
.exportGitTags(missingGitTags
)
1415 """Represent a p4 view ("p4 help views"), and map files in a
1416 repo according to the view."""
1419 """A depot or client path, possibly containing wildcards.
1420 The only one supported is ... at the end, currently.
1421 Initialize with the full path, with //depot or //client."""
1423 def __init__(self
, path
, is_depot
):
1425 self
.is_depot
= is_depot
1426 self
.find_wildcards()
1427 # remember the prefix bit, useful for relative mappings
1428 m
= re
.match("(//[^/]+/)", self
.path
)
1430 die("Path %s does not start with //prefix/" % self
.path
)
1432 if not self
.is_depot
:
1433 # strip //client/ on client paths
1434 self
.path
= self
.path
[len(prefix
):]
1436 def find_wildcards(self
):
1437 """Make sure wildcards are valid, and set up internal
1440 self
.ends_triple_dot
= False
1441 # There are three wildcards allowed in p4 views
1442 # (see "p4 help views"). This code knows how to
1443 # handle "..." (only at the end), but cannot deal with
1444 # "%%n" or "*". Only check the depot_side, as p4 should
1445 # validate that the client_side matches too.
1446 if re
.search(r
'%%[1-9]', self
.path
):
1447 die("Can't handle %%n wildcards in view: %s" % self
.path
)
1448 if self
.path
.find("*") >= 0:
1449 die("Can't handle * wildcards in view: %s" % self
.path
)
1450 triple_dot_index
= self
.path
.find("...")
1451 if triple_dot_index
>= 0:
1452 if triple_dot_index
!= len(self
.path
) - 3:
1453 die("Can handle only single ... wildcard, at end: %s" %
1455 self
.ends_triple_dot
= True
1457 def ensure_compatible(self
, other_path
):
1458 """Make sure the wildcards agree."""
1459 if self
.ends_triple_dot
!= other_path
.ends_triple_dot
:
1460 die("Both paths must end with ... if either does;\n" +
1461 "paths: %s %s" % (self
.path
, other_path
.path
))
1463 def match_wildcards(self
, test_path
):
1464 """See if this test_path matches us, and fill in the value
1465 of the wildcards if so. Returns a tuple of
1466 (True|False, wildcards[]). For now, only the ... at end
1467 is supported, so at most one wildcard."""
1468 if self
.ends_triple_dot
:
1469 dotless
= self
.path
[:-3]
1470 if test_path
.startswith(dotless
):
1471 wildcard
= test_path
[len(dotless
):]
1472 return (True, [ wildcard
])
1474 if test_path
== self
.path
:
1478 def match(self
, test_path
):
1479 """Just return if it matches; don't bother with the wildcards."""
1480 b
, _
= self
.match_wildcards(test_path
)
1483 def fill_in_wildcards(self
, wildcards
):
1484 """Return the relative path, with the wildcards filled in
1485 if there are any."""
1486 if self
.ends_triple_dot
:
1487 return self
.path
[:-3] + wildcards
[0]
1491 class Mapping(object):
1492 def __init__(self
, depot_side
, client_side
, overlay
, exclude
):
1493 # depot_side is without the trailing /... if it had one
1494 self
.depot_side
= View
.Path(depot_side
, is_depot
=True)
1495 self
.client_side
= View
.Path(client_side
, is_depot
=False)
1496 self
.overlay
= overlay
# started with "+"
1497 self
.exclude
= exclude
# started with "-"
1498 assert not (self
.overlay
and self
.exclude
)
1499 self
.depot_side
.ensure_compatible(self
.client_side
)
1507 return "View.Mapping: %s%s -> %s" % \
1508 (c
, self
.depot_side
.path
, self
.client_side
.path
)
1510 def map_depot_to_client(self
, depot_path
):
1511 """Calculate the client path if using this mapping on the
1512 given depot path; does not consider the effect of other
1513 mappings in a view. Even excluded mappings are returned."""
1514 matches
, wildcards
= self
.depot_side
.match_wildcards(depot_path
)
1517 client_path
= self
.client_side
.fill_in_wildcards(wildcards
)
1526 def append(self
, view_line
):
1527 """Parse a view line, splitting it into depot and client
1528 sides. Append to self.mappings, preserving order."""
1530 # Split the view line into exactly two words. P4 enforces
1531 # structure on these lines that simplifies this quite a bit.
1533 # Either or both words may be double-quoted.
1534 # Single quotes do not matter.
1535 # Double-quote marks cannot occur inside the words.
1536 # A + or - prefix is also inside the quotes.
1537 # There are no quotes unless they contain a space.
1538 # The line is already white-space stripped.
1539 # The two words are separated by a single space.
1541 if view_line
[0] == '"':
1542 # First word is double quoted. Find its end.
1543 close_quote_index
= view_line
.find('"', 1)
1544 if close_quote_index
<= 0:
1545 die("No first-word closing quote found: %s" % view_line
)
1546 depot_side
= view_line
[1:close_quote_index
]
1547 # skip closing quote and space
1548 rhs_index
= close_quote_index
+ 1 + 1
1550 space_index
= view_line
.find(" ")
1551 if space_index
<= 0:
1552 die("No word-splitting space found: %s" % view_line
)
1553 depot_side
= view_line
[0:space_index
]
1554 rhs_index
= space_index
+ 1
1556 if view_line
[rhs_index
] == '"':
1557 # Second word is double quoted. Make sure there is a
1558 # double quote at the end too.
1559 if not view_line
.endswith('"'):
1560 die("View line with rhs quote should end with one: %s" %
1563 client_side
= view_line
[rhs_index
+1:-1]
1565 client_side
= view_line
[rhs_index
:]
1567 # prefix + means overlay on previous mapping
1569 if depot_side
.startswith("+"):
1571 depot_side
= depot_side
[1:]
1573 # prefix - means exclude this path
1575 if depot_side
.startswith("-"):
1577 depot_side
= depot_side
[1:]
1579 m
= View
.Mapping(depot_side
, client_side
, overlay
, exclude
)
1580 self
.mappings
.append(m
)
1582 def map_in_client(self
, depot_path
):
1583 """Return the relative location in the client where this
1584 depot file should live. Returns "" if the file should
1585 not be mapped in the client."""
1590 # look at later entries first
1591 for m
in self
.mappings
[::-1]:
1593 # see where will this path end up in the client
1594 p
= m
.map_depot_to_client(depot_path
)
1597 # Depot path does not belong in client. Must remember
1598 # this, as previous items should not cause files to
1599 # exist in this path either. Remember that the list is
1600 # being walked from the end, which has higher precedence.
1601 # Overlap mappings do not exclude previous mappings.
1603 paths_filled
.append(m
.client_side
)
1606 # This mapping matched; no need to search any further.
1607 # But, the mapping could be rejected if the client path
1608 # has already been claimed by an earlier mapping (i.e.
1609 # one later in the list, which we are walking backwards).
1610 already_mapped_in_client
= False
1611 for f
in paths_filled
:
1612 # this is View.Path.match
1614 already_mapped_in_client
= True
1616 if not already_mapped_in_client
:
1617 # Include this file, unless it is from a line that
1618 # explicitly said to exclude it.
1622 # a match, even if rejected, always stops the search
1627 class P4Sync(Command
, P4UserMap
):
1628 delete_actions
= ( "delete", "move/delete", "purge" )
1631 Command
.__init
__(self
)
1632 P4UserMap
.__init
__(self
)
1634 optparse
.make_option("--branch", dest
="branch"),
1635 optparse
.make_option("--detect-branches", dest
="detectBranches", action
="store_true"),
1636 optparse
.make_option("--changesfile", dest
="changesFile"),
1637 optparse
.make_option("--silent", dest
="silent", action
="store_true"),
1638 optparse
.make_option("--detect-labels", dest
="detectLabels", action
="store_true"),
1639 optparse
.make_option("--import-labels", dest
="importLabels", action
="store_true"),
1640 optparse
.make_option("--import-local", dest
="importIntoRemotes", action
="store_false",
1641 help="Import into refs/heads/ , not refs/remotes"),
1642 optparse
.make_option("--max-changes", dest
="maxChanges"),
1643 optparse
.make_option("--keep-path", dest
="keepRepoPath", action
='store_true',
1644 help="Keep entire BRANCH/DIR/SUBDIR prefix during import"),
1645 optparse
.make_option("--use-client-spec", dest
="useClientSpec", action
='store_true',
1646 help="Only sync files that are included in the Perforce Client Spec")
1648 self
.description
= """Imports from Perforce into a git repository.\n
1650 //depot/my/project/ -- to import the current head
1651 //depot/my/project/@all -- to import everything
1652 //depot/my/project/@1,6 -- to import only from revision 1 to 6
1654 (a ... is not needed in the path p4 specification, it's added implicitly)"""
1656 self
.usage
+= " //depot/path[@revRange]"
1658 self
.createdBranches
= set()
1659 self
.committedChanges
= set()
1661 self
.detectBranches
= False
1662 self
.detectLabels
= False
1663 self
.importLabels
= False
1664 self
.changesFile
= ""
1665 self
.syncWithOrigin
= True
1666 self
.importIntoRemotes
= True
1667 self
.maxChanges
= ""
1668 self
.isWindows
= (platform
.system() == "Windows")
1669 self
.keepRepoPath
= False
1670 self
.depotPaths
= None
1671 self
.p4BranchesInGit
= []
1672 self
.cloneExclude
= []
1673 self
.useClientSpec
= False
1674 self
.useClientSpec_from_options
= False
1675 self
.clientSpecDirs
= None
1676 self
.tempBranches
= []
1677 self
.tempBranchLocation
= "git-p4-tmp"
1679 if gitConfig("git-p4.syncFromOrigin") == "false":
1680 self
.syncWithOrigin
= False
1683 # P4 wildcards are not allowed in filenames. P4 complains
1684 # if you simply add them, but you can force it with "-f", in
1685 # which case it translates them into %xx encoding internally.
1686 # Search for and fix just these four characters. Do % last so
1687 # that fixing it does not inadvertently create new %-escapes.
1689 def wildcard_decode(self
, path
):
1690 # Cannot have * in a filename in windows; untested as to
1691 # what p4 would do in such a case.
1692 if not self
.isWindows
:
1693 path
= path
.replace("%2A", "*")
1694 path
= path
.replace("%23", "#") \
1695 .replace("%40", "@") \
1696 .replace("%25", "%")
1699 # Force a checkpoint in fast-import and wait for it to finish
1700 def checkpoint(self
):
1701 self
.gitStream
.write("checkpoint\n\n")
1702 self
.gitStream
.write("progress checkpoint\n\n")
1703 out
= self
.gitOutput
.readline()
1705 print "checkpoint finished: " + out
1707 def extractFilesFromCommit(self
, commit
):
1708 self
.cloneExclude
= [re
.sub(r
"\.\.\.$", "", path
)
1709 for path
in self
.cloneExclude
]
1712 while commit
.has_key("depotFile%s" % fnum
):
1713 path
= commit
["depotFile%s" % fnum
]
1715 if [p
for p
in self
.cloneExclude
1716 if p4PathStartsWith(path
, p
)]:
1719 found
= [p
for p
in self
.depotPaths
1720 if p4PathStartsWith(path
, p
)]
1727 file["rev"] = commit
["rev%s" % fnum
]
1728 file["action"] = commit
["action%s" % fnum
]
1729 file["type"] = commit
["type%s" % fnum
]
1734 def stripRepoPath(self
, path
, prefixes
):
1735 if self
.useClientSpec
:
1736 return self
.clientSpecDirs
.map_in_client(path
)
1738 if self
.keepRepoPath
:
1739 prefixes
= [re
.sub("^(//[^/]+/).*", r
'\1', prefixes
[0])]
1742 if p4PathStartsWith(path
, p
):
1743 path
= path
[len(p
):]
1747 def splitFilesIntoBranches(self
, commit
):
1750 while commit
.has_key("depotFile%s" % fnum
):
1751 path
= commit
["depotFile%s" % fnum
]
1752 found
= [p
for p
in self
.depotPaths
1753 if p4PathStartsWith(path
, p
)]
1760 file["rev"] = commit
["rev%s" % fnum
]
1761 file["action"] = commit
["action%s" % fnum
]
1762 file["type"] = commit
["type%s" % fnum
]
1765 relPath
= self
.stripRepoPath(path
, self
.depotPaths
)
1767 for branch
in self
.knownBranches
.keys():
1769 # add a trailing slash so that a commit into qt/4.2foo doesn't end up in qt/4.2
1770 if relPath
.startswith(branch
+ "/"):
1771 if branch
not in branches
:
1772 branches
[branch
] = []
1773 branches
[branch
].append(file)
1778 # output one file from the P4 stream
1779 # - helper for streamP4Files
1781 def streamOneP4File(self
, file, contents
):
1782 relPath
= self
.stripRepoPath(file['depotFile'], self
.branchPrefixes
)
1783 relPath
= self
.wildcard_decode(relPath
)
1785 sys
.stderr
.write("%s\n" % relPath
)
1787 (type_base
, type_mods
) = split_p4_type(file["type"])
1790 if "x" in type_mods
:
1792 if type_base
== "symlink":
1794 # p4 print on a symlink contains "target\n"; remove the newline
1795 data
= ''.join(contents
)
1796 contents
= [data
[:-1]]
1798 if type_base
== "utf16":
1799 # p4 delivers different text in the python output to -G
1800 # than it does when using "print -o", or normal p4 client
1801 # operations. utf16 is converted to ascii or utf8, perhaps.
1802 # But ascii text saved as -t utf16 is completely mangled.
1803 # Invoke print -o to get the real contents.
1804 text
= p4_read_pipe(['print', '-q', '-o', '-', file['depotFile']])
1807 if type_base
== "apple":
1808 # Apple filetype files will be streamed as a concatenation of
1809 # its appledouble header and the contents. This is useless
1810 # on both macs and non-macs. If using "print -q -o xx", it
1811 # will create "xx" with the data, and "%xx" with the header.
1812 # This is also not very useful.
1814 # Ideally, someday, this script can learn how to generate
1815 # appledouble files directly and import those to git, but
1816 # non-mac machines can never find a use for apple filetype.
1817 print "\nIgnoring apple filetype file %s" % file['depotFile']
1820 # Perhaps windows wants unicode, utf16 newlines translated too;
1821 # but this is not doing it.
1822 if self
.isWindows
and type_base
== "text":
1824 for data
in contents
:
1825 data
= data
.replace("\r\n", "\n")
1826 mangled
.append(data
)
1829 # Note that we do not try to de-mangle keywords on utf16 files,
1830 # even though in theory somebody may want that.
1831 pattern
= p4_keywords_regexp_for_type(type_base
, type_mods
)
1833 regexp
= re
.compile(pattern
, re
.VERBOSE
)
1834 text
= ''.join(contents
)
1835 text
= regexp
.sub(r
'$\1$', text
)
1838 self
.gitStream
.write("M %s inline %s\n" % (git_mode
, relPath
))
1843 length
= length
+ len(d
)
1845 self
.gitStream
.write("data %d\n" % length
)
1847 self
.gitStream
.write(d
)
1848 self
.gitStream
.write("\n")
1850 def streamOneP4Deletion(self
, file):
1851 relPath
= self
.stripRepoPath(file['path'], self
.branchPrefixes
)
1853 sys
.stderr
.write("delete %s\n" % relPath
)
1854 self
.gitStream
.write("D %s\n" % relPath
)
1856 # handle another chunk of streaming data
1857 def streamP4FilesCb(self
, marshalled
):
1859 if marshalled
.has_key('depotFile') and self
.stream_have_file_info
:
1860 # start of a new file - output the old one first
1861 self
.streamOneP4File(self
.stream_file
, self
.stream_contents
)
1862 self
.stream_file
= {}
1863 self
.stream_contents
= []
1864 self
.stream_have_file_info
= False
1866 # pick up the new file information... for the
1867 # 'data' field we need to append to our array
1868 for k
in marshalled
.keys():
1870 self
.stream_contents
.append(marshalled
['data'])
1872 self
.stream_file
[k
] = marshalled
[k
]
1874 self
.stream_have_file_info
= True
1876 # Stream directly from "p4 files" into "git fast-import"
1877 def streamP4Files(self
, files
):
1883 # if using a client spec, only add the files that have
1884 # a path in the client
1885 if self
.clientSpecDirs
:
1886 if self
.clientSpecDirs
.map_in_client(f
['path']) == "":
1889 filesForCommit
.append(f
)
1890 if f
['action'] in self
.delete_actions
:
1891 filesToDelete
.append(f
)
1893 filesToRead
.append(f
)
1896 for f
in filesToDelete
:
1897 self
.streamOneP4Deletion(f
)
1899 if len(filesToRead
) > 0:
1900 self
.stream_file
= {}
1901 self
.stream_contents
= []
1902 self
.stream_have_file_info
= False
1904 # curry self argument
1905 def streamP4FilesCbSelf(entry
):
1906 self
.streamP4FilesCb(entry
)
1908 fileArgs
= ['%s#%s' % (f
['path'], f
['rev']) for f
in filesToRead
]
1910 p4CmdList(["-x", "-", "print"],
1912 cb
=streamP4FilesCbSelf
)
1915 if self
.stream_file
.has_key('depotFile'):
1916 self
.streamOneP4File(self
.stream_file
, self
.stream_contents
)
1918 def make_email(self
, userid
):
1919 if userid
in self
.users
:
1920 return self
.users
[userid
]
1922 return "%s <a@b>" % userid
1925 def streamTag(self
, gitStream
, labelName
, labelDetails
, commit
, epoch
):
1927 print "writing tag %s for commit %s" % (labelName
, commit
)
1928 gitStream
.write("tag %s\n" % labelName
)
1929 gitStream
.write("from %s\n" % commit
)
1931 if labelDetails
.has_key('Owner'):
1932 owner
= labelDetails
["Owner"]
1936 # Try to use the owner of the p4 label, or failing that,
1937 # the current p4 user id.
1939 email
= self
.make_email(owner
)
1941 email
= self
.make_email(self
.p4UserId())
1942 tagger
= "%s %s %s" % (email
, epoch
, self
.tz
)
1944 gitStream
.write("tagger %s\n" % tagger
)
1946 print "labelDetails=",labelDetails
1947 if labelDetails
.has_key('Description'):
1948 description
= labelDetails
['Description']
1950 description
= 'Label from git p4'
1952 gitStream
.write("data %d\n" % len(description
))
1953 gitStream
.write(description
)
1954 gitStream
.write("\n")
1956 def commit(self
, details
, files
, branch
, branchPrefixes
, parent
= ""):
1957 epoch
= details
["time"]
1958 author
= details
["user"]
1959 self
.branchPrefixes
= branchPrefixes
1962 print "commit into %s" % branch
1964 # start with reading files; if that fails, we should not
1968 if [p
for p
in branchPrefixes
if p4PathStartsWith(f
['path'], p
)]:
1969 new_files
.append (f
)
1971 sys
.stderr
.write("Ignoring file outside of prefix: %s\n" % f
['path'])
1973 self
.gitStream
.write("commit %s\n" % branch
)
1974 # gitStream.write("mark :%s\n" % details["change"])
1975 self
.committedChanges
.add(int(details
["change"]))
1977 if author
not in self
.users
:
1978 self
.getUserMapFromPerforceServer()
1979 committer
= "%s %s %s" % (self
.make_email(author
), epoch
, self
.tz
)
1981 self
.gitStream
.write("committer %s\n" % committer
)
1983 self
.gitStream
.write("data <<EOT\n")
1984 self
.gitStream
.write(details
["desc"])
1985 self
.gitStream
.write("\n[git-p4: depot-paths = \"%s\": change = %s"
1986 % (','.join (branchPrefixes
), details
["change"]))
1987 if len(details
['options']) > 0:
1988 self
.gitStream
.write(": options = %s" % details
['options'])
1989 self
.gitStream
.write("]\nEOT\n\n")
1993 print "parent %s" % parent
1994 self
.gitStream
.write("from %s\n" % parent
)
1996 self
.streamP4Files(new_files
)
1997 self
.gitStream
.write("\n")
1999 change
= int(details
["change"])
2001 if self
.labels
.has_key(change
):
2002 label
= self
.labels
[change
]
2003 labelDetails
= label
[0]
2004 labelRevisions
= label
[1]
2006 print "Change %s is labelled %s" % (change
, labelDetails
)
2008 files
= p4CmdList(["files"] + ["%s...@%s" % (p
, change
)
2009 for p
in branchPrefixes
])
2011 if len(files
) == len(labelRevisions
):
2015 if info
["action"] in self
.delete_actions
:
2017 cleanedFiles
[info
["depotFile"]] = info
["rev"]
2019 if cleanedFiles
== labelRevisions
:
2020 self
.streamTag(self
.gitStream
, 'tag_%s' % labelDetails
['label'], labelDetails
, branch
, epoch
)
2024 print ("Tag %s does not match with change %s: files do not match."
2025 % (labelDetails
["label"], change
))
2029 print ("Tag %s does not match with change %s: file count is different."
2030 % (labelDetails
["label"], change
))
2032 # Build a dictionary of changelists and labels, for "detect-labels" option.
2033 def getLabels(self
):
2036 l
= p4CmdList(["labels"] + ["%s..." % p
for p
in self
.depotPaths
])
2037 if len(l
) > 0 and not self
.silent
:
2038 print "Finding files belonging to labels in %s" % `self
.depotPaths`
2041 label
= output
["label"]
2045 print "Querying files for label %s" % label
2046 for file in p4CmdList(["files"] +
2047 ["%s...@%s" % (p
, label
)
2048 for p
in self
.depotPaths
]):
2049 revisions
[file["depotFile"]] = file["rev"]
2050 change
= int(file["change"])
2051 if change
> newestChange
:
2052 newestChange
= change
2054 self
.labels
[newestChange
] = [output
, revisions
]
2057 print "Label changes: %s" % self
.labels
.keys()
2059 # Import p4 labels as git tags. A direct mapping does not
2060 # exist, so assume that if all the files are at the same revision
2061 # then we can use that, or it's something more complicated we should
2063 def importP4Labels(self
, stream
, p4Labels
):
2065 print "import p4 labels: " + ' '.join(p4Labels
)
2067 ignoredP4Labels
= gitConfigList("git-p4.ignoredP4Labels")
2068 validLabelRegexp
= gitConfig("git-p4.labelImportRegexp")
2069 if len(validLabelRegexp
) == 0:
2070 validLabelRegexp
= defaultLabelRegexp
2071 m
= re
.compile(validLabelRegexp
)
2073 for name
in p4Labels
:
2076 if not m
.match(name
):
2078 print "label %s does not match regexp %s" % (name
,validLabelRegexp
)
2081 if name
in ignoredP4Labels
:
2084 labelDetails
= p4CmdList(['label', "-o", name
])[0]
2086 # get the most recent changelist for each file in this label
2087 change
= p4Cmd(["changes", "-m", "1"] + ["%s...@%s" % (p
, name
)
2088 for p
in self
.depotPaths
])
2090 if change
.has_key('change'):
2091 # find the corresponding git commit; take the oldest commit
2092 changelist
= int(change
['change'])
2093 gitCommit
= read_pipe(["git", "rev-list", "--max-count=1",
2094 "--reverse", ":/\[git-p4:.*change = %d\]" % changelist
])
2095 if len(gitCommit
) == 0:
2096 print "could not find git commit for changelist %d" % changelist
2098 gitCommit
= gitCommit
.strip()
2100 # Convert from p4 time format
2102 tmwhen
= time
.strptime(labelDetails
['Update'], "%Y/%m/%d %H:%M:%S")
2104 print "Could not convert label time %s" % labelDetail
['Update']
2107 when
= int(time
.mktime(tmwhen
))
2108 self
.streamTag(stream
, name
, labelDetails
, gitCommit
, when
)
2110 print "p4 label %s mapped to git commit %s" % (name
, gitCommit
)
2113 print "Label %s has no changelists - possibly deleted?" % name
2116 # We can't import this label; don't try again as it will get very
2117 # expensive repeatedly fetching all the files for labels that will
2118 # never be imported. If the label is moved in the future, the
2119 # ignore will need to be removed manually.
2120 system(["git", "config", "--add", "git-p4.ignoredP4Labels", name
])
2122 def guessProjectName(self
):
2123 for p
in self
.depotPaths
:
2126 p
= p
[p
.strip().rfind("/") + 1:]
2127 if not p
.endswith("/"):
2131 def getBranchMapping(self
):
2132 lostAndFoundBranches
= set()
2134 user
= gitConfig("git-p4.branchUser")
2136 command
= "branches -u %s" % user
2138 command
= "branches"
2140 for info
in p4CmdList(command
):
2141 details
= p4Cmd(["branch", "-o", info
["branch"]])
2143 while details
.has_key("View%s" % viewIdx
):
2144 paths
= details
["View%s" % viewIdx
].split(" ")
2145 viewIdx
= viewIdx
+ 1
2146 # require standard //depot/foo/... //depot/bar/... mapping
2147 if len(paths
) != 2 or not paths
[0].endswith("/...") or not paths
[1].endswith("/..."):
2150 destination
= paths
[1]
2152 if p4PathStartsWith(source
, self
.depotPaths
[0]) and p4PathStartsWith(destination
, self
.depotPaths
[0]):
2153 source
= source
[len(self
.depotPaths
[0]):-4]
2154 destination
= destination
[len(self
.depotPaths
[0]):-4]
2156 if destination
in self
.knownBranches
:
2158 print "p4 branch %s defines a mapping from %s to %s" % (info
["branch"], source
, destination
)
2159 print "but there exists another mapping from %s to %s already!" % (self
.knownBranches
[destination
], destination
)
2162 self
.knownBranches
[destination
] = source
2164 lostAndFoundBranches
.discard(destination
)
2166 if source
not in self
.knownBranches
:
2167 lostAndFoundBranches
.add(source
)
2169 # Perforce does not strictly require branches to be defined, so we also
2170 # check git config for a branch list.
2172 # Example of branch definition in git config file:
2174 # branchList=main:branchA
2175 # branchList=main:branchB
2176 # branchList=branchA:branchC
2177 configBranches
= gitConfigList("git-p4.branchList")
2178 for branch
in configBranches
:
2180 (source
, destination
) = branch
.split(":")
2181 self
.knownBranches
[destination
] = source
2183 lostAndFoundBranches
.discard(destination
)
2185 if source
not in self
.knownBranches
:
2186 lostAndFoundBranches
.add(source
)
2189 for branch
in lostAndFoundBranches
:
2190 self
.knownBranches
[branch
] = branch
2192 def getBranchMappingFromGitBranches(self
):
2193 branches
= p4BranchesInGit(self
.importIntoRemotes
)
2194 for branch
in branches
.keys():
2195 if branch
== "master":
2198 branch
= branch
[len(self
.projectName
):]
2199 self
.knownBranches
[branch
] = branch
2201 def listExistingP4GitBranches(self
):
2202 # branches holds mapping from name to commit
2203 branches
= p4BranchesInGit(self
.importIntoRemotes
)
2204 self
.p4BranchesInGit
= branches
.keys()
2205 for branch
in branches
.keys():
2206 self
.initialParents
[self
.refPrefix
+ branch
] = branches
[branch
]
2208 def updateOptionDict(self
, d
):
2210 if self
.keepRepoPath
:
2211 option_keys
['keepRepoPath'] = 1
2213 d
["options"] = ' '.join(sorted(option_keys
.keys()))
2215 def readOptions(self
, d
):
2216 self
.keepRepoPath
= (d
.has_key('options')
2217 and ('keepRepoPath' in d
['options']))
2219 def gitRefForBranch(self
, branch
):
2220 if branch
== "main":
2221 return self
.refPrefix
+ "master"
2223 if len(branch
) <= 0:
2226 return self
.refPrefix
+ self
.projectName
+ branch
2228 def gitCommitByP4Change(self
, ref
, change
):
2230 print "looking in ref " + ref
+ " for change %s using bisect..." % change
2233 latestCommit
= parseRevision(ref
)
2237 print "trying: earliest %s latest %s" % (earliestCommit
, latestCommit
)
2238 next
= read_pipe("git rev-list --bisect %s %s" % (latestCommit
, earliestCommit
)).strip()
2243 log
= extractLogMessageFromGitCommit(next
)
2244 settings
= extractSettingsGitLog(log
)
2245 currentChange
= int(settings
['change'])
2247 print "current change %s" % currentChange
2249 if currentChange
== change
:
2251 print "found %s" % next
2254 if currentChange
< change
:
2255 earliestCommit
= "^%s" % next
2257 latestCommit
= "%s" % next
2261 def importNewBranch(self
, branch
, maxChange
):
2262 # make fast-import flush all changes to disk and update the refs using the checkpoint
2263 # command so that we can try to find the branch parent in the git history
2264 self
.gitStream
.write("checkpoint\n\n");
2265 self
.gitStream
.flush();
2266 branchPrefix
= self
.depotPaths
[0] + branch
+ "/"
2267 range = "@1,%s" % maxChange
2268 #print "prefix" + branchPrefix
2269 changes
= p4ChangesForPaths([branchPrefix
], range)
2270 if len(changes
) <= 0:
2272 firstChange
= changes
[0]
2273 #print "first change in branch: %s" % firstChange
2274 sourceBranch
= self
.knownBranches
[branch
]
2275 sourceDepotPath
= self
.depotPaths
[0] + sourceBranch
2276 sourceRef
= self
.gitRefForBranch(sourceBranch
)
2277 #print "source " + sourceBranch
2279 branchParentChange
= int(p4Cmd(["changes", "-m", "1", "%s...@1,%s" % (sourceDepotPath
, firstChange
)])["change"])
2280 #print "branch parent: %s" % branchParentChange
2281 gitParent
= self
.gitCommitByP4Change(sourceRef
, branchParentChange
)
2282 if len(gitParent
) > 0:
2283 self
.initialParents
[self
.gitRefForBranch(branch
)] = gitParent
2284 #print "parent git commit: %s" % gitParent
2286 self
.importChanges(changes
)
2289 def searchParent(self
, parent
, branch
, target
):
2291 for blob
in read_pipe_lines(["git", "rev-list", "--reverse", "--no-merges", parent
]):
2293 if len(read_pipe(["git", "diff-tree", blob
, target
])) == 0:
2296 print "Found parent of %s in commit %s" % (branch
, blob
)
2303 def importChanges(self
, changes
):
2305 for change
in changes
:
2306 description
= p4Cmd(["describe", str(change
)])
2307 self
.updateOptionDict(description
)
2310 sys
.stdout
.write("\rImporting revision %s (%s%%)" % (change
, cnt
* 100 / len(changes
)))
2315 if self
.detectBranches
:
2316 branches
= self
.splitFilesIntoBranches(description
)
2317 for branch
in branches
.keys():
2319 branchPrefix
= self
.depotPaths
[0] + branch
+ "/"
2323 filesForCommit
= branches
[branch
]
2326 print "branch is %s" % branch
2328 self
.updatedBranches
.add(branch
)
2330 if branch
not in self
.createdBranches
:
2331 self
.createdBranches
.add(branch
)
2332 parent
= self
.knownBranches
[branch
]
2333 if parent
== branch
:
2336 fullBranch
= self
.projectName
+ branch
2337 if fullBranch
not in self
.p4BranchesInGit
:
2339 print("\n Importing new branch %s" % fullBranch
);
2340 if self
.importNewBranch(branch
, change
- 1):
2342 self
.p4BranchesInGit
.append(fullBranch
)
2344 print("\n Resuming with change %s" % change
);
2347 print "parent determined through known branches: %s" % parent
2349 branch
= self
.gitRefForBranch(branch
)
2350 parent
= self
.gitRefForBranch(parent
)
2353 print "looking for initial parent for %s; current parent is %s" % (branch
, parent
)
2355 if len(parent
) == 0 and branch
in self
.initialParents
:
2356 parent
= self
.initialParents
[branch
]
2357 del self
.initialParents
[branch
]
2361 tempBranch
= os
.path
.join(self
.tempBranchLocation
, "%d" % (change
))
2363 print "Creating temporary branch: " + tempBranch
2364 self
.commit(description
, filesForCommit
, tempBranch
, [branchPrefix
])
2365 self
.tempBranches
.append(tempBranch
)
2367 blob
= self
.searchParent(parent
, branch
, tempBranch
)
2369 self
.commit(description
, filesForCommit
, branch
, [branchPrefix
], blob
)
2372 print "Parent of %s not found. Committing into head of %s" % (branch
, parent
)
2373 self
.commit(description
, filesForCommit
, branch
, [branchPrefix
], parent
)
2375 files
= self
.extractFilesFromCommit(description
)
2376 self
.commit(description
, files
, self
.branch
, self
.depotPaths
,
2378 self
.initialParent
= ""
2380 print self
.gitError
.read()
2383 def importHeadRevision(self
, revision
):
2384 print "Doing initial import of %s from revision %s into %s" % (' '.join(self
.depotPaths
), revision
, self
.branch
)
2387 details
["user"] = "git perforce import user"
2388 details
["desc"] = ("Initial import of %s from the state at revision %s\n"
2389 % (' '.join(self
.depotPaths
), revision
))
2390 details
["change"] = revision
2394 fileArgs
= ["%s...%s" % (p
,revision
) for p
in self
.depotPaths
]
2396 for info
in p4CmdList(["files"] + fileArgs
):
2398 if 'code' in info
and info
['code'] == 'error':
2399 sys
.stderr
.write("p4 returned an error: %s\n"
2401 if info
['data'].find("must refer to client") >= 0:
2402 sys
.stderr
.write("This particular p4 error is misleading.\n")
2403 sys
.stderr
.write("Perhaps the depot path was misspelled.\n");
2404 sys
.stderr
.write("Depot path: %s\n" % " ".join(self
.depotPaths
))
2406 if 'p4ExitCode' in info
:
2407 sys
.stderr
.write("p4 exitcode: %s\n" % info
['p4ExitCode'])
2411 change
= int(info
["change"])
2412 if change
> newestRevision
:
2413 newestRevision
= change
2415 if info
["action"] in self
.delete_actions
:
2416 # don't increase the file cnt, otherwise details["depotFile123"] will have gaps!
2417 #fileCnt = fileCnt + 1
2420 for prop
in ["depotFile", "rev", "action", "type" ]:
2421 details
["%s%s" % (prop
, fileCnt
)] = info
[prop
]
2423 fileCnt
= fileCnt
+ 1
2425 details
["change"] = newestRevision
2427 # Use time from top-most change so that all git p4 clones of
2428 # the same p4 repo have the same commit SHA1s.
2429 res
= p4CmdList("describe -s %d" % newestRevision
)
2432 if r
.has_key('time'):
2433 newestTime
= int(r
['time'])
2434 if newestTime
is None:
2435 die("\"describe -s\" on newest change %d did not give a time")
2436 details
["time"] = newestTime
2438 self
.updateOptionDict(details
)
2440 self
.commit(details
, self
.extractFilesFromCommit(details
), self
.branch
, self
.depotPaths
)
2442 print "IO error with git fast-import. Is your git version recent enough?"
2443 print self
.gitError
.read()
2446 def run(self
, args
):
2447 self
.depotPaths
= []
2448 self
.changeRange
= ""
2449 self
.initialParent
= ""
2450 self
.previousDepotPaths
= []
2452 # map from branch depot path to parent branch
2453 self
.knownBranches
= {}
2454 self
.initialParents
= {}
2455 self
.hasOrigin
= originP4BranchesExist()
2456 if not self
.syncWithOrigin
:
2457 self
.hasOrigin
= False
2459 if self
.importIntoRemotes
:
2460 self
.refPrefix
= "refs/remotes/p4/"
2462 self
.refPrefix
= "refs/heads/p4/"
2464 if self
.syncWithOrigin
and self
.hasOrigin
:
2466 print "Syncing with origin first by calling git fetch origin"
2467 system("git fetch origin")
2469 if len(self
.branch
) == 0:
2470 self
.branch
= self
.refPrefix
+ "master"
2471 if gitBranchExists("refs/heads/p4") and self
.importIntoRemotes
:
2472 system("git update-ref %s refs/heads/p4" % self
.branch
)
2473 system("git branch -D p4");
2474 # create it /after/ importing, when master exists
2475 if not gitBranchExists(self
.refPrefix
+ "HEAD") and self
.importIntoRemotes
and gitBranchExists(self
.branch
):
2476 system("git symbolic-ref %sHEAD %s" % (self
.refPrefix
, self
.branch
))
2478 # accept either the command-line option, or the configuration variable
2479 if self
.useClientSpec
:
2480 # will use this after clone to set the variable
2481 self
.useClientSpec_from_options
= True
2483 if gitConfig("git-p4.useclientspec", "--bool") == "true":
2484 self
.useClientSpec
= True
2485 if self
.useClientSpec
:
2486 self
.clientSpecDirs
= getClientSpec()
2488 # TODO: should always look at previous commits,
2489 # merge with previous imports, if possible.
2492 createOrUpdateBranchesFromOrigin(self
.refPrefix
, self
.silent
)
2493 self
.listExistingP4GitBranches()
2495 if len(self
.p4BranchesInGit
) > 1:
2497 print "Importing from/into multiple branches"
2498 self
.detectBranches
= True
2501 print "branches: %s" % self
.p4BranchesInGit
2504 for branch
in self
.p4BranchesInGit
:
2505 logMsg
= extractLogMessageFromGitCommit(self
.refPrefix
+ branch
)
2507 settings
= extractSettingsGitLog(logMsg
)
2509 self
.readOptions(settings
)
2510 if (settings
.has_key('depot-paths')
2511 and settings
.has_key ('change')):
2512 change
= int(settings
['change']) + 1
2513 p4Change
= max(p4Change
, change
)
2515 depotPaths
= sorted(settings
['depot-paths'])
2516 if self
.previousDepotPaths
== []:
2517 self
.previousDepotPaths
= depotPaths
2520 for (prev
, cur
) in zip(self
.previousDepotPaths
, depotPaths
):
2521 prev_list
= prev
.split("/")
2522 cur_list
= cur
.split("/")
2523 for i
in range(0, min(len(cur_list
), len(prev_list
))):
2524 if cur_list
[i
] <> prev_list
[i
]:
2528 paths
.append ("/".join(cur_list
[:i
+ 1]))
2530 self
.previousDepotPaths
= paths
2533 self
.depotPaths
= sorted(self
.previousDepotPaths
)
2534 self
.changeRange
= "@%s,#head" % p4Change
2535 if not self
.detectBranches
:
2536 self
.initialParent
= parseRevision(self
.branch
)
2537 if not self
.silent
and not self
.detectBranches
:
2538 print "Performing incremental import into %s git branch" % self
.branch
2540 if not self
.branch
.startswith("refs/"):
2541 self
.branch
= "refs/heads/" + self
.branch
2543 if len(args
) == 0 and self
.depotPaths
:
2545 print "Depot paths: %s" % ' '.join(self
.depotPaths
)
2547 if self
.depotPaths
and self
.depotPaths
!= args
:
2548 print ("previous import used depot path %s and now %s was specified. "
2549 "This doesn't work!" % (' '.join (self
.depotPaths
),
2553 self
.depotPaths
= sorted(args
)
2558 # Make sure no revision specifiers are used when --changesfile
2560 bad_changesfile
= False
2561 if len(self
.changesFile
) > 0:
2562 for p
in self
.depotPaths
:
2563 if p
.find("@") >= 0 or p
.find("#") >= 0:
2564 bad_changesfile
= True
2567 die("Option --changesfile is incompatible with revision specifiers")
2570 for p
in self
.depotPaths
:
2571 if p
.find("@") != -1:
2572 atIdx
= p
.index("@")
2573 self
.changeRange
= p
[atIdx
:]
2574 if self
.changeRange
== "@all":
2575 self
.changeRange
= ""
2576 elif ',' not in self
.changeRange
:
2577 revision
= self
.changeRange
2578 self
.changeRange
= ""
2580 elif p
.find("#") != -1:
2581 hashIdx
= p
.index("#")
2582 revision
= p
[hashIdx
:]
2584 elif self
.previousDepotPaths
== []:
2585 # pay attention to changesfile, if given, else import
2586 # the entire p4 tree at the head revision
2587 if len(self
.changesFile
) == 0:
2590 p
= re
.sub ("\.\.\.$", "", p
)
2591 if not p
.endswith("/"):
2596 self
.depotPaths
= newPaths
2598 self
.loadUserMapFromCache()
2600 if self
.detectLabels
:
2603 if self
.detectBranches
:
2604 ## FIXME - what's a P4 projectName ?
2605 self
.projectName
= self
.guessProjectName()
2608 self
.getBranchMappingFromGitBranches()
2610 self
.getBranchMapping()
2612 print "p4-git branches: %s" % self
.p4BranchesInGit
2613 print "initial parents: %s" % self
.initialParents
2614 for b
in self
.p4BranchesInGit
:
2618 b
= b
[len(self
.projectName
):]
2619 self
.createdBranches
.add(b
)
2621 self
.tz
= "%+03d%02d" % (- time
.timezone
/ 3600, ((- time
.timezone
% 3600) / 60))
2623 importProcess
= subprocess
.Popen(["git", "fast-import"],
2624 stdin
=subprocess
.PIPE
, stdout
=subprocess
.PIPE
,
2625 stderr
=subprocess
.PIPE
);
2626 self
.gitOutput
= importProcess
.stdout
2627 self
.gitStream
= importProcess
.stdin
2628 self
.gitError
= importProcess
.stderr
2631 self
.importHeadRevision(revision
)
2635 if len(self
.changesFile
) > 0:
2636 output
= open(self
.changesFile
).readlines()
2639 changeSet
.add(int(line
))
2641 for change
in changeSet
:
2642 changes
.append(change
)
2646 # catch "git p4 sync" with no new branches, in a repo that
2647 # does not have any existing p4 branches
2648 if len(args
) == 0 and not self
.p4BranchesInGit
:
2649 die("No remote p4 branches. Perhaps you never did \"git p4 clone\" in here.");
2651 print "Getting p4 changes for %s...%s" % (', '.join(self
.depotPaths
),
2653 changes
= p4ChangesForPaths(self
.depotPaths
, self
.changeRange
)
2655 if len(self
.maxChanges
) > 0:
2656 changes
= changes
[:min(int(self
.maxChanges
), len(changes
))]
2658 if len(changes
) == 0:
2660 print "No changes to import!"
2662 if not self
.silent
and not self
.detectBranches
:
2663 print "Import destination: %s" % self
.branch
2665 self
.updatedBranches
= set()
2667 self
.importChanges(changes
)
2671 if len(self
.updatedBranches
) > 0:
2672 sys
.stdout
.write("Updated branches: ")
2673 for b
in self
.updatedBranches
:
2674 sys
.stdout
.write("%s " % b
)
2675 sys
.stdout
.write("\n")
2677 if gitConfig("git-p4.importLabels", "--bool") == "true":
2678 self
.importLabels
= true
2680 if self
.importLabels
:
2681 p4Labels
= getP4Labels(self
.depotPaths
)
2682 gitTags
= getGitTags()
2684 missingP4Labels
= p4Labels
- gitTags
2685 self
.importP4Labels(self
.gitStream
, missingP4Labels
)
2687 self
.gitStream
.close()
2688 if importProcess
.wait() != 0:
2689 die("fast-import failed: %s" % self
.gitError
.read())
2690 self
.gitOutput
.close()
2691 self
.gitError
.close()
2693 # Cleanup temporary branches created during import
2694 if self
.tempBranches
!= []:
2695 for branch
in self
.tempBranches
:
2696 read_pipe("git update-ref -d %s" % branch
)
2697 os
.rmdir(os
.path
.join(os
.environ
.get("GIT_DIR", ".git"), self
.tempBranchLocation
))
2701 class P4Rebase(Command
):
2703 Command
.__init
__(self
)
2705 optparse
.make_option("--import-labels", dest
="importLabels", action
="store_true"),
2707 self
.importLabels
= False
2708 self
.description
= ("Fetches the latest revision from perforce and "
2709 + "rebases the current work (branch) against it")
2711 def run(self
, args
):
2713 sync
.importLabels
= self
.importLabels
2716 return self
.rebase()
2719 if os
.system("git update-index --refresh") != 0:
2720 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.");
2721 if len(read_pipe("git diff-index HEAD --")) > 0:
2722 die("You have uncommited changes. Please commit them before rebasing or stash them away with git stash.");
2724 [upstream
, settings
] = findUpstreamBranchPoint()
2725 if len(upstream
) == 0:
2726 die("Cannot find upstream branchpoint for rebase")
2728 # the branchpoint may be p4/foo~3, so strip off the parent
2729 upstream
= re
.sub("~[0-9]+$", "", upstream
)
2731 print "Rebasing the current branch onto %s" % upstream
2732 oldHead
= read_pipe("git rev-parse HEAD").strip()
2733 system("git rebase %s" % upstream
)
2734 system("git diff-tree --stat --summary -M %s HEAD" % oldHead
)
2737 class P4Clone(P4Sync
):
2739 P4Sync
.__init
__(self
)
2740 self
.description
= "Creates a new git repository and imports from Perforce into it"
2741 self
.usage
= "usage: %prog [options] //depot/path[@revRange]"
2743 optparse
.make_option("--destination", dest
="cloneDestination",
2744 action
='store', default
=None,
2745 help="where to leave result of the clone"),
2746 optparse
.make_option("-/", dest
="cloneExclude",
2747 action
="append", type="string",
2748 help="exclude depot path"),
2749 optparse
.make_option("--bare", dest
="cloneBare",
2750 action
="store_true", default
=False),
2752 self
.cloneDestination
= None
2753 self
.needsGit
= False
2754 self
.cloneBare
= False
2756 # This is required for the "append" cloneExclude action
2757 def ensure_value(self
, attr
, value
):
2758 if not hasattr(self
, attr
) or getattr(self
, attr
) is None:
2759 setattr(self
, attr
, value
)
2760 return getattr(self
, attr
)
2762 def defaultDestination(self
, args
):
2763 ## TODO: use common prefix of args?
2765 depotDir
= re
.sub("(@[^@]*)$", "", depotPath
)
2766 depotDir
= re
.sub("(#[^#]*)$", "", depotDir
)
2767 depotDir
= re
.sub(r
"\.\.\.$", "", depotDir
)
2768 depotDir
= re
.sub(r
"/$", "", depotDir
)
2769 return os
.path
.split(depotDir
)[1]
2771 def run(self
, args
):
2775 if self
.keepRepoPath
and not self
.cloneDestination
:
2776 sys
.stderr
.write("Must specify destination for --keep-path\n")
2781 if not self
.cloneDestination
and len(depotPaths
) > 1:
2782 self
.cloneDestination
= depotPaths
[-1]
2783 depotPaths
= depotPaths
[:-1]
2785 self
.cloneExclude
= ["/"+p
for p
in self
.cloneExclude
]
2786 for p
in depotPaths
:
2787 if not p
.startswith("//"):
2790 if not self
.cloneDestination
:
2791 self
.cloneDestination
= self
.defaultDestination(args
)
2793 print "Importing from %s into %s" % (', '.join(depotPaths
), self
.cloneDestination
)
2795 if not os
.path
.exists(self
.cloneDestination
):
2796 os
.makedirs(self
.cloneDestination
)
2797 chdir(self
.cloneDestination
)
2799 init_cmd
= [ "git", "init" ]
2801 init_cmd
.append("--bare")
2802 subprocess
.check_call(init_cmd
)
2804 if not P4Sync
.run(self
, depotPaths
):
2806 if self
.branch
!= "master":
2807 if self
.importIntoRemotes
:
2808 masterbranch
= "refs/remotes/p4/master"
2810 masterbranch
= "refs/heads/p4/master"
2811 if gitBranchExists(masterbranch
):
2812 system("git branch master %s" % masterbranch
)
2813 if not self
.cloneBare
:
2814 system("git checkout -f")
2816 print "Could not detect main branch. No checkout/master branch created."
2818 # auto-set this variable if invoked with --use-client-spec
2819 if self
.useClientSpec_from_options
:
2820 system("git config --bool git-p4.useclientspec true")
2824 class P4Branches(Command
):
2826 Command
.__init
__(self
)
2828 self
.description
= ("Shows the git branches that hold imports and their "
2829 + "corresponding perforce depot paths")
2830 self
.verbose
= False
2832 def run(self
, args
):
2833 if originP4BranchesExist():
2834 createOrUpdateBranchesFromOrigin()
2836 cmdline
= "git rev-parse --symbolic "
2837 cmdline
+= " --remotes"
2839 for line
in read_pipe_lines(cmdline
):
2842 if not line
.startswith('p4/') or line
== "p4/HEAD":
2846 log
= extractLogMessageFromGitCommit("refs/remotes/%s" % branch
)
2847 settings
= extractSettingsGitLog(log
)
2849 print "%s <= %s (%s)" % (branch
, ",".join(settings
["depot-paths"]), settings
["change"])
2852 class HelpFormatter(optparse
.IndentedHelpFormatter
):
2854 optparse
.IndentedHelpFormatter
.__init
__(self
)
2856 def format_description(self
, description
):
2858 return description
+ "\n"
2862 def printUsage(commands
):
2863 print "usage: %s <command> [options]" % sys
.argv
[0]
2865 print "valid commands: %s" % ", ".join(commands
)
2867 print "Try %s <command> --help for command specific help." % sys
.argv
[0]
2872 "submit" : P4Submit
,
2873 "commit" : P4Submit
,
2875 "rebase" : P4Rebase
,
2877 "rollback" : P4RollBack
,
2878 "branches" : P4Branches
2883 if len(sys
.argv
[1:]) == 0:
2884 printUsage(commands
.keys())
2888 cmdName
= sys
.argv
[1]
2890 klass
= commands
[cmdName
]
2893 print "unknown command %s" % cmdName
2895 printUsage(commands
.keys())
2898 options
= cmd
.options
2899 cmd
.gitdir
= os
.environ
.get("GIT_DIR", None)
2903 options
.append(optparse
.make_option("--verbose", dest
="verbose", action
="store_true"))
2905 options
.append(optparse
.make_option("--git-dir", dest
="gitdir"))
2907 parser
= optparse
.OptionParser(cmd
.usage
.replace("%prog", "%prog " + cmdName
),
2909 description
= cmd
.description
,
2910 formatter
= HelpFormatter())
2912 (cmd
, args
) = parser
.parse_args(sys
.argv
[2:], cmd
);
2914 verbose
= cmd
.verbose
2916 if cmd
.gitdir
== None:
2917 cmd
.gitdir
= os
.path
.abspath(".git")
2918 if not isValidGitDir(cmd
.gitdir
):
2919 cmd
.gitdir
= read_pipe("git rev-parse --git-dir").strip()
2920 if os
.path
.exists(cmd
.gitdir
):
2921 cdup
= read_pipe("git rev-parse --show-cdup").strip()
2925 if not isValidGitDir(cmd
.gitdir
):
2926 if isValidGitDir(cmd
.gitdir
+ "/.git"):
2927 cmd
.gitdir
+= "/.git"
2929 die("fatal: cannot locate git repository at %s" % cmd
.gitdir
)
2931 os
.environ
["GIT_DIR"] = cmd
.gitdir
2933 if not cmd
.run(args
):
2938 if __name__
== '__main__':