org.sqlite.ExtendedCommand Maven / Gradle / Ivy
//--------------------------------------
// sqlite-jdbc Project
//
// ExtendedCommand.java
// Since: Mar 12, 2010
//
// $URL$
// $Author$
//--------------------------------------
package org.sqlite;
import java.sql.SQLException;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import org.sqlite.core.DB;
/**
* parsing SQLite specific extension of SQL command
*
* @author leo
*
*/
public class ExtendedCommand
{
public static interface SQLExtension
{
public void execute(DB db) throws SQLException;
}
/**
* Parses extended commands of "backup" or "restore" for SQLite database.
* @param sql One of the extended commands:
* backup sourceDatabaseName to destinationFileName OR restore targetDatabaseName from sourceFileName
* @return BackupCommand object if the argument is a backup command; RestoreCommand object if
* the argument is a restore command;
* @throws SQLException
*/
public static SQLExtension parse(String sql) throws SQLException {
if (sql == null)
return null;
if (sql.startsWith("backup"))
return BackupCommand.parse(sql);
else if (sql.startsWith("restore"))
return RestoreCommand.parse(sql);
return null;
}
/**
* Remove the quotation mark from string.
* @param s String with quotation mark.
* @return String with quotation mark removed.
*/
public static String removeQuotation(String s) {
if (s == null)
return s;
if ((s.startsWith("\"") && s.endsWith("\"")) || (s.startsWith("'") && s.endsWith("'")))
return s.substring(1, s.length() - 1);
else
return s;
}
public static class BackupCommand implements SQLExtension
{
public final String srcDB;
public final String destFile;
/**
* Constructs a BackupCommand instance that backup the database to a target file.
* @param srcDB Source database name.
* @param destFile Target file name.
*/
public BackupCommand(String srcDB, String destFile) {
this.srcDB = srcDB;
this.destFile = destFile;
}
private static Pattern backupCmd = Pattern
.compile("backup(\\s+(\"[^\"]*\"|'[^\']*\'|\\S+))?\\s+to\\s+(\"[^\"]*\"|'[^\']*\'|\\S+)");
/**
* Parses SQLite database backup command and creates a BackupCommand object.
* @param sql SQLite database backup command.
* @return BackupCommand object.
* @throws SQLException
*/
public static BackupCommand parse(String sql) throws SQLException {
if (sql != null) {
Matcher m = backupCmd.matcher(sql);
if (m.matches()) {
String dbName = removeQuotation(m.group(2));
String dest = removeQuotation(m.group(3));
if (dbName == null || dbName.length() == 0)
dbName = "main";
return new BackupCommand(dbName, dest);
}
}
throw new SQLException("syntax error: " + sql);
}
public void execute(DB db) throws SQLException {
db.backup(srcDB, destFile, null);
}
}
public static class RestoreCommand implements SQLExtension
{
public final String targetDB;
public final String srcFile;
private static Pattern restoreCmd = Pattern
.compile("restore(\\s+(\"[^\"]*\"|'[^\']*\'|\\S+))?\\s+from\\s+(\"[^\"]*\"|'[^\']*\'|\\S+)");
/**
* Constructs a RestoreCommand instance that restores the database from a given source file.
* @param targetDB Target database name
* @param srcFile Source file name
*/
public RestoreCommand(String targetDB, String srcFile) {
this.targetDB = targetDB;
this.srcFile = srcFile;
}
/**
* Parses SQLite database restore command and creates a RestoreCommand object.
* @param sql SQLite restore backup command
* @return RestoreCommand object.
* @throws SQLException
*/
public static RestoreCommand parse(String sql) throws SQLException {
if (sql != null) {
Matcher m = restoreCmd.matcher(sql);
if (m.matches()) {
String dbName = removeQuotation(m.group(2));
String dest = removeQuotation(m.group(3));
if (dbName == null || dbName.length() == 0)
dbName = "main";
return new RestoreCommand(dbName, dest);
}
}
throw new SQLException("syntax error: " + sql);
}
/**
* @see org.sqlite.ExtendedCommand.SQLExtension#execute(org.sqlite.core.DB)
*/
public void execute(DB db) throws SQLException {
db.restore(targetDB, srcFile, null);
}
}
}