module Soaspec
  module WsdlGenerator
    # Attempt to calculate values of enumeration by looking up type in Schema
    def try_enum_for(type)
      raise "'@wsdl_schemas' must be defined" if @wsdl_schemas.nil?
      custom_type = @wsdl_schemas.xpath("//*[@name='#{type}']")
      if enumeration? custom_type
        prefix = custom_type.first.namespace.prefix
        enumerations = custom_type.xpath("//#{prefix}:enumeration")
        return 'Custom Type' if enumerations.empty?
        @enums_values = []
        enumerations.each { |enum_value| @enums_values << "'#{enum_value['value']}'" }
        "~randomize [#{@enums_values.join(', ')}]"
      else
        'Custom Type'
      end
    end

    # @param [Nokogiri::XML::NodeSet]
    # @return [Boolean] Whether WSDL type is an enumeration
    def enumeration?(type)
      return false unless type.first
      !type.xpath("*/#{type.first.namespace.prefix}:enumeration").empty?
    end

    # Return value of string after a namespace
    # @param [String] string String to parse for part after namespace
    def value_after_namespace(string)
      string.split(':').last
    end

    # Based on WSDL type return a valid value
    # @param [String] type Type without the WSDL
    def fill_in_field_from_type(type)
      case type
      when 'string'
        options[:string_default] # 'test string'
      when 'int'
        2
      when 'boolean'
        true
      when 'double'
        '1.5'
      else
        try_enum_for type
      end
    end

    # @param [Nokogiri::XML::Element]
    # @return [Boolean] True if nokogori element is a complex type, that is, has a complexType element underneath itself
    def complex_type?(element)
      element.children.any? { |child| child.name == 'complexType' }
    end

    # @param [String, Symbol] underscore_separated Snakecase value to be converted to camel case
    def camel_case(underscore_separated)
      underscore_separated.to_s.split('_').collect(&:capitalize).join
    end

    # @param [Hash] op_details Hash with details from WSDL for an operation
    # @return [Nokogiri::XML::NodeSet] List of the root elements in the SOAP body
    def root_elements_for(op_details)
      raise "'@wsdl_schemas' must be defined" if @wsdl_schemas.nil?
      root_element = @wsdl_schemas.at_xpath("//*[@name='#{op_details[:input]}']")
      raise 'Operation has no input defined' if root_element.nil?
      schema_namespace = root_element.namespace.prefix
      root_type = root_element['type']
      if root_type
        @wsdl_schemas.xpath("//*[@name='#{root_type.split(':').last}']//#{schema_namespace}:element")
      else
        return [] unless complex_type? root_element # Empty Array if there are no root elements
        complex_type = root_element.children.find { |c| c.name == 'complexType' }
        sequence = complex_type.children.find { |c| c.name == 'sequence' }
        sequence.xpath("#{schema_namespace}:element")
      end
    end

    # @param [Nokogiri::XML::Element]
    def document_type_for(element, depth = 1)
      # raise "Too far deep for #{element}" unless depth < 6
      return unless depth < 6
      if complex_type? element
        complex_type = element.children.find { |c| c.name == 'complexType' }
        sequence = complex_type.children.find { |c| c.name == 'sequence' }
        sequence.children.select { |node| node.class == Nokogiri::XML::Element }.each do |sub_element|
          document_type_for sub_element, depth += 1
        end
      else
        return "No type seen for #{element}, #{element.class}" unless element['type']
        name = element['name']
        type = value_after_namespace(element['type'])
        @use_camel_case = true if (/[[:upper:]]/.match(name[0]) != nil)
        spaces = '  ' * depth
        @content += "#{spaces}#{name.snakecase}: #{fill_in_field_from_type(type)} # #{type} \n"
      end
    end

    # Makes a yaml string in a '@content' instance variable
    # @param [Nokogiri::XML::NodeSet] list List
    def wsdl_to_yaml_for(list)
      raise "'@content' string must be set" if @content.nil?
      list.each do |element|
        document_type_for element
      end
    end

    # Prompt user for wsdl
    def ask_wsdl
      prompt = <<-EOF
Enter WSDL:
      EOF
      print prompt.chop
      @wsdl = $stdin.gets.strip
      puts
    end

    # Prompt user for Service name for wsdl
    def name_of_wsdl
      prompt = <<-EOF
Enter what you would like to name WSDL (CamelCase):
      EOF
      print prompt.chop
      @name = $stdin.gets.strip
      puts
    end

    # Prompt user to enter basic auth details
    def enter_auth_details
      prompt = <<-EOF
User Name:
      EOF
      print prompt.chop
      @auth_name = $stdin.gets.strip
      puts

      prompt = <<-EOF
User Password:
      EOF
      print prompt.chop
      @auth_password = $stdin.gets.strip
      puts
    end
  end
end