[apt.git] / apt-private / acqprogress.cc

// -*- mode: cpp; mode: fold -*-
// Description								/*{{{*/
// $Id: acqprogress.cc,v 1.24 2003/04/27 01:56:48 doogie Exp $
/* ######################################################################

   Acquire Progress - Command line progress meter 
   
   ##################################################################### */
									/*}}}*/
// Include files							/*{{{*/
#include<config.h>

#include <apt-pkg/acquire-item.h>
#include <apt-pkg/acquire-worker.h>
#include <apt-pkg/configuration.h>
#include <apt-pkg/strutl.h>
#include <apt-pkg/error.h>

#include <stdio.h>
#include <signal.h>
#include <iostream>
#include <unistd.h>

#include "acqprogress.h"
#include <apti18n.h>
									/*}}}*/

using namespace std;

// AcqTextStatus::AcqTextStatus - Constructor				/*{{{*/
// ---------------------------------------------------------------------
/* */
AcqTextStatus::AcqTextStatus(unsigned int &ScreenWidth,unsigned int const Quiet) :
    pkgAcquireStatus(), ScreenWidth(ScreenWidth), ID(0), Quiet(Quiet)
{
   BlankLine[0] = 0;
   // testcases use it to disable pulses without disabling other user messages
   if (Quiet == 0 && _config->FindB("quiet::NoUpdate", false) == true)
      this->Quiet = 1;
}
									/*}}}*/
// AcqTextStatus::Start - Downloading has started			/*{{{*/
// ---------------------------------------------------------------------
/* */
void AcqTextStatus::Start() 
{
   pkgAcquireStatus::Start(); 
   BlankLine[0] = 0;
   ID = 1;
};
									/*}}}*/
// AcqTextStatus::IMSHit - Called when an item got a HIT response	/*{{{*/
// ---------------------------------------------------------------------
/* */
void AcqTextStatus::IMSHit(pkgAcquire::ItemDesc &Itm)
{
   if (Quiet > 1)
      return;

   if (Quiet <= 0)
      cout << '\r' << BlankLine << '\r';   
   
   cout << _("Hit ") << Itm.Description;
   if (Itm.Owner->FileSize != 0)
      cout << " [" << SizeToStr(Itm.Owner->FileSize) << "B]";
   cout << endl;
   Update = true;
};
									/*}}}*/
// AcqTextStatus::Fetch - An item has started to download		/*{{{*/
// ---------------------------------------------------------------------
/* This prints out the short description and the expected size */
void AcqTextStatus::Fetch(pkgAcquire::ItemDesc &Itm)
{
   Update = true;
   if (Itm.Owner->Complete == true)
      return;
   
   Itm.Owner->ID = ID++;
   
   if (Quiet > 1)
      return;

   if (Quiet <= 0)
      cout << '\r' << BlankLine << '\r';
   
   cout << _("Get:") << Itm.Owner->ID << ' ' << Itm.Description;
   if (Itm.Owner->FileSize != 0)
      cout << " [" << SizeToStr(Itm.Owner->FileSize) << "B]";
   cout << endl;
};
									/*}}}*/
// AcqTextStatus::Done - Completed a download				/*{{{*/
// ---------------------------------------------------------------------
/* We don't display anything... */
void AcqTextStatus::Done(pkgAcquire::ItemDesc &Itm)
{
   Update = true;
};
									/*}}}*/
// AcqTextStatus::Fail - Called when an item fails to download		/*{{{*/
// ---------------------------------------------------------------------
/* We print out the error text  */
void AcqTextStatus::Fail(pkgAcquire::ItemDesc &Itm)
{
   if (Quiet > 1)
      return;

   // Ignore certain kinds of transient failures (bad code)
   if (Itm.Owner->Status == pkgAcquire::Item::StatIdle)
      return;
      
   if (Quiet <= 0)
      cout << '\r' << BlankLine << '\r';
   
   if (Itm.Owner->Status == pkgAcquire::Item::StatDone)
   {
      cout << _("Ign ") << Itm.Description << endl;
   }
   else
   {
      cout << _("Err ") << Itm.Description << endl;
      cout << "  " << Itm.Owner->ErrorText << endl;
   }
   
   Update = true;
};
									/*}}}*/
// AcqTextStatus::Stop - Finished downloading				/*{{{*/
// ---------------------------------------------------------------------
/* This prints out the bytes downloaded and the overall average line
   speed */
void AcqTextStatus::Stop()
{
   pkgAcquireStatus::Stop();
   if (Quiet > 1)
      return;

   if (Quiet <= 0)
      cout << '\r' << BlankLine << '\r' << flush;

   if (FetchedBytes != 0 && _error->PendingError() == false)
      ioprintf(cout,_("Fetched %sB in %s (%sB/s)\n"),
	       SizeToStr(FetchedBytes).c_str(),
	       TimeToStr(ElapsedTime).c_str(),
	       SizeToStr(CurrentCPS).c_str());
}
									/*}}}*/
// AcqTextStatus::Pulse - Regular event pulse				/*{{{*/
// ---------------------------------------------------------------------
/* This draws the current progress. Each line has an overall percent
   meter and a per active item status meter along with an overall 
   bandwidth and ETA indicator. */
bool AcqTextStatus::Pulse(pkgAcquire *Owner)
{
   pkgAcquireStatus::Pulse(Owner);
   
   if (Quiet > 0)
      return true;
   
   enum {Long = 0,Medium,Short} Mode = Medium;
   
   char Buffer[sizeof(BlankLine)];
   char *End = Buffer + sizeof(Buffer);
   char *S = Buffer;
   if (ScreenWidth >= sizeof(Buffer))
      ScreenWidth = sizeof(Buffer)-1;

   // Put in the percent done
   sprintf(S,"%.0f%%",((CurrentBytes + CurrentItems)*100.0)/(TotalBytes+TotalItems));

   bool Shown = false;
   for (pkgAcquire::Worker *I = Owner->WorkersBegin(); I != 0;
	I = Owner->WorkerStep(I))
   {
      S += strlen(S);
      
      // There is no item running 
      if (I->CurrentItem == 0)
      {
	 if (I->Status.empty() == false)
	 {
	    snprintf(S,End-S," [%s]",I->Status.c_str());
	    Shown = true;
	 }
	 
	 continue;
      }

      Shown = true;
      
      // Add in the short description
      if (I->CurrentItem->Owner->ID != 0)
	 snprintf(S,End-S," [%lu %s",I->CurrentItem->Owner->ID,
		  I->CurrentItem->ShortDesc.c_str());
      else
	 snprintf(S,End-S," [%s",I->CurrentItem->ShortDesc.c_str());
      S += strlen(S);

      // Show the short mode string
      if (I->CurrentItem->Owner->Mode != 0)
      {
	 snprintf(S,End-S," %s",I->CurrentItem->Owner->Mode);
	 S += strlen(S);
      }
            
      // Add the current progress
      if (Mode == Long)
	 snprintf(S,End-S," %llu",I->CurrentSize);
      else
      {
	 if (Mode == Medium || I->TotalSize == 0)
	    snprintf(S,End-S," %sB",SizeToStr(I->CurrentSize).c_str());
      }
      S += strlen(S);
      
      // Add the total size and percent
      if (I->TotalSize > 0 && I->CurrentItem->Owner->Complete == false)
      {
	 if (Mode == Short)
	    snprintf(S,End-S," %.0f%%",
		     (I->CurrentSize*100.0)/I->TotalSize);
	 else
	    snprintf(S,End-S,"/%sB %.0f%%",SizeToStr(I->TotalSize).c_str(),
		     (I->CurrentSize*100.0)/I->TotalSize);
      }      
      S += strlen(S);
      snprintf(S,End-S,"]");
   }

   // Show something..
   if (Shown == false)
      snprintf(S,End-S,_(" [Working]"));
      
   /* Put in the ETA and cps meter, block off signals to prevent strangeness
      during resizing */
   sigset_t Sigs,OldSigs;
   sigemptyset(&Sigs);
   sigaddset(&Sigs,SIGWINCH);
   sigprocmask(SIG_BLOCK,&Sigs,&OldSigs);
   
   if (CurrentCPS != 0)
   {      
      char Tmp[300];
      unsigned long long ETA = (TotalBytes - CurrentBytes)/CurrentCPS;
      sprintf(Tmp," %sB/s %s",SizeToStr(CurrentCPS).c_str(),TimeToStr(ETA).c_str());
      unsigned int Len = strlen(Buffer);
      unsigned int LenT = strlen(Tmp);
      if (Len + LenT < ScreenWidth)
      {	 
	 memset(Buffer + Len,' ',ScreenWidth - Len);
	 strcpy(Buffer + ScreenWidth - LenT,Tmp);
      }      
   }
   Buffer[ScreenWidth] = 0;
   BlankLine[ScreenWidth] = 0;
   sigprocmask(SIG_SETMASK,&OldSigs,0);

   // Draw the current status
   if (_config->FindB("Apt::Color", false) == true)
      cout << _config->Find("APT::Color::Yellow");
   if (strlen(Buffer) == strlen(BlankLine))
      cout << '\r' << Buffer << flush;
   else
      cout << '\r' << BlankLine << '\r' << Buffer << flush;
   if (_config->FindB("Apt::Color", false) == true)
      cout << _config->Find("APT::Color::Neutral");

   memset(BlankLine,' ',strlen(Buffer));
   BlankLine[strlen(Buffer)] = 0;
   
   Update = false;

   return true;
}
									/*}}}*/
// AcqTextStatus::MediaChange - Media need to be swapped		/*{{{*/
// ---------------------------------------------------------------------
/* Prompt for a media swap */
bool AcqTextStatus::MediaChange(string Media,string Drive)
{
   // If we do not output on a terminal and one of the options to avoid user
   // interaction is given, we assume that no user is present who could react
   // on your media change request
   if (isatty(STDOUT_FILENO) != 1 && Quiet >= 2 &&
       (_config->FindB("APT::Get::Assume-Yes",false) == true ||
	_config->FindB("APT::Get::Force-Yes",false) == true ||
	_config->FindB("APT::Get::Trivial-Only",false) == true))

      return false;

   if (Quiet <= 0)
      cout << '\r' << BlankLine << '\r';
   ioprintf(cout,_("Media change: please insert the disc labeled\n"
		   " '%s'\n"
		   "in the drive '%s' and press enter\n"),
	    Media.c_str(),Drive.c_str());

   char C = 0;
   bool bStatus = true;
   while (C != '\n' && C != '\r')
   {
      int len = read(STDIN_FILENO,&C,1);
      if(C == 'c' || len <= 0)
	 bStatus = false;
   }

   if(bStatus)
      Update = true;
   return bStatus;
}
									/*}}}*/
Commit	Line	Data
b9179170 MV	1	// -- mode: cpp; mode: fold --
	2	// Description /{{{/
	3	// $Id: acqprogress.cc,v 1.24 2003/04/27 01:56:48 doogie Exp $
	4	/* ######################################################################
	5
	6	Acquire Progress - Command line progress meter
	7
	8	##################################################################### */
	9	/}}}/
	10	// Include files /{{{/
	11	#include<config.h>
	12
	13	#include <apt-pkg/acquire-item.h>
	14	#include <apt-pkg/acquire-worker.h>
	15	#include <apt-pkg/configuration.h>
	16	#include <apt-pkg/strutl.h>
	17	#include <apt-pkg/error.h>
	18
	19	#include <stdio.h>
	20	#include <signal.h>
	21	#include <iostream>
	22	#include <unistd.h>
	23
	24	#include "acqprogress.h"
	25	#include <apti18n.h>
	26	/}}}/
	27
	28	using namespace std;
	29
	30	// AcqTextStatus::AcqTextStatus - Constructor /{{{/
	31	// ---------------------------------------------------------------------
	32	/* */
18cce398 DK	33	AcqTextStatus::AcqTextStatus(unsigned int &ScreenWidth,unsigned int const Quiet) :
18cce398 DK	34	pkgAcquireStatus(), ScreenWidth(ScreenWidth), ID(0), Quiet(Quiet)
b9179170 MV	35	{
b9179170 MV	36	BlankLine[0] = 0;
18cce398 DK	37	// testcases use it to disable pulses without disabling other user messages
	38	if (Quiet == 0 && _config->FindB("quiet::NoUpdate", false) == true)
	39	this->Quiet = 1;
b9179170 MV	40	}
	41	/}}}/
	42	// AcqTextStatus::Start - Downloading has started /{{{/
	43	// ---------------------------------------------------------------------
	44	/* */
	45	void AcqTextStatus::Start()
	46	{
	47	pkgAcquireStatus::Start();
	48	BlankLine[0] = 0;
	49	ID = 1;
	50	};
	51	/}}}/
	52	// AcqTextStatus::IMSHit - Called when an item got a HIT response /{{{/
	53	// ---------------------------------------------------------------------
	54	/* */
	55	void AcqTextStatus::IMSHit(pkgAcquire::ItemDesc &Itm)
	56	{
	57	if (Quiet > 1)
	58	return;
	59
	60	if (Quiet <= 0)
	61	cout << '\r' << BlankLine << '\r';
	62
	63	cout << _("Hit ") << Itm.Description;
	64	if (Itm.Owner->FileSize != 0)
	65	cout << " [" << SizeToStr(Itm.Owner->FileSize) << "B]";
	66	cout << endl;
	67	Update = true;
	68	};
	69	/}}}/
	70	// AcqTextStatus::Fetch - An item has started to download /{{{/
	71	// ---------------------------------------------------------------------
	72	/* This prints out the short description and the expected size */
	73	void AcqTextStatus::Fetch(pkgAcquire::ItemDesc &Itm)
	74	{
	75	Update = true;
	76	if (Itm.Owner->Complete == true)
	77	return;
	78
	79	Itm.Owner->ID = ID++;
	80
	81	if (Quiet > 1)
	82	return;
	83
	84	if (Quiet <= 0)
	85	cout << '\r' << BlankLine << '\r';
	86
	87	cout << _("Get:") << Itm.Owner->ID << ' ' << Itm.Description;
	88	if (Itm.Owner->FileSize != 0)
	89	cout << " [" << SizeToStr(Itm.Owner->FileSize) << "B]";
	90	cout << endl;
	91	};
	92	/}}}/
	93	// AcqTextStatus::Done - Completed a download /{{{/
	94	// ---------------------------------------------------------------------
	95	/* We don't display anything... */
	96	void AcqTextStatus::Done(pkgAcquire::ItemDesc &Itm)
	97	{
	98	Update = true;
	99	};
	100	/}}}/
	101	// AcqTextStatus::Fail - Called when an item fails to download /{{{/
	102	// ---------------------------------------------------------------------
	103	/* We print out the error text */
104	void AcqTextStatus::Fail(pkgAcquire::ItemDesc &Itm)
105	{
106	if (Quiet > 1)
107	return;
108
109	// Ignore certain kinds of transient failures (bad code)
110	if (Itm.Owner->Status == pkgAcquire::Item::StatIdle)
111	return;
112
113	if (Quiet <= 0)
114	cout << '\r' << BlankLine << '\r';
115
116	if (Itm.Owner->Status == pkgAcquire::Item::StatDone)
117	{
118	cout << _("Ign ") << Itm.Description << endl;
119	}
120	else
121	{
122	cout << _("Err ") << Itm.Description << endl;
123	cout << " " << Itm.Owner->ErrorText << endl;
124	}
125
126	Update = true;
127	};
128	/}}}/
129	// AcqTextStatus::Stop - Finished downloading /{{{/
130	// ---------------------------------------------------------------------
131	/* This prints out the bytes downloaded and the overall average line
132	speed */
133	void AcqTextStatus::Stop()
134	{
135	pkgAcquireStatus::Stop();
136	if (Quiet > 1)
137	return;
138
139	if (Quiet <= 0)
140	cout << '\r' << BlankLine << '\r' << flush;
141
142	if (FetchedBytes != 0 && _error->PendingError() == false)
143	ioprintf(cout,_("Fetched %sB in %s (%sB/s)\n"),
144	SizeToStr(FetchedBytes).c_str(),
145	TimeToStr(ElapsedTime).c_str(),
146	SizeToStr(CurrentCPS).c_str());
147	}
148	/}}}/
149	// AcqTextStatus::Pulse - Regular event pulse /{{{/
150	// ---------------------------------------------------------------------
151	/* This draws the current progress. Each line has an overall percent
152	meter and a per active item status meter along with an overall
153	bandwidth and ETA indicator. */
154	bool AcqTextStatus::Pulse(pkgAcquire *Owner)
155	{
156	pkgAcquireStatus::Pulse(Owner);
157
158	if (Quiet > 0)
159	return true;
160
161	enum {Long = 0,Medium,Short} Mode = Medium;
162
163	char Buffer[sizeof(BlankLine)];
164	char *End = Buffer + sizeof(Buffer);
165	char *S = Buffer;
166	if (ScreenWidth >= sizeof(Buffer))
167	ScreenWidth = sizeof(Buffer)-1;
168
169	// Put in the percent done
170	sprintf(S,"%.0f%%",((CurrentBytes + CurrentItems)*100.0)/(TotalBytes+TotalItems));
171
172	bool Shown = false;
173	for (pkgAcquire::Worker *I = Owner->WorkersBegin(); I != 0;
174	I = Owner->WorkerStep(I))
175	{
176	S += strlen(S);
177
178	// There is no item running
179	if (I->CurrentItem == 0)
180	{
181	if (I->Status.empty() == false)
182	{
183	snprintf(S,End-S," [%s]",I->Status.c_str());
184	Shown = true;
185	}
186
187	continue;
188	}
189
190	Shown = true;
191
192	// Add in the short description
193	if (I->CurrentItem->Owner->ID != 0)
194	snprintf(S,End-S," [%lu %s",I->CurrentItem->Owner->ID,
195	I->CurrentItem->ShortDesc.c_str());
196	else
197	snprintf(S,End-S," [%s",I->CurrentItem->ShortDesc.c_str());
198	S += strlen(S);
199
200	// Show the short mode string
201	if (I->CurrentItem->Owner->Mode != 0)
202	{
203	snprintf(S,End-S," %s",I->CurrentItem->Owner->Mode);
204	S += strlen(S);
205	}
206
207	// Add the current progress
208	if (Mode == Long)
209	snprintf(S,End-S," %llu",I->CurrentSize);
210	else
211	{
212	if (Mode == Medium \|\| I->TotalSize == 0)
213	snprintf(S,End-S," %sB",SizeToStr(I->CurrentSize).c_str());
214	}
215	S += strlen(S);
216
217	// Add the total size and percent
218	if (I->TotalSize > 0 && I->CurrentItem->Owner->Complete == false)
219	{
220	if (Mode == Short)
221	snprintf(S,End-S," %.0f%%",
222	(I->CurrentSize*100.0)/I->TotalSize);
223	else
224	snprintf(S,End-S,"/%sB %.0f%%",SizeToStr(I->TotalSize).c_str(),
225	(I->CurrentSize*100.0)/I->TotalSize);
226	}
227	S += strlen(S);
228	snprintf(S,End-S,"]");
229	}
230
231	// Show something..
232	if (Shown == false)
233	snprintf(S,End-S,_(" [Working]"));
234
235	/* Put in the ETA and cps meter, block off signals to prevent strangeness
236	during resizing */
237	sigset_t Sigs,OldSigs;
238	sigemptyset(&Sigs);
239	sigaddset(&Sigs,SIGWINCH);
240	sigprocmask(SIG_BLOCK,&Sigs,&OldSigs);
241
242	if (CurrentCPS != 0)
243	{
244	char Tmp[300];
245	unsigned long long ETA = (TotalBytes - CurrentBytes)/CurrentCPS;
246	sprintf(Tmp," %sB/s %s",SizeToStr(CurrentCPS).c_str(),TimeToStr(ETA).c_str());
247	unsigned int Len = strlen(Buffer);
248	unsigned int LenT = strlen(Tmp);
249	if (Len + LenT < ScreenWidth)
250	{
251	memset(Buffer + Len,' ',ScreenWidth - Len);
252	strcpy(Buffer + ScreenWidth - LenT,Tmp);
253	}
254	}
255	Buffer[ScreenWidth] = 0;
256	BlankLine[ScreenWidth] = 0;
257	sigprocmask(SIG_SETMASK,&OldSigs,0);
258
259	// Draw the current status
260	if (_config->FindB("Apt::Color", false) == true)
261	cout << _config->Find("APT::Color::Yellow");
262	if (strlen(Buffer) == strlen(BlankLine))
263	cout << '\r' << Buffer << flush;
264	else
265	cout << '\r' << BlankLine << '\r' << Buffer << flush;
266	if (_config->FindB("Apt::Color", false) == true)
267	cout << _config->Find("APT::Color::Neutral");
268
269	memset(BlankLine,' ',strlen(Buffer));
270	BlankLine[strlen(Buffer)] = 0;
271
272	Update = false;
273
274	return true;
275	}
276	/}}}/
277	// AcqTextStatus::MediaChange - Media need to be swapped /{{{/
278	// ---------------------------------------------------------------------
279	/* Prompt for a media swap */
280	bool AcqTextStatus::MediaChange(string Media,string Drive)
281	{
282	// If we do not output on a terminal and one of the options to avoid user
283	// interaction is given, we assume that no user is present who could react
284	// on your media change request
285	if (isatty(STDOUT_FILENO) != 1 && Quiet >= 2 &&
286	(_config->FindB("APT::Get::Assume-Yes",false) == true \|\|
287	_config->FindB("APT::Get::Force-Yes",false) == true \|\|
288	_config->FindB("APT::Get::Trivial-Only",false) == true))
289
290	return false;
291
292	if (Quiet <= 0)
293	cout << '\r' << BlankLine << '\r';
294	ioprintf(cout,_("Media change: please insert the disc labeled\n"
295	" '%s'\n"
296	"in the drive '%s' and press enter\n"),
297	Media.c_str(),Drive.c_str());
298
299	char C = 0;
300	bool bStatus = true;
301	while (C != '\n' && C != '\r')
302	{
303	int len = read(STDIN_FILENO,&C,1);
304	if(C == 'c' \|\| len <= 0)
305	bStatus = false;
306	}
307
308	if(bStatus)
309	Update = true;
310	return bStatus;
311	}
312	/}}}/