HiddenEye-Legacy/controllers/ngrok_controller.py

from pyngrok import ngrok
from pyngrok import exception as pyngrok_exceptions
from pyngrok import installer as pyngrok_installer
from pyngrok import conf as pyngrok_config


class NgrokController:
    def __init__(self, config_path: str = ".config/ngrok.yml", model=ngrok, exceptions=pyngrok_exceptions, installer=pyngrok_installer, config=pyngrok_config):
        self._model = model
        self._exceptions = exceptions
        self._installer = installer
        self._config = config
        self._config_path = config_path
        self._tunnels = None
        self._ngrok_url = None

    def close_latest_connection(self):
        try:
            self._model.disconnect(self._ngrok_url)
        except self._exceptions.PyngrokError:
            print("Can't find any latest connections.")  # FIXME replace with View entry
            pass

    def maintain_default_config(self):
        self._installer.install_default_config(self._config_path)


    def activate_config_path(self):
        self._config.PyngrokConfig(config_path=self._config_path)

    def establish_connection(self, port='80'):
        self._model.connect(port=port, name='HiddenEye Connection', pyngrok_config=self._config_path)

    def obtain_tunnels(self):
        self._tunnels = self._model.get_tunnels()