org.grible.servlets.app.imp.AnvancedImport.java Source code

Introduction

Here is the source code for org.grible.servlets.app.imp.AnvancedImport.java

Source

/*******************************************************************************
 * Copyright (c) 2013 - 2014 Maksym Barvinskyi.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the GNU Public License v2.0
 * which accompanies this distribution, and is available at
 * http://www.gnu.org/licenses/old-licenses/gpl-2.0.html
 * 
 * Contributors:
 *     Maksym Barvinskyi - initial API and implementation
 ******************************************************************************/
package org.grible.servlets.app.imp;

import java.io.IOException;
import java.io.PrintWriter;

import javax.servlet.ServletException;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.apache.commons.lang3.StringUtils;
import org.grible.dao.PostgresDao;
import org.grible.excel.ExcelFile;
import org.grible.model.Table;
import org.grible.security.Security;
import org.grible.servlets.ServletHelper;
import org.grible.settings.Lang;

/**
 * Servlet implementation class GetStorageValues
 */
@WebServlet("/AnvancedImport")
public class AnvancedImport extends HttpServlet {
    private static final long serialVersionUID = 1L;
    private PostgresDao pDao;

    /**
     * @see HttpServlet#HttpServlet()
     */
    public AnvancedImport() {
        super();
    }

    /**
     * @see HttpServlet#doPost(HttpServletRequest request, HttpServletResponse
     *      response)
     */
    protected void doPost(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException {
        response.setContentType("text/plain");
        PrintWriter out = response.getWriter();
        try {
            if (Security.anyServletEntryCheckFailed(request, response)) {
                return;
            }
            Table currTable = (Table) request.getSession(false).getAttribute("importedTable");
            ExcelFile excelFile = (ExcelFile) request.getSession(false).getAttribute("importedFile");

            String option = request.getParameter("option"); // "addtoend",
            // "addfromrow"

            if (!ServletHelper.isJson()) {
                pDao = new PostgresDao();
            }

            if ("addtoend".equals(option)) {
                String[][] excelValues = excelFile.getValues();
                if (ServletHelper.isJson()) {
                    String[][] oldValues = currTable.getTableJson().getValues();
                    String[][] newValues = new String[oldValues.length + excelValues.length][oldValues[0].length];
                    for (int i = 0; i < newValues.length; i++) {
                        if (i < oldValues.length) {
                            for (int j = 0; j < newValues[0].length; j++) {
                                newValues[i][j] = oldValues[i][j];
                            }
                        } else {
                            for (int j = 0; j < newValues[0].length; j++) {
                                newValues[i][j] = excelValues[i - oldValues.length][j];
                            }
                        }
                    }
                    currTable.getTableJson().setValues(newValues);
                    currTable.save();
                } else {
                    String[][] oldValues = currTable.getValues();
                    String[][] newValues = new String[oldValues.length + excelValues.length][oldValues[0].length];
                    for (int i = 0; i < newValues.length; i++) {
                        if (i < oldValues.length) {
                            for (int j = 0; j < newValues[0].length; j++) {
                                newValues[i][j] = oldValues[i][j];
                            }
                        } else {
                            for (int j = 0; j < newValues[0].length; j++) {
                                newValues[i][j] = excelValues[i - oldValues.length][j];
                            }
                        }
                    }
                    currTable.setValues(newValues);
                    pDao.updateTable(currTable);
                }
            } else if ("addfromrow".equals(option)) {
                int startRow = 0;
                if (request.getParameter("startrow") != null) {
                    if (StringUtils.isNumeric(request.getParameter("startrow"))) {
                        startRow = Integer.parseInt(request.getParameter("startrow"));
                    } else {
                        throw new Exception(Lang.get("error") + ": " + Lang.get("startrownull"));
                    }
                } else {
                    throw new Exception(Lang.get("error") + ": " + Lang.get("startrownull"));
                }

                String[][] excelValues = excelFile.getValues();
                int excelValuesCount = excelValues.length;

                if (ServletHelper.isJson()) {
                    String[][] oldValues = currTable.getTableJson().getValues();

                    int rowsCount = oldValues.length;
                    if (startRow > rowsCount) {
                        throw new Exception(Lang.get("error") + ": " + Lang.get("startrowoutofrange"));
                    }
                    int limit = rowsCount;
                    if (rowsCount < (startRow - 1 + excelValuesCount)) {
                        limit = startRow - 1 + excelValuesCount;
                    }
                    String[][] newValues = new String[limit][oldValues[0].length];

                    for (int i = 0; i < limit; i++) {
                        if ((i < startRow - 1) || (i >= startRow - 1 + excelValuesCount)) {
                            for (int j = 0; j < newValues[0].length; j++) {
                                newValues[i][j] = oldValues[i][j];
                            }
                        } else {
                            for (int j = 0; j < newValues[0].length; j++) {
                                newValues[i][j] = excelValues[i - (startRow - 1)][j];
                            }
                        }
                    }
                    currTable.getTableJson().setValues(newValues);
                    currTable.save();
                } else {
                    String[][] oldValues = currTable.getValues();

                    int rowsCount = oldValues.length;
                    if (startRow > rowsCount) {
                        throw new Exception(Lang.get("error") + ": " + Lang.get("startrowoutofrange"));
                    }
                    int limit = rowsCount;
                    if (rowsCount < (startRow - 1 + excelValuesCount)) {
                        limit = startRow - 1 + excelValuesCount;
                    }
                    String[][] newValues = new String[limit][oldValues[0].length];

                    for (int i = 0; i < limit; i++) {
                        if ((i < startRow - 1) || (i >= startRow - 1 + excelValuesCount)) {
                            for (int j = 0; j < newValues[0].length; j++) {
                                newValues[i][j] = oldValues[i][j];
                            }
                        } else {
                            for (int j = 0; j < newValues[0].length; j++) {
                                newValues[i][j] = excelValues[i - (startRow - 1)][j];
                            }
                        }
                    }
                    currTable.setValues(newValues);
                    pDao.updateTable(currTable);
                }
            }
            request.getSession(true).setAttribute("importedTable", null);
            request.getSession(false).setAttribute("importedFile", null);
            String message = StringUtils.capitalize(currTable.getType().toString().toLowerCase())
                    + " imported successfully.";
            request.getSession(false).setAttribute("importResult", message);
            out.print("success");
        } catch (Exception e) {
            out.print(e.getLocalizedMessage());
            e.printStackTrace();
        }
        out.flush();
        out.close();
    }
}