src/methods.ts

import Prray from './prray'

export async function mapAsync<T, U>(
  arr: Prray<T>,
  func: (currentValue: T, index: number, array: Prray<T>) => Promise<U> | U,
  opts = { concurrency: Infinity },
): Promise<U[]> {
  const result: U[] = []
  await loop<T>(arr, async (value, ix) => (result[ix] = await func(value, ix, arr)), opts)
  return result
}

export function map<T, U>(prr: Prray<T>, func: (currentValue: T, index: number, array: Prray<T>) => U): U[] {
  const result: U[] = []
  for (let ix = 0; ix < prr.length; ix++) {
    result.push(func(prr[ix], ix, prr))
  }
  return result
}

export async function filterAsync<T>(
  arr: Prray<T>,
  func: (currentValue: T, index: number, array: Prray<T>) => Promise<boolean> | boolean,
  opts = { concurrency: Infinity },
): Promise<T[]> {
  const result: T[] = []
  await loop(arr, async (value, ix) => ((await func(value, ix, arr)) ? result.push(value) : null), opts)
  return result
}

export function filter<T>(prr: Prray<T>, func: (currentValue: T, index: number, array: Prray<T>) => boolean): T[] {
  const result: T[] = []
  for (let ix = 0; ix < prr.length; ix++) {
    const value = prr[ix]
    if (func(value, ix, prr)) {
      result.push(value)
    }
  }
  return result
}

export async function reduceAsync<T>(
  prr: Prray<T>,
  func: (accumulator: T, currentValue: T, index: number, prray: Prray<T>) => T | Promise<T>,
): Promise<T>
export async function reduceAsync<T>(
  prr: Prray<T>,
  func: (accumulator: T, currentValue: T, index: number, prray: Prray<T>) => T | Promise<T>,
  initialValue: T,
): Promise<T>
export async function reduceAsync<T, U>(
  prr: Prray<T>,
  func: (accumulator: U, currentValue: T, index: number, prray: Prray<T>) => U | Promise<U>,
  initialValue: U,
): Promise<U>
export async function reduceAsync<T>(
  prr: Prray<T>,
  func: (accumulator: any, currentValue: T, index: number, prray: Prray<T>) => any | Promise<any>,
  initialValue?: any,
): Promise<any> {
  if (prr.length === 0) {
    if (initialValue === undefined) {
      throw TypeError('Reduce of empty array with no initial value')
    } else {
      return initialValue
    }
  }
  let pre = initialValue
  let ix = 0
  if (initialValue === undefined) {
    pre = prr[0]
    ix = 1
  }
  for (ix; ix < prr.length; ix++) {
    const current = prr[ix]
    pre = await func(pre, current, ix, prr)
  }
  return pre
}

export function reduce<T>(
  prr: Prray<T>,
  func: (accumulator: T, currentValue: T, index: number, array: Prray<T>) => T,
): T
export function reduce<T>(
  prr: Prray<T>,
  func: (accumulator: T, currentValue: T, index: number, array: Prray<T>) => T,
  initialValue: T,
): T
export function reduce<T, U>(
  prr: Prray<T>,
  func: (accumulator: U, currentValue: T, index: number, array: Prray<T>) => U,
  initialValue: U,
): U
export function reduce<T>(
  prr: Prray<T>,
  func: (accumulator: any, currentValue: T, index: number, array: Prray<T>) => any,
  initialValue?: any,
): any {
  if (prr.length === 0) {
    if (initialValue === undefined) {
      throw TypeError('Reduce of empty array with no initial value')
    } else {
      return initialValue
    }
  }
  let pre = initialValue
  let ix = 0
  if (initialValue === undefined) {
    pre = prr[0]
    ix = 1
  }
  for (ix; ix < prr.length; ix++) {
    const current = prr[ix]
    pre = func(pre, current, ix, prr)
  }
  return pre
}

export async function reduceRightAsync<T>(
  prr: Prray<T>,
  func: (accumulator: T, currentValue: T, index: number, prray: Prray<T>) => T | Promise<T>,
): Promise<T>
export async function reduceRightAsync<T>(
  prr: Prray<T>,
  func: (accumulator: T, currentValue: T, index: number, prray: Prray<T>) => T | Promise<T>,
  initialValue: T,
): Promise<T>
export async function reduceRightAsync<T, U>(
  prr: Prray<T>,
  func: (accumulator: U, currentValue: T, index: number, prray: Prray<T>) => U | Promise<U>,
  initialValue: U,
): Promise<U>
export async function reduceRightAsync<T>(
  prr: Prray<T>,
  func: (accumulator: any, currentValue: T, index: number, prray: Prray<T>) => any | Promise<any>,
  initialValue?: any,
): Promise<any> {
  if (prr.length === 0) {
    if (initialValue === undefined) {
      throw TypeError('Reduce of empty array with no initial value')
    } else {
      return initialValue
    }
  }
  let pre = initialValue
  let ix = prr.length - 1
  if (initialValue === undefined) {
    pre = prr[prr.length - 1]
    ix = prr.length - 2
  }
  for (ix; ix >= 0; ix--) {
    const current = prr[ix]
    pre = await func(pre, current, ix, prr)
  }
  return pre
}

export function reduceRight<T>(
  prr: Prray<T>,
  func: (accumulator: T, currentValue: T, index: number, array: Prray<T>) => T,
): T
export function reduceRight<T>(
  prr: Prray<T>,
  func: (accumulator: T, currentValue: T, index: number, array: Prray<T>) => T,
  initialValue: T,
): T
export function reduceRight<T, U>(
  prr: Prray<T>,
  func: (accumulator: U, currentValue: T, index: number, array: Prray<T>) => U,
  initialValue: U,
): U
export function reduceRight<T>(
  prr: Prray<T>,
  func: (accumulator: any, currentValue: T, index: number, array: Prray<T>) => any,
  initialValue?: any,
): any {
  if (prr.length === 0) {
    if (initialValue === undefined) {
      throw TypeError('Reduce of empty array with no initial value')
    } else {
      return initialValue
    }
  }
  let pre = initialValue
  let ix = prr.length - 1
  if (initialValue === undefined) {
    pre = prr[prr.length - 1]
    ix = prr.length - 2
  }
  for (ix; ix >= 0; ix--) {
    const current = prr[ix]
    pre = func(pre, current, ix, prr)
  }
  return pre
}

export async function findIndexAsync<T>(
  arr: Prray<T>,
  func: (currentValue: T, index: number, array: Prray<T>) => Promise<boolean> | boolean,
): Promise<number> {
  let result = -1
  await loop(
    arr,
    async (value, ix, _, breakLoop) => {
      if (await func(value, ix, arr)) {
        result = ix
        breakLoop()
      }
    },
    {},
  )
  return result
}

export function findIndex<T>(
  prr: Prray<T>,
  func: (currentValue: T, index: number, array: Prray<T>) => boolean,
): number {
  const length = prr.length
  for (let ix = 0; ix < length; ix++) {
    if (func(prr[ix], ix, prr)) {
      return ix
    }
  }
  return -1
}

export async function findAsync<T>(
  arr: Prray<T>,
  func: (currentValue: T, index: number, array: Prray<T>) => Promise<boolean> | boolean,
): Promise<T | undefined> {
  let result: T | undefined
  await loop(
    arr,
    async (value, ix, _, breakLoop) => {
      if (await func(value, ix, arr)) {
        result = value
        breakLoop()
      }
    },
    {},
  )
  return result
}

export function find<T>(
  prr: Prray<T>,
  func: (currentValue: T, index: number, array: Prray<T>) => boolean,
): T | undefined {
  const length = prr.length
  for (let ix = 0; ix < length; ix++) {
    if (func(prr[ix], ix, prr)) {
      return prr[ix]
    }
  }
  return undefined
}

export async function everyAsync<T>(
  prr: Prray<T>,
  func: (currentValue: T, index: number, prray: Prray<T>) => Promise<boolean> | boolean,
  opts = { concurrency: Infinity },
): Promise<boolean> {
  let result = true
  await loop(
    prr,
    async (value, ix, _, breakLoop) => {
      if (!(await func(value, ix, prr))) {
        result = false
        breakLoop()
      }
    },
    opts,
  )
  return result
}

export function every<T>(prr: Prray<T>, func: (currentValue: T, index: number, prray: Prray<T>) => boolean): boolean {
  for (let ix = 0; ix < prr.length; ix++) {
    if (!func(prr[ix], ix, prr)) {
      return false
    }
  }
  return true
}

export async function someAsync<T>(
  prr: Prray<T>,
  func: (currentValue: T, index: number, prray: Prray<T>) => Promise<boolean> | boolean,
  opts = { concurrency: Infinity },
): Promise<boolean> {
  let result = false
  await loop(
    prr,
    async (value, ix, _, breakLoop) => {
      if (await func(value, ix, prr)) {
        result = true
        breakLoop()
      }
    },
    opts,
  )
  return result
}

export function some<T>(prr: Prray<T>, func: (currentValue: T, index: number, prray: Prray<T>) => boolean): boolean {
  for (let ix = 0; ix < prr.length; ix++) {
    if (func(prr[ix], ix, prr)) {
      return true
    }
  }
  return false
}

export async function sortAsync<T>(arr: Prray<T>, func?: (a: T, b: T) => Promise<number> | number): Promise<Prray<T>> {
  if (!func) {
    return arr.sort()
  }
  if (arr.length < 2) {
    return arr
  }
  // &#25554;&#20837;&#25490;&#24207;
  for (let i = 1; i < arr.length; i++) {
    for (let j = 0; j < i; j++) {
      if ((await func(arr[i], arr[j])) < 0) {
        arr.splice(j, 0, arr[i])
        arr.splice(i + 1, 1)
        break
      }
    }
  }
  return arr
}

export async function forEachAsync<T>(
  prr: Prray<T>,
  func: (currentValue: T, index: number, prray: Prray<T>) => Promise<any> | any,
  opts = { concurrency: Infinity },
): Promise<undefined> {
  await loop(prr, async (value, ix) => func(value, ix, prr), opts)
  return
}

export function forEach<T>(prr: Prray<T>, func: (currentValue: T, index: number, prray: Prray<T>) => any): undefined {
  for (let ix = 0; ix < prr.length; ix++) {
    func(prr[ix], ix, prr)
  }
  return
}

export function loop<T>(
  array: Prray<T>,
  func: (value: T, index: number, array: Prray<T>, breakLoop: () => any) => any,
  { concurrency = Infinity },
) {
  // FEATURE: options { concurrency, timeout, retries, defaults, fallback }

  if (array.length <= concurrency) {
    const promises = array.map((v, ix) => func(v, ix, array, () => null))
    return Promise.all(promises)
  }

  return new Promise<void>((resolve, reject) => {
    const length = array.length
    if (length === 0) {
      resolve()
    }

    let isEnding = false
    let currentIndex = 0
    let workingNum = Math.min(concurrency, length)

    const breakLoop = () => {
      isEnding = true
      resolve()
    }

    const woker = async () => {
      while (!isEnding && currentIndex < length) {
        const ix = currentIndex++
        try {
          await func(array[ix], ix, array, breakLoop)
        } catch (error) {
          isEnding = true
          reject(error)
          return
        }
      }
      workingNum--
      if (workingNum === 0) {
        resolve()
      }
    }

    for (let i = 0; i < Math.min(concurrency, length); i++) {
      woker()
    }
  })
}