import os
import logging.config
import time
import argparse

import features

from app import app, repo_mirror_api
from workers.worker import Worker
from workers.repomirrorworker import process_mirrors
from util.repomirror.validator import RepoMirrorConfigValidator
from util.repomirror.skopeomirror import SkopeoMirror
from util.log import logfile_path


logger = logging.getLogger(__name__)

DEFAULT_MIRROR_INTERVAL = 30


class RepoMirrorWorker(Worker):
  def __init__(self):
    super(RepoMirrorWorker, self).__init__()
    RepoMirrorConfigValidator(app.config.get('FEATURE_REPO_MIRROR', False)).valid()

    self._mirrorer = SkopeoMirror()
    self._next_token = None

    interval = app.config.get('REPO_MIRROR_INTERVAL', DEFAULT_MIRROR_INTERVAL)
    self.add_operation(self._process_mirrors, interval)

  def _process_mirrors(self):
    while True:
      assert app.config.get('FEATURE_REPO_MIRROR', False)

      self._next_token = process_mirrors(self._mirrorer, self._next_token)
      if self._next_token is None:
        break


if __name__ == '__main__':
  if os.getenv('PYDEV_DEBUG', None):
    import pydevd
    host, port = os.getenv('PYDEV_DEBUG').split(':')
    pydevd.settrace(host, port=int(port), stdoutToServer=True, stderrToServer=True, suspend=False)

  logging.config.fileConfig(logfile_path(debug=False), disable_existing_loggers=False)

  parser = argparse.ArgumentParser()
  parser.add_argument('mode', metavar='MODE', type=str, nargs='?', default='',
                      choices=['mirror', ''])
  args = parser.parse_args()

  if not features.REPO_MIRROR:
    logger.debug('Repository mirror disabled; skipping RepoMirrorWorker')
    while True:
      time.sleep(100000)

  worker = RepoMirrorWorker()
  worker.start()