#!/usr/bin/env bash
# find-trackers.sh — Search decompiled source for tracker/analytics SDK usage
set -euo pipefail

usage() {
  cat <<EOF
Usage: find-trackers.sh <source-dir> [OPTIONS]

Search decompiled Java/Kotlin source for analytics and tracker SDK usage.

Arguments:
  <source-dir>    Path to the decompiled sources directory

Options:
  --firebase      Search only for Firebase Analytics
  --adjust        Search only for Adjust SDK
  --appsflyer     Search only for AppsFlyer SDK
  --mixpanel      Search only for Mixpanel SDK
  --amplitude     Search only for Amplitude SDK
  --segment       Search only for Segment SDK
  --braze         Search only for Braze/Appboy SDK
  --clevertap     Search only for CleverTap SDK
  --flurry        Search only for Flurry SDK
  --generic       Search only for generic cross-SDK patterns
  --endpoints     Search only for known tracker endpoints
  --manifest      Search only for AndroidManifest.xml markers
  --entrypoints   Search only for tracker SDK calls in app code (excludes library packages)
  --all           Search all patterns (default)
  -h, --help      Show this help message

Output:
  Results are printed as file:line:match for easy navigation.
EOF
  exit 0
}

SOURCE_DIR=""
SEARCH_FIREBASE=false
SEARCH_ADJUST=false
SEARCH_APPSFLYER=false
SEARCH_MIXPANEL=false
SEARCH_AMPLITUDE=false
SEARCH_SEGMENT=false
SEARCH_BRAZE=false
SEARCH_CLEVERTAP=false
SEARCH_FLURRY=false
SEARCH_GENERIC=false
SEARCH_ENDPOINTS=false
SEARCH_MANIFEST=false
SEARCH_ENTRYPOINTS=false
SEARCH_ALL=true

while [[ $# -gt 0 ]]; do
  case "$1" in
    --firebase)   SEARCH_FIREBASE=true;   SEARCH_ALL=false; shift ;;
    --adjust)     SEARCH_ADJUST=true;     SEARCH_ALL=false; shift ;;
    --appsflyer)  SEARCH_APPSFLYER=true;  SEARCH_ALL=false; shift ;;
    --mixpanel)   SEARCH_MIXPANEL=true;   SEARCH_ALL=false; shift ;;
    --amplitude)  SEARCH_AMPLITUDE=true;  SEARCH_ALL=false; shift ;;
    --segment)    SEARCH_SEGMENT=true;    SEARCH_ALL=false; shift ;;
    --braze)      SEARCH_BRAZE=true;      SEARCH_ALL=false; shift ;;
    --clevertap)  SEARCH_CLEVERTAP=true;  SEARCH_ALL=false; shift ;;
    --flurry)     SEARCH_FLURRY=true;     SEARCH_ALL=false; shift ;;
    --generic)    SEARCH_GENERIC=true;    SEARCH_ALL=false; shift ;;
    --endpoints)  SEARCH_ENDPOINTS=true;  SEARCH_ALL=false; shift ;;
    --manifest)     SEARCH_MANIFEST=true;     SEARCH_ALL=false; shift ;;
    --entrypoints)  SEARCH_ENTRYPOINTS=true;  SEARCH_ALL=false; shift ;;
    --all)          SEARCH_ALL=true; shift ;;
    -h|--help)    usage ;;
    -*)           echo "Error: Unknown option $1" >&2; usage ;;
    *)            SOURCE_DIR="$1"; shift ;;
  esac
done

if [[ -z "$SOURCE_DIR" ]]; then
  echo "Error: No source directory specified." >&2
  usage
fi

if [[ ! -d "$SOURCE_DIR" ]]; then
  echo "Error: Directory not found: $SOURCE_DIR" >&2
  exit 1
fi

GREP_OPTS="-rn --include=*.java --include=*.kt"

section() {
  echo
  echo "==== $1 ===="
  echo
}

run_grep() {
  local pattern="$1"
  # shellcheck disable=SC2086
  grep $GREP_OPTS -E "$pattern" "$SOURCE_DIR" 2>/dev/null || true
}

run_grep_xml() {
  local pattern="$1"
  grep -rn --include="*.xml" -E "$pattern" "$SOURCE_DIR" 2>/dev/null || true
}

# --- Firebase Analytics ---
if [[ "$SEARCH_ALL" == true || "$SEARCH_FIREBASE" == true ]]; then
  section "Firebase Analytics — Initialization"
  run_grep '(FirebaseAnalytics\.getInstance|FirebaseAnalytics\.newInstance|FirebaseApp\.initializeApp)'
  section "Firebase Analytics — Event Logging"
  run_grep '(\.logEvent\s*\(|\.setDefaultEventParameters|FirebaseAnalytics\.Event\.|FirebaseAnalytics\.Param\.)'
  section "Firebase Analytics — User Identity"
  run_grep '(\.setUserId\s*\(|\.setUserProperty\s*\(|\.setAnalyticsCollectionEnabled)'
  section "Firebase Analytics — Consent"
  run_grep '(\.setConsent\s*\(|ConsentType\.|ConsentStatus\.|setAnalyticsCollectionEnabled\s*\(\s*false)'
fi

# --- Adjust ---
if [[ "$SEARCH_ALL" == true || "$SEARCH_ADJUST" == true ]]; then
  section "Adjust — Initialization"
  run_grep '(AdjustConfig\s*\(|Adjust\.onCreate\s*\(|Adjust\.create\s*\(|AdjustConfig\.ENVIRONMENT_)'
  section "Adjust — Event Tracking"
  run_grep '(AdjustEvent\s*\(|Adjust\.trackEvent\s*\(|\.setRevenue\s*\(|\.setCallbackId\s*\()'
  section "Adjust — Attribution & User"
  run_grep '(\.setOnAttributionChangedListener|AdjustAttribution|Adjust\.addSessionCallbackParameter|Adjust\.getAdid\s*\()'
fi

# --- AppsFlyer ---
if [[ "$SEARCH_ALL" == true || "$SEARCH_APPSFLYER" == true ]]; then
  section "AppsFlyer — Initialization"
  run_grep '(AppsFlyerLib\.getInstance|AppsFlyerConversionListener|\.init\s*\(.*AF_DEV_KEY|\.start\s*\()'
  section "AppsFlyer — Event Tracking"
  run_grep '(\.logEvent\s*\(.*AppsFlyerLib|AFInAppEventType\.|AFInAppEventParameterName\.|\.trackEvent\s*\()'
  section "AppsFlyer — User Identity"
  run_grep '(\.setCustomerUserId\s*\(|\.setCustomerIdAndLogSession|\.getAppsFlyerUID\s*\(|\.setAdditionalData\s*\()'
fi

# --- Mixpanel ---
if [[ "$SEARCH_ALL" == true || "$SEARCH_MIXPANEL" == true ]]; then
  section "Mixpanel — Initialization"
  run_grep '(MixpanelAPI\.getInstance\s*\(|MixpanelAPI\.init\s*\()'
  section "Mixpanel — Event Tracking"
  run_grep '(\.track\s*\(|\.trackMap\s*\(|\.timeEvent\s*\(|\.registerSuperProperties\s*\()'
  section "Mixpanel — User Identity & Profile"
  run_grep '(\.identify\s*\(|\.alias\s*\(|\.getPeople\s*\(|\.getDistinctId\s*\(|\.set\s*\(|\.increment\s*\()'
fi

# --- Amplitude ---
if [[ "$SEARCH_ALL" == true || "$SEARCH_AMPLITUDE" == true ]]; then
  section "Amplitude — Initialization"
  run_grep '(Amplitude\.getInstance\s*\(|AmplitudeClient|\.initialize\s*\(.*amplitude|amplitude\.init\s*\()'
  section "Amplitude — Event Tracking"
  run_grep '(\.logEvent\s*\(|\.logRevenue\s*\(|\.setEventProperties|EventOptions\s*\()'
  section "Amplitude — User Identity"
  run_grep '(\.setUserId\s*\(|\.setUserProperties\s*\(|\.setDeviceId\s*\(|\.setGroup\s*\(|Identify\s*\(\))'
fi

# --- Segment ---
if [[ "$SEARCH_ALL" == true || "$SEARCH_SEGMENT" == true ]]; then
  section "Segment — Initialization"
  run_grep '(Analytics\.with\s*\(|Analytics\.Builder|Analytics\.setSingletonInstance|com\.segment\.analytics)'
  section "Segment — Event Tracking"
  run_grep '(\.track\s*\(|\.screen\s*\(|\.group\s*\(|TrackPayload|ScreenPayload)'
  section "Segment — User Identity"
  run_grep '(\.identify\s*\(|\.alias\s*\(|Traits\s*\(\)|IdentifyPayload)'
fi

# --- Braze (formerly Appboy) ---
if [[ "$SEARCH_ALL" == true || "$SEARCH_BRAZE" == true ]]; then
  section "Braze — Initialization"
  run_grep '(Braze\.configure\s*\(|BrazeConfig\.Builder|Appboy\.configure\s*\(|AppboyConfig\.Builder)'
  section "Braze — Event & Purchase Logging"
  run_grep '(\.logCustomEvent\s*\(|\.logPurchase\s*\(|BrazeProperties\s*\(|\.logClick\s*\()'
  section "Braze — User Identity"
  run_grep '(\.getCurrentUser\s*\(|\.changeUser\s*\(|BrazeUser|\.setEmail\s*\(|\.setCustomUserAttribute\s*\()'
fi

# --- CleverTap ---
if [[ "$SEARCH_ALL" == true || "$SEARCH_CLEVERTAP" == true ]]; then
  section "CleverTap — Initialization"
  run_grep '(CleverTapAPI\.getDefaultInstance|CleverTapAPI\.changeCredentials|ActivityLifecycleCallback\.register)'
  section "CleverTap — Event Tracking"
  run_grep '(\.pushEvent\s*\(|\.recordEvent\s*\(|\.pushChargedEvent\s*\()'
  section "CleverTap — User Identity & Profile"
  run_grep '(\.pushProfile\s*\(|\.onUserLogin\s*\(|\.profilePush\s*\(|\.getCleverTapID\s*\()'
fi

# --- Flurry ---
if [[ "$SEARCH_ALL" == true || "$SEARCH_FLURRY" == true ]]; then
  section "Flurry — Initialization"
  run_grep '(FlurryAgent\.Builder|FlurryAgent\.init\s*\(|FlurryAgent\.onStartSession\s*\(|\.withLogEnabled\s*\()'
  section "Flurry — Event Tracking"
  run_grep '(FlurryAgent\.logEvent\s*\(|FlurryAgent\.endTimedEvent|FlurryAgent\.logPayment\s*\()'
  section "Flurry — User Identity"
  run_grep '(FlurryAgent\.setUserId\s*\(|FlurryAgent\.setAge\s*\(|FlurryAgent\.setGender\s*\()'
fi

# --- Generic Cross-SDK Patterns ---
if [[ "$SEARCH_ALL" == true || "$SEARCH_GENERIC" == true ]]; then
  section "Generic — Event Tracking Methods"
  run_grep '\.(trackEvent|logEvent|recordEvent|pushEvent|logCustomEvent)\s*\('
  section "Generic — User Identification"
  run_grep '\.(setUserId|setCustomerUserId|identify|setDistinctId|onUserLogin|changeUser)\s*\('
  section "Generic — Analytics Collection Toggle"
  run_grep '(setAnalyticsCollectionEnabled|setOptOut|setDataCollectionEnabled|setSendingEnabled|optOut|gdprForgetMe)\s*\('
fi

# --- Known Tracker Endpoints ---
if [[ "$SEARCH_ALL" == true || "$SEARCH_ENDPOINTS" == true ]]; then
  section "Known Tracker Endpoints"
  run_grep '"https?://[^"]*(app-measurement\.com|firebase-settings\.crashlytics\.com|firebaselogging|google-analytics\.com/collect)'
  run_grep '"https?://[^"]*(app\.adjust\.(com|io|world)|cdn\.adjust\.com)'
  run_grep '"https?://[^"]*(appsflyer\.com|appsflyersdk\.com|onelink\.me)'
  run_grep '"https?://[^"]*(api\.mixpanel\.com|decide\.mixpanel\.com|api-js\.mixpanel\.com)'
  run_grep '"https?://[^"]*(api\.amplitude\.com|api2\.amplitude\.com|cdn\.amplitude\.com)'
  run_grep '"https?://[^"]*(api\.segment\.(io|com)|cdn-settings\.segment\.com)'
  run_grep '"https?://[^"]*(sdk\.iad-[0-9]+\.braze\.com|rest\.iad-[0-9]+\.braze\.com|appboy\.com)'
  run_grep '"https?://[^"]*(wzrkt\.com|clevertap-prod\.com|in\.clevertap\.com)'
  run_grep '"https?://[^"]*(data\.flurry\.com|flurry\.com/sdk)'
fi

# --- AndroidManifest.xml Markers ---
if [[ "$SEARCH_ALL" == true || "$SEARCH_MANIFEST" == true ]]; then
  section "AndroidManifest — Tracker Services & Receivers"
  run_grep_xml '(com\.google\.firebase\.analytics|com\.google\.android\.gms\.measurement|com\.google\.firebase\.iid)'
  run_grep_xml '(com\.adjust\.|com\.appsflyer\.|com\.mixpanel\.|com\.amplitude\.|com\.segment\.)'
  run_grep_xml '(com\.braze\.|com\.appboy\.|com\.clevertap\.|com\.flurry\.)'
  section "AndroidManifest — Tracker Permissions"
  run_grep_xml '(AD_ID|com\.google\.android\.gms\.permission\.AD_ID|ACCESS_FINE_LOCATION|ACCESS_COARSE_LOCATION|GET_ACCOUNTS)'
  section "AndroidManifest — Tracker Meta-data"
  run_grep_xml '(com\.google\.firebase\.analytics|firebase_analytics_collection_enabled|google_analytics_adid_collection_enabled)'
  run_grep_xml '(ADJUST_|APPSFLYER_|MIXPANEL_|AMPLITUDE_|SEGMENT_|BRAZE_|CLEVERTAP_|FLURRY_)'
fi

# --- Entry Points: Tracker SDK calls from app code only ---
if [[ "$SEARCH_ALL" == true || "$SEARCH_ENTRYPOINTS" == true ]]; then
  section "Entry Points — Tracker SDK calls from app code (library packages excluded)"

  # Build --exclude-dir arguments for known tracker/analytics library packages
  EXCLUDE_DIRS=(
    --exclude-dir="com/google"
    --exclude-dir="com/adjust"
    --exclude-dir="com/appsflyer"
    --exclude-dir="com/mixpanel"
    --exclude-dir="com/amplitude"
    --exclude-dir="com/segment"
    --exclude-dir="com/braze"
    --exclude-dir="com/appboy"
    --exclude-dir="com/clevertap"
    --exclude-dir="com/flurry"
    --exclude-dir="com/facebook"
    --exclude-dir="com/newrelic"
    --exclude-dir="com/crashlytics"
    --exclude-dir="io/sentry"
    --exclude-dir="com/bugsnag"
    --exclude-dir="com/datadog"
  )

  ENTRYPOINT_PATTERN='(FirebaseAnalytics\.getInstance|FirebaseAnalytics\.newInstance|FirebaseApp\.initializeApp|\.logEvent\s*\(|\.setUserId\s*\(|\.setUserProperty\s*\(|\.setAnalyticsCollectionEnabled|Adjust\.onCreate|Adjust\.trackEvent|AdjustConfig\s*\(|AppsFlyerLib\.getInstance|\.init\s*\(.*AF_DEV_KEY|\.start\s*\(.*AppsFlyerLib|MixpanelAPI\.getInstance|\.track\s*\(.*MixpanelAPI|\.identify\s*\(.*Mixpanel|Amplitude\.getInstance|\.logEvent\s*\(.*Amplitude|Analytics\.with\s*\(|\.track\s*\(.*Segment|\.identify\s*\(.*Segment|Braze\.configure|\.logCustomEvent\s*\(|\.changeUser\s*\(.*Braze|CleverTapAPI\.getDefaultInstance|\.pushEvent\s*\(|\.onUserLogin\s*\(|FlurryAgent\.logEvent|FlurryAgent\.setUserId|FlurryAgent\.Builder)'

  # shellcheck disable=SC2086
  grep -rn --include="*.java" --include="*.kt" "${EXCLUDE_DIRS[@]}" -E "$ENTRYPOINT_PATTERN" "$SOURCE_DIR" 2>/dev/null || true

  echo
  echo "NOTE: Only calls from app code are shown above. Library-internal calls are excluded."
  echo "If no results appear, tracker SDKs may only be initialized internally (e.g., via ContentProvider)."
fi

echo
echo "=== Search complete ==="