capsul-flask/capsulflask/shared.py

import re

from flask import current_app
from typing import List

class OnlineHost:
  def __init__(self, id: str, url: str):
    self.id = id
    self.url = url

# I decided to just use dict everywhere instead because I have to use dict to read it from json 
# class SSHHostKey:
#   def __init__(self, key_type=None, content=None, sha256=None):
#     self.key_type = key_type
#     self.content = content
#     self.sha256 = sha256

class VirtualMachine:
  def __init__(self, id, host, ipv4=None, ipv6=None, ssh_host_keys: List[dict] = list()):
    self.id = id
    self.host = host
    self.ipv4 = ipv4
    self.ipv6 = ipv6
    self.ssh_host_keys = ssh_host_keys

class VirtualizationInterface:
  def capacity_avaliable(self, additional_ram_bytes: int) -> bool:
    pass

  def get(self, id: str) -> VirtualMachine:
    pass

  def list_ids(self) -> list:
    pass

  def create(self, email: str, id: str, template_image_file_name: str, vcpus: int, memory: int, ssh_authorized_keys: list):
    pass

  def destroy(self, email: str, id: str):
    pass

  def vm_state_command(self, email: str, id: str, command: str):
    pass

def validate_capsul_id(id):
  if not re.match(r"^(cvm|capsul)-[a-z0-9]{10}$", id):
    raise ValueError(f"vm id \"{id}\" must match \"^capsul-[a-z0-9]{{10}}$\"")

def authorized_as_hub(headers):
  if headers.get('Authorization'):
    auth_header_value = headers.get('Authorization').replace("Bearer ", "")
    return auth_header_value == current_app.config["HUB_TOKEN"]
  return False

def my_exec_info_message(exec_info):
  return "{}: {}".format(".".join([exec_info[0].__module__, exec_info[0].__name__]), exec_info[1])