#!/usr/bin/env bash
# Copyright 2026 Google LLC
# SPDX-License-Identifier: Apache-2.0
# Common utilities for Genkit Python samples
# ==========================================
#
# This script provides shared functions for all sample run.sh scripts.
# Source this file at the beginning of your run.sh:
#
# source "$(dirname "$0")/../_common.sh"
#
# Available functions:
# - print_banner "Title" "emoji" - Print a colorful banner
# - check_env_var "VAR_NAME" "get_url" - Check if env var is set
# - open_browser_for_url "url" - Open browser when URL is ready
# - genkit_start_with_browser [args...] - Start genkit and auto-open browser
# Colors for output
export RED='\033[0;31m'
export GREEN='\033[0;32m'
export YELLOW='\033[1;33m'
export BLUE='\033[0;34m'
export CYAN='\033[0;36m'
export NC='\033[0m' # No Color
# Print a colorful banner
# Usage: print_banner "Title Text" "emoji"
print_banner() {
local title="$1"
local emoji="${2:-✨}"
# Calculate padding for centering (box is 67 chars wide, content is 65)
local content="${emoji} ${title} ${emoji}"
local content_len=${#content}
local padding=$(( (65 - content_len) / 2 ))
local left_pad=$(printf '%*s' "$padding" '')
local right_pad=$(printf '%*s' "$((65 - content_len - padding))" '')
echo -e "${BLUE}"
echo "╔═══════════════════════════════════════════════════════════════╗"
printf "║%s%s%s║\n" "$left_pad" "$content" "$right_pad"
echo "╚═══════════════════════════════════════════════════════════════╝"
echo -e "${NC}"
}
# Check if an environment variable is set
# Usage: check_env_var "GOOGLE_API_KEY" "https://makersuite.google.com/app/apikey"
check_env_var() {
local var_name="$1"
local get_url="$2"
local current_val="${!var_name:-}"
# Prompt if running interactively
# We check -t 0 (stdin is TTY) and also explicit check for /dev/tty availability
if [[ -t 0 ]] && [ -c /dev/tty ]; then
local display_val="${current_val}"
# Simple masking for keys
if [[ "$var_name" == *"API_KEY"* || "$var_name" == *"SECRET"* ]]; then
if [[ -n "$current_val" ]]; then
display_val="******"
fi
fi
echo -en "${BLUE}Enter ${var_name}${NC}"
if [[ -n "$display_val" ]]; then
echo -en " [${YELLOW}${display_val}${NC}]: "
else
echo -n ": "
fi
local input_val
# Safely read from TTY
if read -r input_val < /dev/tty; then
if [[ -n "$input_val" ]]; then
export "$var_name"="$input_val"
fi
fi
# Only print newline if we actually prompted
echo ""
fi
if [[ -z "${!var_name:-}" ]]; then
echo -e "${YELLOW}Warning: ${var_name} not set${NC}"
if [[ -n "$get_url" ]]; then
echo "Get a key from: $get_url"
fi
echo ""
return 1
fi
return 0
}
# Check if we have a GUI/display available
# Returns 0 (true) if GUI is available, 1 (false) otherwise
has_display() {
# Check if running in SSH without X forwarding
if [[ -n "${SSH_CLIENT:-}" || -n "${SSH_TTY:-}" ]]; then
# SSH session - check for X forwarding
if [[ -z "${DISPLAY:-}" ]]; then
return 1 # No display in SSH without X forwarding
fi
fi
# macOS always has a display if not in SSH
if [[ "$(uname)" == "Darwin" ]]; then
return 0
fi
# Linux - check for display server
if [[ -n "${DISPLAY:-}" || -n "${WAYLAND_DISPLAY:-}" ]]; then
return 0
fi
# WSL - check for WSLg or access to Windows
if [[ -n "${WSL_DISTRO_NAME:-}" ]]; then
if command -v wslview &> /dev/null; then
return 0
fi
fi
# No display detected
return 1
}
# Open browser for a given URL
# Works cross-platform: macOS, Linux, Windows (Git Bash/WSL)
# Skips browser opening if no display is available (e.g., SSH sessions)
open_browser_for_url() {
local url="$1"
# Check if we have a display
if ! has_display; then
echo -e "${CYAN}Remote session detected - skipping browser auto-open${NC}"
echo -e "Open manually: ${GREEN}${url}${NC}"
return 0
fi
if command -v open &> /dev/null; then
open "$url" # macOS
elif command -v xdg-open &> /dev/null; then
xdg-open "$url" # Linux
elif command -v wslview &> /dev/null; then
wslview "$url" # WSL
elif command -v start &> /dev/null; then
start "$url" # Windows Git Bash
else
echo -e "${YELLOW}Could not auto-open browser. Please open: ${GREEN}${url}${NC}"
fi
}
# Watch genkit output for the Developer UI URL and open browser
# This function reads from stdin and watches for the URL pattern
_watch_for_devui_url() {
local line
local url_found=false
while IFS= read -r line; do
# Print the line as it comes (pass through)
echo "$line"
# Check for the Genkit Developer UI URL
if [[ "$url_found" == "false" && "$line" == *"Genkit Developer UI:"* ]]; then
# Extract URL - handle both with and without ANSI codes
local url
# Remove ANSI escape codes and extract URL
url=$(echo "$line" | sed 's/\x1b\[[0-9;]*m//g' | grep -oE 'https?://[^ ]+' | head -1)
if [[ -n "$url" ]]; then
url_found=true
# Open browser in background
(
# Small delay to ensure server is fully ready
sleep 1
open_browser_for_url "$url"
) &
fi
fi
done
}
# Start genkit with automatic browser opening
# Usage: genkit_start_with_browser -- [your command after --]
# Example: genkit_start_with_browser -- uv run src/main.py
genkit_start_with_browser() {
echo -e "${BLUE}Starting Genkit Dev UI...${NC}"
echo -e "Browser will open automatically when ready"
echo ""
# Run genkit start and pipe through our URL watcher
# Using stdbuf to disable buffering for real-time output
if command -v stdbuf &> /dev/null; then
stdbuf -oL -eL genkit start "$@" 2>&1 | _watch_for_devui_url
else
# Fallback without stdbuf (may have buffering issues)
genkit start "$@" 2>&1 | _watch_for_devui_url
fi
}
# Install dependencies with uv
install_deps() {
echo -e "${BLUE}Installing dependencies...${NC}"
uv sync
echo ""
}
# Standard help footer
print_help_footer() {
local port="${1:-4000}"
echo ""
echo "Getting Started:"
echo " 1. Set required environment variables"
echo " 2. Run: ./run.sh"
echo " 3. Browser opens automatically to http://localhost:${port}"
}
