@Override
public List<ConnectionProfile> readProfiles(String filename) {
LogMgr.logInfo(
"IniProfileStorage.readProfiles()", "Loading connection profiles from " + filename);
WbProperties props = new WbProperties(1);
BufferedReader reader = null;
List<ConnectionProfile> profiles = new ArrayList<>(25);
WbFile inifile = new WbFile(filename);
try {
File fileDir = inifile.getCanonicalFile().getParentFile();
reader = new BufferedReader(new FileReader(filename));
props.loadFromReader(reader);
Set<String> keys = getProfileKeys(props);
for (String key : keys) {
ConnectionProfile profile = readProfile(fileDir, key, props);
if (profile != null) {
profiles.add(profile);
}
}
} catch (Exception ex) {
LogMgr.logError(
"IniProfileStorage.readProfiles()", "Could not read profiles from: " + filename, ex);
} finally {
FileUtil.closeQuietely(reader);
}
return profiles;
}
private ObjectNameFilter getFilter(WbProperties props, String key, String prop) {
String filterList = props.getProperty(PROP_PREFIX + key + prop);
if (StringUtil.isBlank(filterList)) return null;
boolean inclusion = props.getBoolProperty(PROP_PREFIX + key + prop + ".include", false);
ObjectNameFilter filter = new ObjectNameFilter();
filter.setExpressionList(filterList);
filter.setInclusionFilter(inclusion);
return filter;
}
private Set<String> getProfileKeys(WbProperties props) {
Set<String> uniqueKeys = new TreeSet<>();
List<String> keys = props.getKeysWithPrefix(PROP_PREFIX);
for (String key : keys) {
String[] elements = key.split("\\.");
if (elements.length > 2) {
uniqueKeys.add(elements[1]);
}
}
return uniqueKeys;
}
@Override
public void saveProfiles(List<ConnectionProfile> profiles, String filename) {
LogMgr.logDebug("IniProfileStorage.saveProfiles()", "Saving profiles to: " + filename);
WbProperties props = new WbProperties(2);
// This comparator sorts the "name" attribute at the first place inside the keys for one profile
// This is just for convenience, so that it's easier to read the properties file
Comparator<String> comp =
(String o1, String o2) -> {
int pos1 = o1.indexOf('.', o1.indexOf('.') + 1);
int pos2 = o2.indexOf('.', o2.indexOf('.') + 1);
String base1 = o1.substring(0, pos1);
String base2 = o2.substring(0, pos2);
if (base1.equals(base2)) {
if (o1.endsWith(PROP_NAME) && !o2.endsWith(PROP_NAME)) return -1;
if (!o1.endsWith(PROP_NAME) && o2.endsWith(PROP_NAME)) return 1;
}
return o1.compareTo(o2);
};
props.setSortComparator(comp);
for (int i = 0; i < profiles.size(); i++) {
String key = StringUtil.formatInt(i + 1, 4).toString();
storeProfile(key, profiles.get(i), props);
}
try {
props.saveToFile(new WbFile(filename));
} catch (IOException ex) {
LogMgr.logError(
"IniProfileStorage.saveProfiles()", "Error saving profiles to: " + filename, ex);
}
}
private void setNonDefaultProperty(
WbProperties props, String key, boolean value, boolean defaultValue) {
if (value != defaultValue) {
props.setProperty(key, value);
}
}
private void storeProfile(String key, ConnectionProfile profile, WbProperties props) {
ConnectionProfile defaultValues = new ConnectionProfile();
key = "." + key;
props.setProperty(PROP_PREFIX + key + PROP_URL, profile.getUrl());
props.setProperty(PROP_PREFIX + key + PROP_NAME, profile.getName());
props.setProperty(PROP_PREFIX + key + PROP_DRIVERNAME, profile.getDriverName());
props.setProperty(PROP_PREFIX + key + PROP_DRIVERCLASS, profile.getDriverclass());
props.setProperty(PROP_PREFIX + key + PROP_USERNAME, profile.getUsername());
props.setProperty(PROP_PREFIX + key + PROP_AUTOCOMMMIT, profile.getAutocommit());
props.setProperty(PROP_PREFIX + key + PROP_TAGS, profile.getTagList());
props.setProperty(PROP_PREFIX + key + PROP_STORE_PWD, profile.getStorePassword());
if (profile.getStorePassword()) {
props.setProperty(PROP_PREFIX + key + PROP_PWD, profile.getPassword());
}
props.setProperty(PROP_PREFIX + key + PROP_ICON, profile.getIcon());
props.setProperty(PROP_PREFIX + key + PROP_WORKSPACE, profile.getWorkspaceFile());
props.setProperty(
PROP_PREFIX + key + PROP_ALT_DELIMITER, profile.getAlternateDelimiterString());
props.setProperty(PROP_PREFIX + key + PROP_MACROFILE, profile.getMacroFilename());
props.setProperty(PROP_PREFIX + key + PROP_SCRIPT_CONNECT, profile.getPostConnectScript());
props.setProperty(PROP_PREFIX + key + PROP_SCRIPT_DISCONNECT, profile.getPreDisconnectScript());
setNonDefaultProperty(
props,
PROP_PREFIX + key + PROP_STORECACHE,
profile.getStoreCacheLocally(),
defaultValues.getStoreCacheLocally());
setNonDefaultProperty(
props,
PROP_PREFIX + key + PROP_ROLLBACK_DISCONNECT,
profile.getRollbackBeforeDisconnect(),
defaultValues.getRollbackBeforeDisconnect());
setNonDefaultProperty(
props,
PROP_PREFIX + key + PROP_SEPARATECONNECTION,
profile.getUseSeparateConnectionPerTab(),
defaultValues.getUseSeparateConnectionPerTab());
setNonDefaultProperty(
props,
PROP_PREFIX + key + PROP_IGNOREDROPERRORS,
profile.getIgnoreDropErrors(),
defaultValues.getIgnoreDropErrors());
setNonDefaultProperty(
props,
PROP_PREFIX + key + PROP_TRIMCHARDATA,
profile.getTrimCharData(),
defaultValues.getTrimCharData());
setNonDefaultProperty(
props,
PROP_PREFIX + key + PROP_ORACLESYSDBA,
profile.getOracleSysDBA(),
defaultValues.getOracleSysDBA());
setNonDefaultProperty(
props,
PROP_PREFIX + key + PROP_DETECTOPENTRANSACTION,
profile.getDetectOpenTransaction(),
defaultValues.getDetectOpenTransaction());
setNonDefaultProperty(
props, PROP_PREFIX + key + PROP_READONLY, profile.isReadOnly(), defaultValues.isReadOnly());
setNonDefaultProperty(
props,
PROP_PREFIX + key + PROP_PREVENT_NO_WHERE,
profile.getPreventDMLWithoutWhere(),
defaultValues.getPreventDMLWithoutWhere());
setNonDefaultProperty(
props,
PROP_PREFIX + key + PROP_CONFIRM_UPDATES,
profile.getConfirmUpdates(),
defaultValues.getConfirmUpdates());
setNonDefaultProperty(
props,
PROP_PREFIX + key + PROP_PROMPTUSERNAME,
profile.getPromptForUsername(),
defaultValues.getPromptForUsername());
setNonDefaultProperty(
props,
PROP_PREFIX + key + PROP_EMPTY_STRING_IS_NULL,
profile.getEmptyStringIsNull(),
defaultValues.getEmptyStringIsNull());
setNonDefaultProperty(
props,
PROP_PREFIX + key + PROP_INCLUDE_NULL_ON_INSERT,
profile.getIncludeNullInInsert(),
defaultValues.getIncludeNullInInsert());
setNonDefaultProperty(
props,
PROP_PREFIX + key + PROP_REMOVE_COMMENTS,
profile.getRemoveComments(),
defaultValues.getRemoveComments());
setNonDefaultProperty(
props,
PROP_PREFIX + key + PROP_REMEMEMBER_SCHEMA,
profile.getStoreExplorerSchema(),
defaultValues.getStoreExplorerSchema());
setNonDefaultProperty(
props,
PROP_PREFIX + key + PROP_HIDE_WARNINGS,
profile.isHideWarnings(),
defaultValues.isHideWarnings());
if (StringUtil.stringsAreNotEqual(profile.getGroup(), defaultValues.getGroup())) {
props.setProperty(PROP_PREFIX + key + PROP_GROUP, profile.getGroup());
}
props.setProperty(PROP_PREFIX + key + PROP_SCRIPT_IDLE, profile.getIdleScript());
if (profile.getIdleTime() != defaultValues.getIdleTime()) {
props.setProperty(PROP_PREFIX + key + PROP_IDLE_TIME, Long.toString(profile.getIdleTime()));
}
props.setProperty(
PROP_PREFIX + key + PROP_INFO_COLOR, Settings.colorToString(profile.getInfoDisplayColor()));
Integer fetchSize = profile.getDefaultFetchSize();
if (fetchSize != null) {
props.setProperty(PROP_PREFIX + key + PROP_FETCHSIZE, fetchSize.intValue());
}
Integer timeout = profile.getConnectionTimeout();
if (timeout != null) {
props.setProperty(PROP_PREFIX + key + PROP_CONNECTION_TIMEOUT, timeout.intValue());
}
ObjectNameFilter filter = profile.getSchemaFilter();
String expr = (filter != null ? filter.getFilterString() : null);
if (expr != null && filter != null) {
props.setProperty(PROP_PREFIX + key + PROP_SCHEMA_FILTER, expr);
props.setProperty(
PROP_PREFIX + key + PROP_SCHEMA_FILTER + ".include", filter.isInclusionFilter());
}
filter = profile.getCatalogFilter();
expr = (filter != null ? filter.getFilterString() : null);
if (expr != null && filter != null) {
props.setProperty(PROP_PREFIX + key + PROP_CATALOG_FILTER, expr);
props.setProperty(
PROP_PREFIX + key + PROP_CATALOG_FILTER + ".include", filter.isInclusionFilter());
}
String xml = toXML(profile.getConnectionProperties());
props.setProperty(PROP_PREFIX + key + PROP_CONN_PROPS, xml);
}
private ConnectionProfile readProfile(File baseDir, String key, WbProperties props) {
key = "." + key;
String url = props.getProperty(PROP_PREFIX + key + PROP_URL, null);
String tags = props.getProperty(PROP_PREFIX + key + PROP_TAGS, null);
String name = props.getProperty(PROP_PREFIX + key + PROP_NAME, null);
String driverClass = props.getProperty(PROP_PREFIX + key + PROP_DRIVERCLASS, null);
String driverJar = props.getProperty(PROP_PREFIX + key + PROP_DRIVERJAR, null);
String driverName = props.getProperty(PROP_PREFIX + key + PROP_DRIVERNAME, null);
String group = props.getProperty(PROP_PREFIX + key + PROP_GROUP, null);
String user = props.getProperty(PROP_PREFIX + key + PROP_USERNAME, null);
String pwd = props.getProperty(PROP_PREFIX + key + PROP_PWD, null);
String icon = props.getProperty(PROP_PREFIX + key + PROP_ICON, null);
String wksp = props.getProperty(PROP_PREFIX + key + PROP_WORKSPACE, null);
String delimiter = props.getProperty(PROP_PREFIX + key + PROP_ALT_DELIMITER, null);
String macroFile = props.getProperty(PROP_PREFIX + key + PROP_MACROFILE, null);
String postConnect = props.getProperty(PROP_PREFIX + key + PROP_SCRIPT_CONNECT, null);
String preDisconnect = props.getProperty(PROP_PREFIX + key + PROP_SCRIPT_DISCONNECT, null);
String idleScript = props.getProperty(PROP_PREFIX + key + PROP_SCRIPT_IDLE, null);
String xmlProps = props.getProperty(PROP_PREFIX + key + PROP_CONN_PROPS, null);
String colorValue = props.getProperty(PROP_PREFIX + key + PROP_INFO_COLOR, null);
Properties connProps = toProperties(xmlProps);
Color color = Settings.stringToColor(colorValue);
boolean autoCommit = props.getBoolProperty(PROP_PREFIX + key + PROP_AUTOCOMMMIT, false);
boolean storeCache = props.getBoolProperty(PROP_PREFIX + key + PROP_STORECACHE, false);
boolean storePwd = props.getBoolProperty(PROP_PREFIX + key + PROP_STORE_PWD, true);
boolean rollback = props.getBoolProperty(PROP_PREFIX + key + PROP_ROLLBACK_DISCONNECT, false);
boolean seperateConnection =
props.getBoolProperty(PROP_PREFIX + key + PROP_SEPARATECONNECTION, false);
boolean ignoreDropError =
props.getBoolProperty(PROP_PREFIX + key + PROP_IGNOREDROPERRORS, false);
boolean trimCharData = props.getBoolProperty(PROP_PREFIX + key + PROP_TRIMCHARDATA, false);
boolean sysDBA = props.getBoolProperty(PROP_PREFIX + key + PROP_ORACLESYSDBA, false);
boolean detectOpen =
props.getBoolProperty(PROP_PREFIX + key + PROP_DETECTOPENTRANSACTION, false);
boolean readonly = props.getBoolProperty(PROP_PREFIX + key + PROP_READONLY, false);
boolean preventNoWhere =
props.getBoolProperty(PROP_PREFIX + key + PROP_PREVENT_NO_WHERE, false);
boolean confirmUpdates = props.getBoolProperty(PROP_PREFIX + key + PROP_CONFIRM_UPDATES, false);
boolean promptUsername = props.getBoolProperty(PROP_PREFIX + key + PROP_PROMPTUSERNAME, false);
boolean emptyStringIsNull =
props.getBoolProperty(PROP_PREFIX + key + PROP_EMPTY_STRING_IS_NULL, false);
boolean includeNullInInsert =
props.getBoolProperty(PROP_PREFIX + key + PROP_INCLUDE_NULL_ON_INSERT, true);
boolean removeComments = props.getBoolProperty(PROP_PREFIX + key + PROP_REMOVE_COMMENTS, false);
boolean rememberExplorerSchema =
props.getBoolProperty(PROP_PREFIX + key + PROP_REMEMEMBER_SCHEMA, false);
boolean hideWarnings = props.getBoolProperty(PROP_PREFIX + key + PROP_HIDE_WARNINGS, false);
boolean copyProps = props.getBoolProperty(PROP_PREFIX + key + PROP_COPY_PROPS, false);
int idleTime = props.getIntProperty(PROP_PREFIX + key + PROP_IDLE_TIME, -1);
int size = props.getIntProperty(PROP_PREFIX + key + PROP_FETCHSIZE, -1);
int timeOut = props.getIntProperty(PROP_PREFIX + key + PROP_CONNECTION_TIMEOUT, -1);
Integer fetchSize = null;
if (size >= 0) {
fetchSize = Integer.valueOf(size);
}
Integer connectionTimeOut = null;
if (timeOut > 0) {
connectionTimeOut = Integer.valueOf(timeOut);
}
// if a driver jar was explicitely specified, that jar should be used
// regardless of any registered driver that might be referenced through driverName
if (StringUtil.isNonEmpty(driverJar)) {
WbFile drvFile = new WbFile(driverJar);
if (!drvFile.isAbsolute()) {
drvFile = new WbFile(baseDir, driverJar);
LogMgr.logDebug(
"IniProfileStorage.readProfile()",
"Using full path: "
+ drvFile.getFullPath()
+ " for driver jar "
+ driverJar
+ " from profile "
+ name);
driverJar = drvFile.getFullPath();
} else {
driverJar = drvFile.getFullPath();
}
DbDriver drv = ConnectionMgr.getInstance().registerDriver(driverClass, driverJar);
driverName = drv.getName();
}
ObjectNameFilter schemaFilter = getSchemaFilter(props, key);
ObjectNameFilter catalogFilter = getCatalogFilter(props, key);
ConnectionProfile profile = new ConnectionProfile();
profile.setName(name);
profile.setUsername(user);
profile.setUrl(url);
profile.setInputPassword(pwd);
profile.setDriverclass(driverClass);
profile.setDriverName(driverName);
profile.setGroup(group);
profile.setTagList(tags);
profile.setDefaultFetchSize(fetchSize);
profile.setOracleSysDBA(sysDBA);
profile.setReadOnly(readonly);
profile.setWorkspaceFile(wksp);
profile.setIcon(icon);
profile.setConnectionTimeout(connectionTimeOut);
profile.setRollbackBeforeDisconnect(rollback);
profile.setUseSeparateConnectionPerTab(seperateConnection);
profile.setAlternateDelimiterString(delimiter);
profile.setMacroFilename(macroFile);
profile.setIgnoreDropErrors(ignoreDropError);
profile.setTrimCharData(trimCharData);
profile.setDetectOpenTransaction(detectOpen);
profile.setPreventDMLWithoutWhere(preventNoWhere);
profile.setConfirmUpdates(confirmUpdates);
profile.setPromptForUsername(promptUsername);
profile.setEmptyStringIsNull(emptyStringIsNull);
profile.setIncludeNullInInsert(includeNullInInsert);
profile.setRemoveComments(removeComments);
profile.setStoreExplorerSchema(rememberExplorerSchema);
profile.setHideWarnings(hideWarnings);
profile.setStoreCacheLocally(storeCache);
profile.setAutocommit(autoCommit);
profile.setPreDisconnectScript(preDisconnect);
profile.setPostConnectScript(postConnect);
profile.setIdleScript(idleScript);
profile.setIdleTime(idleTime);
profile.setStorePassword(storePwd);
profile.setCopyExtendedPropsToSystem(copyProps);
profile.setConnectionProperties(connProps);
profile.setInfoDisplayColor(color);
profile.setSchemaFilter(schemaFilter);
profile.setCatalogFilter(catalogFilter);
return profile;
}