org-parse 0.1.1

data/lib/org-parse/node.rb

@@ -0,0 +1,329 @@
+# -*- coding: utf-8 -*-
+
+module OrgParse
+
+  # syntax tree node
+  #
+  # _children_  is Node array
+  #
+  # === Node kind
+  # - _ROOT_ :: RootNode
+  #
+  # lines (StructParser)
+  # - _SECTION_ :: SectionNode
+  # - _HEADLINE_ :: HeadlineNode
+  # - _FN_DEFINE_ :: footnote definition
+  # - _VARIABLE_ :: VarNode
+  # - _WHITELINE_ :: WhitelineNode
+  # - _TEXTBLOCK_ :: TextlineNode's list will become paragraph
+  # - _TEXTLINE_  :: TextlineNode
+  # - _QUOTE_ :: HTML quoted text
+  # - _BLOCK_    :: BlockNode (verse, example, src, html, ...)
+  # - _UNORDERED_LIST_ :: Unorderd list <UL>
+  # - _ENUMLIST_ :: Ordered list <OL>
+  # - _DESCLIST_ :: Description list <DL>
+  # - _LISTITEM_ :: ListitemNode
+  # - _TABLE_ :: TableNode
+  # - _TABLE_SEP_ :: separator between th and td
+  # - _TABLE_ROW_ :: TableRowNode
+  #
+  #  inlines (InlineParser)
+  # - _BOLD_     :: bold
+  # - _ITALIC_   :: italic
+  # - _UNDER_LINE_ :: under line
+  # - _VERBATIM_ :: verbatim
+  # - _STRIKE_THROUGH_ :: strike through
+  # - _CODE_     :: code
+  # - _LINK_     :: LinkNode 参照
+  # - _QUOTE_    :: @<br/> , #+HTML ... 等
+  #
+  class Node
+    # Node type
+    attr_accessor :kind
+    # child nodes
+    attr_accessor :children   
+    # Node value
+    attr_accessor :value 
+    # Parent node
+    attr_accessor :parent
+
+    def initialize(kind = nil, children = [], value = nil)
+      @kind = kind
+      @children = children
+      @value = value
+      @done = false
+      @parent = nil
+      @verse = false
+      @example = false
+      @html = false
+      @src = false
+    end
+
+    def set_verse
+      @verse = true
+    end
+
+    def verse?
+      @verse
+    end
+
+    def set_example
+      @example = true
+    end
+
+    def example?
+      @example
+    end
+
+    def set_src
+      @src = true
+    end
+
+    def src?
+      @src
+    end
+
+    def set_html
+      @html = true
+    end
+
+    def html?
+      @html
+    end
+
+    def done?
+      @done
+    end
+
+    def done
+      @done = true
+    end
+
+    def is_leaf?
+      @children.empty?
+    end
+
+    #def is_container?
+    #  [:SECTION, :TEXTBLOCK, :LISTITEM, :UNORDERED_LIST, :ENUMLIST, :DESCLIST, :BLOCK].include? @kind
+    #end
+
+    def inspect
+      c = ''
+      c  = @children.collect{|i| indent2(i.inspect)}.join("\n") if @children
+      val = @value ? @value.to_s : '(nil)'
+      val = val.is_a?( Array) ? val.join(",") : val.gsub("\n", ' ')
+      "<#{self.class.name} #{@kind}:#{val}>" + (c.empty? ? "" : "\n") + c
+    end
+
+    def section_no_array
+      if @section_no
+        @section_no.split('.')
+      else
+        nil
+      end
+    end
+
+    def set_to_descendant(method, val=nil)
+      @children.each{|node|
+        if val
+          node.send method, *val
+        else
+          node.send method
+        end
+        node.set_to_descendant method, val
+      }
+    end
+
+    private
+
+    def indent2(str)
+      buf = ''
+      str.each_line{|i| buf << "  " << i }
+      buf
+    end
+  end
+
+  # Root
+  #
+  # 
+  class RootNode < Node
+    # options
+    attr_reader :options
+    attr_accessor :section_no
+
+    def initialize(opts)
+      super(:ROOT)
+      @options = opts
+      @section_no = "0"
+    end
+
+    def add(nodes)
+      @children += nodes
+    end
+  end
+
+  # セクション保持用
+  #
+  # _children[0]_ には、必ず Headline Nodeが入る
+  #
+  class SectionNode < Node
+    attr_accessor :section_no
+
+    # コンストラクタ
+    # [_headline_] セクション開始の Headline
+    # [_bodyitems_] セクションに含まれるブロック要素
+    def initialize(headline, bodyitems)
+      @section_no = "0"
+      headline.parent = self
+      super(:SECTION, bodyitems, headline)
+    end
+
+    # Headline のノードを返す
+    def headline
+      @value
+    end
+  end
+
+  # Headline を保持するクラス
+  class HeadlineNode < Node
+    attr_reader :is_comment, :level
+
+    def initialize(children, level)
+      @is_comment = false
+      @level = level.size
+      super(:HEADLINE, children)
+    end
+  end
+
+  # Whiteline
+  class WhitelineNode < Node
+    attr_reader :count
+    def initialize
+      super(:WHITELINE)
+      @value = 1
+    end
+    def increment
+      @value += 1
+    end
+    def count
+      @value
+    end
+  end
+
+
+  class TextlineNode < Node
+    attr_reader :indent
+
+    def initialize(children, vals)
+      @indent = vals[1]
+      super(:TEXTLINE, children, vals[0])
+    end
+
+  end
+
+  # リストアイテム
+  class ListitemNode < Node
+    attr_reader :dt, :type
+
+    def initialize(type, children, value, dt)
+      @type = type
+      @dt = dt
+      super(:LIST_ITEM, children, value)
+    end
+  end
+
+  # BLOCK の情報を保持
+  class BlockNode < Node
+    attr_reader :indent, :block_name, :syntax, :syntax_theme
+    def initialize(vals, children)
+      @block_name = vals[0].upcase
+      @indent = vals[2]
+      @syntax = ''
+      @syntax_theme = ''
+      super(:BLOCK, children, vals[1])
+      case @block_name
+      when 'VERSE'
+        set_to_descendant :set_verse
+      when 'EXAMPLE'
+        set_to_descendant :set_example
+      when 'SRC'
+        set_to_descendant :set_src
+        if vals[1] =~ /SRC\s*([^\s]+)\s+([^\s]+)\s*$/i
+          @syntax = $1.downcase
+          @syntax_theme = $2.downcase
+        elsif vals[1] =~ /SRC\s*(.+)\s*$/i
+          @syntax = $1.downcase
+        end
+      when 'HTML'
+        set_to_descendant :set_html
+      end
+    end
+
+    def line
+      @value
+    end
+  end
+
+  class TableNode < Node
+    attr_reader :hrows
+
+    def initialize(hrows, drows)
+      hrows.each{|h| h.is_head = true }
+      @hrows = hrows
+      super(:TABLE, drows)
+    end
+  end
+
+  # table row
+  class TableRowNode < Node
+    def initialize(cols, head = false)
+      super(:TABLE_ROW, cols, head)
+    end
+
+    def is_head?
+      @value
+    end
+
+    def is_head=(val)
+      @value = val
+    end
+  end
+
+  # Link保持
+  #
+  class LinkNode < Node
+    attr_reader :html_options, :caption
+    
+    def initialize(uri, children = [], vars = [])
+      @html_options = {}
+      @caption = nil
+      vars.each do |v|
+        if v =~ /^CAPTION:(.+)$/
+          @caption = $1.chomp
+        else
+          while v =~ /([^ =]+)=("[^"]+")/
+            @html_options[$1] = $2
+            v = $'
+          end
+        end
+      end
+      super(:LINK, children, uri)
+    end
+
+    # href
+    def uri
+      @value
+    end
+  end
+
+  # オプション変数
+  class VarNode < Node
+    attr_reader :name
+
+    def initialize(name, val)
+      @name = val
+      super(:VARIABLE, [], val)
+    end
+  end
+
+end