summaryrefslogtreecommitdiff
path: root/p2pvr/daemon/maintenance/services.cpp
blob: e5edbc539f068f91e8cb0e685d8069d48a789e31 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
#include <pch.hpp>
#include "../maintenance.h"
#include <siParsers/service.h>
#include <sqlMergeTask.h>
#include <Ice/Communicator.h>
#include <p2Helpers.h>
#include <containerIterator.h>
#include <temporaryIceAdapterObject.h>
#include <privateExecContext.h>

class SiServicesMerger : public SiServicesParser, public PrivateExecContext {
	public:
		SiServicesMerger(DatabaseClient * co) : commonObjects(co) { }

		bool HandleTable(DVBSI::TransportStreamPtr ts)
		{
			Logger()->messagebf(LOG_DEBUG, "Transport Stream Id: %d Original Network Id: %s", ts->TransportStreamId, ts->OriginalNetworkId);
			for (const auto & s : ts->Services) {
				Logger()->messagebf(LOG_DEBUG, "\tService Id: %d Name: %s Type: %d, Provider: %s, DefaultAuthority: %s, RunningStatus %d FreeCaMode %d",
						s->ServiceId, (s->Name ? *s->Name : "?"), (s->Type ? *s->Type : -1),
						(s->ProviderName ? *s->ProviderName : "?"), (s->DefaultAuthority ? *s->DefaultAuthority : "?"),
						s->RunningStatus, s->FreeCaMode);
			}

			DatabaseClient::TxHelper tx(commonObjects);
			SqlMergeTask mergeServices("postgres", "services");
			auto scs = CreateColumns<DVBSI::ServicePtr>(boost::bind(&DatabaseClient::SqlMergeColumnsInserter, &mergeServices, _1, _2));
			// Don't change the list of services available from the network
			mergeServices.doDelete = VariableType(false);
			mergeServices.doInsert = VariableType(false);
			mergeServices.sources.insert(new ContainerIterator<DVBSI::ServiceList>(&ts->Services, scs));
			mergeServices.loadComplete(commonObjects);
			mergeServices.execute(this);
			return false;
		}

	private:
		DatabaseClient * commonObjects;
};

void
Maintenance::UpdateServices(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 SiServicesMerger(this);
	TemporaryIceAdapterObject<P2PVR::RawDataClient> parser(ice.adapter, siparser);

	auto delivery = si->GetDeliveryForSi();
	if (!delivery) {
		throw std::runtime_error("no delivery methods");
	}

	Logger()->messagebf(LOG_DEBUG, "%s: Getting a tuner", __PRETTY_FUNCTION__);
	auto tuner = devs->GetTunerAny(delivery);
	Logger()->messagebf(LOG_DEBUG, "%s: Fetching service list", __PRETTY_FUNCTION__);
	tuner->SendServiceDescriptions(parser);
	Logger()->messagebf(LOG_INFO, "%s: Updated service list", __PRETTY_FUNCTION__);
	devs->ReleaseTuner(tuner);
}