import os
import requests
from util import *

class AuthorStub(object):

    def __init__(self, **kwargs):
        self.authorId = kwargs["authorId"]
        self.name = kwargs.get("name", None)
        self.url = kwargs.get("url", None)

    def __str__(self):
        return self.authorId

    def __eq__(self, other):
        return isinstance(other, AuthorStub) and self.authorId == other.authorId

    def __hash__(self):
        return hash(self.authorId)

    def json(self):
        return {
            "authorId" : self.authorId,
            "name" : self.name,
            "url" : self.url
        }

    def full(self, **kwargs):
        return SemanticScholarAPI.author(self.authorId, **kwargs)

class Author(object):

    def __init__(self, **kwargs):
        self._kwargs = kwargs
        self.authorId = kwargs["authorId"]
        self.name = kwargs.get("name", None)
        self.aliases = kwargs.get("aliases", [])
        self.citationVelocity = kwargs.get("citationVelocity", None)
        self.influentialCitationCount = kwargs.get("influentialCitationCount", None)
        self.url = kwargs.get("url", None)

    def __str__(self):
        return self.authorId

    def __eq__(self, other):
        return isinstance(other, Author) and self.authorId == other.authorId

    def __hash__(self):
        return hash(self.authorId)

    def papers(self):
        for elem in self._kwargs.get("papers", []):
            yield SemanticScholarAPI.paper(elem["paperId"])

    def json(self):
        return self._kwargs

class PaperStub(object):

    def __init__(self, **kwargs):
        self.paperId = kwargs["paperId"]
        self.isInfluential = kwargs.get("isInfluential", False)
        self.title = kwargs.get("title", None)
        self.venue = kwargs.get("venue", None)
        self.year = kwargs.get("year", None)

    def __str__(self):
        return self.paperId

    def __eq__(self, other):
        return isinstance(other, PaperStub) and self.paperId == other.paperId

    def __hash__(self):
        return hash(self.paperId)

    def json(self):
        return {
            "paperId" : self.paperId,
            "isInfluential" : self.isInfluential,
            "title" : self.title,
            "venue" : self.venue,
            "year" : self.year,
        }

    def full(self, **kwargs):
        return SemanticScholarAPI.paper(self.paperId, **kwargs)

class Paper(object):

    def __init__(self, **kwargs):
        self.doi = kwargs.get("doi", None)
        self.citationVelocity = kwargs.get("citationVelocity", None)
        self.influentialCitationCount = kwargs.get("influentialCitationCount", None)
        self.url = kwargs.get("url", None)
        self.authors = [AuthorStub(**elem) for elem in kwargs.get("authors", [])]
        self.citations = [PaperStub(**elem) for elem in kwargs.get("citations", [])]
        self.references = [PaperStub(**elem) for elem in kwargs.get("references", [])]
        self.venue = kwargs.get("venue", None)
        self.references = kwargs.get("references", [])
        self.title = kwargs.get("title", None)
        self.year = kwargs.get("year", None)

    def __str__(self):
        return self.paperId

    def __eq__(self, other):
        return isinstance(other, Paper) and self.paperId == other.paperId

    def __hash__(self):
        return hash(self.paperId)

    def json(self):
        return self._kwargs

class SemanticScholarAPI(object):
    BASE_URL = "http://api.semanticscholar.org/v1"
    AUTHOR_ENDPOINT = "{}/{}".format(BASE_URL, "author")
    PAPER_ENDPOINT = "{}/{}".format(BASE_URL, "paper")
    SEARCH_ENDPOINT = "https://www.semanticscholar.org/api/1/search"
    RAW_PAPER_ENDPOINT = "https://www.semanticscholar.org/api/1/paper"
    headers = {
        'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_14_0) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/70.0.3538.77 Safari/537.36',
    }

    @staticmethod
    def fetch_file(url, fn, **kwargs):
        try:
            resp = requests.get(url, params=kwargs, headers=SemanticScholarAPI.headers, verify=False, timeout=10)
            if resp.status_code != 200:
                return None
        except:
            return None
        size = 0
        with open(fn, 'wb') as f:
            for chunk in resp.iter_content(chunk_size=1024):
                if chunk:
                    size += len(chunk)
                    f.write(chunk)
        return size

    @staticmethod
    def fetch_doi(url, fn, **kwargs):
        try:
            resp = requests.get(url, params=kwargs, headers=SemanticScholarAPI.headers, verify=False, timeout=10)
            if resp.status_code != 200:
                return None, None
        except:
            return None, None
        size = 0
        with open(fn, 'wb') as f:
            for chunk in resp.iter_content(chunk_size=1024):
                if chunk:
                    size += len(chunk)
                    f.write(chunk)
        return size, resp.url

    @staticmethod
    def paper(paper_id, **kwargs):
        url = "{}/{}".format(SemanticScholarAPI.PAPER_ENDPOINT, paper_id)
        resp = requests.get(url, params=kwargs, headers=SemanticScholarAPI.headers)
        return None if resp.status_code != 200 else resp.json() # Paper(**resp.json())

    @staticmethod
    def author(author_id, **kwargs):
        url = "{}/{}".format(SemanticScholarAPI.AUTHOR_ENDPOINT, author_id)
        resp = requests.get(url, params=kwargs, headers=SemanticScholarAPI.headers)
        return None if resp.status_code != 200 else resp.json() # Author(**resp.json())

    @staticmethod
    def pdf_url(paper_id):
      return "http://pdfs.semanticscholar.org/{}/{}.pdf".format(paper_id[:4], paper_id[4:])

    @staticmethod
    def raw_paper(paper_id, **kwargs):
        url = "{}/{}".format(SemanticScholarAPI.RAW_PAPER_ENDPOINT, paper_id)
        resp = requests.get(url, params=kwargs, headers=SemanticScholarAPI.headers)
        return None if resp.status_code != 200 else resp.json() # Paper(**resp.json())

    @staticmethod
    def search(q):
        resp = requests.post(SemanticScholarAPI.SEARCH_ENDPOINT, json={
            'authors': [],
            'coAuthors': [],
            'facets': {},
            'page': 1,
            'pageSize': 10,
            'publicationTypes': [],
            'queryString': q,
            'requireViewablePdf': False,
            'sort': "relevance",
            'venues': [],
            'yearFilter': None,
        }, headers=SemanticScholarAPI.headers)
        # print(resp.status_code)
        return None if resp.status_code != 200 else resp.json()