001/*
002 *  Licensed to the Apache Software Foundation (ASF) under one
003 *  or more contributor license agreements.  See the NOTICE file
004 *  distributed with this work for additional information
005 *  regarding copyright ownership.  The ASF licenses this file
006 *  to you under the Apache License, Version 2.0 (the
007 *  "License"); you may not use this file except in compliance
008 *  with the License.  You may obtain a copy of the License at
009 *  
010 *    http://www.apache.org/licenses/LICENSE-2.0
011 *  
012 *  Unless required by applicable law or agreed to in writing,
013 *  software distributed under the License is distributed on an
014 *  "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
015 *  KIND, either express or implied.  See the License for the
016 *  specific language governing permissions and limitations
017 *  under the License. 
018 *  
019 */
020package org.apache.directory.server.core.trigger;
021
022
023import java.util.Map;
024
025import org.apache.directory.api.ldap.model.entry.Entry;
026import org.apache.directory.api.ldap.model.exception.LdapException;
027import org.apache.directory.api.ldap.model.exception.LdapInvalidDnException;
028import org.apache.directory.api.ldap.model.name.Dn;
029import org.apache.directory.api.ldap.trigger.StoredProcedureParameter;
030import org.apache.directory.server.core.api.interceptor.context.OperationContext;
031
032
033public class AddStoredProcedureParameterInjector extends AbstractStoredProcedureParameterInjector
034{
035    private Dn addedEntryName;
036    private Entry addedEntry;
037
038
039    public AddStoredProcedureParameterInjector( OperationContext opContext, Dn addedEntryName,
040        Entry addedEntry )
041    {
042        super( opContext );
043        this.addedEntryName = addedEntryName;
044        this.addedEntry = addedEntry;
045        Map<Class<?>, MicroInjector> injectors = super.getInjectors();
046        injectors.put( StoredProcedureParameter.Add_ENTRY.class, entryInjector );
047        injectors.put( StoredProcedureParameter.Add_ATTRIBUTES.class, attributesInjector );
048    }
049
050    MicroInjector entryInjector = new MicroInjector()
051    {
052        @Override
053        public Object inject( OperationContext opContext, StoredProcedureParameter param )
054            throws LdapInvalidDnException
055        {
056            // Return a safe copy constructed with user provided name.
057            return opContext.getSession().getDirectoryService().getDnFactory().create( addedEntryName.getName() );
058        }
059    };
060
061    MicroInjector attributesInjector = new MicroInjector()
062    {
063        public Object inject( OperationContext opContext, StoredProcedureParameter param ) throws LdapException
064        {
065            return addedEntry.clone();
066        }
067    };
068
069}