From 0bbdd9f9adf12ad9082a4c49ae1c9a0778b00bb4 Mon Sep 17 00:00:00 2001
From: James Moger <james.moger@gitblit.com>
Date: Wed, 28 Nov 2012 17:39:12 -0500
Subject: [PATCH] Fixed focus for keystore password prompt

---
 src/com/gitblit/FileSettings.java |   64 +++++++++++++++++++++++++++++--
 1 files changed, 59 insertions(+), 5 deletions(-)

diff --git a/src/com/gitblit/FileSettings.java b/src/com/gitblit/FileSettings.java
index e213e80..be1f44f 100644
--- a/src/com/gitblit/FileSettings.java
+++ b/src/com/gitblit/FileSettings.java
@@ -18,10 +18,16 @@
 import java.io.File;
 import java.io.FileInputStream;
 import java.io.FileNotFoundException;
+import java.util.Map;
 import java.util.Properties;
 
+import com.gitblit.utils.FileUtils;
+
 /**
- * Reads GitBlit settings file.
+ * Dynamically loads and reloads a properties file by keeping track of the last
+ * modification date.
+ * 
+ * @author James Moger
  * 
  */
 public class FileSettings extends IStoredSettings {
@@ -30,26 +36,33 @@
 
 	private final Properties properties = new Properties();
 
-	private volatile long lastread;
+	private volatile long lastModified;
+	
+	private volatile boolean forceReload;
 
 	public FileSettings(String file) {
 		super(FileSettings.class);
 		this.propertiesFile = new File(file);
 	}
 
+	/**
+	 * Returns a properties object which contains the most recent contents of
+	 * the properties file.
+	 */
 	@Override
 	protected synchronized Properties read() {
-		if (propertiesFile.exists() && (propertiesFile.lastModified() > lastread)) {
+		if (propertiesFile.exists() && (forceReload || (propertiesFile.lastModified() > lastModified))) {
 			FileInputStream is = null;
 			try {
 				Properties props = new Properties();
 				is = new FileInputStream(propertiesFile);
 				props.load(is);
-				
+
 				// load properties after we have successfully read file
 				properties.clear();
 				properties.putAll(props);
-				lastread = propertiesFile.lastModified();
+				lastModified = propertiesFile.lastModified();
+				forceReload = false;
 			} catch (FileNotFoundException f) {
 				// IGNORE - won't happen because file.exists() check above
 			} catch (Throwable t) {
@@ -67,6 +80,47 @@
 		return properties;
 	}
 
+	/**
+	 * Updates the specified settings in the settings file.
+	 */
+	public synchronized boolean saveSettings(Map<String, String> settings) {
+		String content = FileUtils.readContent(propertiesFile, "\n");
+		for (Map.Entry<String, String> setting:settings.entrySet()) {
+			String regex = "(?m)^(" + regExEscape(setting.getKey()) + "\\s*+=\\s*+)"
+				    + "(?:[^\r\n\\\\]++|\\\\(?:\r?\n|\r|.))*+$";
+			String oldContent = content;
+			content = content.replaceAll(regex, setting.getKey() + " = " + setting.getValue());
+			if (content.equals(oldContent)) {
+				// did not replace value because it does not exist in the file
+				// append new setting to content (issue-85)
+				content += "\n" + setting.getKey() + " = " + setting.getValue();
+			}
+		}
+		FileUtils.writeContent(propertiesFile, content);
+		// manually set the forceReload flag because not all JVMs support real
+		// millisecond resolution of lastModified. (issue-55)		
+		forceReload = true;
+		return true;
+	}
+	
+	private String regExEscape(String input) {
+		return input.replace(".", "\\.");
+	}
+
+	/**
+	 * @return the last modification date of the properties file
+	 */
+	protected long lastModified() {
+		return lastModified;
+	}
+
+	/**
+	 * @return the state of the force reload flag
+	 */
+	protected boolean forceReload() {
+		return forceReload;
+	}
+
 	@Override
 	public String toString() {
 		return propertiesFile.getAbsolutePath();

--
Gitblit v1.9.1