import cheerio from 'cheerio'
import got from 'got'

export async function wallpaper (query: string): Promise<string[]> {
  const data = await got(`https://www.shutterstock.com/search/${query}`, {
    headers: {
      accept:
				'text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.9',
      'accept-encoding': 'gzip, deflate, br',
      'accept-language': 'en-US,en;q=0.9,id;q=0.8',
      // cookie: '_csrf=DLixL776iH1Yv7Ck9wHekk24; _ga=GA1.2.1481444664.1639216586; _gid=GA1.2.348540858.1639216586; _gat=1; _hjFirstSeen=1; _hjSession_2571802=eyJpZCI6ImVkZDUzMWJhLWNjYTgtNDgyMy1hZmUyLWVjNmFhNWMxZjg3ZCIsImNyZWF0ZWQiOjE2MzkyMTY1ODY0Nzl9; _hjAbsoluteSessionInProgress=0; _hjSessionUser_2571802=eyJpZCI6IjIxZGNhYTc5LWRlMTgtNWE5Ni05ZWE2LTdkYjg4NGZhNjIxMSIsImNyZWF0ZWQiOjE2MzkyMTY1ODYyNDMsImV4aXN0aW5nIjp0cnVlfQ==',
      'User-Agent':
				'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/95.0.4638.69 Safari/537.36'
    }
  }).text()
  const $ = cheerio.load(data)
  const results: string[] = [
    ...new Set(
      [
        ...$.html().matchAll(
          /https?:\/\/(image|www)\.shutterstock\.com\/([^"]+)/gim
        )
      ]
        .map((v) => v[0])
        .filter((v) => /.*\.jpe?g|png$/gi.test(v))
    )
  ]
  return results
}

type Ioptionsv2 = { page: number; is4K?: boolean };
export async function wallpaperv2 (
  query: string,
  { page, is4K }: Ioptionsv2 = { page: 1 }
): Promise<string[]> {
  page = page < 2 ? 2 : page
  const data = await got(
		`https://wall.alphacoders.com/by_category.php?id=3&name=${encodeURIComponent(
			query
		).replace(/%20/g, '+')}&quickload=50&page=${page}${is4K ? '&filter=4K+Ultra+HD' : ''
		}`
  ).text()
  const $ = cheerio.load(data)
  const results: string[] = []
  $('div.thumb-container-big').each(function () {
    const img = $(this).find('picture > img').attr('src')
    if (img) results.push(img)
  })
  return results
}

export async function wallpaperv3 (query: string, page: number = 1): Promise<string[]> {
  const html = await got(`https://www.hdwallpapers.in/search/page/${page}?q=${encodeURIComponent(query)}`).text()
  const results: string[] = []
  const $ = cheerio.load(html)
  $('#content > div.page-content.wallpaper > ul > li.wall').each(function () {
    const img = $(this).find('a > img[src]').attr('src')
    if (img) results.push('https://www.hdwallpapers.in' + img)
  })
  return results
}