Tree.cpp

#include <iostream>
#include <vector>
#include <ctime>
#include "map.cpp"
#include "Heap.cpp"
using namespace std;

struct TreeNode
{
    int version_id;
    string content;
    string message;
    time_t created_timestamp;
    time_t snapshot_timestamp;

    // ROLLBACk command needs parent in the case when versionID in not specified
    TreeNode *parent;
    vector<TreeNode *> children;

    TreeNode(int version)
    {
        version_id = version;
        parent = nullptr;
        message = "";
        content = "";
        snapshot_timestamp = 0;
        created_timestamp = time(nullptr);
    }
};

class Tree
{
public:
    TreeNode *root;
    TreeNode *active_version;

    int total_versions;
    map<int, TreeNode *> version_map;

    Tree() {
        root = new TreeNode(0);
        root->message = "Initial Snapshot";
        root->snapshot_timestamp = root->created_timestamp;

        active_version = root;
        version_map[0] = root;
        total_versions = 1;
    }
};