FileSystem.cpp

#include "FileSystem.hpp"
#include <iostream>
using namespace std;

void FileSystem::createFile(string filename){
    auto it = files.find(filename);
    if (it == files.end()){
        cout << "File not found" << endl;
        return;
    }
    File* newFile = new File(filename);
    files[filename] = newFile;
    cout << "File created: " << filename << endl;
}

void FileSystem::readFile(string filename){
    auto it = files.find(filename);
    if (it == files.end()){
        cout << "File not found" << endl;
        return;
    }
    it->second->read();
}

void FileSystem::insertContent(string filename, string content){
    auto it = files.find(filename);
    if (it == files.end()){
        cout << "File not found" << endl;
        return;
    }
    it->second->insert(content);
    cout << "Content inserted into " << filename << endl;
}

void FileSystem::updateFile(string filename, string content){
    auto it = files.find(filename);
    if (it == files.end()){
        cout << "File not found" << endl;
        return;
    }
    it->second->update(content);
    cout << "File updated: " << filename << endl;
}

void FileSystem::createSnapshot(string filename, string message){
    auto it = files.find(filename);
    if (it == files.end()){
        cout << "File not found" << endl;
        return;
    }
    File* file = it->second;
    if (!file->activeVersion){
        cout << "No active version to snapshot" << endl;
        return;
    }
    file->activeVersion->isSnapshot = true;
    file->activeVersion->message = message;
    file->activeVersion->snapshotTimestamp = time(0);
    cout << "Snapshot created for " << filename << " with message: " << message << endl;
}

void FileSystem::rollbackToSnapshot(string filename, int versionID){
    auto it = files.find(filename);
    if (it == files.end()){
        cout << "File not found" << endl;
        return;
    }
    if (it->second->rollback(versionID)){
        cout << "Rolled back " << filename << " to version " << versionID << endl;
    } else {
        cout << "Snapshot version not found" << endl;
    }
    // if version id is not provided, rollback to parent of current active version
    if (versionID == -1 && it->second->activeVersion){
        Version* parent = it->second->activeVersion->parent;
        if (parent && parent->isSnapshot){
            it->second->activeVersion = parent;
            cout << "Rolled back " << filename << " to previous snapshot version " << parent->versionID << endl;
        } else {
            cout << "No previous snapshot version to rollback to" << endl;
        }
    }
}

void FileSystem::viewHistory(string filename){
    auto it = files.find(filename);
    if (it == files.end()){
        cout << "File not found" << endl;
        return;
    }
    it->second->viewHistory();
}

void FileSystem::listRecentFiles(int num) {
    vector<File*> fileList;
    for (const auto& pair : files) {
        fileList.push_back(pair.second);
    }
    sort(fileList.begin(), fileList.end(), [](File* a, File* b) {
        return a->getLastModified() > b->getLastModified();
    });
    cout << "Most recently edited files:" << endl;
    for (int i = 0; i < min(num, (int)fileList.size()); ++i) {
        cout << fileList[i]->getName() << " (Last Modified: " << fileList[i]->getLastModified() << ")" << endl;
    }
}

void FileSystem::listMostVersionedFiles(int num) {
    vector<File*> fileList;
    for (const auto& pair : files) {
        fileList.push_back(pair.second);
    }
    sort(fileList.begin(), fileList.end(), [](File* a, File* b) {
        return a->getVersionCount() > b->getVersionCount();
    });
    cout << "Files with most versions:" << endl;
    for (int i = 0; i < min(num, (int)fileList.size()); ++i) {
        cout << fileList[i]->getName() << " (Versions: " << fileList[i]->getVersionCount() << ")" << endl;
    }
}

void FileSystem::getVersionCount(string filename) {
    auto it = files.find(filename);
    if (it == files.end()){
        cout << "File not found" << endl;
        return;
    }
}