from flask_login import current_user
from flask_restful import Resource, marshal_with
from werkzeug.exceptions import Forbidden

from controllers.console import api
from controllers.console.app.wraps import get_app_model
from controllers.console.wraps import (
    account_initialization_required,
    setup_required,
)
from fields.app_fields_extend import (
    recommended_app_list_fields,
)
from libs.login import login_required
from services.account_service_extend import TenantExtendService
from services.recommended_app_service_extend import RecommendedAppService


# ---------------- start sync app to
class InstalledSyncAppApi(Resource):
    @setup_required
    @login_required
    @account_initialization_required
    @marshal_with(recommended_app_list_fields)
    def get(self):
        """Installed app"""

        app_service = RecommendedAppService()

        return app_service.installed_app_list(current_user.current_tenant_id)


class AppSyncApi(Resource):
    @setup_required
    @login_required
    @account_initialization_required
    @get_app_model
    def put(self, app_model):
        """Sync app"""

        # The role of the current user in the ta table must be admin or owner
        tenant_extend_service = TenantExtendService
        super_admin_id = tenant_extend_service.get_super_admin_id().id
        if super_admin_id != current_user.id:
            raise Forbidden()

        app_service = RecommendedAppService()

        appId = app_service.sync_recommended_app(app_model.id)

        return appId, 200

    @setup_required
    @login_required
    @account_initialization_required
    @get_app_model
    def delete(self, app_model):
        """Delete sync app"""
        # The role of the current user in the ta table must be admin or owner
        tenant_extend_service = TenantExtendService
        super_admin_id = tenant_extend_service.get_super_admin_id().id
        if super_admin_id != current_user.id:
            raise Forbidden()

        app_service = RecommendedAppService()

        app_service.delete_sync_recommended_app(app_model.id)

        return "", 200


# ----------------start sync app------------------------
api.add_resource(AppSyncApi, "/apps/<uuid:app_id>/sync")
api.add_resource(InstalledSyncAppApi, "/installed/apps")
# ---------------- stop sync app ------------------------