import pathlib

from aiohttp import web
from aiohttp_auth import auth

from core.api.decorator import request_mapping
from core.utils.utils import load_config


class Login():

    def __init__(self,cbpi):
        self.cbpi = cbpi
        self.cbpi.register(self)
        cfg = load_config(str(pathlib.Path('.') / 'config' / 'config.yaml'))

        print("######", cfg)
        self.db = {'user': 'password', 'super_user': 'super_password'}

    @request_mapping(path="/logout", name="Logout", method="GET", auth_required=True)
    async def logout_view(self, request):
        await auth.forget(request)
        return web.Response(body='OK'.encode('utf-8'))

    @request_mapping(path="/login",name="Login", method="POST", auth_required=False)
    async def login_view(self, request):
        params = await request.post()

        print(params.get('username', None), params.get('password', None))
        user = params.get('username', None)
        if (user in self.db and
            params.get('password', None) == self.db[user]):

            # User is in our database, remember their login details
            await auth.remember(request, user)
            return web.Response(body='OK'.encode('utf-8'))

        raise web.HTTPForbidden()