# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
# SPDX-License-Identifier: MIT-0
import boto3
import os
import logging
import werkzeug

werkzeug.cached_property = werkzeug.utils.cached_property

from flask import Flask, request, url_for
from flask_restplus import Api, Resource, fields
from flask_cors import CORS


flask_app = Flask(__name__)
flask_app.debug = True
log_level = logging.INFO
flask_app.logger.setLevel(log_level)
# enable CORS
CORS(flask_app, resources={r"/*": {"origins": "*"}})

session = boto3.Session()
dynamodb = session.resource(
    "dynamodb", region_name=os.getenv("PRODUCTS_TABLE_REGION", "eu-west-1")
)
table_name = os.getenv("PRODUCTS_TABLE_NAME", "products")
table = dynamodb.Table(table_name)

# Fix of returning swagger.json on HTTP
@property
def specs_url(self):
    """
    The Swagger specifications absolute url (ie. `swagger.json`)

    :rtype: str
    """
    return url_for(self.endpoint("specs"), _external=False)


Api.specs_url = specs_url
app = Api(
    app=flask_app,
    version="1.0",
    title="Product Catalog",
    description="Complete dictionary of Products available in the Product Catalog",
)

name_space = app.namespace("products", description="Products from Product Catalog")

model = app.model(
    "Name Model",
    {
        "name": fields.String(
            required=True,
            description="Name of the Product",
            help="Product Name cannot be blank.",
        )
    },
)


@name_space.route("/")
class Products(Resource):
    """
    Manipulations with products.
    """

    def get(self):
        """
        List of products.
        Returns a list of products
        """
        try:

            flask_app.logger.info("Get all.....")

            resp = table.scan(AttributesToGet=["id", "name"])

            products = {}
            for item in resp["Items"]:
                products[item["id"]] = item["name"]

            flask_app.logger.info("Get-All Request succeeded")
            return {"products": products}
        except Exception as e:
            flask_app.logger.error(
                "Error 500 Could not retrieve information " + e.__doc__
            )
            name_space.abort(
                500,
                e.__doc__,
                status="Could not retrieve information",
                statusCode="500",
            )


@name_space.route("/ping")
class Ping(Resource):
    def get(self):
        return "healthy"


@name_space.route("/<int:id>")
@name_space.param("id", "Specify the ProductId")
class MainClass(Resource):
    @app.doc(responses={200: "OK", 400: "Invalid Argument", 500: "General Error"})
    def get(self, id=None):
        try:
            resp = table.get_item(Key={"id": str(id)})

            flask_app.logger.info("Get Request succeeded " + resp["Item"]["name"])
            return {"status": "Product Details retrieved", "name": resp["Item"]["name"]}
        except Exception as e:
            flask_app.logger.error(
                "Error 500 Could not retrieve information " + e.__doc__
            )
            name_space.abort(
                500,
                e.__doc__,
                status="Could not retrieve information",
                statusCode="500",
            )

    @app.doc(responses={200: "OK", 400: "Invalid Argument", 500: "General Error"})
    @app.expect(model)
    def post(self, id):
        try:
            product_item = {"id": str(id), "name": request.json["name"]}

            table.put_item(Item=product_item)

            flask_app.logger.info("Post Request succeeded " + request.json["name"])

            return {
                "status": "New Product added to Product Catalog",
                "name": request.json["name"],
            }
        except Exception as e:
            flask_app.logger.error("Error 500 Could not save information " + e.__doc__)
            name_space.abort(
                500, e.__doc__, status="Could not save information", statusCode="500"
            )


if __name__ == "__main__":
    app.run(host="0.0.0.0", debug=True)