"""Enable change stream hook.

A hook to enable change stream in the replica set and the sharded cluster in the multi-tenant
environment.
"""
from time import sleep
from bson.objectid import ObjectId
from pymongo import MongoClient
import os.path
import json

from buildscripts.resmokelib import config
from buildscripts.resmokelib.testing.hooks import interface
from buildscripts.resmokelib.testing.hooks import jsfile


class EnableChangeStream(interface.Hook):
    """Enable change stream hook class.

    Enables change stream in the multi-tenant environment for the replica set and the sharded
    cluster.
    """

    IS_BACKGROUND = False

    def __init__(self, hook_logger, fixture, tenant_id=None):
        """Initialize the EnableChangeCollection."""
        description = "Enables the change stream in the multi-tenant environment."
        self._js_filename = os.path.join("jstests", "hooks", "run_enable_change_stream.js")
        interface.Hook.__init__(self, hook_logger, fixture, description)
        self._fixture = fixture
        self._tenant_id = ObjectId(tenant_id) if tenant_id else None

    def before_test(self, test, test_report):
        """Enables change stream before test suite starts executing the test cases."""
        if hasattr(self._fixture, "mongos"):
            self.logger.info("Enabling change stream in the sharded cluster.")
            self._set_change_collection_state_in_sharded_cluster(test, test_report)
        else:
            self.logger.info("Enabling change stream in the replica sets.")
            self._call_js_hook(self._fixture, test, test_report)

    def _set_change_collection_state_in_sharded_cluster(self, test, test_report):
        for shard in self._fixture.shards:
            self._call_js_hook(shard, test, test_report)
        # TODO SERVER-68341 Remove the sleep. Sleep for some time such that periodic-noop entries
        # get written to change collections. This will ensure that the client open the change
        # stream cursor with the resume token whose timestamp is later than the change collection
        # first entry. Refer to the ticket for more details.
        sleep(5)

    def _call_js_hook(self, fixture, test, test_report):
        shell_options = {"global_vars": {"TestData": {"tenantId": str(self._tenant_id)}}}
        hook_test_case = jsfile.DynamicJSTestCase.create_before_test(
            test.logger, test, self, self._js_filename, shell_options)
        hook_test_case.configure(fixture)
        hook_test_case.run_dynamic_test(test_report)