# -*- coding:utf-8 -*-
import os

from flask import request, abort, current_app, send_from_directory
from werkzeug.utils import secure_filename
import lz4.frame

from api.lib.common_setting.resp_format import ErrFormat
from api.lib.common_setting.upload_file import allowed_file, generate_new_file_name, CommonFileCRUD
from api.resource import APIView

prefix = '/file'

ALLOWED_EXTENSIONS = {
    'txt', 'pdf', 'png', 'jpg', 'jpeg', 'gif', 'xls', 'xlsx', 'doc', 'docx', 'ppt', 'pptx', 'csv', 'svg'
}


class FileExtensionAllowView(APIView):
    url_prefix = (f'{prefix}/allow_extensions',)

    def get(self):
        extensions = current_app.config.get('ALLOWED_EXTENSIONS', ALLOWED_EXTENSIONS)
        extensions = list(extensions)
        return self.jsonify(extensions)


class GetFileView(APIView):
    url_prefix = (f'{prefix}/<string:_filename>',)

    def get(self, _filename):
        file_stream = CommonFileCRUD.get_file(_filename)
        return self.send_file(file_stream, as_attachment=True, download_name=_filename)


class PostFileView(APIView):
    url_prefix = (f'{prefix}',)

    def post(self):
        # check if the post request has the file part
        if 'file' not in request.files:
            abort(400, ErrFormat.no_file_part)

        file = request.files['file']

        if not file:
            abort(400, ErrFormat.file_is_required)
        extension = file.mimetype.split('/')[-1]
        if '+' in extension:
            extension = file.filename.split('.')[-1]
        if file.filename == '':
            filename = f'.{extension}'
        else:
            if extension not in file.filename:
                filename = file.filename + f".{extension}"
            else:
                filename = file.filename

        if allowed_file(filename, current_app.config.get('ALLOWED_EXTENSIONS', ALLOWED_EXTENSIONS)):
            new_filename = generate_new_file_name(filename)
            new_filename = secure_filename(new_filename)
            file_content = file.read()
            compressed_data = lz4.frame.compress(file_content)
            try:
                CommonFileCRUD.add_file(
                    origin_name=filename,
                    file_name=new_filename,
                    binary=compressed_data,
                )

                return self.jsonify(file_name=new_filename)
            except Exception as e:
                current_app.logger.error(e)
                abort(400, ErrFormat.upload_failed.format(e))

        abort(400, ErrFormat.file_type_not_allowed.format(filename))