1#!/usr/bin/env bash
 2
 3# Squawk is a linter for database migrations. It helps identify dangerous patterns, and suggests alternatives.
 4# Squawk flagging an error does not mean that you need to take a different approach, but it does indicate you need to think about what you're doing.
 5# See also: https://squawkhq.com
 6
 7set -e
 8
 9if [ -z "$GITHUB_BASE_REF" ]; then
10  echo 'Not a pull request, skipping squawk modified migrations linting'
11  exit
12fi
13
14SQUAWK_VERSION=0.26.0
15SQUAWK_BIN="./target/squawk-$SQUAWK_VERSION"
16SQUAWK_ARGS="--assume-in-transaction --config script/lib/squawk.toml"
17
18pkgutil --pkg-info com.apple.pkg.RosettaUpdateAuto || /usr/sbin/softwareupdate --install-rosetta --agree-to-license
19# When bootstrapping a brand new CI machine, the `target` directory may not exist yet.
20mkdir -p "./target"
21curl -L -o "$SQUAWK_BIN" "https://github.com/sbdchd/squawk/releases/download/v$SQUAWK_VERSION/squawk-darwin-x86_64"
22chmod +x "$SQUAWK_BIN"
23
24if [ -n "$SQUAWK_GITHUB_TOKEN" ]; then
25    export SQUAWK_GITHUB_REPO_OWNER=$(echo $GITHUB_REPOSITORY | awk -F/ '{print $1}')
26    export SQUAWK_GITHUB_REPO_NAME=$(echo $GITHUB_REPOSITORY | awk -F/ '{print $2}')
27    export SQUAWK_GITHUB_PR_NUMBER=$(echo $GITHUB_REF | awk 'BEGIN { FS = "/" } ; { print $3 }')
28
29    $SQUAWK_BIN $SQUAWK_ARGS upload-to-github $(git diff --name-only origin/$GITHUB_BASE_REF...origin/$GITHUB_HEAD_REF 'crates/collab/migrations/*.sql' 'crates/collab/migrations_llm/*.sql')
30else
31    $SQUAWK_BIN $SQUAWK_ARGS $(git ls-files --others crates/collab/migrations/*.sql crates/collab/migrations_llm/*.sql) $(git diff --name-only main crates/collab/migrations/*.sql crates/collab/migrations_llm/*.sql)
32fi