# This software may be freely redistributed under the terms of the GNU
# general public license.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA

import netifaces
import platform
import os
import socket
import re
import subprocess
import uuid as _uuid
from hashlib import md5 as _md5

from phacter.utils import linux as utils

def ipaddress():
    return netifaces.ifaddresses('eth0')[2][0]['addr']
def fqdn():
    return socket.gethostname()
def processorcount():
    return os.sysconf('SC_NPROCESSORS_ONLN')

def architecture():
    arch = platform.processor()
    if re.compile('i[3456]86').match(arch):
        return 'i386'
    else:
        return arch

def interfaces():
    i = netifaces.interfaces()
    return ",".join([n for n in i if 'lo' not in n])

def domain():
    return ".".join(socket.gethostname().split('.')[1:])
def hostname():
    return socket.gethostname().split('.')[0]
def hardwareisa():
    return platform.processor()
def hardwaremodel():
    return platform.processor()
def id():
    return os.environ['USER']
def pythonversion():
    return platform.python_version()
def kernelrelease():
    return platform.uname()[2]
def kernelversion():
    return platform.uname()[2].split('-')[0]
def _lsbcheck(func):
    s = subprocess.Popen(['/usr/bin/which', 'lsb_release'],stderr=subprocess.PIPE,stdout=subprocess.PIPE)
    if s.stdout.read().strip():
        return func
    return None
@_lsbcheck
def lsbdistcodename():
    return os.popen('lsb_release -c').read().strip().replace('\t','').split(':')[1]
@_lsbcheck
def lsbdistdescription():
    return os.popen('lsb_release -d').read().strip().replace('\t','').split(':')[1]
@_lsbcheck
def lsbdistid():
    return os.popen('lsb_release -i').read().strip().replace('\t','').split(':')[1]
@_lsbcheck
def lsbdistrelease():
    return os.popen('lsb_release -r').read().strip().replace('\t','').split(':')[1]
@_lsbcheck
def lsbrelease():
    lsb_release = os.popen('lsb_release -v').read().strip().replace('\t','').split(':')[1:]
    return ":".join(lsb_release)
@_lsbcheck
def lsbmajdistrelease():
    return os.popen('lsb_release -r').read().strip().replace('\t','').split(':')[1].split('.')[0]
def ps():
    return 'ps -ef'
def permid():
    m = _md5.new(netifaces.ifaddresses('eth0')[17][0]['addr'].lower())
    return m.hexdigest()
def uuid():
    return str(_uuid.uuid4())
def operatingsystem():
    bdistid = lsbdistid()
    if bdistid == 'Ubuntu':
        return bdistid
    elif os.path.isfile('/etc/debian_version'):
        return 'Debian'
    elif os.path.isfile('/etc/gentoo-release'):
        return 'Gentoo'
    elif os.path.isfile('/etc/fedora-release'):
        return 'Fedora'
    elif os.path.isfile('/etc/mandriva-release'):
        return 'Mandriva'
    elif os.path.isfile('/etc/mandrake-release'):
        return 'Mandrake'
    elif os.path.isfile('/etc/redhat-release'):
        f = open('/etc/redhat-release','r')
        if re.compile('centos',re.I).search(f.read()):
            return 'CentOS'
        else:
            return 'RedHat'
        f.close()
    elif os.path.isfile('/etc/SuSE-release'):
        f = open('/etc/SuSE-release','r')
        if re.compile('SUSE LINUX Enterprise Server',re.I).search(f.read()):
            return 'SLES'
        else:
            return 'SuSE'

def operatingsystemrelease():
    os = operatingsystem()
    if os == 'Debian':
        f = open('/etc/debian_version','r')
        v = f.read()
        f.close()
        return v.strip()
    elif os == 'Gentoo':
        f = open('/etc/gentoo-release','r')
        v = f.read() 
        f.close()
        return v.strip()
    elif os == 'Fedora':
        f = open('/etc/fedora-release','r')
        for line in f.readlines():
            if 'Rawhide' in line:
                value = 'Rawhide'
            elif 'release' in line:
                value = line
            else:
                value = 'Unknown'
        f.close()
        return value.strip()
    elif os.path.isfile('/etc/mandriva-release'):
        return 'Mandriva'
    elif os.path.isfile('/etc/mandrake-release'):
        return 'Mandrake'
    elif os == 'CentOS':
        f = open('/etc/redhat-release','r')
        f.close()
        return '5.2'
    elif os == 'RedHat':
        f = open('/etc/redhat-release','r')
        for line in f.readlines():
            if 'Rawhide' in line:
                value = 'Rawhide'
            elif 'release' in line:
                value = line
            else:
                value = 'Unknown'
        f.close()
        return value.strip()
    elif os.path.isfile('/etc/SuSE-release'):
        f = open('/etc/SuSE-release','r')
        if re.compile('SUSE LINUX Enterprise Server',re.I).search(f.read()):
            return 'SLES'
        else:
            return 'SuSE'

def manufacturer():
    dmi_decode = '/usr/sbin/dmidecode'
    if os.path.isfile(dmi_decode):
        try:
            p = subprocess.run(
                dmi_decode, 
                shell=False, 
                timeout=5, 
                check=True, 
                capture_output=True
            )
            for line in p.stdout.decode('utf-8').split('\n'):
                if 'Manufacturer' in line:
                    m = line.split(':')[1].strip()
                    return m
        except:
            pass

#def dealeraddress():
#    return ":".join(utils.getDealerAddress())



#ipaddress_eth0
#macaddress
#macaddress_eth0
#macaddress_eth1
#memoryfree
#memorysize
#netmask
#netmask_eth0
#operatingsystemrelease
#processor0
#processor1
#rubysitedir
#sshdsakey
#sshrsakey
#swapfree
#swapsize
#virtual