be.orbinson.aem.groovy.console.servlets.ScriptDownloadServlet.groovy Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of aem-groovy-console-bundle Show documentation

The newest version!

package be.orbinson.aem.groovy.console.servlets

import be.orbinson.aem.groovy.console.audit.AuditService
import be.orbinson.aem.groovy.console.constants.GroovyConsoleConstants
import org.apache.sling.api.SlingHttpServletRequest
import org.apache.sling.api.SlingHttpServletResponse
import org.apache.sling.api.servlets.SlingSafeMethodsServlet
import org.osgi.service.component.annotations.Component
import org.osgi.service.component.annotations.Reference

import javax.servlet.Servlet

import static javax.servlet.http.HttpServletResponse.SC_BAD_REQUEST

@Component(service = Servlet, immediate = true, property = [
        "sling.servlet.paths=/bin/groovyconsole/download"
])
class ScriptDownloadServlet extends SlingSafeMethodsServlet {

    @Reference
    private AuditService auditService

    @Override
    protected void doGet(SlingHttpServletRequest request, SlingHttpServletResponse response) {
        def userId = request.getParameter(GroovyConsoleConstants.USER_ID)
        def script = request.getParameter(GroovyConsoleConstants.SCRIPT)

        def auditRecord = auditService.getAuditRecord(userId, script)

        if (auditRecord) {
            response.setHeader("Content-Disposition", "attachment; filename=${auditRecord.outputFileName}")

            response.contentType = auditRecord.mediaType
            response.characterEncoding = GroovyConsoleConstants.CHARSET
            response.contentLength = auditRecord.output.length()
            response.outputStream.write(auditRecord.output.getBytes(GroovyConsoleConstants.CHARSET))
        } else {
            response.status = SC_BAD_REQUEST
        }
    }
}