/* Imports */
import fs from 'fs'
import axios from "axios"

/* Definitions */
let htmlEndPoint = 'https://www.gjopen.com/questions?page='

/* Support functions */

function getcookie(){
  try {
    let rawcookie = fs.readFileSync("./src/privatekeys.json")
    let cookie = JSON.parse(rawcookie).goodjudmentopencookie
    if(cookie == undefined){
      throw new Error('No cookie for Good Judgment Open!');
    }
    
    return cookie
  } catch(error) {
    console.log("Error: No cookies for Good Judgment Open on src/privatekeys.json! See the README.md")
    process.exit()
  }
}


async function fetchPage(page, cookie){
  let response  = await axios({
    url: htmlEndPoint+page,
    method: 'GET',
    headers: ({ 
    'Content-Type': 'text/html',
    'Cookie': cookie
    }),
  })
  .then(res => res.data)
  //console.log(response)
  return response
}

async function fetchStats(questionUrl, cookie){
  let response  = await axios({
    url: questionUrl+"/stats",
    method: 'GET',
    headers: ({ 
    'Content-Type': 'text/html',
    'Cookie': cookie,
    'Referer': questionUrl,
    }),
  })
  .then(res => res.data)
  //console.log(response)
  
  // Is binary?
  let isbinary = response.includes("binary?":true")

  let percentage = "none"
  if(isbinary){
    // Crowd percentage
    let htmlElements = response.split("\n")
    let h3Element = htmlElements.filter(str => str.includes("h3"))[0]
    let crowdpercentage = h3Element.split(">")[1].split("<")[0]
    percentage = crowdpercentage
  }

  // Number of forecasts
  let numforecasts = response.split("prediction_sets_count&quot;:")[1].split(",")[0]
  //console.log(numforecasts)
  
  // Number of predictors
  let numforecasters = response.split("predictors_count&quot;:")[1].split(",")[0]
  //console.log(numpredictors)
  
  let result = {
    "Binary question?": isbinary,
    "Percentage": percentage,
    "# Forecasts": numforecasts,
    "# Forecasters": numforecasters
  }
  return result
}

function isEnd(html){
  return html.includes("No questions match your filter")
}

function sleep(ms) {
  return new Promise(resolve => setTimeout(resolve, ms));
}

/* Body */

export async function goodjudgmentopen(){
  let cookie = getcookie()

  let i=1
  let response = await fetchPage(i, cookie)
  let questions = []
  let init = Date.now()
  console.log("Downloading... This might take a couple of minutes. Results will be shown.")
  while(!isEnd(response)){
    console.log(`Page #${i}`)
    let htmlLines = response.split("\n")
    let h5elements = htmlLines.filter(str => str.includes("<h5><a href=")) 
    for(let h5element of h5elements){
      let h5elementSplit = h5element.split('"><span>')
      let url = h5elementSplit[0].split('<a href="')[1]
      let title = h5elementSplit[1].replace('</span></a></h5>', "")
      await sleep(1000 + Math.random()*1000) // don't be as noticeable
      try{
        let moreinfo = await fetchStats(url, cookie)
        if(moreinfo.isbinary){
          if(!moreinfo.crowdpercentage){ // then request again.
            moreinfo = await fetchStats(url, cookie)
          }
        }
        let question = ({
            "Title": title,
            "URL": url,
            "Platform": "Good Judgment Open",
            ...moreinfo
          })
          console.log(question)
          questions.push(question)
      } catch(error){
        console.log(`We encountered some error when fetching the URL: ${url}, so it won't appear on the final json`)
      }
    }
    i=i+1
    console.log("Sleeping for 5secs so as to not be as noticeable to the gjopen servers")
    await sleep(5000 + Math.random()*1000) // don't be a dick to gjopen server
    
    try{
      response = await fetchPage(i,cookie)
    }catch(error){
      console.log(`We encountered some error when fetching page #${i}, so it won't appear on the final json`)
    }
  }
  let string = JSON.stringify(questions,null,  2)
  fs.writeFileSync('./data/goodjudmentopen-questions.json', string);
  
  let end = Date.now()
  let difference = end-init
  console.log(`Took ${difference/1000} seconds, or ${difference/(1000*60)} minutes.`)
}