add media type method

linjunpop · Oct 5, 2012 · d0dd60f · d0dd60f
1 parent fe980b8
commit d0dd60f
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/lib/airplayer/media.rb b/lib/airplayer/media.rb
@@ -19,14 +19,20 @@ def initialize(target)
     end
 
     def open
-      if @type == :file
-        Thread.start { @video_server.start }
-      end
+      Thread.start { @video_server.start } if file?
       @path
     end
 
     def close
-      @video_server.stop if @type == :local
+      @video_server.stop if file?
+    end
+
+    def file?
+      @type == :file
+    end
+
+    def url?
+      @type == :url
     end
   end
 end