env.cpp

Go to the documentation of this file.

// -------------------------------------------------------//
//
// SHAMROCK code for hydrodynamics
// Copyright (c) 2021-2026 Timothée David--Cléris <tim.shamrock@proton.me>
// SPDX-License-Identifier: CeCILL Free Software License Agreement v2.1
// Shamrock is licensed under the CeCILL 2.1 License, see LICENSE for more information
//
// -------------------------------------------------------//

 
#include "sham/term/env.hpp"
#include "sham/term/color.hpp"
#include "sham/term/tty.hpp"
#include <string_view>
#include <vector>
 
namespace {

    static const std::vector<std::string_view> color_support_term{
        "xterm",
        "xterm-256",
        "xterm-256color",
        "xterm-truecolor",
        "vt100",
        "color",
        "ansi",
        "cygwin",
        "linux",
        "xterm-kitty",
        "alacritty"};

    bool term_support_color(sham::term::TermEnvVars vars) {
 
        if (vars.TERM) {
            for (auto term : color_support_term) {
                if (*vars.TERM == term) {
                    return true;
                }
            }
        }
 
        if (vars.COLORTERM) {
            if (*vars.COLORTERM == "truecolor") {
                return true;
            }
            if (*vars.COLORTERM == "24bit") {
                return true;
            }
        }
 
        return false;
    }
 
} // namespace
 
namespace sham::term {
 
    void parse_terminal_support(TermEnvVars vars, const term_parse_callback_t &error_callback) {
        if (term_support_color(vars)) {
            enable_colors();
        } else {
            disable_colors();
        }
 
        bool has_envvar_nocolor = bool(vars.NO_COLOR);
        bool has_envvar_color   = bool(vars.CLICOLOR_FORCE);
 
        if (has_envvar_color && has_envvar_nocolor) {
            throw error_callback(
                "one can not set both NO_COLOR and CLICOLOR_FORCE",
                std::source_location::current());
        }
 
        if (has_envvar_nocolor) {
            disable_colors();
        }
 
        if (has_envvar_color) {
            enable_colors();
        }
 
        auto &res = vars.COLUMN;
 
        int min_sz = 10;
        if (res) {
            try {
                int val = std::stoi(std::string(*res));
                if (val < min_sz) {
                    val = min_sz;
                }
                sham::term::set_tty_columns(val);
            } catch (const std::invalid_argument &a) {
                throw error_callback(
                    "Error : COLUMN is not an integer", std::source_location::current());
            } catch (const std::out_of_range &a) {
                throw error_callback(
                    "Error : COLUMN is out of range", std::source_location::current());
            }
        }
    }
 
} // namespace sham::term