custom_nodes/rgthree-comfy/py/server/utils_server.py

import os
from aiohttp import web

from ..utils import sub_abspath

THIS_DIR = os.path.dirname(os.path.abspath(__file__))
DIR_WEB = os.path.abspath(f'{THIS_DIR}/../../web/')


def get_param(request, param, default=None):
  """Gets a param from a request."""
  return request.rel_url.query[param] if param in request.rel_url.query else default


def is_param_falsy(request, param):
  """Determines if a param is explicitly 0 or false."""
  val = get_param(request, param)
  return val is not None and (val == "0" or val.upper() == "FALSE")


def is_param_truthy(request, param):
  """Determines if a param is explicitly 0 or false."""
  val = get_param(request, param)
  return val is not None and not is_param_falsy(request, param)


def set_default_page_resources(path, routes):
  """Sets up routes for handling static files under a path."""

  @routes.get(f'/rgthree/{path}/{{file}}')
  async def get_resource(request):
    """Returns a resource file."""
    filepath = request.match_info['file']
    abspath = sub_abspath(os.path.join(DIR_WEB, path), filepath)
    if abspath is None:
      return web.HTTPNotFound()
    return web.FileResponse(abspath)

  @routes.get(f'/rgthree/{path}/{{subdir}}/{{file}}')
  async def get_resource_subdir(request):
    """Returns a resource file."""
    filepath = os.path.join(request.match_info['subdir'], request.match_info['file'])
    abspath = sub_abspath(os.path.join(DIR_WEB, path), filepath)
    if abspath is None:
      return web.HTTPNotFound()
    return web.FileResponse(abspath)


def set_default_page_routes(path, routes):
  """ Sets default path handling for a hosted rgthree page. """

  @routes.get(f'/rgthree/{path}')
  async def get_path_redir(request):
    """ Redirects to the path adding a trailing slash. """
    raise web.HTTPFound(f'{request.path}/')

  @routes.get(f'/rgthree/{path}/')
  async def get_path_index(request):
    """ Handles the page's index loading. """
    html = ''
    with open(os.path.join(DIR_WEB, path, 'index.html'), 'r', encoding='UTF-8') as file:
      html = file.read()
    return web.Response(text=html, content_type='text/html')

  set_default_page_resources(path, routes)