#!/usr/bin/env bash
# Create ComfyUI data directories on the data disk.

ensure_comfyui_dirs() {
  local data_root="${1:-/data}"
  mkdir -p \
    "${data_root}/apps/comfyui/storage" \
    "${data_root}/apps/comfyui/models" \
    "${data_root}/apps/comfyui/cache/hf-hub" \
    "${data_root}/apps/comfyui/cache/torch-hub" \
    "${data_root}/apps/comfyui/input" \
    "${data_root}/apps/comfyui/output" \
    "${data_root}/apps/comfyui/custom_nodes" \
    "${data_root}/apps/comfyui/workflows"
}

if [[ "${BASH_SOURCE[0]}" == "${0}" ]]; then
  ensure_comfyui_dirs "${1:-/data}"
fi