cmd.sh

#!/bin/bash

set -euo pipefail

ROOT_DIR="$(git rev-parse --show-toplevel)"
current_dir=$(pwd)

reset() {
	cd $current_dir
}
trap reset EXIT

lint_swift_command() {
	install_swiftformat
	# files: if an arg is provided use it, otherwise .
	# convert arg to a relative path from app/
	if [ -z "${1-}" ]; then
		files="."
	else
		# make path absolute
		if [[ "$1" != /* ]]; then
			files="$current_dir/$1"
		else
			files="$1"
		fi
	fi
	cd "$(git rev-parse --show-toplevel)/app" &&
		mkdir -p .build/caches/swiftformat &&
		$SWIFTFORMAT_PATH --config rules-header.swiftformat "$files" &&
		$SWIFTFORMAT_PATH --config rules.swiftformat "$files" --cache .build/caches/swiftformat
}

# Get list of changed files based on git diff
get_changed_files() {
	git diff --name-only --diff-filter=ACMR HEAD
}

# Ensures node_modules are installed in local-server
setup_local_server() {
	local server_dir="$(git rev-parse --show-toplevel)/local-server"
	if [ ! -d "$server_dir/node_modules" ]; then
		echo "Installing local-server dependencies..."
		cd "$server_dir" && yarn install && yarn build && yarn copy-to-app
	fi
}

lint_ts_command() {
	setup_local_server
	cd "$(git rev-parse --show-toplevel)/local-server" && yarn lint --fix
}

lint_shell_command() {
	cd "$(git rev-parse --show-toplevel)" &&
		git ls-files '*.sh' |
		xargs shfmt -w
}

lint_ruby_command() {
	cd "$(git rev-parse --show-toplevel)" &&
		git ls-files -z -- "*.rb" "*.rake" "**/Gemfile" "**/Rakefile" "**/Fastfile" | xargs -0 bundle exec rubocop --autocorrect
}

lint_yaml_command() {
	cd "$(git rev-parse --show-toplevel)" &&
		git ls-files '*.yml' '*.yaml' |
		grep -v '^\.yamllint\.yml$' |
			xargs npx --yes prettier@latest --write
}

sync_dependencies_command() {
	cd "$(git rev-parse --show-toplevel)/app" &&
		./tools/dependencies/sync.sh "$@"
}

close_xcode() {
	if pgrep -x "Xcode" >/dev/null; then
		# Kill Xcode
		pkill -x "Xcode"
		# Wait for Xcode to close
		while pgrep -x "Xcode" >/dev/null; do
			sleep 0.01
		done
	fi
}

focus_dependency_command() {
	cd "$(git rev-parse --show-toplevel)/app"
	if [ "${SKIP_CLOSE_XCODE:-}" != "true" ]; then
		close_xcode
	fi
	# Reset xcode state
	find . -path '*.xcuserstate' 2>/dev/null | git check-ignore --stdin | xargs -I{} rm {} || true

	./tools/dependencies/focus.sh "$@"
}

build_release_command() {
	cd "$(git rev-parse --show-toplevel)/app" &&
		./tools/release/release.sh "$@"
}

# Install swift format using a specific version (brew doesn't support this).
install_swiftformat() {
	local version="0.58.0"
	local force=false
	if [ "${1-}" = "--force" ]; then
		force=true
	fi

	local target_dir="$HOME/.cmd/dev/swiftformat/$version"
	local target_path="$target_dir/swiftformat"

	# Check if already installed
	if [ -f "$target_path" ] && [ "$force" = false ]; then
		# Set environment variable with binary location
		export SWIFTFORMAT_PATH="$target_path"

		return 0
	fi

	echo "Installing swiftformat $version..."

	# Create tmp directory if needed
	mkdir -p "$target_dir"

	# Download swiftformat
	local download_url="https://github.com/nicklockwood/SwiftFormat/releases/download/$version/swiftformat.zip"
	local tmp_zip="$target_dir/swiftformat.zip"

	curl -L -o "$tmp_zip" "$download_url"

	# Unzip
	unzip -o "$tmp_zip" -d "$target_dir"

	# Clean up zip file
	rm "$tmp_zip"

	# Remove quarantine attribute to avoid Gatekeeper prompt
	xattr -d com.apple.quarantine "$target_path" 2>/dev/null || true

	# Set environment variable with binary location
	export SWIFTFORMAT_PATH="$target_path"

	echo "✅ swiftformat $($SWIFTFORMAT_PATH --version) installed at $target_path."
}

# Xcode has a weird bug where when opening the Xcode project it will not show most files.
# This seems to happen when there's lingering files from nested Swift packages.
# This function attempts to remove such files.
clean_command() {
	# Signal to the file watcher that is should not regenerate files.
	touch "$(git rev-parse --show-toplevel)/.build/disable-watcher"

	close_xcode

	cd "$(git rev-parse --show-toplevel)/app/modules"

	# Remove derived files from swift packages
	find . -type d -name '.build' -exec sh -c 'rm -rf "$1"' _ {} \; 2>/dev/null
	find . -not -path './.git/*' 2>/dev/null |
		git check-ignore --stdin |
		grep 'Package.swift\|Package.resolved' |
		while read file; do rm -rf "$file"; done

	# Remove lock file
	rm "$(git rev-parse --show-toplevel)/.build/disable-watcher"
}

test_swift_command() {
	setup_local_server
	# Extract --module value if provided
	focussed_module=""
	parsed_args=()

	while [[ $# -gt 0 ]]; do
		case "$1" in
		--module)
			focussed_module="$2"
			shift 2 # Skip both --module and its value
			;;
		*)
			parsed_args+=("$1")
			shift
			;;
		esac
	done
	if [ -n "$focussed_module" ]; then
		SKIP_CLOSE_XCODE=true package_swift=$(focus_dependency_command --module "$focussed_module")
		if [ "${#parsed_args[@]}" -gt 0 ]; then
			cd "$(dirname $package_swift)" && swift test -Xswiftc -suppress-warnings --quiet --no-parallel "${parsed_args[@]}"
		else
			cd "$(dirname $package_swift)" && swift test -Xswiftc -suppress-warnings --quiet --no-parallel
		fi
	else
		# run all tests
		if [ "${#parsed_args[@]}" -gt 0 ]; then
			cd "$(git rev-parse --show-toplevel)/app/modules" && swift test -Xswiftc -suppress-warnings --quiet --no-parallel "${parsed_args[@]}"
		else
			cd "$(git rev-parse --show-toplevel)/app/modules" && swift test -Xswiftc -suppress-warnings --quiet --no-parallel
		fi
	fi
}

test_ts_command() {
	setup_local_server
	cd "$(git rev-parse --show-toplevel)/local-server" && yarn test "$@"
}

# Main command dispatcher
command=$1
shift

case "$command" in
lint:swift)
	lint_swift_command "$@"
	;;
lint:ts)
	lint_ts_command "$@"
	;;
lint:shell)
	lint_shell_command "$@"
	;;
lint:rb)
	lint_ruby_command "$@"
	;;
lint:yaml)
	lint_yaml_command "$@"
	;;
lint)
	# Check if --diff flag is provided
	if [ "${1-}" = "--diff" ]; then
		echo "Running lint on changed files only..."
		changed_files=$(get_changed_files)

		if [ -z "$changed_files" ]; then
			echo "No changed files found."
			exit 0
		fi

		# Group files by type
		swift_files=$(echo "$changed_files" | grep '\.swift$' || true)
		ts_files=$(echo "$changed_files" | grep -E 'local-server/.*\.(ts|tsx)$' || true)
		shell_files=$(echo "$changed_files" | grep '\.sh$' || true)
		ruby_files=$(echo "$changed_files" | grep -E '\.(rb|rake)$|Gemfile$|Rakefile$|Fastfile$' || true)
		yaml_files=$(echo "$changed_files" | grep -E '\.(yml|yaml)$' || true)

		# Run linters for changed file types
		[ -n "$swift_files" ] && echo "Linting Swift files..." && lint_swift_command
		[ -n "$ts_files" ] && echo "Linting TypeScript files..." && lint_ts_command
		[ -n "$shell_files" ] && echo "Linting shell files..." && lint_shell_command
		[ -n "$ruby_files" ] && echo "Linting Ruby files..." && lint_ruby_command
		[ -n "$yaml_files" ] && echo "Linting YAML files..." && lint_yaml_command

		echo "✓ Lint complete for changed files"
	else
		lint_swift_command &&
			lint_ts_command &&
			lint_shell_command &&
			lint_ruby_command &&
			lint_yaml_command
	fi
	;;
test:swift)
	test_swift_command "$@"
	;;
test:ts)
	test_ts_command "$@"
	;;
install:swiftformat)
	install_swiftformat "$@"
	;;
test)
	test_swift_command &&
		test_ts_command
	;;
sync:dependencies)
	install_swiftformat &&
		sync_dependencies_command "$@"
	;;
focus)
	focus_dependency_command "$@"
	;;
open:app)
	clean_command &&
		cd "$(git rev-parse --show-toplevel)/app" &&
		xcode_path=$(xcode-select -p) &&
		xcode_path="${xcode_path%%.app*}.app" &&
		open -a "$xcode_path" "./command.xcodeproj" --args -ApplePersistenceIgnoreState YES
	;;
build:release)
	# build the app for release.
	# use --archive to also archive the app.
	build_release_command "$@"
	;;
clean)
	# clean artifacts that might make Xcode behave weirdly and not showing files in the file hierarchy.
	clean_command
	;;
watch)
	# Watch file changes, and update derived files when necessary.
	setup_local_server
	cd "$ROOT_DIR/local-server" && yarn watch
	;;
*)
	echo "Command not found: $command"
	exit 1
	;;
esac