1 files changed, 80 insertions, 0 deletions
diff --git a/p2pvr/lib/maintenance/programAssociations.cpp b/p2pvr/lib/maintenance/programAssociations.cpp
new file mode 100644
index 0000000..4d32a86
--- /dev/null
+++ b/p2pvr/lib/maintenance/programAssociations.cpp
@@ -0,0 +1,80 @@
+#include "../maintenance.h"
+#include "../siParsers/programAssociation.h"
+#include <sqlMergeTask.h>
+#include <Ice/Communicator.h>
+#include "../p2Helpers.h"
+#include "../dvbsiHelpers.h"
+#include "../mapIterator.h"
+#include "../temporaryIceAdapterObject.h"
+
+class SiProgramAssociationHandler : public SiProgramAssociationParser {
+	public:
+		void HandleTable(ProgramAssociationMapPtr pam)
+		{
+			Logger()->messagebf(LOG_DEBUG, "Program association table");
+			BOOST_FOREACH(const auto & pa, *pam) {
+				Logger()->messagebf(LOG_DEBUG, "	%d -> %d", pa.first, pa.second);
+			}
+			BOOST_FOREACH(const auto & pa, *pam) {
+				map[pa.first] = pa.second;
+			}
+		}
+
+		ProgramAssociationMap map;
+};
+
+static
+void
+CreatePATColumns(const ColumnCreator & cc)
+{
+	cc("serviceId", true);
+	cc("programId", false);
+}
+
+void
+Maintenance::UpdateProgramAssociations(short type, const Ice::Current & ice)
+{
+	auto ic = ice.adapter->getCommunicator();
+	auto devs = P2PVR::DevicesPrx::checkedCast(ice.adapter->createProxy(ic->stringToIdentity("GlobalDevices")));
+	auto si = P2PVR::SIPrx::checkedCast(ice.adapter->createProxy(ic->stringToIdentity("SI")));
+
+	if (!devs || !si) {
+		throw std::runtime_error("bad proxy(s)");
+	}
+
+	auto siparser = new SiProgramAssociationHandler();
+	TemporarayIceAdapterObject<P2PVR::RawDataClient> parser(ice.adapter, siparser);
+
+	const auto deliveries = si->GetAllDeliveries(type);
+	if (deliveries.empty()) {
+		throw std::runtime_error("no delivery methods");
+	}
+
+	BOOST_FOREACH(const auto & transport, deliveries) {
+		try {
+			Logger()->messagebf(LOG_DEBUG, "%s: Getting a tuner", __PRETTY_FUNCTION__);
+			auto tuner = devs->GetTunerSpecific(transport, time(NULL) + 300);
+			Logger()->messagebf(LOG_DEBUG, "%s: Fetching associations", __PRETTY_FUNCTION__);
+			tuner->SendProgramAssociationTable(parser);
+			Logger()->messagebf(LOG_INFO, "%s: Updated associations", __PRETTY_FUNCTION__);
+			devs->ReleaseTuner(tuner);
+		}
+		catch (...) {
+			// Tuning can fail
+		}
+	}
+
+	SqlMergeTask mergeServices("postgres", "services");
+	CreatePATColumns(boost::bind(SqlMergeColumnsInserter, &mergeServices, _1, _2));
+	// Don't change the list of services available from the network
+	mergeServices.doDelete = VariableType(false);
+	mergeServices.doInsert = VariableType(false);
+	Columns cols;
+	mergeServices.sources.insert(new MapIterator<ProgramAssociationMap>(CreatePATColumns, &siparser->map));
+	mergeServices.loadComplete(this);
+	mergeServices.execute(NULL);
+	BOOST_FOREACH(const CommonObjects::DataSources::value_type & ds, CommonObjects::datasources) {
+		ds.second->commit();
+	}
+}
+