lib/afr_load.rb in afr_load-0.1.0 vs lib/afr_load.rb in afr_load-0.1.1

@@ -1,7 +1,45 @@
 # encoding: utf-8
 
+require "oga"
+require "httpclient"
+
 require "afr_load/version"
-require "afr_load/afr_load"
+require "afr_load/tv_program"
+require "afr_load/parser.rb"
 
 module AfrLoad
+    class AfrLoad
+        attr_reader :url, :programs, :document
+
+        AFR_LOAD_URL = "http://www.tv-tokyo.co.jp/telecine/oa_afr_load/"
+
+        def initialize(url = AFR_LOAD_URL)
+            @programs = Array.new()
+            @url = url
+        end
+
+        def get_schedule()
+            @document = self.class.fetch_schedule(@url)
+            @programs = Parser.parse(@document).flatten
+            @programs
+        end
+
+        def self.get_schedule(url = AFR_LOAD_URL)
+            afr = self.new(url)
+            afr.get_schedule()
+        end
+
+        def self.get_schedule_from_file(file_path)
+            handler = File.open(file_path)
+            document = Oga.parse_html(handler)
+        end
+
+        def self.fetch_schedule(url = AFR_LOAD_URL)
+            Oga.parse_html(Enumerator.new do |yielder|
+                HTTPClient.get(url) do |chunk|
+                    yielder << chunk.force_encoding("utf-8")
+                end
+            end)
+        end
+    end
 end