1132 lines
30 KiB
Python
Executable File
1132 lines
30 KiB
Python
Executable File
# coding: utf-8
|
||
|
||
|
||
import time
|
||
import os
|
||
import sys
|
||
import re
|
||
import psutil
|
||
|
||
web_dir = os.getcwd() + "/web"
|
||
if os.path.exists(web_dir):
|
||
sys.path.append(web_dir)
|
||
os.chdir(web_dir)
|
||
|
||
import core.mw as mw
|
||
|
||
|
||
|
||
|
||
app_debug = False
|
||
if mw.isAppleSystem():
|
||
app_debug = True
|
||
|
||
|
||
def getPluginName():
|
||
return 'gogs'
|
||
|
||
|
||
def getPluginDir():
|
||
return mw.getPluginDir() + '/' + getPluginName()
|
||
|
||
|
||
def getServerDir():
|
||
return mw.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(':', 1)
|
||
tmp[t[0]] = t[1]
|
||
elif args_len > 1:
|
||
for i in range(len(args)):
|
||
t = args[i].split(':', 1)
|
||
tmp[t[0]] = t[1]
|
||
|
||
return tmp
|
||
|
||
|
||
def checkArgs(data, ck=[]):
|
||
for i in range(len(ck)):
|
||
if not ck[i] in data:
|
||
return (False, mw.returnJson(False, '参数:(' + ck[i] + ')没有!'))
|
||
return (True, mw.returnJson(True, 'ok'))
|
||
|
||
|
||
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 = mw.execShell(
|
||
"ps -ef|grep " + getPluginName() + " |grep -v grep | grep -v python | awk '{print $2}'")
|
||
if data[0] == '':
|
||
return 'stop'
|
||
return 'start'
|
||
|
||
|
||
def getHomeDir():
|
||
if mw.isAppleSystem():
|
||
user = mw.execShell(
|
||
"who | sed -n '2, 1p' |awk '{print $1}'")[0].strip()
|
||
return '/Users/' + user
|
||
else:
|
||
return '/root'
|
||
|
||
|
||
def getRunUser():
|
||
if mw.isAppleSystem():
|
||
user = mw.execShell("who | sed -n '2, 1p' |awk '{print $1}'")[0].strip()
|
||
return user
|
||
else:
|
||
return 'root'
|
||
|
||
__SR = '''#!/bin/bash
|
||
PATH=/bin:/sbin:/usr/bin:/usr/sbin:/usr/local/bin:/usr/local/sbin:~/bin
|
||
export PATH
|
||
export USER=%s
|
||
export HOME=%s && ''' % ( getRunUser(), getHomeDir())
|
||
|
||
|
||
def contentReplace(content):
|
||
|
||
service_path = mw.getServerDir()
|
||
content = content.replace('{$ROOT_PATH}', mw.getFatherDir())
|
||
content = content.replace('{$SERVER_PATH}', service_path)
|
||
content = content.replace('{$RUN_USER}', getRunUser())
|
||
content = content.replace('{$HOME_DIR}', getHomeDir())
|
||
|
||
return content
|
||
|
||
|
||
def initDreplace():
|
||
|
||
file_tpl = getInitdConfTpl()
|
||
service_path = mw.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 = mw.readFile(file_tpl)
|
||
content = contentReplace(content)
|
||
mw.writeFile(file_bin, content)
|
||
mw.execShell('chmod +x ' + file_bin)
|
||
|
||
# conf_bin = getConf()
|
||
# if not os.path.exists(conf_bin):
|
||
# mw.execShell('mkdir -p ' + getServerDir() + '/custom/conf')
|
||
# conf_tpl = getConfTpl()
|
||
# content = mw.readFile(conf_tpl)
|
||
# content = contentReplace(content)
|
||
# mw.writeFile(conf_bin, content)
|
||
|
||
# systemd
|
||
systemDir = mw.systemdCfgDir()
|
||
systemService = systemDir + '/gogs.service'
|
||
systemServiceTpl = getPluginDir() + '/init.d/gogs.service.tpl'
|
||
if os.path.exists(systemDir) and not os.path.exists(systemService):
|
||
service_path = mw.getServerDir()
|
||
se_content = mw.readFile(systemServiceTpl)
|
||
se_content = se_content.replace('{$SERVER_PATH}', service_path)
|
||
mw.writeFile(systemService, se_content)
|
||
mw.execShell('systemctl daemon-reload')
|
||
|
||
log_path = getServerDir() + '/log'
|
||
if not os.path.exists(log_path):
|
||
os.mkdir(log_path)
|
||
|
||
return file_bin
|
||
|
||
|
||
def getRootUrl():
|
||
content = mw.readFile(getConf())
|
||
rep = r'ROOT_URL\\s*=\\s*(.*)'
|
||
tmp = re.search(rep, content)
|
||
if tmp:
|
||
return tmp.groups()[0]
|
||
|
||
rep = r'EXTERNAL_URL\\s*=\s*(.*)'
|
||
tmp = re.search(rep, content)
|
||
if tmp:
|
||
return tmp.groups()[0]
|
||
return ''
|
||
|
||
|
||
def getSshPort():
|
||
content = mw.readFile(getConf())
|
||
rep = r'SSH_PORT\\s*=\\s*(.*)'
|
||
tmp = re.search(rep, content)
|
||
if not tmp:
|
||
return ''
|
||
return tmp.groups()[0]
|
||
|
||
|
||
def getHttpPort():
|
||
content = mw.readFile(getConf())
|
||
rep = r'HTTP_PORT\\s*=\\s*(.*)'
|
||
tmp = re.search(rep, content)
|
||
if not tmp:
|
||
return ''
|
||
return tmp.groups()[0]
|
||
|
||
|
||
def getRootPath():
|
||
content = mw.readFile(getConf())
|
||
rep = r'ROOT\\s*=\\s*(.*)'
|
||
tmp = re.search(rep, content)
|
||
if not tmp:
|
||
return ''
|
||
return tmp.groups()[0]
|
||
|
||
|
||
def getDbConfValue():
|
||
conf = getConf()
|
||
if not os.path.exists(conf):
|
||
return {}
|
||
|
||
content = mw.readFile(conf)
|
||
rep_scope = r"\\[database\\](.*?)\\["
|
||
tmp = re.findall(rep_scope, content, re.S)
|
||
|
||
rep = r'(\\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):
|
||
host = conf['HOST'].split(':')
|
||
# pymysql
|
||
db = mw.getMyORM()
|
||
# MySQLdb |
|
||
# db = mw.getMyORMDb()
|
||
|
||
db.setPort(int(host[1]))
|
||
db.setUser(conf['USER'])
|
||
|
||
if 'PASSWD' in conf:
|
||
db.setPwd(conf['PASSWD'])
|
||
else:
|
||
db.setPwd(conf['PASSWORD'])
|
||
|
||
db.setDbName(conf['NAME'])
|
||
# db.setSocket(getSocketFile())
|
||
db.setCharset("utf8")
|
||
return db
|
||
|
||
|
||
def pSqliteDb(conf):
|
||
# print(conf)
|
||
import db
|
||
psDb = db.Sql()
|
||
|
||
# 默认
|
||
gsdir = getServerDir() + '/data'
|
||
dbname = 'gogs'
|
||
if conf['PATH'][0] == '/':
|
||
# 绝对路径
|
||
pass
|
||
else:
|
||
path = conf['PATH'].split('/')
|
||
gsdir = getServerDir() + '/' + path[0]
|
||
dbname = path[1].split('.')[0]
|
||
|
||
# print(gsdir, dbname)
|
||
psDb.dbPos(gsdir, dbname)
|
||
return psDb
|
||
|
||
|
||
def getGogsDbType(conf):
|
||
|
||
if 'DB_TYPE' in conf:
|
||
return conf['DB_TYPE']
|
||
|
||
if 'TYPE' in conf:
|
||
return conf['TYPE']
|
||
|
||
return 'NONE'
|
||
|
||
|
||
def pQuery(sql):
|
||
conf = getDbConfValue()
|
||
gtype = getGogsDbType(conf)
|
||
if gtype == 'sqlite3':
|
||
db = pSqliteDb(conf)
|
||
data = db.query(sql, []).fetchall()
|
||
return data
|
||
elif gtype == 'mysql':
|
||
db = pMysqlDb(conf)
|
||
return db.query(sql)
|
||
|
||
print("仅支持mysql|sqlite3配置")
|
||
exit(0)
|
||
|
||
|
||
def isSqlError(mysqlMsg):
|
||
# 检测数据库执行错误
|
||
_mysqlMsg = str(mysqlMsg)
|
||
# print _mysqlMsg
|
||
if "MySQLdb" in _mysqlMsg:
|
||
return mw.returnData(False, 'MySQLdb组件缺失! <br>进入SSH命令行输入： pip install mysql-python')
|
||
if "2002," in _mysqlMsg:
|
||
return mw.returnData(False, '数据库连接失败,请检查数据库服务是否启动!')
|
||
if "using password:" in _mysqlMsg:
|
||
return mw.returnData(False, '数据库管理密码错误!')
|
||
if "Connection refused" in _mysqlMsg:
|
||
return mw.returnData(False, '数据库连接失败,请检查数据库服务是否启动!')
|
||
if "1133," in _mysqlMsg:
|
||
return mw.returnData(False, '数据库用户不存在!')
|
||
if "1007," in _mysqlMsg:
|
||
return mw.returnData(False, '数据库已经存在!')
|
||
if "1044," in _mysqlMsg:
|
||
return mw.returnData(False, mysqlMsg[1])
|
||
if "2003," in _mysqlMsg:
|
||
return mw.returnData(False, "Can't connect to MySQL server on '127.0.0.1' (61)")
|
||
return mw.returnData(True, 'OK')
|
||
|
||
|
||
def gogsOp(method):
|
||
file = initDreplace()
|
||
|
||
if not mw.isAppleSystem():
|
||
data = mw.execShell('systemctl ' + method + ' gogs')
|
||
if data[1] == '':
|
||
return 'ok'
|
||
return 'fail'
|
||
|
||
data = mw.execShell(__SR + file + ' ' + method)
|
||
if data[1] == '':
|
||
return 'ok'
|
||
return data[0]
|
||
|
||
|
||
def start():
|
||
return gogsOp('start')
|
||
|
||
|
||
def stop():
|
||
return gogsOp('stop')
|
||
|
||
|
||
def restart():
|
||
return gogsOp('restart')
|
||
|
||
|
||
def reload():
|
||
return gogsOp('reload')
|
||
|
||
|
||
def initdStatus():
|
||
if mw.isAppleSystem():
|
||
return "Apple Computer does not support"
|
||
|
||
shell_cmd = 'systemctl status gogs | grep loaded | grep "enabled;"'
|
||
data = mw.execShell(shell_cmd)
|
||
if data[0] == '':
|
||
return 'fail'
|
||
return 'ok'
|
||
|
||
|
||
def initdInstall():
|
||
if mw.isAppleSystem():
|
||
return "Apple Computer does not support"
|
||
|
||
mw.execShell('systemctl enable gogs')
|
||
return 'ok'
|
||
|
||
|
||
def initdUinstall():
|
||
if mw.isAppleSystem():
|
||
return "Apple Computer does not support"
|
||
|
||
mw.execShell('systemctl disable gogs')
|
||
return 'ok'
|
||
|
||
|
||
def runLog():
|
||
log_path = getServerDir() + '/log/gogs.log'
|
||
return log_path
|
||
|
||
|
||
def postReceiveLog():
|
||
log_path = getServerDir() + '/log/hooks/post-receive.log'
|
||
return log_path
|
||
|
||
|
||
def getGogsConf():
|
||
conf = getConf()
|
||
if not os.path.exists(conf):
|
||
return mw.returnJson(False, "请先安装初始化!<br/>默认地址:http://" + mw.getLocalIp() + ":3000")
|
||
|
||
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 = mw.readFile(conf)
|
||
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 mw.returnJson(True, 'OK', 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 = mw.readFile(filename)
|
||
for g in gets:
|
||
if g in args:
|
||
rep = g + '\\s*=\\s*(.*)'
|
||
val = g + ' = ' + args[g]
|
||
conf = re.sub(rep, val, conf)
|
||
mw.writeFile(filename, conf)
|
||
reload()
|
||
return mw.returnJson(True, '设置成功')
|
||
|
||
|
||
def gogsEditTpl():
|
||
data = {}
|
||
data['post_receive'] = getPluginDir() + '/hook/post-receive.tpl'
|
||
data['commit'] = getPluginDir() + '/hook/commit.tpl'
|
||
return mw.getJson(data)
|
||
|
||
|
||
def userList():
|
||
|
||
conf = getConf()
|
||
if not os.path.exists(conf):
|
||
return mw.returnJson(False, "请先安装初始化!<br/>默认地址:http://" + mw.getLocalIp() + ":3000")
|
||
|
||
conf = getDbConfValue()
|
||
gtype = getGogsDbType(conf)
|
||
if gtype != 'mysql':
|
||
return mw.returnJson(False, "仅支持mysql数据操作!")
|
||
|
||
import math
|
||
args = getArgs()
|
||
|
||
data = checkArgs(args, ['page', 'page_size'])
|
||
if not data[0]:
|
||
return data[1]
|
||
|
||
page = int(args['page'])
|
||
page_size = int(args['page_size'])
|
||
search = ''
|
||
if 'search' in args:
|
||
search = args['search']
|
||
|
||
user_where1 = ''
|
||
user_where2 = ''
|
||
if search != '':
|
||
user_where1 = ' where name like "%' + search + '%"'
|
||
user_where2 = ' where name like "%' + search + '%"'
|
||
|
||
data = {}
|
||
|
||
data['root_url'] = getRootUrl()
|
||
|
||
start = (page - 1) * page_size
|
||
list_count = pQuery('select count(id) as num from user' + user_where1)
|
||
count = list_count[0]["num"]
|
||
list_data = pQuery(
|
||
'select id,name,email from user ' + user_where2 + ' order by id desc limit ' + str(start) + ',' + str(page_size))
|
||
data['list'] = mw.getPage({'count': count, 'p': page,
|
||
'row': page_size, 'tojs': 'gogsUserList'})
|
||
data['page'] = page
|
||
data['page_size'] = page_size
|
||
data['page_count'] = int(math.ceil(count / page_size))
|
||
data['data'] = list_data
|
||
return mw.returnJson(True, 'OK', data)
|
||
|
||
|
||
def checkRepoListIsHasScript(data):
|
||
path = getRootPath()
|
||
for x in range(len(data)):
|
||
name = data[x]['name'] + '/' + data[x]['repo'] + '.git'
|
||
path_tmp = path + '/' + name + '/custom_hooks/post-receive'
|
||
if os.path.exists(path_tmp):
|
||
data[x]['has_hook'] = True
|
||
else:
|
||
data[x]['has_hook'] = False
|
||
return data
|
||
|
||
|
||
def repoList():
|
||
|
||
conf = getConf()
|
||
if not os.path.exists(conf):
|
||
return mw.returnJson(False, "请先安装初始化!<br/>默认地址:http://" + mw.getLocalIp() + ":3000")
|
||
|
||
conf = getDbConfValue()
|
||
gtype = getGogsDbType(conf)
|
||
if gtype != 'mysql':
|
||
return mw.returnJson(False, "仅支持mysql数据操作!")
|
||
|
||
import math
|
||
args = getArgs()
|
||
|
||
data = checkArgs(args, ['page', 'page_size'])
|
||
if not data[0]:
|
||
return data[1]
|
||
|
||
page = int(args['page'])
|
||
page_size = int(args['page_size'])
|
||
search = ''
|
||
if 'search' in args:
|
||
search = args['search']
|
||
|
||
data = {}
|
||
|
||
data['root_url'] = getRootUrl()
|
||
|
||
repo_where1 = ''
|
||
repo_where2 = ''
|
||
if search != '':
|
||
repo_where1 = ' where name like "%' + search + '%"'
|
||
repo_where2 = ' where r.name like "%' + search + '%"'
|
||
|
||
start = (page - 1) * page_size
|
||
list_count = pQuery(
|
||
'select count(id) as num from repository' + repo_where1)
|
||
count = list_count[0]["num"]
|
||
sql = 'select r.id,r.owner_id,r.name as repo, u.name from repository r left join user u on r.owner_id=u.id ' + repo_where2 + ' order by r.id desc limit ' + \
|
||
str(start) + ',' + str(page_size)
|
||
# print(sql)
|
||
list_data = pQuery(sql)
|
||
# print(list_data)
|
||
list_data = checkRepoListIsHasScript(list_data)
|
||
|
||
data['list'] = mw.getPage({'count': count, 'p': page, 'row': page_size, 'tojs': 'gogsRepoListPage'})
|
||
data['page'] = page
|
||
data['page_size'] = page_size
|
||
data['page_count'] = int(math.ceil(count / page_size))
|
||
data['data'] = list_data
|
||
return mw.returnJson(True, 'OK', data)
|
||
|
||
|
||
def getAllUserProject(user, search=''):
|
||
path = getRootPath() + '/' + user
|
||
dlist = []
|
||
if os.path.exists(path):
|
||
for filename in os.listdir(path):
|
||
tmp = {}
|
||
filePath = path + '/' + filename
|
||
if os.path.isdir(filePath):
|
||
if search == '':
|
||
tmp['name'] = filename.replace('.git', '')
|
||
dlist.append(tmp)
|
||
else:
|
||
if filename.find(search) != -1:
|
||
tmp['name'] = filename.replace('.git', '')
|
||
dlist.append(tmp)
|
||
return dlist
|
||
|
||
|
||
def checkProjectListIsHasScript(user, data):
|
||
path = getRootPath() + '/' + user
|
||
for x in range(len(data)):
|
||
name = data[x]['name'] + '.git'
|
||
path_tmp = path + '/' + name + '/custom_hooks/post-receive'
|
||
if os.path.exists(path_tmp):
|
||
data[x]['has_hook'] = True
|
||
else:
|
||
data[x]['has_hook'] = False
|
||
return data
|
||
|
||
|
||
def userProjectList():
|
||
import math
|
||
args = getArgs()
|
||
# print args
|
||
|
||
page = 1
|
||
page_size = 5
|
||
search = ''
|
||
|
||
if not 'name' in args:
|
||
return mw.returnJson(False, '缺少参数name')
|
||
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']
|
||
|
||
data = {}
|
||
|
||
ulist = getAllUserProject(args['name'])
|
||
dlist_sum = len(ulist)
|
||
|
||
start = (page - 1) * page_size
|
||
ret_data = ulist[start:start + page_size]
|
||
ret_data = checkProjectListIsHasScript(args['name'], ret_data)
|
||
|
||
data['root_url'] = getRootUrl()
|
||
data['data'] = ret_data
|
||
data['args'] = args
|
||
data['list'] = mw.getPage(
|
||
{'count': dlist_sum, 'p': page, 'row': page_size, 'tojs': 'userProjectListPage'})
|
||
|
||
return mw.returnJson(True, 'OK', data)
|
||
|
||
|
||
def projectScriptEdit():
|
||
args = getArgs()
|
||
|
||
if not 'user' in args:
|
||
return mw.returnJson(True, 'username missing')
|
||
|
||
if not 'name' in args:
|
||
return mw.returnJson(True, 'project name missing')
|
||
|
||
user = args['user']
|
||
name = args['name'] + '.git'
|
||
post_receive = getRootPath() + '/' + user + '/' + name + \
|
||
'/custom_hooks/commit'
|
||
if os.path.exists(post_receive):
|
||
return mw.returnJson(True, 'OK', {'path': post_receive})
|
||
else:
|
||
return mw.returnJson(False, 'file does not exist')
|
||
|
||
|
||
def projectScriptLoad():
|
||
args = getArgs()
|
||
if not 'user' in args:
|
||
return mw.returnJson(True, 'username missing')
|
||
|
||
if not 'name' in args:
|
||
return mw.returnJson(True, 'project name missing')
|
||
|
||
user = args['user']
|
||
name = args['name'] + '.git'
|
||
|
||
path = getRootPath() + '/' + user + '/' + name
|
||
post_receive_tpl = getPluginDir() + '/hook/post-receive.tpl'
|
||
post_receive = path + '/custom_hooks/post-receive'
|
||
|
||
if not os.path.exists(path + '/custom_hooks'):
|
||
mw.execShell('mkdir -p ' + path + '/custom_hooks')
|
||
|
||
pct_content = mw.readFile(post_receive_tpl)
|
||
pct_content = pct_content.replace('{$PATH}', path + '/custom_hooks')
|
||
mw.writeFile(post_receive, pct_content)
|
||
mw.execShell('chmod 777 ' + post_receive)
|
||
|
||
commit_tpl = getPluginDir() + '/hook/commit.tpl'
|
||
commit = path + '/custom_hooks/commit'
|
||
|
||
codeDir = mw.getFatherDir() + '/git'
|
||
|
||
cc_content = mw.readFile(commit_tpl)
|
||
|
||
gitPath = getRootPath()
|
||
cc_content = cc_content.replace('{$GITROOTURL}', gitPath)
|
||
cc_content = cc_content.replace('{$CODE_DIR}', codeDir)
|
||
cc_content = cc_content.replace('{$USERNAME}', user)
|
||
cc_content = cc_content.replace('{$PROJECT}', args['name'])
|
||
cc_content = cc_content.replace('{$WEB_ROOT}', mw.getWwwDir())
|
||
mw.writeFile(commit, cc_content)
|
||
mw.execShell('chmod 777 ' + commit)
|
||
|
||
return 'ok'
|
||
|
||
|
||
def projectScriptUnload():
|
||
args = getArgs()
|
||
data = checkArgs(args, ['user', 'name'])
|
||
if not data[0]:
|
||
return data[1]
|
||
|
||
user = args['user']
|
||
name = args['name'] + '.git'
|
||
|
||
post_receive = getRootPath() + '/' + user + '/' + name + \
|
||
'/custom_hooks/post-receive'
|
||
mw.execShell('rm -f ' + post_receive)
|
||
|
||
commit = getRootPath() + '/' + user + '/' + name + \
|
||
'/custom_hooks/commit'
|
||
mw.execShell('rm -f ' + commit)
|
||
return 'ok'
|
||
|
||
|
||
def projectScriptDebug():
|
||
args = getArgs()
|
||
data = checkArgs(args, ['user', 'name'])
|
||
if not data[0]:
|
||
return data[1]
|
||
|
||
user = args['user']
|
||
name = args['name'] + '.git'
|
||
commit_log = getRootPath() + '/' + user + '/' + name + \
|
||
'/custom_hooks/sh.log'
|
||
|
||
data = {}
|
||
if os.path.exists(commit_log):
|
||
data['status'] = True
|
||
data['path'] = commit_log
|
||
else:
|
||
data['status'] = False
|
||
data['msg'] = '没有日志文件'
|
||
|
||
return mw.getJson(data)
|
||
|
||
|
||
def projectScriptRun():
|
||
args = getArgs()
|
||
data = checkArgs(args, ['user', 'name'])
|
||
if not data[0]:
|
||
return data[1]
|
||
|
||
user = args['user']
|
||
name = args['name'] + '.git'
|
||
|
||
path = getRootPath() + '/' + user + '/' + name
|
||
commit_sh = path + '/custom_hooks/commit'
|
||
commit_log = path + '/custom_hooks/sh.log'
|
||
script_run = 'sh -x ' + commit_sh + ' 2>' + commit_log
|
||
|
||
if not os.path.exists(commit_sh):
|
||
return mw.returnJson(False, '脚本文件不存在!')
|
||
|
||
mw.execShell(script_run)
|
||
return mw.returnJson(True, '脚本文件执行成功,观察日志!')
|
||
|
||
|
||
def projectScriptSelf():
|
||
args = getArgs()
|
||
data = checkArgs(args, ['user', 'name'])
|
||
if not data[0]:
|
||
return data[1]
|
||
|
||
user = args['user']
|
||
name = args['name'] + '.git'
|
||
|
||
custom_hooks = getRootPath() + '/' + user + '/' + \
|
||
name + '/custom_hooks'
|
||
|
||
self_path = custom_hooks + '/self'
|
||
if not os.path.exists(self_path):
|
||
os.mkdir(self_path)
|
||
|
||
self_logs_path = custom_hooks + '/self_logs'
|
||
if not os.path.exists(self_logs_path):
|
||
os.mkdir(self_logs_path)
|
||
|
||
self_hook_file = custom_hooks + '/self_hook.sh'
|
||
self_hook_exist = False
|
||
if os.path.exists(self_hook_file):
|
||
self_hook_exist = True
|
||
|
||
dlist = []
|
||
if os.path.exists(self_path):
|
||
for filename in os.listdir(self_path):
|
||
tmp = {}
|
||
filePath = self_path + '/' + filename
|
||
if os.path.isfile(filePath):
|
||
tmp['path'] = filePath
|
||
tmp['name'] = os.path.basename(filePath)
|
||
tmp['is_hidden'] = False
|
||
if tmp['name'].endswith('.txt'):
|
||
tmp['is_hidden'] = True
|
||
|
||
dlist.append(tmp)
|
||
|
||
dlist_sum = len(dlist)
|
||
# print(dlist)
|
||
rdata = {}
|
||
rdata['data'] = dlist
|
||
rdata['self_hook'] = self_hook_exist
|
||
rdata['list'] = mw.getPage(
|
||
{'count': dlist_sum, 'p': 1, 'row': 100, 'tojs': 'self_page'})
|
||
|
||
return mw.returnJson(True, 'ok', rdata)
|
||
|
||
|
||
def projectScriptSelf_Create():
|
||
args = getArgs()
|
||
data = checkArgs(args, ['user', 'name', 'file'])
|
||
if not data[0]:
|
||
return data[1]
|
||
|
||
user = args['user']
|
||
name = args['name'] + '.git'
|
||
file = args['file']
|
||
|
||
self_path = path = getRootPath() + '/' + user + '/' + name + '/custom_hooks/self'
|
||
|
||
if not os.path.exists(self_path):
|
||
os.mkdir(self_path)
|
||
|
||
abs_file = self_path + '/' + file + '.sh'
|
||
if os.path.exists(abs_file):
|
||
return mw.returnJson(False, '脚本已经存在!')
|
||
|
||
mw.writeFile(abs_file, "#!/bin/bash\necho `date +'%Y-%m-%d %H:%M:%S'`\n")
|
||
|
||
rdata = {}
|
||
rdata['abs_file'] = abs_file
|
||
return mw.returnJson(True, '创建文件成功!', rdata)
|
||
|
||
|
||
def projectScriptSelf_Del():
|
||
args = getArgs()
|
||
data = checkArgs(args, ['user', 'name', 'file'])
|
||
if not data[0]:
|
||
return data[1]
|
||
|
||
user = args['user']
|
||
name = args['name'] + '.git'
|
||
file = args['file']
|
||
|
||
custom_hooks = getRootPath() + '/' + user + '/' + \
|
||
name + '/custom_hooks'
|
||
self_path = custom_hooks + '/self'
|
||
|
||
if not os.path.exists(self_path):
|
||
os.mkdir(self_path)
|
||
|
||
abs_file = self_path + '/' + file
|
||
# print(abs_file)
|
||
if not os.path.exists(abs_file):
|
||
return mw.returnJson(False, '脚本已经删除!')
|
||
|
||
os.remove(abs_file)
|
||
|
||
# 日志也删除
|
||
log_file = custom_hooks + '/self_logs/' + file + '.log'
|
||
if os.path.exists(log_file):
|
||
os.remove(log_file)
|
||
|
||
return mw.returnJson(True, '脚本删除成功!')
|
||
|
||
|
||
def projectScriptSelf_Logs():
|
||
args = getArgs()
|
||
data = checkArgs(args, ['user', 'name', 'file'])
|
||
if not data[0]:
|
||
return data[1]
|
||
|
||
user = args['user']
|
||
name = args['name'] + '.git'
|
||
file = args['file']
|
||
|
||
self_path = path = getRootPath() + '/' + user + '/' + \
|
||
name + '/custom_hooks/self_logs'
|
||
|
||
if not os.path.exists(self_path):
|
||
os.mkdir(self_path)
|
||
|
||
logs_file = self_path + '/' + file + '.log'
|
||
if os.path.exists(logs_file):
|
||
rdata = {}
|
||
rdata['path'] = logs_file
|
||
return mw.returnJson(True, 'ok', rdata)
|
||
|
||
return mw.returnJson(False, '日志不存在!')
|
||
|
||
|
||
def projectScriptSelf_Run():
|
||
args = getArgs()
|
||
data = checkArgs(args, ['user', 'name', 'file'])
|
||
if not data[0]:
|
||
return data[1]
|
||
|
||
user = args['user']
|
||
name = args['name'] + '.git'
|
||
file = args['file']
|
||
|
||
custom_hooks = getRootPath() + '/' + user + '/' + \
|
||
name + '/custom_hooks'
|
||
self_path = custom_hooks + '/self/' + file
|
||
self_logs_path = custom_hooks + '/self_logs/' + file + '.log'
|
||
|
||
shell = "sh -x " + self_path + " 2>" + self_logs_path + ' &'
|
||
mw.execShell(shell)
|
||
return mw.returnJson(True, '执行成功!')
|
||
|
||
|
||
def projectScriptSelf_Rename():
|
||
args = getArgs()
|
||
data = checkArgs(args, ['user', 'name', 'o_file', 'n_file'])
|
||
if not data[0]:
|
||
return data[1]
|
||
|
||
user = args['user']
|
||
name = args['name'] + '.git'
|
||
o_file = args['o_file']
|
||
n_file = args['n_file']
|
||
|
||
custom_hooks = getRootPath() + '/' + user + '/' + \
|
||
name + '/custom_hooks'
|
||
self_path = custom_hooks + '/self'
|
||
|
||
if not os.path.exists(self_path):
|
||
os.mkdir(self_path)
|
||
|
||
o_file_abs = self_path + '/' + o_file + '.sh'
|
||
if not os.path.exists(o_file_abs):
|
||
return mw.returnJson(False, '原文件已经不存在了!')
|
||
|
||
n_file_abs = self_path + '/' + n_file + '.sh'
|
||
|
||
os.rename(o_file_abs, n_file_abs)
|
||
|
||
# 日志也删除
|
||
log_file = custom_hooks + '/self_logs/' + o_file + '.sh.log'
|
||
if os.path.exists(log_file):
|
||
os.remove(log_file)
|
||
|
||
return mw.returnJson(True, '重命名成功!')
|
||
|
||
|
||
def projectScriptSelf_Enable():
|
||
args = getArgs()
|
||
data = checkArgs(args, ['user', 'name', 'enable'])
|
||
if not data[0]:
|
||
return data[1]
|
||
|
||
user = args['user']
|
||
name = args['name'] + '.git'
|
||
enable = args['enable']
|
||
|
||
custom_path = getRootPath() + '/' + user + '/' + \
|
||
name + '/custom_hooks'
|
||
|
||
# 替换commit配置
|
||
commit_path = custom_path + '/commit'
|
||
note = '#Gogs Script Don`t Remove and Change'
|
||
|
||
self_file = custom_path + '/self_hook.sh'
|
||
self_hook_tpl = getPluginDir() + '/hook/self_hook.tpl'
|
||
|
||
if enable == '1':
|
||
content = mw.readFile(self_hook_tpl)
|
||
content = content.replace('{$HOOK_DIR}', custom_path + '/self')
|
||
content = content.replace(
|
||
'{$HOOK_LOGS_DIR}', custom_path + '/self_logs')
|
||
mw.writeFile(self_file, content)
|
||
mw.execShell("chmod 777 " + self_file)
|
||
|
||
commit_content = mw.readFile(commit_path)
|
||
commit_content += "\n\n" + "bash " + self_file + " " + note
|
||
mw.writeFile(commit_path, commit_content)
|
||
|
||
return mw.returnJson(True, '开启成功!')
|
||
else:
|
||
commit_content = mw.readFile(commit_path)
|
||
rep = ".*" + note
|
||
commit_content = re.sub(rep, '', commit_content, re.M)
|
||
commit_content = commit_content.strip()
|
||
mw.writeFile(commit_path, commit_content)
|
||
if os.path.exists(self_file):
|
||
os.remove(self_file)
|
||
return mw.returnJson(True, '关闭成功!')
|
||
|
||
|
||
def projectScriptSelf_Status():
|
||
args = getArgs()
|
||
data = checkArgs(args, ['user', 'name', 'file', 'status'])
|
||
if not data[0]:
|
||
return data[1]
|
||
|
||
user = args['user']
|
||
name = args['name'] + '.git'
|
||
file = args['file']
|
||
status = args['status']
|
||
|
||
custom_hooks = getRootPath() + '/' + user + '/' + \
|
||
name + '/custom_hooks'
|
||
self_path = custom_hooks + '/self'
|
||
|
||
if not os.path.exists(self_path):
|
||
os.mkdir(self_path)
|
||
|
||
# 日志也删除
|
||
log_file = custom_hooks + '/self_logs/' + file + '.log'
|
||
if os.path.exists(log_file):
|
||
os.remove(log_file)
|
||
|
||
if status == '1':
|
||
file_abs = self_path + '/' + file
|
||
file_text_abs = self_path + '/' + file + '.txt'
|
||
os.rename(file_abs, file_text_abs)
|
||
return mw.returnJson(True, '开始禁用成功!')
|
||
else:
|
||
file_abs = self_path + '/' + file.strip('.txt')
|
||
file_text_abs = self_path + '/' + file
|
||
os.rename(file_text_abs, file_abs)
|
||
return mw.returnJson(True, '开始使用成功!')
|
||
|
||
return mw.returnJson(True, '禁用成功!')
|
||
|
||
|
||
def getRsaPublic():
|
||
path = getHomeDir()
|
||
path += '/.ssh/id_rsa.pub'
|
||
|
||
content = mw.readFile(path)
|
||
|
||
data = {}
|
||
data['mw'] = content
|
||
return mw.getJson(data)
|
||
|
||
|
||
def getTotalStatistics():
|
||
st = status()
|
||
data = {}
|
||
if st.strip() == 'start':
|
||
list_count = pQuery('select count(id) as num from repository')
|
||
count = list_count[0]["num"]
|
||
data['status'] = True
|
||
data['count'] = count
|
||
data['ver'] = mw.readFile(getServerDir() + '/version.pl').strip()
|
||
return mw.returnJson(True, 'ok', data)
|
||
|
||
data['status'] = False
|
||
data['count'] = 0
|
||
return mw.returnJson(False, 'fail', data)
|
||
|
||
|
||
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 == 'post_receive_log':
|
||
print(postReceiveLog())
|
||
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 == 'gogs_edit_tpl':
|
||
print(gogsEditTpl())
|
||
elif func == 'user_list':
|
||
print(userList())
|
||
elif func == 'repo_list':
|
||
print(repoList())
|
||
elif func == 'user_project_list':
|
||
print(userProjectList())
|
||
elif func == 'project_script_edit':
|
||
print(projectScriptEdit())
|
||
elif func == 'project_script_load':
|
||
print(projectScriptLoad())
|
||
elif func == 'project_script_unload':
|
||
print(projectScriptUnload())
|
||
elif func == 'project_script_debug':
|
||
print(projectScriptDebug())
|
||
elif func == 'project_script_run':
|
||
print(projectScriptRun())
|
||
elif func == 'project_script_self':
|
||
print(projectScriptSelf())
|
||
elif func == 'project_script_self_create':
|
||
print(projectScriptSelf_Create())
|
||
elif func == 'project_script_self_del':
|
||
print(projectScriptSelf_Del())
|
||
elif func == 'project_script_self_logs':
|
||
print(projectScriptSelf_Logs())
|
||
elif func == 'project_script_self_run':
|
||
print(projectScriptSelf_Run())
|
||
elif func == 'project_script_self_rename':
|
||
print(projectScriptSelf_Rename())
|
||
elif func == 'project_script_self_enable':
|
||
print(projectScriptSelf_Enable())
|
||
elif func == 'project_script_self_status':
|
||
print(projectScriptSelf_Status())
|
||
elif func == 'get_rsa_public':
|
||
print(getRsaPublic())
|
||
elif func == 'get_total_statistics':
|
||
print(getTotalStatistics())
|
||
else:
|
||
print('fail')
|