yolanda/trunk/gnutube_daemon.pl

#!/usr/bin/perl -w

use Proc::Daemon;
use DBI;
use Digest::SHA;
use File::Copy;

$database = 'gnutube';
$dbhost = 'localhost';
$dbuser = 'root';
$dbpass = '';
$gnutube_root = '/var/www/gnutube';

#TODO: deamonize by uncommenting this line
#Proc::Daemon::Init;

$LOG = "/var/www/gnutube/daemon.log";


sub appendlog
{
	if (open(FILE, ">>$LOG"))
	{
		print FILE scalar(localtime)." ".$$." ".join(" ",@_)."\n";
		close FILE;
	}
}

sub interrupt
{
	appendlog(@_);
	die;
}

$dbh = DBI->connect("DBI:mysql:$database:$dbhost", $dbuser, $dbpass) or interrupt "could not connect to db";

#video status:
# 0 - new entry - nothing done yet
# 1 - valid public video
# 2 - error: invalid audio and/or video stream
# 3 - error: file not found
# 4 - error: file is not a video
# 5 - error: video is a duplicate

while(1)
{
	#get fresh video id from db
	my $sth = $dbh->prepare(qq{select id, title, description, userid, timestamp from uploaded where status = 0 limit 1}) or interrupt $dbh->errstr;
	$sth->execute() or interrupt $dbh->errstr;
	my ($id, $title, $description, $userid, $timestamp) = $sth->fetchrow_array();
	$sth->finish() or interrupt $dbh->errstr;
	
	if($id)
	{
		$info = `ffplay -stats -an -vn -nodisp $gnutube_root/tmp/$id 2>&1`;
		
		if($info =~ /ignoring/)
		{
			appendlog $id, "invalid stream";
			
			#write status 2 to uploaded table
			$sth = $dbh->prepare(qq{update uploaded set status = ? where id = ?}) or interrupt $dbh->errstr;
			$sth->execute(2, $id) or interrupt $dbh->errstr;
			$sth->finish() or interrupt $dbh->errstr;
			unlink "$gnutube_root/tmp/$id";
		}
		elsif ($info =~ /I\/O error occured/)
		{
			appendlog $id, "file not found";
			
			#write status 3 to uploaded table
			$sth = $dbh->prepare(qq{update uploaded set status = ? where id = ?}) or interrupt $dbh->errstr;
			$sth->execute(3, $id) or interrupt $dbh->errstr;
			$sth->finish() or interrupt $dbh->errstr;
			unlink "$gnutube_root/tmp/$id";
		}
		elsif ($info =~ /Unknown format/ or $info =~ /could not find codec parameters/)
		{
			appendlog $id, "file is no video";
			
			#write status 4 to uploaded table
			$sth = $dbh->prepare(qq{update uploaded set status = ? where id = ?}) or interrupt $dbh->errstr;
			$sth->execute(4, $id) or interrupt $dbh->errstr;
			$sth->finish() or interrupt $dbh->errstr;
			unlink "$gnutube_root/tmp/$id";
		}
		else
		{
			$sha = new Digest::SHA(256);
			$sha->addfile("$gnutube_root/tmp/$id");
			$sha = $sha->hexdigest;
			
			#check if this hash is already in database
			my $sth = $dbh->prepare(qq{select id from videos where hash = ? limit 1}) or interrupt $dbh->errstr;
			$sth->execute($sha) or interrupt $dbh->errstr;
			my ($resultid) = $sth->fetchrow_array();
			$sth->finish() or interrupt $dbh->errstr;
			
			#if so, then video is a duplicate
			if($resultid)
			{
				appendlog "$id, video already uploaded: $resultid";
				
				#write status 5 to uploaded table
				$sth = $dbh->prepare(qq{update uploaded set status = ? where id = ?}) or interrupt $dbh->errstr;
				$sth->execute(5, $id) or interrupt $dbh->errstr;
				$sth->finish() or interrupt $dbh->errstr;
				unlink "$gnutube_root/tmp/$id";
			}
			else
			{
				($container, $duration) = $info =~ /Input \#0, (\w+),.+?\n.+?Duration: (\d{2}:\d{2}:\d{2}\.\d)/;
				
				#these two regexes have to be applied seperately because nobody knows which stream (audio or video) comes first
				($audio) = $info =~ /Audio: (\w+)/;
				($video, $width, $height, $fps) = $info =~ /Video: (\w+),.+?(\d+)x(\d+),.+?(\d+\.\d+) fps/;
				
				if(!$audio or !$video or !$duration)
				{
					appendlog $id, "a stream is missing or video is corrupt";
					
					#write status 2 to uploaded table
					$sth = $dbh->prepare(qq{update uploaded set status = ? where id = ?}) or interrupt $dbh->errstr;
					$sth->execute(2, $id) or interrupt $dbh->errstr;
					$sth->finish() or interrupt $dbh->errstr;
					unlink "$gnutube_root/tmp/$id";
				}
				else
				{
					#TODO: maybe delete entry from uploaded table after successful upload?
					$filesize = -s "$gnutube_root/tmp/$id";
					
					#convert hh:mm:ss.s duration to full seconds - thanks perl for making this so damn easy!
					#don't want to know how this would look in python or php... hell I don't even have to create extra variables!
					$duration =~ /^(\d{2}):(\d{2}):(\d{2})\.(\d)$/;
					$duration = int($1*3600 + $2*60 + $3 + $4/10 + .5);
					
					#create thumbnail
					$thumbnailsec = int($duration/3 + .5);
					system "ffmpeg -i $gnutube_root/tmp/$id -vcodec mjpeg -vframes 1 -an -f rawvideo -ss $thumbnailsec -s 320x240 $gnutube_root/video-stills/$id";
					
					#check if the upload already is in the right format
					if ($container eq 'ogg' and $video eq 'theora' and $audio eq 'vorbis')
					{
						appendlog $id, "file already is ogg-theora/vorbis";
						
						#write status 1 to uploaded table
						$sth = $dbh->prepare(qq{update uploaded set status = ? where id = ?}) or interrupt $dbh->errstr;
						$sth->execute(1, $id) or interrupt $dbh->errstr;
						$sth->finish() or interrupt $dbh->errstr;
						
						#add video to videos table
						$sth = $dbh->prepare(qq{insert into videos (id, title, description, userid, timestamp,
												hash, filesize, duration, width, height, fps)
												values (?, ?, ?, ?, ?, ?, ?, ?, ?,  ?,  ?)}) or interrupt $dbh->errstr;
						$sth->execute($id, $title, $description, $userid, $timestamp, $sha, $filesize, $duration, $width, $height, $fps) or interrupt $dbh->errstr;
						$sth->finish() or interrupt $dbh->errstr;
						
						#move video
						move "$gnutube_root/tmp/$id", "$gnutube_root/videos/$id";
					}
					else #encode video
					{
						#FIXME: remove endtime - this is for testing only
						system "ffmpeg2theora --optimize --videobitrate 1000 --audiobitrate 64 --sharpness 0 --endtime 10 --output $gnutube_root/videos/$id $gnutube_root/tmp/$id 2>&1";
						appendlog $id, $audio, $video, $width, $height, $fps, $duration, $sha;

						#write status 1 to uploaded table
						$sth = $dbh->prepare(qq{update uploaded set status = ? where id = ?}) or interrupt $dbh->errstr;
						$sth->execute(1, $id) or interrupt $dbh->errstr;
						$sth->finish() or interrupt $dbh->errstr;
						
						#add video to videos table
						$sth = $dbh->prepare(qq{insert into videos (id, title, description, userid, timestamp,
												hash, filesize, duration, width, height, fps)
												values (?, ?, ?, ?, ?, ?, ?, ?, ?,  ?,  ?)}) or interrupt $dbh->errstr;
						$sth->execute($id, $title, $description, $userid, $timestamp, $sha, $filesize, $duration, $width, $height, $fps) or interrupt $dbh->errstr;
						$sth->finish() or interrupt $dbh->errstr;
						
						#delete temp file
						unlink "$gnutube_root/tmp/$id";
					}
				}
			}
		}
	}
	else
	{
		sleep 10;
	}
}