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.ArrayList;
024 import java.util.List;
025 import java.util.Map;
026
027 import org.apache.directory.server.core.entry.ClonedServerEntry;
028 import org.apache.directory.server.core.interceptor.context.ModifyOperationContext;
029 import org.apache.directory.server.core.interceptor.context.OperationContext;
030 import org.apache.directory.server.core.partition.ByPassConstants;
031 import org.apache.directory.shared.ldap.entry.Modification;
032 import org.apache.directory.shared.ldap.entry.ServerEntry;
033 import org.apache.directory.shared.ldap.exception.LdapException;
034 import org.apache.directory.shared.ldap.exception.LdapInvalidDnException;
035 import org.apache.directory.shared.ldap.name.DN;
036 import org.apache.directory.shared.ldap.trigger.StoredProcedureParameter;
037
038
039 public class ModifyStoredProcedureParameterInjector extends AbstractStoredProcedureParameterInjector
040 {
041 private DN modifiedEntryName;
042 private List<Modification> modifications;
043 private ServerEntry oldEntry;
044
045
046 public ModifyStoredProcedureParameterInjector( ModifyOperationContext opContext ) throws Exception
047 {
048 super( opContext );
049 modifiedEntryName = opContext.getDn();
050 modifications = opContext.getModItems();
051 this.oldEntry = getEntry( opContext );
052 Map<Class<?>, MicroInjector> injectors = super.getInjectors();
053 injectors.put( StoredProcedureParameter.Modify_OBJECT.class, $objectInjector );
054 injectors.put( StoredProcedureParameter.Modify_MODIFICATION.class, $modificationInjector );
055 injectors.put( StoredProcedureParameter.Modify_OLD_ENTRY.class, $oldEntryInjector );
056 injectors.put( StoredProcedureParameter.Modify_NEW_ENTRY.class, $newEntryInjector );
057 }
058
059
060 MicroInjector $objectInjector = new MicroInjector()
061 {
062 public Object inject( OperationContext opContext, StoredProcedureParameter param ) throws LdapInvalidDnException
063 {
064 // Return a safe copy constructed with user provided name.
065 return new DN( modifiedEntryName.getName() );
066 }
067 };
068
069
070 MicroInjector $modificationInjector = new MicroInjector()
071 {
072 public Object inject( OperationContext opContext, StoredProcedureParameter param ) throws LdapException
073 {
074 List<Modification> newMods = new ArrayList<Modification>();
075
076 for ( Modification mod:modifications )
077 {
078 newMods.add( mod.clone() );
079 }
080
081 return newMods;
082 }
083 };
084
085
086 MicroInjector $oldEntryInjector = new MicroInjector()
087 {
088 public Object inject( OperationContext opContext, StoredProcedureParameter param ) throws LdapException
089 {
090 return oldEntry;
091 }
092 };
093
094
095 MicroInjector $newEntryInjector = new MicroInjector()
096 {
097 public Object inject( OperationContext opContext, StoredProcedureParameter param ) throws Exception
098 {
099 return getEntry( opContext );
100 }
101 };
102
103
104 private ClonedServerEntry getEntry( OperationContext opContext ) throws Exception
105 {
106 /**
107 * Using LOOKUP_EXCLUDING_OPR_ATTRS_BYPASS here to exclude operational attributes
108 * especially subentry related ones like "triggerExecutionSubentries".
109 */
110 return opContext.lookup( modifiedEntryName, ByPassConstants.LOOKUP_EXCLUDING_OPR_ATTRS_BYPASS );
111 }
112 }