# Check if a parameter is provided
if [ -z "$1" ]; then
  echo "Error: An artifact ID, local path, or remote URL is required."
  echo "Usage: ./view-trace.sh <artifactId|localPath|remotePath>"
  exit 1
fi

# Capture the parameter
input=$1
tmp_dir=$(mktemp -d)

# Function to download artifact by ID
download_artifact() {
  local artifact_id=$1
  echo "Downloading artifact ID $artifact_id..."
  gh api /repos/ORG_NAME/REPO_NAME/actions/artifacts/$artifact_id/zip > "$tmp_dir/artifact.zip"
  if [ $? -ne 0 ]; then
    echo "Error: Failed to download artifact ID $artifact_id."
    exit 1
  fi
}

# Function to handle remote URL
download_remote() {
  local url=$1
  echo "Extracting artifact ID from URL..."
  # Extract the artifact ID using regex
  if [[ $url =~ artifacts/([0-9]+) ]]; then
    artifact_id="${BASH_REMATCH[1]}"
    echo "Artifact ID extracted: $artifact_id"
    download_artifact "$artifact_id"
  else
    echo "Error: Failed to extract artifact ID from URL."
    exit 1
  fi
}

# upload-artifact double-zips zip files
extract_inner_zip() {
  echo "Extracting inner artifact ZIP..."
  unzip -q "$tmp_dir/artifact.zip" -d "$tmp_dir"
  rm "$tmp_dir/artifact.zip"
  if [ $? -ne 0 ]; then
    echo "Error: Failed to extract artifact."
    exit 1
  fi
}

# Determine the input type
if [[ $input =~ ^[0-9]+$ ]]; then
  # Numeric input: Treat as artifact ID
  download_artifact "$input"
  extract_inner_zip
elif [[ $input =~ ^https?:// ]]; then
  # URL input: Treat as remote path
  download_remote "$input"
  extract_inner_zip
elif [[ $input =~ ^/ ]]; then
  # Local path: Verify file exists
  if [ -f "$input" ]; then
    echo "Using local file $input..."
    cp "$input" "$tmp_dir/artifact.zip"
  else
    echo "Error: Local file $input does not exist."
    exit 1
  fi
else
  echo "Error: Invalid input. Provide an artifact ID, local path, or remote URL."
  exit 1
fi

# Find the inner ZIP file (randomly named)
inner_zip=$(find "$tmp_dir" -name "*.zip" | head -n 1)
if [ -z "$inner_zip" ]; then
  echo "Error: No inner ZIP file found in the artifact."
  exit 1
fi

# Open the extracted folder with Playwright trace viewer
echo "Opening trace viewer..."
yarn playwright show-trace "$inner_zip"
if [ $? -ne 0 ]; then
  echo "Error: Failed to open trace viewer."
  exit 1
fi

echo "Trace viewer closed. Temporary files are in: $tmp_dir"