Tvheadend

#!/usr/bin/php-cgi -q

<?php

/**

* This fix was written for HTS Tvheadend 4.1-2415~ge5f5a42-dirty

* I'm using it with incron (https://linux.die.net/man/5/incrontab) this way:

* When new service appears, MUX file is changed. This event is the trigger to run this script and it's triggered by incron.

* Example of incrontab configuration

* /home/ivan/.hts/tvheadend/input/iptv/networks/36a38dsw1eed25461d765ave4c60d9c9/muxes IN_ALL_EVENTS /home/ivan/.hts/tvheadend/scripts/service_channel_fix.php $$ $@ $# $% $&

* When some file in directory /home/ivan/.hts/tvheadend/input/iptv/networks/36a38dsw1eed25461d765ave4c60d9c9/muxes changes (IN_ALL_EVENTS)

* incron will trigger this script with 5 parameters.

* Read incron documentation for further information about all those parameters.

* Third arugment is the file name that changed and it is MUX ID of service that was created and it is a key to have this working.

*/

ob_start();

register_shutdown_function('writeOutputToLog', LOCK_FILE);

echo "(" . __FILE__ . ":" . __LINE__ . ") " . date("Y-m-d H:i:s") . ": Process started with PID: `" . getmypid() . "`\n";

define('LOG_FILE', '/tmp/service_channel_fix.log');

define('USER', ''); //use your admin username

define('PASS', ''); //use your admin password

define('AUTH_STRING', (strlen(USER) && strlen(PASS)) ? USER . ':' . PASS . '@' : '');

define('SERVER_ADDR', '127.0.0.1');

define('SERVER_PORT', '9981');

define('SERVICE_GRIG_LIMIT', 999999999);

define('SERVICES_URL', 'http://' . AUTH_STRING . SERVER_ADDR . ':' . SERVER_PORT . '/api/mpegts/service/grid');

define('SAVE_ID_NODE_URL',  'http://' . AUTH_STRING . SERVER_ADDR . ':' . SERVER_PORT . '/api/idnode/save');

define('DELETE_ID_NODE_URL',  'http://' . AUTH_STRING . SERVER_ADDR . ':' . SERVER_PORT . '/api/idnode/delete');

$curlDefaultOpt = array(

	CURLOPT_RETURNTRANSFER => true

);

echo "(" . __FILE__ . ":" . __LINE__ . ") " . date("Y-m-d H:i:s") . ": Passed arguments: " . print_r($argv, true) . "\n";

$muxId = $argv[3]; //third arugment passed from incron regarding name of file that changed. It is also a MUX ID

if (!strlen($muxId)) die("(" . __FILE__ . ":" . __LINE__ . ") " . date("Y-m-d H:i:s") . ": MUX ID is empty\n");

$services = getServices($muxId); //getting all services related to that mux ID

if (!count($services)) die("(" . __FILE__ . ":" . __LINE__ . ") " . date("Y-m-d H:i:s") . ": No services found for MUX ID: `" . $muxId . "`\n");

$channelIds = getChannelIds($services); //getting all channels stored for services. This should return array made mainly from only one channel ID

if (!count($channelIds)) die("(" . __FILE__ . ":" . __LINE__ . ") " . date("Y-m-d H:i:s") . ": No channel IDs found for services of MUX ID: `" . $muxId . "`\n");

echo "(" . __FILE__ . ":" . __LINE__ . ") " . date("Y-m-d H:i:s") . ": Channel IDs: " . print_r($channelIds, true) . "\n";

$newServiceIds = getNewServiceIds($services); //getting all service IDs that are new. These IDs we need to pair with our channel(s)

if (!count($newServiceIds)) die("(" . __FILE__ . ":" . __LINE__ . ") " . date("Y-m-d H:i:s") . ": No new channel IDs found for services of MUX ID: `" . $muxId . "`\n");

$oldServiceIds = getOldServiceIds($services); //getting all service IDs that are old, not working. These IDs we need to delete because they are not necessary anymore

echo "(" . __FILE__ . ":" . __LINE__ . ") " . date("Y-m-d H:i:s") . ": New service IDs: " . print_r($newServiceIds, true) . "\n";

echo "(" . __FILE__ . ":" . __LINE__ . ") " . date("Y-m-d H:i:s") . ": Saving channels to services\n";

saveServiceForChannel($newServiceIds, $channelIds);

echo "(" . __FILE__ . ":" . __LINE__ . ") " . date("Y-m-d H:i:s") . ": Old service IDs to delete: " . print_r($oldServiceIds, true) . "\n";

echo "(" . __FILE__ . ":" . __LINE__ . ") " . date("Y-m-d H:i:s") . ": Deleting services\n";

if (count($oldServiceIds))

	deleteServices($oldServiceIds);

echo "(" . __FILE__ . ":" . __LINE__ . ") " . date("Y-m-d H:i:s") . ": Process finished\n";

exit;

/**

* Parses all services and looks only for services that are realted to mux ID

* @param string muxId ID of your MUX

* @return array Array of service objects

*/

function getServices($muxId) {

	$services		= array();

	$allServices	= getAllServices();

	foreach($allServices->{'entries'} as $serviceKey => $service) {

		if ($service->{'multiplex_uuid'} == $muxId) {

			$services[] = $service;

		}

	}

	return $services;

}

/**

* Requests all services from TVheadend through JSON API

* @return array Array of all service objects

*/

function getAllServices() {

	global $curlDefaultOpt;

	$curlResource					= curl_init(SERVICES_URL);

	$curlOpts						= $curlDefaultOpt;

	$curlOpts[CURLOPT_POST]			= true;

	$curlOpts[CURLOPT_POSTFIELDS]	= 'start=0&limit=' . SERVICE_GRIG_LIMIT;

	curl_setopt_array($curlResource, $curlOpts);

	$answer = curl_exec($curlResource);

	curl_close($curlResource);

	return json_decode($answer);

}

/**

* Parses services and gets distinct values of all channels from them

* @param array services Array of service objects

* @return array Array of distinct channel values

*/

function getChannelIds($services) {

	$channels = array();

	foreach($services as $serviceKey => $service) {

		foreach($service->{'channel'} as $channelId) {

			if (!in_array($channelId, $channels))

				$channels[] = $channelId; //this is already an array

		}

	}

	return $channels;

}

/**

* Parses service object and looks for services that have empty channel array.

* These are new services that need set the channel(s)

* @param array services Array of service objects

* @return array Array of service IDs

*/

function getNewServiceIds($services) {

	$newServices = array();

	foreach($services as $serviceKey => $service) {

		if (!count($service->{'channel'})) {

			$newServices[] = $service->{'uuid'};

		}

	}

	return $newServices;

}

/**

* Parses service object and looks for services that don't have empty channel array.

* These are old services that don't work and we want to delete them later

* @param array services Array of service objects

* @return array Array of service IDs

*/

function  getOldServiceIds($services) {

	$oldServices = array();

	foreach($services as $serviceKey => $service) {

		if (count($service->{'channel'})) {

			$oldServices[] = $service->{'uuid'};

		}

	}

	return $oldServices;

}

/**

* Saves channnel(s) for service(s)

* @param array newServiceIds Array of service objects

* @param array channelIds Array of distinct channel values

* @return void

*/

function saveServiceForChannel($newServiceIds, $channelIds) {

	foreach($newServiceIds as $newServiceId) {

		echo "(" . __FILE__ . ":" . __LINE__ . ") " . date("Y-m-d H:i:s") . ": Saving service ID `" . print_r($newServiceId, true) . "` for channels " . print_r($channelIds, true) . "\n";

		saveIdNodes($newServiceId, $channelIds);

	}

}

/**

* Requests TVheadend to store channel ID(s) to service

* @param string newServiceId ID of service

* @param array channelIds Array of distinct channel values

* @return array Array returned from TVheadend as a response

*/

function saveIdNodes($newServiceId, $channelIds) {

	global $curlDefaultOpt;

	$curlResource					= curl_init(SAVE_ID_NODE_URL);

	$curlOpts						= $curlDefaultOpt;

	$curlOpts[CURLOPT_POST]			= true;

	//only this way the server accepted the parameters. Don't use postfields as php array. Use this string format to get this work

	$curlOpts[CURLOPT_POSTFIELDS]	= 'node=' . json_encode(array(array('channel' => $channelIds, 'uuid' => $newServiceId)));

	curl_setopt_array($curlResource, $curlOpts);

	curl_setopt($curlResource, CURLINFO_HEADER_OUT, true);

	$answer = curl_exec($curlResource);

	curl_close($curlResource);

	return json_decode($answer);

}

/**

* Requests TVheadend to delete service ID(s)

* @param array oldServiceIds ID of old service(s) to delete

* @return array Array returned from TVheadend as a response

*/

function deleteServices($oldServiceIds) {

	global $curlDefaultOpt;

	$curlResource					= curl_init(DELETE_ID_NODE_URL);

	$curlOpts						= $curlDefaultOpt;

	$curlOpts[CURLOPT_POST]			= true;

	//only this way the server accepted the parameters. Don't use postfields as php array. Use this string format to get this work

	$curlOpts[CURLOPT_POSTFIELDS]	= 'uuid=' . json_encode($oldServiceIds);

	curl_setopt_array($curlResource, $curlOpts);

	curl_setopt($curlResource, CURLINFO_HEADER_OUT, true);

	$answer = curl_exec($curlResource);

	curl_close($curlResource);

	return json_decode($answer);

}

/**

* All outputs of this file will be written to log file defined

* @return void

*/

function writeOutputToLog() {

	$output = ob_get_clean();

	file_put_contents(LOG_FILE, $output, FILE_APPEND);

}