Ahenk/opt/ahenk/base/registration/Registration.py

#!/usr/bin/python3
# -*- coding: utf-8 -*-
# Author: Volkan Şahin <volkansah.in> <bm.volkansahin@gmail.com>

from base.Scope import Scope
from base.messaging.MessageSender import MessageSender
from uuid import getnode as get_mac
import json, uuid, netifaces, socket, datetime


class Registration():


    #TODO keep catches and set logs
    def __init__(self):
        scope = Scope().getInstance()
        self.conf_manager = scope.getConfigurationManager()
        self.logger=scope.getLogger()
        self.message_manager=scope.getMessageManager()
        self.event_manager = scope.getEventManager()

        self.event_manager.register_event('confirm_registration',self.confirm_registration)

        if self.conf_manager.has_section('REGISTRATION'):
            if self.conf_manager.get('REGISTRATION', 'registered')=='false':
                self.re_register()
            else:
                self.logger.debug('[Registration] Ahenk already registered')
        else:
            self.register(True)

    def registration_request(self):
        self.logger.debug('[Registration] Requesting registration')
        message_sender=MessageSender(self.get_registration_request_message(),None)
        message_sender.connect_to_server()

    def ldap_registration_request(self):
        self.logger.debug('[Registration] Requesting LDAP registration')
        message_sender=MessageSender(self.get_ldap_registration_request_message(),None)
        message_sender.connect_to_server()

    def confirm_registration(self,reg_reply):
        self.logger.debug('[Registration] Reading registration reply')
        j = json.loads(reg_reply)
        self.logger.debug('[Registration]'+j['message'])
        status =str(j['status']).lower()
        dn=str(j['agentDn']).lower()
        self.logger.debug('[Registration] Registration status: '+str(status))

        if str(status)=='registered' or str(status)=='registered_without_ldap':
            self.logger.debug('dn:'+dn)
            self.update_conf_file(dn)
        elif str(status)=='registration_error':
            self.logger.info('[Registration] Registration is failed. New registration request will send')
            self.re_register()
            self.registration_request()
        elif str(status)=='already_exists':
            self.update_conf_file(dn)
            self.logger.info('[Registration] Ahenk already registered')


    def update_conf_file(self,dn=None):
        self.logger.debug('[Registration] Registration configuration is updating...')
        if self.conf_manager.has_section('CONNECTION') and self.conf_manager.get('REGISTRATION', 'from') is not None:
            self.conf_manager.set('CONNECTION', 'uid',self.conf_manager.get('REGISTRATION', 'from'))
            self.conf_manager.set('CONNECTION', 'password',self.conf_manager.get('REGISTRATION', 'password'))
            self.conf_manager.set('REGISTRATION', 'dn',dn)
            self.conf_manager.set('REGISTRATION', 'registered','true')
            with open('/etc/ahenk/ahenk.conf', 'w') as configfile:
                self.conf_manager.write(configfile)
            self.logger.debug('[Registration] Registration configuration file is updated')

    def is_registered(self):
        if self.conf_manager.has_section('REGISTRATION') and (self.conf_manager.get('REGISTRATION', 'registered')=='true'):
            self.logger.debug('registered')
            return True
        else:
            self.logger.debug('not registered')
            return False

    def is_ldap_registered(self):
        if self.is_registered() and self.conf_manager.get('REGISTRATION', 'dn')!='' and self.conf_manager.get('REGISTRATION', 'dn') is not None:
            return True
        else:
            return False

    def get_registration_request_message(self):
        if self.conf_manager.has_section('REGISTRATION'):
            return self.message_manager.registration_msg()
        else:
            self.logger.error('[Registration] Registration section not found while trying to registration request')
            return None

    def get_ldap_registration_request_message(self):
        if self.conf_manager.has_section('REGISTRATION'):
            return self.message_manager.ldap_registration_msg()
        else:
            self.logger.error('[Registration] Registration section not found while trying to ldap registration request')
            return None

    def register(self,uuid_depend_mac):
        if self.conf_manager.has_section('REGISTRATION'):
            self.logger.info('[Registration] Registration section is already created')
        else:
            self.logger.debug('[Registration] Creating Registration section')
            self.conf_manager.add_section('REGISTRATION')
            self.conf_manager.set('REGISTRATION', 'from',str(self.generate_uuid(uuid_depend_mac)))
            self.conf_manager.set('REGISTRATION', 'macAddresses',str(':'.join(("%012X" % get_mac())[i:i+2] for i in range(0, 12, 2))))
            self.conf_manager.set('REGISTRATION', 'ipAddresses',str(self.get_ipAddresses()))
            self.conf_manager.set('REGISTRATION', 'hostname',str(socket.gethostname()))
            self.conf_manager.set('REGISTRATION', 'timestamp',str(datetime.datetime.now().strftime("%d-%m-%Y %I:%M")))
            self.conf_manager.set('REGISTRATION', 'password',str(self.generate_password()))
            self.conf_manager.set('REGISTRATION', 'dn','')
            self.conf_manager.set('REGISTRATION', 'registered','false')

            #TODO self.conf_manager.configurationFilePath attribute error ? READ olacak
            self.logger.debug('[Registration] Parameters were set up, section will write to configuration file')
            with open('/etc/ahenk/ahenk.conf', 'w') as configfile:
                self.conf_manager.write(configfile)
            self.logger.debug('[Registration] REGISTRATION section wrote to configuration file successfully')

    def unregister(self):
        self.logger.debug('[Registration] Ahenk is unregistering...')
        if self.conf_manager.has_section('REGISTRATION'):
            #TODO open this block if you want to be aware about unregistration
            #message_sender=MessageSender(self.message_manager.unregister_msg(),None)
            #message_sender.connect_to_server()

            self.conf_manager.remove_section('REGISTRATION')
            self.conf_manager.set('CONNECTION', 'uid','')
            self.conf_manager.set('CONNECTION', 'password','')
            with open('/etc/ahenk/ahenk.conf', 'w') as configfile:
                self.conf_manager.write(configfile)
            self.logger.debug('[Registration] Ahenk is unregistered')

    def re_register(self):
        self.logger.debug('[Registration] Reregistrating...')
        self.unregister()
        self.register(False)

    def generate_uuid(self,depend_mac=True):
        if depend_mac is False:
            self.logger.debug('[Registration] uuid creating randomly')
            return uuid.uuid4() # make a random UUID
        else:
            self.logger.debug('[Registration] uuid creating according to mac address')
            return uuid.uuid3(uuid.NAMESPACE_DNS, str(get_mac()))# make a UUID using an MD5 hash of a namespace UUID and a mac address

    def generate_password(self):
        return uuid.uuid4()

    def get_ipAddresses(self):
        self.logger.debug('[Registration] looking for network interfaces')
        ip_address=""
        for interface in netifaces.interfaces():
            if(str(interface) != "lo"):
                ip_address+=str(netifaces.ifaddresses(interface)[netifaces.AF_INET])
        self.logger.debug('[Registration] returning ip addresses from every interfaces')
        return ip_address