# -*- coding: utf-8; -*-
################################################################################
#
#  Rattail -- Retail Software Framework
#  Copyright © 2010-2022 Lance Edgar
#
#  This file is part of Rattail.
#
#  Rattail 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.
#
#  Rattail 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
#  Rattail.  If not, see <http://www.gnu.org/licenses/>.
#
################################################################################
"""
DataSync Views
"""

import asyncio
import json

from tailbone.views.asgi import WebsocketView


class DatasyncWS(WebsocketView):

    async def status(self, scope, receive, send):
        app = self.get_rattail_app()
        datasync_handler = app.get_datasync_handler()

        # is user allowed to see this?
        if not await self.authorize(scope, receive, send, 'datasync.status'):
            return

        # this tracks when client disconnects
        state = {'disconnected': False}

        async def wait_for_disconnect():
            message = await receive()
            if message['type'] == 'websocket.disconnect':
                state['disconnected'] = True

        # watch for client disconnect, while we do other things
        asyncio.create_task(wait_for_disconnect())

        # do the rest forever, until client disconnects
        while not state['disconnected']:

            # give client latest supervisor process info
            info = datasync_handler.get_supervisor_process_info()
            await send({'type': 'websocket.send',
                        'subtype': 'datasync.supervisor_process_info',
                        'text': json.dumps(info)})

            # pause for 1 second
            await asyncio.sleep(1)

    @classmethod
    def defaults(cls, config):
        cls._defaults(config)

    @classmethod
    def _defaults(cls, config):

        # status
        config.add_tailbone_websocket('datasync.status',
                                      cls, attr='status')


def defaults(config, **kwargs):
    base = globals()

    DatasyncWS = kwargs.get('DatasyncWS', base['DatasyncWS'])
    DatasyncWS.defaults(config)


def includeme(config):
    defaults(config)