from flask import request,jsonify
from flask_restx import Resource, Namespace, fields
from database.database import DB
import datetime
import jwt


users = {}

Auth = Namespace(
    name="Auth",
    description="사용자 인증을 위한 API",
)


user_fields = Auth.model('User', {  # Model 객체 생성
    'id': fields.String(description='a User Name', required=True, example="id")
})


user_fields_auth = Auth.inherit('User Auth', user_fields, {
    'password': fields.String(description='Password', required=True)

})


get_phone_number = Auth.inherit('get a phone number of an user', {
    'id' : fields.String(description="user id", required=True)
})


get_email = Auth.inherit('get an email of an user', {
    'id' : fields.String(description="user id", required=True)
})

user_fields_register = Auth.inherit('User reigster', user_fields, {
    'password': fields.String(description='Password', required=True),'email': fields.String(description='email', required=True),'phone': fields.String(description='phone', required=True)

})



@Auth.route('/id')
class AuthCheck(Resource):
    @Auth.doc(responses={200: 'Success'})
    @Auth.doc(responses={500: 'Register Failed'})
    def post(self):
        db=DB()
        id = request.json['id']
        value=db.db_check_id(id)
        if value != None:
            return {
                "message": "중복 아이디가 있습니다"
            }, 500
        else:
            return {
                'message': '사용가능한 아이디입니다'  # str으로 반환하여 return
            }, 200


@Auth.route('/register')
class AuthRegister(Resource):
    @Auth.expect(user_fields_register)
    @Auth.doc(responses={200: 'Success'})
    @Auth.doc(responses={500: 'Register Failed'})
    def post(self):
        user_manager = DB()
        # Extract data from the request
        data = request.json
        id_ = data['id']
        password = data['password']
        user_email = data['email']
        # sex = data['user_sex']
        phone = data['phone']

        # Prepare data for registration
        user_data = {
            'username': id_,
            'password': password,
            'email': user_email,
            # 'sex': sex,
            'phone': phone
        }

        # Call the register_user method from the UserManager instance
        result, status_code = user_manager.register_user(user_data)

        # Return the appropriate response based on the result from UserManager
        if status_code == 200:
            return result, 200
        else:
            return result, 500

@Auth.route('/retrive_phone_number')
class AuthRegister(Resource):
    @Auth.expect(get_phone_number)
    @Auth.doc(responses={200: 'Success'})
    @Auth.doc(responses={500: 'Register Failed'})
    def post(self):
        user_manager = DB()
        data = request.json
        id_ = data['id']
        query_input = {
            "username" : id_
        }
        result, status_code = user_manager.get_phone_number(query_input)

        if status_code == 200:
            return result, 200
        else:
            return result, 500


@Auth.route('/retrive_email')
class AuthRegister(Resource):
    @Auth.expect(get_email)
    @Auth.doc(responses={200: 'Success'})
    @Auth.doc(responses={500: 'Register Failed'})
    def post(self):
        user_manager = DB()
        data = request.json
        id_ = data['id']
        query_input = {
            "username" : id_
        }
        result, status_code = user_manager.get_email(query_input)

        if status_code == 200:
            return result, 200
        else:
            return result, 500



@Auth.route('/login')
class AuthLogin(Resource):
    @Auth.expect(user_fields_auth)
    @Auth.doc(responses={200: 'Login Successful'})
    @Auth.doc(responses={401: 'Unauthorized'})
    @Auth.doc(responses={500: 'Login Failed'})
    def post(self):
        user_manager = DB()
        # Extract data from the request
        data = request.json
        id_ = data['id']
        password = data['password']

        # Prepare data for authentication
        user_data = {
            'username': id_,
            'password': password
        }

        # Call the login_user method from the UserManager instance
        result, status_code = user_manager.login_user(user_data)

        if result['status'] == 'success':
            payload = {
                'id': id_,
                'exp': datetime.datetime.utcnow() + datetime.timedelta(days=14)
            }
            token = jwt.encode(payload, "secret", algorithm='HS256')
            return {'result': 'success', 'token': token}, 200
        else :
            return {'result': 'fail', 'msg': '아이디/비밀번호가 일치하지 않습니다.'}, 401


@Auth.route('/withdraw')
class AuthWithdraw(Resource):
    def post(self):
         db=DB()
         id = request.json['token']
         payload = jwt.decode(id, "secret", algorithms=['HS256'])
         db.db_delete_id(payload['id'])
         return {'secession':'success'}, 200