frontend: all paths work locally now

frontend/frontend.py

@@ -36,12 +36,18 @@ def index(path):
     if not path:
         form = DataForm()
         return render_template("index.html", form=form, data="")
+    if current_app.debug:
+        result_url = url_for(
+            '.get_data', path=path
+        )
+    else:
+        result_url = url_for(
+            '.get_data', path=path, _external=True, _scheme='https'
+        )
     return render_template(
         "index.html",
         form="",
-        result_url=url_for(
-            '.get_data', path=path, _external=True, _scheme='https'
-        )
+        result_url=result_url
     )
 
 
@@ -51,17 +57,19 @@ def get_data(path):
     if not token:
         return redirect(url_for(".index"))
 
-    vault_uri = os.environ.get("VAULT_URI", None)
-    if not vault_uri:
-        flash("Missing VAULT_URI")
-        return redirect(url_for(".index"))
-
-    try:
-        cubby = hvac.Client(url=vault_uri, token=token)
-        result = cubby.read("cubbyhole/%s" % token)
-    except hvac.exceptions.Forbidden:
-        return jsonify(result="link expired")
-    secret = base64.b64decode(result["data"]["wrap"]).decode()
+    if current_app.debug:
+        secret = "123\n123"
+    else:
+        vault_uri = os.environ.get("VAULT_URI", None)
+        if not vault_uri:
+            flash("Missing VAULT_URI")
+            return redirect(url_for(".index"))
+        try:
+            cubby = hvac.Client(url=vault_uri, token=token)
+            result = cubby.read("cubbyhole/%s" % token)
+        except hvac.exceptions.Forbidden:
+            return jsonify(result="link expired")
+        secret = base64.b64decode(result["data"]["wrap"]).decode()
     return jsonify(result=secret)
 
 
@@ -73,19 +81,25 @@ def add_entry():
         secret_data = base64.b64encode(form.secrets.data.encode()).decode()
         root_token = current_app.get_token()
 
-        vault_uri = os.environ.get("VAULT_URI", None)
-        if not vault_uri:
-            flash("Missing VAULT_URI")
-            return redirect(url_for(".index"))
-
-        vault = hvac.Client(url=vault_uri, token=root_token)
-        token = vault.create_token(
-            lease="24h", num_uses=2, renewable=False, no_default_policy=True
-        )
-        token_id = token["auth"]["client_token"]
+        if current_app.debug:
+            token_id="0000"
+        else:
+            vault_uri = os.environ.get("VAULT_URI", None)
+            if not vault_uri:
+                flash("Missing VAULT_URI")
+                return redirect(url_for(".index"))
+
+            vault = hvac.Client(url=vault_uri, token=root_token)
+            token = vault.create_token(
+                lease="24h",
+                num_uses=2,
+                renewable=False,
+                no_default_policy=True,
+            )
+            token_id = token["auth"]["client_token"]
 
-        cubby = hvac.Client(url=vault_uri, token=token_id)
-        cubby.write("cubbyhole/%s" % token_id, wrap=secret_data)
+            cubby = hvac.Client(url=vault_uri, token=token_id)
+            cubby.write("cubbyhole/%s" % token_id, wrap=secret_data)
         flash("Successfully saved")
 
         return render_template("success.html", token=token_id)