<?

// Yuba batch helper
//                                      //
//////////////////////////////////////////

$prefs_file = "/Users/".get_current_user()."/Library/Preferences/yuba_prefs.php";
$p = unserialize(file_get_contents($prefs_file));

require("functions.php");
require("filetypes.php");

$mode = $argv[1];
$fid = $argv[2];
$pathname = $argv[3];
$bpath = $argv[4];
$mytime = $argv[5];

$tmpdir = "/tmp/yuba/".$mytime;
if (!is_dir($tmpdir)) { mkdir($tmpdir,0777,true); }

$file = pathinfo($pathname, PATHINFO_BASENAME);
$ext = pathinfo($pathname, PATHINFO_EXTENSION);

$shellpath = escapeshellarg($pathname);

$dbp = new PDO("sqlite:".$bpath."/pool.sqlite3");
$dbp->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);
$dbp->setAttribute(PDO::ATTR_DEFAULT_FETCH_MODE, PDO::FETCH_ASSOC);

echo "\nBatch (".$mode."): ".basename($pathname);

switch ($mode) {
	
	// Icons //////////////////////////////////////////

	case "icons":
	
	if ($p['icon_tool'] == 0) {
		$tool = "qltool";
		} elseif ($p['icon_tool'] == 1) {
		$tool = "qlmanage";
		}
	
	if (in_array($ext, $p['i_skip'])) {
	// file extension is in the skip list
		echo " ->i_skip";
		break;
		}
	
	$tfile = $tmpdir."/".substr($fid,0,2)."/".$file.".png";
	if (!is_dir(dirname($tfile))) { @mkdir(dirname($tfile)); }
	// workaround for qlmanage naming convention

	$rowid = @$dbp->query("SELECT rowid FROM icons WHERE fid='".$fid."'")->fetch()['rowid'];
	// check for existing icon generation attempt

	if (!$rowid) {
	// no prior attempt
	
		$stmt = $dbp->prepare("INSERT INTO icons VALUES (:fid, :hash, :created, :relative_path, :tool)");
		echo " ->generate";
		
		} elseif ($rowid && $p['icons'] == 2) {
		// prior attempt but rebuild mode
		
		$stmt = $dbp->prepare("UPDATE icons SET fid = :fid, hash = :hash, created = :created, relative_path = :relative_path, tool = :tool WHERE rowid = ".$rowid);
		echo " ->rebuild";
		
		} else {
		// prior attempt
		
		echo " ->skip";
		break;
		
		}

	$stmt->BindValue(":fid",$fid);
	
	if ($tool == "qltool") {
			
		$cmd = $bin_qltool." di ".$shellpath." ".$p['icon_size']." ".$p['icon_size']." | base64 --decode > ".escapeshellarg($tfile);
			
		} elseif ($tool == "qlmanage") {
		
		// > code for custom qlgenerator bindings goes here <
		
		//$cmd = "qlmanage -ti -s ".$p['icon_size']." -o ".dirname($tfile)." ".$shellpath;
		$cmd = "qlmanage -ti -f ".floor($p['icon_size']/128)." -o ".dirname($tfile)." ".$shellpath;	
			
		}

	shell_exec($cmd);
	
	if (@filesize($tfile)) {
			
		$hash = md5_file($tfile);
		$stmt->BindValue(":hash",$hash);
			
		if ($row = @$dbp->query("SELECT * FROM icons WHERE hash='".$hash."'")->fetchAll()[0]) {
		// check for another icon with the same hash
				
			$stmt->BindValue(":created",$row['created']);
			$stmt->BindValue(":relative_path",$row['relative_path']);
			$stmt->BindValue(":tool",$row['tool']);
			
			echo " ->recycle";
			
			} else {
				
			$dfile = $bpath."/icons/".substr($hash, 0, 2)."/".$hash.".png";
			if (!is_dir(dirname($dfile))) { @mkdir(dirname($dfile)); }
			
			echo " ->use";
			
			if ($tool == "qltool") {
			// qltool makes 2x icons for some reason
				
				//shell_exec("sips -z ".$p['icon_size']." ".$p['icon_size']." ".$tfile." --out ".$dfile);
				shell_exec($bin_convert." ".escapeshellarg($tfile)." -scale 50% -strip -define png:compression-level=9 ".$dfile);
				
				} else {
				
				//rename($tfile,$dfile);
				shell_exec($bin_convert." ".escapeshellarg($tfile)." -strip -define png:compression-level=9 ".$dfile);
					
				}
				
			$stmt->BindValue(":created",time());
			$stmt->BindValue(":relative_path",substr($dfile,strlen($bpath)));
			$stmt->BindValue(":tool",$tool);
				
			}
				
		} else {
		
		echo " ->discard";
		
		}
			
	$stmt->execute();
	
	echo " ->db";
	
	break;
	
	case "thumbs":
	
	// Thumbs //////////////////////////////////////////
	
	// "bad" filesizes
	$discard = array(	3953,
						4977,
						5019,
						6059,
						6616,
						17393	);
	
	$tfile = $tmpdir."/".substr($fid,0,2)."/".$fid.".jpg";
	$tpfile = $tmpdir."/".substr($fid,0,2)."/".$file.".png"; // workaround for qlmanage naming convention
	if (!is_dir(dirname($tfile))) { @mkdir(dirname($tfile)); }
	
	if (in_array($ext, $p['t_skip'])) {
	// file extension is in the skip list
		echo " ->t_skip";
		break;
		}
		
	if ($p['thumbs'] == 1 && file_exists($tfile)) {
	// if this is not a rebuild, first check for an existing thumb file (faster)
		echo " ->skip_fast";
		break;
		}
		
	$rowid = @$dbp->query("SELECT rowid FROM thumbs WHERE fid='".$fid."'")->fetch()['rowid'];
	// check for existing thumb generation attempt
	
	if (!$rowid) {
	// no prior attempt
	
		$stmt = $dbp->prepare("INSERT INTO thumbs VALUES (:fid, :created, :relative_path, :width, :height, :tool)");
		echo " ->generate";
		
		} elseif ($rowid && $p['thumbs'] == 2) {
		// prior attempt but rebuild mode
		
		$stmt = $dbp->prepare("UPDATE thumbs SET fid = :fid, created = :created, relative_path = :relative_path, width = :width, height = :height, tool = :tool WHERE rowid = ".$rowid);
		echo " ->rebuild";
		
		} else {
		// prior attempt
		
		echo " ->skip";
		break;
		
		}
	
	$cmd['sips'] = "sips -s format jpeg -s formatOptions 80 --resampleHeightWidthMax ".$p['thumb_size']." ".$shellpath." --out ".$tfile; // add ?? "-d profile --deleteColorManagementProperties"
	$cmd['vips'] = $bin_vips." ".$shellpath." -o ".$tfile."[Q=90,optimize_coding] --size=".$p['thumb_size'];
	
	$cmd['sox'] = $bin_sox." ".$shellpath." -n trim 0 $(".$bin_exiftool." -s -s -s -duration# ".$shellpath." | awk '{print $1/10}') spectrogram -o - | ".$bin_convert." - -crop 800x515+58+30 -scale 515x515! +dither -colors 16 ".$tfile;
	$cmd['ffmpeg'] = $bin_ffmpeg." -ss $(( $(".$bin_mediainfo." --Inform='Video;%Duration%' ".$shellpath." | cut -d'.' -f1) / 10000 )) -i ".$shellpath." -vframes 1 -filter:v scale='400:-1' -q:v 3 ".$tfile;
	
	$cmd['ql-thumbnail'] = $bin_qlthumb." ".$shellpath." ".$tfile." public.jpeg ".$p['thumb_size']." ".$p['thumb_size']." .8";
	$cmd['qlmanage'] = "qlmanage -t -s ".$p['thumb_size']." -o ".dirname($tpfile)." ".$shellpath."; sips -s format jpeg -s formatOptions 80 ".escapeshellarg($tpfile)." --out ".$tfile;
	
	if (in_array($ext, $p['t_files']['sips'])) {
		$external_tool = "sips";
		} elseif (in_array($ext, $p['t_files']['ffmpeg'])) {
		$external_tool = "ffmpeg";
		} elseif (in_array($ext, $p['t_files']['sox'])) {
		$external_tool = "sox";
		} else {
		$external_tool = null;
		}
	switch ($p['thumb_priority']) {
		// external tool priority
		case 0: $priority = array($external_tool,"ql-thumbnail","qlmanage"); break;
		// ql-thumbnail priority
		case 1: $priority = array("ql-thumbnail","qlmanage",$external_tool); break;
		// qlmanage priority
		case 2: $priority = array("qlmanage","ql-thumbnail",$external_tool); break;
		}
	
	$stmt->BindValue(":fid",$fid);
	$stmt->BindValue(":created",time());
	
	$dfile = $bpath."/thumbs/".substr($fid, 0, 2)."/".$fid.".jpg";;
	if (!is_dir(dirname($dfile))) { @mkdir(dirname($dfile)); }
	
	foreach ($priority as $tool) {

		if (empty($cmd[$tool])) { continue; }
		
		shell_exec($cmd[$tool]." 2>&1");
		echo " ->".$tool;
				
		$checksize = @filesize($tfile);
		if ($checksize && !in_array($checksize,$discard)) {
			echo " ->use";
			$stmt->BindValue(":tool",$tool);
			$stmt->BindValue(":relative_path",substr($dfile, strlen($bpath)));
			list($width, $height) = getimagesize($tfile);
			$stmt->BindValue(":width",$width);
			$stmt->BindValue(":height",$height);
			rename($tfile,$dfile);
			break;
			} else {
			echo " ->discard";
			}

		}

	$stmt->execute();
	
	echo " ->db";
	
	break;
	
	}

echo "\n";

?>