module CukeModeler

  # A class modeling the Doc String of a Step.

  class DocString

    include Raw
    include Nested


    # The content type associated with the doc string
    attr_accessor :content_type

    # Deprecated
    #
    # The contents of the doc string
    attr_accessor :contents

    # The contents of the doc string
    attr_accessor :contents_text


    # Creates a new DocString object and, if *source* is provided, populates
    # the object.
    def initialize(source = nil)
      @contents = []
      @contents_text = ''

      parsed_doc_string = process_source(source)

      build_doc_string(parsed_doc_string) if parsed_doc_string
    end

    # Returns a gherkin representation of the doc string.
    def to_s
      text = "\"\"\"#{content_type_output_string}\n"
      text << contents_output_string
      text << '"""'
    end


    private


    def process_source(source)
      case
        when source.is_a?(String)
          parse_doc_string(source)
        else
          source
      end
    end

    def parse_doc_string(source_text)
      base_file_string = "Feature:\nScenario:\n* step\n"
      source_text = base_file_string + source_text

      parsed_file = Parsing::parse_text(source_text)

      parsed_file.first['elements'].first['steps'].first['doc_string']
    end

    def build_doc_string(doc_string)
      populate_content_type(doc_string)
      populate_contents(doc_string)
      populate_raw_element(doc_string)
    end

    def populate_content_type(doc_string)
      @content_type = doc_string['content_type'] == "" ? nil : doc_string['content_type']
    end

    def populate_contents(doc_string)
      @contents = doc_string['value'].split($/, -1)
      @contents_text = doc_string['value']
    end

    def content_type_output_string
      content_type ? " #{content_type}" : ''
    end

    def contents_output_string
      contents_text.empty? ? '' : contents_text.gsub('"""', '\"\"\"') + "\n"
    end

  end
end