/*
* #%L
* ACS AEM Tools Bundle
* %%
* Copyright (C) 2013 Adobe
* %%
* Licensed 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.
* #L%
*/
package com.adobe.acs.tools.fiddle.impl;
import org.apache.felix.scr.annotations.Reference;
import org.apache.felix.scr.annotations.sling.SlingServlet;
import org.apache.sling.api.SlingConstants;
import org.apache.sling.api.SlingHttpServletRequest;
import org.apache.sling.api.SlingHttpServletResponse;
import org.apache.sling.api.request.RequestDispatcherOptions;
import org.apache.sling.api.resource.Resource;
import org.apache.sling.api.resource.ResourceResolver;
import org.apache.sling.api.servlets.SlingAllMethodsServlet;
import org.apache.sling.api.wrappers.SlingHttpServletRequestWrapper;
import org.osgi.service.event.Event;
import org.osgi.service.event.EventAdmin;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import javax.jcr.Node;
import javax.jcr.RepositoryException;
import javax.jcr.Session;
import javax.servlet.RequestDispatcher;
import javax.servlet.ServletException;
import java.io.IOException;
import java.util.Collections;
import java.util.Iterator;
import java.util.Map;
@SuppressWarnings("serial")
@SlingServlet(resourceTypes = "acs-tools/components/aemfiddle", selectors = "run", methods = "POST")
public class RunFiddleServlet extends SlingAllMethodsServlet {
private static final Logger log = LoggerFactory.getLogger(RunFiddleServlet.class);
private static final String VAR_CLASSES = "/var/classes";
private static final String COMPILED_JSP = "org/apache/jsp/apps/acs_002dtools/components/aemfiddle";
@Reference
private EventAdmin eventAdmin;
@Override
protected void doPost(SlingHttpServletRequest request, SlingHttpServletResponse response)
throws ServletException, IOException {
// Clear any previously compiled fiddle scripts as they have a tendency to execute on first executions
this.clearCompiledFiddle(request.getResourceResolver());
final Resource resource = getResource(request);
final String data = request.getParameter("scriptdata");
final String ext = request.getParameter("scriptext");
InMemoryScript script = InMemoryScript.set(ext, data);
// doing this as a synchronous event so we ensure that
// the JSP has been invalidated
Map<String, String> props = Collections.singletonMap(
SlingConstants.PROPERTY_PATH, script.getPath());
eventAdmin.sendEvent(new Event(SlingConstants.TOPIC_RESOURCE_CHANGED, props));
final RequestDispatcherOptions options = new RequestDispatcherOptions();
options.setForceResourceType(Constants.PSEDUO_COMPONENT_PATH);
options.setReplaceSelectors("");
// Suppress ACS AEM Commons - Component Error Handler from capturing errors
request.setAttribute("com.adobe.acs.commons.wcm.component-error-handler.suppress", true);
RequestDispatcher dispatcher = request.getRequestDispatcher(resource, options);
GetRequest getRequest = new GetRequest(request);
dispatcher.forward(getRequest, response);
}
private Resource getResource(SlingHttpServletRequest request) {
String path = request.getParameter("resource");
if (path == null || "".equals(path)) {
return request.getResource();
} else {
Resource resource = request.getResourceResolver().resolve(path);
if (resource != null) {
return resource;
} else {
return request.getResource();
}
}
}
private void clearCompiledFiddle(final ResourceResolver resourceResolver) {
final Resource varClasses = resourceResolver.getResource(VAR_CLASSES);
final Session session = resourceResolver.adaptTo(Session.class);
if (varClasses != null) {
/* /var/classes structure builds out under a UUID that is different between AEM instances */
final Iterator<Resource> iterator = varClasses.listChildren();
while (iterator.hasNext()) {
final Resource varClass = iterator.next();
final Resource fiddleResource = varClass.getChild(COMPILED_JSP);
if (fiddleResource != null) {
final Node node = fiddleResource.adaptTo(Node.class);
if (node != null) {
try {
/* Removed the aemfiddle folder that contains the script */
node.remove();
} catch (RepositoryException e) {
log.error("Could not remove compiled AEM Fiddle scripts: {}", e.getMessage());
}
}
}
}
try {
/* Only save if something was removed */
if (session.hasPendingChanges()) {
session.save();
}
} catch (RepositoryException e) {
log.error("Could not save removal of compiled AEM Fiddle scripts: {}", e.getMessage());
}
}
}
private static class GetRequest extends SlingHttpServletRequestWrapper {
public GetRequest(SlingHttpServletRequest wrappedRequest) {
super(wrappedRequest);
}
@Override
public String getMethod() {
return "GET";
}
}
}