starting to build console controller and views

capsulflask/console.py

@@ -9,8 +9,69 @@ from flask import session
 from flask import render_template
 from flask_mail import Message
 from werkzeug.exceptions import abort
+from nanoid import generate
+
+from capsulflask.auth import account_required
 
 from capsulflask.db import get_model
 
 bp = Blueprint("console", __name__, url_prefix="/console")
 
+def makeCapsulId():
+  lettersAndNumbers = generate(alphabet="1234567890qwertyuiopasdfghjklzxcvbnm", size=10)
+  return f"capsul-{lettersAndNumbers}"
+
+@bp.route("/")
+@account_required
+def index():
+  return render_template("console.html", vms=get_model().list_vms_for_account(session["account"]))
+
+@bp.route("/create", methods=("GET", "POST"))
+@account_required
+def create():
+  db_model = get_model()
+  vm_sizes = db_model.vm_sizes_dict()
+  operating_systems = db_model.operating_systems_dict()
+  ssh_public_keys = db_model.list_ssh_public_keys_for_account(session["account"])
+  error = None
+
+  if request.method == "POST":
+    size = request.form["size"]
+    os = request.form["os"]
+    if not size:
+      error = "Size is required"
+    elif size not in vm_sizes:
+      error = f"Invalid size {size}"
+
+    if not os:
+      error = "OS is required"
+    elif os not in operating_systems:
+      error = f"Invalid os {os}"
+
+    if error is None:
+      id = makeCapsulId()
+      db_model.create_vm(
+        email=session["account"], 
+        id=id, 
+        size=size, 
+        os=os
+      )
+      current_app.config["VIRTUALIZATION_MODEL"].create(
+        email = session["account"],
+        id=id,
+        template_image_file_name=operating_systems[os].template_image_file_name,
+        vcpus=vm_sizes[size].vcpus,
+        memory=vm_sizes[size].memory
+      )
+
+  return render_template(
+    "create.html",
+    ssh_public_keys=ssh_public_keys,
+    operating_systems=operating_systems,
+    vm_sizes=vm_sizes
+  )
+
+@bp.route("/billing")
+@account_required
+def faq():
+  return render_template("billing.html")