import functools

from flask import Blueprint, current_app, request, jsonify
from flask.views import MethodView
from flask_login import current_user

from . import db
# FIXME
from .fbin import get_file

app = Blueprint('api', __name__)

# TODO: Implement this stuff.

def makejson(f):
    @functools.wraps(f)
    def wrapper(*args, **kwargs):
        r = f(*args, **kwargs)
        if isinstance(r, dict):
            r = jsonify(r)
        return r
    return wrapper

def api_login_required(f):
    def wrapper(*args, **kwargs):
        if not current_user.is_authenticated:
            return {
                'status': False,
                'message': 'Not authenticated'
            }
        return f(*args, **kwargs)
    return wrapper

class FileAPI(MethodView):
    decorators = [api_login_required, makejson]

    def put(self, hash):
        f = get_file(hash, user_id = current_user.get_id())
        if not f:
            return {
                'status': False,
                'message': 'File not found'
            }
        filename = request.form.get('filename')
        if not filename:
            return {
                'status': False,
                'message': 'Empty or missing filename',
            }
        with db.session_scope() as sess:
            f.filename = filename
            sess.add(f)
        return {
            'status': True,
        }

    def delete(self, hash):
        pass

file_api_view = FileAPI.as_view('file_api')
app.add_url_rule('/file/<hash>', view_func = file_api_view, methods = ['PUT', 'DELETE'])