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     */
020    package org.apache.directory.server.core.trigger;
021    
022    
023    import java.util.Map;
024    
025    import org.apache.directory.server.core.interceptor.context.OperationContext;
026    import org.apache.directory.shared.ldap.entry.ServerEntry;
027    import org.apache.directory.shared.ldap.exception.LdapException;
028    import org.apache.directory.shared.ldap.exception.LdapInvalidDnException;
029    import org.apache.directory.shared.ldap.name.DN;
030    import org.apache.directory.shared.ldap.trigger.StoredProcedureParameter;
031    
032    
033    public class AddStoredProcedureParameterInjector extends AbstractStoredProcedureParameterInjector
034    {
035        private DN addedEntryName;
036        private ServerEntry addedEntry;
037        
038        
039        public AddStoredProcedureParameterInjector( OperationContext opContext, DN addedEntryName, 
040            ServerEntry 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        
051        MicroInjector $entryInjector = new MicroInjector()
052        {
053            public Object inject( OperationContext opContext, StoredProcedureParameter param ) throws LdapInvalidDnException
054            {
055                // Return a safe copy constructed with user provided name.
056                return new DN( addedEntryName.getName() );
057            }
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    }