Skip to content

Commit

Permalink
Merge pull request #11816 from hercules-ci/fix-logger-ask-eof
Browse files Browse the repository at this point in the history
`ProgressBar::ask`: accept EOF, as a no
  • Loading branch information
roberth authored Nov 6, 2024
2 parents deb1db8 + c6c8d2a commit 1af94bf
Show file tree
Hide file tree
Showing 5 changed files with 25 additions and 10 deletions.
2 changes: 1 addition & 1 deletion src/libmain/progress-bar.cc
Original file line number Diff line number Diff line change
Expand Up @@ -543,7 +543,7 @@ class ProgressBar : public Logger
auto state(state_.lock());
if (!state->active) return {};
std::cerr << fmt("\r\e[K%s ", msg);
auto s = trim(readLine(STDIN_FILENO));
auto s = trim(readLine(STDIN_FILENO, true));
if (s.size() != 1) return {};
draw(*state);
return s[0];
Expand Down
7 changes: 6 additions & 1 deletion src/libutil/file-descriptor.hh
Original file line number Diff line number Diff line change
Expand Up @@ -77,8 +77,13 @@ void writeFull(Descriptor fd, std::string_view s, bool allowInterrupts = true);

/**
* Read a line from a file descriptor.
*
* @param fd The file descriptor to read from
* @param eofOk If true, return an unterminated line if EOF is reached. (e.g. the empty string)
*
* @return A line of text ending in `\n`, or a string without `\n` if `eofOk` is true and EOF is reached.
*/
std::string readLine(Descriptor fd);
std::string readLine(Descriptor fd, bool eofOk = false);

/**
* Write a line to a file descriptor.
Expand Down
10 changes: 7 additions & 3 deletions src/libutil/unix/file-descriptor.cc
Original file line number Diff line number Diff line change
Expand Up @@ -47,7 +47,7 @@ void writeFull(int fd, std::string_view s, bool allowInterrupts)
}


std::string readLine(int fd)
std::string readLine(int fd, bool eofOk)
{
std::string s;
while (1) {
Expand All @@ -58,8 +58,12 @@ std::string readLine(int fd)
if (rd == -1) {
if (errno != EINTR)
throw SysError("reading a line");
} else if (rd == 0)
throw EndOfFile("unexpected EOF reading a line");
} else if (rd == 0) {
if (eofOk)
return s;
else
throw EndOfFile("unexpected EOF reading a line");
}
else {
if (ch == '\n') return s;
s += ch;
Expand Down
10 changes: 10 additions & 0 deletions tests/functional/flakes/config.sh
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,17 @@ cat <<EOF > flake.nix
EOF

# Without --accept-flake-config, the post hook should not run.
# To test variations in stderr tty-ness, we run the command in different ways,
# none of which should block on stdin or accept the `nixConfig`s.
nix build < /dev/null
nix build < /dev/null 2>&1 | cat
# EOF counts as no, even when interactive (throw EOF error before)
if type -p script >/dev/null && script -q -c true /dev/null; then
echo "script is available and GNU-like, so we can ensure a tty"
script -q -c 'nix build < /dev/null' /dev/null
else
echo "script is not available or not GNU-like, so we skip testing with an added tty"
fi
(! [[ -f post-hook-ran ]])
TODO_NixOS
clearStore
Expand Down
6 changes: 1 addition & 5 deletions tests/functional/package.nix
Original file line number Diff line number Diff line change
@@ -1,7 +1,6 @@
{ lib
, stdenv
, mkMesonDerivation
, releaseTools

, meson
, ninja
Expand All @@ -16,10 +15,6 @@
, nix-expr
, nix-cli

, rapidcheck
, gtest
, runCommand

, busybox-sandbox-shell ? null

# Configuration Options
Expand Down Expand Up @@ -60,6 +55,7 @@ mkMesonDerivation (finalAttrs: {
# etc.
busybox-sandbox-shell
# For Overlay FS tests need `mount`, `umount`, and `unshare`.
# For `script` command (ensuring a TTY)
# TODO use `unixtools` to be precise over which executables instead?
util-linux
];
Expand Down

0 comments on commit 1af94bf

Please sign in to comment.