fixed stuff. whatevre

Author: Halcyon

hanse_web.py

@@ -1,14 +1,15 @@
-from flask import Flask, request, send_from_directory, render_template, send_file
+from flask import Flask, request, send_from_directory, render_template, send_file, redirect
 from werkzeug.contrib.cache import SimpleCache
 
 app = Flask(__name__, static_url_path='')
 cache = SimpleCache()
-
+app_root = "hanseweb"
 base_path = "images/"
 
 import numpy as np
 from io import BytesIO
 import glob
+import requests
 
 from AnsiGraphics import AnsiGraphics
 from AnsiImage import AnsiImage
@@ -20,50 +21,82 @@
 for i in range(16):
     col = np.array(np.floor(np.array(ansi_graphics.cga_colour(i)) * 255.0), 'int')
     pal_entry = ".fg" + str(i) + "{\n"
-    pal_entry += "    color: rgb(" + str(col[0]) + ", " + str(col[1]) + ", " + str(col[2]) + ");\n"
+    pal_entry += "    color: rgba(" + str(col[0]) + ", " + str(col[1]) + ", " + str(col[2]) + ", 0);\n"
     pal_entry += "}\n\n"
     pal_entry += ".bg" + str(i) + "{\n"
-    pal_entry += "    background: rgb(" + str(col[0]) + ", " + str(col[1]) + ", " + str(col[2]) + ");\n"
+    pal_entry += "    background: rgba(" + str(col[0]) + ", " + str(col[1]) + ", " + str(col[2]) + ", 0);\n"
     pal_entry += "}\n\n"
     pal_styles += pal_entry
 
+def get_remote_file(url, max_size = 200*1024):
+    r = requests.get(url, stream=True)
+    r.raise_for_status()
+
+    if int(r.headers.get('Content-Length')) > max_size:
+        raise ValueError('response too large')
 
-@app.route('/')
+    size = 0
+    content = b""
+    for chunk in r.iter_content(1024):
+        size += len(chunk)
+        if size > max_size:
+            raise ValueError('response too large')
+        content += chunk
+    return content
+
+def load_ansi(path):
+    if ".." in path or path[0] == '/':
+        raise(ValueError("dangerous."))
+              
+    wide_mode = False
+    if request.args.get('wide', False) != False:
+        wide_mode = True    
+    
+    ansi_image = AnsiImage(ansi_graphics)
+    ansi_image.clear_image(1, 1)
+    
+    if path[0:4] == 'http':
+        ansi_data = get_remote_file(path)
+        ansi_image.parse_ans(ansi_data, wide_mode = wide_mode)
+    else:
+        ansi_image.load_ans(base_path + path, wide_mode = wide_mode)
+    return ansi_image
+
+@app.route('/', methods = ['GET', 'POST'])
 def file_list():
+    if request.form.get('load_url', None) != None:
+        return(redirect('/view/' + request.form.get('load_url', None), code=302))
+        
     file_list = list(glob.glob(base_path + '*.ans'))
     list_html = '<h1>Files</h1><div style="text-align: left; font-size: 28px; width: 600px;">'
     for ansi_file in file_list:
         ansi_file = ansi_file[len(base_path):]
-        list_html += '--> <a href="/view/' + ansi_file + '">' + ansi_file + '</a><br/>'
-    list_html += '<a href="/gallery">gallery</a></div>'
-    return(render_template("default.html", title="files", content=list_html))
+        list_html += '--> <a href="/' + app_root + '/view/' + ansi_file + '">' + ansi_file + '</a><br/>'
+    list_html += '<a href="/' + app_root + '/gallery">gallery</a>'
+    list_html += '<form method="post" action="/' + app_root + '">url: <input type="text" name="load_url" style="width: 600px;"></input> <input type="submit" value="load"></input></form></div>'
+    return(render_template("default.html", title="files", content=list_html, app_root=app_root))
 
 @app.route('/gallery')
 def gallery():
     file_list = list(glob.glob(base_path + '*.ans'))
     list_html = '<h1>Gallery</h1><div style="text-align: left; font-size: 28px;" class="gallery">'
     for ansi_file in file_list:
         ansi_file = ansi_file[len(base_path):]
-        list_html += '<a href="/view/' + ansi_file + '"><img src="/image/' + ansi_file + '?thumb"></img></a>'
-    list_html += '<a href="/"><-- back</a></div>'
-    return(render_template("default.html", title="gallery", content=list_html))
+        list_html += '<a href="/' + app_root + '/view/' + ansi_file + '"><img src="/' + app_root + '/image/' + ansi_file + '?thumb"></img></a>'
+    list_html += '<a href="/' + app_root + '"><-- back</a></div>'
+    return(render_template("default.html", title="gallery", content=list_html, app_root=app_root))
 
 @app.route('/view/<path:path>')
 def render_ansi(path):
-    wide_mode = False
-    if request.args.get('wide', False) != False:
-        wide_mode = True    
-        
     try:
-        ansi_image = AnsiImage(ansi_graphics)
-        ansi_image.clear_image(80, 24)
-        ansi_image.load_ans(base_path + path, wide_mode = wide_mode)
-        width, height = ansi_image.get_size()
+        ansi_image = load_ansi(path)
     except:
-       return(render_template("default.html", title="Loading error, sorry."))
+        return(render_template("default.html", title="Loading error, sorry.", content="<h3>Loading error, sorry</h3>", app_root = app_root))
+    width, height = ansi_image.get_size()
 
     html_ansi = ""
-    html_ansi += '<div style="text-align: left; font-size: 28px;"><a href="/"><-- back</a></div>'
+    html_ansi += '<div style="text-align: left; font-size: 28px;"><a href="/' + app_root + '"><-- back</a></div>'
+    html_ansi += '<div style="display:inline-block; background:url(' + "'/" + app_root + '/image/' + path + "'" + ');">'
     for y in range(height):
         for x in range(width):
             char = ansi_image.get_cell(x, y)
@@ -73,18 +106,27 @@ def render_ansi(path):
             html_ansi += chr(char[0])
             html_ansi += '</span>'
         html_ansi += "\n"
-    return(render_template("default.html", title=path, styles=pal_styles, content=html_ansi, show_dl=True))
+    html_ansi += '</div>'
+    return(render_template("default.html", title=path, styles=pal_styles, content=html_ansi, show_dl=True, app_root=app_root))
+
+@app.route('/ansi/<path:path>')
+def send_ansi(path):
+    try:
+        ansi_image = load_ansi(path)
+    except:
+        return(render_template("default.html", title="Loading error, sorry.", content="<h3>Loading error, sorry</h3>", app_root=app_root))
+    img_io = BytesIO()
+    img_io.write(ansi_image.to_ans())
+    img_io.seek(0)
+    return send_file(img_io, mimetype='plain/text')
 
 @app.route('/image/<path:path>')
 def render_ansi_png(path):
     transparent = False
-    wide_mode = False
     thumb = False
 
     if request.args.get('transparent', None) != None:
         transparent = True
-    if request.args.get('wide', None) != None:
-        wide_mode = True    
     if request.args.get('thumb', None) != None:
         thumb = True    
 
@@ -93,14 +135,13 @@ def render_ansi_png(path):
         bitmap = cached
     else:
         try:
-            ansi_image = AnsiImage(ansi_graphics)
-            ansi_image.clear_image(1000, 1000)
-            ansi_image.load_ans(base_path + path, wide_mode = wide_mode)
-            bitmap = ansi_image.to_bitmap(transparent = transparent)
+            ansi_image = load_ansi(path)
         except:
-            return(render_template("default.html", title="Loading error, sorry."))
+            return(render_template("default.html", title="Loading error, sorry.", content="<h3>Loading error, sorry</h3>", app_root=app_root))
+
+        bitmap = ansi_image.to_bitmap(transparent = transparent)
         if thumb == True:
-            bitmap.thumbnail((128, 128))
+            bitmap.thumbnail((256, 256))
         cache.set(request.url, bitmap)
 
     img_io = BytesIO()