tivohmo 0.1.0

data/lib/tivohmo/adapters/filesystem/application.rb

@@ -0,0 +1,23 @@
+require 'active_support/core_ext/string/inflections'
+require 'tivohmo/adapters/streamio'
+
+module TivoHMO
+  module Adapters
+    module Filesystem
+
+      # An Application based on a filesystem
+      class Application < FolderContainer
+        include TivoHMO::API::Application
+        include GemLogger::LoggerSupport
+        include MonitorMixin
+
+        def initialize(identifier)
+          super(identifier)
+          self.metadata_class = TivoHMO::Adapters::StreamIO::Metadata
+          self.transcoder_class = TivoHMO::Adapters::StreamIO::Transcoder
+        end
+      end
+
+    end
+  end
+end

data/lib/tivohmo/adapters/filesystem/file_item.rb

@@ -0,0 +1,29 @@
+#require 'streamio-ffmpeg'
+
+module TivoHMO
+  module Adapters
+    module Filesystem
+
+      # An Item based on a filesystem file
+      class FileItem
+        include TivoHMO::API::Item
+        include GemLogger::LoggerSupport
+
+        attr_reader :full_path
+
+        def initialize(identifier)
+          @full_path = File.expand_path(identifier)
+          raise ArgumentError, "Must provide an existing file" unless File.file?(full_path)
+
+          super(full_path)
+
+          self.title = File.basename(self.identifier)
+          self.modified_at = File.mtime(self.identifier)
+          self.created_at = File.ctime(self.identifier)
+        end
+
+      end
+
+    end
+  end
+end

data/lib/tivohmo/adapters/filesystem/folder_container.rb

@@ -0,0 +1,105 @@
+require 'active_support/core_ext/string/inflections'
+require 'listen'
+
+module TivoHMO
+  module Adapters
+    module Filesystem
+
+      # A Container based on a filesystem folder
+      class FolderContainer
+        include TivoHMO::API::Container
+        include GemLogger::LoggerSupport
+        include MonitorMixin
+
+        VIDEO_EXTENSIONS = %w[
+          tivo mpg avi wmv mov flv f4v vob mp4 m4v mkv
+          ts tp trp 3g2 3gp 3gp2 3gpp amv asf avs bik bix box bsf
+          dat dif divx dmb dpg dv dvr-ms evo eye flc fli flx gvi ivf
+          m1v m21 m2t m2ts m2v m2p m4e mjp mjpeg mod moov movie mp21
+          mpe mpeg mpv mpv2 mqv mts mvb nsv nuv nut ogm qt rm rmvb
+          rts scm smv ssm svi vdo vfw vid viv vivo vp6 vp7 vro webm
+          wm wmd wtv yuv
+        ]
+
+        attr_reader :full_path
+
+        attr_accessor :allowed_item_types,
+                      :allowed_item_extensions
+
+        def initialize(identifier)
+          @full_path = File.expand_path(identifier)
+          raise ArgumentError,
+                "Must provide an existing directory: #{full_path}" unless File.directory?(full_path)
+
+          super(full_path)
+
+          self.allowed_item_types = %i[file dir]
+          self.allowed_item_extensions = VIDEO_EXTENSIONS
+
+          self.title = File.basename(self.identifier).titleize
+          self.modified_at = File.mtime(self.identifier)
+          self.created_at = File.ctime(self.identifier)
+
+          setup_change_listener
+        end
+
+        def children
+          synchronize do
+            if super.blank?
+              folders = []
+              files = []
+
+              Dir["#{self.full_path}/*"].each do |path|
+                if allowed_container?(path)
+                  folders << FolderContainer.new(path)
+                elsif allowed_item?(path)
+                  files << FileItem.new(path)
+                else
+                  logger.debug "Ignoring: #{path}"
+                end
+              end
+
+              (folders + files).each {|c| add_child(c) }
+            end
+          end
+
+          super
+        end
+
+        protected
+
+        def setup_change_listener
+          logger.debug "Setting up change listener on #{identifier}"
+          @listener = Listen.to(identifier, ignore: /\//) do |modified, added, removed|
+            logger.debug "Detected filesystem change on #{identifier}"
+            logger.debug "modified: #{modified}"
+            logger.debug "added: #{added}"
+            logger.debug "removed: #{removed}"
+
+            # TODO: be more intelligent instead of just wiping children to cause the refresh
+            self.refresh
+
+            # cleanup - not strictly correct as this listener won't necessarily get triggered
+            # if self is removed from the parent
+            @listener.stop unless root.find(title_path)
+            logger.debug "Completed filesystem refresh on #{identifier}"
+          end
+          @listener.start
+        end
+
+        def allowed_container?(path)
+          File.directory?(path) && allowed_item_types.include?(:dir)
+        end
+
+        def allowed_item?(path)
+          ext = File.extname(path).gsub(/^./, '')
+          File.file?(path) &&
+              allowed_item_types.include?(:file) &&
+              allowed_item_extensions.include?(ext)
+        end
+
+      end
+
+    end
+  end
+end

data/lib/tivohmo/adapters/filesystem.rb

@@ -0,0 +1,3 @@
+require_relative 'filesystem/file_item'
+require_relative 'filesystem/folder_container'
+require_relative 'filesystem/application'

data/lib/tivohmo/adapters/plex/application.rb

@@ -0,0 +1,41 @@
+module TivoHMO
+  module Adapters
+    module Plex
+
+      class Application
+        include TivoHMO::API::Application
+        include GemLogger::LoggerSupport
+        include MonitorMixin
+
+        attr_reader :server
+
+        def initialize(identifier)
+          host, port = identifier.to_s.split(':')
+          host ||= 'localhost'
+          port ||= 32400
+          super("Plex[#{host}:#{port}]")
+
+          self.metadata_class = TivoHMO::Adapters::Plex::Metadata
+          self.transcoder_class = TivoHMO::Adapters::Plex::Transcoder
+          self.title = self.identifier
+
+          @server = ::Plex::Server.new(host, port)
+        end
+
+        def children
+          synchronize do
+            if super.blank?
+              Array(server.library.sections).each do |section|
+                add_child(Section.new(section))
+              end
+            end
+          end
+
+          super
+        end
+
+      end
+
+    end
+  end
+end

data/lib/tivohmo/adapters/plex/category.rb

@@ -0,0 +1,72 @@
+module TivoHMO
+  module Adapters
+    module Plex
+
+      class Category
+        include TivoHMO::API::Container
+        include GemLogger::LoggerSupport
+        include MonitorMixin
+
+        attr_reader :delegate
+        attr_accessor :category_type, :category_value
+
+        def initialize(delegate, category_type, category_value=nil)
+          # delegate is a Plex::Section
+          @delegate = delegate
+
+          super(delegate.key)
+
+          self.category_type = category_type
+          self.category_value = category_value
+
+          if category_value
+            self.title = category_value[:title]
+          else
+            self.title = category_type.to_s.titleize
+          end
+
+          self.modified_at = Time.at(delegate.updated_at.to_i)
+          self.created_at = Time.at(delegate.updated_at.to_i)
+        end
+
+        def children
+          synchronize do
+
+            delegate.refresh
+            new_modified_at = delegate.updated_at.to_i
+            if new_modified_at > modified_at.to_i
+              logger.info "Plex section was updated, refreshing"
+              self.modified_at = Time.at(new_modified_at)
+              super.clear
+            end
+
+            if super.blank?
+
+              if category_value
+                listing = delegate.send(category_type, category_value[:key])
+              else
+                listing = delegate.send(category_type)
+              end
+
+              Array(listing).each do |media|
+                if media.is_a?(::Plex::Movie)
+                  add_child(Movie.new(media))
+                elsif media.is_a?(::Plex::Episode)
+                  add_child(Episode.new(media))
+                elsif media.is_a?(::Plex::Show)
+                  add_child(Show.new(media))
+                else
+                  logger.error "Unknown type for #{media.class} in #{self.title}"
+                end
+              end
+            end
+          end
+
+          super
+        end
+
+      end
+
+    end
+  end
+end

data/lib/tivohmo/adapters/plex/episode.rb

@@ -0,0 +1,26 @@
+module TivoHMO
+  module Adapters
+    module Plex
+
+      class Episode
+        include TivoHMO::API::Item
+        include GemLogger::LoggerSupport
+
+        attr_reader :delegate
+
+        def initialize(delegate)
+          # delegate is a Plex::Episode
+          @delegate = delegate
+
+          super(delegate.key)
+
+          self.title = delegate.title
+          self.modified_at = Time.at(delegate.updated_at.to_i)
+          self.created_at = Time.at(delegate.added_at.to_i)
+        end
+
+      end
+
+    end
+  end
+end

data/lib/tivohmo/adapters/plex/metadata.rb

@@ -0,0 +1,24 @@
+module TivoHMO
+  module Adapters
+    module Plex
+
+      class Metadata
+        include TivoHMO::API::Metadata
+        include GemLogger::LoggerSupport
+
+        def initialize(item)
+          super
+
+          begin
+            self.description = item.delegate.summary
+            self.duration = (item.delegate.duration.to_i / 1000).to_i
+          rescue => e
+            logger.error "Failed to read plex metadata: #{e}"
+          end
+        end
+
+      end
+
+    end
+  end
+end

data/lib/tivohmo/adapters/plex/movie.rb

@@ -0,0 +1,26 @@
+module TivoHMO
+  module Adapters
+    module Plex
+
+      class Movie
+        include TivoHMO::API::Item
+        include GemLogger::LoggerSupport
+
+        attr_reader :delegate
+
+        def initialize(delegate)
+          # delegate is a Plex::Movie
+          @delegate = delegate
+
+          super(delegate.key)
+
+          self.title = delegate.title
+          self.modified_at = Time.at(delegate.updated_at.to_i)
+          self.created_at = Time.at(delegate.added_at.to_i)
+        end
+
+      end
+
+    end
+  end
+end

data/lib/tivohmo/adapters/plex/qualified_category.rb

@@ -0,0 +1,51 @@
+module TivoHMO
+  module Adapters
+    module Plex
+
+      class QualifiedCategory
+        include TivoHMO::API::Container
+        include GemLogger::LoggerSupport
+        include MonitorMixin
+
+        attr_reader :delegate
+        attr_accessor :category_type, :category_qualifier
+
+        def initialize(delegate, category_type, category_qualifier)
+          # delegate is a Plex::Section
+          @delegate = delegate
+
+          super(delegate.key)
+
+          self.category_type = category_type
+          self.category_qualifier = category_qualifier
+          self.title = category_type.to_s.titleize
+          self.modified_at = Time.at(delegate.updated_at.to_i)
+          self.created_at = Time.at(delegate.updated_at.to_i)
+        end
+
+        def children
+          synchronize do
+
+            delegate.refresh
+            new_modified_at = delegate.updated_at.to_i
+            if new_modified_at > modified_at.to_i
+              logger.info "Plex section was updated, refreshing"
+              self.modified_at = Time.at(new_modified_at)
+              super.clear
+            end
+
+            if super.blank?
+              Array(delegate.send(category_qualifier)).each do |category_value|
+                add_child(Category.new(delegate, category_type, category_value))
+              end
+            end
+          end
+
+          super
+        end
+
+      end
+
+    end
+  end
+end

data/lib/tivohmo/adapters/plex/season.rb

@@ -0,0 +1,39 @@
+module TivoHMO
+  module Adapters
+    module Plex
+
+      class Season
+        include TivoHMO::API::Container
+        include GemLogger::LoggerSupport
+        include MonitorMixin
+
+        attr_reader :delegate
+
+        def initialize(delegate)
+          # delegate is a Plex::Season
+          @delegate = delegate
+
+          super(delegate.key)
+
+          self.title = delegate.title
+#          self.modified_at = Time.at(delegate.updated_at.to_i)
+#          self.created_at = Time.at(delegate.added_at.to_i)
+        end
+
+        def children
+          synchronize do
+            if super.blank?
+              Array(delegate.episodes).each do |media|
+                add_child(Episode.new(media))
+              end
+            end
+          end
+
+          super
+        end
+
+      end
+
+    end
+  end
+end

data/lib/tivohmo/adapters/plex/section.rb

@@ -0,0 +1,48 @@
+module TivoHMO
+  module Adapters
+    module Plex
+
+      class Section
+        include TivoHMO::API::Container
+        include GemLogger::LoggerSupport
+        include MonitorMixin
+
+        attr_reader :delegate
+
+        def initialize(delegate)
+          # delegate is a Plex::Section
+          @delegate = delegate
+
+          super(delegate.key)
+
+          self.title = delegate.title
+          self.modified_at = Time.at(delegate.updated_at.to_i)
+          self.created_at = Time.at(delegate.updated_at.to_i)
+        end
+
+        def children
+          synchronize do
+            if super.blank?
+              add_child(Category.new(delegate, :recently_added))
+              #add_child(Category.new(delegate, :unwatched))
+              add_child(Category.new(delegate, :newest))
+              add_child(Category.new(delegate, :on_deck))
+              add_child(Category.new(delegate, :recently_viewed))
+              add_child(Category.new(delegate, :all))
+              add_child(QualifiedCategory.new(delegate, :by_genre, :genres))
+              add_child(QualifiedCategory.new(delegate, :by_year, :years))
+              add_child(QualifiedCategory.new(delegate, :by_first_character, :first_characters))
+              add_child(QualifiedCategory.new(delegate, :by_collection, :collections))
+              add_child(QualifiedCategory.new(delegate, :by_folder, :folders))
+              add_child(QualifiedCategory.new(delegate, :by_content_rating, :content_ratings))
+            end
+          end
+
+          super
+        end
+
+      end
+
+    end
+  end
+end

data/lib/tivohmo/adapters/plex/show.rb

@@ -0,0 +1,39 @@
+module TivoHMO
+  module Adapters
+    module Plex
+
+      class Show
+        include TivoHMO::API::Container
+        include GemLogger::LoggerSupport
+        include MonitorMixin
+
+        attr_reader :delegate
+
+        def initialize(delegate)
+          # delegate is a Plex::Show
+          @delegate = delegate
+
+          super(delegate.key)
+
+          self.title = delegate.title
+          self.modified_at = Time.at(delegate.updated_at.to_i)
+          self.created_at = Time.at(delegate.added_at.to_i)
+        end
+
+        def children
+          synchronize do
+            if super.blank?
+              Array(delegate.seasons).each do |media|
+                add_child(Season.new(media))
+              end
+            end
+          end
+
+          super
+        end
+
+      end
+
+    end
+  end
+end

data/lib/tivohmo/adapters/plex/transcoder.rb

@@ -0,0 +1,19 @@
+require 'tivohmo/adapters/streamio'
+
+module TivoHMO
+  module Adapters
+    module Plex
+
+      class Transcoder < TivoHMO::Adapters::StreamIO::Transcoder
+        include GemLogger::LoggerSupport
+
+        def initialize(item)
+          super(item)
+          self.source_filename = item.delegate.medias.first.parts.first.file
+        end
+
+      end
+
+    end
+  end
+end

data/lib/tivohmo/adapters/plex.rb

@@ -0,0 +1,11 @@
+require 'plex-ruby'
+require_relative 'plex/movie'
+require_relative 'plex/episode'
+require_relative 'plex/season'
+require_relative 'plex/show'
+require_relative 'plex/category'
+require_relative 'plex/qualified_category'
+require_relative 'plex/section'
+require_relative 'plex/application'
+require_relative 'plex/metadata'
+require_relative 'plex/transcoder'

data/lib/tivohmo/adapters/streamio/metadata.rb

@@ -0,0 +1,26 @@
+module TivoHMO
+  module Adapters
+    module StreamIO
+
+      # Extracts some basic metadata using the streamio gem
+      class Metadata
+        include TivoHMO::API::Metadata
+        include GemLogger::LoggerSupport
+
+        attr_accessor :movie
+
+        def initialize(item)
+          super(item)
+          begin
+            self.movie = FFMPEG::Movie.new(item.identifier)
+            self.duration = movie.duration.to_i
+          rescue => e
+            logger.error "Failed to read movie metadata: #{e}"
+          end
+        end
+
+      end
+
+    end
+  end
+end