#!/bin/bash
set -e

for DIR_ITEM in $(yarn workspaces list --json | jq -r '.location'); do
  DIR=$(echo -n "$DIR_ITEM" | tr -d '\r\n')
  if [ -f "$DIR/package.json" ]; then
    echo "Setting version for $DIR"
    jq ".version = \"$1\"" "$DIR"/package.json > tmp.json && mv tmp.json "$DIR"/package.json
  else
    echo "No package.json found in: $DIR"
  fi
done

update_app_version_in_helm_charts() {
  local file_path=$1
  local new_version=$2

  # Check if file exists
  if [ ! -f "$file_path" ]; then
    echo "Error: File does not exist at $file_path."
    return 1
  fi

  echo "Updating $file_path with appVersion $new_version"

  # Use sed to replace the appVersion value with the new version.
  sed -i.bak -E "s/^appVersion:[[:space:]]+[\"']?.*[\"']?$/appVersion: \"$new_version\"/" "$file_path" || {
    # If the sed command fails, print an error message and exit with a non-zero status
    echo "Error: Failed to update the appVersion in $file_path."
    return 1
  }

  echo "appVersion in $file_path updated to $new_version"

  rm "$file_path".bak
}

update_app_stream_version() {
  local file_path=$1
  local new_version=$2

  # Check if file exists
  if [ ! -f "$file_path" ]; then
    echo "Error: File does not exist at $file_path."
    return 1
  fi

  echo "Updating $file_path with appVersion $new_version"
  # version is at
  # <releases>
  #   <release version="0.21.0" date="yyyy-MM-dd">
  #     <url>https://github.com/toeverything/AFFiNE/releases/tag/v0.21.0</url>
  #   </release>
  # </releases>
  # We need to update the version and the url
  current_date=$(date +"%Y-%m-%d")

  # Use sed to update the version, date, and URL in the releases section
  sed -i.bak -E "s|<release version=\"[^\"]*\" date=\"[^\"]*\">|<release version=\"$new_version\" date=\"$current_date\">|" "$file_path" &&
  sed -i.bak -E "s|<url>https://github.com/toeverything/AFFiNE/releases/tag/v[^<]*</url>|<url>https://github.com/toeverything/AFFiNE/releases/tag/v$new_version</url>|" "$file_path" || {
    # If the sed command fails, print an error message and exit with a non-zero status
    echo "Error: Failed to update the appVersion."
    return 1
  }

  echo "appVersion in $file_path updated to $new_version"

  rm "$file_path".bak
}

update_ios_marketing_version() {
  local file_path=$1
  # Normalize inputs like "v0.26.4-beta.1" to "0.26.4"
  local new_version=$(echo "$2" | sed -E 's/^v//; s/-.*$//')

  # Check if file exists
  if [ ! -f "$file_path" ]; then
    echo "Error: File does not exist at $file_path."
    return 1
  fi

  echo "Updating $file_path with MARKETING_VERSION $new_version"

  # Use sed to replace the MARKETING_VERSION value with the new version
  sed -i.bak -E "s/MARKETING_VERSION = [^;]*;/MARKETING_VERSION = $new_version;/g" "$file_path" || {
    # If the sed command fails, print an error message and exit with a non-zero status
    echo "Error: Failed to update the MARKETING_VERSION."
    return 1
  }

  echo "MARKETING_VERSION in $file_path updated to $new_version"

  rm "$file_path".bak
}

# Derive a date-based iOS MARKETING_VERSION from the latest stable/beta tag.
# Apple requires CFBundleShortVersionString to increase monotonically. Using
# date-based versions (YYYY.M.D) derived from the last stable/beta release tag
# ensures this. The user-facing App Store version is set separately in
# App Store Connect.
get_ios_version_from_git() {
  # Find the most recent stable/beta tag reachable from HEAD (exclude canary/nightly)
  local latest_tag
  latest_tag=$(git describe --tags --match 'v[0-9]*' \
    --exclude '*canary*' --exclude '*nightly*' \
    --abbrev=0 HEAD 2>/dev/null)

  if [ -z "$latest_tag" ]; then
    # No stable/beta tag found, fall back to today's date
    date +"%Y.%-m.%-d"
    return
  fi

  # Get the tag creation date (tagger date for annotated tags, commit date for lightweight)
  local tag_date
  tag_date=$(git for-each-ref --format='%(creatordate:short)' "refs/tags/$latest_tag")

  if [ -z "$tag_date" ]; then
    date +"%Y.%-m.%-d"
    return
  fi

  # Format as YYYY.M.D (no leading zeros for month/day)
  local year month day
  year=$(echo "$tag_date" | cut -d'-' -f1)
  month=$((10#$(echo "$tag_date" | cut -d'-' -f2)))
  day=$((10#$(echo "$tag_date" | cut -d'-' -f3)))

  echo "${year}.${month}.${day}"
}

new_version=$1

if [ -n "$IOS_APP_VERSION" ]; then
  # Manual override via environment variable
  ios_new_version=$IOS_APP_VERSION
elif echo "$new_version" | grep -qE '(canary|nightly)'; then
  # Canary/nightly: use the date of the last stable/beta tag
  ios_new_version=$(get_ios_version_from_git)
else
  # Stable/beta release: use today's date
  ios_new_version=$(date +"%Y.%-m.%-d")
fi

echo "iOS MARKETING_VERSION: $ios_new_version (app version: $new_version)"

update_app_version_in_helm_charts ".github/helm/affine/Chart.yaml" "$new_version"
update_app_version_in_helm_charts ".github/helm/affine/charts/graphql/Chart.yaml" "$new_version"
update_app_version_in_helm_charts ".github/helm/affine/charts/front/Chart.yaml" "$new_version"
update_app_version_in_helm_charts ".github/helm/affine/charts/doc/Chart.yaml" "$new_version"

update_app_stream_version "packages/frontend/apps/electron/resources/affine.metainfo.xml" "$new_version"

update_ios_marketing_version "packages/frontend/apps/ios/App/App.xcodeproj/project.pbxproj" "$ios_new_version"