# Copyright (C) 2021 Wind River Systems, Inc.
#
#  Licensed under the Apache License, Version 2.0 (the "License");
#  you may not use this file except in compliance with the License.
#  You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
#  Unless required by applicable law or agreed to in writing, software
#  distributed under the License is distributed on an "AS IS" BASIS,
#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#  See the License for the specific language governing permissions and
#  limitations under the License.

import cotyledon

from o2app import bootstrap
from o2common.service.watcher.base import WatcherTree
from o2common.service.watcher.worker import PollWorker

from o2ims.service.watcher.ocloud_watcher import OcloudWatcher
from o2ims.service.watcher.ocloud_watcher import DmsWatcher
from o2ims.service.watcher.resourcepool_watcher import ResourcePoolWatcher
from o2ims.adapter.clients.ocloud_sa_client import StxSaDmsClient
from o2ims.adapter.clients.ocloud_sa_client import StxSaOcloudClient
from o2ims.adapter.clients.ocloud_sa_client import StxSaResourcePoolClient

from o2ims.service.watcher.pserver_watcher import PServerWatcher
from o2ims.adapter.clients.ocloud_sa_client import StxPserverClient

from o2ims.service.watcher.pserver_cpu_watcher import PServerCpuWatcher
from o2ims.adapter.clients.ocloud_sa_client import StxCpuClient

from o2ims.service.watcher.pserver_mem_watcher import PServerMemWatcher
from o2ims.adapter.clients.ocloud_sa_client import StxMemClient

from o2ims.service.watcher.pserver_if_watcher import PServerIfWatcher
from o2ims.adapter.clients.ocloud_sa_client import StxIfClient

from o2ims.service.watcher.pserver_port_watcher import PServerPortWatcher
from o2ims.adapter.clients.ocloud_sa_client import StxPortClient

from o2common.helper import o2logging
logger = o2logging.get_logger(__name__)

# r = redis.Redis(**config.get_redis_host_and_port())


class WatcherService(cotyledon.Service):
    def __init__(self, worker_id, args=None) -> None:
        super().__init__(worker_id)
        self.args = args
        self.bus = bootstrap.bootstrap()
        self.worker = PollWorker()

    def run(self):
        try:
            root = WatcherTree(OcloudWatcher(
                StxSaOcloudClient(), self.bus))
            root.addchild(
                DmsWatcher(StxSaDmsClient(), self.bus))

            child_respool = root.addchild(
                ResourcePoolWatcher(StxSaResourcePoolClient(),
                                    self.bus))
            child_pserver = child_respool.addchild(
                PServerWatcher(StxPserverClient(), self.bus))
            child_pserver.addchild(
                PServerCpuWatcher(StxCpuClient(), self.bus))
            child_pserver.addchild(
                PServerMemWatcher(StxMemClient(), self.bus))
            child_pserver.addchild(
                PServerIfWatcher(StxIfClient(), self.bus))
            child_pserver.addchild(
                PServerPortWatcher(StxPortClient(), self.bus))

            self.worker.add_watcher(root)

            self.worker.start()
        except Exception as ex:
            logger.warning("WorkerService Exception:" + str(ex))
        finally:
            self.worker.stop()


def start_watchers(sm: cotyledon.ServiceManager = None):
    watchersm = sm if sm else cotyledon.ServiceManager()
    watchersm.add(WatcherService, workers=1, args=())
    watchersm.run()


def main():
    logger.info("Resource watcher starting")
    start_watchers()


if __name__ == "__main__":
    main()