app.py

import json
from flask import Flask, send_from_directory, request, jsonify, redirect, url_for
from flask_restful import Api, Resource, reqparse
from flask_cors import CORS #comment this on deployment
import sentry_sdk
from sentry_sdk.integrations.flask import FlaskIntegration
from api.imageManager import store_reviews, get_product, get_product_review
from datetime import datetime, timedelta, timezone
from flask_jwt_extended import create_access_token,get_jwt,get_jwt_identity, \
                               unset_jwt_cookies, jwt_required, JWTManager
from api.db import getClient
import hashlib
import os


db = getClient()

#from api.greet import greet
sentry_sdk.init(
    dsn="https://7f5de5cc77444f6f893d648f9a16c7de@o4504119699636224.ingest.sentry.io/4504119699636224",
    integrations=[
        FlaskIntegration(),
    ],

    # Set traces_sample_rate to 1.0 to capture 100%
    # of transactions for performance monitoring.
    # We recommend adjusting this value in production.
    traces_sample_rate=1.0
)
app = Flask(__name__, static_url_path='/', static_folder='build')
CORS(app) #comment this on deployment
api = Api(app)

app.config["JWT_SECRET_KEY"] = "please-remember-to-change-me"
app.config["JWT_ACCESS_TOKEN_EXPIRES"] = timedelta(hours=1)
jwt = JWTManager(app)


@app.after_request
def refresh_expiring_jwts(response):
    try:
        exp_timestamp = get_jwt()["exp"]
        now = datetime.now(timezone.utc)
        target_timestamp = datetime.timestamp(now + timedelta(minutes=30))
        if target_timestamp > exp_timestamp:
            access_token = create_access_token(identity=get_jwt_identity())
            data = response.get_json()
            if type(data) is dict:
                data["access_token"] = access_token
                response.data = json.dumps(data)
        return response
    except (RuntimeError, KeyError):
        # Case where there is not a valid JWT. Just return the original respone
        return response


@app.route('/token', methods=["POST"])
def create_token():
    username = request.json.get("username", None)
    password = request.json.get("password", None)

    
    existing_user = db.users.find_one({"username": username})
    if existing_user == None:
        return {"msg": "Wrong username or password"}, 401    
    else:
        newHash = hashlib.pbkdf2_hmac('sha256', password.encode(
            'utf-8'), existing_user["salt"], 100000)            
        if existing_user["hash"] != newHash:
            return {"msg": "Wrong username or password"}, 401

        access_token = create_access_token(identity=username)
        response = {"access_token": access_token}
        return response


@app.route('/register', methods=["POST"])
def register():
    print(request.json)
    username = request.json.get("username", None)
    password = request.json.get("password", None)    
    
    if db.users.find_one({"username": username}) == None:
        print({"username": username, "password": password})        
        
        pwSalt = os.urandom(32)
        pwHash = hashlib.pbkdf2_hmac(
            'sha256', password.encode('utf-8'), pwSalt, 100000)
        
        db.users.insert_one({"username": username, "salt": pwSalt, "hash": pwHash})
        return {"status": "success"}
    else:        
        return {"status": "fail"}

@app.route("/logout", methods=["POST"])
def logout():
    response = jsonify({"msg": "logout successful"})
    unset_jwt_cookies(response)
    return response

@app.errorhandler(404)
def not_found(e):
    print("Not Found")
    return send_from_directory(app.static_folder,'index.html'), 201

@app.route('/')
# @jwt_required()
def index():
    print("Index")
    return send_from_directory(app.static_folder,'index.html'), 201

@app.route('/debug-sentry')
def trigger_error():
    division_by_zero = 1 / 0

@app.route('/image-api', methods = ['POST', 'GET'])
def image_portal():

    print("Image api")
    requestBody = request.get_json()
    
    if requestBody['option'] == 'store-review':
        print("Save image")
        store_reviews(requestBody['content'])
        response = {
            'content': 'received'
        }
        return response, 200
    elif requestBody['option'] == 'get-reviews-for-product':
        print("Retrieve image")
        reviewInfo = get_product_review(requestBody['content'])
        
        response = {
            'content': reviewInfo
        }
        return response, 200

@app.route('/product-api', methods = ['POST', 'GET'])
def product_portal():

    requestBody = request.get_json()

    if requestBody['option'] == 'store-product':
        
        store_reviews(requestBody['content'])
        response = {
            'content': 'received'
        }
        return response, 200

    elif requestBody['option'] == 'get-product':
       
        product = get_product(requestBody['content'])
        
        response = {
            'content': product
        }
        return response, 200

#api.add_resource(greet, '/greeting')
def get_app():
    return app