report_exporter.cpp

Go to the documentation of this file.

#include "report_exporter.hpp"
#include "i18n.hpp"
 
#include <fstream>
#include <sstream>
 
namespace {
 
    std::string escape_json(const std::string& s) {
        std::ostringstream oss;
        for (char c : s) {
            switch (c) {
                case '"':  oss << "\\\""; break;
                case '\\': oss << "\\\\"; break;
                case '\b': oss << "\\b";  break;
                case '\f': oss << "\\f";  break;
                case '\n': oss << "\\n";  break;
                case '\r': oss << "\\r";  break;
                case '\t': oss << "\\t";  break;
                default:
                    if (static_cast<unsigned char>(c) < 0x20) {
                        oss << "\\u";
                        const char* hex = "0123456789abcdef";
                        oss << '0' << '0'
                        << hex[(static_cast<unsigned char>(c) >> 4) & 0xF]
                        << hex[static_cast<unsigned char>(c) & 0xF];
                    } else {
                        oss << c;
                    }
            }
        }
        return oss.str();
    }
 
    const char* kind_to_string(ResultKind kind) {
        switch (kind) {
            case ResultKind::Partition:
                return "Partition";
            case ResultKind::LostPartitionCandidate:
                return "LostPartitionCandidate";
            case ResultKind::FilesystemCandidate:
                return "FilesystemCandidate";
            case ResultKind::FilesystemVolume:
                return "FilesystemVolume";
            case ResultKind::NtfsVolume:
                return "NtfsVolume";
            case ResultKind::NtfsMetadata:
                return "NtfsMetadata";
            case ResultKind::Info:
                return "Info";
            case ResultKind::Warning:
                return "Warning";
            case ResultKind::Error:
                return "Error";
            default:
                return tr("Unknown");
        }
    }
 
} // namespace
 
bool ReportExporter::save_json(const ScanReport& report,
                               const std::string& path,
                               std::string& error_out) {
    std::ofstream out(path, std::ios::binary);
    if (!out) {
        error_out = trs("Unable to open destination file");
        return false;
    }
 
    out << "{\n";
    out << "  \"input_path\": \"" << escape_json(report.input_path) << "\",\n";
    out << "  \"input_size\": " << report.input_size << ",\n";
    out << "  \"logs\": [\n";
    for (size_t i = 0; i < report.logs.size(); ++i) {
        out << "    \"" << escape_json(report.logs[i]) << "\""
        << (i + 1 < report.logs.size() ? "," : "") << "\n";
    }
    out << "  ],\n";
    out << "  \"results\": [\n";
    for (size_t i = 0; i < report.results.size(); ++i) {
        const auto& r = report.results[i];
        out << "    {\n";
        out << "      \"kind\": \"" << kind_to_string(r.kind) << "\",\n";
        out << "      \"offset_bytes\": " << r.offset_bytes << ",\n";
        out << "      \"size_bytes\": " << r.size_bytes << ",\n";
        out << "      \"type\": \"" << escape_json(r.type) << "\",\n";
        out << "      \"source\": \"" << escape_json(r.source) << "\",\n";
        out << "      \"detail\": \"" << escape_json(r.detail) << "\",\n";
        out << "      \"confidence\": " << r.confidence << "\n";
        out << "    }" << (i + 1 < report.results.size() ? "," : "") << "\n";
    }
    out << "  ]\n";
    out << "}\n";
    return true;
                               }
 
                               bool ReportExporter::save_text(const ScanReport& report,
                                                              const std::string& path,
                                                              std::string& error_out) {
                                   std::ofstream out(path);
                                   if (!out) {
                                       error_out = trs("Unable to open destination file");
                                       return false;
                                   }
 
                                   out << "BastionGuard Recovery Data\n";
                                   out << "Input path: " << report.input_path << "\n";
                                   out << "Input size: " << report.input_size << " bytes\n\n";
                                   out << "Logs\n----\n";
                                   for (const auto& line : report.logs) {
                                       out << line << '\n';
                                   }
 
                                   out << "\nResults\n-------\n";
                                   for (const auto& r : report.results) {
                                       out << '[' << kind_to_string(r.kind) << "] "
                                       << r.type
                                       << " | source=" << r.source
                                       << " | offset=" << r.offset_bytes
                                       << " | size=" << r.size_bytes
                                       << " | confidence=" << r.confidence
                                       << " | detail=" << r.detail
                                       << '\n';
                                   }
 
                                   return true;
                                                              }