syobocal 0.10.0 → 0.14.0

data/lib/syobocal/comment/parser.rb

@@ -0,0 +1,103 @@
+module Syobocal
+  module Comment
+    class Parser
+      ELEMENT_CLASSES = [
+        Element::Header2,
+        Element::Header1,
+        Element::List,
+        Element::Row,
+        Element::Blank,
+        Element::TextNode, # NOTE Sentinel
+      ]
+
+      def initialize(comment)
+        @comment = comment
+      end
+
+      def parse
+        return @parse if defined? @parse
+
+        return @parse = Element::Root.new([]) unless @comment
+
+        elements = @comment.each_line.map do |line|
+          line.chomp!
+          ELEMENT_CLASSES.find { |clazz| clazz.match?(line) }.parse(line)
+        end
+
+        @parse = Element::Root.new(elements)
+      end
+
+      def staffs
+        return @staffs if defined? @staffs
+
+        rows = sections.find { |section| section.staff_section? }&.rows || []
+
+        @staffs = create_staff_list(rows)
+      end
+
+      def casts
+        return @casts if defined? @casts
+
+        rows = sections.find { |section| section.cast_section? }&.rows || []
+
+        @casts = create_cast_list(rows)
+      end
+
+      def musics
+        return @musics if defined? @musics
+
+        @musics = create_musics(all_sections)
+      end
+
+      def links
+        return @links if defined? @links
+
+        @links = sections.find { |section| section.link_section? }&.links || []
+      end
+
+      def sections
+        return @sections if defined? @sections
+
+        @sections = Section.create_sections(parse.elements)
+      end
+
+      def all_sections
+        return enum_for(:all_sections) unless block_given?
+
+        sections.each do |section|
+          yield section
+
+          section.sub_sections.each do |sub_section|
+            yield sub_section
+          end
+        end
+      end
+
+      private
+
+      def create_staff_list(rows)
+        rows.map do |row|
+          role = row.attr_node.inner_text
+          people = row.value_node.split.map { |str| Person.parse(str) }
+
+          Staff.new(role, people)
+        end
+      end
+
+      def create_cast_list(rows)
+        rows.map do |row|
+          character = row.attr_node.inner_text
+          people = row.value_node.split.map { |str| Person.parse(str) }
+
+          Cast.new(character, people)
+        end
+      end
+
+      def create_musics(sections)
+        sections.each_with_object([]) do |section, musics|
+          musics << section.to_music if section.music_section?
+        end
+      end
+    end
+  end
+end

data/lib/syobocal/comment/person.rb

@@ -0,0 +1,33 @@
+module Syobocal
+  module Comment
+    class Person
+      PERSON_SEPARATOR = "、"
+      ENCODED_SEPARATOR = "\t"
+
+      attr_reader :name, :note
+
+      def initialize(name, note)
+        @name, @note = name, note
+      end
+
+      def ==(other)
+        other.instance_of?(self.class) && other.name == name && other.note == note
+      end
+
+      def self.parse(str)
+        _, name, note = *(str.match(/\A([^\(\)]+?)(?:\((.*?)\))?\Z/).to_a)
+
+        Person.new(name, note)
+      end
+
+      def self.multi_parse(str)
+        Helper::Fragment.parse(str).to_a.map{|f|
+          name = f.text
+          note = f&.child&.to_s
+
+          Person.new(name, note)
+        }
+      end
+    end
+  end
+end

data/lib/syobocal/comment/section.rb

@@ -0,0 +1,136 @@
+module Syobocal
+  module Comment
+    class Section
+      attr_reader :header, :elements, :sub_sections
+
+      class RestrictedOperation < StandardError; end
+
+      def initialize(header, elements)
+        @header, @elements = header, elements
+        @sub_sections = []
+      end
+
+      def sub_section?
+        @header.instance_of? Element::Header2
+      end
+
+      def add_subsection(sub_section)
+        raise RestrictedOperation if sub_section?
+        raise RestrictedOperation unless sub_section.sub_section?
+
+        @sub_sections << sub_section
+      end
+
+      STAFF_WORD = "スタッフ"
+      CAST_WORD = "キャスト"
+      LINK_WORD = "リンク"
+      MUSIC_WORDS = %w(テーマ ソング 歌 曲)
+      MUSIC_TITLE_REGEXP = /\A(.*)「(.+?)」\Z/
+
+      def staff_section?
+        header.text_node.inner_text.include?(STAFF_WORD)
+      end
+
+      def cast_section?
+        header.text_node.inner_text.include?(CAST_WORD)
+      end
+
+      def link_section?
+        header.text_node.inner_text.include?(LINK_WORD)
+      end
+
+      def music_section?
+        str = header.text_node.inner_text
+
+        MUSIC_WORDS.any? { |keyword| str.include?(keyword) } && str.match(MUSIC_TITLE_REGEXP)
+      end
+
+      def to_music
+        m = header.text_node.inner_text.match(MUSIC_TITLE_REGEXP)
+        title = m[2]
+        category = m[1]
+
+        data_list = rows.map do |row|
+          attr = row.attr_node.inner_text
+
+          attr_fragment = Helper::Fragment.parse(attr)
+
+          if attr_fragment.to_a.size == 1
+            attr_text = attr_fragment.text
+            attr_note = attr_fragment&.child&.to_s
+          else
+            attr_text = attr_fragment.to_s
+            attr_note = nil
+          end
+
+          value = row.value_node.inner_text
+          people = Person.multi_parse(value)
+
+          MusicData.new(attr, attr_text, attr_note, value, people)
+        end
+
+        Music.new(title, category, data_list)
+      end
+
+      def rows
+        elements.select { |element| element.is_a? Element::Row }
+      end
+
+      def links
+        elements.each_with_object([]) { |element, links|
+          case element
+          when Element::List
+            links << element.text_node.text_elements.select { |elm| elm.instance_of? Element::Link }
+          when Element::TextNode
+            links << element.text_elements.select { |elm| elm.instance_of? Element::Link }
+          end
+        }.flatten
+      end
+
+      def self.create_sections(elements)
+        sections = []
+        current_header = nil
+        buffered_elements = []
+
+        elements.each do |element|
+          case element
+          when Element::Header1, Element::Header2
+            if current_header
+              sections << Section.new(current_header, buffered_elements)
+              buffered_elements = []
+            end
+
+            current_header = element
+          else
+            buffered_elements << element if current_header
+          end
+        end
+
+        if current_header
+          sections << Section.new(current_header, buffered_elements)
+        end
+
+        structure_sections(sections)
+      end
+
+      def self.structure_sections(sections)
+        root_sections = []
+        parent_section = nil
+
+        sections.each do |section|
+          if section.sub_section?
+            # NOTE parent_sectionがないパターンは不正なので無視する
+            parent_section.add_subsection(section) if parent_section
+          else
+            root_sections << section
+            parent_section = section
+          end
+        end
+
+        root_sections
+      end
+
+      private_class_method :structure_sections
+    end
+  end
+end

data/lib/syobocal/comment/staff.rb

@@ -0,0 +1,15 @@
+module Syobocal
+  module Comment
+    class Staff
+      attr_reader :role, :people
+
+      def initialize(role, people)
+        @role, @people = role, people
+      end
+
+      def ==(other)
+        other.instance_of?(self.class) && other.role == role && other.people == people
+      end
+    end
+  end
+end

data/lib/syobocal/db.rb

@@ -3,14 +3,14 @@ module Syobocal
     module Mapper
       def map(elm)
         result = {}
-        elm.each_element{|child|
-          set(result, to_snake(child.name).to_sym, child, @map[child.name]) 
+        elm.each_element { |child|
+          set(result, to_snake(child.name).to_sym, child, @map[child.name])
         }
         result
       end
 
       def to_snake(name)
-        name.gsub(/([a-z])([A-Z])/){ $1 + '_' + $2 }.downcase
+        name.gsub(/([a-z])([A-Z])/) { $1 + "_" + $2 }.downcase
       end
 
       def set(hash, key, elm, type)
@@ -36,11 +36,11 @@ module Syobocal
 
     module TitleLookup
       def get(params = {})
-        parse(open(url(params)))
+        parse(URI.open(url(params)))
       end
 
       def url(params)
-        'http://cal.syoboi.jp/db.php?Command=TitleLookup' + Syobocal::Util.format_params_amp(params)
+        "http://cal.syoboi.jp/db.php?Command=TitleLookup" + Syobocal::Util.format_params_amp(params)
       end
 
       def parse(xml)
@@ -48,10 +48,10 @@ module Syobocal
 
         result = LookupResult.new
 
-        result.code = xml.elements['TitleLookupResponse/Result/Code'].text.to_i
-        result.message = xml.elements['TitleLookupResponse/Result/Message'].text
+        result.code = xml.elements["TitleLookupResponse/Result/Code"].text.to_i
+        result.message = xml.elements["TitleLookupResponse/Result/Message"].text
 
-        xml.elements.each('TitleLookupResponse/TitleItems/TitleItem'){|item|
+        xml.elements.each("TitleLookupResponse/TitleItems/TitleItem") { |item|
           mapper = Mapper.new
           result << mapper.map(item)
         }
@@ -91,11 +91,11 @@ module Syobocal
 
     module ProgLookup
       def get(params = {})
-        parse(open(url(params)))
+        parse(URI.open(url(params)))
       end
 
       def url(params)
-        'http://cal.syoboi.jp/db.php?Command=ProgLookup' + Syobocal::Util.format_params_amp(params)
+        "http://cal.syoboi.jp/db.php?Command=ProgLookup" + Syobocal::Util.format_params_amp(params)
       end
 
       def parse(xml)
@@ -103,10 +103,10 @@ module Syobocal
 
         result = LookupResult.new
 
-        result.code = xml.elements['ProgLookupResponse/Result/Code'].text.to_i
-        result.message = xml.elements['ProgLookupResponse/Result/Message'].text
+        result.code = xml.elements["ProgLookupResponse/Result/Code"].text.to_i
+        result.message = xml.elements["ProgLookupResponse/Result/Message"].text
 
-        xml.elements.each('ProgLookupResponse/ProgItems/ProgItem'){|item|
+        xml.elements.each("ProgLookupResponse/ProgItems/ProgItem") { |item|
           mapper = Mapper.new
           result << mapper.map(item)
         }
@@ -143,11 +143,11 @@ module Syobocal
 
     module ChLookup
       def get(params = {})
-        parse(open(url(params)))
+        parse(URI.open(url(params)))
       end
 
       def url(params)
-        'http://cal.syoboi.jp/db.php?Command=ChLookup' + Syobocal::Util.format_params_amp(params)
+        "http://cal.syoboi.jp/db.php?Command=ChLookup" + Syobocal::Util.format_params_amp(params)
       end
 
       def parse(xml)
@@ -155,10 +155,10 @@ module Syobocal
 
         result = LookupResult.new
 
-        result.code = xml.elements['ChLookupResponse/Result/Code'].text.to_i
-        result.message = xml.elements['ChLookupResponse/Result/Message'].text
+        result.code = xml.elements["ChLookupResponse/Result/Code"].text.to_i
+        result.message = xml.elements["ChLookupResponse/Result/Message"].text
 
-        xml.elements.each('ChLookupResponse/ChItems/ChItem'){|item|
+        xml.elements.each("ChLookupResponse/ChItems/ChItem") { |item|
           mapper = Mapper.new
           result << mapper.map(item)
         }
@@ -189,11 +189,11 @@ module Syobocal
 
     module ChGroupLookup
       def get(params = {})
-        parse(open(url(params)))
+        parse(URI.open(url(params)))
       end
 
       def url(params)
-        'http://cal.syoboi.jp/db.php?Command=ChGroupLookup' + Syobocal::Util.format_params_amp(params)
+        "http://cal.syoboi.jp/db.php?Command=ChGroupLookup" + Syobocal::Util.format_params_amp(params)
       end
 
       def parse(xml)
@@ -201,10 +201,10 @@ module Syobocal
 
         result = LookupResult.new
 
-        result.code = xml.elements['ChGroupLookupResponse/Result/Code'].text.to_i
-        result.message = xml.elements['ChGroupLookupResponse/Result/Message'].text
+        result.code = xml.elements["ChGroupLookupResponse/Result/Code"].text.to_i
+        result.message = xml.elements["ChGroupLookupResponse/Result/Message"].text
 
-        xml.elements.each('ChGroupLookupResponse/ChGroupItems/ChGroupItem'){|item|
+        xml.elements.each("ChGroupLookupResponse/ChGroupItems/ChGroupItem") { |item|
           mapper = Mapper.new
           result << mapper.map(item)
         }
@@ -231,11 +231,11 @@ module Syobocal
 
     module TitleViewCount
       def get(params = {})
-        parse(open(url(params)))
+        parse(URI.open(url(params)))
       end
 
       def url(params)
-        'http://cal.syoboi.jp/db.php?Command=TitleViewCount' + Syobocal::Util.format_params_amp(params)
+        "http://cal.syoboi.jp/db.php?Command=TitleViewCount" + Syobocal::Util.format_params_amp(params)
       end
 
       def parse(xml)
@@ -247,11 +247,11 @@ module Syobocal
 
     module TitleRankHistory
       def get(params = {})
-        parse(open(url(params)))
+        parse(URI.open(url(params)))
       end
 
       def url(params)
-        'http://cal.syoboi.jp/db.php?Command=TitleRankHistory' + Syobocal::Util.format_params_amp(params)
+        "http://cal.syoboi.jp/db.php?Command=TitleRankHistory" + Syobocal::Util.format_params_amp(params)
       end
 
       def parse(xml)
@@ -263,11 +263,11 @@ module Syobocal
 
     module TitlePointHistory
       def get(params = {})
-        parse(open(url(params)))
+        parse(URI.open(url(params)))
       end
 
       def url(params)
-        'http://cal.syoboi.jp/db.php?Command=TitlePointHistory' + Syobocal::Util.format_params_amp(params)
+        "http://cal.syoboi.jp/db.php?Command=TitlePointHistory" + Syobocal::Util.format_params_amp(params)
       end
 
       def parse(xml)
@@ -279,11 +279,11 @@ module Syobocal
 
     module TitlePointTop
       def get(params = {})
-        parse(open(url(params)))
+        parse(URI.open(url(params)))
       end
 
       def url(params)
-        'http://cal.syoboi.jp/db.php?Command=TitlePointTop' + Syobocal::Util.format_params_amp(params)
+        "http://cal.syoboi.jp/db.php?Command=TitlePointTop" + Syobocal::Util.format_params_amp(params)
       end
 
       def parse(xml)
@@ -298,21 +298,21 @@ module Syobocal
 
       result = TableResult.new
 
-      result.code = xml.elements['TableData/Result/Code'].text.to_i
-      result.message = xml.elements['TableData/Result/Message'].text
+      result.code = xml.elements["TableData/Result/Code"].text.to_i
+      result.message = xml.elements["TableData/Result/Message"].text
 
       result.columns = []
       if result.code == 200
-        result.title = xml.elements['TableData/Title'].text
-        result.type = xml.elements['TableData/Type'].text
+        result.title = xml.elements["TableData/Title"].text
+        result.type = xml.elements["TableData/Type"].text
 
-        xml.elements.each('TableData/Columns/Value'){|item|
+        xml.elements.each("TableData/Columns/Value") { |item|
           result.columns << item.text
         }
 
-        xml.elements.each('TableData/Line'){|line|
+        xml.elements.each("TableData/Line") { |line|
           line_data = {}
-          line.elements.each_with_index{|value, index|
+          line.elements.each_with_index { |value, index|
             key = result.columns[index]
             if key == "Date"
               line_data[key] = Date.parse(value.text)

data/lib/syobocal/json.rb

@@ -2,11 +2,11 @@ module Syobocal
   module JSON
     module TitleMedium
       def get(params = {})
-        parse(open(url(params)))
+        parse(URI.open(url(params)))
       end
 
       def url(params = {})
-        'http://cal.syoboi.jp/json.php?Req=TitleMedium' + Syobocal::Util.format_params_amp(params)
+        "http://cal.syoboi.jp/json.php?Req=TitleMedium" + Syobocal::Util.format_params_amp(params)
       end
 
       def parse(json)
@@ -18,11 +18,11 @@ module Syobocal
 
     module TitleLarge
       def get(params = {})
-        parse(open(url(params)))
+        parse(URI.open(url(params)))
       end
 
       def url(params = {})
-        'http://cal.syoboi.jp/json.php?Req=TitleLarge' + Syobocal::Util.format_params_amp(params)
+        "http://cal.syoboi.jp/json.php?Req=TitleLarge" + Syobocal::Util.format_params_amp(params)
       end
 
       def parse(json)
@@ -34,11 +34,11 @@ module Syobocal
 
     module TitleFull
       def get(params = {})
-        parse(open(url(params)))
+        parse(URI.open(url(params)))
       end
 
       def url(params = {})
-        'http://cal.syoboi.jp/json.php?Req=TitleFull' + Syobocal::Util.format_params_amp(params)
+        "http://cal.syoboi.jp/json.php?Req=TitleFull" + Syobocal::Util.format_params_amp(params)
       end
 
       def parse(json)
@@ -50,11 +50,11 @@ module Syobocal
 
     module ProgramByPID
       def get(params = {})
-        parse(open(url(params)))
+        parse(URI.open(url(params)))
       end
 
       def url(params = {})
-        'http://cal.syoboi.jp/json.php?Req=ProgramByPID' + Syobocal::Util.format_params_amp(params)
+        "http://cal.syoboi.jp/json.php?Req=ProgramByPID" + Syobocal::Util.format_params_amp(params)
       end
 
       def parse(json)
@@ -66,11 +66,11 @@ module Syobocal
 
     module ProgramByCount
       def get(params = {})
-        parse(open(url(params)))
+        parse(URI.open(url(params)))
       end
 
       def url(params = {})
-        'http://cal.syoboi.jp/json.php?Req=ProgramByCount' + Syobocal::Util.format_params_amp(params)
+        "http://cal.syoboi.jp/json.php?Req=ProgramByCount" + Syobocal::Util.format_params_amp(params)
       end
 
       def parse(json)
@@ -82,11 +82,11 @@ module Syobocal
 
     module ProgramByDate
       def get(params = {})
-        parse(open(url(params)))
+        parse(URI.open(url(params)))
       end
 
       def url(params = {})
-        'http://cal.syoboi.jp/json.php?Req=ProgramByDate' + Syobocal::Util.format_params_amp(params)
+        "http://cal.syoboi.jp/json.php?Req=ProgramByDate" + Syobocal::Util.format_params_amp(params)
       end
 
       def parse(json)
@@ -98,11 +98,11 @@ module Syobocal
 
     module SubTitles
       def get(params = {})
-        parse(open(url(params)))
+        parse(URI.open(url(params)))
       end
 
       def url(params = {})
-        'http://cal.syoboi.jp/json.php?Req=SubTitles' + Syobocal::Util.format_params_amp(params)
+        "http://cal.syoboi.jp/json.php?Req=SubTitles" + Syobocal::Util.format_params_amp(params)
       end
 
       def parse(json)
@@ -114,11 +114,11 @@ module Syobocal
 
     module ChFilter
       def get(params = {})
-        parse(open(url(params)))
+        parse(URI.open(url(params)))
       end
 
       def url(params = {})
-        'http://cal.syoboi.jp/json.php?Req=ChFilter' + Syobocal::Util.format_params_amp(params)
+        "http://cal.syoboi.jp/json.php?Req=ChFilter" + Syobocal::Util.format_params_amp(params)
       end
 
       def parse(json)
@@ -130,11 +130,11 @@ module Syobocal
 
     module ChIDFilter
       def get(params = {})
-        parse(open(url(params)))
+        parse(URI.open(url(params)))
       end
 
       def url(params = {})
-        'http://cal.syoboi.jp/json.php?Req=ChIDFilter' + Syobocal::Util.format_params_amp(params)
+        "http://cal.syoboi.jp/json.php?Req=ChIDFilter" + Syobocal::Util.format_params_amp(params)
       end
 
       def parse(json)

data/lib/syobocal/rss.rb

@@ -1,11 +1,11 @@
 module Syobocal
   module RSS
     def get(params = {})
-      parse(open(url(params)))
+      parse(URI.open(url(params)))
     end
 
     def url(params = {})
-      'http://cal.syoboi.jp/rss.php' + Syobocal::Util.format_params(params)
+      "http://cal.syoboi.jp/rss.php" + Syobocal::Util.format_params(params)
     end
 
     def parse(rss)
@@ -13,25 +13,25 @@ module Syobocal
 
       result = Result.new
 
-      channel = rss.elements['rdf:RDF/channel']
-      result.title = channel.elements['title'].text
-      result.link = channel.elements['link'].text
-      result.description = channel.elements['description'].text
-      
-      rss.elements.each('rdf:RDF/item'){|item|
-        tv = item.elements['tv:feed']
+      channel = rss.elements["rdf:RDF/channel"]
+      result.title = channel.elements["title"].text
+      result.link = channel.elements["link"].text
+      result.description = channel.elements["description"].text
+
+      rss.elements.each("rdf:RDF/item") { |item|
+        tv = item.elements["tv:feed"]
         result << {
           :about => item.attribute("rdf:about").to_s,
-          :title => item.elements['title'].text,
-          :link => item.elements['link'].text,
-          :description => item.elements['description'].text,
-          :dc_date => Time.parse(item.elements['dc:date'].text),
-          :dc_publisher => item.elements['dc:publisher'].text,
-          :tv_genre => tv.elements['tv:genre'].text,
-          :tv_start_datetime => Time.parse(tv.elements['tv:startDatetime'].text),
-          :tv_end_datetime => Time.parse(tv.elements['tv:endDatetime'].text),
-          :tv_iepg_url => tv.elements['tv:iepgUrl'].text,
-          :tv_performer => tv.elements['tv:performer'].text,
+          :title => item.elements["title"].text,
+          :link => item.elements["link"].text,
+          :description => item.elements["description"].text,
+          :dc_date => Time.parse(item.elements["dc:date"].text),
+          :dc_publisher => item.elements["dc:publisher"].text,
+          :tv_genre => tv.elements["tv:genre"].text,
+          :tv_start_datetime => Time.parse(tv.elements["tv:startDatetime"].text),
+          :tv_end_datetime => Time.parse(tv.elements["tv:endDatetime"].text),
+          :tv_iepg_url => tv.elements["tv:iepgUrl"].text,
+          :tv_performer => tv.elements["tv:performer"].text,
         }
       }