aboutsummaryrefslogtreecommitdiff
path: root/libs/installer_bsplugins/src/TESData/FileEntry.h
blob: 0ee692259ad60defd4e8cb26470806c73e86608f (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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
#ifndef TESDATA_FILEENTRY_H
#define TESDATA_FILEENTRY_H

#include "Record.h"
#include "RecordPath.h"
#include "TESFile/Type.h"

#include <boost/container/flat_map.hpp>

#include <cstdint>
#include <functional>
#include <memory>
#include <optional>
#include <shared_mutex>
#include <string>
#include <variant>
#include <vector>

namespace cont = boost::container;

namespace TESData
{

using TESFileHandle = int;

class FileEntry final
{
public:
  struct TreeItem
  {
    using Key =
        std::variant<TESFile::GroupData, std::uint32_t, std::string, TESFile::Type>;

    const TreeItem* parent{nullptr};
    std::string name;
    TESFile::Type formType{};
    std::optional<TESFile::GroupData> group;
    std::shared_ptr<Record> record;
    cont::flat_map<Key, std::shared_ptr<TreeItem>> children;
  };

  FileEntry(TESFileHandle handle, const std::string& name);

  [[nodiscard]] TESFileHandle handle() const { return m_Handle; }
  [[nodiscard]] const std::string& name() const { return m_Name; }
  [[nodiscard]] TreeItem* dataRoot() const { return m_Root.get(); }
  [[nodiscard]] const std::vector<std::string>& files() const { return m_Files; }

  void
  forEachRecord(std::function<void(const std::shared_ptr<const Record>&)> func) const;

  std::shared_ptr<Record> createRecord(const RecordPath& path, const std::string& name,
                                       TESFile::Type formType);
  void addRecord(const RecordPath& path, const std::string& name,
                 TESFile::Type formType, std::shared_ptr<Record> record);
  void addChildGroup(const RecordPath&);

  [[nodiscard]] std::shared_ptr<Record> findRecord(const RecordPath& path) const;
  [[nodiscard]] std::shared_ptr<TreeItem> findItem(const RecordPath& path) const;

private:
  std::shared_ptr<TreeItem> createHierarchy(const RecordPath& path);

  TESFileHandle m_Handle;
  std::string m_Name;
  std::shared_ptr<TreeItem> m_Root;
  std::vector<std::string> m_Files;
  mutable std::shared_mutex m_Mutex;
};

}  // namespace TESData

#endif  // TESDATA_FILEENTRY_H