From f5fa7958b07d0e585542b712b13f7ad9b47a8be9 Mon Sep 17 00:00:00 2001 From: James Moger <james.moger@gitblit.com> Date: Fri, 23 Dec 2011 08:28:15 -0500 Subject: [PATCH] Improved logging of GitblitUserService --- src/com/gitblit/GitBlit.java | 84 ++++++++++++++++++++++++++++++++++++----- 1 files changed, 73 insertions(+), 11 deletions(-) diff --git a/src/com/gitblit/GitBlit.java b/src/com/gitblit/GitBlit.java index 2719734..c818dd5 100644 --- a/src/com/gitblit/GitBlit.java +++ b/src/com/gitblit/GitBlit.java @@ -25,11 +25,14 @@ import java.text.MessageFormat; import java.util.ArrayList; import java.util.Arrays; +import java.util.Collection; import java.util.Collections; import java.util.HashMap; +import java.util.HashSet; import java.util.List; import java.util.Map; import java.util.Map.Entry; +import java.util.Set; import java.util.concurrent.ConcurrentHashMap; import java.util.concurrent.Executors; import java.util.concurrent.ScheduledExecutorService; @@ -293,6 +296,16 @@ */ public static File getProposalsFolder() { return getFileOrFolder(Keys.federation.proposalsFolder, "proposals"); + } + + /** + * Returns the path of the Groovy folder. This method checks to see if + * Gitblit is running on a cloud service and may return an adjusted path. + * + * @return the Groovy scripts folder path + */ + public static File getGroovyScriptsFolder() { + return getFileOrFolder(Keys.groovy.scriptsFolder, "groovy"); } /** @@ -731,6 +744,8 @@ "gitblit", null, "preReceiveScript"))); model.postReceiveScripts = new ArrayList<String>(Arrays.asList(config.getStringList( "gitblit", null, "postReceiveScript"))); + model.mailingLists = new ArrayList<String>(Arrays.asList(config.getStringList( + "gitblit", null, "mailingList"))); } r.close(); return model; @@ -950,8 +965,16 @@ config.setString("gitblit", null, "federationStrategy", repository.federationStrategy.name()); config.setBoolean("gitblit", null, "isFederated", repository.isFederated); - config.setStringList("gitblit", null, "preReceiveScript", repository.preReceiveScripts); - config.setStringList("gitblit", null, "postReceiveScript", repository.postReceiveScripts); + if (repository.preReceiveScripts != null) { + config.setStringList("gitblit", null, "preReceiveScript", repository.preReceiveScripts); + } + if (repository.postReceiveScripts != null) { + config.setStringList("gitblit", null, "postReceiveScript", + repository.postReceiveScripts); + } + if (repository.mailingLists != null) { + config.setStringList("gitblit", null, "mailingList", repository.mailingLists); + } try { config.save(); } catch (IOException e) { @@ -1416,12 +1439,54 @@ } /** + * Returns the list of all available Groovy push hook scripts that are not + * already specified globally for all repositories. Script files must have + * .groovy extension + * + * @return list of available hook scripts + */ + public List<String> getAvailableScripts() { + File groovyFolder = getGroovyScriptsFolder(); + File[] files = groovyFolder.listFiles(new FileFilter() { + @Override + public boolean accept(File pathname) { + return pathname.isFile() && pathname.getName().endsWith(".groovy"); + } + }); + + Set<String> globals = new HashSet<String>(); + String[] keys = { Keys.groovy.preReceiveScripts, Keys.groovy.postReceiveScripts }; + for (String key : keys) { + for (String script : getStrings(key)) { + if (script.endsWith(".groovy")) { + globals.add(script.substring(0, script.lastIndexOf('.'))); + } else { + globals.add(script); + } + } + } + + // create list of available scripts by excluding scripts that are + // globally specified + List<String> scripts = new ArrayList<String>(); + if (files != null) { + for (File file : files) { + String script = file.getName().substring(0, file.getName().lastIndexOf('.')); + if (!globals.contains(script)) { + scripts.add(script); + } + } + } + return scripts; + } + + /** * Notify the administrators by email. * * @param subject * @param message */ - public void sendEmailToAdministrators(String subject, String message) { + public void sendMailToAdministrators(String subject, String message) { try { Message mail = mailExecutor.createMessageForAdministrators(); if (mail != null) { @@ -1441,8 +1506,8 @@ * @param message * @param toAddresses */ - public void sendEmail(String subject, String message, ArrayList<String> toAddresses) { - this.sendEmail(subject, message, toAddresses.toArray(new String[0])); + public void sendMail(String subject, String message, Collection<String> toAddresses) { + this.sendMail(subject, message, toAddresses.toArray(new String[0])); } /** @@ -1452,7 +1517,7 @@ * @param message * @param toAddresses */ - public void sendEmail(String subject, String message, String... toAddresses) { + public void sendMail(String subject, String message, String... toAddresses) { try { Message mail = mailExecutor.createMessage(toAddresses); if (mail != null) { @@ -1478,6 +1543,7 @@ setting.currentValue = settings.getString(key, ""); } } + settingsModel.pushScripts = getAvailableScripts(); return settingsModel; } @@ -1569,11 +1635,7 @@ loginService = (IUserService) realmClass.newInstance(); } } catch (Throwable t) { - loginService = new UserServiceWrapper() { - @Override - public void setupService(IStoredSettings settings) { - } - }; + loginService = new GitblitUserService(); } setUserService(loginService); mailExecutor = new MailExecutor(settings); -- Gitblit v1.9.1