import gc
import logging
import os
import shutil
import tempfile

from morpheus.utils.logger import configure_logging
from morpheus.config import Config
from morpheus.config import CppConfig
from morpheus.pipeline import LinearPipeline
from morpheus.stages.input.file_source_stage import FileSourceStage
from morpheus.stages.output.write_to_file_stage import WriteToFileStage

MORPHEUS_ROOT_DIR = os.environ['MORPHEUS_ROOT']
TEST_DF = os.path.join(MORPHEUS_ROOT_DIR, "tests", "tests_data", "filter_probs.csv")


def run_pipe(out_file: str):
    config = Config()
    pipe = LinearPipeline(config)
    pipe.set_source(FileSourceStage(config, filename=TEST_DF))
    pipe.add_stage(WriteToFileStage(config, filename=out_file, overwrite=False))
    pipe.run()


def wrapper(logger: logging.Logger):
    tmp_dir = tempfile.mkdtemp()
    out_file = os.path.join(tmp_dir, "test.csv")

    logger.debug("Running pipe")
    try:
        run_pipe(out_file)
    finally:
        logger.debug("Pipe finished, cleaning up")
        shutil.rmtree(tmp_dir)
        logger.debug("Calling gc.collect()")
        gc.collect()
        logger.debug("gc.collect() - done")


if __name__ == "__main__":
    configure_logging(log_level=logging.DEBUG)
    logger = logging.getLogger("morpheus.tests.test_destructors")
    #CppConfig.set_should_use_cpp(False)
    # gc.set_debug(gc.DEBUG_LEAK)
    wrapper(logger)