FildFd: Introduce a Flush() function and call it from Close()

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

diff --git a/apt-private/acqprogress.cc b/apt-private/acqprogress.cc
index 14a53eacb495e61078baf2d36a542e4bbb0184a2..a33d51c7178d60d631586c03f768d2c46ed74dcf 100644 (file)
--- a/apt-private/acqprogress.cc
+++ b/apt-private/acqprogress.cc
@@ -1,10 +1,9 @@
 // -*- mode: cpp; mode: fold -*-
 // Description                                                         /*{{{*/
 // -*- 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 
-   
+   Acquire Progress - Command line progress meter
+

    ##################################################################### */
                                                                        /*}}}*/
 // Include files                                                       /*{{{*/
    ##################################################################### */
                                                                        /*}}}*/
 // Include files                                                       /*{{{*/


@@ -23,23 +22,23 @@
 #include <stdio.h>
 #include <signal.h>
 #include <iostream>
 #include <stdio.h>
 #include <signal.h>
 #include <iostream>

+#include <sstream>

 #include <unistd.h>
 
 #include <apti18n.h>
                                                                        /*}}}*/
 
 #include <unistd.h>
 
 #include <apti18n.h>
                                                                        /*}}}*/
 

-using namespace std;
-

 // AcqTextStatus::AcqTextStatus - Constructor                          /*{{{*/
 // ---------------------------------------------------------------------
 /* */
 // AcqTextStatus::AcqTextStatus - Constructor                          /*{{{*/
 // ---------------------------------------------------------------------
 /* */

-AcqTextStatus::AcqTextStatus(unsigned int &ScreenWidth,unsigned int const Quiet) :
-    pkgAcquireStatus(), ScreenWidth(ScreenWidth), ID(0), Quiet(Quiet)
+AcqTextStatus::AcqTextStatus(std::ostream &out, unsigned int &ScreenWidth,unsigned int const Quiet) :
+    pkgAcquireStatus(), out(out), ScreenWidth(ScreenWidth), LastLineLength(0), 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;
    // testcases use it to disable pulses without disabling other user messages
    if (Quiet == 0 && _config->FindB("quiet::NoUpdate", false) == true)
       this->Quiet = 1;

+   if (Quiet < 2 && _config->FindB("quiet::NoProgress", false) == true)
+      this->Quiet = 2;

 }
                                                                        /*}}}*/
 // AcqTextStatus::Start - Downloading has started                      /*{{{*/
 }
                                                                        /*}}}*/
 // AcqTextStatus::Start - Downloading has started                      /*{{{*/


@@ -48,10 +47,20 @@ AcqTextStatus::AcqTextStatus(unsigned int &ScreenWidth,unsigned int const Quiet)
 void AcqTextStatus::Start()
 {
    pkgAcquireStatus::Start();
 void AcqTextStatus::Start()
 {
    pkgAcquireStatus::Start();

-   BlankLine[0] = 0;
+   LastLineLength = 0;

    ID = 1;
 }
                                                                        /*}}}*/
    ID = 1;
 }
                                                                        /*}}}*/

+void AcqTextStatus::AssignItemID(pkgAcquire::ItemDesc &Itm)            /*{{{*/
+{
+   /* In theory calling it from Fetch() would be enough, but to be
+      safe we call it from IMSHit and Fail as well.
+      Also, an Item can pass through multiple stages, so ensure
+      that it keeps the same number */
+   if (Itm.Owner->ID == 0)
+      Itm.Owner->ID = ID++;
+}
+                                                                       /*}}}*/

 // AcqTextStatus::IMSHit - Called when an item got a HIT response      /*{{{*/
 // ---------------------------------------------------------------------
 /* */
 // AcqTextStatus::IMSHit - Called when an item got a HIT response      /*{{{*/
 // ---------------------------------------------------------------------
 /* */


@@ -60,11 +69,12 @@ void AcqTextStatus::IMSHit(pkgAcquire::ItemDesc &Itm)
    if (Quiet > 1)
       return;
 
    if (Quiet > 1)
       return;
 

-   if (Quiet <= 0)
-      cout << '\r' << BlankLine << '\r';
+   AssignItemID(Itm);
+   clearLastLine();

 
 

-   cout << _("Hit ") << Itm.Description;
-   cout << endl;
+   // TRANSLATOR: Very short word to be displayed before unchanged files in 'apt-get update'
+   ioprintf(out, _("Hit:%lu %s"), Itm.Owner->ID, Itm.Description.c_str());
+   out << std::endl;

    Update = true;
 }
                                                                        /*}}}*/
    Update = true;
 }
                                                                        /*}}}*/


@@ -76,27 +86,28 @@ void AcqTextStatus::Fetch(pkgAcquire::ItemDesc &Itm)
    Update = true;
    if (Itm.Owner->Complete == true)
       return;
    Update = true;
    if (Itm.Owner->Complete == true)
       return;

-
-   Itm.Owner->ID = ID++;
+   AssignItemID(Itm);

 
    if (Quiet > 1)
       return;
 
 
    if (Quiet > 1)
       return;
 

-   if (Quiet <= 0)
-      cout << '\r' << BlankLine << '\r';
+   clearLastLine();

 
 

-   cout << _("Get:") << Itm.Owner->ID << ' ' << Itm.Description;
+   // TRANSLATOR: Very short word to be displayed for files processed in 'apt-get update'
+   // Potentially replaced later by "Hit:", "Ign:" or "Err:" if something (bad) happens
+   ioprintf(out, _("Get:%lu %s"), Itm.Owner->ID, Itm.Description.c_str());

    if (Itm.Owner->FileSize != 0)
    if (Itm.Owner->FileSize != 0)

-      cout << " [" << SizeToStr(Itm.Owner->FileSize) << "B]";
-   cout << endl;
+      out << " [" << SizeToStr(Itm.Owner->FileSize) << "B]";
+   out << std::endl;

 }
                                                                        /*}}}*/
 // AcqTextStatus::Done - Completed a download                          /*{{{*/
 // ---------------------------------------------------------------------
 /* We don't display anything... */
 }
                                                                        /*}}}*/
 // AcqTextStatus::Done - Completed a download                          /*{{{*/
 // ---------------------------------------------------------------------
 /* We don't display anything... */

-void AcqTextStatus::Done(pkgAcquire::ItemDesc &/*Itm*/)
+void AcqTextStatus::Done(pkgAcquire::ItemDesc &Itm)

 {
    Update = true;
 {
    Update = true;

+   AssignItemID(Itm);

 }
                                                                        /*}}}*/
 // AcqTextStatus::Fail - Called when an item fails to download         /*{{{*/
 }
                                                                        /*}}}*/
 // AcqTextStatus::Fail - Called when an item fails to download         /*{{{*/


@@ -107,24 +118,25 @@ void AcqTextStatus::Fail(pkgAcquire::ItemDesc &Itm)
    if (Quiet > 1)
       return;
 
    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';
+   AssignItemID(Itm);
+   clearLastLine();

 
 

-   if (Itm.Owner->Status == pkgAcquire::Item::StatDone)
+   if (Itm.Owner->Status == pkgAcquire::Item::StatDone || Itm.Owner->Status == pkgAcquire::Item::StatIdle)

    {
    {

-      cout << _("Ign ") << Itm.Description << endl;
+      // TRANSLATOR: Very short word to be displayed for files in 'apt-get update'
+      // which failed to download, but the error is ignored (compare "Err:")
+      ioprintf(out, _("Ign:%lu %s"), Itm.Owner->ID, Itm.Description.c_str());

       if (Itm.Owner->ErrorText.empty() == false &&
            _config->FindB("Acquire::Progress::Ignore::ShowErrorText", false) == true)
       if (Itm.Owner->ErrorText.empty() == false &&
            _config->FindB("Acquire::Progress::Ignore::ShowErrorText", false) == true)

-        cout << "  " << Itm.Owner->ErrorText << endl;
+        out << std::endl << "  " << Itm.Owner->ErrorText;
+      out << std::endl;

    }
    else
    {
    }
    else
    {

-      cout << _("Err ") << Itm.Description << endl;
-      cout << "  " << Itm.Owner->ErrorText << endl;
+      // TRANSLATOR: Very short word to be displayed for files in 'apt-get update'
+      // which failed to download and the error is critical (compare "Ign:")
+      ioprintf(out, _("Err:%lu %s"), Itm.Owner->ID, Itm.Description.c_str());
+      out << std::endl << "  " << Itm.Owner->ErrorText << std::endl;

    }
 
    Update = true;
    }
 
    Update = true;


@@ -140,14 +152,13 @@ void AcqTextStatus::Stop()
    if (Quiet > 1)
       return;
 
    if (Quiet > 1)
       return;
 

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

 
    if (_config->FindB("quiet::NoStatistic", false) == true)
       return;
 
    if (FetchedBytes != 0 && _error->PendingError() == false)
 
    if (_config->FindB("quiet::NoStatistic", false) == true)
       return;
 
    if (FetchedBytes != 0 && _error->PendingError() == false)

-      ioprintf(cout,_("Fetched %sB in %s (%sB/s)\n"),
+      ioprintf(out,_("Fetched %sB in %s (%sB/s)\n"),

               SizeToStr(FetchedBytes).c_str(),
               TimeToStr(ElapsedTime).c_str(),
               SizeToStr(CurrentCPS).c_str());
               SizeToStr(FetchedBytes).c_str(),
               TimeToStr(ElapsedTime).c_str(),
               SizeToStr(CurrentCPS).c_str());


@@ -156,7 +167,7 @@ void AcqTextStatus::Stop()
 // AcqTextStatus::Pulse - Regular event pulse                          /*{{{*/
 // ---------------------------------------------------------------------
 /* This draws the current progress. Each line has an overall percent
 // 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 
+   meter and a per active item status meter along with an overall

    bandwidth and ETA indicator. */
 bool AcqTextStatus::Pulse(pkgAcquire *Owner)
 {
    bandwidth and ETA indicator. */
 bool AcqTextStatus::Pulse(pkgAcquire *Owner)
 {


@@ -165,79 +176,67 @@ bool AcqTextStatus::Pulse(pkgAcquire *Owner)
    if (Quiet > 0)
       return true;
 
    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%%", Percent);
-
-   bool Shown = false;
-   for (pkgAcquire::Worker *I = Owner->WorkersBegin(); I != 0;
-       I = Owner->WorkerStep(I))
+   std::string Line;

    {
    {

-      S += strlen(S);
-
-      // There is no item running
-      if (I->CurrentItem == 0)
+      std::stringstream S;
+      for (pkgAcquire::Worker *I = Owner->WorkersBegin(); I != 0;
+           I = Owner->WorkerStep(I))

       {
       {

-        if (I->Status.empty() == false)
+        // There is no item running
+        if (I->CurrentItem == 0)

         {
         {

-           snprintf(S,End-S," [%s]",I->Status.c_str());
-           Shown = true;
-        }
-
-        continue;
-      }
-
-      Shown = true;
+           if (I->Status.empty() == false)
+              S << " [" << I->Status << "]";

 
 

-      // 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);
+           continue;
+        }

 
 

-      // Show the short mode string
-      if (I->CurrentItem->Owner->ActiveSubprocess.empty() == false)
-      {
-        snprintf(S,End-S, " %s", I->CurrentItem->Owner->ActiveSubprocess.c_str());
-        S += strlen(S);
-      }
+        // Add in the short description
+        S << " [";
+        if (I->CurrentItem->Owner->ID != 0)
+           S << I->CurrentItem->Owner->ID << " ";
+        S << I->CurrentItem->ShortDesc;

 
 

-      // 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);
+        // Show the short mode string
+        if (I->CurrentItem->Owner->ActiveSubprocess.empty() == false)
+           S << " " << I->CurrentItem->Owner->ActiveSubprocess;

 
 

-      // 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);
+        enum {Long = 0,Medium,Short} Mode = Medium;
+        // Add the current progress
+        if (Mode == Long)
+           S << " " << I->CurrentSize;

         else
         else

-           snprintf(S,End-S,"/%sB %.0f%%",SizeToStr(I->TotalSize).c_str(),
+        {
+           if (Mode == Medium || I->TotalSize == 0)
+              S << " " << SizeToStr(I->CurrentSize) << "B";
+        }
+
+        // Add the total size and percent
+        if (I->TotalSize > 0 && I->CurrentItem->Owner->Complete == false)
+        {
+           if (Mode == Short)
+              ioprintf(S, " %.0f%%", (I->CurrentSize*100.0)/I->TotalSize);
+           else
+              ioprintf(S, "/%sB %.0f%%", SizeToStr(I->TotalSize).c_str(),

                     (I->CurrentSize*100.0)/I->TotalSize);
                     (I->CurrentSize*100.0)/I->TotalSize);

+        }
+        S << "]";

       }
       }

-      S += strlen(S);
-      snprintf(S,End-S,"]");
-   }

 
 

-   // Show something..
-   if (Shown == false)
-      snprintf(S,End-S,_(" [Working]"));
+      // Show at least something
+      Line = S.str();
+      S.clear();
+      if (Line.empty() == true)
+        Line = _(" [Working]");
+   }
+   // Put in the percent done
+   {
+      std::stringstream S;
+      ioprintf(S, "%.0f%%", Percent);
+      S << Line;
+      Line = S.str();
+      S.clear();
+   }

 
    /* Put in the ETA and cps meter, block off signals to prevent strangeness
       during resizing */
 
    /* Put in the ETA and cps meter, block off signals to prevent strangeness
       during resizing */


@@ -248,34 +247,33 @@ bool AcqTextStatus::Pulse(pkgAcquire *Owner)
 
    if (CurrentCPS != 0)
    {
 
    if (CurrentCPS != 0)
    {

-      char Tmp[300];

       unsigned long long ETA = (TotalBytes - CurrentBytes)/CurrentCPS;
       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)
+      std::string Tmp = " " + SizeToStr(CurrentCPS) + "B/s " + TimeToStr(ETA);
+      size_t alignment = Line.length() + Tmp.length();
+      if (alignment < ScreenWidth)

       {
       {

-        memset(Buffer + Len,' ',ScreenWidth - Len);
-        strcpy(Buffer + ScreenWidth - LenT,Tmp);
+        alignment = ScreenWidth - alignment;
+        for (size_t i = 0; i < alignment; ++i)
+           Line.append(" ");
+        Line.append(Tmp);

       }
    }
       }
    }

-   Buffer[ScreenWidth] = 0;
-   BlankLine[ScreenWidth] = 0;
+   if (Line.length() > ScreenWidth)
+      Line.erase(ScreenWidth);

    sigprocmask(SIG_SETMASK,&OldSigs,0);
 
    // Draw the current status
    if (_config->FindB("Apt::Color", false) == true)
    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;
+      out << _config->Find("APT::Color::Yellow");
+   if (LastLineLength > Line.length())
+      clearLastLine();

    else
    else

-      cout << '\r' << BlankLine << '\r' << Buffer << flush;
+      out << '\r';
+   out << Line << std::flush;

    if (_config->FindB("Apt::Color", false) == true)
    if (_config->FindB("Apt::Color", false) == true)

-      cout << _config->Find("APT::Color::Neutral") << flush;
-
-   memset(BlankLine,' ',strlen(Buffer));
-   BlankLine[strlen(Buffer)] = 0;
+      out << _config->Find("APT::Color::Neutral") << std::flush;

 
 

+   LastLineLength = Line.length();

    Update = false;
 
    return true;
    Update = false;
 
    return true;


@@ -284,7 +282,7 @@ bool AcqTextStatus::Pulse(pkgAcquire *Owner)
 // AcqTextStatus::MediaChange - Media need to be swapped               /*{{{*/
 // ---------------------------------------------------------------------
 /* Prompt for a media swap */
 // AcqTextStatus::MediaChange - Media need to be swapped               /*{{{*/
 // ---------------------------------------------------------------------
 /* Prompt for a media swap */

-bool AcqTextStatus::MediaChange(string Media,string Drive)
+bool AcqTextStatus::MediaChange(std::string Media, std::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
 {
    // 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


@@ -296,11 +294,10 @@ bool AcqTextStatus::MediaChange(string Media,string Drive)
 
       return false;
 
 
       return false;
 

-   if (Quiet <= 0)
-      cout << '\r' << BlankLine << '\r';
-   ioprintf(cout,_("Media change: please insert the disc labeled\n"
+   clearLastLine();
+   ioprintf(out,_("Media change: please insert the disc labeled\n"

                   " '%s'\n"
                   " '%s'\n"

-                  "in the drive '%s' and press enter\n"),
+                  "in the drive '%s' and press [Enter]\n"),

            Media.c_str(),Drive.c_str());
 
    char C = 0;
            Media.c_str(),Drive.c_str());
 
    char C = 0;


@@ -317,3 +314,17 @@ bool AcqTextStatus::MediaChange(string Media,string Drive)
    return bStatus;
 }
                                                                        /*}}}*/
    return bStatus;
 }
                                                                        /*}}}*/

+void AcqTextStatus::clearLastLine() {                                  /*{{{*/
+   if (Quiet > 0 || LastLineLength == 0)
+      return;
+
+   // do not try to clear more than the (now smaller) screen
+   if (LastLineLength > ScreenWidth)
+      LastLineLength = ScreenWidth;
+
+   out << '\r';
+   for (size_t i = 0; i < LastLineLength; ++i)
+      out << ' ';
+   out << '\r' << std::flush;
+}
+                                                                       /*}}}*/