import os

from log import Log

import sys
import importlib
import importlib.machinery

class GcpModuleFinder(importlib.abc.MetaPathFinder):
    _MODULES = [
        'google.cloud.bigquery',
        'google.cloud.storage',
        'google.cloud.automl_v1beta1',
        'google.cloud.translate',
        'google.cloud.translate_v2',
        'google.cloud.translate_v3',
        'google.cloud.language',
        'google.cloud.language_v1',
        'google.cloud.videointelligence',
        'google.cloud.videointelligence_v1',
        'google.cloud.vision',
        'google.cloud.vision_v1',
    ]
    _KAGGLE_GCP_PATH = 'kaggle_gcp.py'
    def __init__(self):
        pass

    def _is_called_from_kaggle_gcp(self):
        import inspect
        for frame in inspect.stack():
            if os.path.basename(frame.filename) == self._KAGGLE_GCP_PATH:
                return True
        return False

    def find_spec(self, fullname, path, target=None):
        if fullname in self._MODULES:
            # If being called from kaggle_gcp, don't return our
            # monkeypatched module to avoid circular dependency,
            # since we call kaggle_gcp to load the module.
            if self._is_called_from_kaggle_gcp():
                return None
            return importlib.machinery.ModuleSpec(fullname, GcpModuleLoader())


class GcpModuleLoader(importlib.abc.Loader):
    def __init__(self):
        pass

    def create_module(self, spec):
        """Create the gcp module from the spec.
        """
        import kaggle_gcp
        _LOADERS = {
            'google.cloud.bigquery': kaggle_gcp.init_bigquery,
            'google.cloud.storage': kaggle_gcp.init_gcs,
            'google.cloud.automl_v1beta1': kaggle_gcp.init_automl,
            'google.cloud.translate': kaggle_gcp.init_translation_v3,
            'google.cloud.translate_v2': kaggle_gcp.init_translation_v2,
            'google.cloud.translate_v3': kaggle_gcp.init_translation_v3,
            'google.cloud.language': kaggle_gcp.init_natural_language,
            'google.cloud.language_v1': kaggle_gcp.init_natural_language,
            'google.cloud.videointelligence': kaggle_gcp.init_video_intelligence,
            'google.cloud.videointelligence_v1': kaggle_gcp.init_video_intelligence,
            'google.cloud.vision': kaggle_gcp.init_vision,
            'google.cloud.vision_v1': kaggle_gcp.init_vision
        }
        monkeypatch_gcp_module = _LOADERS[spec.name]()
        return monkeypatch_gcp_module

    def exec_module(self, module):
        pass

if not hasattr(sys, 'frozen'):
    sys.meta_path.insert(0, GcpModuleFinder())