git.saurik.com Git - apt.git/blame_incremental

... / ...

Commit	Line	Data
	1	#include <apt-pkg/iprogress.h>
	2	#include <apt-pkg/strutl.h>
	3
	4	#include <termios.h>
	5	#include <sys/ioctl.h>
	6
	7	namespace APT {
	8	namespace Progress {
	9
	10	void PackageManagerFancy::SetupTerminalScrollArea(int nr_rows)
	11	{
	12	// scroll down a bit to avoid visual glitch when the screen
	13	// area shrinks by one row
	14	std::cout << "\n";
	15
	16	// save cursor
	17	std::cout << "\033[s";
	18
	19	// set scroll region (this will place the cursor in the top left)
	20	std::cout << "\033[1;" << nr_rows - 1 << "r";
	21
	22	// restore cursor but ensure its inside the scrolling area
	23	std::cout << "\033[u";
	24	static const char *move_cursor_up = "\033[1A";
	25	std::cout << move_cursor_up;
	26
	27	std::flush(std::cout);
	28	}
	29
	30	PackageManagerFancy::PackageManagerFancy()
	31	: nr_terminal_rows(-1)
	32	{
	33	struct winsize win;
	34	if(ioctl(STDOUT_FILENO, TIOCGWINSZ, (char *)&win) == 0)
	35	{
	36	nr_terminal_rows = win.ws_row;
	37	}
	38	}
	39
	40	void PackageManagerFancy::Started()
	41	{
	42	if (nr_terminal_rows > 0)
	43	SetupTerminalScrollArea(nr_terminal_rows);
	44	}
	45
	46	void PackageManagerFancy::Finished()
	47	{
	48	if (nr_terminal_rows > 0)
	49	{
	50	SetupTerminalScrollArea(nr_terminal_rows + 1);
	51
	52	// override the progress line (sledgehammer)
	53	static const char* clear_screen_below_cursor = "\033[J";
	54	std::cout << clear_screen_below_cursor;
	55	}
	56	}
	57
	58	void PackageManagerFancy::StatusChanged(std::string PackageName,
	59	unsigned int StepsDone,
	60	unsigned int TotalSteps)
	61	{
	62	int reporting_steps = _config->FindI("DpkgPM::Reporting-Steps", 1);
	63	float percentage = StepsDone/(float)TotalSteps * 100.0;
	64
	65	if(percentage < (last_reported_progress + reporting_steps))
	66	return;
	67
	68	std::string progress_str;
	69	strprintf(progress_str, "Progress: [%3i%%]", (int)percentage);
	70
	71	int row = nr_terminal_rows;
	72
	73	static string save_cursor = "\033[s";
	74	static string restore_cursor = "\033[u";
	75
	76	static string set_bg_color = "\033[42m"; // green
	77	static string set_fg_color = "\033[30m"; // black
	78
	79	static string restore_bg = "\033[49m";
	80	static string restore_fg = "\033[39m";
	81
	82	std::cout << save_cursor
	83	// move cursor position to last row
	84	<< "\033[" << row << ";0f"
	85	<< set_bg_color
	86	<< set_fg_color
	87	<< progress_str
	88	<< restore_cursor
	89	<< restore_bg
	90	<< restore_fg;
	91	std::flush(std::cout);
	92	last_reported_progress = percentage;
	93	}
	94
	95	void PackageManagerText::StatusChanged(std::string PackageName,
	96	unsigned int StepsDone,
	97	unsigned int TotalSteps)
	98	{
	99	int reporting_steps = _config->FindI("DpkgPM::Reporting-Steps", 1);
	100	float percentage = StepsDone/(float)TotalSteps * 100.0;
	101
	102	if(percentage < (last_reported_progress + reporting_steps))
	103	return;
	104
	105	std::string progress_str;
	106	strprintf(progress_str, "Progress: [%3i%%]", (int)percentage);
	107
	108	std::cout << progress_str << "\r\n";
	109	std::flush(std::cout);
	110
	111	last_reported_progress = percentage;
	112	}
	113
	114
	115	}; // namespace progress
	116	}; // namespace apt