files.cpp

#include<iostream>
#include<vector>
#include<fstream>
#include<string>
#include<filesystem>
#include<algorithm>
#include "files.h"
#include "parsing.h"
#include "logging.h"
#include "play.h"
#include "platform_utils.h"

using namespace std;
namespace fs = std::filesystem;

vector<string> readFile(string fileName) {
	dbg(LOG_NOTE, "Reading file: " + fileName);
	ifstream file(fileName);
	vector<string> lines;
	string line;
	if (file.is_open()) {
		while (getline(file, line)) {
			lines.push_back(line);
		}
		file.close();
	} else {
		cerr << "Failed to open file: " << fileName << endl;
	}
	return lines;
}


vector<string> sitsToFileStrings(vector<Situation> sits) {
	vector<string> stringsToAdd;
	for (Situation sit : sits) {
		stringsToAdd.push_back("S" + to_string(sit.situationNum) + ". " + sit.title);
		for (Option opt : sit.options) {
			stringsToAdd.push_back("O" + to_string(opt.referSituationNum) + ". " + opt.optionText);
		}
	}

	return stringsToAdd;
}

void writeFile(vector<string> lines, string filename, bool saveToDocuments) {
	// Determine target path
	fs::path filePath(filename);
	if (filePath.extension() != ".story") {
		filePath += ".story";
	}
	if (saveToDocuments) {
		std::string documentsDir = PlatformUtils::getDocumentsPath();
		filePath = fs::path(documentsDir) / filePath.filename();
	}
	ofstream file(filePath, ofstream::out | ofstream::trunc);
	if (!file.is_open()) {
		dbg(LOG_ERROR, "ERROR opening file: " + filePath.string());
		cerr << "Error opening file: " << filePath.string() << endl;
		return;
	}
	for (const auto& line : lines) {
		file << line << "\n";
	}
	file.close();
}

vector<string> getStoryFilesFromDir(string dir) {
	vector<string> fileNames;

	try {
		fs::path path(dir);

		if (!fs::exists(path) || !fs::is_directory(path)) {
			throw runtime_error("Invalid directory path: " + dir);
		}

		// Iterate through directory entries
		for (const auto& entry : fs::directory_iterator(path)) {
			// Check if it's a regular file and has .story extension
			if (entry.is_regular_file() && entry.path().extension() == ".story") {
				// Add the full path as string to our vector
				fileNames.push_back(entry.path().string());
			}
		}

		// Sort the files alphabetically
		sort(fileNames.begin(), fileNames.end());

	} catch (const exception& e) {
		throw runtime_error("Error scanning directory: " + string(e.what()));
	}

	return fileNames;
}

string getShortFileName(const string& path) {
	return fs::path(path).filename().string();
}

void selectNewFile(string& storyPath, vector<string>& lines, vector<Situation>& story, string homeDir) {
	// get files from home dir
	dbg(LOG_NOTE, "Getting .story files from home dir " + homeDir);
	vector<string> fileNames = getStoryFilesFromDir(homeDir);
	if (fileNames.size() == 0) {
		dbg(LOG_ERROR, "No files in home directory");
		// ERROR SAY TO USER
		cout << "\033[2J\033[H"; // clear screen
		cout << "No files found in documents folder, create a new story or download testStory.story from the website.";
        cin.ignore(std::numeric_limits<std::streamsize>::max(), '\n');
		return;
	}
	vector<string> shortFileNames;
	for (int i = 0; i < fileNames.size(); i++) {
		dbg(LOG_NOTE, "Found story file: " + fileNames[i]);
		shortFileNames.push_back(getShortFileName(fileNames[i]));
	}

	dbg(LOG_NOTE, "Asking what file...");
	size_t selectedFileIndex = showOptions(shortFileNames, "Select file:");

	// set as new story
	dbg(LOG_WARNING, "Setting " + fileNames[selectedFileIndex] + " as new story.");
	storyPath = fileNames[selectedFileIndex];
	lines = readFile(storyPath);
	story = parseStoryFile(lines);
	dbg(LOG_NOTE, "Successfully set as new story, going back to main menu.");
}