se-scraper/examples/pluggable.js

module.exports = class Pluggable {
    constructor(options = {}) {
        const {
            chromeFlags = [
                '--no-sandbox',
                '--disable-setuid-sandbox',
                '--disable-dev-shm-usage',
                '--disable-accelerated-2d-canvas',
                '--disable-gpu',
                '--window-size=1920x1080',
                '--hide-scrollbars',
                '--user-agent=Chrome',
            ],
            userAgent = 'Chrome',
            headless = true,
        } = options;

        this.chromeFlags = chromeFlags;
        this.userAgent = userAgent;
        this.headless = headless;

        this.chromeFlags.push(this.userAgent);
    }

    async close_browser() {
        await this.browser.close();
    }

    // Callback invoked after metadata has been gathered
    async handle_metadata(args) {
        // store scraping metadata somewhere
    }

    // Callback invoked after all keywords have been scraped
    async handle_results(args) {
        // store the results somewhere
    }

    // Callback invoked before a keyword is scraped.
    async before_keyword_scraped(args) {
        console.log('before keyword scraped.');
    }

    // Callback invoked after a keyword has been scraped.
    // TODO: implement this
    async after_keyword_scraped(args) {
        console.log('after keyword scraped.')
    }

    async start_browser(args={}) {
        const puppeteer = require('puppeteer');

        let launch_args = {
            args: args.chromeFlags || this.chromeFlags,
            headless: args.headless,
        };

        if (launch_args.headless === undefined) {
            launch_args.headless = this.headless;
        }

        this.browser = await puppeteer.launch(launch_args);
        console.log('Loaded custom function get_browser()');
        console.log(launch_args);

        return this.browser;
    }
};