copy-user-id-from-browser.sh

#!/bin/bash

# Required parameters:
# @raycast.schemaVersion 1
# @raycast.title Copy UserID from Admin Page
# @raycast.mode fullOutput

# Optional parameters:
# @raycast.icon 🆔

# Documentation:
# @raycast.description Extract the numeric UserID from the currently open admin page in a Chromium-based browser and copy it to the clipboard.
# @raycast.author Jorrit Harmamny

set -euo pipefail

# Prefer these browsers in order if multiple are running
BROWSERS=(
  "Google Chrome"
  "Brave Browser"
  "Microsoft Edge"
  "Arc"
  "Chromium"
  "Dia"
)

get_page_text() {
  local app_name="$1"
  /usr/bin/osascript - "$app_name" <<'APPLESCRIPT'
on run argv
  set appName to item 1 of argv
  try
    using terms from application "Google Chrome"
      tell application appName
        if (count of windows) is 0 then error "No windows"
        set theTab to active tab of front window
        set pageURL to URL of theTab
        set pageTitle to title of theTab
        set pageText to ""
        try
          set pageText to execute theTab javascript "
            try {
              const text = [];

              // New admin UI: profile page — .ud-stat-value--mono holds '#ID'
              const monoStat = document.querySelector('.ud-stat-value--mono');
              if (monoStat) {
                const val = monoStat.textContent.trim().replace(/^#/, '');
                if (/^\\d{4,}$/.test(val)) text.push('ID:' + val);
              }

              // New admin UI: search results — XHR to first profile link in results table
              if (text.length === 0 && window.location.search.includes('commit=Filter')) {
                const scope = document.querySelector('#index_table_users tbody') || document.querySelector('#index_table_users') || document.querySelector('table.index_table') || document;
                const link = Array.from(scope.querySelectorAll('a[href]'))
                  .find(function(a) {
                    return /\\/admin\\/users\\/(?!new(?:[^a-z]|$))(?!new_)[^?#\\/]+$/.test(a.href);
                  });
                if (link) {
                  const xhr = new XMLHttpRequest();
                  xhr.open('GET', link.href, false);
                  xhr.send();
                  if (xhr.status === 200) {
                    const tmp = document.createElement('div');
                    tmp.innerHTML = xhr.responseText;
                    const el = tmp.querySelector('.ud-stat-value--mono');
                    if (el) {
                      const val = el.textContent.trim().replace(/^#/, '');
                      if (/^\\d{4,}$/.test(val)) text.push('ID:' + val);
                    }
                  }
                }
              }

              // Old admin UI: table rows with numeric IDs in first cell
              if (text.length === 0) {
                document.querySelectorAll('tr').forEach(function(row) {
                  const cells = row.querySelectorAll('td');
                  if (cells.length > 0) {
                    const firstCell = cells[0].textContent.trim();
                    if (/^\\d{1,10}$/.test(firstCell)) text.push('ID:' + firstCell);
                  }
                });
              }

              // Data attributes
              document.querySelectorAll('[data-user-id], [data-id]').forEach(function(el) {
                const uid = el.getAttribute('data-user-id') || el.getAttribute('data-id');
                if (uid && /^\\d{1,10}$/.test(uid)) text.push('ID:' + uid);
              });

              if (text.length === 0) text.push(document.body.innerText || '');
              text.join('\\n');
            } catch(e) {
              document.body.innerText || '';
            }
          "
          if pageText is missing value then set pageText to ""
        end try
        return pageURL & "\n---TITLE---\n" & pageTitle & "\n---TEXT---\n" & pageText
      end tell
    end using terms from
  on error errMsg number errNum
    return "ERROR: " & errMsg
  end try
end run
APPLESCRIPT
}

running_app=""
content=""

for app in "${BROWSERS[@]}"; do
  result=$(get_page_text "$app") || true
  if [[ "$result" != ERROR:* ]]; then
    running_app="$app"
    content="$result"
    break
  fi
done

if [[ -z "$running_app" || -z "$content" ]]; then
  echo "No supported Chromium-based browser found with an open page."
  exit 1
fi

page_url=$(printf "%s" "$content" | awk 'NR==1{print; exit}')
page_title=$(printf "%s" "$content" | awk '/^---TITLE---$/{getline; print; exit}')
page_text=$(printf "%s" "$content" | awk 'f{print} /^---TEXT---$/{f=1}')

echo "Debug: Using browser: $running_app" >&2

# Try several extraction strategies:
# 1) Explicit ID: prefix injected by our JavaScript (highest confidence)
from_id_prefix=$(printf "%s" "$page_text" | grep -Eo '^ID:[0-9]{4,12}' | head -n1 | grep -Eo '[0-9]+' || true)

# 2) From URL paths and query parameters (numeric IDs only — old admin UI)
from_url=$(printf "%s\n%s\n" "$page_url" "$page_title" | grep -Eo '/users/([0-9]{1,10})|[?&]id=([0-9]{1,10})|/admin/users/([0-9]{1,10})|user_id=([0-9]{1,10})' | grep -Eo '[0-9]{1,10}' | head -n1 || true)

# 3) From page text with expanded patterns
from_context=$(printf "%s" "$page_text" | grep -Eio '(ID:|user.?id|user.?number|uid|^[0-9]{1,10}$)[:#\s]*[0-9]{1,10}|\bid[:#\s]*[0-9]{1,10}' | grep -Eo '[0-9]{1,10}' | head -n1 || true)

# 2b) Try to extract from title if it contains "User ID" or similar
from_title=$(printf "%s" "$page_title" | grep -Eio 'user.?id[:#\s]*[0-9]{1,10}' | grep -Eo '[0-9]{1,10}' | head -n1 || true)

# 2c) Parse StackBlitz admin rows where the ID is the first column
from_table=$(printf "%s" "$page_text" | python3 - <<'PY' || true
import sys
import re

text = sys.stdin.read()
lines = [ln.strip() for ln in text.splitlines()]
month_prefix = ('jan', 'feb', 'mar', 'apr', 'may', 'jun', 'jul', 'aug', 'sep', 'oct', 'nov', 'dec')
candidates = []

for idx, line in enumerate(lines):
    if not line:
        continue
    m = re.match(r'^(\d{1,10})\b(.*)', line)
    if not m:
        continue
    user_id = m.group(1)
    rest = m.group(2).strip()
    if not rest:
        continue
    lower = rest.lower()
    if lower.startswith(month_prefix):
        continue
    score = 1
    if 'sentry' in lower or 'username' in lower or 'email' in lower or '@' in rest:
        score += 3
    if idx + 1 < len(lines):
        neighbor = lines[idx + 1].lower()
        if '@' in lines[idx + 1]:
            score += 2
        if 'view' in neighbor:
            score += 1
    if idx + 2 < len(lines):
        if 'view' in lines[idx + 2].lower():
            score += 1
    candidates.append((score, idx, user_id))

if candidates:
    candidates.sort(key=lambda item: (-item[0], item[1]))
    print(candidates[0][2])
PY
)

# Debug with more context
echo "Debug: Extraction attempts:" >&2
echo "ID prefix extraction attempt: '$from_id_prefix'" >&2
echo "URL extraction attempt: '$from_url'" >&2
echo "Context extraction attempt: '$from_context'" >&2
echo "Full URL: $page_url" >&2
echo "Page text length: $(echo "$page_text" | wc -c) bytes" >&2
echo "Page text sample:" >&2
printf "%s" "$page_text" | head -c 500 >&2
echo >&2

# 3) Fallback: Try to find any number that looks like a user ID (skip years)
from_bigint=$(printf "%s" "$page_text" | python3 - <<'PY' || true
import sys
import re

text = sys.stdin.read()
for match in re.finditer(r'\b\d{4,}\b', text):
    val = match.group()
    num = int(val)
    if 1900 <= num <= 2100:
        continue
    print(val)
    break
PY
)

# 4) If page text was empty, try selection copy
from_selection=""
if [[ -z "$from_url" && -z "$from_context" && -z "$from_bigint" ]]; then
  if [[ -z "$page_text" ]]; then
    echo "Attempting clipboard selection method..." >&2
    sel=$(\
      /usr/bin/osascript - "$running_app" <<'APPLESCRIPT'
on run argv
  set appName to item 1 of argv
  tell application "System Events"
    tell process appName
      set frontmost to true
      keystroke "c" using {command down}
    end tell
  end tell
  delay 0.2
  return "OK"
end run
APPLESCRIPT
    ) || true
    if [[ "$sel" == OK ]]; then
      from_selection=$(pbpaste | grep -Eo '\b[0-9]{4,}\b' | head -n1 || true)
      echo "Selection content: $(pbpaste | head -c 50)..." >&2
    fi
  fi
fi

USERID=""
if [[ -n "$from_id_prefix" ]]; then
  USERID="$from_id_prefix"
elif [[ -n "$from_url" ]]; then
  USERID="$from_url"
elif [[ -n "$from_table" ]]; then
  USERID="$from_table"
elif [[ -n "$from_title" ]]; then
  USERID="$from_title"
elif [[ -n "$from_context" ]]; then
  USERID="$from_context"
elif [[ -n "$from_bigint" ]]; then
  USERID="$from_bigint"
elif [[ -n "$from_selection" ]]; then
  USERID="$from_selection"
fi

if [[ -z "$USERID" ]]; then
  echo "Could not find a numeric UserID on the current page."
  echo "URL: $page_url"
  echo "Title: $page_title"
  if [[ -z "$page_text" ]]; then
    echo "Trying fallback: copy entire page text..."
    all_copy=$(\
      /usr/bin/osascript - "$running_app" <<'APPLESCRIPT'
on run argv
  set appName to item 1 of argv
  try
    using terms from application "Google Chrome"
      tell application appName to activate
    end using terms from
    tell application "System Events"
      keystroke "a" using {command down}
      delay 0.05
      keystroke "c" using {command down}
    end tell
    delay 0.2
    return "OK"
  on error errMsg
    return "ERROR: " & errMsg
  end try
end run
APPLESCRIPT
    ) || true
    if [[ "$all_copy" == OK ]]; then
      # For StackBlitz admin pages, look for IDs at the start of a line (table format)
      # or after common patterns like "Id" headers
      clipboard_content=$(pbpaste)
      
      echo "Clipboard content sample (first 500 chars):" >&2
      echo "$clipboard_content" | head -c 500 >&2
      echo >&2
      
      # Try multiple extraction strategies
      # 1. Parse rows similar to the page text parser
      candidate=$(printf "%s" "$clipboard_content" | python3 - <<'PY' || true
import sys
import re

text = sys.stdin.read()
lines = [ln.strip() for ln in text.splitlines()]
month_prefix = ('jan', 'feb', 'mar', 'apr', 'may', 'jun', 'jul', 'aug', 'sep', 'oct', 'nov', 'dec')
candidates = []

for idx, line in enumerate(lines):
    if not line:
        continue
    m = re.match(r'^(\d{1,10})\b(.*)', line)
    if not m:
        continue
    user_id = m.group(1)
    rest = m.group(2).strip()
    if not rest:
        continue
    lower = rest.lower()
    if lower.startswith(month_prefix):
        continue
    score = 1
    if 'sentry' in lower or 'username' in lower or 'email' in lower or '@' in rest:
        score += 3
    if idx + 1 < len(lines):
        neighbor = lines[idx + 1].lower()
        if '@' in lines[idx + 1]:
            score += 2
        if 'view' in neighbor:
            score += 1
    if idx + 2 < len(lines):
        if 'view' in lines[idx + 2].lower():
            score += 1
    candidates.append((score, idx, user_id))

if candidates:
    candidates.sort(key=lambda item: (-item[0], item[1]))
    print(candidates[0][2])
PY
      )
      
      # 2. ID in table format (number followed by tabs/spaces and other fields)
      if [[ -z "$candidate" ]]; then
        candidate=$(echo "$clipboard_content" | grep -E '^[0-9]{1,10}[[:space:]]' | grep -Eo '^[0-9]{1,10}' | head -n1 || true)
      fi
      
      # 2. ID after "Id" column header in table
      if [[ -z "$candidate" ]]; then
        candidate=$(echo "$clipboard_content" | grep -A 3 -E '^Id[[:space:]]' | grep -Eo '^[0-9]{1,10}' | head -n1 || true)
      fi
      
      # 3. Look for lines with email and extract any digits before/after
      if [[ -z "$candidate" ]]; then
        # Find line with email, then look for 4+ digit number on same or adjacent lines
        email_line=$(echo "$clipboard_content" | grep -n '@' | head -n1 | cut -d: -f1)
        if [[ -n "$email_line" ]]; then
          # Get context around email line (5 lines before and after)
          start_line=$((email_line-5))
          if (( start_line < 1 )); then
            start_line=1
          fi
          candidate=$(echo "$clipboard_content" | sed -n "${start_line},$((email_line+5))p" | grep -Eo '\b[0-9]{1,10}\b' | head -n1 || true)
        fi
      fi
      
      # 4. Try to find number-email pattern in a single line
      if [[ -z "$candidate" ]]; then
        candidate=$(echo "$clipboard_content" | grep -E '[0-9]{1,10}.*@' | grep -Eo '\b[0-9]{1,10}\b' | head -n1 || true)
      fi
      
      # 5. Fallback: just find any 4+ digit number (but skip common false positives like years)
      if [[ -z "$candidate" ]]; then
        candidate=$(echo "$clipboard_content" | grep -Eo '\b[0-9]{4,}\b' | head -n1 || true)
      fi

      # 6. As a last resort, open view-source, copy the HTML, and parse for IDs
      if [[ -z "$candidate" ]]; then
        echo "Attempting view-source extraction..." >&2
        view_source=$(\
          /usr/bin/osascript - "$running_app" <<'APPLESCRIPT'
on run argv
  set appName to item 1 of argv
  try
    using terms from application "Google Chrome"
      tell application appName to activate
    end using terms from
    tell application "System Events"
      tell process appName
        set frontmost to true
        keystroke "u" using {command down, option down}
      end tell
    end tell
    delay 0.5
    tell application "System Events"
      tell process appName
        keystroke "a" using {command down}
        delay 0.05
        keystroke "c" using {command down}
        delay 0.05
        keystroke "w" using {command down}
      end tell
    end tell
    delay 0.2
    return "OK"
  on error errMsg
    return "ERROR: " & errMsg
  end try
end run
APPLESCRIPT
        ) || true
        if [[ "$view_source" == OK ]]; then
          clipboard_html=$(pbpaste)
          candidate=$(printf "%s" "$clipboard_html" | python3 - <<'PY' || true
import sys
import re

text = sys.stdin.read()
patterns = [
    r'data-(?:user-)?id="(\d{1,10})"',
    r'<td[^>]*\bcol-id\b[^>]*>\s*(\d{1,10})\s*<',
    r'href="/admin/users/(\d{1,10})"',
]
candidates = []
for pattern in patterns:
    for match in re.finditer(pattern, text, flags=re.IGNORECASE):
        candidates.append(match.group(1))
        if len(candidates) > 10:
            break
    if candidates:
        break

if not candidates:
    match = re.search(r'\b(\d{5,})\b', text)
    if match:
        candidates.append(match.group(1))

if candidates:
    print(candidates[0])
PY
          )
        fi
      fi
      
      echo "Clipboard extraction result: '$candidate'" >&2
      
      if [[ -n "$candidate" ]]; then
        USERID="$candidate"
      fi
    fi
    if [[ -z "$USERID" ]]; then
      echo "Tip: Enable 'Allow JavaScript from Apple Events' in your browser's Developer settings, or select the row with the numeric ID and rerun."
      exit 2
    fi
  else
    exit 2
  fi
fi

printf "%s" "$USERID" | pbcopy

# Capture username from /admin/users/<username> profile URL (used for the
# token-allocation form, which keys on username rather than ID).
USERNAME=""
if [[ "$page_url" =~ /admin/users/([^?#/]+)/?$ ]]; then
  candidate="${BASH_REMATCH[1]}"
  if [[ ! "$candidate" =~ ^[0-9]+$ && "$candidate" != "new" && ! "$candidate" =~ ^new_ ]]; then
    USERNAME="$candidate"
  fi
fi

echo "Detected browser: $running_app"
echo "Page: $page_title"
echo "URL: $page_url"
echo "Copied UserID to clipboard: $USERID"
[[ -n "$USERNAME" ]] && echo "Detected username: $USERNAME"

# Post-action menu. Use `choose from list` (vs display dialog's 3-button cap)
# so we can add Token Allocation alongside Rate Limits / Reset Tokens.
# Cancel must NOT exit non-zero (set -e + bolt-admin.sh's retry loop would
# otherwise re-spawn this dialog up to max_attempts times).
prompt_text="UserID: ${USERID}"
[[ -n "$USERNAME" ]] && prompt_text="${prompt_text} (${USERNAME})"
menu_choice=$(/usr/bin/osascript \
  -e 'on run argv' \
  -e '  set promptText to item 1 of argv' \
  -e '  tell application "System Events" to activate' \
  -e '  set theList to {"Rate Limits", "Reset Tokens", "Add Token Allocation"}' \
  -e '  set choiceResult to choose from list theList with title "Bolt Admin" with prompt promptText default items {"Rate Limits"} OK button name "Open" cancel button name "Cancel"' \
  -e '  if choiceResult is false then return ""' \
  -e '  return (item 1 of choiceResult) as text' \
  -e 'end run' \
  -- "$prompt_text" 2>/dev/null) || menu_choice=""

case "$menu_choice" in
  "Rate Limits")
    # Open rate limits page (requires authentication)
    open "https://bolt.new/api/rate-limits/$USERID"
    ;;
    
  "Reset Tokens")
    # First open the current rate limits page
    open "https://bolt.new/api/rate-limits/$USERID"
    sleep 0.8
    
    reset_choice=$(/usr/bin/osascript -e 'tell application "System Events" to activate' \
      -e "display dialog \"Choose token reset type for ${USERID}\" with title \"Reset Tokens\" buttons {\"Cancel\", \"All\", \"Monthly\"} default button \"Monthly\" cancel button \"Cancel\"" \
      -e 'return button returned of result' 2>/dev/null) || reset_choice=""
    
    case "$reset_choice" in
      "Monthly")
        confirm=$(/usr/bin/osascript -e 'tell application "System Events" to activate' \
          -e "display dialog \"Confirm monthly token reset for ${USERID}?\" with title \"Confirm Reset\" buttons {\"Cancel\", \"Confirm\"} default button \"Confirm\" cancel button \"Cancel\"" \
          -e 'return button returned of result' 2>/dev/null) || confirm=""
        
        if [[ "$confirm" == "Confirm" ]]; then
          # Reset monthly tokens
          open "https://bolt.new/api/rate-limits/reset/$USERID/month"
          echo "Opened monthly token reset endpoint."
          sleep 1
          # Show updated rate limits
          open "https://bolt.new/api/rate-limits/$USERID"
        else
          echo "Cancelled."
        fi
        ;;
        
      "All")
        confirm=$(/usr/bin/osascript -e 'tell application "System Events" to activate' \
          -e "display dialog \"Confirm ALL token reset (including rollovers) for ${USERID}?\" with title \"Confirm Reset\" buttons {\"Cancel\", \"Confirm\"} default button \"Confirm\" cancel button \"Cancel\"" \
          -e 'return button returned of result' 2>/dev/null) || confirm=""
        
        if [[ "$confirm" == "Confirm" ]]; then
          # Reset all tokens
          open "https://bolt.new/api/rate-limits/reset/$USERID/all"
          echo "Opened ALL token reset endpoint."
          sleep 1
          # Show updated rate limits
          open "https://bolt.new/api/rate-limits/$USERID"
        else
          echo "Cancelled."
        fi
        ;;
      *)
        echo "Cancelled."
        ;;
    esac
    ;;

  "Add Token Allocation")
    if [[ -z "$USERNAME" ]]; then
      /usr/bin/osascript \
        -e 'tell application "System Events" to activate' \
        -e 'display alert "No username available" message "Run the lookup from a /admin/users/<username> profile page first."' \
        2>/dev/null || true
    else
      millions=$(/usr/bin/osascript \
        -e 'tell application "System Events" to activate' \
        -e "display dialog \"Token amount (millions, 1-200) for ${USERNAME}\" with title \"Token Allocation\" default answer \"10\" buttons {\"Cancel\", \"OK\"} default button \"OK\" cancel button \"Cancel\"" \
        -e 'return text returned of result' 2>/dev/null) || millions=""
      millions=$(printf '%s' "$millions" | tr -d '[:space:]')

      if [[ -z "$millions" ]]; then
        echo "Cancelled."
      elif ! [[ "$millions" =~ ^([1-9]|[1-9][0-9]|1[0-9]{2}|200)$ ]]; then
        /usr/bin/osascript \
          -e 'tell application "System Events" to activate' \
          -e "display alert \"Invalid token amount\" message \"Enter an integer between 1 and 200 (millions). Got: ${millions}\"" \
          2>/dev/null || true
      else
        tokens=$(( millions * 1000000 ))
        # starts_at = now (UTC), expires_at = now + 2 months (UTC).
        # macOS BSD `date -v` advances calendar months with end-of-month clamping.
        starts_at=$(date -u +"%Y-%m-%d %H:%M:%S UTC")
        expires_at=$(date -u -v+2m +"%Y-%m-%d %H:%M:%S UTC")

        echo "Opening allocation form for ${USERNAME}: ${tokens} tokens, expires ${expires_at}"
        open "https://stackblitz.com/admin/user_token_allocations/new"
        sleep 2

        # Build the JS payload in bash so we control quoting; pass it as one argv
        # to osascript and have AppleScript just hand it to `execute theTab javascript`.
        # Username/tokens/date are sanitized inputs, but escape defensively anyway.
        js_escape() { python3 -c "import json,sys;print(json.dumps(sys.argv[1]))" "$1"; }
        js_user=$(js_escape "$USERNAME")
        js_tokens=$(js_escape "$tokens")
        js_expires=$(js_escape "$expires_at")
        js_code="(function(){function setVal(id,v){var el=document.getElementById(id);if(!el)return false;el.focus();el.value=v;el.dispatchEvent(new Event('input',{bubbles:true}));el.dispatchEvent(new Event('change',{bubbles:true}));el.blur();return true;}var ok=true;ok=setVal('user_token_allocation_user_id',${js_user})&&ok;ok=setVal('user_token_allocation_label','support-tokens')&&ok;ok=setVal('user_token_allocation_tokens',${js_tokens})&&ok;ok=setVal('user_token_allocation_expires_at',${js_expires})&&ok;return ok;})()"

        /usr/bin/osascript - "$running_app" "$js_code" <<'APPLESCRIPT' >/dev/null 2>&1 || true
on run argv
  set appName to item 1 of argv
  set jsCode to item 2 of argv
  try
    using terms from application "Google Chrome"
      tell application appName
        if (count of windows) is 0 then error "No windows"
        set theTab to active tab of front window
        execute theTab javascript jsCode
      end tell
    end using terms from
  end try
end run
APPLESCRIPT
      fi
    fi
    ;;

  *)
    # Cancel or closed dialog
    ;;
esac