main.cpp

#include <iostream>
#include <vector>
#include <chrono>
#include <thread>
#include <limits>
#include "Project.h"
#include "Task.h"
#include "FileManager.h"

void clearScreen() {
    #ifdef _WIN32
        system("cls");
    #else
        system("clear");
    #endif
}

void beep() {
    std::cout << '\a' << std::flush;
    for (int i = 0; i < 7; i++) {
        std::cout << '\a' << std::flush;
        std::this_thread::sleep_for(std::chrono::milliseconds(200));
    }
}

void startTimer(std::chrono::minutes duration) {
    int seconds = duration.count() * 60;
    while (seconds > 0) {
        std::cout << "Time remaining: " << seconds << " seconds\r" << std::flush;
        std::this_thread::sleep_for(std::chrono::seconds(1));
        seconds--;
    }
    std::cout << "\nTime's up!" << std::endl;
    beep();
}

void showMainMenu(std::vector<Project>& projects) {
    clearScreen();
    std::cout << "Please choose one of the following options:\n\n";
    std::cout << "1. View existing projects\n";
    std::cout << "2. Add new project\n";
    std::cout << "3. Select project\n";
    std::cout << "4. Exit\n\n";
    std::cout << "Enter your choice: ";
}

void showProjects(const std::vector<Project>& projects) {
    clearScreen();
    std::cout << "List of projects:\n\n";
    for (size_t i = 0; i < projects.size(); i++) {
        std::cout << i+1 << ". " << projects[i].getName() 
                  << " (Total time: " << projects[i].getTotalTimeSpent().count() << " mins)\n";
    }
    std::cout << "\nPress enter to continue...";
    std::cin.ignore();
    std::cin.get();
}

void addProject(std::vector<Project>& projects) {
    clearScreen();
    std::string name;
    std::cout << "Please enter name of the project: ";
    std::cin.ignore();
    std::getline(std::cin, name);
    projects.emplace_back(name);
    FileManager::saveProjects(projects);
    std::cout << "\nProject added successfully!\n";
    std::this_thread::sleep_for(std::chrono::seconds(1));
}

void showProjectMenu(Project& project) {
    clearScreen();
    std::cout << "Project: " << project.getName() << "\n\n";
    std::cout << "1. List all tasks\n";
    std::cout << "2. Add task\n";
    std::cout << "3. Choose task for focus session\n";
    std::cout << "4. Back to main menu\n\n";
    std::cout << "Enter your choice: ";
}

void showTasks(Project& project) {
    clearScreen();
    auto& tasks = project.getTasks();
    std::cout << "Tasks in project " << project.getName() << ":\n\n";
    for (size_t i = 0; i < tasks.size(); i++) {
        std::cout << i+1 << ". " << tasks[i].getName() 
                  << " (Time spent: " << tasks[i].getTimeSpent().count() << " mins)\n";
    }
    std::cout << "\nPress enter to continue...";
    std::cin.ignore();
    std::cin.get();
}

void addTask(Project& project, std::vector<Project>& projects) {
    clearScreen();
    std::string name, details;
    std::cout << "Please enter task name: ";
    std::cin.ignore();
    std::getline(std::cin, name);
    std::cout << "Enter task details (optional): ";
    std::getline(std::cin, details);
    
    project.addTask(Task(name, details));
    FileManager::saveProjects(projects);
    std::cout << "\nTask added successfully!\n";
    std::this_thread::sleep_for(std::chrono::seconds(1));
}

void startFocusSession(Project& project, std::vector<Project>& projects) {
    auto& tasks = project.getTasks();
    if (tasks.empty()) {
        std::cout << "No tasks available in this project.\n";
        std::this_thread::sleep_for(std::chrono::seconds(1));
        return;
    }

    clearScreen();
    std::cout << "Select a task for focus session:\n\n";
    for (size_t i = 0; i < tasks.size(); i++) {
        std::cout << i+1 << ". " << tasks[i].getName() << "\n";
    }
    
    int choice;
    std::cout << "\nEnter task number: ";
    std::cin >> choice;
    
    if (choice < 1 || choice > static_cast<int>(tasks.size())) {
        std::cout << "Invalid choice!\n";
        std::this_thread::sleep_for(std::chrono::seconds(1));
        return;
    }
    
    Task& selectedTask = tasks[choice-1];
    
    int minutes;
    std::cout << "Enter the amount of time to focus (minutes): ";
    std::cin >> minutes;
    
    startTimer(std::chrono::minutes(minutes));
    selectedTask.addTimeSpent(std::chrono::minutes(minutes));
    FileManager::saveProjects(projects);
    
    char restart;
    std::cout << "Do you want to restart focus time on same task? (y/n): ";
    std::cin >> restart;
    
    if (restart == 'y' || restart == 'Y') {
        startFocusSession(project, projects);
    }
}

int main() {
    std::vector<Project> projects = FileManager::loadProjects();
    
    while (true) {
        showMainMenu(projects);
        
        int choice;
        std::cin >> choice;
        
        if (std::cin.fail()) {
            std::cin.clear();
            std::cin.ignore(std::numeric_limits<std::streamsize>::max(), '\n');
            continue;
        }
        
        switch (choice) {
            case 1:
                showProjects(projects);
                break;
            case 2:
                addProject(projects);
                break;
            case 3: {
                if (projects.empty()) {
                    std::cout << "No projects available. Please add a project first.\n";
                    std::this_thread::sleep_for(std::chrono::seconds(1));
                    break;
                }
                
                showProjects(projects);
                clearScreen();
                std::cout << "Select a project:\n\n";
                for (size_t i = 0; i < projects.size(); i++) {
                    std::cout << i+1 << ". " << projects[i].getName() << "\n";
                }
                
                int projChoice;
                std::cout << "\nEnter project number: ";
                std::cin >> projChoice;
                
                if (projChoice < 1 || projChoice > static_cast<int>(projects.size())) {
                    std::cout << "Invalid choice!\n";
                    std::this_thread::sleep_for(std::chrono::seconds(1));
                    break;
                }
                
                Project& selectedProject = projects[projChoice-1];
                
                while (true) {
                    showProjectMenu(selectedProject);
                    std::cin >> choice;
                    
                    if (std::cin.fail()) {
                        std::cin.clear();
                        std::cin.ignore(std::numeric_limits<std::streamsize>::max(), '\n');
                        continue;
                    }
                    
                    switch (choice) {
                        case 1:
                            showTasks(selectedProject);
                            break;
                        case 2:
                            addTask(selectedProject, projects);
                            break;
                        case 3:
                            startFocusSession(selectedProject, projects);
                            break;
                        case 4:
                            break;
                        default:
                            std::cout << "Invalid choice!\n";
                            std::this_thread::sleep_for(std::chrono::seconds(1));
                    }
                    
                    if (choice == 4) break;
                }
                break;
            }
            case 4:
                return 0;
            default:
                std::cout << "Invalid choice!\n";
                std::this_thread::sleep_for(std::chrono::seconds(1));
        }
    }
    
    return 0;
}