| /* |
| * 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.sling.models.impl.injectors; |
| |
| import java.lang.reflect.AnnotatedElement; |
| import java.lang.reflect.Type; |
| |
| import javax.servlet.http.HttpServletRequest; |
| import javax.servlet.http.HttpServletResponse; |
| |
| import org.apache.sling.api.SlingHttpServletRequest; |
| import org.apache.sling.api.SlingHttpServletResponse; |
| import org.apache.sling.api.resource.Resource; |
| import org.apache.sling.api.resource.ResourceResolver; |
| import org.apache.sling.api.scripting.SlingBindings; |
| import org.apache.sling.api.scripting.SlingScriptHelper; |
| import org.apache.sling.models.annotations.injectorspecific.InjectionStrategy; |
| import org.apache.sling.models.annotations.injectorspecific.SlingObject; |
| import org.apache.sling.models.spi.AcceptsNullName; |
| import org.apache.sling.models.spi.DisposalCallbackRegistry; |
| import org.apache.sling.models.spi.Injector; |
| import org.apache.sling.models.spi.injectorspecific.AbstractInjectAnnotationProcessor2; |
| import org.apache.sling.models.spi.injectorspecific.InjectAnnotationProcessor2; |
| import org.apache.sling.models.spi.injectorspecific.StaticInjectAnnotationProcessorFactory; |
| import org.jetbrains.annotations.NotNull; |
| import org.osgi.framework.Constants; |
| import org.osgi.service.component.annotations.Component; |
| |
| /** |
| * Injects common Sling objects that can be derived from either a SlingHttpServletRequest, a ResourceResolver or a |
| * Resource. |
| * Documentation see {@link SlingObject}. |
| */ |
| @Component(property=Constants.SERVICE_RANKING+":Integer="+Integer.MAX_VALUE, service={Injector.class, StaticInjectAnnotationProcessorFactory.class, AcceptsNullName.class}) |
| public final class SlingObjectInjector implements Injector, StaticInjectAnnotationProcessorFactory, AcceptsNullName { |
| |
| /** |
| * Injector name |
| */ |
| public static final @NotNull String NAME = "sling-object"; |
| |
| @Override |
| public @NotNull String getName() { |
| return NAME; |
| } |
| |
| @Override |
| public Object getValue(final @NotNull Object adaptable, final String name, final @NotNull Type type, final @NotNull AnnotatedElement element, |
| final @NotNull DisposalCallbackRegistry callbackRegistry) { |
| |
| // only class types are supported |
| if (!(type instanceof Class<?>)) { |
| return null; |
| } |
| Class<?> requestedClass = (Class<?>) type; |
| |
| // validate input |
| if (adaptable instanceof SlingHttpServletRequest) { |
| SlingHttpServletRequest request = (SlingHttpServletRequest) adaptable; |
| if (requestedClass.equals(ResourceResolver.class)) { |
| return request.getResourceResolver(); |
| } |
| if (requestedClass.equals(Resource.class) && element.isAnnotationPresent(SlingObject.class)) { |
| return request.getResource(); |
| } |
| if (requestedClass.equals(SlingHttpServletRequest.class) || requestedClass.equals(HttpServletRequest.class)) { |
| return request; |
| } |
| if (requestedClass.equals(SlingHttpServletResponse.class) |
| || requestedClass.equals(HttpServletResponse.class)) { |
| return getSlingHttpServletResponse(request); |
| } |
| if (requestedClass.equals(SlingScriptHelper.class)) { |
| return getSlingScriptHelper(request); |
| } |
| } else if (adaptable instanceof ResourceResolver) { |
| ResourceResolver resourceResolver = (ResourceResolver) adaptable; |
| if (requestedClass.equals(ResourceResolver.class)) { |
| return resourceResolver; |
| } |
| } else if (adaptable instanceof Resource) { |
| Resource resource = (Resource) adaptable; |
| if (requestedClass.equals(ResourceResolver.class)) { |
| return resource.getResourceResolver(); |
| } |
| if (requestedClass.equals(Resource.class) && element.isAnnotationPresent(SlingObject.class)) { |
| return resource; |
| } |
| } |
| |
| return null; |
| } |
| |
| private SlingScriptHelper getSlingScriptHelper(final SlingHttpServletRequest request) { |
| SlingBindings bindings = (SlingBindings) request.getAttribute(SlingBindings.class.getName()); |
| if (bindings != null) { |
| return bindings.getSling(); |
| } |
| return null; |
| } |
| |
| private SlingHttpServletResponse getSlingHttpServletResponse(final SlingHttpServletRequest request) { |
| SlingScriptHelper scriptHelper = getSlingScriptHelper(request); |
| if (scriptHelper != null) { |
| return scriptHelper.getResponse(); |
| } |
| return null; |
| } |
| |
| @Override |
| @SuppressWarnings({ "unused", "null" }) |
| public InjectAnnotationProcessor2 createAnnotationProcessor(final AnnotatedElement element) { |
| // check if the element has the expected annotation |
| SlingObject annotation = element.getAnnotation(SlingObject.class); |
| if (annotation != null) { |
| return new SlingObjectAnnotationProcessor(annotation); |
| } |
| return null; |
| } |
| |
| private static class SlingObjectAnnotationProcessor extends AbstractInjectAnnotationProcessor2 { |
| |
| private final SlingObject annotation; |
| |
| public SlingObjectAnnotationProcessor(final SlingObject annotation) { |
| this.annotation = annotation; |
| } |
| |
| @Override |
| public InjectionStrategy getInjectionStrategy() { |
| return annotation.injectionStrategy(); |
| } |
| |
| @Override |
| @SuppressWarnings("deprecation") |
| public Boolean isOptional() { |
| return annotation.optional(); |
| } |
| } |
| } |