javascript 从puppeteer导出所有用户/会话数据

Posted 2021-05-03

async getChromeUserData() {
    assert(this.userDataDir, 'userDataDir is empty');

    try {
      const buf = await streamToBuffer(await tar.c({
        gzip: true,
        cwd: this.userDataDir,
        filter(path) {
          if (/BrowserMetrics-active\.pma$/i.test(path)) return false;
          if (/\/Favicons(-journal)?$/.test(path)) return false;
          if (/\/History(-journal)?$/.test(path)) return false;
          if (/\/Top Sites(-journal)?$/.test(path)) return false;
          if (/\/Last Session$/.test(path)) return false;
          if (/\/Last Tabs$/.test(path)) return false;
          if (/\/Visited Links$/.test(path)) return false;
          if (/\/ShaderCache\//.test(path)) return false;
          if (/\/ShaderCache$/.test(path)) return false;
          if (/\/GPUCache\//.test(path)) return false;
          if (/\/GPUCache$/.test(path)) return false;
          return true;
        },
      }, ['.']));

      return buf;
    } catch (error) {
      this.logger.error('error-getChromeUserData', {
        error,
      });
      return Buffer.from('');
    }
  }
  
  
  
  async export() {
    assert(this.page, 'page is empty');

    // eslint-disable-next-line no-underscore-dangle
    const { cookies } = (await this.page._client.send('Network.getAllCookies'));

    return new Promise(async (resolve, reject) => {
      try {
        return zlib.gzip(JSON.stringify({
          chromeUserData: (await this.getChromeUserData()).toString('base64'),
          puppeteerCookies: cookies,
          cookies: cookies.map(c => ({
            domain: c.domain,
            expires: new Date((c.expires || Date.now() / 1000) * 1000).toGMTString(),
            expiry: c.expires,
            httponly: c.httpOnly,
            name: c.name,
            path: c.path,
            secure: c.secure,
            value: c.value,
          })),
        }), (err, buf) => {
          if (err) return reject(err);
          return resolve(buf.toString('base64'));
        });
      } catch (error) {
        return reject(error);
      }
    });
  }
}