From cd8f9d14c972764e5e14ea43811744652a11dcd2 Mon Sep 17 00:00:00 2001
From: James Moger <james.moger@gitblit.com>
Date: Fri, 28 Oct 2011 08:09:55 -0400
Subject: [PATCH] Documentation

---
 src/com/gitblit/client/GitblitManager.java |  247 ++++++++++++++++++++++++++++++++-----------------
 1 files changed, 161 insertions(+), 86 deletions(-)

diff --git a/src/com/gitblit/client/GitblitManager.java b/src/com/gitblit/client/GitblitManager.java
index 47ff6ea..f16616a 100644
--- a/src/com/gitblit/client/GitblitManager.java
+++ b/src/com/gitblit/client/GitblitManager.java
@@ -16,10 +16,9 @@
 package com.gitblit.client;
 
 import java.awt.BorderLayout;
+import java.awt.Cursor;
 import java.awt.Dimension;
 import java.awt.EventQueue;
-import java.awt.Font;
-import java.awt.GridLayout;
 import java.awt.Point;
 import java.awt.event.ActionEvent;
 import java.awt.event.ActionListener;
@@ -28,23 +27,30 @@
 import java.awt.event.WindowEvent;
 import java.io.File;
 import java.io.IOException;
+import java.net.ConnectException;
 import java.text.MessageFormat;
+import java.text.SimpleDateFormat;
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.Comparator;
+import java.util.Date;
 import java.util.LinkedHashMap;
+import java.util.List;
+import java.util.Locale;
 import java.util.Map;
 import java.util.Set;
+import java.util.TimeZone;
 
 import javax.swing.ImageIcon;
 import javax.swing.JFrame;
-import javax.swing.JLabel;
 import javax.swing.JMenu;
 import javax.swing.JMenuBar;
 import javax.swing.JMenuItem;
 import javax.swing.JOptionPane;
 import javax.swing.JPanel;
-import javax.swing.JPasswordField;
 import javax.swing.JTabbedPane;
-import javax.swing.JTextField;
 import javax.swing.KeyStroke;
+import javax.swing.SwingWorker;
 import javax.swing.UIManager;
 
 import org.eclipse.jgit.errors.ConfigInvalidException;
@@ -54,6 +60,7 @@
 import org.eclipse.jgit.util.FS;
 
 import com.gitblit.Constants;
+import com.gitblit.GitBlitException.ForbiddenException;
 import com.gitblit.utils.StringUtils;
 
 /**
@@ -62,19 +69,21 @@
  * @author James Moger
  * 
  */
-public class GitblitManager extends JFrame {
+public class GitblitManager extends JFrame implements RegistrationsDialog.RegistrationListener {
 
 	private static final long serialVersionUID = 1L;
+	private final SimpleDateFormat dateFormat;
 	private JTabbedPane serverTabs;
 	private File configFile = new File(System.getProperty("user.home"), ".gitblit/config");
-	private GitblitRegistration localhost = new GitblitRegistration("default",
-			"https://localhost:8443", "admin", "admin".toCharArray());
 
 	private Map<String, GitblitRegistration> registrations = new LinkedHashMap<String, GitblitRegistration>();
 	private JMenu recentMenu;
+	private int maxRecentCount = 5;
 
 	private GitblitManager() {
 		super();
+		dateFormat = new SimpleDateFormat("yyyy-MM-dd'T'HH:mm:ss'Z'", Locale.US);
+		dateFormat.setTimeZone(TimeZone.getTimeZone("UTC"));
 	}
 
 	private void initialize() {
@@ -86,6 +95,11 @@
 			@Override
 			public void windowClosing(WindowEvent event) {
 				saveSizeAndPosition();
+			}
+
+			@Override
+			public void windowOpened(WindowEvent event) {
+				manageRegistrations();
 			}
 		});
 
@@ -142,45 +156,23 @@
 		}
 	}
 
-	public void setVisible(boolean value) {
-		if (value) {
-			if (registrations.size() == 0) {
-				// default prompt
-				loginPrompt(localhost);
-			} else if (registrations.size() == 1) {
-				// single registration prompt
-				GitblitRegistration reg = registrations.values().iterator().next();
-				loginPrompt(reg);
-			}
-			super.setVisible(value);
-		}
-	}
-
 	private JMenuBar setupMenu() {
 		JMenuBar menuBar = new JMenuBar();
 		JMenu serversMenu = new JMenu(Translation.get("gb.servers"));
 		menuBar.add(serversMenu);
 		recentMenu = new JMenu(Translation.get("gb.recent"));
 		serversMenu.add(recentMenu);
-		JMenuItem login = new JMenuItem(Translation.get("gb.login") + "...");
-		login.setAccelerator(KeyStroke.getKeyStroke(KeyEvent.VK_L, KeyEvent.CTRL_DOWN_MASK, false));
-		login.addActionListener(new ActionListener() {
+
+		JMenuItem manage = new JMenuItem(Translation.get("gb.manage") + "...");
+		manage.setAccelerator(KeyStroke.getKeyStroke(KeyEvent.VK_M, KeyEvent.CTRL_DOWN_MASK, false));
+		manage.addActionListener(new ActionListener() {
 			public void actionPerformed(ActionEvent event) {
-				loginPrompt(localhost);
+				manageRegistrations();
 			}
 		});
-		serversMenu.add(login);
-		return menuBar;
-	}
+		serversMenu.add(manage);
 
-	private JPanel newLabelPanel(String text, JTextField field) {
-		JLabel label = new JLabel(text);
-		label.setFont(label.getFont().deriveFont(Font.BOLD));
-		label.setPreferredSize(new Dimension(75, 10));
-		JPanel jpanel = new JPanel(new BorderLayout());
-		jpanel.add(label, BorderLayout.WEST);
-		jpanel.add(field, BorderLayout.CENTER);
-		return jpanel;
+		return menuBar;
 	}
 
 	private JPanel getCenterPanel() {
@@ -192,61 +184,99 @@
 		return panel;
 	}
 
-	private boolean loginPrompt(GitblitRegistration reg) {
-		JTextField urlField = new JTextField(reg.url, 30);
-		JTextField nameField = new JTextField(reg.name);
-		JTextField accountField = new JTextField(reg.account);
-		JPasswordField passwordField = new JPasswordField(new String(reg.password));
-
-		JPanel panel = new JPanel(new GridLayout(0, 1, 5, 5));
-		panel.add(newLabelPanel(Translation.get("gb.name"), nameField));
-		panel.add(newLabelPanel(Translation.get("gb.url"), urlField));
-		panel.add(newLabelPanel(Translation.get("gb.username"), accountField));
-		panel.add(newLabelPanel(Translation.get("gb.password"), passwordField));
-
-		int result = JOptionPane.showConfirmDialog(GitblitManager.this, panel,
-				Translation.get("gb.login"), JOptionPane.OK_CANCEL_OPTION);
-		if (result != JOptionPane.OK_OPTION) {
-			return false;
-		}
-		String url = urlField.getText();
-		if (StringUtils.isEmpty(url)) {
-			return false;
-		}
-		reg = new GitblitRegistration(nameField.getText(), url, accountField.getText(),
-				passwordField.getPassword());
-		boolean success = login(reg);
-		registrations.put(reg.name, reg);
-		rebuildRecentMenu();
-		return success;
+	private void manageRegistrations() {
+		RegistrationsDialog dialog = new RegistrationsDialog(new ArrayList<GitblitRegistration>(
+				registrations.values()), this);
+		dialog.setLocationRelativeTo(GitblitManager.this);
+		dialog.setVisible(true);
 	}
 
-	private boolean login(GitblitRegistration reg) {
-		try {
-			GitblitPanel panel = new GitblitPanel(reg);
-			panel.login();
-			serverTabs.addTab(reg.name, panel);
-			int idx = serverTabs.getTabCount() - 1;
-			serverTabs.setSelectedIndex(idx);
-			serverTabs.setTabComponentAt(idx, new ClosableTabComponent(reg.name, null, serverTabs,
-					panel));
-			saveRegistration(reg);
-			return true;
-		} catch (IOException e) {
-			JOptionPane.showMessageDialog(GitblitManager.this, e.getMessage(),
-					Translation.get("gb.error"), JOptionPane.ERROR_MESSAGE);
+	@Override
+	public void login(GitblitRegistration reg) {
+		if (!reg.savePassword && (reg.password == null || reg.password.length == 0)) {
+			// prompt for password
+			EditRegistrationDialog dialog = new EditRegistrationDialog(this, reg, true);
+			dialog.setLocationRelativeTo(GitblitManager.this);
+			dialog.setVisible(true);
+			reg = dialog.getRegistration();
+			if (reg == null) {
+				// user canceled
+				return;
+			}
 		}
-		return false;
+		
+		// login
+		setCursor(Cursor.getPredefinedCursor(Cursor.WAIT_CURSOR));
+		final GitblitRegistration registration = reg;
+		final GitblitPanel panel = new GitblitPanel(registration);
+		SwingWorker<Boolean, Void> worker = new SwingWorker<Boolean, Void>() {
+
+			@Override
+			protected Boolean doInBackground() throws IOException {
+				panel.login();
+				return true;
+			}
+
+			@Override
+			protected void done() {
+				try {
+					boolean success = get();
+					serverTabs.addTab(registration.name, panel);
+					int idx = serverTabs.getTabCount() - 1;
+					serverTabs.setSelectedIndex(idx);
+					serverTabs.setTabComponentAt(idx, new ClosableTabComponent(registration.name,
+							null, serverTabs, panel));
+					registration.lastLogin = new Date();
+					saveRegistration(registration.name, registration);
+					registrations.put(registration.name, registration);
+					rebuildRecentMenu();
+					if (!registration.savePassword) {
+						// clear password
+						registration.password = null;
+					}
+				} catch (Throwable t) {
+					Throwable cause = t.getCause();
+					if (cause instanceof ConnectException) {
+						JOptionPane.showMessageDialog(GitblitManager.this, cause.getMessage(),
+								Translation.get("gb.error"), JOptionPane.ERROR_MESSAGE);
+					} else if (cause instanceof ForbiddenException) {
+						JOptionPane
+								.showMessageDialog(
+										GitblitManager.this,
+										"This Gitblit server does not allow RPC Management or Administration",
+										Translation.get("gb.error"), JOptionPane.ERROR_MESSAGE);
+					} else {
+						Utils.showException(GitblitManager.this, t);
+					}
+				} finally {
+					setCursor(Cursor.getPredefinedCursor(Cursor.DEFAULT_CURSOR));
+				}
+			}
+		};
+		worker.execute();
 	}
 
 	private void rebuildRecentMenu() {
 		recentMenu.removeAll();
 		ImageIcon icon = new ImageIcon(getClass().getResource("/gitblt-favicon.png"));
-		for (final GitblitRegistration reg : registrations.values()) {
+		List<GitblitRegistration> list = new ArrayList<GitblitRegistration>(registrations.values());
+		Collections.sort(list, new Comparator<GitblitRegistration>() {
+			@Override
+			public int compare(GitblitRegistration o1, GitblitRegistration o2) {
+				return o2.lastLogin.compareTo(o1.lastLogin);
+			}
+		});
+		if (list.size() > maxRecentCount) {
+			list = list.subList(0, maxRecentCount);
+		}
+		for (int i = 0; i < list.size(); i++) {
+			final GitblitRegistration reg = list.get(i);
 			JMenuItem item = new JMenuItem(reg.name, icon);
+			item.setAccelerator(KeyStroke.getKeyStroke(KeyEvent.VK_1 + i, KeyEvent.CTRL_DOWN_MASK,
+					false));
 			item.addActionListener(new ActionListener() {
 				public void actionPerformed(ActionEvent e) {
-					loginPrompt(reg);
+					login(reg);
 				}
 			});
 			recentMenu.add(item);
@@ -258,11 +288,22 @@
 			StoredConfig config = getConfig();
 			Set<String> servers = config.getSubsections("servers");
 			for (String server : servers) {
+				Date lastLogin = new Date(0);
+				String date = config.getString("servers", server, "lastLogin");
+				if (!StringUtils.isEmpty(date)) {
+					lastLogin = dateFormat.parse(date);
+				}
 				String url = config.getString("servers", server, "url");
 				String account = config.getString("servers", server, "account");
-				char[] password = new String(Base64.decode(config.getString("servers", server,
-						"password"))).toCharArray();
+				char[] password;
+				String pw = config.getString("servers", server, "password");
+				if (StringUtils.isEmpty(pw)) {
+					password = new char[0];
+				} else {
+					password = new String(Base64.decode(pw)).toCharArray();
+				}
 				GitblitRegistration reg = new GitblitRegistration(server, url, account, password);
+				reg.lastLogin = lastLogin;
 				registrations.put(reg.name, reg);
 			}
 		} catch (Throwable t) {
@@ -270,17 +311,51 @@
 		}
 	}
 
-	private void saveRegistration(GitblitRegistration reg) {
+	@Override
+	public boolean saveRegistration(String name, GitblitRegistration reg) {
 		try {
 			StoredConfig config = getConfig();
+			if (!StringUtils.isEmpty(name) && !name.equals(reg.name)) {
+				// delete old registration
+				registrations.remove(name);
+				config.unsetSection("servers", name);
+			}
+
+			// update registration
 			config.setString("servers", reg.name, "url", reg.url);
 			config.setString("servers", reg.name, "account", reg.account);
-			config.setString("servers", reg.name, "password",
-					Base64.encodeBytes(new String(reg.password).getBytes("UTF-8")));
+			if (reg.savePassword) {
+				config.setString("servers", reg.name, "password",
+						Base64.encodeBytes(new String(reg.password).getBytes("UTF-8")));
+			} else {
+				config.setString("servers", reg.name, "password", "");
+			}
+			if (reg.lastLogin != null) {
+				config.setString("servers", reg.name, "lastLogin", dateFormat.format(reg.lastLogin));
+			}
 			config.save();
+			return true;
 		} catch (Throwable t) {
 			Utils.showException(GitblitManager.this, t);
 		}
+		return false;
+	}
+
+	@Override
+	public boolean deleteRegistrations(List<GitblitRegistration> list) {
+		boolean success = false;
+		try {
+			StoredConfig config = getConfig();
+			for (GitblitRegistration reg : list) {
+				config.unsetSection("servers", reg.name);
+				registrations.remove(reg.name);
+			}
+			config.save();
+			success = true;
+		} catch (Throwable t) {
+			Utils.showException(GitblitManager.this, t);
+		}
+		return success;
 	}
 
 	private StoredConfig getConfig() throws IOException, ConfigInvalidException {

--
Gitblit v1.9.1