mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-15 02:41:59 +00:00
44 lines
1.4 KiB
C++
44 lines
1.4 KiB
C++
#include <unistd.h>
|
|
#include <sys/ioctl.h>
|
|
#if defined(OS_SUNOS)
|
|
# include <sys/termios.h>
|
|
#endif
|
|
#include <Common/Exception.h>
|
|
#include <Common/TerminalSize.h>
|
|
#include <boost/program_options.hpp>
|
|
|
|
|
|
namespace DB::ErrorCodes
|
|
{
|
|
extern const int SYSTEM_ERROR;
|
|
}
|
|
|
|
uint16_t getTerminalWidth()
|
|
{
|
|
struct winsize terminal_size {};
|
|
if (isatty(STDIN_FILENO))
|
|
{
|
|
if (ioctl(STDIN_FILENO, TIOCGWINSZ, &terminal_size))
|
|
DB::throwFromErrno("Cannot obtain terminal window size (ioctl TIOCGWINSZ)", DB::ErrorCodes::SYSTEM_ERROR);
|
|
}
|
|
else if (isatty(STDERR_FILENO))
|
|
{
|
|
if (ioctl(STDERR_FILENO, TIOCGWINSZ, &terminal_size))
|
|
DB::throwFromErrno("Cannot obtain terminal window size (ioctl TIOCGWINSZ)", DB::ErrorCodes::SYSTEM_ERROR);
|
|
}
|
|
/// Default - 0.
|
|
return terminal_size.ws_col;
|
|
}
|
|
|
|
po::options_description createOptionsDescription(const std::string & caption, uint16_t 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<uint16_t>(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);
|
|
}
|