Yan Zhulanow
121 lines
4.4 KiB
Java
121 lines
4.4 KiB
Java
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();
|
|
}
|
|
}
|
|
}
|