001/*
002 *  Copyright 2014 Anyware Services
003 *
004 *  Licensed under the Apache License, Version 2.0 (the "License");
005 *  you may not use this file except in compliance with the License.
006 *  You may obtain a copy of the License at
007 *
008 *      http://www.apache.org/licenses/LICENSE-2.0
009 *
010 *  Unless required by applicable law or agreed to in writing, software
011 *  distributed under the License is distributed on an "AS IS" BASIS,
012 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013 *  See the License for the specific language governing permissions and
014 *  limitations under the License.
015 */
016package org.ametys.core.captcha;
017
018import java.awt.image.BufferedImage;
019import java.io.IOException;
020
021import javax.imageio.ImageIO;
022
023import org.apache.cocoon.ProcessingException;
024import org.apache.cocoon.environment.ObjectModelHelper;
025import org.apache.cocoon.environment.Request;
026import org.apache.cocoon.reading.AbstractReader;
027import org.xml.sax.SAXException;
028
029/**
030 * Read an image captcha given by its key
031 */
032public class BufferedImageReader extends AbstractReader
033{
034    /** The name of the request attribute which must contains the image */
035    public static final String REQUEST_ATTRIBUTE = BufferedImageReader.class.getName();
036    
037    @Override
038    public void generate() throws IOException, SAXException, ProcessingException
039    {
040        Request request = ObjectModelHelper.getRequest(objectModel);
041        
042        BufferedImage image = (BufferedImage) request.getAttribute(REQUEST_ATTRIBUTE);
043        
044        ImageIO.write(image, "PNG", out);
045        out.flush();
046        out.close();
047    }
048
049    @Override
050    public String getMimeType()
051    {
052        return "image/png";
053    }
054}