module Seda

# File lib/seda.rb, line 29
def self.contains(collection, target)
  collection.each do |element|
    if element == target
      return true
    end
  end
  false
end

# File lib/seda.rb, line 39
def self.defaults(*obj)
  final_object = (eval(local_variables[0].to_s))[0]
  iterables = (eval(local_variables[0].to_s))[1..-1]

  if final_object.class == Hash && iterables.class == Array
    iterables.each do |element|
      if element.class == Hash
        for k in element
          if final_object[k[0].to_sym] == nil
            final_object[k[0].to_sym] = k[1]
          end
        end
      elsif element.class == Array
        element.each do |item|
          if final_object[item.to_sym] == nil
            final_object[item.to_sym] = item
          end
        end
      else
        if final_object[element.to_sym] == nil
          final_object[element.to_sym] = element
        end
      end
    end
  else
    raise ArgumentError.new('Must pass in the destination Hash as first argument, and additional argument objects to extend values from')
  end

  return final_object
end

# File lib/seda.rb, line 383
def self.delay(some_method, wait, *some_method_args)
  args = eval(local_variables[2].to_s)

  if some_method.class == String
    some_method = some_method.to_sym
    some_method = method(some_method)
  elsif some_method.class == Symbol
    some_method = method(some_method)
  end

  if some_method.class == Method
    sleep (wait.to_f/1000)
    some_method.call(*args)
  else
    raise ArgumentError.new('First argument must reference a method')
  end
end

# File lib/seda.rb, line 359
def self.difference(*arrays)
  checker = (eval(local_variables[0].to_s))[0]
  iterables = (eval(local_variables[0].to_s))[1..-1]

  arr = []
  seen = {}

  checker.each do |element|
    seen[element] = element
    arr << element
  end

  iterables.each do |inner_arr|
    inner_arr.each do |element|
      if seen[element]
        arr = arr.reject{|el| el == element}
      end
    end
  end

  arr
end

# File lib/seda.rb, line 402
def self.every(collection, some_method)
  collection.each do |item|
    if some_method.class == Method
      if !some_method.call(item)
        return false
      end
    elsif some_method.class == Symbol
      if !method(some_method).call(item)
        return false
      end
    elsif some_method.class == String
      if !method(some_method.to_sym).call(item)
        return false
      end
    else
      raise ArgumentError.new('First argument must reference a method')
    end
  end
  return true
end

# File lib/seda.rb, line 71
def self.extend(*obj)
  initial_obj = (eval(local_variables[0].to_s))[0]
  iterables = (eval(local_variables[0].to_s))[1..-1]
  final_object = {}

  if initial_obj.class == Hash
    initial_obj.each do |k, v|
      final_object[k.to_s] = v
    end

    if iterables.class == Array
      iterables.each do |element|
        if element.class == Hash
          for k in element
            final_object[k[0].to_s] = k[1]
          end
        elsif element.class == Array
          element.each do |item|
            final_object[item.to_s] = item
          end
        else
          final_object[element.to_s] = element
        end
      end
    end
  else
    raise ArgumentError.new('Must pass in the destination Hash as first argument, and additional argument objects to extend values from')
  end

  return final_object
end

# File lib/seda.rb, line 148
def self.filter(collection, some_method)
  arr = []
  result_obj = {}

  if collection.class == Array
    collection.each_with_index do |element, idx|
      if some_method.class == String
        if method(some_method.to_sym).call(collection, idx)
          arr << element
        end
      elsif some_method.class == Symbol
        if method(some_method).call(collection, idx)
          arr << element
        end
      elsif some_method.class == Method
        if some_method.call(collection, idx)
          arr << element
        end
      else
        raise ArgumentError.new('The second argument must be a method. You could pass in the actual method, a string of the method name, or a symbol of the method name. Both arguments are needed.')
      end
    end
    return arr
  elsif collection.class == Hash
    for k in collection
      if some_method.class == String
        if method(some_method.to_sym).call(k[1])
          result_obj[k[0]] = k[1]
        end
      elsif some_method.class == Symbol
        if method(some_method).call(k[1])
          result_obj[k[0]] = k[1]
        end
      elsif some_method.class == Method
        if some_method.call(k[1])
          result_obj[k[0]] = k[1]
        end
      else
        raise ArgumentError.new('The second argument must be a method. You could pass in the actual method, a string of the method name, or a symbol of the method name. Both arguments are needed.')
      end
    end
    return result_obj
  else
     raise ArgumentError.new('Must provide an Array, or a Hash as the first Argument.')
  end
end

# File lib/seda.rb, line 9
def self.first(arr, n_first_elements)
  n_first_elements == nil ? arr[0] : arr.slice(0, n_first_elements)
end

# File lib/seda.rb, line 424
def self.flatten(nested_array, result = [])
  nested_array.each do |element|
    if element.class == Array
      self.flatten(element, result)
    else
      result << element
    end
  end

  result
end

# File lib/seda.rb, line 5
def self.identity(value)
  value
end

# File lib/seda.rb, line 13
def self.index_of(arr, target)
  if(arr == nil)
    return arr
  end

  result = -1

  arr.each_with_index do |element, i|
    if(element == target && result == -1)
      result = i
    end
  end

  result
end

# File lib/seda.rb, line 103
def self.intersection(*arrays)
  result = []
  seen = {}

  iterables = (eval(local_variables[0].to_s))[0..-1]

  iterables.flatten.each do |element|
    seen[element] == nil ? seen[element] = 1 : seen[element] += 1
  end

  for k in seen
    if k[1] == iterables.length
      result << k[0]
    end
  end

  result
end

# File lib/seda.rb, line 123
def self.last(arr, n)
  if(n > arr.length)
    arr
  else
    n == nil ? arr[arr.length - 1] : arr.slice(arr.length - n, arr.length)
  end
end

# File lib/seda.rb, line 437
def self.map(collection, some_method)
  arr = []

  if collection.class == Array
    collection.each_with_index do |element, idx|
      if some_method.class == String
        arr << method(some_method.to_sym).call(element, idx, collection)
      elsif some_method.class == Symbol
        arr << method(some_method).call(element, idx, collection)
      elsif some_method.class == Method
        arr << some_method.call(element, idx, collection)
      else
        raise ArgumentError.new('The second argument must be a method. You could pass in the actual method, a string of the method name, or a symbol of the method name. Both arguments are needed.')
      end
    end
  elsif collection.class == Hash
    for k in collection
      if some_method.class == String
        arr << method(some_method.to_sym).call(k[1], k[0], collection)
      elsif some_method.class == Symbol
        arr << method(some_method).call(k[1], k[0], collection)
      elsif some_method.class == Method
        arr << some_method.call(k[1], k[0], collection)
      else
        raise ArgumentError.new('The second argument must be a method. You could pass in the actual method, a string of the method name, or a symbol of the method name. Both arguments are needed.')
      end
    end
  else
     raise ArgumentError.new('Must provide an Array, or a Hash as the first Argument.')
  end

  arr
end

# File lib/seda.rb, line 472
def self.memoize(some_method, *some_method_args)
  storage = {}
  result = ''
  b = eval(local_variables[1].to_s)

  if some_method.class == String
    some_method = method(some_method.to_sym)
  elsif some_method.class == Symbol
    some_method = method(some_method)
  end

  result = lambda{|a = b, arg = b.to_s|
    if !storage[arg]
      storage[arg] = some_method.call(*a)
    end

    return storage[arg]
  }

  return result.call(b)
end

# File lib/seda.rb, line 132
def self.pluck(collection, key)
  collection.map{|item| item[key]}
end

# File lib/seda.rb, line 245
def self.reduce(collection, some_method)
  accumulator = ''
  cloned_collection = Marshal.load(Marshal.dump(collection))
  counter = 0
  optional_result_object = cloned_collection.class.new()

  if cloned_collection.class == Array
    accumulator = cloned_collection[0]
    optional_result_object << accumulator

    cloned_collection.each_with_index do |element, idx|

      if counter == 0
        counter = 1
        next
      end

      if some_method.class == String
        args = method(some_method).parameters.map { |arg| arg[1].to_s }

        if args.length < 3
          accumulator = method(some_method.to_sym).call(accumulator, element)
          optional_result_object << accumulator
        else
          accumulator = method(some_method.to_sym).call(accumulator, element, idx)
          optional_result_object << accumulator
        end

      elsif some_method.class == Symbol
        args = method(some_method).parameters.map { |arg| arg[1].to_s }

        if args.length < 3
          accumulator = method(some_method).call(accumulator, element)
          optional_result_object << accumulator
        else
          accumulator = method(some_method).call(accumulator, element, idx)
          optional_result_object << accumulator
        end

      elsif some_method.class == Method
        args = some_method.parameters.map { |arg| arg[1].to_s }

        if args.length < 3
          accumulator = some_method.call(accumulator, element)
          optional_result_object << accumulator
        else
          accumulator = some_method.call(accumulator, element, idx)
          optional_result_object << accumulator
        end

      end

    end

  elsif cloned_collection.class == Hash
    accumulator = cloned_collection.values[0]

    for k in cloned_collection

      if counter == 0
        optional_result_object[k[0]] = accumulator
        counter = 1
        next
      else

        if some_method.class == String
          args = some_method.parameters.map { |arg| arg[1].to_s }

          if args.length < 3
            accumulator = method(some_method.to_sym).call(accumulator, k[1])
            optional_result_object[k[0]] = accumulator
          else
            accumulator = method(some_method.to_sym).call(accumulator, k[1], counter)
            optional_result_object[k[0]] = accumulator
          end
        elsif some_method.class == Symbol
          args = some_method.parameters.map { |arg| arg[1].to_s }

          if args.length < 3
            accumulator = method(some_method).call(accumulator, k[1])
            optional_result_object[k[0]] = accumulator
          else
            accumulator = method(some_method).call(accumulator, k[1], counter)
            optional_result_object[k[0]] = accumulator
          end
        elsif some_method.class == Method
          args = some_method.parameters.map { |arg| arg[1].to_s }

          if args.length < 3
            accumulator = some_method.call(accumulator, k[1])
            optional_result_object[k[0]] = accumulator
          else
            accumulator = some_method.call(accumulator, k[1], counter)
            optional_result_object[k[0]] = accumulator
          end
        end

      end
      counter = counter + 1

    end

  else
    return cloned_collection
  end

  if accumulator.class == Float || accumulator.class == Integer
    accumulator
  else
    cloned_collection
  end
end

# File lib/seda.rb, line 196
def self.reject(collection, some_method)
  arr = []
  result_obj = {}

  if collection.class == Array
    collection.each_with_index do |element, idx|
      if some_method.class == String
        if !method(some_method.to_sym).call(collection, idx)
          arr << element
        end
      elsif some_method.class == Symbol
        if !method(some_method).call(collection, idx)
          arr << element
        end
      elsif some_method.class == Method
        if !some_method.call(collection, idx)
          arr << element
        end
      else
        raise ArgumentError.new('The second argument must be a method. You could pass in the actual method, a string of the method name, or a symbol of the method name. Both arguments are needed.')
      end
    end

    return arr
  elsif collection.class == Hash
    for k in collection
      if some_method.class == String
        if !method(some_method.to_sym).call(k[1])
          result_obj[k[0]] = k[1]
        end
      elsif some_method.class == Symbol
        if !method(some_method).call(k[1])
          result_obj[k[0]] = k[1]
        end
      elsif some_method.class == Method
        if !some_method.call(k[1])
          result_obj[k[0]] = k[1]
        end
      else
        raise ArgumentError.new('The second argument must be a method. You could pass in the actual method, a string of the method name, or a symbol of the method name. Both arguments are needed.')
      end
    end
    return result_obj
  else
     raise ArgumentError.new('Must provide an Array, or a Hash as the first Argument.')
  end
end

# File lib/seda.rb, line 136
def self.shuffle(arr)
  counter = arr.length - 1

  while counter > 0
    random_index = rand(counter)
    arr[counter], arr[random_index] = arr[random_index], arr[counter]
    counter -= 1
  end

  arr
end

# File lib/seda.rb, line 495
def self.some(collection, some_method)
  result = true
  collection.each do |item|
    if some_method.class == Method
      result = some_method.call(item)
    elsif some_method.class == Symbol
      result = method(some_method).call(item)
    elsif some_method.class == String
      result = method(some_method.to_sym).call(item)
    else
      raise ArgumentError.new('First argument must reference a method')
    end
  end

  return result
end

# File lib/seda.rb, line 513
def self.sort_by_max(collection)
  if collection.class == Hash
    result = {}
    arr = collection.sort{|a, b| b <=> a }
    arr.flatten.uniq.inject({}) do |memo, item|
      result[item] = item
    end

    return result
  elsif collection.class == Array
    collection.sort!{|a, b| b <=> a }
  end
end

# File lib/seda.rb, line 527
def self.sort_by_min(collection)
  if collection.class == Hash
    result = {}
    arr = collection.sort{|a, b| a <=> b }
    arr.flatten.uniq.inject({}) do |memo, item|
      result[item] = item
    end

    return result
  elsif collection.class == Array
    collection.sort!{|a, b| a <=> b }
  end
end