gbdbupdate does bugs too and background option

1 files changed, 11 insertions, 1 deletions

diff --git a/gentoobrowse-api/util/update.cpp b/gentoobrowse-api/util/update.cpp
index 43f894a..bc0082d 100644
--- a/gentoobrowse-api/util/update.cpp
+++ b/gentoobrowse-api/util/update.cpp
@@ -8,9 +8,11 @@ int
 main(int c, char ** v)
 {
 	std::string endpoint;
+	bool background;
 	po::options_description opts("Gentoo Browse Util::Update options");
 	opts.add_options()
 		("endpoint", po::value(&endpoint)->default_value("tcp -p 9001"), "Service endpoint")
+		("background,b", po::value(&background)->default_value(false)->zero_tokens(), "Background")
 		("help,h", "Show help")
 		;
 
@@ -27,7 +29,15 @@ main(int c, char ** v)
 	auto m = Gentoo::MaintenancePrx::checkedCast(ic->stringToProxy("maintenance:" + endpoint));
 	m->ice_ping();
 
-	m->refreshPackageTree();
+	auto rpt = m->begin_refreshPackageTree();
+	auto rb = m->begin_refreshBugs();
+	rpt->waitForSent();
+	rb->waitForSent();
+
+	if (!background) {
+		m->end_refreshPackageTree(rpt);
+		m->end_refreshBugs(rb);
+	}
 
 	ic->destroy();
 	return 0;