nathanielks · June 3, 2023 17:24 · nadnerb · Jul 23, 2015 · nathanielks · Jul 23, 2015
diff --git a/README.md b/README.md
diff --git a/terraform-env.sh b/terraform-env.sh
 #!/bin/bash

 function help {
    echo "usage: ${0} <environment> <action> [<args>]"
    exit 1
 }

 function contains_element () {
    local i
    for i in "${@:2}"; do
        [[ "$i" == "$1" ]] && return 0
    done
    return 1
 }

 #All of the args are mandatory.
 if [ $# -lt 1 ]; then
    help
 fi

 # Let's check the existence of the config file
 if [ ! -f $CFG_FILE ]; then
    echo "Error: $CFG_FILE does not exist. You'll need to create a config file so we know where to set up the remote config."
    exit 1
 fi

 source ./terraform.cfg

 # Let's set up our environment
 if [[ $default_environment && ${default_environment-x} ]]
 then
    ENVIRONMENT=$default_environment
    ACTION=$1
    ADDTL_PARAMS=${*:2}
 else
    ENVIRONMENT=$1
    ACTION=$2
    ADDTL_PARAMS=${*:3}
 fi

 # Let's set up our variables
 ALLOWS_VARFILE=(apply plan push refresh destroy)
 ENV_FILE=.terraform/environment
 VARS_FILE=vars/$ENVIRONMENT
 VARS_FILE_FLAG=
 BUCKET_KEY=$bucket_prefix/state/$ENVIRONMENT
 PREVIOUS_ENVIRONMENT=$([ -f $ENV_FILE ] && echo "$(<$ENV_FILE)" || echo "previous")


 # Let's check to see if a vars file exists for the requested environment before proceeding
 if [ ! -f $VARS_FILE ]; then
    echo "Error: $VARS_FILE does not exist. You'll need to create a vars file for the requested environment before continuing."
    exit 1
 fi

 # Checks if current action allows a varfile to be passed
 contains_element "$ACTION" "${ALLOWS_VARFILE[@]}"
 if [ $? -eq 0 ]; then
    VARS_FILE_FLAG="-var-file=$VARS_FILE"
 fi

 # Let's check if the requested environment is different from the previous environment
 if [ $PREVIOUS_ENVIRONMENT != $ENVIRONMENT ]; then

    # Move current state out of the way to make room for the new state
    mv -f .terraform/terraform.tfstate .terraform/terraform.tfstate.$PREVIOUS_ENVIRONMENT &>2
    mv -f .terraform/terraform.tfstate.backup .terraform/terraform.tfstate.backup.$PREVIOUS_ENVIRONMENT &>2

    # Let's log the new environment for later
    echo $ENVIRONMENT > $ENV_FILE

    # Set up remote configuration and pull latest version
    terraform remote config -backend S3 -backend-config="bucket=$bucket" -backend-config="key=$BUCKET_KEY" -backend-config="region=$region"

 fi

 # Let's do work!
 terraform $ACTION $VARS_FILE_FLAG $ADDTL_PARAMS
diff --git a/terraform.cfg b/terraform.cfg
 bucket=some_secret_bucket
 bucket_prefix=example
 region=us-east-1
 default_environment=dev # This is optional. if set, specifying the environment before running commands isn't necessary
	#!/bin/bash

	function help {
	echo "usage: ${0} <environment> <action> [<args>]"
	exit 1
	}

	function contains_element () {
	local i
	for i in "${@:2}"; do
	[[ "$i" == "$1" ]] && return 0
	done
	return 1
	}

	#All of the args are mandatory.
	if [ $# -lt 1 ]; then
	help
	fi

	# Let's check the existence of the config file
	if [ ! -f $CFG_FILE ]; then
	echo "Error: $CFG_FILE does not exist. You'll need to create a config file so we know where to set up the remote config."
	exit 1
	fi

	source ./terraform.cfg

	# Let's set up our environment
	if [[ $default_environment && ${default_environment-x} ]]
	then
	ENVIRONMENT=$default_environment
	ACTION=$1
	ADDTL_PARAMS=${*:2}
	else
	ENVIRONMENT=$1
	ACTION=$2
	ADDTL_PARAMS=${*:3}
	fi

	# Let's set up our variables
	ALLOWS_VARFILE=(apply plan push refresh destroy)
	ENV_FILE=.terraform/environment
	VARS_FILE=vars/$ENVIRONMENT
	VARS_FILE_FLAG=
	BUCKET_KEY=$bucket_prefix/state/$ENVIRONMENT
	PREVIOUS_ENVIRONMENT=$([ -f $ENV_FILE ] && echo "$(<$ENV_FILE)" \|\| echo "previous")


	# Let's check to see if a vars file exists for the requested environment before proceeding
	if [ ! -f $VARS_FILE ]; then
	echo "Error: $VARS_FILE does not exist. You'll need to create a vars file for the requested environment before continuing."
	exit 1
	fi

	# Checks if current action allows a varfile to be passed
	contains_element "$ACTION" "${ALLOWS_VARFILE[@]}"
	if [ $? -eq 0 ]; then
	VARS_FILE_FLAG="-var-file=$VARS_FILE"
	fi

	# Let's check if the requested environment is different from the previous environment
	if [ $PREVIOUS_ENVIRONMENT != $ENVIRONMENT ]; then

	# Move current state out of the way to make room for the new state
	mv -f .terraform/terraform.tfstate .terraform/terraform.tfstate.$PREVIOUS_ENVIRONMENT &>2
	mv -f .terraform/terraform.tfstate.backup .terraform/terraform.tfstate.backup.$PREVIOUS_ENVIRONMENT &>2

	# Let's log the new environment for later
	echo $ENVIRONMENT > $ENV_FILE

	# Set up remote configuration and pull latest version
	terraform remote config -backend S3 -backend-config="bucket=$bucket" -backend-config="key=$BUCKET_KEY" -backend-config="region=$region"

	fi

	# Let's do work!
	terraform $ACTION $VARS_FILE_FLAG $ADDTL_PARAMS
	bucket=some_secret_bucket
	bucket_prefix=example
	region=us-east-1
	default_environment=dev # This is optional. if set, specifying the environment before running commands isn't necessary