from flask import request, jsonify

from storage.storage_service import (
    upload_device_type_icon,
    upload_device_avatar,
)


def upload_device_type_icon_controller():
    try:
        file = request.files.get("file")
        name = request.form.get("name")

        icon_url = upload_device_type_icon(file, name)

        return jsonify({
            "success": True,
            "message": "Device type icon uploaded successfully",
            "url": icon_url
        }), 201

    except ValueError as e:
        return jsonify({
            "success": False,
            "message": str(e)
        }), 400

    except Exception as e:
        return jsonify({
            "success": False,
            "message": "Failed to upload device type icon",
            "error": str(e)
        }), 500


def upload_device_avatar_controller():
    try:
        file = request.files.get("file")
        name = request.form.get("name")

        avatar_url = upload_device_avatar(file, name)

        return jsonify({
            "success": True,
            "message": "Device avatar uploaded successfully",
            "url": avatar_url
        }), 201

    except ValueError as e:
        return jsonify({
            "success": False,
            "message": str(e)
        }), 400

    except Exception as e:
        return jsonify({
            "success": False,
            "message": "Failed to upload device avatar",
            "error": str(e)
        }), 500