import axios from 'axios';
import get from 'lodash/get';
import low from 'lowdb';
import FileAsync from 'lowdb/adapters/FileAsync';

const adapter = new FileAsync('db.json');

const xhr = axios.create({
  headers: {
    Accept: 'application/json, text/plain, */*',
    Referer: 'https://www.leafly.com/explore',
  },
});

const getPage = async num => {
  const url = `https://www.leafly.com/explore/page-${num}`;
  const response = await xhr.get(url, {
    responseType: 'json',
  });

  // transform strain data
  const strains = response.data.Model.Strains.filter(strain => strain.Name != null).map(strain => ({
    id: parseInt(strain.Id, 10),
    name: strain.Name,
    symbol: strain.Symbol,
    category: get(strain, 'Category', 'unknown').toLowerCase(),
    profile: get(strain, 'CannabinoidProfile', 'none').toLowerCase(),
    rating: strain.Rating,
    rating_count: strain.RatingCount,
    effects: get(strain, 'Tags', []).map(tag => tag.DisplayLabel.toLowerCase()),
    negative_effects: get(strain, 'NegativeEffects', []).map(tag => tag.DisplayLabel.toLowerCase()),
    flavors: get(strain, 'Flavors', []).map(tag => tag.DisplayLabel.toLowerCase()),
    uses: get(strain, 'Symptoms', []).map(tag => tag.DisplayLabel.toLowerCase()),
    conditions: get(strain, 'Conditions', []).map(tag => tag.DisplayLabel.toLowerCase()),
  }));

  return {
    strains,
    page: response.data.Model.PagingContext,
  };
};

export default async function scrapeLeafly(startFrom = 1, endAt = Infinity) {
  let pageNum = startFrom;
  let finished = false;
  const db = await low(adapter);

  await db.defaults({ strains: [] }).write();

  while (!finished) {
    console.log(`Fetching page ${pageNum}`);
    const data = await getPage(pageNum);

    data.strains.forEach(async strain => {
      // check for value
      const doc = db
        .get('strains')
        .filter({ id: strain.id })
        .first()
        .value();

      if (!doc) {
        console.log(`Adding ${strain.id}, ${strain.name}`);
        await db
          .get('strains')
          .push(strain)
          .write();
      }
    });

    if (pageNum >= endAt || !data.strains.length || data.page.isLastPage) finished = true;
    pageNum += 1;
  }
}