const cheerio = require("cheerio");
const axios = require("axios");
const fs = require("fs");
const request = require("request");
const path = require('path');
// 图片存储位置
const fileImgUrl = "./images/";
// 获取HTML文档的内容，内容的获取跟jQuery一样
let httpUrl = "https://www.pkdoutu.com/";
axios.get(httpUrl).then(res => {
    // 判断文件夹是否存在，不存在则创建
    console.log(fs.existsSync(fileImgUrl), "文件是否存在")
    if (!fs.existsSync(fileImgUrl)) {
        fs.mkdir(fileImgUrl, { recursive: true }, (err) => {
            if (err) {
                throw err;
            } else {

            }
        });
    }

    let $ = cheerio.load(res.data);
    $('.random_article>.col-sm-3>img').each((index, element) => {
        console.log($(element).attr("data-original"));
        let imgUrl = $(element).attr("data-original");
        let imgOrgin = imgUrl.split('/')[imgUrl.split('/').length - 1];
        axios.get(imgUrl, {
            // 设置返回的类型是二进制
            responseType: 'arraybuffer',
        }).then(res => {
            // 下载当前图片文件到文件夹中
            fs.writeFile(`${fileImgUrl}/${imgOrgin}`, res.data, 'binary', (err) => {
                if (err) {
                    console.log(`${index} error`, err);
                } else {
                    console.log(`${index} 下载成功`);

                }
            })
        });


    })
})