beizhedenglong / leetcode-site-generator Goto Github PK

const baseUrl = 'https://leetcode-cn.com';
const graphqlUrl = `${baseUrl}/graphql`;

const login = async () => {
  const browser = await puppeteer.launch({ headless: false });
  const page = await browser.newPage();
  await page.goto('https://leetcode-cn.com/accounts/login/');
  await page.waitForNavigation({
    timeout: 0,
  });
  const cookies = await page.cookies();
  await browser.close();
  return cookies.reduce((acc, cookie) => {
    const { name } = cookie;
    acc[name] = cookie;
    return acc;
  }, {});
};

const getQuestionData = async (titleSlug) => {
  const qglRequest = await createGqlRequest();
  // interface from 'https://leetcode-cn.com/problems/add-two-numbers/'
  const json = await qglRequest(`{
    questionData($titleSlug: "${titleSlug}"){
      question{
        topicTags
      }
    }
  }`);
  const question = (json.questionData && json.questionData.question) || {};
  return question;
};