57 lines
1.6 KiB
Python
57 lines
1.6 KiB
Python
from flask import Blueprint, current_app, render_template, request, make_response
|
|
|
|
from api.http_client import HttpClient
|
|
from operations.get_instances import get_organization
|
|
|
|
bp = Blueprint("main", __name__)
|
|
|
|
|
|
def _mask(s):
|
|
if not s or len(s) < 8:
|
|
return ""
|
|
return s[:4] + "*" * (len(s) - 8) + s[-4:]
|
|
|
|
|
|
@bp.route("/", methods=["GET", "POST"])
|
|
def index():
|
|
env_token = current_app.config["NUBES_API_TOKEN"]
|
|
user_token = request.cookies.get("token") or request.form.get("token") or ""
|
|
active_token = user_token or env_token
|
|
|
|
org = None
|
|
error = None
|
|
action = request.form.get("action", "")
|
|
|
|
if action == "clear":
|
|
resp = make_response(render_template("index.html",
|
|
organization=None, error=None,
|
|
env_token_masked=_mask(env_token),
|
|
has_user_token=False))
|
|
resp.delete_cookie("token")
|
|
return resp
|
|
|
|
if action == "save":
|
|
user_token = request.form.get("token", "")
|
|
active_token = user_token or env_token
|
|
resp = make_response()
|
|
resp.set_cookie("token", user_token, max_age=60*60*24*365)
|
|
# redirect to GET
|
|
resp.headers["Location"] = "/"
|
|
resp.status_code = 302
|
|
return resp
|
|
|
|
if active_token:
|
|
try:
|
|
client = HttpClient(
|
|
current_app.config["NUBES_API_ENDPOINT"],
|
|
active_token,
|
|
)
|
|
org = get_organization(client)
|
|
except Exception as e:
|
|
error = str(e)
|
|
|
|
return render_template("index.html",
|
|
organization=org, error=error,
|
|
env_token_masked=_mask(env_token),
|
|
has_user_token=bool(user_token))
|