3 # hashlib is only available in python >= 2.5
12 sys
.path
.insert(0, os
.getenv("GITPYTHONLIB","."))
14 from git_remote_helpers
.util
import die
, debug
, warn
15 from git_remote_helpers
.git
.repo
import GitRepo
16 from git_remote_helpers
.git
.exporter
import GitExporter
17 from git_remote_helpers
.git
.importer
import GitImporter
18 from git_remote_helpers
.git
.non_local
import NonLocalGit
20 def get_repo(alias
, url
):
21 """Returns a git repository object initialized for usage.
29 hasher
.update(repo
.path
)
30 repo
.hash = hasher
.hexdigest()
32 repo
.get_base_path
= lambda base
: os
.path
.join(
33 base
, 'info', 'fast-import', repo
.hash)
35 prefix
= 'refs/testgit/%s/' % alias
36 debug("prefix: '%s'", prefix
)
38 repo
.gitdir
= os
.environ
["GIT_DIR"]
42 repo
.exporter
= GitExporter(repo
)
43 repo
.importer
= GitImporter(repo
)
44 repo
.non_local
= NonLocalGit(repo
)
49 def local_repo(repo
, path
):
50 """Returns a git repository object initalized for usage.
55 local
.non_local
= None
56 local
.gitdir
= repo
.gitdir
57 local
.alias
= repo
.alias
58 local
.prefix
= repo
.prefix
59 local
.hash = repo
.hash
60 local
.get_base_path
= repo
.get_base_path
61 local
.exporter
= GitExporter(local
)
62 local
.importer
= GitImporter(local
)
67 def do_capabilities(repo
, args
):
68 """Prints the supported capabilities.
73 print "refspec refs/heads/*:%s*" % repo
.prefix
75 print # end capabilities
78 def do_list(repo
, args
):
79 """Lists all known references.
81 Bug: This will always set the remote head to master for non-local
82 repositories, since we have no way of determining what the remote
83 head is at clone time.
87 debug("? refs/heads/%s", ref
)
88 print "? refs/heads/%s" % ref
91 debug("@refs/heads/%s HEAD" % repo
.head
)
92 print "@refs/heads/%s HEAD" % repo
.head
94 debug("@refs/heads/master HEAD")
95 print "@refs/heads/master HEAD"
100 def update_local_repo(repo
):
101 """Updates (or clones) a local repo.
107 path
= repo
.non_local
.clone(repo
.gitdir
)
108 repo
.non_local
.update(repo
.gitdir
)
109 repo
= local_repo(repo
, path
)
113 def do_import(repo
, args
):
114 """Exports a fast-import stream from testgit for git to import.
118 die("Import needs exactly one ref")
121 die("Need gitdir to import")
123 repo
= update_local_repo(repo
)
124 repo
.exporter
.export_repo(repo
.gitdir
, args
)
129 def do_export(repo
, args
):
130 """Imports a fast-import stream from git to testgit.
134 die("Need gitdir to export")
136 dirname
= repo
.get_base_path(repo
.gitdir
)
138 if not os
.path
.exists(dirname
):
141 path
= os
.path
.join(dirname
, 'testgit.marks')
143 if os
.path
.exists(path
):
149 update_local_repo(repo
)
150 repo
.importer
.do_import(repo
.gitdir
)
153 repo
.non_local
.push(repo
.gitdir
)
157 'capabilities': do_capabilities
,
165 """Cleans up the url.
168 if value
.startswith('testgit::'):
174 def read_one_line(repo
):
175 """Reads and processes one command.
178 line
= sys
.stdin
.readline()
183 warn("Unexpected EOF")
186 cmdline
= cmdline
.strip().split()
188 # Blank line means we're about to quit
192 debug("Got command '%s' with args '%s'", cmd
, ' '.join(cmdline
))
194 if cmd
not in COMMANDS
:
195 die("Unknown command, %s", cmd
)
205 """Starts a new remote helper for the specified repository.
209 die("Expecting exactly three arguments.")
212 if os
.getenv("GIT_DEBUG_TESTGIT"):
213 import git_remote_helpers
.util
214 git_remote_helpers
.util
.DEBUG
= True
216 alias
= sanitize(args
[1])
217 url
= sanitize(args
[2])
219 if not alias
.isalnum():
220 warn("non-alnum alias '%s'", alias
)
226 repo
= get_repo(alias
, url
)
228 debug("Got arguments %s", args
[1:])
233 more
= read_one_line(repo
)
235 if __name__
== '__main__':
236 sys
.exit(main(sys
.argv
))