Java tutorial
/* * Copyright (C) 2010 {Apertum}Projects. web: www.apertum.ru email: info@apertum.ru * * This program is free software: you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation, either version 3 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program. If not, see <http://www.gnu.org/licenses/>. */ package ru.apertum.qsystem.reports.net; import java.io.File; import java.io.FileInputStream; import java.io.FileNotFoundException; import java.io.IOException; import java.io.InputStream; import java.io.OutputStream; import java.util.Locale; import org.apache.http.HttpException; import org.apache.http.HttpRequest; import org.apache.http.HttpResponse; import org.apache.http.HttpStatus; import org.apache.http.MethodNotSupportedException; import org.apache.http.entity.EntityTemplate; import org.apache.http.protocol.HttpContext; import org.apache.http.protocol.HttpRequestHandler; import ru.apertum.qsystem.client.forms.FAbout; import ru.apertum.qsystem.common.Uses; import ru.apertum.qsystem.common.QLog; import ru.apertum.qsystem.reports.common.RepResBundle; import ru.apertum.qsystem.reports.common.Response; import ru.apertum.qsystem.reports.model.QReportsList; /** * * @author Evgeniy Egorov */ public class HttpQSystemReportsHandler implements HttpRequestHandler { @Override public void handle(HttpRequest request, HttpResponse response, HttpContext context) throws HttpException, IOException { final String method = request.getRequestLine().getMethod().toUpperCase(Locale.ENGLISH); if (!method.equals("GET") && !method.equals("HEAD") && !method.equals("POST")) { throw new MethodNotSupportedException(method + " method not supported"); } // ?? ? Response result = QReportsList.getInstance().generate(request); // ? if (result == null) { String subject = NetUtil.getUrl(request); if ("/".equals(subject)) { subject = "/login.html"; } int dot = subject.lastIndexOf("."); final String ext = subject.substring(dot + 1); String contentType = "; charset=UTF-8"; if ("htm".equals(ext) || "html".equals(ext)) { contentType = "text/" + ext + contentType; } else { if ("pdf".equals(ext) || "rtf".equals(ext) || "doc".equals(ext) || "xlsx".equals(ext)) { contentType = "application/" + ext + contentType; } else { if ("gif".equals(ext) || "jpeg".equals(ext) || "jpg".equals(ext) || "ico".equals(ext) || "xpm".equals(ext)) { contentType = "image/" + ext + contentType; } } } // ?? "/ru/apertum/qsystem/reports/web/" final InputStream inStream = getClass() .getResourceAsStream("/ru/apertum/qsystem/reports/web" + subject); if (inStream == null) { QLog.l().logRep() .warn("?? : \"/ru/apertum/qsystem/reports/web" + subject + "\""); // ??, // subject = subject.substring(1); File anyFile = new File(subject); if (anyFile.exists()) { QLog.l().logRep().info(" : \"" + subject + "\""); FileInputStream fInStream = null; try { fInStream = new FileInputStream(anyFile); } catch (FileNotFoundException ex) { final String err = " \"" + subject + "\" "; QLog.l().logRep().error("err " + ex); result = new Response((err + ex).getBytes(), contentType); response.setStatusCode(HttpStatus.SC_INTERNAL_SERVER_ERROR); } try { result = new Response(Uses.readInputStream(fInStream), contentType); response.setStatusCode(HttpStatus.SC_OK); } catch (IOException ex) { final String err = " \"" + subject + "\" "; QLog.l().logRep().error("err " + ex); result = new Response((err + ex).getBytes(), contentType); response.setStatusCode(HttpStatus.SC_INTERNAL_SERVER_ERROR); } } else { // anyFile = new File( Uses.TEMP_FOLDER + File.separator + "temphtml.html_files" + File.separator + subject); if (anyFile.exists()) { QLog.l().logRep().info(" : \"" + Uses.TEMP_FOLDER + File.separator + "temphtml.html_files" + File.separator + subject + "\""); FileInputStream fInStream = null; try { fInStream = new FileInputStream(anyFile); } catch (FileNotFoundException ex) { final String err = " \"" + Uses.TEMP_FOLDER + File.separator + "temphtml.html_files" + File.separator + subject + "\" "; QLog.l().logRep().error("err " + ex); result = new Response((err + ex).getBytes(), contentType); response.setStatusCode(HttpStatus.SC_INTERNAL_SERVER_ERROR); } try { result = new Response(Uses.readInputStream(fInStream), contentType); response.setStatusCode(HttpStatus.SC_OK); } catch (IOException ex) { final String err = " \"" + Uses.TEMP_FOLDER + File.separator + "temphtml.html_files" + File.separator + subject + "\" "; QLog.l().logRep().error("err " + ex); result = new Response((err + ex).getBytes(), contentType); response.setStatusCode(HttpStatus.SC_INTERNAL_SERVER_ERROR); } anyFile.delete(); } else { // . ? ?? ( ???) QLog.l().logRep() .error("?? : \"" + Uses.TEMP_FOLDER + File.separator + "temphtml.html_files" + File.separator + subject + "\""); final String s = "<html><head><meta http-equiv = \"Content-Type\" content = \"text/html; charset=utf-8\" ></head><p align=center>?? .</p></html>"; result = new Response(s.getBytes()); } } } else { QLog.l().logRep().info(" ??: \"" + subject + "\""); try { result = new Response(Uses.readInputStream(inStream)); if ("/login.html".equals(subject)) { result.setData( RepResBundle.getInstance().prepareString(new String(result.getData(), "UTF-8")) .replaceFirst(Uses.ANCHOR_USERS_FOR_REPORT, QReportsList.getInstance().getHtmlUsersList()) .replaceFirst(Uses.ANCHOR_PROJECT_NAME_FOR_REPORT, Uses.getLocaleMessage("project.name" + FAbout.getCMRC_SUFF())) .getBytes("UTF-8")); //"Cp1251" } } catch (IOException ex) { QLog.l().logRep().error(" ? ??. " + ex); result = new Response((" ? ??. " + ex).getBytes()); response.setStatusCode(HttpStatus.SC_INTERNAL_SERVER_ERROR); } response.setStatusCode(HttpStatus.SC_OK); } } // : QLog.l().logRep().trace(" " + result.getData().length + " ? \"" + request.getRequestLine().getUri() + "\"."); final byte[] result2 = result.getData(); final EntityTemplate body = new EntityTemplate((final OutputStream outstream) -> { outstream.write(result2); outstream.flush(); }); body.setContentType(result.getContentType()); response.setEntity(body); NetUtil.freeEntityContent(request); } }