Examples with ReportResults com.tremolosecurity.provisioning.service.util.ReportResults used on opensource projects

Example 1 with ReportResults

use of com.tremolosecurity.provisioning.service.util.ReportResults in project OpenUnison by TremoloSecurity.

the class ScaleMain method generateReport.

private void generateReport(HttpFilterRequest request, HttpFilterResponse response, Gson gson, ReportType reportToRun, AuthInfo userData, Connection db) throws SQLException, IOException {
    PreparedStatement ps = null;
    ResultSet rs = null;
    try {
        if (logger.isDebugEnabled()) {
            logger.debug("Report SQL : '" + reportToRun.getSql() + "'");
        }
        ps = db.prepareStatement(reportToRun.getSql());
        int i = 1;
        for (String paramType : reportToRun.getParamater()) {
            switch(paramType) {
                case "currentUser":
                    String userid = userData.getAttribs().get(this.scaleConfig.getUidAttributeName()).getValues().get(0);
                    if (logger.isDebugEnabled()) {
                        logger.debug("Current User : '" + userid + "'");
                    }
                    ps.setString(i, userid);
                    break;
                case "userKey":
                    if (logger.isDebugEnabled()) {
                        logger.debug("User Key : '" + request.getParameter("userKey") + "'");
                    }
                    ps.setString(i, request.getParameter("userKey").getValues().get(0));
                    break;
                case "beginDate":
                    String beginDate = request.getParameter("beginDate").getValues().get(0);
                    if (logger.isDebugEnabled()) {
                        logger.debug("Begin Date : '" + beginDate + "'");
                    }
                    Date d = new Date(Long.parseLong(beginDate));
                    ps.setDate(i, d);
                    break;
                case "endDate":
                    String endDate = request.getParameter("endDate").getValues().get(0);
                    if (logger.isDebugEnabled()) {
                        logger.debug("End Date : '" + endDate + "'");
                    }
                    Date de = new Date(Long.parseLong(endDate));
                    ps.setDate(i, de);
                    break;
            }
            i++;
        }
        rs = ps.executeQuery();
        String groupingVal = null;
        ReportResults res = new ReportResults();
        res.setName(reportToRun.getName());
        res.setDescription(reportToRun.getDescription());
        res.setDataFields(reportToRun.getDataFields());
        res.setHeaderFields(reportToRun.getHeaderFields());
        res.setGrouping(new ArrayList<ReportGrouping>());
        ReportGrouping grouping = null;
        if (!reportToRun.isGroupings()) {
            grouping = new ReportGrouping();
            grouping.setData(new ArrayList<Map<String, String>>());
            grouping.setHeader(new HashMap<String, String>());
            res.getGrouping().add(grouping);
        }
        logger.debug("Running report");
        while (rs.next()) {
            if (logger.isDebugEnabled()) {
                logger.debug("New row");
            }
            HashMap<String, String> row = new HashMap<String, String>();
            for (String dataField : reportToRun.getDataFields()) {
                if (logger.isDebugEnabled()) {
                    logger.debug("Field - " + dataField + "='" + rs.getString(dataField) + "'");
                }
                row.put(dataField, rs.getString(dataField));
            }
            if (reportToRun.isGroupings()) {
                String rowID = rs.getString(reportToRun.getGroupBy());
                if (logger.isDebugEnabled()) {
                    logger.debug("Grouping Val : '" + groupingVal + "'");
                    logger.debug("Group By : '" + reportToRun.getGroupBy() + "'");
                    logger.debug("Value of Group By in row : '" + rowID + "'");
                }
                if (groupingVal == null || !groupingVal.equals(rowID)) {
                    grouping = new ReportGrouping();
                    grouping.setData(new ArrayList<Map<String, String>>());
                    grouping.setHeader(new HashMap<String, String>());
                    res.getGrouping().add(grouping);
                    for (String headerField : reportToRun.getHeaderFields()) {
                        grouping.getHeader().put(headerField, rs.getString(headerField));
                    }
                    groupingVal = rowID;
                }
            }
            grouping.getData().add(row);
        }
        if (request.getParameter("excel") != null && request.getParameter("excel").getValues().get(0).equalsIgnoreCase("true")) {
            UUID id = UUID.randomUUID();
            String sid = id.toString();
            Map<String, String> map = new HashMap<String, String>();
            map.put("reportid", sid);
            request.getSession().setAttribute(sid, res);
            String json = gson.toJson(map);
            if (logger.isDebugEnabled()) {
                logger.debug("JSON : " + json);
            }
            response.setContentType("application/json");
            ScaleJSUtils.addCacheHeaders(response);
            response.getWriter().print(json);
            response.getWriter().flush();
        } else {
            ProvisioningResult pres = new ProvisioningResult();
            pres.setSuccess(true);
            pres.setReportResults(res);
            String json = gson.toJson(res);
            if (logger.isDebugEnabled()) {
                logger.debug("JSON : " + json);
            }
            response.setContentType("application/json");
            ScaleJSUtils.addCacheHeaders(response);
            response.getWriter().print(json);
            response.getWriter().flush();
        }
    } finally {
        if (rs != null) {
            try {
                rs.close();
            } catch (Throwable t) {
            }
        }
        if (ps != null) {
            try {
                ps.close();
            } catch (Throwable t) {
            }
        }
        if (db != null) {
            try {
                db.close();
            } catch (Throwable t) {
            }
        }
    }
}

Example 2 with ReportResults

use of com.tremolosecurity.provisioning.service.util.ReportResults in project OpenUnison by TremoloSecurity.

the class ScaleMain method exportToExcel.

private void exportToExcel(HttpFilterRequest request, HttpFilterResponse response, Gson gson) throws IOException {
    int lastslash = request.getRequestURI().lastIndexOf('/');
    int secondlastslash = request.getRequestURI().lastIndexOf('/', lastslash - 1);
    String id = request.getRequestURI().substring(secondlastslash + 1, lastslash);
    ReportResults res = (ReportResults) request.getSession().getAttribute(id);
    if (res == null) {
        response.setStatus(404);
        ScaleError error = new ScaleError();
        error.getErrors().add("Report no longer available");
        ScaleJSUtils.addCacheHeaders(response);
        response.getWriter().print(gson.toJson(error).trim());
        response.getWriter().flush();
    } else {
        response.setHeader("Cache-Control", "private, no-store, no-cache, must-revalidate");
        response.setHeader("Pragma", "no-cache");
        Workbook wb = new XSSFWorkbook();
        Font font = wb.createFont();
        font.setBold(true);
        Font titleFont = wb.createFont();
        titleFont.setBold(true);
        titleFont.setFontHeightInPoints((short) 16);
        Sheet sheet = wb.createSheet(WorkbookUtil.createSafeSheetName(res.getName()));
        // Create a header
        Row row = sheet.createRow(0);
        Cell cell = row.createCell(0);
        RichTextString title = new XSSFRichTextString(res.getName());
        title.applyFont(titleFont);
        sheet.addMergedRegion(new CellRangeAddress(0, 0, 0, 3));
        cell.setCellValue(title);
        row = sheet.createRow(1);
        cell = row.createCell(0);
        cell.setCellValue(res.getDescription());
        sheet.addMergedRegion(new CellRangeAddress(1, 1, 0, 3));
        row = sheet.createRow(2);
        cell = row.createCell(0);
        // cell.setCellValue(new DateTime().toString("MMMM Do, YYYY h:mm:ss a"));
        sheet.addMergedRegion(new CellRangeAddress(2, 2, 0, 3));
        row = sheet.createRow(3);
        int rowNum = 4;
        if (res.getGrouping().isEmpty()) {
            row = sheet.createRow(rowNum);
            cell = row.createCell(0);
            cell.setCellValue("There is no data for this report");
        } else {
            for (ReportGrouping group : res.getGrouping()) {
                for (String colHeader : res.getHeaderFields()) {
                    row = sheet.createRow(rowNum);
                    cell = row.createCell(0);
                    RichTextString rcolHeader = new XSSFRichTextString(colHeader);
                    rcolHeader.applyFont(font);
                    cell.setCellValue(rcolHeader);
                    cell = row.createCell(1);
                    cell.setCellValue(group.getHeader().get(colHeader));
                    rowNum++;
                }
                row = sheet.createRow(rowNum);
                int cellNum = 0;
                for (String colHeader : res.getDataFields()) {
                    cell = row.createCell(cellNum);
                    RichTextString rcolHeader = new XSSFRichTextString(colHeader);
                    rcolHeader.applyFont(font);
                    cell.setCellValue(rcolHeader);
                    cellNum++;
                }
                rowNum++;
                for (Map<String, String> dataRow : group.getData()) {
                    cellNum = 0;
                    row = sheet.createRow(rowNum);
                    for (String colHeader : res.getDataFields()) {
                        cell = row.createCell(cellNum);
                        cell.setCellValue(dataRow.get(colHeader));
                        cellNum++;
                    }
                    rowNum++;
                }
                row = sheet.createRow(rowNum);
                rowNum++;
            }
        }
        response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
        wb.write(response.getOutputStream());
    }
}

Example 3 with ReportResults

use of com.tremolosecurity.provisioning.service.util.ReportResults in project OpenUnison by TremoloSecurity.

the class GenerateReport method generateReportData.

private void generateReportData(HttpServletRequest req, HttpServletResponse resp, Gson gson, ReportType reportToRun, Connection db) throws SQLException, IOException {
    PreparedStatement ps;
    ResultSet rs;
    if (logger.isDebugEnabled()) {
        logger.debug("Report SQL : '" + reportToRun.getSql() + "'");
    }
    ps = db.prepareStatement(reportToRun.getSql());
    int i = 1;
    for (String paramType : reportToRun.getParamater()) {
        switch(paramType) {
            case "currentUser":
                if (logger.isDebugEnabled()) {
                    logger.debug("Current User : '" + req.getParameter("currentUser") + "'");
                }
                ps.setString(i, req.getParameter("currentUser"));
                break;
            case "userKey":
                if (logger.isDebugEnabled()) {
                    logger.debug("User Key : '" + req.getParameter("userKey") + "'");
                }
                ps.setString(i, req.getParameter("userKey"));
                break;
            case "beginDate":
                String beginDate = req.getParameter("beginDate");
                if (logger.isDebugEnabled()) {
                    logger.debug("Begin Date : '" + beginDate + "'");
                }
                Date d = new Date(DateTime.parse(beginDate).getMillis());
                ps.setDate(i, d);
                break;
            case "endDate":
                String endDate = req.getParameter("endDate");
                if (logger.isDebugEnabled()) {
                    logger.debug("End Date : '" + endDate + "'");
                }
                Date de = new Date(DateTime.parse(endDate).getMillis());
                ps.setDate(i, de);
                break;
        }
        i++;
    }
    rs = ps.executeQuery();
    String groupingVal = null;
    ReportResults res = new ReportResults();
    res.setName(reportToRun.getName());
    res.setDescription(reportToRun.getDescription());
    res.setDataFields(reportToRun.getDataFields());
    res.setHeaderFields(reportToRun.getHeaderFields());
    res.setGrouping(new ArrayList<ReportGrouping>());
    ReportGrouping grouping = null;
    if (!reportToRun.isGroupings()) {
        grouping = new ReportGrouping();
        grouping.setData(new ArrayList<Map<String, String>>());
        grouping.setHeader(new HashMap<String, String>());
        res.getGrouping().add(grouping);
    }
    logger.debug("Running report");
    while (rs.next()) {
        if (logger.isDebugEnabled()) {
            logger.debug("New row");
        }
        HashMap<String, String> row = new HashMap<String, String>();
        for (String dataField : reportToRun.getDataFields()) {
            if (logger.isDebugEnabled()) {
                logger.debug("Field - " + dataField + "='" + rs.getString(dataField) + "'");
            }
            row.put(dataField, rs.getString(dataField));
        }
        if (reportToRun.isGroupings()) {
            String rowID = rs.getString(reportToRun.getGroupBy());
            if (logger.isDebugEnabled()) {
                logger.debug("Grouping Val : '" + groupingVal + "'");
                logger.debug("Group By : '" + reportToRun.getGroupBy() + "'");
                logger.debug("Value of Group By in row : '" + rowID + "'");
            }
            if (groupingVal == null || !groupingVal.equals(rowID)) {
                grouping = new ReportGrouping();
                grouping.setData(new ArrayList<Map<String, String>>());
                grouping.setHeader(new HashMap<String, String>());
                res.getGrouping().add(grouping);
                for (String headerField : reportToRun.getHeaderFields()) {
                    grouping.getHeader().put(headerField, rs.getString(headerField));
                }
                groupingVal = rowID;
            }
        }
        grouping.getData().add(row);
    }
    ProvisioningResult pres = new ProvisioningResult();
    pres.setSuccess(true);
    pres.setReportResults(res);
    String json = gson.toJson(pres);
    if (logger.isDebugEnabled()) {
        logger.debug("JSON : " + json);
    }
    resp.getOutputStream().print(json);
}