1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44 package org.eclipse.jgit.pgm;
45
46 import java.io.File;
47 import java.net.InetSocketAddress;
48 import java.net.URISyntaxException;
49 import java.text.MessageFormat;
50 import java.util.ArrayList;
51 import java.util.List;
52 import java.util.concurrent.Executors;
53
54 import org.eclipse.jgit.internal.ketch.KetchLeader;
55 import org.eclipse.jgit.internal.ketch.KetchLeaderCache;
56 import org.eclipse.jgit.internal.ketch.KetchPreReceive;
57 import org.eclipse.jgit.internal.ketch.KetchSystem;
58 import org.eclipse.jgit.internal.ketch.KetchText;
59 import org.eclipse.jgit.lib.Repository;
60 import org.eclipse.jgit.pgm.internal.CLIText;
61 import org.eclipse.jgit.storage.file.FileBasedConfig;
62 import org.eclipse.jgit.storage.file.WindowCacheConfig;
63 import org.eclipse.jgit.storage.pack.PackConfig;
64 import org.eclipse.jgit.transport.DaemonClient;
65 import org.eclipse.jgit.transport.DaemonService;
66 import org.eclipse.jgit.transport.ReceivePack;
67 import org.eclipse.jgit.transport.resolver.FileResolver;
68 import org.eclipse.jgit.transport.resolver.ReceivePackFactory;
69 import org.eclipse.jgit.transport.resolver.ServiceNotEnabledException;
70 import org.eclipse.jgit.util.FS;
71 import org.kohsuke.args4j.Argument;
72 import org.kohsuke.args4j.Option;
73
74 @Command(common = true, usage = "usage_exportRepositoriesOverGit")
75 class Daemon extends TextBuiltin {
76 @Option(name = "--config-file", metaVar = "metaVar_configFile", usage = "usage_configFile")
77 File configFile;
78
79 @Option(name = "--port", metaVar = "metaVar_port", usage = "usage_portNumberToListenOn")
80 int port = org.eclipse.jgit.transport.Daemon.DEFAULT_PORT;
81
82 @Option(name = "--listen", metaVar = "metaVar_hostName", usage = "usage_hostnameOrIpToListenOn")
83 String host;
84
85 @Option(name = "--timeout", metaVar = "metaVar_seconds", usage = "usage_abortConnectionIfNoActivity")
86 int timeout = -1;
87
88 @Option(name = "--enable", metaVar = "metaVar_service", usage = "usage_enableTheServiceInAllRepositories")
89 List<String> enable = new ArrayList<>();
90
91 @Option(name = "--disable", metaVar = "metaVar_service", usage = "usage_disableTheServiceInAllRepositories")
92 List<String> disable = new ArrayList<>();
93
94 @Option(name = "--allow-override", metaVar = "metaVar_service", usage = "usage_configureTheServiceInDaemonServicename")
95 List<String> canOverride = new ArrayList<>();
96
97 @Option(name = "--forbid-override", metaVar = "metaVar_service", usage = "usage_configureTheServiceInDaemonServicename")
98 List<String> forbidOverride = new ArrayList<>();
99
100 @Option(name = "--export-all", usage = "usage_exportWithoutGitDaemonExportOk")
101 boolean exportAll;
102
103 @Option(name = "--ketch", metaVar = "metaVar_ketchServerType", usage = "usage_ketchServerType")
104 KetchServerType ketchServerType;
105
106 enum KetchServerType {
107 LEADER;
108 }
109
110 @Argument(required = true, metaVar = "metaVar_directory", usage = "usage_directoriesToExport")
111 List<File> directory = new ArrayList<>();
112
113
114 @Override
115 protected boolean requiresRepository() {
116 return false;
117 }
118
119
120 @Override
121 protected void run() throws Exception {
122 PackConfig packConfig = new PackConfig();
123
124 if (configFile != null) {
125 if (!configFile.exists()) {
126 throw die(MessageFormat.format(
127 CLIText.get().configFileNotFound,
128 configFile.getAbsolutePath()));
129 }
130
131 FileBasedConfig cfg = new FileBasedConfig(configFile, FS.DETECTED);
132 cfg.load();
133 new WindowCacheConfig().fromConfig(cfg).install();
134 packConfig.fromConfig(cfg);
135 }
136
137 int threads = packConfig.getThreads();
138 if (threads <= 0)
139 threads = Runtime.getRuntime().availableProcessors();
140 if (1 < threads)
141 packConfig.setExecutor(Executors.newFixedThreadPool(threads));
142
143 final FileResolver<DaemonClient> resolver = new FileResolver<>();
144 for (File f : directory) {
145 outw.println(MessageFormat.format(CLIText.get().exporting, f.getAbsolutePath()));
146 resolver.exportDirectory(f);
147 }
148 resolver.setExportAll(exportAll);
149
150 final org.eclipse.jgit.transport.Daemon d;
151 d = new org.eclipse.jgit.transport.Daemon(
152 host != null ? new InetSocketAddress(host, port)
153 : new InetSocketAddress(port));
154 d.setPackConfig(packConfig);
155 d.setRepositoryResolver(resolver);
156 if (0 <= timeout)
157 d.setTimeout(timeout);
158
159 for (String n : enable)
160 service(d, n).setEnabled(true);
161 for (String n : disable)
162 service(d, n).setEnabled(false);
163
164 for (String n : canOverride)
165 service(d, n).setOverridable(true);
166 for (String n : forbidOverride)
167 service(d, n).setOverridable(false);
168 if (ketchServerType == KetchServerType.LEADER) {
169 startKetchLeader(d);
170 }
171 d.start();
172 outw.println(MessageFormat.format(CLIText.get().listeningOn, d.getAddress()));
173 }
174
175 private static DaemonService service(
176 final org.eclipse.jgit.transport.Daemon d,
177 final String n) {
178 final DaemonService svc = d.getService(n);
179 if (svc == null)
180 throw die(MessageFormat.format(CLIText.get().serviceNotSupported, n));
181 return svc;
182 }
183
184 private void startKetchLeader(org.eclipse.jgit.transport.Daemon daemon) {
185 KetchSystem system = new KetchSystem();
186 final KetchLeaderCacheaderCache.html#KetchLeaderCache">KetchLeaderCache leaders = new KetchLeaderCache(system);
187 final ReceivePackFactory<DaemonClient> factory;
188
189 factory = daemon.getReceivePackFactory();
190 daemon.setReceivePackFactory((DaemonClient req, Repository repo) -> {
191 ReceivePack rp = factory.create(req, repo);
192 KetchLeader leader;
193 try {
194 leader = leaders.get(repo);
195 } catch (URISyntaxException err) {
196 throw new ServiceNotEnabledException(
197 KetchText.get().invalidFollowerUri, err);
198 }
199 rp.setPreReceiveHook(new KetchPreReceive(leader));
200 return rp;
201 });
202 }
203 }