/*
    * Copyright 2013 smartics, Kronseder & Reiner GmbH
    *
    * Licensed under the Apache License, Version 2.0 (the "License");
    * you may not use this file except in compliance with the License.
    * You may obtain a copy of the License at
    *
    *     http://www.apache.org/licenses/LICENSE-2.0
    *
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS,
   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   * See the License for the specific language governing permissions and
   * limitations under the License.
   */
  package de.smartics.properties.jboss.extension.resources.components;
  
  import static javax.xml.stream.XMLStreamConstants.END_ELEMENT;
  
  import javax.xml.stream.XMLStreamException;
  
  import org.jboss.as.controller.AttributeDefinition;
  import org.jboss.as.controller.OperationFailedException;
  import org.jboss.as.controller.OperationStepHandler;
  import org.jboss.as.controller.SimpleAttributeDefinition;
  import org.jboss.as.controller.SimpleAttributeDefinitionBuilder;
  import org.jboss.as.controller.parsing.ParseUtils;
  import org.jboss.as.controller.registry.AttributeAccess;
  import org.jboss.as.controller.registry.ManagementResourceRegistration;
  import org.jboss.dmr.ModelNode;
  import org.jboss.dmr.ModelType;
  import org.jboss.staxmapper.XMLExtendedStreamReader;
  import org.jboss.staxmapper.XMLExtendedStreamWriter;
  
  import de.smartics.properties.api.config.ds.DataSourceConfiguration;
  import de.smartics.properties.spi.config.config.PropertiesConfiguration;
  
  /**
   * Defines the data source information of a configuration. This information
   * allows to access a data source containing property definitions.
   * <p>
   * Example XML fragment of a data source element within a configuration:
   * </p>
   *
   * <pre>
   * {@markupExample "Data Source Element"
   * &lt;datasource&gt;
   *   &lt;jndiName&gt;java:jboss/datasources/propertiesDS&lt;/jndiName&gt;
   *   &lt;dropTable&gt;false&lt;/dropTable&gt;
   *   &lt;createTable&gt;true&lt;/createTable&gt;
   * &lt;/datasource&gt;}
   * </pre>
   */
  public final class DataSourceComponent extends AbstractComponent
  {
    // ********************************* Fields *********************************
  
    // --- constants ------------------------------------------------------------
  
    /**
     * The name of the configuration XML element that holds the data source
     * information.
     * <p>
     * The value of this constant is {@value}.
     * </p>
     */
    public static final String DS_ELEMENT_NAME = "datasource";
  
    /**
     * The attribute that holds the JNDI name of the data source.
     */
    public static final SimpleAttributeDefinition JNDI_NAME =
        new SimpleAttributeDefinitionBuilder("datasource.jndiName",
            ModelType.STRING, true).setXmlName("jndiName")
            .setAllowExpression(true)
            .setFlags(AttributeAccess.Flag.RESTART_ALL_SERVICES).build();
  
    /**
     * The attribute that holds the <code>drop table</code> flag of the data
     * source.
     */
    public static final SimpleAttributeDefinition DROP_TABLE =
        new SimpleAttributeDefinitionBuilder("datasource.dropTable",
            ModelType.BOOLEAN, true).setXmlName("dropTable")
            .setAllowExpression(true)
            .setFlags(AttributeAccess.Flag.RESTART_ALL_SERVICES).build();
  
    /**
     * The attribute that holds the <code>create table</code> flag of the data
     * source.
     */
    public static final SimpleAttributeDefinition CREATE_TABLE =
        new SimpleAttributeDefinitionBuilder("datasource.createTable",
            ModelType.BOOLEAN, true).setXmlName("createTable")
            .setAllowExpression(true)
            .setFlags(AttributeAccess.Flag.RESTART_ALL_SERVICES).build();
  
    private static final AttributeDefinition[] DEFINITIONS = { JNDI_NAME,
                                                              DROP_TABLE,
                                                             CREATE_TABLE };
 
   // --- members --------------------------------------------------------------
 
   // ****************************** Initializer *******************************
 
   // ****************************** Constructors ******************************
 
   /**
    * Default constructor.
    */
   public DataSourceComponent(final OperationStepHandler writeHandler)
   {
     super(writeHandler);
   }
 
   // ****************************** Inner Classes *****************************
 
   // ********************************* Methods ********************************
 
   // --- init -----------------------------------------------------------------
 
   // --- get&set --------------------------------------------------------------
 
   // --- business -------------------------------------------------------------
 
   public void registerAttributes(
       final ManagementResourceRegistration resourceRegistration)
   {
     resourceRegistration.registerReadWriteAttribute(JNDI_NAME, null,
         writeHandler);
     resourceRegistration.registerReadWriteAttribute(DROP_TABLE, null,
         writeHandler);
     resourceRegistration.registerReadWriteAttribute(CREATE_TABLE, null,
         writeHandler);
   }
 
   public void populateModel(final ModelNode operation, final ModelNode model)
     throws OperationFailedException
   {
     JNDI_NAME.validateAndSet(operation, model);
     DROP_TABLE.validateAndSet(operation, model);
     CREATE_TABLE.validateAndSet(operation, model);
   }
 
   public void read(final XMLExtendedStreamReader reader,
       final ModelNode addOperation) throws XMLStreamException
   {
     ParseUtils.requireNoAttributes(reader);
 
     while (reader.hasNext() && reader.nextTag() != END_ELEMENT)
     {
       if (reader.isStartElement())
       {
         final String localName = reader.getLocalName();
 
         if (DataSourceComponent.JNDI_NAME.getXmlName().equals(localName))
         {
           readAttribute(reader, addOperation, DataSourceComponent.JNDI_NAME);
         }
         else if (DataSourceComponent.DROP_TABLE.getXmlName().equals(localName))
         {
           readAttribute(reader, addOperation, DataSourceComponent.DROP_TABLE);
         }
         else if (DataSourceComponent.CREATE_TABLE.getXmlName()
             .equals(localName))
         {
           readAttribute(reader, addOperation, DataSourceComponent.CREATE_TABLE);
         }
         else
         {
           throw ParseUtils.unexpectedElement(reader);
         }
       }
     }
   }
 
   public void write(final XMLExtendedStreamWriter writer, final ModelNode config)
     throws XMLStreamException
   {
     if (isAtLeastOneDefined(config, DEFINITIONS))
     {
       writer.writeStartElement(DataSourceComponent.DS_ELEMENT_NAME);
 
       for (final AttributeDefinition definition : DEFINITIONS)
       {
         writeAttribute(writer, config, definition);
       }
 
       writer.writeEndElement();
     }
   }
 
   public void apply(final PropertiesConfiguration propertiesConfig,
       final ModelNode model)
   {
     final DataSourceConfiguration dataSourceConfig =
         propertiesConfig.getDataSourceConfiguration();
 
     final String jndiName = getAttribute(model, JNDI_NAME);
     dataSourceConfig.setJndiName(jndiName);
 
     final String createTableAtt = getAttribute(model, CREATE_TABLE);
     if (createTableAtt != null)
     {
       final boolean createTable = Boolean.parseBoolean(createTableAtt);
       dataSourceConfig.setCreateTable(createTable);
     }
 
     final String dropTableAtt = getAttribute(model, DROP_TABLE);
     if (dropTableAtt != null)
     {
       final boolean dropTable = Boolean.parseBoolean(dropTableAtt);
       dataSourceConfig.setDropTable(dropTable);
     }
   }
 
   // --- object basics --------------------------------------------------------
 
 }