[apt.git] / apt-pkg / install-progress.cc

#include <config.h>

#include <apt-pkg/configuration.h>
#include <apt-pkg/fileutl.h>
#include <apt-pkg/strutl.h>
#include <apt-pkg/install-progress.h>

#include <signal.h>
#include <unistd.h>
#include <iostream>
#include <string>
#include <vector>
#include <sys/ioctl.h>
#include <sstream>
#include <fcntl.h>
#include <algorithm>
#include <stdio.h>

#include <apti18n.h>

namespace APT {
namespace Progress {


/* Return a APT::Progress::PackageManager based on the global
 * apt configuration (i.e. APT::Status-Fd and APT::Status-deb822-Fd)
 */
PackageManager* PackageManagerProgressFactory()
{
   // select the right progress
   int status_fd = _config->FindI("APT::Status-Fd", -1);
   int status_deb822_fd = _config->FindI("APT::Status-deb822-Fd", -1);

   APT::Progress::PackageManager *progress = NULL;
   if (status_deb822_fd > 0)
      progress = new APT::Progress::PackageManagerProgressDeb822Fd(
         status_deb822_fd);
   else if (status_fd > 0)
      progress = new APT::Progress::PackageManagerProgressFd(status_fd);
   else if(_config->FindB("Dpkg::Progress-Fancy", false) == true)
      progress = new APT::Progress::PackageManagerFancy();
   else if (_config->FindB("Dpkg::Progress", 
                           _config->FindB("DpkgPM::Progress", false)) == true)
      progress = new APT::Progress::PackageManagerText();
   else
      progress = new APT::Progress::PackageManager();
   return progress;
}

bool PackageManager::StatusChanged(std::string /*PackageName*/,
                                   unsigned int StepsDone,
                                   unsigned int TotalSteps,
                                   std::string /*HumanReadableAction*/)
{
   int reporting_steps = _config->FindI("DpkgPM::Reporting-Steps", 1);
   percentage = StepsDone/(float)TotalSteps * 100.0;
   strprintf(progress_str, _("Progress: [%3i%%]"), (int)percentage);

   if(percentage < (last_reported_progress + reporting_steps))
      return false;

   return true;
}

PackageManagerProgressFd::PackageManagerProgressFd(int progress_fd)
   : StepsDone(0), StepsTotal(1)
{
   OutStatusFd = progress_fd;
}

void PackageManagerProgressFd::WriteToStatusFd(std::string s)
{
   if(OutStatusFd <= 0)
      return;
   FileFd::Write(OutStatusFd, s.c_str(), s.size());   
}

void PackageManagerProgressFd::StartDpkg()
{
   if(OutStatusFd <= 0)
      return;

   // FIXME: use SetCloseExec here once it taught about throwing
   //        exceptions instead of doing _exit(100) on failure
   fcntl(OutStatusFd,F_SETFD,FD_CLOEXEC); 

   // send status information that we are about to fork dpkg
   std::ostringstream status;
   status << "pmstatus:dpkg-exec:" 
          << (StepsDone/float(StepsTotal)*100.0) 
          << ":" << _("Running dpkg")
          << std::endl;
   WriteToStatusFd(status.str());
}

APT_CONST void PackageManagerProgressFd::Stop()
{
}

void PackageManagerProgressFd::Error(std::string PackageName,
                                     unsigned int StepsDone,
                                     unsigned int TotalSteps,
                                     std::string ErrorMessage)
{
   std::ostringstream status;
   status << "pmerror:" << PackageName
          << ":"  << (StepsDone/float(TotalSteps)*100.0) 
          << ":" << ErrorMessage
          << std::endl;
   WriteToStatusFd(status.str());
}

void PackageManagerProgressFd::ConffilePrompt(std::string PackageName,
                                              unsigned int StepsDone,
                                              unsigned int TotalSteps,
                                              std::string ConfMessage)
{
   std::ostringstream status;
   status << "pmconffile:" << PackageName
          << ":"  << (StepsDone/float(TotalSteps)*100.0) 
          << ":" << ConfMessage
          << std::endl;
   WriteToStatusFd(status.str());
}


bool PackageManagerProgressFd::StatusChanged(std::string PackageName, 
                                             unsigned int xStepsDone,
                                             unsigned int xTotalSteps,
                                             std::string pkg_action)
{
   StepsDone = xStepsDone;
   StepsTotal = xTotalSteps;

   // build the status str
   std::ostringstream status;
   status << "pmstatus:" << StringSplit(PackageName, ":")[0]
          << ":"  << (StepsDone/float(StepsTotal)*100.0) 
          << ":" << pkg_action
          << std::endl;
   WriteToStatusFd(status.str());

   if(_config->FindB("Debug::APT::Progress::PackageManagerFd", false) == true)
      std::cerr << "progress: " << PackageName << " " << xStepsDone
                << " " << xTotalSteps << " " << pkg_action
                << std::endl;


   return true;
}


PackageManagerProgressDeb822Fd::PackageManagerProgressDeb822Fd(int progress_fd)
   : StepsDone(0), StepsTotal(1)
{
   OutStatusFd = progress_fd;
}

void PackageManagerProgressDeb822Fd::WriteToStatusFd(std::string s)
{
   FileFd::Write(OutStatusFd, s.c_str(), s.size());   
}

void PackageManagerProgressDeb822Fd::StartDpkg()
{
   // FIXME: use SetCloseExec here once it taught about throwing
   //        exceptions instead of doing _exit(100) on failure
   fcntl(OutStatusFd,F_SETFD,FD_CLOEXEC); 

   // send status information that we are about to fork dpkg
   std::ostringstream status;
   status << "Status: " << "progress" << std::endl
          << "Percent: " << (StepsDone/float(StepsTotal)*100.0) << std::endl
          << "Message: " << _("Running dpkg") << std::endl
          << std::endl;
   WriteToStatusFd(status.str());
}

APT_CONST void PackageManagerProgressDeb822Fd::Stop()
{
}

void PackageManagerProgressDeb822Fd::Error(std::string PackageName,
                                     unsigned int StepsDone,
                                     unsigned int TotalSteps,
                                     std::string ErrorMessage)
{
   std::ostringstream status;
   status << "Status: " << "Error" << std::endl
          << "Package:" << PackageName << std::endl
          << "Percent: "  << (StepsDone/float(TotalSteps)*100.0) << std::endl
          << "Message: " << ErrorMessage << std::endl
          << std::endl;
   WriteToStatusFd(status.str());
}

void PackageManagerProgressDeb822Fd::ConffilePrompt(std::string PackageName,
                                              unsigned int StepsDone,
                                              unsigned int TotalSteps,
                                              std::string ConfMessage)
{
   std::ostringstream status;
   status << "Status: " << "ConfFile" << std::endl
          << "Package:" << PackageName << std::endl
          << "Percent: "  << (StepsDone/float(TotalSteps)*100.0) << std::endl
          << "Message: " << ConfMessage << std::endl
          << std::endl;
   WriteToStatusFd(status.str());
}


bool PackageManagerProgressDeb822Fd::StatusChanged(std::string PackageName, 
                                             unsigned int xStepsDone,
                                             unsigned int xTotalSteps,
                                             std::string message)
{
   StepsDone = xStepsDone;
   StepsTotal = xTotalSteps;

   // build the status str
   std::ostringstream status;
   status << "Status: " << "progress" << std::endl
          << "Package: " << PackageName << std::endl
          << "Percent: "  << (StepsDone/float(StepsTotal)*100.0) << std::endl
          << "Message: " << message << std::endl
          << std::endl;
   WriteToStatusFd(status.str());

   return true;
}


PackageManagerFancy::PackageManagerFancy()
   : child_pty(-1)
{
   // setup terminal size
   old_SIGWINCH = signal(SIGWINCH, PackageManagerFancy::staticSIGWINCH);
   instances.push_back(this);
}
std::vector<PackageManagerFancy*> PackageManagerFancy::instances;

PackageManagerFancy::~PackageManagerFancy()
{
   instances.erase(find(instances.begin(), instances.end(), this));
   signal(SIGWINCH, old_SIGWINCH);
}

void PackageManagerFancy::staticSIGWINCH(int signum)
{
   std::vector<PackageManagerFancy *>::const_iterator I;
   for(I = instances.begin(); I != instances.end(); ++I)
      (*I)->HandleSIGWINCH(signum);
}

int PackageManagerFancy::GetNumberTerminalRows()
{
   struct winsize win;
   // FIXME: get from "child_pty" instead?
   if(ioctl(STDOUT_FILENO, TIOCGWINSZ, (char *)&win) != 0)
      return -1;

   if(_config->FindB("Debug::InstallProgress::Fancy", false) == true)
      std::cerr << "GetNumberTerminalRows: " << win.ws_row << std::endl;
   
   return win.ws_row;
}

void PackageManagerFancy::SetupTerminalScrollArea(int nr_rows)
{
     if(_config->FindB("Debug::InstallProgress::Fancy", false) == true)
        std::cerr << "SetupTerminalScrollArea: " << nr_rows << std::endl;

     // scroll down a bit to avoid visual glitch when the screen
     // area shrinks by one row
     std::cout << "\n";
         
     // save cursor
     std::cout << "\033[s";
         
     // set scroll region (this will place the cursor in the top left)
     std::cout << "\033[0;" << nr_rows - 1 << "r";
            
     // restore cursor but ensure its inside the scrolling area
     std::cout << "\033[u";
     static const char *move_cursor_up = "\033[1A";
     std::cout << move_cursor_up;

     // ensure its flushed
     std::flush(std::cout);

     // setup tty size to ensure xterm/linux console are working properly too
     // see bug #731738
     struct winsize win;
     ioctl(child_pty, TIOCGWINSZ, (char *)&win);
     win.ws_row = nr_rows - 1;
     ioctl(child_pty, TIOCSWINSZ, (char *)&win);
}

void PackageManagerFancy::HandleSIGWINCH(int)
{
   int nr_terminal_rows = GetNumberTerminalRows();
   SetupTerminalScrollArea(nr_terminal_rows);
}

void PackageManagerFancy::Start(int a_child_pty)
{
   child_pty = a_child_pty;
   int nr_terminal_rows = GetNumberTerminalRows();
   if (nr_terminal_rows > 0)
      SetupTerminalScrollArea(nr_terminal_rows);
}

void PackageManagerFancy::Stop()
{
   int nr_terminal_rows = GetNumberTerminalRows();
   if (nr_terminal_rows > 0)
   {
      SetupTerminalScrollArea(nr_terminal_rows + 1);

      // override the progress line (sledgehammer)
      static const char* clear_screen_below_cursor = "\033[J";
      std::cout << clear_screen_below_cursor;
   }
   child_pty = -1;
}

bool PackageManagerFancy::StatusChanged(std::string PackageName, 
                                        unsigned int StepsDone,
                                        unsigned int TotalSteps,
                                        std::string HumanReadableAction)
{
   if (!PackageManager::StatusChanged(PackageName, StepsDone, TotalSteps,
          HumanReadableAction))
      return false;

   int row = GetNumberTerminalRows();

   static std::string save_cursor = "\033[s";
   static std::string restore_cursor = "\033[u";

   static std::string set_bg_color = "\033[42m"; // green
   static std::string set_fg_color = "\033[30m"; // black

   static std::string restore_bg =  "\033[49m";
   static std::string restore_fg = "\033[39m";

   std::cout << save_cursor
      // move cursor position to last row
             << "\033[" << row << ";0f" 
             << set_bg_color
             << set_fg_color
             << progress_str
             << restore_cursor
             << restore_bg
             << restore_fg;
   std::flush(std::cout);
   last_reported_progress = percentage;

   return true;
}

bool PackageManagerText::StatusChanged(std::string PackageName, 
                                       unsigned int StepsDone,
                                       unsigned int TotalSteps,
                                       std::string HumanReadableAction)
{
   if (!PackageManager::StatusChanged(PackageName, StepsDone, TotalSteps, HumanReadableAction))
      return false;

   std::cout << progress_str << "\r\n";
   std::flush(std::cout);
                   
   last_reported_progress = percentage;

   return true;
}


} // namespace progress
} // namespace apt
Commit	Line	Data
453b82a3 DK	1	#include <config.h>
453b82a3 DK	2
e6ad8031 MV	3	#include <apt-pkg/configuration.h>
e6ad8031 MV	4	#include <apt-pkg/fileutl.h>
31f97d7b	5	#include <apt-pkg/strutl.h>
af36becc	6	#include <apt-pkg/install-progress.h>
c7ea1eba	7
453b82a3 DK	8	#include <signal.h>
	9	#include <unistd.h>
	10	#include <iostream>
	11	#include <string>
	12	#include <vector>
31f97d7b	13	#include <sys/ioctl.h>
e6ad8031	14	#include <sstream>
5e9458e2	15	#include <fcntl.h>
5ed88785	16	#include <algorithm>
c23e6cd5	17	#include <stdio.h>
e96e4e9c	18
453b82a3 DK	19	#include <apti18n.h>
453b82a3 DK	20
31f97d7b MV	21	namespace APT {
	22	namespace Progress {
	23
61f954bf MV	24
	25	/* Return a APT::Progress::PackageManager based on the global
	26	* apt configuration (i.e. APT::Status-Fd and APT::Status-deb822-Fd)
	27	*/
bd5f39b3 MV	28	PackageManager* PackageManagerProgressFactory()
	29	{
	30	// select the right progress
	31	int status_fd = _config->FindI("APT::Status-Fd", -1);
	32	int status_deb822_fd = _config->FindI("APT::Status-deb822-Fd", -1);
	33
	34	APT::Progress::PackageManager *progress = NULL;
	35	if (status_deb822_fd > 0)
	36	progress = new APT::Progress::PackageManagerProgressDeb822Fd(
	37	status_deb822_fd);
	38	else if (status_fd > 0)
	39	progress = new APT::Progress::PackageManagerProgressFd(status_fd);
	40	else if(_config->FindB("Dpkg::Progress-Fancy", false) == true)
	41	progress = new APT::Progress::PackageManagerFancy();
	42	else if (_config->FindB("Dpkg::Progress",
	43	_config->FindB("DpkgPM::Progress", false)) == true)
	44	progress = new APT::Progress::PackageManagerText();
	45	else
	46	progress = new APT::Progress::PackageManager();
	47	return progress;
	48	}
	49
65512241	50	bool PackageManager::StatusChanged(std::string /PackageName/,
e6ad8031 MV	51	unsigned int StepsDone,
e6ad8031 MV	52	unsigned int TotalSteps,
65512241	53	std::string /HumanReadableAction/)
6c5ae8ed MV	54	{
	55	int reporting_steps = _config->FindI("DpkgPM::Reporting-Steps", 1);
	56	percentage = StepsDone/(float)TotalSteps * 100.0;
	57	strprintf(progress_str, _("Progress: [%3i%%]"), (int)percentage);
	58
	59	if(percentage < (last_reported_progress + reporting_steps))
	60	return false;
	61
	62	return true;
	63	}
	64
e6ad8031	65	PackageManagerProgressFd::PackageManagerProgressFd(int progress_fd)
65dbd5a1	66	: StepsDone(0), StepsTotal(1)
e6ad8031 MV	67	{
	68	OutStatusFd = progress_fd;
	69	}
	70
f9935b1c MV	71	void PackageManagerProgressFd::WriteToStatusFd(std::string s)
	72	{
	73	if(OutStatusFd <= 0)
	74	return;
	75	FileFd::Write(OutStatusFd, s.c_str(), s.size());
	76	}
	77
e45c4617	78	void PackageManagerProgressFd::StartDpkg()
e6ad8031	79	{
5e9458e2 MV	80	if(OutStatusFd <= 0)
	81	return;
	82
	83	// FIXME: use SetCloseExec here once it taught about throwing
	84	// exceptions instead of doing _exit(100) on failure
	85	fcntl(OutStatusFd,F_SETFD,FD_CLOEXEC);
e6ad8031 MV	86
e6ad8031 MV	87	// send status information that we are about to fork dpkg
f9935b1c MV	88	std::ostringstream status;
	89	status << "pmstatus:dpkg-exec:"
	90	<< (StepsDone/float(StepsTotal)*100.0)
	91	<< ":" << _("Running dpkg")
	92	<< std::endl;
	93	WriteToStatusFd(status.str());
e6ad8031 MV	94	}
e6ad8031 MV	95
a02db58f	96	APT_CONST void PackageManagerProgressFd::Stop()
e6ad8031	97	{
e6ad8031 MV	98	}
	99
	100	void PackageManagerProgressFd::Error(std::string PackageName,
	101	unsigned int StepsDone,
	102	unsigned int TotalSteps,
	103	std::string ErrorMessage)
	104	{
	105	std::ostringstream status;
	106	status << "pmerror:" << PackageName
	107	<< ":" << (StepsDone/float(TotalSteps)*100.0)
	108	<< ":" << ErrorMessage
	109	<< std::endl;
f9935b1c	110	WriteToStatusFd(status.str());
e6ad8031 MV	111	}
	112
	113	void PackageManagerProgressFd::ConffilePrompt(std::string PackageName,
	114	unsigned int StepsDone,
	115	unsigned int TotalSteps,
	116	std::string ConfMessage)
	117	{
	118	std::ostringstream status;
	119	status << "pmconffile:" << PackageName
	120	<< ":" << (StepsDone/float(TotalSteps)*100.0)
	121	<< ":" << ConfMessage
	122	<< std::endl;
f9935b1c	123	WriteToStatusFd(status.str());
e6ad8031 MV	124	}
	125
	126
	127	bool PackageManagerProgressFd::StatusChanged(std::string PackageName,
	128	unsigned int xStepsDone,
	129	unsigned int xTotalSteps,
	130	std::string pkg_action)
	131	{
	132	StepsDone = xStepsDone;
	133	StepsTotal = xTotalSteps;
	134
	135	// build the status str
	136	std::ostringstream status;
dd640f3c	137	status << "pmstatus:" << StringSplit(PackageName, ":")[0]
e6ad8031 MV	138	<< ":" << (StepsDone/float(StepsTotal)*100.0)
	139	<< ":" << pkg_action
	140	<< std::endl;
f9935b1c	141	WriteToStatusFd(status.str());
65dbd5a1 MV	142
	143	if(_config->FindB("Debug::APT::Progress::PackageManagerFd", false) == true)
	144	std::cerr << "progress: " << PackageName << " " << xStepsDone
	145	<< " " << xTotalSteps << " " << pkg_action
	146	<< std::endl;
	147
	148
e6ad8031 MV	149	return true;
	150	}
	151
c7ea1eba MV	152
	153	PackageManagerProgressDeb822Fd::PackageManagerProgressDeb822Fd(int progress_fd)
	154	: StepsDone(0), StepsTotal(1)
	155	{
	156	OutStatusFd = progress_fd;
	157	}
	158
	159	void PackageManagerProgressDeb822Fd::WriteToStatusFd(std::string s)
	160	{
	161	FileFd::Write(OutStatusFd, s.c_str(), s.size());
	162	}
	163
790d41f6	164	void PackageManagerProgressDeb822Fd::StartDpkg()
c7ea1eba MV	165	{
	166	// FIXME: use SetCloseExec here once it taught about throwing
	167	// exceptions instead of doing _exit(100) on failure
	168	fcntl(OutStatusFd,F_SETFD,FD_CLOEXEC);
	169
	170	// send status information that we are about to fork dpkg
	171	std::ostringstream status;
	172	status << "Status: " << "progress" << std::endl
	173	<< "Percent: " << (StepsDone/float(StepsTotal)*100.0) << std::endl
	174	<< "Message: " << _("Running dpkg") << std::endl
	175	<< std::endl;
	176	WriteToStatusFd(status.str());
	177	}
	178
a02db58f	179	APT_CONST void PackageManagerProgressDeb822Fd::Stop()
c7ea1eba	180	{
c7ea1eba MV	181	}
	182
	183	void PackageManagerProgressDeb822Fd::Error(std::string PackageName,
	184	unsigned int StepsDone,
	185	unsigned int TotalSteps,
	186	std::string ErrorMessage)
	187	{
	188	std::ostringstream status;
	189	status << "Status: " << "Error" << std::endl
	190	<< "Package:" << PackageName << std::endl
	191	<< "Percent: " << (StepsDone/float(TotalSteps)*100.0) << std::endl
	192	<< "Message: " << ErrorMessage << std::endl
	193	<< std::endl;
	194	WriteToStatusFd(status.str());
	195	}
	196
	197	void PackageManagerProgressDeb822Fd::ConffilePrompt(std::string PackageName,
	198	unsigned int StepsDone,
	199	unsigned int TotalSteps,
	200	std::string ConfMessage)
	201	{
	202	std::ostringstream status;
	203	status << "Status: " << "ConfFile" << std::endl
	204	<< "Package:" << PackageName << std::endl
	205	<< "Percent: " << (StepsDone/float(TotalSteps)*100.0) << std::endl
	206	<< "Message: " << ConfMessage << std::endl
	207	<< std::endl;
	208	WriteToStatusFd(status.str());
	209	}
	210
	211
	212	bool PackageManagerProgressDeb822Fd::StatusChanged(std::string PackageName,
	213	unsigned int xStepsDone,
	214	unsigned int xTotalSteps,
	215	std::string message)
	216	{
	217	StepsDone = xStepsDone;
	218	StepsTotal = xTotalSteps;
	219
	220	// build the status str
	221	std::ostringstream status;
	222	status << "Status: " << "progress" << std::endl
	223	<< "Package: " << PackageName << std::endl
	224	<< "Percent: " << (StepsDone/float(StepsTotal)*100.0) << std::endl
	225	<< "Message: " << message << std::endl
	226	<< std::endl;
	227	WriteToStatusFd(status.str());
	228
	229	return true;
	230	}
	231
5ed88785 MV	232
	233	PackageManagerFancy::PackageManagerFancy()
	234	: child_pty(-1)
	235	{
	236	// setup terminal size
	237	old_SIGWINCH = signal(SIGWINCH, PackageManagerFancy::staticSIGWINCH);
	238	instances.push_back(this);
	239	}
	240	std::vector<PackageManagerFancy*> PackageManagerFancy::instances;
	241
	242	PackageManagerFancy::~PackageManagerFancy()
	243	{
	244	instances.erase(find(instances.begin(), instances.end(), this));
	245	signal(SIGWINCH, old_SIGWINCH);
	246	}
	247
	248	void PackageManagerFancy::staticSIGWINCH(int signum)
	249	{
	250	std::vector<PackageManagerFancy *>::const_iterator I;
9ce3cfc9	251	for(I = instances.begin(); I != instances.end(); ++I)
5ed88785 MV	252	(*I)->HandleSIGWINCH(signum);
	253	}
	254
e96e4e9c MV	255	int PackageManagerFancy::GetNumberTerminalRows()
	256	{
	257	struct winsize win;
5ed88785	258	// FIXME: get from "child_pty" instead?
e96e4e9c MV	259	if(ioctl(STDOUT_FILENO, TIOCGWINSZ, (char *)&win) != 0)
e96e4e9c MV	260	return -1;
5ed88785 MV	261
	262	if(_config->FindB("Debug::InstallProgress::Fancy", false) == true)
	263	std::cerr << "GetNumberTerminalRows: " << win.ws_row << std::endl;
e96e4e9c MV	264
	265	return win.ws_row;
	266	}
c7ea1eba	267
db78c60c	268	void PackageManagerFancy::SetupTerminalScrollArea(int nr_rows)
31f97d7b	269	{
5ed88785 MV	270	if(_config->FindB("Debug::InstallProgress::Fancy", false) == true)
	271	std::cerr << "SetupTerminalScrollArea: " << nr_rows << std::endl;
	272
31f97d7b MV	273	// scroll down a bit to avoid visual glitch when the screen
	274	// area shrinks by one row
	275	std::cout << "\n";
	276
	277	// save cursor
	278	std::cout << "\033[s";
	279
	280	// set scroll region (this will place the cursor in the top left)
5ed88785	281	std::cout << "\033[0;" << nr_rows - 1 << "r";
31f97d7b MV	282
	283	// restore cursor but ensure its inside the scrolling area
	284	std::cout << "\033[u";
	285	static const char *move_cursor_up = "\033[1A";
	286	std::cout << move_cursor_up;
db78c60c	287
5ed88785	288	// ensure its flushed
31f97d7b	289	std::flush(std::cout);
31f97d7b	290
5ed88785 MV	291	// setup tty size to ensure xterm/linux console are working properly too
	292	// see bug #731738
	293	struct winsize win;
	294	ioctl(child_pty, TIOCGWINSZ, (char *)&win);
	295	win.ws_row = nr_rows - 1;
	296	ioctl(child_pty, TIOCSWINSZ, (char *)&win);
e96e4e9c MV	297	}
	298
	299	void PackageManagerFancy::HandleSIGWINCH(int)
	300	{
	301	int nr_terminal_rows = GetNumberTerminalRows();
	302	SetupTerminalScrollArea(nr_terminal_rows);
31f97d7b MV	303	}
31f97d7b MV	304
5ed88785	305	void PackageManagerFancy::Start(int a_child_pty)
31f97d7b	306	{
5ed88785	307	child_pty = a_child_pty;
e96e4e9c	308	int nr_terminal_rows = GetNumberTerminalRows();
db78c60c MV	309	if (nr_terminal_rows > 0)
db78c60c MV	310	SetupTerminalScrollArea(nr_terminal_rows);
31f97d7b MV	311	}
31f97d7b MV	312
a22fdebf	313	void PackageManagerFancy::Stop()
31f97d7b	314	{
e96e4e9c	315	int nr_terminal_rows = GetNumberTerminalRows();
db78c60c MV	316	if (nr_terminal_rows > 0)
	317	{
	318	SetupTerminalScrollArea(nr_terminal_rows + 1);
31f97d7b	319
db78c60c MV	320	// override the progress line (sledgehammer)
	321	static const char* clear_screen_below_cursor = "\033[J";
	322	std::cout << clear_screen_below_cursor;
	323	}
5ed88785	324	child_pty = -1;
31f97d7b MV	325	}
31f97d7b MV	326
6c5ae8ed	327	bool PackageManagerFancy::StatusChanged(std::string PackageName,
31f97d7b	328	unsigned int StepsDone,
e6ad8031 MV	329	unsigned int TotalSteps,
e6ad8031 MV	330	std::string HumanReadableAction)
31f97d7b	331	{
e6ad8031 MV	332	if (!PackageManager::StatusChanged(PackageName, StepsDone, TotalSteps,
e6ad8031 MV	333	HumanReadableAction))
6c5ae8ed	334	return false;
31f97d7b	335
e96e4e9c	336	int row = GetNumberTerminalRows();
31f97d7b	337
453b82a3 DK	338	static std::string save_cursor = "\033[s";
	339	static std::string restore_cursor = "\033[u";
	340
	341	static std::string set_bg_color = "\033[42m"; // green
	342	static std::string set_fg_color = "\033[30m"; // black
	343
	344	static std::string restore_bg = "\033[49m";
	345	static std::string restore_fg = "\033[39m";
	346
31f97d7b MV	347	std::cout << save_cursor
	348	// move cursor position to last row
	349	<< "\033[" << row << ";0f"
	350	<< set_bg_color
	351	<< set_fg_color
	352	<< progress_str
	353	<< restore_cursor
	354	<< restore_bg
	355	<< restore_fg;
	356	std::flush(std::cout);
	357	last_reported_progress = percentage;
6c5ae8ed MV	358
6c5ae8ed MV	359	return true;
31f97d7b MV	360	}
31f97d7b MV	361
6c5ae8ed	362	bool PackageManagerText::StatusChanged(std::string PackageName,
31f97d7b	363	unsigned int StepsDone,
e6ad8031 MV	364	unsigned int TotalSteps,
e6ad8031 MV	365	std::string HumanReadableAction)
31f97d7b	366	{
e6ad8031	367	if (!PackageManager::StatusChanged(PackageName, StepsDone, TotalSteps, HumanReadableAction))
6c5ae8ed	368	return false;
31f97d7b MV	369
	370	std::cout << progress_str << "\r\n";
	371	std::flush(std::cout);
	372
	373	last_reported_progress = percentage;
6c5ae8ed MV	374
6c5ae8ed MV	375	return true;
31f97d7b MV	376	}
	377
	378
c7ea1eba	379
d3e8fbb3 DK	380	} // namespace progress
d3e8fbb3 DK	381	} // namespace apt