handbrake-cluster-webui/lib/RippingCluster/Source/Plugin/HandBrake.class.php

<?php

class RippingCluster_Source_Plugin_HandBrake implements RippingCluster_Source_IPlugin {
    
    const PLUGIN_NAME = "HandBrake";
    
    const PM_TITLE    = 0;
    const PM_CHAPTER  = 1;
    const PM_AUDIO    = 2;
    const PM_SUBTITLE = 3;
    
    public static function init() {
        // Nothing to do
    }
    
    public static function name() {
        return self::PLUGIN_NAME;
    }
    
    public static function enumerate() {
        $config = RippingCluster_Main::instance()->config();
        $directory = $config->get('source.handbrake.dir');
        
        if (!is_dir($directory)) {
            throw new RippingCluster_Exception_InvalidSourceDirectory($directory);
        }
        
        $sources = array();
        
        $iterator = new RippingCluster_Utility_DvdDirectoryIterator(new RippingCluster_Utility_VisibleFilesIterator(new DirectoryIterator($directory)));
        foreach ($iterator as /** @var SplFileInfo */ $source_vts) {
            $sources[] = self::load($source_vts->getPathname(), false);
        }
        
        return $sources;
    }
    
    /**
     * 
     * 
     * @param string $source
     * @param bool $scan
     * @param bool $use_cache 
     * @return RippingCluster_Source
     */
    public static function load($source_filename, $scan = true, $use_cache = true) {
        $cache = RippingCluster_Main::instance()->cache();

        // Ensure the source is a valid directory, and lies below the configured source_dir
        if ( ! self::isValidSource($source_filename)) {
            return new RippingCluster_Source($source_filename, self::name(), false);
        }
            
        $source = null;
        if ($use_cache && $cache->exists($source_filename)) {
            $source = unserialize($cache->fetch($source_filename));
        } else {
            $source = new RippingCluster_Source($source_filename, self::name(), true);
            
            if ($scan) {
                $source_shell = escapeshellarg($source_filename);
                $handbrake_cmd = "HandBrakeCLI -i {$source_shell} -t 0";
                list($retval, $handbrake_output, $handbrake_error) = RippingCluster_ForegroundTask::execute($handbrake_cmd);
                
                // Process the output
                $lines = explode("\n", $handbrake_error);
                $title = null;
                $mode = self::PM_TITLE;
        
                foreach ($lines as $line) {
                    // Skip any line that doesn't begin with a + (with optional leading whitespace)
                    if ( ! preg_match('/\s*\+/', $line)) {
                        continue;
                    }
        
                    $matches = array();
                    switch (true) {
                       case preg_match('/^\+ title (?P<id>\d+):$/', $line, $matches): {
                            if ($title) {
                                $source->addTitle($title);
                            }
                            
                            $mode = self::PM_TITLE;
                            $title = new RippingCluster_Rips_SourceTitle($matches['id']);
                        } break;
                        
                        case $title && preg_match('/^  \+ chapters:$/', $line): {
                            $mode = self::PM_CHAPTER;    
                        } break;
                        
                        case $title && preg_match('/^  \+ audio tracks:$/', $line): {
                            $mode = self::PM_AUDIO;
                            
                        } break;
                        
                        case $title && preg_match('/^  \+ subtitle tracks:$/', $line): {
                            $mode = self::PM_SUBTITLE;
                        } break;
                        
                        case $title && $mode == self::PM_TITLE && preg_match('/^  \+ duration: (?P<duration>\d+:\d+:\d+)$/', $line, $matches): {
                            $title->setDuration($matches['duration']);
                        } break;
                        
                        case $title && $mode == self::PM_TITLE && preg_match('/^  \+ angle\(s\) (?P<angles>\d+)$/', $line, $matches): {
                            $title->setAngles($matches['angles']);
                        } break;
                        
                        //"  + size: 720x576, pixel aspect: 64/45, display aspect: 1.78, 25.000 fps"
                        case $title && $mode == self::PM_TITLE && preg_match('/^  \+ size: (?P<width>\d+)x(?P<height>\d+), pixel aspect: (?P<pixel_aspect>\d+\/\d+), display aspect: (?P<display_aspect>[\d\.]+), (?<framerate>[\d\.]+) fps$/', $line, $matches): {
                            $title->setDisplayInfo(
                                $matches['width'], $matches['height'], $matches['pixel_aspect'],
                                $matches['display_aspect'], $matches['framerate']
                            );
                        } break;
                        
                        case $title && $mode == self::PM_TITLE && preg_match('/^  \+ autocrop: (?P<autocrop>(?:\d+\/?){4})$/', $line, $matches): {
                            $title->setAutocrop($matches['autocrop']);
                        } break;
                        
                        case $title && $mode == self::PM_CHAPTER && preg_match('/^    \+ (?P<id>\d+): cells \d+->\d+, \d+ blocks, duration (?P<duration>\d+:\d+:\d+)$/', $line, $matches): {
                            $title->addChapter($matches['id'], $matches['duration']);
                        } break;
        
                        case $title && $mode == self::PM_AUDIO && preg_match('/^    \+ (?P<id>\d+), (?P<name>.+) \((?P<format>.+)\) \((?P<channels>(.+ ch|Dolby Surround))\) \((?P<language>.+)\), (?P<samplerate>\d+)Hz, (?P<bitrate>\d+)bps$/', $line, $matches): {
                            $title->addAudioTrack(
                                new RippingCluster_Rips_SourceAudioTrack(
                                    $matches['id'], $matches['name'], $matches['format'], $matches['channels'],
                                    $matches['language'], $matches['samplerate'], $matches['bitrate']
                                )
                            );
                        } break;
                        
                        case $title && $mode == self::PM_SUBTITLE && preg_match('/^    \+ (?P<id>\d+), (?P<name>.+) \((?P<language>.+)\) \((?P<format>.+)\)$/', $line, $matches): {
                            $title->addSubtitleTrack(
                                new RippingCluster_Rips_SourceSubtitleTrack(
                                    $matches['id'], $matches['name'], $matches['language'], $matches['format']
                                )
                            );
                        } break;
                        
                        default: {
                            // Ignore this unmatched line
                        } break;
        
                    }
                }
                
                // Handle the last title found as a special case
                if ($title) {
                    $source->addTitle($title);
                }
    
                // If requested, store the new source object in the cache
                if ($use_cache) {
                    $source->cache();
                }
            }
        }

        return $source;
    }
    
    public static function loadEncoded($encoded_filename, $scan = true, $use_cache = true) {
        // Decode the filename
        $source_filename = base64_decode(str_replace('-', '/', $encoded_filename));

        return self::load($source_filename, $scan, $use_cache);
    }
    
    public static function isValidSource($source_filename) {
        $config = RippingCluster_Main::instance()->config();
        
        // Ensure the source is a valid directory, and lies below the configured source_dir
        if ( ! is_dir($source_filename)) {
            return false;
        }
        $real_source_filename = realpath($source_filename);
        
        $source_basedir = $config->get('rips.source_dir');
        $real_source_basedir = realpath($source_basedir);
        
        if (substr($real_source_filename, 0, strlen($real_source_basedir)) != $real_source_basedir) {
            return false;
        }
        
        return true;
    }
    
}

?>