#!/usr/bin/python3
import logging
from gi.repository import Gst

from lib.config import Config
from lib.tcpsingleconnection import TCPSingleConnection

class AVSource(TCPSingleConnection):
	def __init__(self, name, port, outputs=None, has_audio=True, has_video=True):
		self.log = logging.getLogger('AVSource['+name+']')
		super().__init__(port)

		if outputs is None:
			outputs = [name]

		assert has_audio or has_video

		self.name = name
		self.has_audio = has_audio
		self.has_video = has_video
		self.outputs = outputs

	def on_accepted(self, conn, addr):
		pipeline = """
			fdsrc fd={fd} !
			queue !
			matroskademux name=demux
		""".format(
			fd=conn.fileno()
		)

		if self.has_audio:
			pipeline += """
				demux. !
				{acaps} !
				queue !
				tee name=atee
			""".format(
				acaps=Config.get('mix', 'audiocaps')
			)

			for output in self.outputs:
				pipeline += """
					atee. ! queue ! interaudiosink channel=audio_{output}
				""".format(
					output=output
				)

		if self.has_video:
			pipeline += """
				demux. !
				{vcaps} !
				queue !
				tee name=vtee
			""".format(
				vcaps=Config.get('mix', 'videocaps')
			)

			for output in self.outputs:
				pipeline += """
					vtee. ! queue ! intervideosink channel=video_{output}
				""".format(
					output=output
				)

		self.log.debug('Launching Source-Pipeline:\n%s', pipeline)
		self.receiverPipeline = Gst.parse_launch(pipeline)

		self.log.debug('Binding End-of-Stream-Signal on Source-Pipeline')
		self.receiverPipeline.bus.add_signal_watch()
		self.receiverPipeline.bus.connect("message::eos", self.on_eos)
		self.receiverPipeline.bus.connect("message::error", self.on_error)

		self.all_video_caps = Gst.Caps.from_string('video/x-raw')
		self.video_caps = Gst.Caps.from_string(Config.get('mix', 'videocaps'))

		self.all_audio_caps = Gst.Caps.from_string('audio/x-raw')
		self.audio_caps = Gst.Caps.from_string(Config.get('mix', 'audiocaps'))

		demux = self.receiverPipeline.get_by_name('demux')
		demux.connect('pad-added', self.on_pad_added)

		self.receiverPipeline.set_state(Gst.State.PLAYING)

	def on_pad_added(self, demux, src_pad):
		caps = src_pad.query_caps(None)
		self.log.debug('demuxer added pad w/ caps: %s', caps.to_string())
		if caps.can_intersect(self.all_audio_caps):
			self.log.debug('new demuxer-pad is a audio-pad, testing against configured audio-caps')
			if not caps.can_intersect(self.audio_caps):
				self.log.warning('the incoming connection presented a video-stream that is not compatible to the configured caps')
				self.log.warning('   incoming caps:   %s', caps.to_string())
				self.log.warning('   configured caps: %s', self.audio_caps.to_string())


		elif caps.can_intersect(self.all_video_caps):
			self.log.debug('new demuxer-pad is a video-pad, testing against configured video-caps')
			if not caps.can_intersect(self.video_caps):
				self.log.warning('the incoming connection presented a video-stream that is not compatible to the configured caps')
				self.log.warning('   incoming caps:   %s', caps.to_string())
				self.log.warning('   configured caps: %s', self.video_caps.to_string())

	def on_eos(self, bus, message):
		self.log.debug('Received End-of-Stream-Signal on Source-Pipeline')
		if self.currentConnection is not None:
			self.disconnect()

	def on_error(self, bus, message):
		self.log.debug('Received Error-Signal on Source-Pipeline')
		(error, debug) = message.parse_error()
		self.log.debug('Error-Details: #%u: %s', error.code, debug)

		if self.currentConnection is not None:
			self.disconnect()

	def disconnect(self):
		self.receiverPipeline.set_state(Gst.State.NULL)
		self.receiverPipeline = None
		self.close_connection()