mirror of
https://github.com/CheckPointSW/InviZzzible
synced 2026-06-08 10:35:43 +00:00
chkp-alexeybu
f6a9ebceef
Removed configuration file for the HTML report template Scripts migrated to Python 3
96 lines
2.1 KiB
C++
96 lines
2.1 KiB
C++
#include "report.h"
|
|
#include "helper.h"
|
|
#include <fstream>
|
|
#include <sstream>
|
|
|
|
|
|
bool Report::load() {
|
|
// loading content of files
|
|
std::ifstream ifsb(bootstrap_f);
|
|
if (!ifsb.is_open()) {
|
|
// load from the pre generated file
|
|
#include "data_bootstrap_css.h"
|
|
bootstrap_content = std::string(bootstrap_data);
|
|
}
|
|
else {
|
|
bootstrap_content.assign((std::istreambuf_iterator<char>(ifsb)), (std::istreambuf_iterator<char>()));
|
|
}
|
|
|
|
std::ifstream ifsr(report_f);
|
|
if (!ifsr.is_open()) {
|
|
#include "data_report_html.h"
|
|
report = std::string(report_data);
|
|
}
|
|
else {
|
|
report.assign((std::istreambuf_iterator<char>(ifsr)), (std::istreambuf_iterator<char>()));
|
|
}
|
|
|
|
// find bootstrap expression
|
|
return string_replace_substring(report, bootstrap_s, bootstrap_content);
|
|
}
|
|
|
|
bool Report::dump(const std::string &fname) const {
|
|
std::ofstream ofs(fname);
|
|
if (!ofs.is_open())
|
|
return false;
|
|
|
|
std::string freport = report;
|
|
if (!string_replace_substring(freport, module_s, ""))
|
|
return false;
|
|
|
|
ofs << freport;
|
|
ofs.close();
|
|
|
|
return true;
|
|
}
|
|
|
|
bool Report::add_entry(const std::vector<std::string> &columns) {
|
|
if (columns.size() != column_names.size() && columns.size() > 3)
|
|
return false;
|
|
|
|
std::stringstream table_entry;
|
|
|
|
// FIXME: make entries instead of numbers
|
|
if (columns[3] == "YES")
|
|
table_entry << "<tr class = \"danger\">";
|
|
else if (columns[3] == "NO")
|
|
table_entry << "<tr class = \"success\">";
|
|
else return false;
|
|
|
|
// generate table report entry
|
|
for (auto &c : columns) {
|
|
table_entry << "\n<td>" << c << "</td>\n";
|
|
}
|
|
|
|
table_entry << "</tr>\n";
|
|
table_entries.push_back(table_entry.str());
|
|
|
|
return true;
|
|
}
|
|
|
|
bool Report::flush(const std::string &header_name) {
|
|
std::stringstream s;
|
|
|
|
// generate report for specific module
|
|
s << "<h2 class = \"text-primary\">" << header_name << "</h2>\n";
|
|
s << "<table class=\"table\">\n";
|
|
s << "<thead>\n<tr>\n";
|
|
|
|
for (auto &cn : column_names)
|
|
s << "<th>" << cn << "</th>\n";
|
|
|
|
s << "</tr>\n</thead>\n";
|
|
|
|
s << "<tbody>\n";
|
|
|
|
for (auto &te : table_entries)
|
|
s << te << "\n";
|
|
|
|
s << "</tbody>\n</table>\n";
|
|
|
|
table_entries = {};
|
|
|
|
// add to current report data
|
|
return string_replace_substring(report, module_s, s.str() + module_s);
|
|
}
|