blob: f4ba3ea088d5ee2d93acbbb39238c8d01a6e3808 [file]
import javax.swing.*;
import java.io.IOException;
import java.io.InputStream;
import java.net.URL;
import java.net.URLConnection;
import java.util.*;
import java.util.stream.Collectors;
public class KotlinPrePushHook {
private static final String BRANCH_DATA_URL = "https://raw.githubusercontent.com/Kotlin/kotlin-branch-status/master/status.txt";
private static final String MESSAGE_PREFIX = "MESSAGE.";
private static final List<String> KOTLIN_GIT_REPOS = Arrays.asList(
"git@github.com:JetBrains/kotlin.git",
"https://github.com/JetBrains/kotlin.git");
private static final String REF_PREFIX = "refs/heads/";
public static void main(String[] args) throws Exception {
if (args.length != 2) {
error("Usage: <remote refs> <target repository>, got " + Arrays.toString(args));
}
List<String> remoteRefs = Arrays
.stream(args[0].split(","))
.filter(ref -> ref.startsWith(REF_PREFIX))
.map(ref -> ref.substring(REF_PREFIX.length())).collect(Collectors.toList());
String targetRepo = args[1];
if (!KOTLIN_GIT_REPOS.contains(targetRepo)) {
return;
}
check(remoteRefs);
}
private static void check(List<String> remoteRefs) throws Exception {
Map<String, String> branchProperties = getProperties(new URL(BRANCH_DATA_URL));
if (branchProperties.isEmpty()) {
return;
}
Map<String, String> messages = new HashMap<>();
Map<String, String> branches = new HashMap<>();
for (Map.Entry<String, String> entry : branchProperties.entrySet()) {
String key = entry.getKey();
String value = entry.getValue();
if (key.startsWith(MESSAGE_PREFIX)) {
messages.put(key.substring(MESSAGE_PREFIX.length()), value);
} else {
branches.put(key, value);
}
}
for (Map.Entry<String, String> entry : branches.entrySet()) {
String branchName = entry.getKey();
if (remoteRefs.contains(branchName)) {
String message = messages.get(entry.getValue());
if (message == null) {
error("Invalid message key '" + entry.getValue() + "', expected one of these: " + messages.keySet());
}
//noinspection ConstantConditions
if (!showWarning(message, branchName)) {
error("Push aborted.");
}
}
}
}
private static void error(String message) {
//noinspection UseOfSystemOutOrSystemErr
System.err.println(message);
System.exit(1);
}
private static boolean showWarning(String reason, String targetBranch) {
String baseMessage = "You are about to commit to '" + targetBranch + "',\n" +
"and it is probably not the best idea.\n" +
"Please think twice before pressing 'Yes'.";
String confirmationMessage = "Do you still want to commit to '" + targetBranch + "'?";
String completeMessage = baseMessage + "\n\nReason: " + reason.replace("\\n", "\n") + "\n\n" + confirmationMessage;
int result = JOptionPane.showOptionDialog(
null, completeMessage, "Friendly warning",
JOptionPane.YES_NO_OPTION, JOptionPane.WARNING_MESSAGE, null,
new String[] { "Yes", "No" }, "No");
return result == JOptionPane.YES_OPTION;
}
private static Map<String, String> getProperties(URL url) throws Exception {
URLConnection connection = url.openConnection();
try (InputStream is = connection.getInputStream()) {
Properties properties = new Properties();
properties.load(is);
Map<String, String> propertyMap = new HashMap<>();
for (Map.Entry<Object, Object> entry : properties.entrySet()) {
if (entry.getKey() instanceof String && entry.getValue() instanceof String) {
propertyMap.put((String) entry.getKey(), (String) entry.getValue());
}
}
return propertyMap;
} catch (IOException e) {
System.err.println("Can't fetch " + BRANCH_DATA_URL + " (" + e.getMessage() + ").");
System.err.println("Pre-push hook won't work.");
return Collections.emptyMap();
}
}
}