syobocal 0.10.0 → 0.11.0

data/lib/syobocal/comment/element/text.rb

@@ -0,0 +1,21 @@
+module Syobocal
+  module Comment
+    module Element
+      class Text
+        attr_reader :str
+
+        def initialize(str)
+          @str = str
+        end
+
+        def ==(other)
+          other.instance_of?(self.class) && other.str == str
+        end
+
+        def self.create(text)
+          new(text)
+        end
+      end
+    end
+  end
+end

data/lib/syobocal/comment/element/text_node.rb

@@ -0,0 +1,55 @@
+module Syobocal
+  module Comment
+    module Element
+      class TextNode
+        attr_reader :text_elements
+
+        SUBJECT_SEPARATOR = "、"
+        ENCODED_SEPARATOR = "\t"
+
+        def initialize(text_elements)
+          @text_elements = text_elements
+        end
+
+        def inner_text
+          text_elements.map(&:str).join("")
+        end
+
+        def ==(other)
+          other.instance_of?(self.class) && other.text_elements == text_elements
+        end
+
+        def split
+          buffer = ""
+
+          text_elements.each do |node|
+            case node
+            when Element::Text
+              buffer << node.str.gsub(SUBJECT_SEPARATOR, ENCODED_SEPARATOR)
+            when Element::Link
+              buffer << node.str
+            end
+          end
+
+          buffer.scan(/[^#{ENCODED_SEPARATOR}\(]+(?:\(.*?\))?/).map { |str| str.gsub(ENCODED_SEPARATOR, SUBJECT_SEPARATOR).strip }
+        end
+
+        def self.match?(line)
+          true
+        end
+
+        def self.parse(text)
+          elements = text.split(/(\[\[[^\[\]]*?\]\])/).select { |s| !s.empty? }.map do |s|
+            if s.match(/\A\[\[[^\[\]]*?\]\]\Z/)
+              Link.create(s)
+            else
+              Text.create(s)
+            end
+          end
+
+          Element::TextNode.new(elements)
+        end
+      end
+    end
+  end
+end

data/lib/syobocal/comment/music.rb

@@ -0,0 +1,15 @@
+module Syobocal
+  module Comment
+    class Music
+      attr_reader :title, :category, :data_list
+
+      def initialize(title, category, data_list)
+        @title, @category, @data_list = title, category, data_list
+      end
+
+      def ==(other)
+        other.instance_of?(self.class) && other.title == title && other.category == category && other.data_list == data_list
+      end
+    end
+  end
+end

data/lib/syobocal/comment/music_data.rb

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

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,24 @@
+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
+    end
+  end
+end

data/lib/syobocal/comment/section.rb

@@ -0,0 +1,123 @@
+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
+          value = row.value_node.inner_text
+          MusicData.new(attr, value)
+        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)
@@ -40,7 +40,7 @@ module Syobocal
       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)
         }
@@ -95,7 +95,7 @@ module Syobocal
       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)
         }
@@ -147,7 +147,7 @@ module Syobocal
       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)
         }
@@ -193,7 +193,7 @@ module Syobocal
       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)
         }
@@ -235,7 +235,7 @@ module Syobocal
       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)
@@ -251,7 +251,7 @@ module Syobocal
       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)
@@ -267,7 +267,7 @@ module Syobocal
       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)
@@ -283,7 +283,7 @@ module Syobocal
       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)