import time
import json
import requests
from decouple import config



class wordtune():
    def __init__(self, token, draftId):
        super(wordtune, self).__init__()
        self.token = token
        self.draftId = draftId
        self.url = "https://api.wordtune.com/rewrite"

    # Generate Payload
    def payload_generator(self, inp):
        inp_len = len(inp)

        payload = json.dumps({
            "text": inp,
            "action": "REWRITE",
            "start": 0,
            "end": inp_len,
            "selection": {
                "wholeText": inp,
                "bulletText": "",
                "start": 0,
                "end": inp_len
            },
            "draftId": self.draftId,
            "emailAccount": None,
            "emailMetadata": {},
            "lookaheadIndex": 0
        })

        payload_without_whitespace = payload.replace(
            ": ", ":").replace(", ", ",")
        # Calulate content_length header
        self.content_length = str(len(payload_without_whitespace))

        return payload
    
    # Generate Headers
    def headers_generator(self):
        headers = {
            'host': 'api.wordtune.com',
            'connection': 'keep-alive',
            'content-length': self.content_length,
            'sec-ch-ua': '" Not;A Brand";v="99", "Google Chrome";v="91", "Chromium";v="91"',
            'dnt': '1',
            'x-wordtune-origin': 'null',
            'x-wordtune-version': '0.0.1',
            'sec-ch-ua-mobile': '?0',
            'user-agent': 'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.164 Safari/537.36',
            'content-type': 'application/json',
            'x-wordtune': '1',
            'token': self.token,
            'accept': '*/*',
            'origin': 'https://app.wordtune.com',
            'sec-fetch-site': 'same-site',
            'sec-fetch-mode': 'cors',
            'sec-fetch-dest': 'empty',
            'referer': 'https://app.wordtune.com/',
            'accept-encoding': 'gzip, deflate',
            'accept-language': 'en-US,en;q=0.9'
        }

        return headers

    # Performing Post HTTP request
    def requests(self, payload, headers):
        # Make delay for security reason - Account Suspension Prevention
        time.sleep(2)
        response = requests.request(
            "POST", self.url, headers=headers, data=payload)
        response_json = response.json()

        return response_json

    # Get list of suggestions
    def get_suggestion(self, response_json, inp):
        suggestion_list = []

        for suggestion in response_json['suggestions']:
            wordtune_changedText = suggestion[0]
            changedText_start = suggestion[1][0]
            changedText_end = suggestion[1][1]
            rephrased = inp[0:changedText_start] + \
                wordtune_changedText + inp[changedText_end:-1]
            suggestion_list.append(rephrased)

        return suggestion_list


def main():
    # environmental variables must be in .env file in the same dir as this file
    token = config('TOKEN')
    draftId = config('DRAFTID')
    paraphrase = wordtune(token, draftId)
    
    inps = ['How are you?', 'How should I take Acetaminophen?']
    for inp in inps:
        payload = paraphrase.payload_generator(inp)
        headers = paraphrase.headers_generator()
        response = paraphrase.requests(payload, headers)
        suggestion_list = paraphrase.get_suggestion(response, inp)
        print(suggestion_list)


if __name__ == '__main__':
    main()