diff options
Diffstat (limited to 'test/scratch.cc')
| -rw-r--r-- | test/scratch.cc | 200 |
1 files changed, 197 insertions, 3 deletions
diff --git a/test/scratch.cc b/test/scratch.cc index 53681db16..c9ddc2ea4 100644 --- a/test/scratch.cc +++ b/test/scratch.cc @@ -1,9 +1,202 @@ #include <apt-pkg/acquire-item.h> +#include <apt-pkg/acquire-worker.h> #include <apt-pkg/init.h> #include <apt-pkg/error.h> -#include <signal.h> #include <strutl.h> +#include <signal.h> +#include <stdio.h> + +class AcqTextStatus : public pkgAcquireStatus +{ + unsigned int ScreenWidth; + char BlankLine[300]; + unsigned long ID; + + public: + + virtual void IMSHit(pkgAcquire::ItemDesc &Itm); + virtual void Fetch(pkgAcquire::ItemDesc &Itm); + virtual void Done(pkgAcquire::ItemDesc &Itm); + virtual void Fail(pkgAcquire::ItemDesc &Itm); + virtual void Start() {pkgAcquireStatus::Start(); BlankLine[0] = 0; ID = 1;}; + virtual void Stop(); + + void Pulse(pkgAcquire *Owner); +}; + +// AcqTextStatus::IMSHit - Called when an item got a HIT response /*{{{*/ +// --------------------------------------------------------------------- +/* */ +void AcqTextStatus::IMSHit(pkgAcquire::ItemDesc &Itm) +{ + cout << '\r' << BlankLine << '\r'; + cout << "Hit " << Itm.Description; + if (Itm.Owner->FileSize != 0) + cout << " [" << SizeToStr(Itm.Owner->FileSize) << ']'; + 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++; + + cout << '\r' << BlankLine << '\r'; + cout << hex << Itm.Owner->ID << dec << " Get " << Itm.Description; + if (Itm.Owner->FileSize != 0) + cout << " [" << SizeToStr(Itm.Owner->FileSize) << ']'; + 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) +{ + cout << '\r' << BlankLine << '\r'; + 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(); + cout << '\r' << BlankLine << '\r'; + + if (FetchedBytes == 0) + cout << flush; + else + cout << "Fetched " << SizeToStr(FetchedBytes) << " in " << + TimeToStr(ElapsedTime) << " (" << SizeToStr(CurrentCPS) << + "/s)" << endl; +} + /*}}}*/ +// 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. */ +void AcqTextStatus::Pulse(pkgAcquire *Owner) +{ + pkgAcquireStatus::Pulse(Owner); + + enum {Long = 0,Medium,Short} Mode = Long; + + ScreenWidth = 140; + char Buffer[300]; + char *End = Buffer + sizeof(Buffer); + char *S = Buffer; + + // Put in the percent done + sprintf(S,"%ld%%",long(double(CurrentBytes*100.0)/double(TotalBytes))); + + 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()); + continue; + } + + // Add in the short description + if (I->CurrentItem->Owner->ID != 0) + snprintf(S,End-S," [%x %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," %u",I->CurrentSize); + else + { + if (Mode == Medium || I->TotalSize == 0) + snprintf(S,End-S," %s",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 == Long) + snprintf(S,End-S,"/%u %u%%",I->TotalSize, + long(double(I->CurrentSize*100.0)/double(I->TotalSize))); + else + { + if (Mode == Medium) + snprintf(S,End-S,"/%s %u%%",SizeToStr(I->TotalSize).c_str(), + long(double(I->CurrentSize*100.0)/double(I->TotalSize))); + else + snprintf(S,End-S," %u%%", + long(double(I->CurrentSize*100.0)/double(I->TotalSize))); + } + } + S += strlen(S); + snprintf(S,End-S,"]"); + } + + // Put in the ETA and cps meter + if (CurrentCPS != 0) + { + char Tmp[300]; + unsigned long ETA = (unsigned long)((TotalBytes - CurrentBytes)/CurrentCPS); + sprintf(Tmp," %s/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; + + // Draw the current status + if (strlen(Buffer) == strlen(BlankLine)) + cout << '\r' << Buffer << flush; + else + cout << '\r' << BlankLine << '\r' << Buffer << flush; + memset(BlankLine,' ',strlen(Buffer)); + BlankLine[strlen(Buffer)] = 0; + + Update = false; +} + /*}}}*/ + int main(int argc,char *argv[]) { signal(SIGPIPE,SIG_IGN); @@ -21,7 +214,8 @@ int main(int argc,char *argv[]) pkgInitialize(*_config); pkgSourceList List; - pkgAcquire Fetcher; + AcqTextStatus Stat; + pkgAcquire Fetcher(&Stat); List.ReadMainList(); pkgSourceList::const_iterator I; @@ -31,7 +225,7 @@ int main(int argc,char *argv[]) if (_error->PendingError() == true) break; } - + Fetcher.Run(); _error->DumpErrors(); |