#-*- coding: utf-8 -*-

# Copyright 2008-2010 Calculate Ltd. http://www.calculate-linux.org
#
#  Licensed under the Apache License, Version 2.0 (the "License");
#  you may not use this file except in compliance with the License.
#  You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
#  Unless required by applicable law or agreed to in writing, software
#  distributed under the License is distributed on an "AS IS" BASIS,
#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#  See the License for the specific language governing permissions and
#  limitations under the License.

import os
import sys
import re
# Вывод в строку ввода
import readline
# Файловый объект из строки
import cStringIO
import termios
# Для ввода символа
import tty
# Работа со временем
import time

from cl_utils import getpathenv, runOsCommand

"""общие функции серверных программ"""

def unicList(listEl):
    """Уникальный список с сохранением порядка"""
    retList = []
    [not x in retList and retList.append(x) for x in listEl]
    return retList


def dialogYn(message):
    """Вывод сообщения, ожидание нажатия Y или n

    если Y - True если n - False"""

    def getChar():
        fd = sys.stdin.fileno()
        oldSet = termios.tcgetattr(fd)
        tty.setraw(fd)
        char = sys.stdin.read(1)
        termios.tcsetattr(fd, termios.TCSADRAIN, oldSet)
        return char

    def term(char):
        if ord(char) == 3:
            return None
        if char == "Y":
            return True
        elif char == "n":
            return False
        else:
            char = getChar()
            return term(char)

    sys.stdout.write(message + ":")
    char = getChar()
    res = term(char)
    sys.stdout.write("\n")
    return res


def dialogYesNo(message):
    """Вывод сообщения, ожидание набора Yes или No (в любом регистре)

    если Yes - True, если No - False"""
    sys.stdout.write(message + ": ")
    strIn=sys.stdin.readline().lower().strip()
    sys.stdout.write("\n")
    if "yes" == strIn:
        return True
    elif "no" == strIn:
        return False
    else:
        return dialogYesNo(message)

def rawInput(promptText="", inputText=""):
    """Создает поле ввода

    promptText - текст перед полем ввода
    inputText - текст в поле ввода
    """
    if inputText:
        # Записываем текст для последующего вывода в строке ввода
        readline.set_pre_input_hook(lambda:\
                                readline.insert_text(inputText) or\
                                readline.redisplay())
    strInput = ""
    if promptText:
        # Получаем текст введенный пользователем
        strInput = raw_input(promptText)
    else:
        strInput = raw_input()
    if inputText:
        # Сбрасываем строку ввода
        readline.set_pre_input_hook(None)
    return strInput

def execProg(cmdStrProg, inStr=False, retFull=True, envProg={}):
    """Выполняет внешнюю программу

    Параметры:
    cmdStrProg     внешняя программа
    inStr          данные передаваемые программе на страндартный вход.
    Возвращаемые параметры:
    строка которую выведет внешняя программа или False в случае ошибки
    """
    env_path = {"PATH":getpathenv()}
    env = {}
    env.update(os.environ.items() + env_path.items() + envProg.items())
    retCode,programOut = runOsCommand(cmdStrProg,inStr,retFull,env)
    if not retCode:
        return programOut
    return False

def genSleep(timeSleep=(0.2, 0.4, 0.8)):
    """Генератор задержек"""
    for t in timeSleep:
        time.sleep(t)
        yield(t)

def stringIsJpeg(string):
    """Определяет является ли строка - jpeg изображением"""
    if len(string)<8:
        return False
    FD = cStringIO.StringIO(string)
    isJpeg = False
    FD.seek(0, 0)
    (firstByte, secondByte) = FD.read(2)
    if (ord(firstByte) == 0xff and ord(secondByte) == 0xd8):
        (firstByte, secondByte) = FD.read(2)
        if (ord(firstByte) == 0xff and ord(secondByte) == 0xe0):
            isJpeg = True
    return isJpeg

def isCorrectStringNet(strNetworks, checkNet=True):
    """Проверяет на корректность строку доверительных сетей

    Выводит cписок сетей
    """
    splNet = strNetworks.replace(","," ").split(" ")
    if checkNet:
        checkIP = False
        res=re.compile("^\d\d?\d?\.\d\d?\d?\.\d\d?\d?\.\d\d?\d?\/\d\d?$")
    else:
        checkIP = True
        res=re.compile("^\d\d?\d?\.\d\d?\d?\.\d\d?\d?\.\d\d?\d?$")
    flagError = False
    networks = []
    for i in splNet:
        r = i.strip()
        if not r:
            continue
        find =res.search(r)
        if not find:
            flagError = True
            break
        else:
            splIP = map(lambda x: 255>=int(x.split("/")[0]) and\
                                x.split("/")[0], find.group().split("."))
            if not splIP[0] or splIP[0] and int(splIP[0]) == 0:
                flagError = True
                break
            if not splIP[3] or splIP[3] and int(splIP[3]) == 255:
                flagError = True
                break
            if checkNet:
                netList = r.split("/")
                if len(netList)==2:
                    try:
                        netMaskInt = int(netList[1])
                    except:
                        flagError = True
                        break
                    if netMaskInt>31 or netMaskInt<4:
                        flagError = True
                        break
                else:
                    flagError = True
                    break
            if checkIP and splIP[3] and int(splIP[3]) == 0:
                flagError = True
                break
            for t in splIP:
                if t == False:
                    flagError = True
                    break
        if flagError:
            break
        networks.append(r)
    if flagError:
        return False
    else:
        return list(set(networks))

def searchLineInFile(name, fileName, numEl=0):
    """Ищет строку в которой есть название разделенное ':'

    в файле похожем на /etc/passwd"""
    if os.path.exists(fileName):
        FD = open(fileName)
        lines = FD.readlines()
        FD.close()
        lineFound = ""
        for line in lines:
            if name == line.split(":")[numEl]:
                lineFound = line
                break
    if lineFound:
        return lineFound
    else:
        return False

def getMaxInFile(fileName, numMin, numMax, numEl=2):
    """Получаем максимальный номер из файла похожего на /etc/group"""
    lst = []
    lst.append(0)
    if os.path.exists(fileName):
        FD = open(fileName)
        lines = FD.readlines()
        FD.close()
        for line in lines:
            if not ':' in line:
                continue
            num = int(line.split(":")[numEl])
            if num<=numMax and num>=numMin:
                lst.append(num)
        return max(lst)
    return False

def chortToFullName(listNames, domain):
        """Из списка коротких имен получаем cписок полных имен

        К коротким именам добавляем домен, длинные выдаем как есть
        """
        listFillNames = []
        for name in listNames:
            if "." in name:
                listFillNames.append(name)
            else:
                listFillNames.append("%s.%s" %(name,domain))
        return listFillNames

def addInfoUser(name, uid, gid, comment):
        """Добавляем информацию о пользователе"""
        class user():
            """Информация о пользователе"""
            name = ""
            uid = ""
            gid = ""
            comment = ""
        us = user()
        us.name = name
        us.uid = uid
        us.gid = gid
        us.comment = comment
        return us

def addInfoGroup(name, gid, comment, rid="", type="", sid=""):
        """Добавляем информацию о группе"""
        class group():
            """Информация о группе"""
            name = ""
            gid = ""
            comment = ""
            rid = ""
            type = ""
            sid = ""
        gr = group()
        gr.name = name
        gr.gid = gid
        gr.comment = comment
        gr.rid = rid
        gr.type = type
        gr.sid = sid
        return gr