// 引入一些需要用到的庫以及一些聲明
import * as puppeteer from 'puppeteer' // 引入Puppeteer
import mongo from '../lib/mongoDb' // 需要用到的 mongodb庫，用來存取爬取的數(shù)據(jù)
import chalk from 'chalk' // 一個美化 console 輸出的庫

const log = console.log // 縮寫 console.log
const TOTAL_PAGE = 50 // 定義需要爬取的網(wǎng)頁數(shù)量，對應(yīng)頁面下部的跳轉(zhuǎn)鏈接

// 定義要爬去的數(shù)據(jù)結(jié)構(gòu)
interface IWriteData { 
 link: string // 爬取到的商品詳情鏈接
 picture: string // 爬取到的圖片鏈接
 price: number // 價格，number類型，需要從爬取下來的數(shù)據(jù)進(jìn)行轉(zhuǎn)型
 title: string // 爬取到的商品標(biāo)題
}

// 格式化的進(jìn)度輸出 用來顯示當(dāng)前爬取的進(jìn)度
function formatProgress (current: number): string { 
 let percent = (current / TOTAL_PAGE) * 100
 let done = ~~(current / TOTAL_PAGE * 40)
 let left = 40 - done
 let str = `當(dāng)前進(jìn)度：[${''.padStart(done, '=')}${''.padStart(left, '-')}]  ${percent}%`
 return str
}

// 因為我們需要用到大量的 await 語句，因此在外層包裹一個 async function
async function main() {
 // Do something
}
main()

// 進(jìn)入代碼的主邏輯
async function main() {
 // 首先通過Puppeteer啟動一個瀏覽器環(huán)境
 const browser = await puppeteer.launch()
 log(chalk.green('服務(wù)正常啟動'))
 // 使用 try catch 捕獲異步中的錯誤進(jìn)行統(tǒng)一的錯誤處理
 try {
  // 打開一個新的頁面
  const page = await browser.newPage()
  // 監(jiān)聽頁面內(nèi)部的console消息
  page.on('console', msg => {
   if (typeof msg === 'object') {
    console.dir(msg)
   } else {
    log(chalk.blue(msg))
   }
  })

  // 打開我們剛剛看見的淘寶頁面
  await page.goto('https://s.taobao.com/search?q=gtx1080&imgfile=&js=1&stats_click=search_radio_all%3A1&initiative_id=staobaoz_20180416&ie=utf8')
  log(chalk.yellow('頁面初次加載完畢'))

  // 使用一個 for await 循環(huán)，不能一個時間打開多個網(wǎng)絡(luò)請求，這樣容易因為內(nèi)存過大而掛掉
  for (let i = 1; i <= TOTAL_PAGE; i++) {
   // 找到分頁的輸入框以及跳轉(zhuǎn)按鈕
   const pageInput = await page.$(`.J_Input[type='number']`)
   const submit = await page.$('.J_Submit')
   // 模擬輸入要跳轉(zhuǎn)的頁數(shù)
   await pageInput.type('' + i)
   // 模擬點擊跳轉(zhuǎn)
   await submit.click()
   // 等待頁面加載完畢，這里設(shè)置的是固定的時間間隔，之前使用過page.waitForNavigation()，但是因為等待的時間過久導(dǎo)致報錯（Puppeteer默認(rèn)的請求超時是30s,可以修改）,因為這個頁面總有一些不需要的資源要加載，而我的網(wǎng)絡(luò)最近日了狗，會導(dǎo)致超時，因此我設(shè)定等待2.5s就夠了
   await page.waitFor(2500)

   // 清除當(dāng)前的控制臺信息
   console.clear()
   // 打印當(dāng)前的爬取進(jìn)度
   log(chalk.yellow(formatProgress(i)))
   log(chalk.yellow('頁面數(shù)據(jù)加載完畢'))

   // 處理數(shù)據(jù)，這個函數(shù)的實現(xiàn)在下面
   await handleData()
   // 一個頁面爬取完畢以后稍微歇歇，不然太快淘寶會把你當(dāng)成機(jī)器人彈出驗證碼（雖然我們本來就是機(jī)器人）
   await page.waitFor(2500)
  }

  // 所有的數(shù)據(jù)爬取完畢后關(guān)閉瀏覽器
  await browser.close()
  log(chalk.green('服務(wù)正常結(jié)束'))

  // 這是一個在內(nèi)部聲明的函數(shù)，之所以在內(nèi)部聲明而不是外部，是因為在內(nèi)部可以獲取相關(guān)的上下文信息，如果在外部聲明我還要傳入 page 這個對象
  async function handleData() {
   // 現(xiàn)在我們進(jìn)入瀏覽器內(nèi)部搞些事情，通過page.evaluate方法，該方法的參數(shù)是一個函數(shù)，這個函數(shù)將會在頁面內(nèi)部運行，這個函數(shù)的返回的數(shù)據(jù)將會以Promise的形式返回到外部 
   const list = await page.evaluate(() => {
    
    // 先聲明一個用于存儲爬取數(shù)據(jù)的數(shù)組
    const writeDataList: IWriteData[] = []

    // 獲取到所有的商品元素
    let itemList = document.querySelectorAll('.item.J_MouserOnverReq')
    // 遍歷每一個元素，整理需要爬取的數(shù)據(jù)
    for (let item of itemList) {
     // 首先聲明一個爬取的數(shù)據(jù)結(jié)構(gòu)
     let writeData: IWriteData = {
      picture: undefined,
      link: undefined,
      title: undefined,
      price: undefined
     }

     // 找到商品圖片的地址
     let img = item.querySelector('img')
     writeData.picture = img.src

     // 找到商品的鏈接
     let link: HTMLAnchorElement = item.querySelector('.pic-link.J_ClickStat.J_ItemPicA')
     writeData.link = link.href

     // 找到商品的價格，默認(rèn)是string類型 通過~~轉(zhuǎn)換為整數(shù)number類型
     let price = item.querySelector('strong')
     writeData.price = ~~price.innerText
     
     // 找到商品的標(biāo)題，淘寶的商品標(biāo)題有高亮效果，里面有很多的span標(biāo)簽，不過一樣可以通過innerText獲取文本信息
     let title: HTMLAnchorElement = item.querySelector('.title>a')
 
     writeData.title = title.innerText

     // 將這個標(biāo)簽頁的數(shù)據(jù)push進(jìn)剛才聲明的結(jié)果數(shù)組
     writeDataList.push(writeData)
    }
    // 當(dāng)前頁面所有的返回給外部環(huán)境
    return writeDataList
    
   })
   // 得到數(shù)據(jù)以后寫入到mongodb
   const result = await mongo.insertMany('GTX1080', list)

   log(chalk.yellow('寫入數(shù)據(jù)庫完畢'))
  }

 } catch (error) {
  // 出現(xiàn)任何錯誤，打印錯誤消息并且關(guān)閉瀏覽器
  console.log(error)
  log(chalk.red('服務(wù)意外終止'))
  await browser.close()
 } finally {
  // 最后要退出進(jìn)程
  process.exit(0)
 }
}