server.py

#!/bin/bash
# -*- coding: utf-8 -*-
# this file includes basic minecraft server class functions

from subprocess import Popen, PIPE
import select
import fcntl, os
import time
import sys
import traceback
import threading
import mcdplugin
from mcdlog import *
import serverinfoparser

stop_flag = 0

def notice():
  print('thanks for using MCDaemon,it\'s open source and u can find it here:')
  print('https://github.com/kafuuchino-desu/MCDaemon')
  print('please notice that this software is still in alpha version,it may not work well')
  print('this software is maintained by chino_desu,welcome for your issues and PRs')
  
def listplugins(plugins):
  result = ''
  result = result + 'loaded plugins:\n'
  for singleplugin in plugins.plugins:
    result = result +str(singleplugin) + '\n'
  result = result +'loaded startup plugins:\n'
  for singleplugin in plugins.startupPlugins:
    result = result +str(singleplugin) + '\n'
  result = result + 'loaded onPlayerJoin plugins:\n'
  for singleplugin in plugins.onPlayerJoinPlugins:
    result = result + str(singleplugin) + '\n'
  result = result +'loaded onPlayerLeavePlugins plugins:\n'
  for singleplugin in plugins.onPlayerLeavePlugins:
    result = result +str(singleplugin) + '\n'
  return result

def getInput(server):
  inp = ''
  while True:
    inp = raw_input()
    if inp != '' :
      if inp == 'stop':
        server.cmdstop()
      elif inp == 'MCDReload':
        print('reloading')
        plugins.initPlugins()
        plugins_inf = listplugins(plugins)
        for singleline in plugins_inf.splitlines():
          print(singleline)
      else:
        server.execute(inp)

class Server(object):
  def __init__(self):
    self.start()

  def start(self):
    self.process = Popen('./start.sh', stdin=PIPE, stdout=PIPE, stderr=PIPE, shell=True)
    flags = fcntl.fcntl(self.process.stdout, fcntl.F_GETFL)
    fcntl.fcntl(self.process.stdout, fcntl.F_SETFL, flags | os.O_NONBLOCK)
    log('Server Running at PID:'+str(self.process.pid))

  def tick(self):
    try:
      global stop_flag
      receive = self.recv()
      if receive != '':
        print(receive)
        for line in receive.splitlines():
          if line[11:].startswith('[Server Shutdown Thread/INFO]: Stopping server') or line[11:].startswith('[Server thread/INFO]: Stopping server'): #sometimes this two message will only show one of them
            if stop_flag > 0:
              log('Plugin called a reboot')
            else:
              log('Server stopped by itself.Exiting...')
              sys.exit(0)
            stop_flag -= 1
          if line[11:].startswith('[Server Watchdog/FATAL]: A single server tick'):
            exitlog('single tick took too long for server and watchdog forced the server off', 1)
            sys.exit(0)
          result = serverinfoparser.parse(line)
          if (result.isPlayer == 1) and (result.content == '!!MCDReload'):
            try:
              self.say('[MCDaemon] :Reloading plugins')
              plugins.initPlugins()
              plugins_inf = listplugins(plugins)
              for singleline in plugins_inf.splitlines():
                server.say(singleline)
            except:
              server.say('error initalizing plugins,check console for detailed information')
              errlog('error initalizing plugins,printing traceback.', traceback.format_exc())
          elif (result.isPlayer == 0) and(result.content.endswith('joined the game')):
            player = result.content.split(' ')[0]
            for singleplugin in plugins.onPlayerJoinPlugins:
              try:
                t =threading.Thread(target=singleplugin.onPlayerJoin,args=(server, player))
                t.setDaemon(True)
                t.start()
              except:
                errlog('error processing plugin: ' + str(singleplugin), traceback.format_exc())
          elif (result.isPlayer == 0) and(result.content.endswith('left the game')):
            player = result.content.split(' ')[0]
            for singleplugin in plugins.onPlayerLeavePlugins:
              try:
                t =threading.Thread(target=singleplugin.onPlayerLeave,args=(server, player))
                t.setDaemon(True)
                t.start()
              except:
                errlog('error processing plugin: ' + str(singleplugin), traceback.format_exc())
          for singleplugin in plugins.plugins:
            t =threading.Thread(target=self.callplugin,args=(result, singleplugin))
            t.setDaemon(True)
            t.start()
        time.sleep(1)
    except (KeyboardInterrupt, SystemExit):
      self.stop()
      sys.exit(0) 

  def send(self, data): #send a string to STDIN
    self.process.stdin.write(data)
    self.process.stdin.flush()

  def execute(self, data , tail = '\n'): #puts a command in STDIN with \n to execute
    self.send(data + tail)

  def recv(self, t=0.1): #returns latest STDOUT
    r = ''
    pr = self.process.stdout
    while True:
      if not select.select([pr], [], [], 0.1)[0]:
        time.sleep(t)
        continue
      r = pr.read()
      return r.rstrip()
    return r.rstrip()

  def cmdstop(self): #stop the server using command
    self.send('stop\n')

  def forcestop(self): #stop the server using pclose, donnt use it until necessary
    try:
      self.process.kill()
    except:
      raise RuntimeError
      
  def stop(self):
    global stop_flag
    stop_flag = 2
    self.cmdstop()
    try:
      self.forcestop()
      log('forced server to stop')
    except:
      pass
    
  def say(self, data):
    self.execute('tellraw @a {"text":"' + str(data) + '"}')

  def tell(self, player, data):
    self.execute('tellraw '+ player + ' {"text":"' + str(data) + '"}')
    
  def callplugin(self, result, plugin):
    try:
      plugin.onServerInfo(self, result)
    except:
      errlog('error processing plugin: ' + str(plugin), traceback.format_exc())
    

if __name__ == "__main__":
  notice()
  log('initalizing plugins')
  try:
    import mcdplugin
    plugins = mcdplugin.mcdplugin()
    plugins_inf = listplugins(plugins)
    print(plugins_inf)
  except:
    errlog('error initalizing plugins,printing traceback.', traceback.format_exc())
    sys.exit(0)
  try:
    server = Server()
  except:
    exitlog('failed to initalize the server.', 1, traceback.format_exc())
    sys.exit(0)
  for singleplugin in plugins.startupPlugins:
    try:
      t =threading.Thread(target=singleplugin.onServerStartup,args=(server, ))
      t.setDaemon(True)
      t.start()
    except:
      errlog('error initalizing startup plugins,printing traceback.', traceback.format_exc())
  cmd =threading.Thread(target=getInput,args=(server, ))
  cmd.setDaemon(True)
  cmd.start()
  while True:
    try:
      server.tick()
    except (SystemExit,IOError):
      log('server stopped')
      sys.exit(0)
    except:
      errlog('error ticking MCD')
      print(traceback.format_exc())
      server.stop()
      sys.exit(0)