Makefile

PAPERTRAIL_HOST ?= logs.papertrailapp.com
PAPERTRAIL_PORT ?= 1234
LAMBDA_NAME ?= system_name
LOG_GROUP ?= log_group_name
WAIT_FOR_FLUSH ?= true
CWTP_DEBUG ?= false

ALNUM_LOG_GROUP = $(shell echo $(LOG_GROUP) | sed 's/[^[:alnum:]]/_/g')
AWS_ACCOUNT_ID = $(shell aws sts get-caller-identity --output text --query Account)

all: lambda log

deps:
	rm -rf node_modules
	yarn --prod

env:
	rm -f env.json
	echo "{\"papertrailHost\": \"$(PAPERTRAIL_HOST)\", \"papertrailPort\": $(PAPERTRAIL_PORT), \"lambdaName\": \"$(LAMBDA_NAME)\", \"logGroup\": \"$(LOG_GROUP)\", \"waitForFlush\": $(WAIT_FOR_FLUSH), \"debug\": $(CWTP_DEBUG)}" > env.json

create-zip:
	rm -f code.zip
	zip code.zip -r index.js env.json node_modules

lambda: deps env create-zip
	aws lambda create-function --publish \
	--function-name $(LAMBDA_NAME) \
	--runtime nodejs6.10 \
	--handler index.handler \
	--zip-file fileb://code.zip \
	--role arn:aws:iam::$(AWS_ACCOUNT_ID):role/lambda_basic_execution

deploy: deps env create-zip
	aws lambda update-function-code --publish \
	--function-name $(LAMBDA_NAME) \
	--zip-file fileb://code.zip

	aws lambda update-alias \
	--function-name "${LAMBDA_NAME}" \
	--name "${LAMBDA_ALIAS}" \
	--function-version `aws lambda list-versions-by-function --function "${LAMBDA_NAME}" --max-items 10000 | node -e "let stdin=''; process.stdin.on('data',(chunk)=>{stdin+=chunk}).on('end',()=>{console.log(JSON.parse(stdin).Versions.pop().Version)})"`

log:
	aws lambda add-permission \
	--function-name $(LAMBDA_NAME) \
	--statement-id $(ALNUM_LOG_GROUP)__$(LAMBDA_NAME) \
	--principal logs.$(AWS_DEFAULT_REGION).amazonaws.com \
	--action lambda:InvokeFunction \
	--source-arn arn:aws:logs:$(AWS_DEFAULT_REGION):$(AWS_ACCOUNT_ID):log-group:$(LOG_GROUP):* \
	--source-account $(AWS_ACCOUNT_ID)

	aws logs put-subscription-filter \
	--log-group-name $(LOG_GROUP) \
	--destination-arn arn:aws:lambda:$(AWS_DEFAULT_REGION):$(AWS_ACCOUNT_ID):function:$(LAMBDA_NAME) \
	--filter-name LambdaStream_$(LAMBDA_NAME) \
	--filter-pattern ""

clean:
	rm -f code.zip env.json

# Test sends through test CloudTrail log data, gzipped and base64 encoded. It should have two
# lines: 'first test message' and 'second test message'.
test: env
	docker run --name lambda --rm -v "${PWD}":/var/task lambci/lambda:nodejs6.10 index.handler '{"awslogs":{"data":"H4sIAAAAAAAAAHWPwQqCQBCGX0Xm7EFtK+smZBEUgXoLCdMhFtKV3akI8d0bLYmibvPPN3wz00CJxmQnTO41whwWQRIctmEcB6sQbFC3CjW3XW8kxpOpP+OC22d1Wml1qZkQGtoMsScxaczKN3plG8zlaHIta5KqWsozoTYw3/djzwhpLwivWFGHGpAFe7DL68JlBUk+l7KSN7tCOEJ4M3/qOI49vMHj+zCKdlFqLaU2ZHV2a4Ct/an0/ivdX8oYc1UVX860fQDQiMdxRQEAAA=="}}'

destroy:
	aws lambda delete-function \
	--function-name $(LAMBDA_NAME)