package nl.astraeus.jdbc.web;

import nl.astraeus.jdbc.util.IOUtils;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.InputStream;

/**
 * User: rnentjes
 * Date: 3/28/12
 * Time: 3:05 PM
 */
public class ResourceServlet extends HttpServlet {

    @Override
    protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
        String uri = req.getRequestURI();

        uri = "nl/astraeus/jdbc/web" + uri;
        InputStream in = Thread.currentThread().getContextClassLoader().getResourceAsStream(uri);

        if (in == null) {
            resp.sendError(404, "Cannot find resource '" + uri + "'.");
        } else {
            try {
                if (uri.endsWith("js")) {
                    resp.setContentType("text/javascript");
                } else if (uri.endsWith("css")) {
                    resp.setContentType("text/css");
                } else if (uri.endsWith("png")) {
                    resp.setContentType("image/png");
                } else if (uri.endsWith("jpg")) {
                    resp.setContentType("image/jpeg");
                } else if (uri.endsWith("gif")) {
                    resp.setContentType("image/gif");
                }

                IOUtils.copy(in, resp.getOutputStream());
            } finally {
                in.close();
            }
        }
    }

}