#!/usr/bin/env python3
import redis
import os
import json
import socket
import time
import logging


class RedisSuperStorage:
    def __init__(self, redis_url: str, timeout: int):
        self.r = redis.from_url(redis_url)
        self._timeout = timeout

    def get_consumer_list(self) -> dict:
        keys = self.r.keys('consumer_*')

        list_of_customers = {}

        for key in keys:
            info = json.loads((self.r.get(key) or b"{}").decode('utf-8'))

            if info:
                list_of_customers[info['uuid']] = info

        return list_of_customers

    def get_producer_list(self) -> dict:
        keys = self.r.keys('producer_*')

        list_of_producer_ip = {}

        for key in keys:
            ip = (self.r.get(key) or b"").decode('utf-8')

            if ip:
                list_of_producer_ip[key.decode('utf-8')] = ip

        return list_of_producer_ip

    def is_ip_changed(self) -> bool:
        old_ip = self.r.get('current_ip')

        if old_ip:
            old_ip = old_ip.decode('utf-8')

        host_name = socket.gethostname()
        current_ip = socket.gethostbyname(host_name)

        if current_ip != old_ip:
            logging.info(f'IP changed: {old_ip} -> {current_ip}')
            self.r.set('current_ip', current_ip.encode('utf-8'))

        return current_ip != old_ip

    def update_consumer(self, uuid: str, ip: str):

        cust_key = f"consumer_{uuid}"

        info = {
            "uuid": uuid,
            "ip": ip,
            "last_seen": time.time()
        }

        self.r.set(cust_key, json.dumps(info).encode('utf-8'))
        self.r.expire(cust_key, os.environ["CUSTOMER_TIMEOUT"])