import logging
from datetime import datetime, timedelta

from app import app, instance_keys
from workers.servicekeyworker.models_pre_oci import pre_oci_model as model
from workers.worker import Worker

logger = logging.getLogger(__name__)


class ServiceKeyWorker(Worker):
  def __init__(self):
    super(ServiceKeyWorker, self).__init__()
    self.add_operation(self._refresh_service_key,
                       app.config.get('INSTANCE_SERVICE_KEY_REFRESH', 60) * 60)

  def _refresh_service_key(self):
    """
    Refreshes the instance's active service key so it doesn't get garbage collected.
    """
    expiration = timedelta(minutes=instance_keys.service_key_expiration)
    logger.debug('Starting refresh of automatic service keys')
    model.set_key_expiration(instance_keys.local_key_id, datetime.utcnow() + expiration)
    logger.debug('Finished refresh of automatic service keys')


if __name__ == "__main__":
  worker = ServiceKeyWorker()
  worker.start()