# coding:utf-8

import sys
import io
import os
import time

sys.path.append(os.getcwd() + "/class/core")
import public


def status():
    data = public.execShell(
        "ps -ef|grep memcached |grep -v grep | grep -v python | awk '{print $2}'")
    if data[0] == '':
        return 'stop'
    return 'start'


def initDreplace():

    file_tpl = os.getcwd() + '/plugins/memcached/init.d/memcached.tpl'
    file_bin = getConf()

    if os.path.exists(file_bin):
        return file_bin

    content = public.readFile(file_tpl)

    service_path = os.path.dirname(os.getcwd())
    content = content.replace('{$PATH}', service_path)

    public.writeFile(file_bin, content)
    public.execShell('chmod +x ' + file_bin)
    return file_bin


def start():
    file = initDreplace()
    data = public.execShell(file + ' start')
    if data[1] == '':
        return 'ok'
    return 'fail'


def stop():
    file = initDreplace()
    data = public.execShell(file + ' stop')
    if data[1] == '':
        return 'ok'
    return 'fail'


def restart():
    file = initDreplace()
    data = public.execShell(file + ' reload')
    if data[1] == '':
        return 'ok'
    return 'fail'


def reload():
    file = initDreplace()
    data = public.execShell(file + ' reload')
    if data[1] == '':
        return 'ok'
    return 'fail'


def runInfo():
    # 获取memcached状态
    import telnetlib
    import re
    tn = telnetlib.Telnet('127.0.0.1', 11211)
    tn.write(b"stats\n")
    tn.write(b"quit\n")
    data = tn.read_all()
    if type(data) == bytes:
        data = data.decode('utf-8')
    data = data.replace('STAT', '').replace('END', '').split("\n")
    result = {}
    res = ['cmd_get', 'get_hits', 'get_misses', 'limit_maxbytes', 'curr_items', 'bytes',
           'evictions', 'limit_maxbytes', 'bytes_written', 'bytes_read', 'curr_connections']
    for d in data:
        if len(d) < 3:
            continue
        t = d.split()
        if not t[0] in res:
            continue
        result[t[0]] = int(t[1])
    result['hit'] = 1
    if result['get_hits'] > 0 and result['cmd_get'] > 0:
        result['hit'] = float(result['get_hits']) / \
            float(result['cmd_get']) * 100

    conf = public.readFile(getConf())
    result['bind'] = re.search('IP=(.+)', conf).groups()[0]
    result['port'] = int(re.search('PORT=(\d+)', conf).groups()[0])
    result['maxconn'] = int(re.search('MAXCONN=(\d+)', conf).groups()[0])
    result['cachesize'] = int(
        re.search('CACHESIZE=(\d+)', conf).groups()[0])
    return public.getJson(result)


def getConf():
    path = os.getcwd() + "/plugins/memcached/init.d/memcached.tpl"
    return path

if __name__ == "__main__":
    func = sys.argv[1]
    if func == 'run_info':
        print runInfo()
    elif func == 'conf':
        print getConf()
    elif func == 'status':
        print status()
    elif func == 'start':
        print start()
    elif func == 'stop':
        print stop()
    elif func == 'restart':
        print restart()
    elif func == 'reload':
        print reload()