/*

   Derby - Class org.apache.derby.jdbc.ReferenceableDataSource

   Licensed to the Apache Software Foundation (ASF) under one or more
   contributor license agreements.  See the NOTICE file distributed with
   this work for additional information regarding copyright ownership.
   The ASF licenses this file to You 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 org.apache.derby.jdbc;

import java.lang.reflect.Method;
import java.util.Enumeration;
import javax.naming.RefAddr;
import javax.naming.Reference;
import javax.naming.spi.ObjectFactory;

/**
 * The data source factory for Derby embedded driver data sources.
 * <p>
 * This factory reconstructs a Derby data source object when it is retrieved
 * from JNDI. References are needed since many naming services don't have the
 * ability to store Java objects in their serialized form. When a data source
 * object is bound in this type of naming service the
 * {@link javax.naming.Reference} for that object is actually stored by the
 * JNDI implementation, not the data source object itself.
 * <p>
 * A JNDI administrator is responsible for making sure that both the object
 * factory and data source implementation classes provided by a JDBC driver
 * vendor are accessible to the JNDI service provider at runtime.
 * <p>
 * An object factory implements the {@link javax.naming.spi.ObjectFactory}
 * interface. This interface contains a single method, {@code getObjectInstance}
 * which is called by a JNDI service provider to reconstruct an object when that
 * object is retrieved from JNDI. A JDBC driver vendor should provide an object
 * factory as part of their JDBC product.
 *
 * @see org.apache.derby.jdbc.EmbeddedDataSource
 * @see org.apache.derby.jdbc.EmbeddedConnectionPoolDataSource
 * @see org.apache.derby.jdbc.EmbeddedXADataSource
 */

public class ReferenceableDataSource extends BasicEmbeddedDataSource40
                                     implements ObjectFactory {

    private static final long serialVersionUID = 1872877359127597176L;

    /**
     * Reconstructs a Derby embedded-driver data source object from a JNDI data
     * source reference.
     * <p>
     * The {@code getObjectInstance} method is passed a reference that
     * corresponds to the object being retrieved as its first parameter. The
     * other parameters are optional in the case of JDBC data source objects.
     * The object factory should use the information contained in the reference
     * to reconstruct the data source. If for some reason, a data source
     * object cannot be reconstructed from the reference, a value of
     * {@code null} may be returned. This allows other object factories that may
     * be registered in JNDI to be tried. If an exception is thrown then no
     * other object factories are tried.
     *
     * @param refObj the possibly {@code null} object containing location or
     *      reference information that can be used in creating an object
     * @param name the name of this object relative to {@code nameContext}, or
     *      {@code null} if no name is specified
     * @param nameContext context relative to which the name parameter is
     *      specified, or {@code null} if name is relative to the default
     *      initial context.
     * @param environment possibly {@code null} environment that is used in
     *      creating the object.
     *
     * @return Object created, or {@code null} if no attempt to create the
     *      object is made.
     * @throws Exception if recreating the object fails
     */
    @Override
    public Object getObjectInstance(Object refObj,
                                    javax.naming.Name name,
                                    javax.naming.Context nameContext,
                                    java.util.Hashtable<?,?> environment)
            throws java.lang.Exception {

        Object ds = null;
        if (refObj instanceof javax.naming.Reference) {
            javax.naming.Reference ref = (javax.naming.Reference) refObj;

            // See if this object belongs to Derby.
            String className = ref.getClassName();
            if (className != null &&
                    className.startsWith("org.apache.derby.jdbc.Embedded")) {
                // Create the proper data source object shell.
                ds = Class.forName(className).getConstructor().newInstance();

                // Fill in the data source object shell with values from the
                // jndi reference.
                setBeanProperties(ds, ref);
            }
        }
        return ds;
    }

    // Reflect lookup for Java bean method taking a single String arg
    private static final Class<?>[] STRING_ARG = { "".getClass() };
    // Reflect lookup for Java bean method taking a single int arg
    private static final Class<?>[] INT_ARG = { Integer.TYPE };
    // Reflect lookup for Java bean method taking a single boolean arg
    private static final Class<?>[] BOOLEAN_ARG = { Boolean.TYPE };
    // Reflect lookup for Java bean method taking a single short arg
    private static final Class<?>[] SHORT_ARG = { Short.TYPE };

    // Set the Java bean properties for an object from its Reference. The
    // Reference contains a set of StringRefAddr values with the key being the
    // bean name and the value a String representation of the bean's value. This
    // code looks for setXXX() method where the set method corresponds to the
    // standard bean naming scheme and has a single parameter of type String,
    // int, boolean or short.
    private static void setBeanProperties(Object ds, Reference ref)
            throws Exception {

        for (Enumeration<RefAddr> e = ref.getAll(); e.hasMoreElements();) {

            RefAddr attribute = e.nextElement();

            String propertyName = attribute.getType();

            String value = (String) attribute.getContent();

            String methodName = "set"
                    + propertyName.substring(0, 1).toUpperCase(
                            java.util.Locale.ENGLISH)
                    + propertyName.substring(1);

            Method m;

            Object argValue;
            try {
                m = ds.getClass().getMethod(methodName, STRING_ARG);
                argValue = value;
            } catch (NoSuchMethodException nsme) {
                try {
                    m = ds.getClass().getMethod(methodName, INT_ARG);
                    argValue = Integer.valueOf(value);
                } catch (NoSuchMethodException nsme2) {
                    try {
                        m = ds.getClass().getMethod(methodName, BOOLEAN_ARG);
                        argValue = Boolean.valueOf(value);
                    } catch (NoSuchMethodException nsme3) {
                        m = ds.getClass().getMethod(methodName, SHORT_ARG);
                        argValue = Short.valueOf(value);
                    }
                }
            }
            m.invoke(ds, new Object[] { argValue });
        }
    }
}
