added addl wrappers
This commit is contained in:
reynold
321
sf-data-export
Executable file
321
sf-data-export
Executable file
@@ -0,0 +1,321 @@
|
||||
#!/bin/bash
|
||||
|
||||
# Data export wrapper for Salesforce CLI
|
||||
# Provides streamlined data export functionality with SOQL query support
|
||||
|
||||
# Color codes for output formatting
|
||||
readonly RED='\033[0;31m'
|
||||
readonly GREEN='\033[0;32m'
|
||||
readonly YELLOW='\033[0;33m'
|
||||
readonly BLUE='\033[0;34m'
|
||||
readonly CYAN='\033[0;36m'
|
||||
readonly GRAY='\033[0;37m'
|
||||
readonly NC='\033[0m' # No Color
|
||||
|
||||
# Function to display usage information
|
||||
show_usage() {
|
||||
echo -e "${BLUE}sf-data-export - Data Export Wrapper for Salesforce CLI${NC}"
|
||||
echo ""
|
||||
echo "USAGE:"
|
||||
echo " sf-data-export [OPTIONS]"
|
||||
echo ""
|
||||
echo "OPTIONS:"
|
||||
echo " -q, --query QUERY SOQL query to export data"
|
||||
echo " -f, --file FILE File containing SOQL query"
|
||||
echo " -s, --sobject SOBJECT Standard object query (exports all records)"
|
||||
echo " -o, --output FILE Output file path (default: export.csv)"
|
||||
echo " -t, --target-org ORG Target org username or alias"
|
||||
echo " --format FORMAT Output format: csv, json (default: csv)"
|
||||
echo " --bulk Use bulk API for large datasets"
|
||||
echo " --wait MINUTES Wait time in minutes (default: 10)"
|
||||
echo " -v, --verbose Enable verbose output"
|
||||
echo " -h, --help Show this help message"
|
||||
echo ""
|
||||
echo "EXAMPLES:"
|
||||
echo " sf-data-export --query \"SELECT Id, Name FROM Account LIMIT 100\""
|
||||
echo " sf-data-export --sobject Account --format json --output accounts.json"
|
||||
echo " sf-data-export --file queries/contacts.soql --bulk --wait 15"
|
||||
echo " sf-data-export --query \"SELECT Id FROM User\" --target-org production"
|
||||
echo ""
|
||||
echo "This script automatically checks for Salesforce CLI installation."
|
||||
}
|
||||
|
||||
# Function to check if Salesforce CLI is installed
|
||||
check_salesforce_cli() {
|
||||
if ! command -v sf &> /dev/null; then
|
||||
return 1
|
||||
fi
|
||||
return 0
|
||||
}
|
||||
|
||||
# Function to run sf-check diagnostics
|
||||
run_salesforce_check() {
|
||||
if [[ -f "sf-check" ]]; then
|
||||
echo -e "${YELLOW}Running Salesforce CLI diagnostics...${NC}"
|
||||
./sf-check
|
||||
elif [[ -f "sf-check.sh" ]]; then
|
||||
echo -e "${YELLOW}Running Salesforce CLI diagnostics...${NC}"
|
||||
bash sf-check.sh
|
||||
elif [[ -f "sf-check.ps1" ]]; then
|
||||
echo -e "${YELLOW}Running Salesforce CLI diagnostics...${NC}"
|
||||
pwsh sf-check.ps1
|
||||
else
|
||||
echo -e "${RED}Salesforce CLI not found and no diagnostic script available.${NC}"
|
||||
echo -e "${RED}Please install the Salesforce CLI: https://developer.salesforce.com/tools/salesforcecli${NC}"
|
||||
fi
|
||||
}
|
||||
|
||||
# Function to validate SOQL query
|
||||
validate_query() {
|
||||
local query="$1"
|
||||
|
||||
# Basic validation - check if it starts with SELECT
|
||||
if [[ ! "$query" =~ ^[[:space:]]*[Ss][Ee][Ll][Ee][Cc][Tt] ]]; then
|
||||
echo -e "${RED}Error: Query must start with SELECT${NC}"
|
||||
return 1
|
||||
fi
|
||||
|
||||
return 0
|
||||
}
|
||||
|
||||
# Function to build standard object query
|
||||
build_sobject_query() {
|
||||
local sobject="$1"
|
||||
|
||||
# For common objects, use sensible field selections
|
||||
case "$sobject" in
|
||||
"Account")
|
||||
echo "SELECT Id, Name, Type, Industry, Phone, Website, BillingCity, BillingState, BillingCountry FROM Account"
|
||||
;;
|
||||
"Contact")
|
||||
echo "SELECT Id, FirstName, LastName, Email, Phone, AccountId, Account.Name FROM Contact"
|
||||
;;
|
||||
"Lead")
|
||||
echo "SELECT Id, FirstName, LastName, Email, Phone, Company, Status, Source FROM Lead"
|
||||
;;
|
||||
"Opportunity")
|
||||
echo "SELECT Id, Name, AccountId, Account.Name, Amount, CloseDate, StageName, Probability FROM Opportunity"
|
||||
;;
|
||||
"Case")
|
||||
echo "SELECT Id, CaseNumber, Subject, Status, Priority, Origin, AccountId, Account.Name, ContactId, Contact.Name FROM Case"
|
||||
;;
|
||||
"User")
|
||||
echo "SELECT Id, Name, Email, Username, Profile.Name, IsActive, LastLoginDate FROM User"
|
||||
;;
|
||||
*)
|
||||
# Generic query for other objects
|
||||
echo "SELECT Id, Name FROM $sobject"
|
||||
;;
|
||||
esac
|
||||
}
|
||||
|
||||
# Initialize variables
|
||||
QUERY=""
|
||||
QUERY_FILE=""
|
||||
SOBJECT=""
|
||||
OUTPUT_FILE="export.csv"
|
||||
TARGET_ORG=""
|
||||
FORMAT="csv"
|
||||
USE_BULK=false
|
||||
WAIT_TIME="10"
|
||||
VERBOSE=false
|
||||
|
||||
# Parse command line arguments
|
||||
while [[ $# -gt 0 ]]; do
|
||||
case $1 in
|
||||
-q|--query)
|
||||
QUERY="$2"
|
||||
shift 2
|
||||
;;
|
||||
-f|--file)
|
||||
QUERY_FILE="$2"
|
||||
shift 2
|
||||
;;
|
||||
-s|--sobject)
|
||||
SOBJECT="$2"
|
||||
shift 2
|
||||
;;
|
||||
-o|--output)
|
||||
OUTPUT_FILE="$2"
|
||||
shift 2
|
||||
;;
|
||||
-t|--target-org)
|
||||
TARGET_ORG="$2"
|
||||
shift 2
|
||||
;;
|
||||
--format)
|
||||
FORMAT="$2"
|
||||
shift 2
|
||||
;;
|
||||
--bulk)
|
||||
USE_BULK=true
|
||||
shift
|
||||
;;
|
||||
--wait)
|
||||
WAIT_TIME="$2"
|
||||
shift 2
|
||||
;;
|
||||
-v|--verbose)
|
||||
VERBOSE=true
|
||||
shift
|
||||
;;
|
||||
-h|--help)
|
||||
show_usage
|
||||
exit 0
|
||||
;;
|
||||
*)
|
||||
echo -e "${RED}Unknown option: $1${NC}"
|
||||
echo ""
|
||||
show_usage
|
||||
exit 1
|
||||
;;
|
||||
esac
|
||||
done
|
||||
|
||||
# Silently check for Salesforce CLI
|
||||
if ! check_salesforce_cli; then
|
||||
run_salesforce_check
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Validate that exactly one query method is specified
|
||||
query_methods=0
|
||||
[[ -n "$QUERY" ]] && ((query_methods++))
|
||||
[[ -n "$QUERY_FILE" ]] && ((query_methods++))
|
||||
[[ -n "$SOBJECT" ]] && ((query_methods++))
|
||||
|
||||
if [[ $query_methods -eq 0 ]]; then
|
||||
echo -e "${RED}Error: Must specify one of: --query, --file, or --sobject${NC}"
|
||||
echo ""
|
||||
echo -e "${YELLOW}Usage examples:${NC}"
|
||||
echo -e "${GRAY} sf-data-export --query \"SELECT Id, Name FROM Account\"${NC}"
|
||||
echo -e "${GRAY} sf-data-export --file queries/accounts.soql${NC}"
|
||||
echo -e "${GRAY} sf-data-export --sobject Account${NC}"
|
||||
echo ""
|
||||
echo -e "${YELLOW}Use --help for detailed usage information.${NC}"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
if [[ $query_methods -gt 1 ]]; then
|
||||
echo -e "${RED}Error: Can only specify one of: --query, --file, or --sobject${NC}"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Validate format
|
||||
if [[ "$FORMAT" != "csv" && "$FORMAT" != "json" ]]; then
|
||||
echo -e "${RED}Error: Format must be 'csv' or 'json'${NC}"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Determine the final query
|
||||
FINAL_QUERY=""
|
||||
if [[ -n "$QUERY" ]]; then
|
||||
FINAL_QUERY="$QUERY"
|
||||
echo -e "${GREEN}Using inline query${NC}"
|
||||
elif [[ -n "$QUERY_FILE" ]]; then
|
||||
if [[ ! -f "$QUERY_FILE" ]]; then
|
||||
echo -e "${RED}Error: Query file not found: $QUERY_FILE${NC}"
|
||||
exit 1
|
||||
fi
|
||||
FINAL_QUERY=$(cat "$QUERY_FILE")
|
||||
echo -e "${GREEN}Using query from file: $QUERY_FILE${NC}"
|
||||
elif [[ -n "$SOBJECT" ]]; then
|
||||
FINAL_QUERY=$(build_sobject_query "$SOBJECT")
|
||||
echo -e "${GREEN}Using standard query for $SOBJECT${NC}"
|
||||
fi
|
||||
|
||||
# Validate the query
|
||||
if ! validate_query "$FINAL_QUERY"; then
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Build the sf command
|
||||
SF_ARGS=("data" "export")
|
||||
|
||||
# Add the query
|
||||
SF_ARGS+=("--query" "$FINAL_QUERY")
|
||||
|
||||
# Add optional parameters
|
||||
if [[ -n "$TARGET_ORG" ]]; then
|
||||
SF_ARGS+=("--target-org" "$TARGET_ORG")
|
||||
echo -e "${CYAN}Target org: $TARGET_ORG${NC}"
|
||||
fi
|
||||
|
||||
if [[ "$USE_BULK" == true ]]; then
|
||||
SF_ARGS+=("--bulk")
|
||||
echo -e "${YELLOW}Using Bulk API${NC}"
|
||||
fi
|
||||
|
||||
if [[ "$WAIT_TIME" != "10" ]]; then
|
||||
SF_ARGS+=("--wait" "$WAIT_TIME")
|
||||
fi
|
||||
|
||||
# Set output file and format
|
||||
case "$FORMAT" in
|
||||
"csv")
|
||||
SF_ARGS+=("--output-file" "$OUTPUT_FILE")
|
||||
;;
|
||||
"json")
|
||||
SF_ARGS+=("--output-file" "$OUTPUT_FILE")
|
||||
SF_ARGS+=("--json")
|
||||
;;
|
||||
esac
|
||||
|
||||
echo -e "${CYAN}Output format: $FORMAT${NC}"
|
||||
echo -e "${CYAN}Output file: $OUTPUT_FILE${NC}"
|
||||
|
||||
# Add verbose flag if requested
|
||||
if [[ "$VERBOSE" == true ]]; then
|
||||
SF_ARGS+=("--verbose")
|
||||
fi
|
||||
|
||||
# Display export information
|
||||
echo ""
|
||||
echo -e "${BLUE}📊 Starting Data Export${NC}"
|
||||
echo -e "${BLUE}=======================${NC}"
|
||||
|
||||
# Show query preview if verbose
|
||||
if [[ "$VERBOSE" == true ]]; then
|
||||
echo ""
|
||||
echo -e "${YELLOW}📝 SOQL Query:${NC}"
|
||||
echo -e "${GRAY}----------------------------------------${NC}"
|
||||
echo -e "${GRAY}$FINAL_QUERY${NC}"
|
||||
echo -e "${GRAY}----------------------------------------${NC}"
|
||||
fi
|
||||
|
||||
# Display the command being run
|
||||
echo ""
|
||||
echo -e "${GRAY}Executing: sf ${SF_ARGS[*]}${NC}"
|
||||
echo ""
|
||||
|
||||
# Execute the command
|
||||
if sf "${SF_ARGS[@]}"; then
|
||||
EXPORT_EXIT_CODE=0
|
||||
else
|
||||
EXPORT_EXIT_CODE=$?
|
||||
fi
|
||||
|
||||
echo ""
|
||||
if [[ $EXPORT_EXIT_CODE -eq 0 ]]; then
|
||||
echo -e "${GREEN}✅ Data export completed successfully!${NC}"
|
||||
|
||||
# Show file information if it exists
|
||||
if [[ -f "$OUTPUT_FILE" ]]; then
|
||||
FILE_SIZE=$(du -h "$OUTPUT_FILE" | cut -f1)
|
||||
if [[ "$FORMAT" == "csv" ]]; then
|
||||
# Count records (excluding header)
|
||||
RECORD_COUNT=$(($(wc -l < "$OUTPUT_FILE") - 1))
|
||||
echo -e "${CYAN}📁 Exported $RECORD_COUNT records to $OUTPUT_FILE ($FILE_SIZE)${NC}"
|
||||
else
|
||||
echo -e "${CYAN}📁 Data exported to $OUTPUT_FILE ($FILE_SIZE)${NC}"
|
||||
fi
|
||||
fi
|
||||
|
||||
if [[ "$VERBOSE" == true ]]; then
|
||||
echo -e "${YELLOW}💡 Use a spreadsheet application or text editor to view the exported data${NC}"
|
||||
fi
|
||||
else
|
||||
echo -e "${RED}❌ Data export failed with exit code: $EXPORT_EXIT_CODE${NC}"
|
||||
echo -e "${YELLOW}💡 Check query syntax and permissions${NC}"
|
||||
exit $EXPORT_EXIT_CODE
|
||||
fi
|
||||
Reference in New Issue
Block a user