SSH keys CRUD working

capsulflask/console.py

@@ -1,3 +1,4 @@
+import re
 from flask import Blueprint
 from flask import flash
 from flask import current_app
@@ -24,7 +25,62 @@ def makeCapsulId():
 @bp.route("/")
 @account_required
 def index():
-  return render_template("console.html", vms=get_model().list_vms_for_account(session["account"]))
+  return render_template("capsuls.html", vms=get_model().list_vms_for_account(session["account"]))
+
+@bp.route("/ssh", methods=("GET", "POST"))
+@account_required
+def ssh_public_keys():
+  db_model = get_model()
+  error = None
+
+  if request.method == "POST":
+    method = request.form["method"]
+
+    name = request.form["name"]
+    if not name or len(name.strip()) < 1:
+      error = "Name is required"
+    elif not re.match(r"^[0-9A-Za-z_ -]+$", name):
+      error = "Name must match \"^[0-9A-Za-z_ -]+$\""
+
+    if method == "POST":
+      content = request.form["content"]
+      if not content or len(content.strip()) < 1:
+        error = "Content is required"
+      else:
+        content = content.replace("\r", "").replace("\n", "")
+        if not re.match(r"^(ssh|ecdsa)-[0-9A-Za-z+/_=@ -]+$", content):
+          error = "Content must match \"^(ssh|ecdsa)-[0-9A-Za-z+/_=@ -]+$\""
+
+      if db_model.ssh_public_key_name_exists(session["account"], name):
+        error = "A key with that name already exists"
+
+      if error is None:
+        db_model.create_ssh_public_key(session["account"], name, content)
+
+    elif method == "DELETE":
+
+      if error is None:
+        db_model.delete_ssh_public_key(session["account"], name)
+
+  if error:
+    flash(error)
+
+  # keys_list=
+  # for key in keys_list:
+  #   if len(key['content']) > 40:
+  #     print(key['content'])
+  #     print(f"{key['content'][:20]}...{key['content'][len(key['content'])-20:]}")
+  #     key["content"] = 
+
+  # return 
+
+  return render_template(
+    "ssh-public-keys.html", 
+    ssh_public_keys=map(
+      lambda x: dict(name=x['name'], content=f"{x['content'][:20]}...{x['content'][len(x['content'])-20:]}"), 
+      db_model.list_ssh_public_keys_for_account(session["account"])
+    )
+  )
 
 @bp.route("/create", methods=("GET", "POST"))
 @account_required
@@ -64,8 +120,11 @@ def create():
         memory=vm_sizes[size].memory
       )
 
+  if error:
+    flash(error)
+
   return render_template(
-    "create.html",
+    "create-capsul.html",
     ssh_public_keys=ssh_public_keys,
     operating_systems=operating_systems,
     vm_sizes=vm_sizes