initial commit

server.py

@@ -0,0 +1,131 @@
+import argparse
+import uuid
+import json
+import socket
+import os
+
+
+# non standards, in requirements.txt
+from flask import Flask, request, Markup, render_template, redirect, url_for
+import requests
+import github
+
+
+
+
+class Anonymous_Github:
+    def __init__(self, github_token="0e8f5af6801d89e533f5c045920e928b4535d41e", host="127.0.0.1", port=5000,
+                 config_dir='./repositories'):
+        self.github_token = github_token if github_token != "" else os.environ["GITHUB_AUTH_TOKEN"]
+        self.host = host
+        self.port = port
+        self.config_dir = config_dir
+        if config_dir[0:2] == "./":
+            self.config_dir = os.path.join(os.path.dirname(os.path.realpath(__file__)), config_dir[2:])
+        if not os.path.exists(self.config_dir):
+            os.makedirs(self.config_dir)
+        self.application = self.create_flask_application()
+        self.set_public_url()
+        self.github = github.Github(login_or_token=self.github_token)
+
+    def set_public_url(self):
+        if self.host == "0.0.0.0":
+            self.public_url = "http://" + socket.getfqdn() + ":" + str(self.port)
+        else:
+            self.public_url = "http://" + self.host + ":" + str(self.port)
+
+
+
+    def create_flask_application(self):
+        application = Flask(__name__)
+        application.log = {}
+        application.killurl = str(uuid.uuid4())
+
+        @application.template_filter('file_render', )
+        def file_render(file, terms):
+            def removeTerms(content, terms):
+                for term in terms:
+                    content = content.replace(term, "XXX")
+                return content
+            if ".md" in file.name:
+                return removeTerms(Markup(self.github.render_markdown(file.decoded_content)), terms)
+            if ".jpg" in file.name or ".png" in file.name or ".png" in file.name or ".gif" in file.name:
+                return Markup("<img src='%s' alt='%s'>" % (file.url, file.name))
+            if ".html" in file.name:
+                return removeTerms(Markup(file.decoded_content), terms)
+            if ".txt" in file.name:
+                return removeTerms(Markup("<pre>" + file.decoded_content + "</pre>"), terms)
+            return Markup("<a href='%s'>Download %s</a>" % (file.url, file.name))
+
+        @application.route('/' + application.killurl, methods=['POST'])
+        def seriouslykill():
+            func = request.environ.get('werkzeug.server.shutdown')
+            func()
+            return "Shutting down..."
+
+        @application.route('/repository/<id>', methods=['GET'], defaults={'path': ''})
+        @application.route('/repository/<id>/', methods=['GET'], defaults={'path': ''})
+        @application.route('/repository/<id>/<path:path>', methods=['GET'])
+        def repository(id, path):
+            config_path = self.config_dir + "/" + str(id) + "/config.json"
+            if not os.path.exists(config_path):
+                return render_template('404.html'), 404
+            with open(config_path) as f:
+                data = json.load(f)
+                repo = data['repository']\
+                    .replace("http://github.com/", "")\
+                    .replace("https://github.com/", "")
+                if repo[-1] == '/':
+                    repo = repo[:-1]
+                g_repo = self.github.get_repo(repo)
+                current_folder = g_repo.get_contents(path)
+                current_file = None
+                if type(current_folder) is github.ContentFile.ContentFile:
+                    current_file = current_folder
+                    current_folder = g_repo.get_contents(os.path.dirname(path))
+                else:
+                    for f in current_folder:
+                        if f.name.lower() == "readme.md" or f.name.lower() == "index.html":
+                            current_file = f
+                            break
+                return render_template('repo.html', terms=data["terms"], current_repository=id, current_file=current_file, current_folder=current_folder)
+
+        @application.route('/', methods=['GET'])
+        def index():
+            return render_template('index.html')
+
+        @application.route('/', methods=['POST'])
+        def add_repository():
+            id = str(uuid.uuid4())
+            repo = request.form['githubRepository']
+            terms = request.form['terms']
+
+            config_path = self.config_dir + "/" + str(id)
+            os.mkdir(config_path)
+            with open(config_path + "/config.json", 'w') as outfile:
+                json.dump({
+                    "id": id,
+                    "repository": repo,
+                    "terms": terms.split("\n")
+                }, outfile)
+            return redirect(url_for('repository', id=id))
+
+        return application
+
+    def run(self, **keywords):
+        self.application.run(host=self.host, port=self.port, **keywords)
+
+
+def initParser():
+    parser = argparse.ArgumentParser(description='Start Anonymous Github')
+    parser.add_argument('-token', required=True, help='GitHuh token')
+    parser.add_argument('-host', help='The hostname', default="127.0.0.1")
+    parser.add_argument('-port', help='The port of the application', default=5000)
+    parser.add_argument('-config_dir', help='The repository that will contains the configuration files',
+                        default='./repositories')
+    return parser.parse_args()
+
+
+if __name__ == "__main__":
+    args = initParser()
+    Anonymous_Github(github_token=args.token, host=args.host, port=args.port, config_dir=args.config_dir).run()