From 8c08188c88af23111d729d54d3639074659be21a Mon Sep 17 00:00:00 2001
From: n07070 <social+git@n07070.xyz>
Date: Sat, 12 Mar 2022 04:28:17 +0100
Subject: [PATCH] Add web routes, web class

---
 src/main.py | 97 +++++++++++++++++++++++++++++++++++++++--------------
 1 file changed, 72 insertions(+), 25 deletions(-)

diff --git a/src/main.py b/src/main.py
index fdf6f6c..a1d3c9b 100644
--- a/src/main.py
+++ b/src/main.py
@@ -7,11 +7,16 @@
 # Then we build the web interface, using the simple Jinja2 templating.
 
 # Following are the librairies we import,
-from flask import Flask, request         # Used for the web framework
-from printer import Printer     # The wrapper for the printer class
-import toml                     # Used for the config file parsing
-import pprint
-import time
+from flask import Flask, request, render_template, flash, abort, redirect, url_for, make_response, jsonify       # Used for the web framework
+from flask_limiter import Limiter
+from flask_limiter.util import get_remote_address
+from printer import Printer             # The wrapper for the printer class
+from web import Web                     # Wrapper for the web routes and API
+import toml                             # Used for the config file parsing
+import pprint                           # To pretty print JSON
+import time                             # To sleep
+import os                               # For VARS from the shell.
+# Variables
 
 # Load the configuration file
 try:
@@ -25,7 +30,13 @@ except toml.TomlDecodeError:
 except Exception as e:
     print("Error while loading file : " + str(e))
 
-pprint.pprint(configuration_file)
+# Define the USB connections here.
+vendor_id = configuration_file["printer"]["vendor_id"]
+device_id = configuration_file["printer"]["device_id"]
+
+# Output the config file
+if os.getenv('LIPY_DEBUG') == True:
+    pprint.pprint(configuration_file)
 
 # We define the app module used by Flask
 app = Flask(__name__)
@@ -34,16 +45,24 @@ app.secret_key = configuration_file["secrets"]["flask_secret_key"]
 
 # Printer connection
 # Uses the class defined in the printer.py file
-
-# Define the USB connections here.
-vendor_id = configuration_file["printer"]["vendor_id"]
-device_id = configuration_file["printer"]["device_id"]
-
 printer = Printer(app,0x04b8, 0x0e28)
 printer.init_printer()
 
+# Web routes
+web = Web(app, printer)
 
 
+limiter = Limiter(
+    app,
+    key_func=get_remote_address,
+    default_limits=["200 per day", "50 per hour"]
+)
+
+@app.route('/')
+@limiter.limit("1/second", override_defaults=False)
+def index():
+    return render_template('index.html')
+
 # API routes
 # The api has the following methods
 # api/print/{sms,img,letter,qr,barcode}
@@ -51,22 +70,50 @@ printer.init_printer()
 # api/status/{paper,ping,stats}
 
 # If you just call the api route, you get a help back.
-@app.route('/')
-def index():
-    return "Welcome ! Please type your message in the URL bar in such way : /api/print/sms?txt=<message here>"
-
-
 @app.route('/api')
-def api_help():
-    return "Welcome to the API's help page"
-
 @app.route('/api/print')
+@limiter.limit("1/second", override_defaults=False)
 def api_index():
-    return "Welcome to the printing software's API. Please see the index page to get started."
+    return render_template("api.html")
 
-@app.route('/api/print/sms', methods=['GET'])
+
+@app.route('/api/print/sms', methods=['POST'])
+@limiter.limit("2/minute", override_defaults=False)
 def api_print_sms():
-    texte = request.args.get("txt",type=str)
-    app.logger.debug("Printing : " + texte)
-    time.sleep(10)
-    return  printer.print_sms(texte)
+    try:
+        txt = request.form["txt"]
+        sign = request.form["signature"]
+    except Exception as e:
+        flash(e)
+        redirect(url_for('index'))
+
+    flash(web.print_sms(txt,sign))
+    return redirect(url_for('index'))
+
+@app.route('/api/print/image', methods=['POST'])
+@limiter.limit("2/minute", override_defaults=False)
+def api_print_image():
+    flash(web.print_image())
+    return "print image"
+
+@app.route('/login')
+@limiter.limit("1/second", override_defaults=False)
+def login_page():
+    # web.login(username,password)
+    return render_template("index.html")
+
+@app.route('/logout')
+@limiter.limit("1/second", override_defaults=False)
+def logout_page():
+    # web.logout(username, password)
+    return render_template("index.html")
+
+@app.errorhandler(429)
+def ratelimit_handler(e):
+    flash("Rate limit reached, please slow down :) ( Currently at "+ e.description + ")", 'error')
+    return render_template('index.html')
+
+@app.route("/ping")
+@limiter.exempt
+def ping():
+    return "PONG"