#!/bin/bash

# IMAP/SMTP Email Skill Setup Helper

echo "================================"
echo "  IMAP/SMTP Email Skill Setup"
echo "================================"
echo ""
echo "This script will help you create a .env file with your email credentials."
echo ""

# Prompt for email provider
echo "Select your email provider:"
echo ""
echo "  --- Global ---"
echo "  1) Gmail"
echo "  2) Outlook / Hotmail"
echo "  3) Yahoo Mail"
echo "  4) iCloud Mail"
echo "  5) Zoho Mail"
echo "  6) Fastmail"
echo "  7) AOL Mail"
echo "  8) GMX Mail"
echo "  9) ProtonMail (Bridge)"
echo ""
echo "  --- Japan (日本) ---"
echo " 10) Yahoo! JAPAN"
echo " 11) OCN"
echo " 12) BIGLOBE"
echo " 13) @nifty"
echo " 14) plala"
echo " 15) So-net"
echo " 16) au Mail (@au.com)"
echo " 17) SAKURA Internet"
echo ""
echo "  --- China (中国) ---"
echo " 18) 163.com"
echo " 19) vip.163.com"
echo " 20) 126.com"
echo " 21) vip.126.com"
echo " 22) 188.com"
echo " 23) vip.188.com"
echo " 24) yeah.net"
echo " 25) QQ Mail"
echo ""
echo " 99) Custom"
echo ""
read -p "Enter choice: " PROVIDER_CHOICE

case $PROVIDER_CHOICE in
  1)
    IMAP_HOST="imap.gmail.com"
    IMAP_PORT="993"
    SMTP_HOST="smtp.gmail.com"
    SMTP_PORT="587"
    SMTP_SECURE="false"
    IMAP_TLS="true"
    echo ""
    echo "⚠️  Gmail requires an App Password — your regular Google password will NOT work."
    echo "   1. Go to: https://myaccount.google.com/apppasswords"
    echo "   2. Generate an App Password (requires 2-Step Verification enabled)"
    echo "   3. Use the generated 16-character password below"
    echo ""
    ;;
  2)
    IMAP_HOST="outlook.office365.com"
    IMAP_PORT="993"
    SMTP_HOST="smtp.office365.com"
    SMTP_PORT="587"
    SMTP_SECURE="false"
    IMAP_TLS="true"
    ;;
  3)
    IMAP_HOST="imap.mail.yahoo.com"
    IMAP_PORT="993"
    SMTP_HOST="smtp.mail.yahoo.com"
    SMTP_PORT="465"
    SMTP_SECURE="true"
    IMAP_TLS="true"
    ;;
  4)
    IMAP_HOST="imap.mail.me.com"
    IMAP_PORT="993"
    SMTP_HOST="smtp.mail.me.com"
    SMTP_PORT="587"
    SMTP_SECURE="false"
    IMAP_TLS="true"
    echo ""
    echo "⚠️  iCloud requires an App-Specific Password."
    echo "   1. Go to: https://appleid.apple.com/account/manage"
    echo "   2. Generate an App-Specific Password"
    echo "   3. Use the generated password below"
    echo ""
    ;;
  5)
    IMAP_HOST="imap.zoho.com"
    IMAP_PORT="993"
    SMTP_HOST="smtp.zoho.com"
    SMTP_PORT="465"
    SMTP_SECURE="true"
    IMAP_TLS="true"
    ;;
  6)
    IMAP_HOST="imap.fastmail.com"
    IMAP_PORT="993"
    SMTP_HOST="smtp.fastmail.com"
    SMTP_PORT="465"
    SMTP_SECURE="true"
    IMAP_TLS="true"
    ;;
  7)
    IMAP_HOST="imap.aol.com"
    IMAP_PORT="993"
    SMTP_HOST="smtp.aol.com"
    SMTP_PORT="465"
    SMTP_SECURE="true"
    IMAP_TLS="true"
    ;;
  8)
    IMAP_HOST="imap.gmx.com"
    IMAP_PORT="993"
    SMTP_HOST="mail.gmx.com"
    SMTP_PORT="587"
    SMTP_SECURE="false"
    IMAP_TLS="true"
    ;;
  9)
    IMAP_HOST="127.0.0.1"
    IMAP_PORT="1143"
    SMTP_HOST="127.0.0.1"
    SMTP_PORT="1025"
    SMTP_SECURE="false"
    IMAP_TLS="false"
    echo ""
    echo "⚠️  ProtonMail requires ProtonMail Bridge to be running locally."
    echo ""
    ;;
  10)
    IMAP_HOST="imap.mail.yahoo.co.jp"
    IMAP_PORT="993"
    SMTP_HOST="smtp.mail.yahoo.co.jp"
    SMTP_PORT="465"
    SMTP_SECURE="true"
    IMAP_TLS="true"
    echo ""
    echo "⚠️  Yahoo! JAPANはアプリパスワードが必要です。"
    echo "   Yahoo! JAPAN Mail設定 → IMAP/POP/SMTPアクセスを有効にしてください。"
    echo ""
    ;;
  11)
    IMAP_HOST="imap.ocn.ne.jp"
    IMAP_PORT="993"
    SMTP_HOST="smtp.ocn.ne.jp"
    SMTP_PORT="465"
    SMTP_SECURE="true"
    IMAP_TLS="true"
    ;;
  12)
    IMAP_HOST="mail.biglobe.ne.jp"
    IMAP_PORT="993"
    SMTP_HOST="mail.biglobe.ne.jp"
    SMTP_PORT="465"
    SMTP_SECURE="true"
    IMAP_TLS="true"
    ;;
  13)
    IMAP_HOST="imap.nifty.com"
    IMAP_PORT="993"
    SMTP_HOST="smtp.nifty.com"
    SMTP_PORT="587"
    SMTP_SECURE="false"
    IMAP_TLS="true"
    ;;
  14)
    IMAP_HOST="imap.plala.or.jp"
    IMAP_PORT="993"
    SMTP_HOST="secure.plala.or.jp"
    SMTP_PORT="465"
    SMTP_SECURE="true"
    IMAP_TLS="true"
    ;;
  15)
    IMAP_HOST="imap.so-net.ne.jp"
    IMAP_PORT="993"
    SMTP_HOST="mail.so-net.ne.jp"
    SMTP_PORT="587"
    SMTP_SECURE="false"
    IMAP_TLS="true"
    ;;
  16)
    IMAP_HOST="imap.au.com"
    IMAP_PORT="993"
    SMTP_HOST="smtp.au.com"
    SMTP_PORT="465"
    SMTP_SECURE="true"
    IMAP_TLS="true"
    ;;
  17)
    IMAP_HOST="imap.sakura.ne.jp"
    IMAP_PORT="993"
    SMTP_HOST="smtp.sakura.ne.jp"
    SMTP_PORT="587"
    SMTP_SECURE="false"
    IMAP_TLS="true"
    ;;
  18)
    IMAP_HOST="imap.163.com"
    IMAP_PORT="993"
    SMTP_HOST="smtp.163.com"
    SMTP_PORT="465"
    SMTP_SECURE="true"
    IMAP_TLS="true"
    ;;
  19)
    IMAP_HOST="imap.vip.163.com"
    IMAP_PORT="993"
    SMTP_HOST="smtp.vip.163.com"
    SMTP_PORT="465"
    SMTP_SECURE="true"
    IMAP_TLS="true"
    ;;
  20)
    IMAP_HOST="imap.126.com"
    IMAP_PORT="993"
    SMTP_HOST="smtp.126.com"
    SMTP_PORT="465"
    SMTP_SECURE="true"
    IMAP_TLS="true"
    ;;
  21)
    IMAP_HOST="imap.vip.126.com"
    IMAP_PORT="993"
    SMTP_HOST="smtp.vip.126.com"
    SMTP_PORT="465"
    SMTP_SECURE="true"
    IMAP_TLS="true"
    ;;
  22)
    IMAP_HOST="imap.188.com"
    IMAP_PORT="993"
    SMTP_HOST="smtp.188.com"
    SMTP_PORT="465"
    SMTP_SECURE="true"
    IMAP_TLS="true"
    ;;
  23)
    IMAP_HOST="imap.vip.188.com"
    IMAP_PORT="993"
    SMTP_HOST="smtp.vip.188.com"
    SMTP_PORT="465"
    SMTP_SECURE="true"
    IMAP_TLS="true"
    ;;
  24)
    IMAP_HOST="imap.yeah.net"
    IMAP_PORT="993"
    SMTP_HOST="smtp.yeah.net"
    SMTP_PORT="465"
    SMTP_SECURE="true"
    IMAP_TLS="true"
    ;;
  25)
    IMAP_HOST="imap.qq.com"
    IMAP_PORT="993"
    SMTP_HOST="smtp.qq.com"
    SMTP_PORT="587"
    SMTP_SECURE="false"
    IMAP_TLS="true"
    ;;
  99)
    read -p "IMAP Host: " IMAP_HOST
    read -p "IMAP Port: " IMAP_PORT
    read -p "SMTP Host: " SMTP_HOST
    read -p "SMTP Port: " SMTP_PORT
    read -p "Use TLS for IMAP? (true/false): " IMAP_TLS
    read -p "Use SSL for SMTP? (true/false): " SMTP_SECURE
    ;;
  *)
    echo "Invalid choice"
    exit 1
    ;;
esac

echo ""
read -p "Email address: " EMAIL
read -s -p "Password / App Password / Authorization Code: " PASSWORD
echo ""
read -p "Accept self-signed certificates? (y/n): " ACCEPT_CERT
if [ "$ACCEPT_CERT" = "y" ]; then
  REJECT_UNAUTHORIZED="false"
else
  REJECT_UNAUTHORIZED="true"
fi

read -p "Allowed directories for reading files (comma-separated, e.g. ~/Downloads,~/Documents): " ALLOWED_READ_DIRS
read -p "Allowed directories for saving attachments (comma-separated, e.g. ~/Downloads): " ALLOWED_WRITE_DIRS

# Create .env file
cat > .env << EOF
# IMAP Configuration
IMAP_HOST=$IMAP_HOST
IMAP_PORT=$IMAP_PORT
IMAP_USER=$EMAIL
IMAP_PASS=$PASSWORD
IMAP_TLS=$IMAP_TLS
IMAP_REJECT_UNAUTHORIZED=$REJECT_UNAUTHORIZED
IMAP_MAILBOX=INBOX

# SMTP Configuration
SMTP_HOST=$SMTP_HOST
SMTP_PORT=$SMTP_PORT
SMTP_SECURE=$SMTP_SECURE
SMTP_USER=$EMAIL
SMTP_PASS=$PASSWORD
SMTP_FROM=$EMAIL
SMTP_REJECT_UNAUTHORIZED=$REJECT_UNAUTHORIZED

# File access whitelist (security)
ALLOWED_READ_DIRS=${ALLOWED_READ_DIRS:-$HOME/Downloads,$HOME/Documents}
ALLOWED_WRITE_DIRS=${ALLOWED_WRITE_DIRS:-$HOME/Downloads}
EOF

echo ""
echo "✅ Created .env file"
chmod 600 .env
echo "✅ Set .env file permissions to 600 (owner read/write only)"
echo ""
echo "Testing connections..."
echo ""

# Test IMAP connection
echo "Testing IMAP..."
if node scripts/imap.js list-mailboxes >/dev/null 2>&1; then
    echo "✅ IMAP connection successful!"
else
    echo "❌ IMAP connection test failed"
    echo "   Please check your credentials and settings"
fi

# Test SMTP connection
echo ""
echo "Testing SMTP..."
echo "  (This will send a test email to your own address: $EMAIL)"
if node scripts/smtp.js test >/dev/null 2>&1; then
    echo "✅ SMTP connection successful!"
else
    echo "❌ SMTP connection test failed"
    echo "   Please check your credentials and settings"
fi

echo ""
echo "Setup complete! Try:"
echo "  node scripts/imap.js check"
echo "  node scripts/smtp.js send --to recipient@example.com --subject Test --body 'Hello World'"