Fix TTY detection

Author: supechicken

component/client.rb

@@ -12,7 +12,7 @@ def restore_console
 def runas_client(argv)
   $mode     = :client
   $tty_attr = %x[/bin/stty -g].chomp
-  is_tty    = $stdout.isatty && $stderr.isatty
+  is_tty    = $stdin.isatty && $stdout.isatty && $stderr.isatty
   socket    = UNIXSocket.open(SOCKET_PATH) # connect to daemon
 
   # send stdin/stdout/stderr to daemon
@@ -29,7 +29,7 @@ def runas_client(argv)
   })
 
   # listen to terminal resize event
-  trap('WINCH') { send_event(socket, 'termResize', { newsize: IO.console.winsize }) }
+  trap('WINCH') { send_event(socket, 'termResize', { newsize: IO.console.winsize }) } if is_tty
 
   # ignore following signals
   %w[HUP PIPE QUIT TERM INT].each {|sig| trap(sig) {} }

component/daemon.rb

@@ -33,7 +33,7 @@ def runas_daemon(argv)
 
       client_request = JSON.parse(socket.gets, symbolize_names: true)
       cmdline        = ['/usr/bin/sudo'].concat(client_request[:arg])
-      open_pty       = client_stdout.isatty && client_stderr.isatty
+      open_pty       = client_stdin.isatty && client_stdout.isatty && client_stderr.isatty
       process_env    = client_request[:env].transform_keys(&:to_s)
 
       if open_pty

lib/function.rb

@@ -2,8 +2,8 @@ def send_event(sock, event, args = {})
   sock.puts({ event: event }.merge(args).to_json)
 end
 
-def message(message, loglevel: :verbose)
+def message(m, loglevel: :verbose)
   return unless $log || $verbose || loglevel != :verbose
 
-  warn "[#{$mode}]: #{message}"
+  warn m.lines(chomp: true).map {|line| "[#{$mode}]: #{line}" }
 end