trading_bot_v3/app/api/trading/route.ts

import { NextRequest, NextResponse } from 'next/server'
import { driftTradingService } from '../../../lib/drift-trading'

export async function POST(req: NextRequest) {
  try {
    const params = await req.json()
    
    // Ensure user is logged in before executing trade
    const loginStatus = await driftTradingService.login()
    if (!loginStatus.isLoggedIn) {
      return NextResponse.json(
        { error: `Cannot execute trade: ${loginStatus.error}` }, 
        { status: 401 }
      )
    }

    const result = await driftTradingService.executeTrade(params)
    return NextResponse.json(result)
  } catch (e: any) {
    return NextResponse.json({ error: e.message }, { status: 500 })
  }
}

export async function GET() {
  try {
    // Ensure user is logged in before getting positions
    const loginStatus = await driftTradingService.login()
    if (!loginStatus.isLoggedIn) {
      return NextResponse.json(
        { error: `Cannot get positions: ${loginStatus.error}` }, 
        { status: 401 }
      )
    }

    const positions = await driftTradingService.getPositions()
    return NextResponse.json({ positions })
  } catch (e: any) {
    return NextResponse.json({ error: e.message }, { status: 500 })
  }
}