ClickHouse/dbms/src/Common/TerminalDisplaying.cpp

#include <unistd.h>
#include <sys/ioctl.h>
#include <Common/Exception.h>
#include <Common/TerminalDisplaying.h>

namespace po = boost::program_options;


namespace DB::ErrorCodes
{
    extern const int SYSTEM_ERROR;
}

unsigned short int getTerminalWidth()
{
    if (isatty(STDIN_FILENO))
    {
        winsize terminal_size {};

        if (ioctl(STDIN_FILENO, TIOCGWINSZ, &terminal_size))
            DB::throwFromErrno("Cannot obtain terminal window size (ioctl TIOCGWINSZ)", DB::ErrorCodes::SYSTEM_ERROR);

        return terminal_size.ws_col;
    }
    return 0;
}

po::options_description setOptionsDescription(const std::string & caption, unsigned short terminal_width)
{
    unsigned line_length = po::options_description::m_default_line_length;
    unsigned min_description_length = line_length / 2;
    std::string longest_option_desc = "--http_native_compression_disable_checksumming_on_decompress";

    line_length = std::max(static_cast<unsigned short>(longest_option_desc.size()), terminal_width);
    min_description_length = std::min(min_description_length, line_length - 2);

    return po::options_description(caption, line_length, min_description_length);
}