/*
* Copyright (c) 2010, SQL Power Group Inc.
*
* This file is part of Power*Architect.
*
* Power*Architect is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 3 of the License, or
* (at your option) any later version.
*
* Power*Architect is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
package ca.sqlpower.enterprise;
import java.io.IOException;
import java.net.MalformedURLException;
import java.net.URISyntaxException;
import java.net.URL;
import java.util.HashMap;
import java.util.Map;
import javax.swing.SwingUtilities;
import javax.swing.event.HyperlinkEvent;
import javax.swing.event.HyperlinkListener;
import org.apache.http.auth.AuthScope;
import org.apache.http.auth.UsernamePasswordCredentials;
import org.apache.http.client.CookieStore;
import org.apache.http.client.methods.HttpOptions;
import org.apache.http.client.methods.HttpUriRequest;
import org.apache.http.impl.client.BasicResponseHandler;
import org.apache.http.impl.client.DefaultHttpClient;
import org.apache.http.params.BasicHttpParams;
import org.apache.http.params.HttpConnectionParams;
import org.apache.http.params.HttpParams;
import org.json.JSONException;
import org.json.JSONObject;
import ca.sqlpower.enterprise.client.SPServerInfo;
import ca.sqlpower.enterprise.client.ServerProperties;
import ca.sqlpower.util.BrowserUtil;
import ca.sqlpower.util.HTMLUserPrompter;
import ca.sqlpower.util.UserPrompter.UserPromptOptions;
import ca.sqlpower.util.UserPrompter.UserPromptResponse;
import ca.sqlpower.util.Version;
public abstract class ServerInfoProvider {
public static final String defaultWatermarkMessage = "<html>This version of SQL Power Architect is for EVALUATION PURPOSES ONLY. To obtain a full Production License, please visit <a href=\"http://www.sqlpower.ca/architect-e\">www.sqlpower.ca/architect-e</a></html>";
private static Map<String,Version> version = new HashMap<String, Version>();
private static Map<String,Boolean> licenses = new HashMap<String, Boolean>();
private static Map<String,String> watermarkMessages = new HashMap<String, String>();
public static Version getServerVersion(
String host,
String port,
String path,
String username,
String password,
CookieStore cookieStore) throws MalformedURLException,IOException
{
init(toURL(host, port, path), username, password, cookieStore);
return version.get(generateServerKey(host, port, path, username, password));
}
public static boolean isServerLicensed(SPServerInfo infos, CookieStore cookieStore)
throws MalformedURLException,IOException
{
return isServerLicensed(
infos.getServerAddress(),
String.valueOf(infos.getPort()),
infos.getPath(),
infos.getUsername(),
infos.getPassword(), cookieStore);
}
public static boolean isServerLicensed(
String host,
String port,
String path,
String username,
String password,
CookieStore cookieStore) throws MalformedURLException,IOException
{
init(toURL(host, port, path), username, password, cookieStore);
return licenses.get(generateServerKey(host, port, path, username, password));
}
private static URL toURL(
String host,
String port,
String path) throws MalformedURLException
{
// Build the base URL
StringBuilder sb = new StringBuilder();
sb.append("http://");
sb.append(host);
sb.append(":");
sb.append(port);
sb.append(path);
sb.append(path.endsWith("/")?"":"/");
sb.append(ClientSideSessionUtils.REST_TAG);
sb.append("/serverinfo");
// Spawn a connection object
return new URL(sb.toString());
}
private static void init(URL url, String username, String password, CookieStore cookieStore) throws IOException {
if (version.containsKey(generateServerKey(url, username, password))) return;
try {
HttpParams params = new BasicHttpParams();
HttpConnectionParams.setConnectionTimeout(params, 2000);
DefaultHttpClient httpClient = new DefaultHttpClient(params);
httpClient.setCookieStore(cookieStore);
httpClient.getCredentialsProvider().setCredentials(
new AuthScope(url.getHost(), AuthScope.ANY_PORT),
new UsernamePasswordCredentials(username, password));
HttpUriRequest request = new HttpOptions(url.toURI());
String responseBody = httpClient.execute(request, new BasicResponseHandler());
// Decode the message
String serverVersion;
Boolean licensedServer;
final String watermarkMessage;
try {
JSONObject jsonObject = new JSONObject(responseBody);
serverVersion = jsonObject.getString(ServerProperties.SERVER_VERSION.toString());
licensedServer = jsonObject.getBoolean(ServerProperties.SERVER_LICENSED.toString());
watermarkMessage = jsonObject.getString(ServerProperties.SERVER_WATERMARK_MESSAGE.toString());
} catch (JSONException e) {
throw new IOException(e.getMessage());
}
// Save found values
version.put(generateServerKey(url, username, password), new Version(serverVersion));
licenses.put(generateServerKey(url, username, password), licensedServer);
watermarkMessages.put(generateServerKey(url, username, password), watermarkMessage);
// Notify the user if the server is not licensed.
if (!licensedServer || (watermarkMessage != null && watermarkMessage.trim().length() > 0)) {
SwingUtilities.invokeLater(new Runnable() {
public void run() {
HyperlinkListener hyperlinkListener = new HyperlinkListener() {
@Override
public void hyperlinkUpdate(HyperlinkEvent e) {
try {
if (e.getEventType() == HyperlinkEvent.EventType.ACTIVATED) {
if (e.getURL() != null) {
BrowserUtil.launch(e.getURL().toString());
}
}
} catch (IOException ex) {
throw new RuntimeException(ex);
}
}
};
HTMLUserPrompter htmlPrompter = new HTMLUserPrompter(
UserPromptOptions.OK,
UserPromptResponse.OK,
null,
watermarkMessage,
hyperlinkListener,
"OK");
htmlPrompter.promptUser("");
}
});
}
} catch (URISyntaxException e) {
throw new IOException(e.getLocalizedMessage());
}
}
public static String getWatermarkMessage(SPServerInfo infos, CookieStore cookieStore)
throws MalformedURLException,IOException
{
return getWatermarkMessage(
infos.getServerAddress(),
String.valueOf(infos.getPort()),
infos.getPath(),
infos.getUsername(),
infos.getPassword(), cookieStore);
}
public static String getWatermarkMessage(
String host,
String port,
String path,
String username,
String password,
CookieStore cookieStore)
{
String message = defaultWatermarkMessage;
try {
if (!isServerLicensed(host,port,path,username,password,cookieStore)) {
message = watermarkMessages.get(generateServerKey(host, port, path, username, password));
} else {
message = "";
}
} catch (Exception e) {
// no op
}
return message;
}
private static String generateServerKey(
String host,
String port,
String path,
String username,
String password) throws MalformedURLException
{
return generateServerKey(
toURL(host, port, path),
username,
password);
}
private static String generateServerKey(
URL url,
String username,
String password)
{
return
String.valueOf(
url.toString()
.concat(username)
.concat(password)
.hashCode());
}
}