Makefile

# Load environment variables from .env file
include .env

.PHONY: tests tests-basic lint install mypy update ui-install ui-run docker

# Existing commands
tests:
	poetry run pytest

tests-basic:
	poetry run pytest tests/basic
	poetry run pytest -s tests/test_api.py
	poetry run pytest tests/test_runner_caching.py
	poetry run pytest -s tests/test_pandas_accessors.py

lint:
	poetry run ruff check docetl/* --fix

install:
	pip install poetry
	poetry install --all-extras

mypy:
	poetry run mypy

update:
	poetry update

# UI-related commands
UI_DIR := ./website 

install-ui:
	cd $(UI_DIR) && npm install

run-ui-dev:
	@echo "Starting server..."
	@python server/app/main.py & \
	echo "Starting UI development server..." && \
	cd $(UI_DIR) && HOST=${FRONTEND_HOST}  PORT=${FRONTEND_PORT} npm run dev

run-ui:
	@echo "Starting server..."
	@python server/app/main.py & \
	echo "Building UI..." && \
	cd $(UI_DIR) && npm run build && HOST=${FRONTEND_HOST}  PORT=${FRONTEND_PORT} NEXT_PUBLIC_FRONTEND_ALLOWED_HOSTS=${FRONTEND_ALLOWED_HOSTS} npm run start

# Single Docker command to build and run
docker:
	docker volume create docetl-data
	docker build -t docetl .

	@if [ -n "$${AWS_PROFILE}" ]; then \
		echo "[INFO] Detected AWS_PROFILE — including AWS credentials."; \
		DOCKER_AWS_FLAGS="-v ~/.aws:/root/.aws:ro \
			-e AWS_PROFILE=$${AWS_PROFILE} \
			-e AWS_REGION=$${AWS_REGION:-us-west-2}"; \
	else \
		echo "[INFO] No AWS_PROFILE set — skipping AWS credentials."; \
		DOCKER_AWS_FLAGS=""; \
	fi && \
	docker run --rm -it \
		-p 3000:3000 \
		-p 8000:8000 \
		-v docetl-data:/docetl-data \
		$$DOCKER_AWS_FLAGS \
		-e FRONTEND_HOST=0.0.0.0 \
		-e FRONTEND_PORT=3000 \
		-e BACKEND_HOST=0.0.0.0 \
		-e BACKEND_PORT=8000 \
		docetl

# Add new command for cleaning up docker resources
docker-clean:
	docker volume rm docetl-data

# Test AWS connectivity
test-aws:
	@if ! command -v aws > /dev/null; then \
		echo "[WARNING] AWS CLI is not installed locally. Skipping local AWS credentials test."; \
	else \
		if [ ! -d ~/.aws ]; then \
			echo "[ERROR] AWS directory ~/.aws not found!"; \
			echo "👉 Configure AWS CLI first: https://docs.aws.amazon.com/cli/latest/userguide/cli-configure-files.html"; \
			exit 1; \
		fi; \
		if [ -z "$${AWS_PROFILE}" ]; then \
			echo "[WARNING] AWS_PROFILE is not set, using 'default' profile."; \
		fi; \
		if [ -z "$${AWS_REGION}" ]; then \
			echo "[WARNING] AWS_REGION is not set, using 'us-west-2' region."; \
		fi; \
		AWS_PROFILE_FLAG=$${AWS_PROFILE:+--profile $$AWS_PROFILE}; \
		echo "[INFO] Testing AWS credentials locally via cli..."; \
		if ! aws sts get-caller-identity $$AWS_PROFILE_FLAG > /dev/null; then \
			echo "[ERROR] Local AWS credentials test failed!"; \
			exit 1; \
		else \
			echo "[SUCCESS] Local AWS credentials are valid."; \
		fi; \
	fi
	@echo "[INFO] Testing AWS credentials inside Docker..."; \
	if ! docker run --rm -it \
		-v ~/.aws:/root/.aws:ro \
		-e AWS_PROFILE=$${AWS_PROFILE:-default} \
		-e AWS_REGION=$${AWS_REGION:-us-west-2} \
		amazon/aws-cli sts get-caller-identity > /dev/null; then \
		echo "[ERROR] Docker AWS credentials test failed!"; \
		exit 1; \
	else \
		echo "[SUCCESS] Docker AWS credentials are valid."; \
	fi

# Help command
help:
	@echo "Available commands:"
	@echo "  make tests        : Run all tests"
	@echo "  make tests-basic  : Run basic tests"
	@echo "  make lint         : Run linter"
	@echo "  make install      : Install dependencies using Poetry"
	@echo "  make mypy         : Run mypy for type checking"
	@echo "  make update       : Update dependencies"
	@echo "  make install-ui   : Install UI dependencies"
	@echo "  make run-ui-dev   : Run UI development server"
	@echo "  make run-ui       : Run UI production server"
	@echo "  make docker       : Build and run docetl in Docker"
	@echo "  make docker-clean : Remove docetl Docker volume"
	@echo "  make test-aws     : Test AWS credentials configuration"
	@echo "  make help         : Show this help message"