mirror of
https://github.com/MrLowPr/overdrive.git
synced 2026-10-18 21:01:17 +02:00
57 lines
3.4 KiB
C++
57 lines
3.4 KiB
C++
#include <dirent.h>
|
|
#include "archives.hpp"
|
|
#include "archive_preflight.hpp"
|
|
#include <chrono>
|
|
#include <cstdlib>
|
|
#include <iostream>
|
|
#include <thread>
|
|
// ARCHIVE_TRACE=1 prints the extraction steps, the throughput breakdown among them.
|
|
void archives::trace(const char* step,const char* detail){if(std::getenv("ARCHIVE_TRACE"))std::cerr<<step<<" "<<detail<<"\n";}
|
|
// ARCHIVE_PIPELINE_BYTES=0 exercises the inline writer.
|
|
std::size_t archives::pipeline_bytes(){const char* v=std::getenv("ARCHIVE_PIPELINE_BYTES");return v?std::strtoull(v,nullptr,10):32u<<20;}
|
|
// ARCHIVE_NO_MEMORY=1 reports a heap without room for an in-process extraction.
|
|
bool archives::hold_memory(bool hold){return !hold||!std::getenv("ARCHIVE_NO_MEMORY");}
|
|
// ARCHIVE_DOWNLOADING=1 asks rar-extract for the thread count used while downloads run.
|
|
bool archives::downloading(){return std::getenv("ARCHIVE_DOWNLOADING")!=nullptr;}
|
|
// True once `s` has ended. ARCHIVE_PHASES=1 prints to stderr each phase the polls see, then the files synced
|
|
// of the total and the declared total at the end.
|
|
bool ended(const archives::Snapshot& s){
|
|
const bool last=s.state=="completed"||s.state=="failed"||s.state=="cancelled";
|
|
static std::string seen;if(!std::getenv("ARCHIVE_PHASES"))return last;
|
|
if(s.phase!=seen)std::cerr<<"phase "<<(seen=s.phase)<<"\n";
|
|
if(last){std::cerr<<"synced "<<s.synced<<"/"<<s.sync_total<<"\ntotal "<<s.total<<"\n";seen.clear();}
|
|
return last;}
|
|
bool archives::list_directory(const char* path,void(*visit)(const char*,void*),void* user){
|
|
DIR* d=opendir(path);if(!d)return false;while(auto* e=readdir(d))visit(e->d_name,user);closedir(d);return true;}
|
|
int main(int argc,char** argv){
|
|
if(argc>2&&std::string(argv[1])=="inspect"){
|
|
int first=2;
|
|
if(argc>4&&std::string(argv[2])=="--budget"){archives::inspect_budget=std::chrono::milliseconds(std::stol(argv[3]));first=4;}
|
|
const auto inspection=archives::inspect(std::vector<std::string>(argv+first,argv+argc));
|
|
std::cout<<inspection.kind<<"\n"<<inspection.refusal<<"\n";return 0;}
|
|
if(argc<4)return 2;
|
|
archives::Request request;request.destination=argv[1];request.max_bytes=std::stoull(argv[2]);
|
|
// --stream: later volumes may appear while it runs; --cancel-after <ms> cancels it then; --again <worker>
|
|
// runs the same request once more in this process with that rar-extract.
|
|
long cancel_after=-1;const char* again=nullptr;
|
|
for(int i=3;i<argc;i++){
|
|
if(std::string(argv[i])=="--password"&&i+1<argc)request.password=argv[++i];
|
|
else if(std::string(argv[i])=="--stream")request.stream=true;
|
|
else if(std::string(argv[i])=="--cancel-after"&&i+1<argc)cancel_after=std::stol(argv[++i]);
|
|
else if(std::string(argv[i])=="--again"&&i+1<argc)again=argv[++i];
|
|
else request.sources.emplace_back(argv[i]);
|
|
}
|
|
const auto copy=request;
|
|
std::string error;auto id=archives::enqueue(std::move(request),error);
|
|
if(!id){std::cout<<"failed\n"<<error;return 0;}
|
|
auto started=std::chrono::steady_clock::now();
|
|
while(true){for(const auto& s:archives::poll())if(ended(s)){
|
|
std::cout<<s.state<<"\n"<<s.written<<"\n"<<s.error;
|
|
if(!again){archives::stop();return 0;}
|
|
std::cout<<"\n";setenv("PS5_REACT_RAR_WORKER",again,1);again=nullptr;cancel_after=-1;
|
|
if(!(id=archives::enqueue(copy,error))){std::cout<<"failed\n"<<error;return 0;}
|
|
started=std::chrono::steady_clock::now();}
|
|
if(cancel_after>=0&&std::chrono::steady_clock::now()-started>=std::chrono::milliseconds(cancel_after)){archives::cancel(id);cancel_after=-1;}
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(10));}
|
|
}
|