server.py 4.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148
  1. #!/usr/bin/python
  2. # -*- coding: utf-8 -*-
  3. from __future__ import print_function
  4. from BaseHTTPServer import BaseHTTPRequestHandler, HTTPServer
  5. from storage import Storage
  6. import json
  7. import logging
  8. import re
  9. from SocketServer import ThreadingMixIn
  10. class BatcaveHttpRequestHandler(BaseHTTPRequestHandler):
  11. def parse_url_pathquery(self):
  12. url = re.match(r'^\/(?P<path>.+?)(\?(?P<query>.+))?$', self.path)
  13. if url is None:
  14. logging.warn('Failed to parse URL \'' + str(self.path) + '\'.')
  15. return ( None, None )
  16. path = url.group('path')
  17. query = {}
  18. if not url.group('query') is None:
  19. for m in re.finditer(r'(?P<key>.+?)=(?P<value>.+?)(&|$)', url.group('query')):
  20. query[m.group('key')] = m.group('value')
  21. return ( path, query )
  22. def do_GET(self):
  23. path, query = self.parse_url_pathquery()
  24. if path == '':
  25. self.respond_index(query)
  26. return
  27. if path == 'list':
  28. self.respond_list(query)
  29. return
  30. m = re.match(r'node/([a-f0-9]{12})\.json$', path)
  31. if m != None:
  32. self.respond_node(m.group(1))
  33. return
  34. self.send_error(404, 'The URL \'{0}\' was not found here.'.format(path))
  35. def send_nocache_headers(self):
  36. self.send_header('Cache-Control', 'no-cache, no-store, must-revalidate')
  37. self.send_header('Pragma', 'no-cache')
  38. self.send_header('Expires', '0')
  39. def respond_index(self, query):
  40. storagy = self.server.storage
  41. self.send_response(200)
  42. self.send_header('Content-Type', 'text/html; charset=utf-8')
  43. self.send_nocache_headers()
  44. self.end_headers()
  45. self.wfile.write('<!DOCTYPE html><html><head><title>BATCAVE</title></head>\n')
  46. self.wfile.write('<body>\n')
  47. self.wfile.write('<H1 title="Batman/Alfred Transmission Collection, Aggregation & Value Engine">BATCAVE</H1>\n')
  48. self.wfile.write('<p>Dies ist ein interner Hintergrund-Dienst. Er wird nur von anderen Diensten\n')
  49. self.wfile.write('angesprochen und sollte aus einer Mehrzahl von Gr&uuml;nden nicht &ouml;ffentlich\n')
  50. self.wfile.write('zug&auml;nglich sein.</p>\n')
  51. self.wfile.write('<H2>Status</H2>\n')
  52. self.wfile.write('Daten: <span id="datacount" class="value">')
  53. self.wfile.write(len(storage.data))
  54. self.wfile.write('</span>\n')
  55. self.wfile.write('<H2>API</H2>\n')
  56. self.wfile.write('<p>Grundsätzlich ist das Antwort-Format JSON und alle Daten sind Live-Daten (kein Cache) die ggf. etwas Bearbeitungs-Zeit erfordern.</p>')
  57. self.wfile.write('<dl>\n')
  58. self.wfile.write('<dt><a href="/nodes.json">nodes.json</a></dt><dd>zur Verwendung mit ffmap (MACs anonymisiert)</dd>\n')
  59. self.wfile.write('<dt><a href="/node/ff00ff00ff00.json">/node/&lt;id&gt;.json</a></dt><dd><u>alle</u> vorhandenen Information zu der gewünschten Node</dd>\n')
  60. self.wfile.write('</dl>\n')
  61. self.wfile.write('</body></html>')
  62. def respond_list(self, query):
  63. storage = self.server.storage
  64. self.send_response(200)
  65. self.send_header('Content-Type', 'text/html')
  66. self.send_nocache_headers()
  67. self.end_headers()
  68. self.wfile.write('<!DOCTYPE html><html><head><title>BATCAVE</title></head>\n')
  69. self.wfile.write('<body>\n')
  70. self.wfile.write('<H1>BATCAVE - LIST</H1>\n')
  71. self.wfile.write('<table>\n')
  72. self.wfile.write('<thead><tr><th>ID</th><th>Name</th></tr></thead>\n')
  73. self.wfile.write('<tbody>\n')
  74. data = storage.data
  75. if 'sort' in query:
  76. if query['sort'] == 'name':
  77. sorteddata = sorted(data, key=lambda x: data[x]['hostname'].lower())
  78. data = sorteddata
  79. elif query['sort'] == 'id':
  80. sorteddata = sorted(data)
  81. data = sorteddata
  82. for nodeid in data:
  83. nodename = storage.data[nodeid]['hostname'] if 'hostname' in storage.data[nodeid] else '&lt;?&gt;'
  84. self.wfile.write('<tr><td><a href="/node/' + nodeid + '.json">' + nodeid + '</a></td><td>' + nodename + '</td></tr>')
  85. self.wfile.write('</tbody>\n')
  86. self.wfile.write('</table>\n')
  87. def respond_node(self, nodeid):
  88. storage = self.server.storage
  89. if nodeid == 'ff00ff00ff00':
  90. self.send_response(200)
  91. self.send_header('Content-Type', 'text/json')
  92. self.send_nocache_headers()
  93. self.end_headers()
  94. self.wfile.write(json.dumps({
  95. 'name': 'API-Example',
  96. 'nodeid': nodeid,
  97. 'META': 'Dies ist ein minimaler Beispiel-Datensatz. Herzlichen Glückwunsch, du hast das Prinzip der API kapiert.',
  98. }))
  99. return
  100. if not nodeid in storage.data:
  101. self.send_error(404, 'No node with id \'' + nodeid + '\' present.')
  102. return
  103. self.send_response(200)
  104. self.send_header('Content-Type', 'text/json')
  105. self.send_nocache_headers()
  106. self.end_headers()
  107. self.wfile.write(json.dumps(storage.data[nodeid]))
  108. class ApiServer(ThreadingMixIn, HTTPServer):
  109. def __init__(self, endpoint, storage):
  110. HTTPServer.__init__(self, endpoint, BatcaveHttpRequestHandler)
  111. self.storage = storage
  112. def __str__(self):
  113. return 'ApiServer on {0}'.format(self.server_address)
  114. if __name__ == '__main__':
  115. dummystorage = Storage()
  116. server = ApiServer(('0.0.0.0', 8888), dummystorage)
  117. print("Server:", str(server))
  118. server.serve_forever()