120 lines
5.2 KiB
Python
Executable File
120 lines
5.2 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import ast # Needed for localhost cmd strings
|
|
import json
|
|
import os
|
|
import re
|
|
import sys
|
|
modules = {}
|
|
try:
|
|
import git
|
|
modules['git'] = True
|
|
except ImportError:
|
|
import subprocess
|
|
modules['git'] = False
|
|
try:
|
|
import paramiko
|
|
import socket
|
|
modules['ssh'] = True
|
|
except ImportError:
|
|
modules['ssh'] = False
|
|
|
|
|
|
|
|
repos = {}
|
|
repos['bdisk'] = {'remotecmds': {'g.rainwreck.com': {'gitbot': {'cmds': ['git -C /var/lib/gitbot/clonerepos/BDisk pull',
|
|
'git -C /var/lib/gitbot/clonerepos/BDisk pull --tags',
|
|
'asciidoctor /var/lib/gitbot/clonerepos/BDisk/docs/manual/HEAD.adoc -o /srv/http/bdisk/index.html']}}}}
|
|
repos['test'] = {'remotecmds': {'g.rainwreck.com': {'gitbot': {'cmds': ['echo $USER']}}}}
|
|
repos['games-site'] = {'remotecmds': {'games.square-r00t.net':
|
|
{'gitbot':
|
|
{'cmds': ['cd /srv/http/games-site && git pull']}}}}
|
|
repos['aif-ng'] = {'cmds': [['asciidoctor', '/opt/git/repo.checkouts/aif-ng/docs/README.adoc', '-o', '/srv/http/aif/index.html']]}
|
|
|
|
def execHook(gitinfo = False):
|
|
if not gitinfo:
|
|
gitinfo = getGitInfo()
|
|
repo = gitinfo['repo'].lower()
|
|
print('Executing hooks for {0}:{1}...'.format(repo, gitinfo['branch']))
|
|
print('This commit: {0}\nLast commit: {1}'.format(gitinfo['currev'], gitinfo['oldrev']))
|
|
# Execute local commands first
|
|
if 'cmds' in repos[repo].keys():
|
|
for cmd in repos[repo]['cmds']:
|
|
print('\tExecuting {0}...'.format(' '.join(cmd)))
|
|
subprocess.call(cmd)
|
|
if 'remotecmds' in repos[repo].keys():
|
|
for host in repos[repo]['remotecmds'].keys():
|
|
if 'port' in repos[repo]['remotecmds'][host].keys():
|
|
port = int(repos[repo]['remotecmds'][host]['port'])
|
|
else:
|
|
port = 22
|
|
for user in repos[repo]['remotecmds'][host].keys():
|
|
print('{0}@{1}:'.format(user, host))
|
|
if paramikomodule:
|
|
ssh = paramiko.SSHClient()
|
|
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
|
|
ssh.connect(host, username = user, port = port)
|
|
try:
|
|
for cmd in repos[repo]['remotecmds'][host][user]['cmds']:
|
|
print('\tExecuting \'{0}\'...'.format(cmd))
|
|
stdin, stdout, stderr = ssh.exec_command(cmd)
|
|
stdout = stdout.read().decode('utf-8')
|
|
stderr = stderr.read().decode('utf-8')
|
|
print(stdout)
|
|
if stderr != '':
|
|
print(stderr)
|
|
except paramiko.AuthenticationException:
|
|
print('({0}@{1}) AUTHENTICATION FAILED!'.format(user, host))
|
|
except paramiko.BadHostKeyException:
|
|
print('({0}@{1}) INCORRECT HOSTKEY!'.format(user, host))
|
|
except paramiko.SSHException:
|
|
print('({0}@{1}) FAILED TO ESTABLISH SSH!'.format(user, host))
|
|
except socket.error:
|
|
print('({0}@{1}) SOCKET CONNECTION FAILURE! (DNS, timeout/firewall, etc.)'.format(user, host))
|
|
else:
|
|
for cmd in repos[repo]['remotecmds'][host][user]['cmds']:
|
|
try:
|
|
print('\tExecuting \'{0}\'...'.format(cmd))
|
|
subprocess.call(['ssh', '{0}@{1}'.format(user, host), cmd])
|
|
except:
|
|
print('({0}@{1}) An error occurred!'.format(user, host))
|
|
|
|
def getGitInfo():
|
|
refs = sys.argv[1].split('/')
|
|
gitinfo = {}
|
|
if refs[1] == 'tags':
|
|
gitinfo['branch'] = False
|
|
gitinfo['tag'] = refs[2]
|
|
elif refs[1] == 'heads':
|
|
gitinfo['branch'] = refs[2]
|
|
gitinfo['tag'] = False
|
|
gitinfo['repo'] = os.environ['GL_REPO']
|
|
gitinfo['user'] = os.environ['GL_USER']
|
|
clientinfo = os.environ['SSH_CONNECTION'].split()
|
|
gitinfo['ssh'] = {'client': {'ip': clientinfo[0], 'port': clientinfo[1]},
|
|
'server': {'ip': clientinfo[2], 'port': clientinfo[3]},
|
|
'user': os.environ['USER']
|
|
}
|
|
if os.environ['GIT_DIR'] == '.':
|
|
gitinfo['dir'] = os.environ['PWD']
|
|
else:
|
|
#gitinfo['dir'] = os.path.join(os.environ['GL_REPO_BASE'], gitinfo['repo'], '.git')
|
|
gitinfo['dir'] = os.path.abspath(os.path.expanduser(os.environ['GIT_DIR']))
|
|
if gitmodule:
|
|
# This is preferred, because it's a lot more faster and a lot more flexible.
|
|
#https://gitpython.readthedocs.io/en/stable
|
|
gitobj = git.Repo(gitinfo['dir'])
|
|
commits = list(gitobj.iter_commits(gitobj.head.ref.name, max_count = 2))
|
|
else:
|
|
commits = subprocess.check_output(['git', 'rev-parse', 'HEAD..HEAD^1']).decode('utf-8').splitlines()
|
|
gitinfo['oldrev'] = re.sub('^\^', '', commits[1])
|
|
gitinfo['currev'] = re.sub('^\^', '', commits[0])
|
|
return(gitinfo)
|
|
#sys.exit(0)
|
|
|
|
def main():
|
|
execHook()
|
|
|
|
if __name__ == '__main__':
|
|
main()
|