discourse-custom-wizard-unlocked/lib/custom_wizard/mapper.rb

class CustomWizard::Mapper
  attr_accessor :inputs, :data, :user
  
  USER_FIELDS = ['name', 'username', 'email', 'date_of_birth', 'title', 'locale', 'trust_level']
  PROFILE_FIELDS = ['location', 'website', 'bio_raw']
  
  def self.user_fields
    USER_FIELDS + PROFILE_FIELDS
  end
  
  OPERATORS = {
    equal: '==',
    greater: '>',
    less: '<',
    greater_or_equal: '>=',
    less_or_equal: '<=',
    regex: '=~'
  }
 
  def initialize(params)
    @inputs = params[:inputs] || {}
    @data = params[:data] || {}
    @user = params[:user]
    @opts = params[:opts] || {}
  end
  
  def perform
    multiple = @opts[:multiple]
    perform_result = multiple ? [] : nil
    
    inputs.each do |input|
      input_type = input['type']
      pairs = input['pairs']

      if (input_type === 'conditional' && validate_pairs(pairs)) || input_type === 'assignment'
        output = input['output']
        output_type = input['output_type']
        
        result = build_result(map_field(output, output_type), input_type)
        
        if multiple
          perform_result.push(result)
        else
          perform_result = result
          break
        end
      end
      
      if input_type === 'validation'
        result = build_result(validate_pairs(pairs), input_type)
        
        if multiple
          perform_result.push(result)
        else
          perform_result = result
          break
        end
      end
      
      if input_type === 'association'
        result = build_result(map_pairs(pairs), input_type)
        
        if multiple
          perform_result.push(result)
        else
          perform_result = result
          break
        end
      end
    end
                  
    perform_result
  end
  
  def build_result(result, type)
    if @opts[:with_type]
      {
        type: type,
        result: result
      }
    else
      result
    end
  end
  
  def validate_pairs(pairs)
    failed = false
    
    pairs.each do |pair|
      key = map_field(pair['key'], pair['key_type'])
      connector = pair['connector']
      operator = map_operator(connector)
      value = cast_value(
        key,
        interpolate(map_field(pair['value'], pair['value_type'])),
        connector
      )
                        
      begin
        failed = !cast_result(key.public_send(operator, value), connector)
      rescue NoMethodError
        #
      end
    end
    
    !failed
  end
  
  def cast_value(key, value, connector)
    if connector == 'regex'
      Regexp.new(value)
    else
      if key.is_a?(String)
        value.to_s
      elsif key.is_a?(Integer)
        value.to_i
      else
        value
      end
    end
  end
  
  def cast_result(result, connector)
    if connector == 'regex'
      result == 0 ? true : false
    else
      result
    end
  end
  
  def map_pairs(pairs)
    result = []
    
    pairs.each do |pair|
      key = map_field(pair['key'], pair['key_type'])
      value = map_field(pair['value'], pair['value_type'])
      
      if key && value
        result.push(
          key: key,
          value: value
        )
      end
    end
    
    result
  end
  
  def map_operator(connector)
    OPERATORS[connector.to_sym] || '=='
  end
  
  def map_field(value, type)
    method = "map_#{type}"
  
    if self.respond_to?(method)
      self.send(method, value)
    else
      value
    end
  end
  
  def map_wizard_field(value)
    data && !data.key?("submitted_at") && data[value]
  end

  def map_user_field(value)
    if value.include?('user_field_')
      UserCustomField.where(user_id: user.id, name: value).pluck(:value).first
    elsif PROFILE_FIELDS.include?(value)
      UserProfile.find_by(user_id: user.id).send(value)
    elsif USER_FIELDS.include?(value)
      User.find(user.id).send(value)
    end
  end
  
  def interpolate(string)
    result = string.gsub(/u\{(.*?)\}/) do |match|
      result = ''
      result = user.send($1) if USER_FIELDS.include?($1)
      result = user.user_profile.send($1) if PROFILE_FIELDS.include?($1)
      result
    end

    result = result.gsub(/w\{(.*?)\}/) { |match| recurse(data, [*$1.split('.')]) }
    
    result.gsub(/v\{(.*?)\}/) do |match|
      attrs = $1.split(':')
      key = attrs.first
      format = attrs.length > 1 ? attrs.last : nil
      val = nil
      
      if key == 'time'
        time_format = format.present? ? format : "%B %-d, %Y"
        val = Time.now.strftime(time_format)
      end

      val
    end
  end
  
  def recurse(data, keys)
    k = keys.shift
    result = data[k]
    keys.empty? ? result : self.recurse(result, keys)
  end
end