#!/bin/bash
#
# post-message.sh - Send a message to Microsoft Teams with error handling
#
# Usage:
#   ./post-message.sh <channel-id> <message>
#   ./post-message.sh --channel-name <name> <message>
#
# Example:
#   ./post-message.sh "19:abc123@thread.tacv2" "Hello from script!"
#   ./post-message.sh --channel-name General "Deployment completed"
#
# NOTE: Teams tokens expire in 60-90 minutes! This script handles token refresh.

set -euo pipefail

# Colors for output
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
NC='\033[0m' # No Color

# Parse arguments
CHANNEL_ID=""
CHANNEL_NAME=""
MESSAGE=""

if [ "$1" = "--channel-name" ]; then
  if [ $# -lt 3 ]; then
    echo "Usage: $0 --channel-name <name> <message>"
    echo ""
    echo "Example:"
    echo "  $0 --channel-name General 'Build completed'"
    echo ""
    echo "NOTE: Teams tokens expire in 60-90 minutes!"
    exit 1
  fi
  CHANNEL_NAME="$2"
  MESSAGE="$3"
elif [ $# -lt 2 ]; then
  echo "Usage: $0 <channel-id> <message>"
  echo "       $0 --channel-name <name> <message>"
  echo ""
  echo "Examples:"
  echo "  $0 '19:abc123@thread.tacv2' 'Hello world!'"
  echo "  $0 --channel-name General 'Build completed'"
  echo ""
  echo "NOTE: Teams tokens expire in 60-90 minutes!"
  exit 1
else
  CHANNEL_ID="$1"
  MESSAGE="$2"
fi

# Function to refresh token if needed
refresh_token_if_needed() {
  local status
  status=$(agent-teams auth status 2>&1)
  
  # Check if token is expired or expiring soon
  if echo "$status" | jq -e '.error' > /dev/null 2>&1; then
    echo -e "${YELLOW}Token expired or invalid, refreshing...${NC}"
    agent-teams auth extract
    return 0
  fi
  
  local expires_soon
  expires_soon=$(echo "$status" | jq -r '.token_expires_soon // false')
  
  if [ "$expires_soon" = "true" ]; then
    echo -e "${YELLOW}Token expiring soon, refreshing...${NC}"
    agent-teams auth extract
  fi
}

# Function to send message with retry logic
send_message() {
  local channel_id=$1
  local message=$2
  local max_attempts=3
  local attempt=1
  local token_refresh_attempts=0
  local max_token_refresh=2
  
  while [ $attempt -le $max_attempts ]; do
    echo -e "${YELLOW}Attempt $attempt/$max_attempts...${NC}"
    
    # Send message and capture result
    RESULT=$(agent-teams message send "$channel_id" "$message" 2>&1)
    
    # Check if successful (has an 'id' field)
    if echo "$RESULT" | jq -e '.id' > /dev/null 2>&1; then
      echo -e "${GREEN}Message sent successfully!${NC}"
      
      # Extract message details
      MSG_ID=$(echo "$RESULT" | jq -r '.id')
      
      echo ""
      echo "Message details:"
      echo "  Channel: $channel_id"
      echo "  Message ID: $MSG_ID"
      
      return 0
    fi
    
    # Extract error information
    if echo "$RESULT" | jq -e '.error' > /dev/null 2>&1; then
      ERROR_MSG=$(echo "$RESULT" | jq -r '.error // "Unknown error"')
      echo -e "${RED}Failed: $ERROR_MSG${NC}"
      
      # Handle token expiry - refresh and retry (with a limit on refresh attempts)
      if echo "$ERROR_MSG" | grep -Eqi "expired|401|unauthorized"; then
        token_refresh_attempts=$((token_refresh_attempts + 1))
        if [ $token_refresh_attempts -le $max_token_refresh ]; then
          echo -e "${YELLOW}Token expired, refreshing (attempt $token_refresh_attempts/$max_token_refresh)...${NC}"
          agent-teams auth extract
          continue
        else
          echo -e "${RED}Token refresh failed after $max_token_refresh attempts${NC}"
          return 1
        fi
      fi
      
      # Don't retry on certain errors
      if echo "$ERROR_MSG" | grep -q "Not authenticated"; then
        echo ""
        echo "Not authenticated. Run:"
        echo "  agent-teams auth extract"
        return 1
      fi
      
      if echo "$ERROR_MSG" | grep -q "Channel not found"; then
        echo ""
        echo "Channel '$channel_id' not found. Check channel ID."
        echo "List channels with: agent-teams channel list"
        return 1
      fi
    else
      echo -e "${RED}Unexpected error: $RESULT${NC}"
    fi
    
    # Exponential backoff before retry
    if [ $attempt -lt $max_attempts ]; then
      SLEEP_TIME=$((attempt * 2))
      echo "Retrying in ${SLEEP_TIME}s..."
      sleep $SLEEP_TIME
    fi
    
    attempt=$((attempt + 1))
  done
  
  echo -e "${RED}Failed after $max_attempts attempts${NC}"
  return 1
}

# Check if agent-teams is installed
if ! command -v agent-teams &> /dev/null; then
  echo -e "${RED}Error: agent-teams not found${NC}"
  echo ""
  echo "Install it with:"
  echo "  bun install -g agent-teams"
  exit 1
fi

# Check and refresh authentication
echo "Checking authentication..."
refresh_token_if_needed

AUTH_STATUS=$(agent-teams auth status 2>&1)

if echo "$AUTH_STATUS" | jq -e '.error' > /dev/null 2>&1; then
  echo -e "${RED}Not authenticated!${NC}"
  echo ""
  echo "Run this to authenticate:"
  echo "  agent-teams auth extract"
  exit 1
fi

USER_NAME=$(echo "$AUTH_STATUS" | jq -r '.user // "Unknown"')
TEAM_ID=$(echo "$AUTH_STATUS" | jq -r '.current_team // ""')
TOKEN_AGE=$(echo "$AUTH_STATUS" | jq -r '.token_age_minutes // "unknown"')
echo -e "${GREEN}Authenticated as: $USER_NAME${NC}"
echo -e "Token age: ${TOKEN_AGE} minutes"

if [ -z "$TEAM_ID" ]; then
  echo -e "${RED}No team selected!${NC}"
  echo ""
  echo "Run this to select a team:"
  echo "  agent-teams team list"
  echo "  agent-teams team switch <team-id>"
  exit 1
fi

# If channel name provided, look up channel ID
if [ -n "$CHANNEL_NAME" ]; then
  echo "Looking up channel #$CHANNEL_NAME..."
  CHANNELS=$(agent-teams channel list 2>&1)
  
  if echo "$CHANNELS" | jq -e '.error' > /dev/null 2>&1; then
    echo -e "${RED}Failed to list channels${NC}"
    exit 1
  fi
  
  CHANNEL_ID=$(echo "$CHANNELS" | jq -r --arg name "$CHANNEL_NAME" '.[] | select(.name==$name) | .id')
  
  if [ -z "$CHANNEL_ID" ]; then
    echo -e "${RED}Channel #$CHANNEL_NAME not found${NC}"
    echo ""
    echo "Available channels:"
    echo "$CHANNELS" | jq -r '.[] | "  #\(.name) (\(.id))"'
    exit 1
  fi
  
  echo -e "${GREEN}Found channel: $CHANNEL_ID${NC}"
fi

echo ""

# Send the message
echo "Sending message to channel $CHANNEL_ID..."
echo "Message: $MESSAGE"
echo ""

send_message "$CHANNEL_ID" "$MESSAGE"