zipmark/rspec_api_documentation

module RspecApiDocumentation
  module Writers
    class CombinedTextWriter < Writer
      def write
        index.examples.each do |rspec_example|
          example = CombinedTextExample.new(rspec_example)
          FileUtils.mkdir_p(configuration.docs_dir.join(example.resource_name))
          File.open(configuration.docs_dir.join(example.resource_name, "index.txt"), "a+") do |f|
            f.print example.description
            f.print example.parameters

            example.requests.each_with_index do |(request, response), i|
              f.puts "Request:"
              f.puts request
              f.puts
              f.puts "Response:"
              f.puts response

              if i + 1 < example.requests.count
                f.puts
              end
            end

            unless rspec_example == index.examples.last
              f.puts
              f.puts
            end
          end
        end
      end

      def self.format_hash(hash, separator="=")
        hash.sort_by { |k, v| k }.inject("") do |out, (k, v)|
          out << "  #{k}#{separator}#{v}\n"
        end
      end
    end

    class CombinedTextExample
      attr_reader :example

      def initialize(example)
        @example = example
      end

      def resource_name
        example.resource_name.to_s.downcase.gsub(/\s+/, '_')
      end

      def description
        example.description + "\n" + "-" * example.description.length + "\n\n"
      end

      def parameters
        return "" unless example.metadata[:parameters]
        "Parameters:\n" + example.metadata[:parameters].inject("") do |out, parameter|
          out << "  * #{parameter[:name]} - #{parameter[:description]}\n"
        end + "\n"
      end

      def requests
        return [] unless example.metadata[:requests]
        example.metadata[:requests].map do |request|
          [format_request(request), format_response(request)]
        end
      end

      private
      def format_request(request)
        [
          [
            "  #{request[:request_method]} #{request[:request_path]}",
            format_hash(request[:request_headers], ": ")
          ],
          [
            format_string(request[:request_body]) || format_hash(request[:request_query_parameters])
          ]
        ].map { |x| x.compact.join("\n") }.reject(&:blank?).join("\n\n") + "\n"
      end

      def format_response(request)
        [
          [
            "  Status: #{request[:response_status]} #{request[:response_status_text]}",
            format_hash(request[:response_headers], ": ")
          ],
          [
            format_string(request[:response_body])
          ]
        ].map { |x| x.compact.join("\n") }.reject(&:blank?).join("\n\n") + "\n"
      end

      def format_string(string)
        return unless string
        string.split("\n").map { |line| "  #{line}" }.join("\n")
      end

      def format_hash(hash, separator="=")
        return unless hash
        hash.sort_by { |k, v| k }.map do |k, v|
          "  #{k}#{separator}#{v}"
        end.join("\n")
      end
    end
  end
end