#!/usr/bin/python
#coding utf8

import os
import sys
import time

def getRuningProgramPids(progname):
    pids = []
    lines = os.popen('ps -ef | grep %s' % progname).readlines()
    for l in lines:
        line = ''
        oldc = ''
        for c in l.strip():
            if c in [' ', '\t'] and c == oldc:
                continue
            oldc = c
            line += c
        line = line.split(' ')

        if line[7] == './%s' % progname:
            pids.append(line[1])
    return pids

def getExePath(pid):
    return os.popen('ls -l /proc/%d | grep "exe ->" | cut -d " " -f 7-' % int(pid)).read()

def getExeCmdLine(pid):
    return os.popen('cat /proc/%d/cmdline' % int(pid)).read()

def stop(instance_id):
    roomserver_ids = getRuningProgramPids('roomserver')
    pids = roomserver_ids
    for pid in pids:
        exepath = getExePath(pid)
        cmdline = getExeCmdLine(pid)
        if cmdline == ("./roomserver\0-i\0%d\0" % instance_id):
            os.popen('kill -9 %d' % int(pid))

def listserver():
    roomserver_ids = getRuningProgramPids('roomserver')
    pids = roomserver_ids
    for pid in pids:
        exepath = getExePath(pid)
        cmdline = getExeCmdLine(pid)
        print pid, exepath, cmdline

def restartServer(str_instance_ids):
    instance_ids = str_instance_ids.split(',')
    for instance_id in instance_ids:
        instance_id = int(instance_id)
        stop(instance_id)
        time.sleep(0.5)
        print 'roomserver %d starting......' % instance_id
        cmd = 'sh start_instance.sh %d' % (instance_id)
        os.popen(cmd)
        time.sleep(0.5)

def printHelp():
    print 'usuage: [restart]'

def main(argv):
    if len(argv) == 1:
        printHelp()
    else:
        if argv[1] == 'restart':
            restartServer(argv[2])
        elif argv[1] == 'stop':
            stop()
        elif argv[1] == 'list':
            listserver()

if __name__ == '__main__':
    main(sys.argv)