Tvheadend

#!/bin/bash

# matchIcons.sh, [email protected], 20140116

#

# Takes a Tvheadend configuration and a directory containing channel icons (png), scans

# the channels in the config and tries to match them to the icon file names, creates

# symlinks in the icon directory for non-exact matches, writes file:// URLs to the

# configuration. 

# The orignal channel configuration is NOT touched by default. Instead we work on 

# a copy that needs to be moved into place manually before restarting Tvheadend.

#

# Tested only on recent Tvheadend 3.9.x versions (dvb rewrite)!

# Feedback welcome!

#

# config section

htsconf="/home/hts/.hts/tvheadend"

newchannel="$htsconf/channel.new$$"

icondir="/home/hts/channelicons"

# function secion

function simplify {

	echo "$1" | sed -e 's/[^a-zA-Z0-9]//g' -e 's/\(.*\)/\L\1/' -e 's/.*\///'

}

function exist {

	[ -f "$1" -o -h "$1" ] && [ -s "$1" ]

}

iconlist=$(

	ls -1 "$icondir" | 

	while read line; do

		if [[ "$icondir/$line" == *.png ]] && exist "$icondir/$line"; then

			printf "%s\t%s\n" "$line" $(simplify "${line%.png}") 

		fi

	done

)

function name2icon {

        simplename=$(simplify "$1")

	echo "$iconlist" | awk -F'\t' -vname="$simplename" '{if ($2==name) print $1;}' | head -1

}

function service2name {

	grep '"svcname"' "$htsconf/input/linuxdvb/networks/"*"/muxes/"*"/services/$1" |

	awk -F\" '{print $4}' |

	sed -e 's/\//-/g'

}

function patchicon {

	channelfile="$1"

	iconfile="$2"

	exist "$channelfile" || return

	exist "$iconfile" || return

	url=$(echo "file://$iconfile" | sed -e 's/"/\\"/g')

	savetxt=$(grep -v -Ee '{|}|"icon"' "$channelfile")

	echo '{' >"$channelfile"

	echo "        \"icon\": \"$url\"," >>"$channelfile"

	echo "$savetxt" >>"$channelfile"

	echo '}' >>"$channelfile"

}

############## MAIN

[ -d "$newchannel" ] && echo "$newcchannel exists!" && exit 1

echo "Copying current channel dir to $newchannel..."

cp -rp "$htsconf/channel" "$newchannel" || exit 1

ls -1 "$newchannel/"* |

while read channelfile; do

	exist "$channelfile" || continue

	name=$(service2name $(grep -A 1 '"services"' "$channelfile" | tail -1 | awk -F\" '{print $2}'))

	icon=$(name2icon "$name")

	! exist "$icondir/$icon" && icon=$(name2icon "${name% [hH][dD]}")

	echo "Processing '$name'..."

	if exist "$icondir/$icon"; then

		if ! exist "$icondir/$name.png"; then

			echo "  Creating symlink $icondir/$name.png -> $icondir/$icon"

			ln -sf "$icondir/$icon" "$icondir/$name.png"

		fi

		echo "  Setting URL 'file://$icondir/$name.png'..."

		patchicon "$channelfile" "$icondir/$name.png"

	else

		echo "  No icon found for '$name'."

	fi

done

echo "

Finished creating $newchannel.

Next steps: move in place and restart tvheadend.

Enjoy!

"