import config
import requests
from base64 import b64encode
from garfpy import logger


class Kroger:
    def __init__(self):
        self.client_id = config.CLIENT_ID
        self.client_secret = config.CLIENT_SECRET
        self.auth = b64encode(
            f"{self.client_id}:{self.client_secret}".encode()
        ).decode()

    def kroger_token(self):
        headers = {
            "Content-Type": "application/x-www-form-urlencoded",
            "Authorization": f"Basic {self.auth}",
        }

        response = requests.post(
            "https://api.kroger.com/v1/connect/oauth2/token",
            headers=headers,
            data={"grant_type": "client_credentials", "scope": "product.compact"},
        )

        response.raise_for_status()
        return response.json()["access_token"]

    def find_store(self, zipcode, kroken):
        headers = {
            "Authorization": f"Bearer {kroken}",
        }
        params = {
            "filter.zipCode.near": zipcode,
            "filter.limit": 1,
        }
        response = requests.get(
            "https://api.kroger.com/v1/locations", headers=headers, params=params
        )
        return response.json()

    def search_product(self, product, loc_id, kroken):
        logger.info(f"Searching for {product}...")
        headers = {
            "Authorization": f"Bearer {kroken}",
        }
        params = {
            "filter.term": product,
            "filter.locationId": loc_id,
            "filter.limit": 10,
        }
        response = requests.get(
            "https://api.kroger.com/v1/products", headers=headers, params=params
        )
        return response.json()

    def garfshop(self, query):
        try:
            query = query.split()
            kroken = self.kroger_token()
            product = query[-2]
            zipcode = query[-1]
            loc_data = self.find_store(zipcode, kroken)
            loc_id = loc_data["data"][0]["locationId"]
            store_name = loc_data["data"][0]["name"]
            product_query = self.search_product(product, loc_id, kroken)
            products = product_query["data"]
            sorted_products = sorted(
                products, key=lambda item: item["items"][0]["price"]["regular"]
            )
            response = f"Prices for `{product}` at `{store_name}` near `{zipcode}`:\n"
            for item in sorted_products:
                product_name = item["description"]
                price = item["items"][0]["price"]["regular"]
                response += f"- `${price}`: {product_name} \n"
            return response
        except Exception as e:
            return e