from flask import Blueprint, request, jsonify
from app.database import db
from app.models import Documento


documentos_bp = Blueprint('documentos', __name__, url_prefix='/documentos')

@documentos_bp.route('/', methods=['GET'])
def get_documentos():
    documentos = Documento.query.all()
    documentos_data = [
        {
            'id': str(doc.id),
            'nome_arquivo': doc.nome_arquivo,
            'caminho_arquivo': doc.caminho_arquivo,
            'usuario_id': doc.usuario_id,
            'criado_em': doc.criado_em
        }
        for doc in documentos
    ]
    return jsonify(documentos_data), 200

@documentos_bp.route('/', methods=['POST'])
def create_documento():
    data = request.get_json()
    if not all(k in data for k in ('nome_arquivo', 'caminho_arquivo', 'usuario_id')):
        return jsonify({'error': 'Campos obrigatórios ausentes'}), 400

    try:
        novo_documento = Documento(
            nome_arquivo=data['nome_arquivo'],
            caminho_arquivo=data['caminho_arquivo'],
            usuario_id=data['usuario_id']
        )
        db.session.add(novo_documento)
        db.session.commit()

        return jsonify({
            'message': 'Documento criado com sucesso',
            'id': str(novo_documento.id)
        }), 201
    except Exception as e:
        db.session.rollback()
        return jsonify({'error': str(e)}), 500