package org.apache.torque.mojo;

 import java.util.Properties;

 import org.apache.maven.plugin.MojoExecutionException;

 import org.apache.maven.settings.Server;

 import org.kuali.db.DatabaseCommand;

 import static org.apache.commons.lang.StringUtils.*;

 /**

  * Common logic for running SQL commands on a database

  */

 public abstract class AbstractDBACommandMojo extends AbstractSQLExecutorMojo {

         public static final String DATABASE_PROPERTY = "database";

         public static final String DATABASE_PW_PROPERTY = "databasePassword";

         public static final String DATABASE_USERNAME_PROPERTY = "databaseUsername";

         /**

          * URL to connect directly to the database server itself (ie: no database specified). This is optional as it is

          * automatically detected in almost all cases from the <code>url</code>. If <code>serverUrl</code> is explicitly

          * supplied it overrides the <code>serverUrl</code> chosen by the automatic detection logic.

          * 

          * @parameter expression="${serverUrl}"

          */

         String serverUrl;

         /**

          * The name of the database to DROP/CREATE. If not specified, this defaults to a database name that is compatible

          * with ${targetDatabase} based on platform specific logic that converts the artifact id.<br>

          * <br>

          * 

          * For example:<br>

          * ks-embedded-db is converted to KSEMBEDDED for Oracle, and ksembedded for MySQL)

          * 

          * @parameter expression="${database}"

          */

         String database;

         /**

          * The user to DROP/CREATE when issuing DBA commands for creating/dropping a user. If not specified, this defaults

          * to a user that is compatible with ${targetDatabase} based on platform specific logic that converts the artifact

          * id.<br>

          * <br>

          * 

          * For example:<br>

          * ks-embedded-db is converted to KSEMBEDDED for Oracle, and ksembedded for MySQL

          * 

          * @parameter expression="${databaseUser}"

          */

         String databaseUser;

         /**

          * The password for the user that is DROPPED/CREATED. If not specified, this defaults to a password that is

          * compatible with ${targetDatabase} based on platform specific logic that converts the artifact id.<br>

          * <br>

          * 

          * For example:<br>

          * ks-embedded-db is converted to KSEMBEDDED for Oracle, and ksembedded for MySQL

          * 

          * @parameter expression="${databasePassword}"

          */

         String databasePassword;

         /**

          * Lookup DBA credentials in settings.xml using this key. If nothing is found under

          * <code>impex.dba.${project.artifactId}</code> a second attempt will be made to locate a set of credentials under

          * <code>impex.dba.${url}</code>

          * 

          * @parameter expression="${dbaSettingsKey}" default-value="impex.dba.${project.artifactId}"

          */

         String dbaSettingsKey;

         protected String getTransactionDescription(DatabaseCommand command) {

                 return command + " " + getDatabase();

         }

         protected void updateConfiguration() throws MojoExecutionException {

                 super.updateConfiguration();

                 if (isEmpty(database)) {

                         database = platform.getSchemaName(getProject().getArtifactId());

                 }

                 if (isEmpty(databasePassword)) {

                         databasePassword = platform.getSchemaName(getProject().getArtifactId());

                 }

                 if (isEmpty(databaseUser)) {

                         databaseUser = platform.getSchemaName(getProject().getArtifactId());

                 }

                 if (isEmpty(serverUrl)) {

                         serverUrl = platform.getServerUrl(url);

                 }

         }

         @Override

         protected String getUpdatedPassword(Server server, String password) {

                 // They already gave us a password, don't mess with it

                 if (!isEmpty(password)) {

                         return password;

                 }

                 if (server != null) {

                         // We've successfully located a server in settings.xml, use the password from that

                         getLog().info("Located a password in settings.xml under the server id '" + server.getId() + "' Password: " + getDisplayPassword(server.getPassword()));

                         return server.getPassword();

                 }

                 // Do not return a default value

                 return null;

         }

         @Override

         protected String getUpdatedUsername(Server server, String username) {

                 // They already gave us a username, don't mess with it

                 if (!isEmpty(username)) {

                         return username;

                 }

                 if (server != null) {

                         // We've successfully located a server in settings.xml, use the username from that

                         getLog().info("Located a username in settings.xml under the server id '" + server.getId() + "' Username: " + server.getUsername());

                         return server.getUsername();

                 }

                 // Do not return a default value

                 return null;

         }

         @Override

         protected Properties getContextProperties() {

                 Properties properties = super.getContextProperties();

                 properties.setProperty(DATABASE_PROPERTY, getDatabase());

                 properties.setProperty(DATABASE_PW_PROPERTY, getDatabasePassword());

                 properties.setProperty(DATABASE_USERNAME_PROPERTY, getDatabaseUser());

                 return properties;

         }

         @Override

         protected Server getServerFromSettingsKey() {

                 Server server = getSettings().getServer(dbaSettingsKey);

                 if (server != null) {

                         return server;

                 }

                 String settingsKey = "impex.dba." + getUrl();

                 return getSettings().getServer(settingsKey);

         }

         @Override

         protected void validateConfiguration() throws MojoExecutionException {

                 super.validateConfiguration();

                 if (isEmpty(database)) {

                         throw new MojoExecutionException("\n\nNo database was specified.\nSpecify a database in the plugin configuration or as a system property.\n\nFor example:\n-Ddatabase=MYDB\n\n.");

                 }

         }

         public String getDatabase() {

                 return database;

         }

         public void setDatabase(String schema) {

                 this.database = schema;

         }

         public String getDatabasePassword() {

                 return databasePassword;

         }

         public void setDatabasePassword(String databasePassword) {

                 this.databasePassword = databasePassword;

         }

         public String getDatabaseUser() {

                 return databaseUser;

         }

         public void setDatabaseUser(String databaseUsername) {

                 this.databaseUser = databaseUsername;

         }

         public String getServerUrl() {

                 return serverUrl;

         }

         public void setServerUrl(String serverUrl) {

                 this.serverUrl = serverUrl;

         }

         public String getDbaSettingsKey() {

                 return dbaSettingsKey;

         }

         public void setDbaSettingsKey(String dbaSettingsKey) {

                 this.dbaSettingsKey = dbaSettingsKey;

         }

 }