stackql-google-cloud-shell.sh

#!/bin/sh

show_usage() {
    echo "Script to run StackQL in Google Cloud Shell"
    echo     
    echo "Usage:"
    echo "  ./stackql-google-cloud-shell.sh [shell | exec] [flags]"
    echo
    echo "  Command (optional):"
    echo "      'shell' (default) enters the StackQL command shell to execute queries interactively."
    echo "      'exec' is used to execute StackQL queries or files to provide batch outputs"
    echo "          (such as CSV or JSON output files). If not specified, 'shell' is assumed."
    echo
    echo "  Flags:"
    echo "      StackQL args are optional global flags, documented at https://stackql.io/docs/command-line-usage/global-flags"
    echo
    echo "  Examples:"
    echo "      # Launch the StackQL shell using interactive authentication"
    echo "      sh stackql-google-cloud-shell.sh"
    echo
    echo "      # Execute a query from a file, writing the output to a CSV file with interactive authentication"    
    echo "      sh stackql-google-cloud-shell.sh exec --infile /path/to/query.sql --output csv --outfile /path/to/output.csv"
    echo
}

pull_google_docs() {
    echo "Pulling latest Google Cloud provider..."
    ./stackql exec "REGISTRY PULL google"
}

CMD="shell" # Default to 'shell' command
FLAGS=""

# Parse command line arguments
while [ $# -gt 0 ]; do
    case "$1" in
        shell|exec)
            CMD="$1"
            shift # Move past the command
            ;;
        *)
            # Check if the argument contains spaces
            if echo "$1" | grep -q " "; then
                # Argument contains spaces, wrap it in quotes
                FLAGS="$FLAGS \"$1\""
            else
                # Argument does not contain spaces, add as is
                FLAGS="$FLAGS $1"
            fi
            shift # Move past each flag
            ;;
    esac
done

# Set authentication for Google Cloud
AUTH='{ "google": { "type": "interactive" }}'

# Execute the StackQL command
if [ "$CMD" = "shell" ]; then
    pull_google_docs
    echo "Entering StackQL shell..."
    eval "./stackql shell --auth='${AUTH}' $FLAGS"
elif [ "$CMD" = "exec" ]; then
    pull_google_docs
    echo "Executing StackQL query..."
    eval "./stackql exec --auth='${AUTH}' $FLAGS"
else
    show_usage
    echo
    echo "Error: invalid command ($CMD)"    
    exit 1
fi