spring-website/github-webhook.py at master · Tim-Blokdijk/spring-website · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
#!/usr/bin/env python


"""
	a simple webserver that waits for a post and does a git pull then
"""
import time
import cgi
import BaseHTTPServer
import json
import subprocess
import os

HOST_NAME = '0.0.0.0'
ALLOWED_IPS = { '207.97.227.253', '50.57.128.197', '108.171.174.178', '204.232.175.75' }
PORT_NUMBER = 9999
GIT_REPO = os.path.expanduser('~/www')
LOG_FILE = os.path.expanduser('~/webhook.log')

f = open(LOG_FILE, 'a')

def log(msg, s = {}):
	f.write("%s %s\n" % (time.asctime(), str(msg)))
	f.flush()

class MyHandler(BaseHTTPServer.BaseHTTPRequestHandler):
	def log_message(self, format,*args):
		log(self.client_address[0] +" "+format %( args))
	def is_allowed(self):
		if not self.client_address[0] in ALLOWED_IPS:
			self.send_response(403)
			self.send_header("Content-type", "text/html")
			self.end_headers()
			self.wfile.write("Access denied\n")
			self.log_message("Access denied from %s", (self.client_address[0]))
			return False
		return True

	def do_HEAD(s):
		if not s.is_allowed():
			return
		s.send_response(200)
		s.send_header("Content-type", "text/html")
		s.end_headers()
	def do_GET(s):
		if not s.is_allowed():
			return
		"""Respond to a GET request."""
		s.send_response(200)
		s.send_header("Content-type", "text/html")
		s.end_headers()
		s.wfile.write("<html><head><title>Title goes here.</title></head>")
		s.wfile.write("<body><p>This is a test.</p>")
		# If someone went to "http://something.somewhere.net/foo/bar/",
		# then s.path equals "/foo/bar/".
		s.wfile.write("<p>You accessed path: %s</p>" % s.path)
		s.wfile.write("</body></html>")
	def do_POST(self):
		if not self.is_allowed():
			return

		if self.path != "/github-webhook":
			self.log_message("Invalid path")
			return
		# Parse the form data posted
		form = cgi.FieldStorage(
			fp=self.rfile,
			headers=self.headers,
			environ={'REQUEST_METHOD':'POST',
				'CONTENT_TYPE':self.headers['Content-Type'],
			})

		# Begin the response
		self.send_response(200)
		self.end_headers()
		self.wfile.write('Path: %s\n' % self.path)
		self.wfile.write('Form data:\n')
		if not 'payload' in form:
			self.log_message("Invalid request")
			return
		data = json.loads(form['payload'].value)
		if 'repository' in data and 'name' in data['repository'] and data['repository']['name'] == 'spring-website':
			self.checkout(self)
		#else:
		#	dumpPost(form)
	def dumpPost(self, form):

		# Echo back information about what was posted in the form
		for field in form.keys():
			field_item = form[field]
			if field_item.filename:
				# The field contains an uploaded file
				file_data = field_item.file.read()
				file_len = len(file_data)
				del file_data
				self.wfile.write('\tUploaded %s (%d bytes)\n' % (field,
							 file_len))
			else:
				# Regular form value
				self.wfile.write('\t%s=%s\n' % (field, form[field].value))
				self.log_message('\t%s=%s\n' % (field, form[field].value))

	def checkout(self, s):
		""" update the git repository """
		origWD = os.getcwd()
		os.chdir(GIT_REPO)
		output = ""
		try:
			output = subprocess.check_output(["git", "pull"])
		except CalledProcessError as e:
			output = "process failed: " + str(e.output)
		except Exception as e:
			output = "Exception: %s", str(e)
		os.chdir(origWD)
		output = output.replace("\n", "")
		self.log_message("git pull: " + output)

if __name__ == '__main__':
	server_class = BaseHTTPServer.HTTPServer
	try:
		httpd = server_class((HOST_NAME, PORT_NUMBER), MyHandler)
		log("Server Starts - %s:%s" % (HOST_NAME, PORT_NUMBER))
		httpd.serve_forever()
	except KeyboardInterrupt:
		httpd.server_close()
		pass
	log("Server Stops - %s:%s" % (HOST_NAME, PORT_NUMBER))
	f.close