cursor.hpp

Go to the documentation of this file.

#ifndef __CPP_CONSOLE_CURSOR_HPP__
#define __CPP_CONSOLE_CURSOR_HPP__
 
#include <iostream>
 
#ifdef __WIN32
#include <Windows.h>
#else
#include <stdio.h>
#include <sys/ioctl.h>
#include <termios.h>
#include <unistd.h>
#endif
 
#include "ansi.hpp"
#include "console.hpp"
#include "data.hpp"
 
namespace console {
ConsoleCoord getCursorPosition() {
#ifdef __WIN32
    CONSOLE_SCREEN_BUFFER_INFO cbsi;
    GetConsoleScreenBufferInfo(GetStdHandle(STD_OUTPUT_HANDLE), &cbsi);
    return {cbsi.dwCursorPosition.X, cbsi.dwCursorPosition.Y};
#else
    ConsoleCoord coord;
    struct termios term, term_orig;
    tcgetattr(STDIN_FILENO, &term);
    term_orig = term;
 
    term.c_lflag &= ~(ICANON | ECHO);
    tcsetattr(STDIN_FILENO, TCSANOW, &term);
    std::cout << "\033[6n";
    std::cin.ignore();
 
    char ignore;
    std::cin >> ignore >> coord.X;
    std::cin.ignore();
    std::cin >> coord.Y;
 
    tcsetattr(STDIN_FILENO, TCSANOW, &term_orig);
    return coord;
#endif
}
 
void moveCursorHome() {
    if (isANSIEnabled()) {
        std::cout << "\033[H";
        return;
    }
#ifdef __WIN32
    HANDLE hConsole = GetStdHandle(STD_OUTPUT_HANDLE);
    if (hConsole == INVALID_HANDLE_VALUE) {
        std::cerr << "Error: Unable to get console handle." << std::endl;
        return;
    }
    COORD homeCoords = {0, 0};
    SetConsoleCursorPosition(hConsole, homeCoords);
#endif
}
 
void moveCursorToPosition(int row, int col) {
    if (isANSIEnabled()) {
        std::cout << "\033[" << std::to_string(row) << ";"
                  << std::to_string(col) << "H";
        return;
    }
#ifdef __WIN32
    HANDLE hConsole = GetStdHandle(STD_OUTPUT_HANDLE);
    if (hConsole == INVALID_HANDLE_VALUE) {
        std::cerr << "Error: Unable to get console handle." << std::endl;
        return;
    }
    COORD coord;
    coord.X = row;
    coord.Y = col;
    SetConsoleCursorPosition(hConsole, coord);
#endif
}
 
void moveCursorUp(int n) {
    if (isANSIEnabled()) {
        std::cout << "\033[" << std::to_string(n) << "A";
        return;
    }
#ifdef __WIN32
    ConsoleCoord coord = getCursorPosition();
    moveCursorToPosition(coord.X, coord.Y - n);
#endif
}
 
void moveCursorDown(int n) {
    if (isANSIEnabled()) {
        std::cout << "\033[" << std::to_string(n) << "B";
        return;
    }
#ifdef __WIN32
    ConsoleCoord coord = getCursorPosition();
    moveCursorToPosition(coord.X, coord.Y + n);
#endif
}
 
void moveCursorRight(int n) {
    if (isANSIEnabled()) {
        std::cout << "\033[" << std::to_string(n) << "C";
        return;
    }
#ifdef __WIN32
    ConsoleCoord coord = getCursorPosition();
    moveCursorToPosition(coord.X + n, coord.Y);
#endif
}
 
void moveCursorLeft(int n) {
    if (isANSIEnabled()) {
        std::cout << "\033[" << std::to_string(n) << "D";
        return;
    }
#ifdef __WIN32
    ConsoleCoord coord = getCursorPosition();
    moveCursorToPosition(coord.X - n, coord.Y);
#endif
}
 
void moveCursorDownAndToStart(int n) {
    if (isANSIEnabled()) {
        std::cout << "\033[" << std::to_string(n) << "E";
        return;
    }
#ifdef __WIN32
    ConsoleCoord coord = getCursorPosition();
    moveCursorToPosition(0, coord.Y + n);
#endif
}
 
void moveCursorUpAndToStart(int n) {
    if (isANSIEnabled()) {
        std::cout << "\033[" << std::to_string(n) << "F";
        return;
    }
#ifdef __WIN32
    ConsoleCoord coord = getCursorPosition();
    moveCursorToPosition(0, coord.Y - n);
#endif
}
 
void moveCursorToColumn(int n) {
    if (isANSIEnabled()) {
        std::cout << "\033[" << std::to_string(n) << "G";
        return;
    }
#ifdef __WIN32
    ConsoleCoord coord = getCursorPosition();
    moveCursorToPosition(coord.X, n);
#endif
}
}  // namespace console
 
#endif