summaryrefslogtreecommitdiff
path: root/src/core/ResultExtractor.hpp
blob: 916bea7589aee11dbb0c6aa3daab7a5d4c789008 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
// Copyright (C) 2020-2022 Joel Rosdahl and other contributors
//
// See doc/AUTHORS.adoc for a complete list of contributors.
//
// 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, write to the Free Software Foundation, Inc., 51
// Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA

#pragma once

#include "Fd.hpp"

#include <core/Result.hpp>

#include <functional>
#include <optional>
#include <string>

namespace core {

// This class extracts the parts of a result entry to a directory.
class ResultExtractor : public Result::Deserializer::Visitor
{
public:
  using GetRawFilePathFunction = std::function<std::string(uint8_t)>;

  //`result_path` should be the path to the local result entry file if the
  // result comes from local storage.
  ResultExtractor(
    const std::string& output_directory,
    std::optional<GetRawFilePathFunction> get_raw_file_path = std::nullopt);

  void on_embedded_file(uint8_t file_number,
                        Result::FileType file_type,
                        nonstd::span<const uint8_t> data) override;
  void on_raw_file(uint8_t file_number,
                   Result::FileType file_type,
                   uint64_t file_size) override;

private:
  std::string m_output_directory;
  std::optional<GetRawFilePathFunction> m_get_raw_file_path;
};

} // namespace core