379 lines
9.7 KiB
Python
Executable File
379 lines
9.7 KiB
Python
Executable File
# coding: utf-8
|
||
|
||
|
||
import time
|
||
import os
|
||
import sys
|
||
import re
|
||
|
||
sys.path.append("/usr/local/lib/python2.7/site-packages")
|
||
import psutil
|
||
|
||
sys.path.append(os.getcwd() + "/class/core")
|
||
import public
|
||
|
||
|
||
app_debug = False
|
||
if public.getOs() == 'darwin':
|
||
app_debug = True
|
||
|
||
|
||
def getPluginName():
|
||
return 'gogs'
|
||
|
||
|
||
def getPluginDir():
|
||
return public.getPluginDir() + '/' + getPluginName()
|
||
|
||
sys.path.append(getPluginDir() + "/class")
|
||
import mysql
|
||
|
||
|
||
def getServerDir():
|
||
return public.getServerDir() + '/' + getPluginName()
|
||
|
||
|
||
def getInitDFile():
|
||
if app_debug:
|
||
return '/tmp/' + getPluginName()
|
||
return '/etc/init.d/' + getPluginName()
|
||
|
||
|
||
def getArgs():
|
||
args = sys.argv[2:]
|
||
tmp = {}
|
||
args_len = len(args)
|
||
|
||
if args_len == 1:
|
||
t = args[0].strip('{').strip('}')
|
||
t = t.split(':')
|
||
tmp[t[0]] = t[1]
|
||
elif args_len > 1:
|
||
for i in range(len(args)):
|
||
t = args[i].split(':')
|
||
tmp[t[0]] = t[1]
|
||
|
||
return tmp
|
||
|
||
|
||
def getInitdConfTpl():
|
||
path = getPluginDir() + "/init.d/gogs.tpl"
|
||
return path
|
||
|
||
|
||
def getInitdConf():
|
||
path = getServerDir() + "/init.d/gogs"
|
||
return path
|
||
|
||
|
||
def getConf():
|
||
path = getServerDir() + "/custom/conf/app.ini"
|
||
return path
|
||
|
||
|
||
def getConfTpl():
|
||
path = getPluginDir() + "/conf/app.ini"
|
||
return path
|
||
|
||
|
||
def status():
|
||
data = public.execShell(
|
||
"ps -ef|grep " + getPluginName() + " |grep -v grep | grep -v python | awk '{print $2}'")
|
||
if data[0] == '':
|
||
return 'stop'
|
||
return 'start'
|
||
|
||
|
||
def contentReplace(content):
|
||
user = 'root'
|
||
if public.isAppleSystem():
|
||
user = public.execShell(
|
||
"who | sed -n '2, 1p' |awk '{print $1}'")[0].strip()
|
||
content = content.replace('{$HOME_DIR}', '/Users/' + user)
|
||
else:
|
||
content = content.replace('{$HOME_DIR}', '/root')
|
||
|
||
service_path = public.getServerDir()
|
||
content = content.replace('{$ROOT_PATH}', public.getRootDir())
|
||
content = content.replace('{$SERVER_PATH}', service_path)
|
||
content = content.replace('{$RUN_USER}', user)
|
||
|
||
return content
|
||
|
||
|
||
def initDreplace():
|
||
|
||
file_tpl = getInitdConfTpl()
|
||
service_path = public.getServerDir()
|
||
|
||
initD_path = getServerDir() + '/init.d'
|
||
if not os.path.exists(initD_path):
|
||
os.mkdir(initD_path)
|
||
file_bin = initD_path + '/' + getPluginName()
|
||
|
||
if not os.path.exists(file_bin):
|
||
content = public.readFile(file_tpl)
|
||
content = contentReplace(content)
|
||
public.writeFile(file_bin, content)
|
||
public.execShell('chmod +x ' + file_bin)
|
||
|
||
conf_bin = getConf()
|
||
if not os.path.exists(conf_bin):
|
||
public.execShell('mkdir -p ' + getServerDir() + '/custom/conf')
|
||
conf_tpl = getConfTpl()
|
||
content = public.readFile(conf_tpl)
|
||
content = contentReplace(content)
|
||
public.writeFile(conf_bin, content)
|
||
|
||
log_path = getServerDir() + '/log'
|
||
if not os.path.exists(log_path):
|
||
os.mkdir(log_path)
|
||
|
||
return file_bin
|
||
|
||
|
||
def getDbConfValue():
|
||
content = public.readFile(getConf())
|
||
|
||
rep_scope = "\[database\](.*?)\["
|
||
tmp = re.findall(rep_scope, content, re.S)
|
||
|
||
rep = '(\w*)\s*=\s*(.*)'
|
||
tmp = re.findall(rep, tmp[0])
|
||
r = {}
|
||
for x in range(len(tmp)):
|
||
k = tmp[x][0]
|
||
v = tmp[x][1]
|
||
r[k] = v
|
||
return r
|
||
|
||
|
||
def pMysqlDb():
|
||
conf = getDbConfValue()
|
||
|
||
host = conf['HOST'].split(':')
|
||
conn = mysql.mysql()
|
||
|
||
conn.setHost(host[0])
|
||
conn.setUser(conf['USER'])
|
||
conn.setPwd(conf['PASSWD'])
|
||
conn.setPort(int(host[1]))
|
||
conn.setDb(conf['NAME'])
|
||
return conn
|
||
|
||
|
||
def isSqlError(mysqlMsg):
|
||
# 检测数据库执行错误
|
||
_mysqlMsg = str(mysqlMsg)
|
||
# print _mysqlMsg
|
||
if "MySQLdb" in _mysqlMsg:
|
||
return public.returnData(False, 'MySQLdb组件缺失! <br>进入SSH命令行输入： pip install mysql-python')
|
||
if "2002," in _mysqlMsg:
|
||
return public.returnData(False, '数据库连接失败,请检查数据库服务是否启动!')
|
||
if "using password:" in _mysqlMsg:
|
||
return public.returnData(False, '数据库管理密码错误!')
|
||
if "Connection refused" in _mysqlMsg:
|
||
return public.returnData(False, '数据库连接失败,请检查数据库服务是否启动!')
|
||
if "1133," in _mysqlMsg:
|
||
return public.returnData(False, '数据库用户不存在!')
|
||
if "1007," in _mysqlMsg:
|
||
return public.returnData(False, '数据库已经存在!')
|
||
if "1044," in _mysqlMsg:
|
||
return public.returnData(False, mysqlMsg[1])
|
||
if "2003," in _mysqlMsg:
|
||
return public.returnData(False, mysqlMsg[1])
|
||
return public.returnData(True, 'OK')
|
||
|
||
|
||
def start():
|
||
|
||
is_frist = True
|
||
conf_bin = getConf()
|
||
if os.path.exists(conf_bin):
|
||
is_frist = False
|
||
|
||
file = initDreplace()
|
||
|
||
if is_frist:
|
||
return "第一次启动Gogs,默认使用MySQL连接!<br>可以在配置文件中重新设置,再启动!"
|
||
|
||
conn = pMysqlDb()
|
||
list_table = conn.query('show tables')
|
||
data = isSqlError(list_table)
|
||
if not data['status']:
|
||
return data['msg']
|
||
|
||
data = public.execShell(file + ' start')
|
||
if data[1] == '':
|
||
return 'ok'
|
||
return data[0]
|
||
|
||
|
||
def stop():
|
||
file = initDreplace()
|
||
data = public.execShell(file + ' stop')
|
||
if data[1] == '':
|
||
return 'ok'
|
||
return data[1]
|
||
|
||
|
||
def restart():
|
||
file = initDreplace()
|
||
data = public.execShell(file + ' reload')
|
||
if data[1] == '':
|
||
return 'ok'
|
||
return data[1]
|
||
|
||
|
||
def reload():
|
||
file = initDreplace()
|
||
data = public.execShell(file + ' reload')
|
||
if data[1] == '':
|
||
return 'ok'
|
||
return data[1]
|
||
|
||
|
||
def initdStatus():
|
||
if not app_debug:
|
||
os_name = public.getOs()
|
||
if os_name == 'darwin':
|
||
return "Apple Computer does not support"
|
||
initd_bin = getInitDFile()
|
||
if os.path.exists(initd_bin):
|
||
return 'ok'
|
||
return 'fail'
|
||
|
||
|
||
def initdInstall():
|
||
import shutil
|
||
if not app_debug:
|
||
os_name = public.getOs()
|
||
if os_name == 'darwin':
|
||
return "Apple Computer does not support"
|
||
|
||
mem_bin = initDreplace()
|
||
initd_bin = getInitDFile()
|
||
shutil.copyfile(mem_bin, initd_bin)
|
||
public.execShell('chmod +x ' + initd_bin)
|
||
return 'ok'
|
||
|
||
|
||
def initdUinstall():
|
||
if not app_debug:
|
||
os_name = public.getOs()
|
||
if os_name == 'darwin':
|
||
return "Apple Computer does not support"
|
||
initd_bin = getInitDFile()
|
||
os.remove(initd_bin)
|
||
return 'ok'
|
||
|
||
|
||
def runLog():
|
||
log_path = getServerDir() + '/log/gogs.log'
|
||
return log_path
|
||
|
||
|
||
def getGogsConf():
|
||
gets = [
|
||
{'name': 'DOMAIN', 'type': -1, 'ps': '服务器域名'},
|
||
{'name': 'ROOT_URL', 'type': -1, 'ps': '公开的完整URL路径'},
|
||
{'name': 'HTTP_ADDR', 'type': -1, 'ps': '应用HTTP监听地址'},
|
||
{'name': 'HTTP_PORT', 'type': -1, 'ps': '应用 HTTP 监听端口号'},
|
||
|
||
{'name': 'START_SSH_SERVER', 'type': 2, 'ps': '启动内置SSH服务器'},
|
||
{'name': 'SSH_PORT', 'type': -1, 'ps': 'SSH 端口号'},
|
||
|
||
{'name': 'REQUIRE_SIGNIN_VIEW', 'type': 2, 'ps': '强制登录浏览'},
|
||
{'name': 'ENABLE_CAPTCHA', 'type': 2, 'ps': '启用验证码服务'},
|
||
{'name': 'DISABLE_REGISTRATION', 'type': 2, 'ps': '禁止注册,只能由管理员创建帐号'},
|
||
{'name': 'ENABLE_NOTIFY_MAIL', 'type': 2, 'ps': '是否开启邮件通知'},
|
||
|
||
{'name': 'FORCE_PRIVATE', 'type': 2, 'ps': '强制要求所有新建的仓库都是私有'},
|
||
|
||
{'name': 'SHOW_FOOTER_BRANDING', 'type': 2, 'ps': 'Gogs推广信息'},
|
||
{'name': 'SHOW_FOOTER_VERSION', 'type': 2, 'ps': 'Gogs版本信息'},
|
||
{'name': 'SHOW_FOOTER_TEMPLATE_LOAD_TIME', 'type': 2, 'ps': 'Gogs模板加载时间'},
|
||
]
|
||
conf = public.readFile(getConf())
|
||
result = []
|
||
|
||
for g in gets:
|
||
rep = g['name'] + '\s*=\s*(.*)'
|
||
tmp = re.search(rep, conf)
|
||
if not tmp:
|
||
continue
|
||
g['value'] = tmp.groups()[0]
|
||
result.append(g)
|
||
return public.getJson(result)
|
||
|
||
|
||
def submitGogsConf():
|
||
gets = ['DOMAIN', 'ROOT_URL', 'HTTP_ADDR',
|
||
'HTTP_PORT', 'START_SSH_SERVER', 'SSH_PORT',
|
||
'REQUIRE_SIGNIN_VIEW', 'FORCE_PRIVATE',
|
||
'ENABLE_CAPTCHA', 'DISABLE_REGISTRATION', 'ENABLE_NOTIFY_MAIL',
|
||
'SHOW_FOOTER_BRANDING', 'SHOW_FOOTER_VERSION',
|
||
'SHOW_FOOTER_TEMPLATE_LOAD_TIME']
|
||
args = getArgs()
|
||
filename = getConf()
|
||
conf = public.readFile(filename)
|
||
for g in gets:
|
||
if g in args:
|
||
rep = g + '\s*=\s*(.*)'
|
||
val = g + ' = ' + args[g]
|
||
conf = re.sub(rep, val, conf)
|
||
public.writeFile(filename, conf)
|
||
reload()
|
||
return public.returnJson(True, '设置成功')
|
||
|
||
|
||
def userList():
|
||
args = getArgs()
|
||
page = 1
|
||
page_size = 10
|
||
search = ''
|
||
if 'page' in args:
|
||
page = int(args['page'])
|
||
|
||
if 'page_size' in args:
|
||
page_size = int(args['page_size'])
|
||
|
||
if 'search' in args:
|
||
search = args['search']
|
||
|
||
return public.returnJson(True, '设置成功')
|
||
|
||
|
||
if __name__ == "__main__":
|
||
func = sys.argv[1]
|
||
if func == 'status':
|
||
print status()
|
||
elif func == 'start':
|
||
print start()
|
||
elif func == 'stop':
|
||
print stop()
|
||
elif func == 'restart':
|
||
print restart()
|
||
elif func == 'reload':
|
||
print reload()
|
||
elif func == 'initd_status':
|
||
print initdStatus()
|
||
elif func == 'initd_install':
|
||
print initdInstall()
|
||
elif func == 'initd_uninstall':
|
||
print initdUinstall()
|
||
elif func == 'run_log':
|
||
print runLog()
|
||
elif func == 'conf':
|
||
print getConf()
|
||
elif func == 'init_conf':
|
||
print getInitdConf()
|
||
elif func == 'get_gogs_conf':
|
||
print getGogsConf()
|
||
elif func == 'submit_gogs_conf':
|
||
print submitGogsConf()
|
||
elif func == 'user_list':
|
||
print userList()
|
||
else:
|
||
print 'fail'
|