1 /* 2 * Copyright (c) 2019, Oracle and/or its affiliates. All rights reserved. 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 4 * 5 * This code is free software; you can redistribute it and/or modify it 6 * under the terms of the GNU General Public License version 2 only, as 7 * published by the Free Software Foundation. 8 * 9 * This code is distributed in the hope that it will be useful, but WITHOUT 10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 12 * version 2 for more details (a copy is included in the LICENSE file that 13 * accompanied this code). 14 * 15 * You should have received a copy of the GNU General Public License version 16 * 2 along with this work; if not, write to the Free Software Foundation, 17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 18 * 19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 20 * or visit www.oracle.com if you need additional information or have any 21 * questions. 22 */ 23 package org.openjdk.skara.cli; 24 25 import org.openjdk.skara.args.*; 26 import org.openjdk.skara.vcs.*; 27 28 import java.io.*; 29 import java.net.URI; 30 import java.nio.file.*; 31 import java.util.*; 32 import java.util.logging.*; 33 34 public class GitSync { 35 private static IOException die(String message) { 36 System.err.println(message); 37 System.exit(1); 38 return new IOException("will never reach here"); 39 } 40 41 private static int fetch() throws IOException, InterruptedException { 42 var pb = new ProcessBuilder("git", "fetch"); 43 pb.inheritIO(); 44 return pb.start().waitFor(); 45 } 46 47 private static int pull() throws IOException, InterruptedException { 48 var pb = new ProcessBuilder("git", "pull"); 49 pb.inheritIO(); 50 return pb.start().waitFor(); 51 } 52 53 public static void main(String[] args) throws IOException, InterruptedException { 54 var flags = List.of( 55 Option.shortcut("") 56 .fullname("from") 57 .describe("REMOTE") 58 .helptext("Fetch changes from this remote") 59 .optional(), 60 Option.shortcut("") 61 .fullname("to") 62 .describe("REMOTE") 63 .helptext("Push changes to this remote") 64 .optional(), 65 Option.shortcut("") 66 .fullname("branches") 67 .describe("BRANCHES") 68 .helptext("Comma separated list of branches to sync") 69 .optional(), 70 Switch.shortcut("") 71 .fullname("pull") 72 .helptext("Pull current branch from origin after successful sync") 73 .optional(), 74 Switch.shortcut("") 75 .fullname("fetch") 76 .helptext("Fetch current branch from origin after successful sync") 77 .optional(), 78 Switch.shortcut("m") 79 .fullname("mercurial") 80 .helptext("Force use of mercurial") 81 .optional(), 82 Switch.shortcut("") 83 .fullname("verbose") 84 .helptext("Turn on verbose output") 85 .optional(), 86 Switch.shortcut("") 87 .fullname("debug") 88 .helptext("Turn on debugging output") 89 .optional(), 90 Switch.shortcut("v") 91 .fullname("version") 92 .helptext("Print the version of this tool") 93 .optional() 94 ); 95 96 var parser = new ArgumentParser("git sync", flags); 97 var arguments = parser.parse(args); 98 99 if (arguments.contains("version")) { 100 System.out.println("git-sync version: " + Version.fromManifest().orElse("unknown")); 101 System.exit(0); 102 } 103 104 if (arguments.contains("verbose") || arguments.contains("debug")) { 105 var level = arguments.contains("debug") ? Level.FINER : Level.FINE; 106 Logging.setup(level); 107 } 108 109 var cwd = Paths.get("").toAbsolutePath(); 110 var repo = Repository.get(cwd).orElseThrow(() -> 111 die("error: no repository found at " + cwd.toString()) 112 ); 113 114 var remotes = repo.remotes(); 115 116 String upstream = null; 117 if (arguments.contains("from")) { 118 upstream = arguments.get("from").asString(); 119 } else { 120 var lines = repo.config("sync.from"); 121 if (lines.size() == 1 && remotes.contains(lines.get(0))) { 122 upstream = lines.get(0); 123 } else { 124 die("No remote provided to fetch from, please set the --from flag"); 125 } 126 } 127 var upstreamPullPath = remotes.contains(upstream) ? 128 Remote.toURI(repo.pullPath(upstream)) : URI.create(upstream); 129 130 String origin = null; 131 if (arguments.contains("to")) { 132 origin = arguments.get("to").asString(); 133 } else { 134 var lines = repo.config("sync.to"); 135 if (lines.size() == 1) { 136 if (!remotes.contains(lines.get(0))) { 137 die("The given remote to push to, " + lines.get(0) + ", does not exist"); 138 } else { 139 origin = lines.get(0); 140 } 141 } else { 142 origin = "origin"; 143 } 144 } 145 var originPushPath = Remote.toURI(repo.pushPath(origin)); 146 147 var branches = new HashSet<String>(); 148 if (arguments.contains("branches")) { 149 var requested = arguments.get("branches").asString().split(","); 150 for (var branch : requested) { 151 branches.add(branch.trim()); 152 } 153 } 154 155 for (var branch : repo.remoteBranches(upstream)) { 156 var name = branch.name(); 157 if (!branches.isEmpty() && !branches.contains(name)) { 158 System.out.println("Skipping branch " + name); 159 continue; 160 } 161 System.out.print("Syncing " + upstream + "/" + name + " to " + origin + "/" + name + "... "); 162 System.out.flush(); 163 var fetchHead = repo.fetch(upstreamPullPath, branch.hash().hex()); 164 repo.push(fetchHead, originPushPath, name); 165 System.out.println("done"); 166 } 167 168 if (arguments.contains("fetch")) { 169 int err = fetch(); 170 if (err != 0) { 171 System.exit(err); 172 } 173 } 174 175 if (arguments.contains("pull")) { 176 int err = pull(); 177 if (err != 0) { 178 System.exit(err); 179 } 180 } 181 } 182 }