Onionr/src/apiservers/private/register_private_blueprints.py

"""Onionr - Private P2P Communication.

This file registers blueprints for the private api server
"""
from gevent import spawn
from gevent import sleep

from httpapi import security, friendsapi, profilesapi, configapi, insertblock
from httpapi import miscclientapi, onionrsitesapi, apiutils
from httpapi import directconnections
from httpapi import themeapi
from httpapi.sse.private import private_sse_blueprint

"""
    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <https://www.gnu.org/licenses/>.
"""


def register_private_blueprints(private_api, app):
    """Register private API plask blueprints."""
    app.register_blueprint(security.client.ClientAPISecurity(
        private_api).client_api_security_bp)
    app.register_blueprint(friendsapi.friends)
    app.register_blueprint(profilesapi.profile_BP)
    app.register_blueprint(configapi.config_BP)
    app.register_blueprint(insertblock.ib)
    app.register_blueprint(miscclientapi.getblocks.client_get_blocks)
    app.register_blueprint(miscclientapi.endpoints.PrivateEndpoints(
        private_api).private_endpoints_bp)
    app.register_blueprint(miscclientapi.motd.bp)
    app.register_blueprint(onionrsitesapi.site_api)
    app.register_blueprint(apiutils.shutdown.shutdown_bp)
    app.register_blueprint(miscclientapi.staticfiles.static_files_bp)
    app.register_blueprint(directconnections.DirectConnectionManagement(
        private_api).direct_conn_management_bp)
    app.register_blueprint(themeapi.theme_blueprint)
    app.register_blueprint(private_sse_blueprint)

    def _add_events_bp():
        while True:
            try:
                private_api._too_many
                break
            except AttributeError:
                sleep(0.2)
        app.register_blueprint(
            private_api._too_many.get_by_string('DaemonEventsBP').flask_bp)

    spawn(_add_events_bp)
    return app