osm2pgsql/src/util.cpp at c06615483439f026679b6aa0ebbe655ca31f45ee · joto/osm2pgsql · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
/**
 * SPDX-License-Identifier: GPL-2.0-or-later
 *
 * This file is part of osm2pgsql (https://osm2pgsql.org/).
 *
 * Copyright (C) 2006-2026 by the osm2pgsql developer community.
 * For a full list of authors see the git log.
 */

#include "util.hpp"

#include <iostream>

#ifdef _WIN32
#include <windows.h>
#elif __has_include(<termios.h>)
#include <termios.h>
#include <unistd.h>
#endif

namespace util {

std::string human_readable_duration(uint64_t seconds)
{
    if (seconds < 60UL) {
        return fmt::format("{}s", seconds);
    }

    if (seconds < (60UL * 60UL)) {
        return fmt::format("{}s ({}m {}s)", seconds, seconds / 60,
                           seconds % 60);
    }

    auto const secs = seconds % 60;
    auto const mins = seconds / 60;
    return fmt::format("{}s ({}h {}m {}s)", seconds, mins / 60, mins % 60,
                       secs);
}

std::string human_readable_duration(std::chrono::microseconds duration)
{
    return human_readable_duration(static_cast<uint64_t>(
        std::chrono::duration_cast<std::chrono::seconds>(duration).count()));
}

std::string get_password()
{
#ifdef _WIN32
    HANDLE const handle_stdin = GetStdHandle(STD_INPUT_HANDLE);
    DWORD mode = 0;
    GetConsoleMode(handle_stdin, &mode);
    SetConsoleMode(handle_stdin, mode & (~ENABLE_ECHO_INPUT));
#elif __has_include(<termios.h>)
    termios orig_flags{};
    tcgetattr(STDIN_FILENO, &orig_flags);
    termios flags = orig_flags;
    flags.c_lflag &= ~static_cast<unsigned int>(ECHO);
    tcsetattr(STDIN_FILENO, TCSANOW, &flags);
#endif

    std::string password;
    std::cout << "Password:";
    std::getline(std::cin, password);
    std::cout << "\n";

#ifdef _WIN32
    SetConsoleMode(handle_stdin, mode);
#elif __has_include(<termios.h>)
    tcsetattr(STDIN_FILENO, TCSANOW, &orig_flags);
#endif

    return password;
}

string_joiner_t::string_joiner_t(char delim, char quote, char before,
                                 char after)
: m_delim(delim), m_quote(quote), m_before(before), m_after(after)
{
    if (m_before) {
        m_result += m_before;
    }
}

void string_joiner_t::add(std::string const &item)
{
    if (m_quote) {
        m_result += m_quote;
        m_result += item;
        m_result += m_quote;
    } else {
        m_result += item;
    }
    m_result += m_delim;
}

std::string string_joiner_t::operator()()
{
    if (m_result.size() == 1 && m_before) {
        m_result.clear();
    } else if (m_result.size() > 1) {
        if (m_after) {
            m_result.back() = m_after;
        } else {
            m_result.resize(m_result.size() - 1);
        }
    }
    return std::move(m_result);
}

std::string join(std::vector<std::string> const &items, char delim, char quote,
                 char before, char after)
{
    string_joiner_t joiner{delim, quote, before, after};
    for (auto const &item : items) {
        joiner.add(item);
    }
    return joiner();
}

} // namespace util