blob: 64d10b6ca14a52420de41d3b49e6cec12232981e [file] [log] [blame]
Elliott Hughesf5cdc1d2015-10-27 16:03:15 -07001// Copyright 2013 Google Inc. All Rights Reserved.
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7// http://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14
15#include "line_printer.h"
16
17#include <stdio.h>
18#include <stdlib.h>
19#ifdef _WIN32
20#include <windows.h>
21#else
22#include <unistd.h>
23#include <sys/ioctl.h>
24#include <termios.h>
25#include <sys/time.h>
26#endif
27
28// Make sure printf is really adb_printf which works for UTF-8 on Windows.
29#include <sysdeps.h>
30
31// Stuff from ninja's util.h that's needed below.
32#include <vector>
33using namespace std;
34string ElideMiddle(const string& str, size_t width) {
35 const int kMargin = 3; // Space for "...".
36 string result = str;
37 if (result.size() + kMargin > width) {
38 size_t elide_size = (width - kMargin) / 2;
39 result = result.substr(0, elide_size)
40 + "..."
41 + result.substr(result.size() - elide_size, elide_size);
42 }
43 return result;
44}
45
Elliott Hughes78a37a52015-12-08 16:01:15 -080046LinePrinter::LinePrinter() : have_blank_line_(true) {
Elliott Hughesf5cdc1d2015-10-27 16:03:15 -070047#ifndef _WIN32
48 const char* term = getenv("TERM");
David Pursell58805362015-10-28 14:29:51 -070049 smart_terminal_ = unix_isatty(1) && term && string(term) != "dumb";
Elliott Hughesf5cdc1d2015-10-27 16:03:15 -070050#else
51 // Disable output buffer. It'd be nice to use line buffering but
52 // MSDN says: "For some systems, [_IOLBF] provides line
53 // buffering. However, for Win32, the behavior is the same as _IOFBF
54 // - Full Buffering."
55 setvbuf(stdout, NULL, _IONBF, 0);
56 console_ = GetStdHandle(STD_OUTPUT_HANDLE);
57 CONSOLE_SCREEN_BUFFER_INFO csbi;
58 smart_terminal_ = GetConsoleScreenBufferInfo(console_, &csbi);
59#endif
60}
61
Elliott Hughes78a37a52015-12-08 16:01:15 -080062static void Out(const std::string& s) {
63 // Avoid printf and C strings, since the actual output might contain null
64 // bytes like UTF-16 does (yuck).
65 fwrite(s.data(), 1, s.size(), stdout);
66}
67
Elliott Hughesf5cdc1d2015-10-27 16:03:15 -070068void LinePrinter::Print(string to_print, LineType type) {
Elliott Hughes78a37a52015-12-08 16:01:15 -080069 if (!smart_terminal_) {
Elliott Hughes6e6e4602016-01-08 15:47:07 -080070 Out(to_print + "\n");
Elliott Hughesf5cdc1d2015-10-27 16:03:15 -070071 return;
72 }
73
Elliott Hughes78a37a52015-12-08 16:01:15 -080074 // Print over previous line, if any.
75 // On Windows, calling a C library function writing to stdout also handles
76 // pausing the executable when the "Pause" key or Ctrl-S is pressed.
77 printf("\r");
Elliott Hughesf5cdc1d2015-10-27 16:03:15 -070078
Elliott Hughes78a37a52015-12-08 16:01:15 -080079 if (type == INFO) {
Elliott Hughesf5cdc1d2015-10-27 16:03:15 -070080#ifdef _WIN32
81 CONSOLE_SCREEN_BUFFER_INFO csbi;
82 GetConsoleScreenBufferInfo(console_, &csbi);
83
Spencer Low50f5bf12015-11-12 15:20:15 -080084 // TODO: std::wstring to_print_wide; if (!android::base::UTF8ToWide(to_print, &to_print_wide)...
Elliott Hughesf5cdc1d2015-10-27 16:03:15 -070085 // TODO: wstring ElideMiddle.
86 to_print = ElideMiddle(to_print, static_cast<size_t>(csbi.dwSize.X));
87 // We don't want to have the cursor spamming back and forth, so instead of
88 // printf use WriteConsoleOutput which updates the contents of the buffer,
89 // but doesn't move the cursor position.
90 COORD buf_size = { csbi.dwSize.X, 1 };
91 COORD zero_zero = { 0, 0 };
92 SMALL_RECT target = {
93 csbi.dwCursorPosition.X, csbi.dwCursorPosition.Y,
94 static_cast<SHORT>(csbi.dwCursorPosition.X + csbi.dwSize.X - 1),
95 csbi.dwCursorPosition.Y
96 };
97 vector<CHAR_INFO> char_data(csbi.dwSize.X);
98 for (size_t i = 0; i < static_cast<size_t>(csbi.dwSize.X); ++i) {
99 // TODO: UnicodeChar instead of AsciiChar, to_print_wide[i].
100 char_data[i].Char.AsciiChar = i < to_print.size() ? to_print[i] : ' ';
101 char_data[i].Attributes = csbi.wAttributes;
102 }
103 // TODO: WriteConsoleOutputW.
104 WriteConsoleOutput(console_, &char_data[0], buf_size, zero_zero, &target);
105#else
106 // Limit output to width of the terminal if provided so we don't cause
107 // line-wrapping.
108 winsize size;
109 if ((ioctl(0, TIOCGWINSZ, &size) == 0) && size.ws_col) {
110 to_print = ElideMiddle(to_print, size.ws_col);
111 }
Elliott Hughes78a37a52015-12-08 16:01:15 -0800112 Out(to_print);
Elliott Hughesf5cdc1d2015-10-27 16:03:15 -0700113 printf("\x1B[K"); // Clear to end of line.
114 fflush(stdout);
115#endif
116
117 have_blank_line_ = false;
118 } else {
Elliott Hughes78a37a52015-12-08 16:01:15 -0800119 Out(to_print);
120 Out("\n");
121 have_blank_line_ = true;
Elliott Hughesf5cdc1d2015-10-27 16:03:15 -0700122 }
123}
124
Elliott Hughes78a37a52015-12-08 16:01:15 -0800125void LinePrinter::KeepInfoLine() {
126 if (!have_blank_line_) Out("\n");
Josh Gaof35fede2016-08-04 11:43:00 -0700127 have_blank_line_ = true;
Elliott Hughesf5cdc1d2015-10-27 16:03:15 -0700128}