import { mcp } from '@routecraft/ai'
import { craft, noop } from '@routecraft/routecraft'
import { z } from 'zod'

// Tool 1: Search flights
craft()
  .id('search-flights')
  .from(mcp('search-flights', {
    description: 'Search for available flights',
    schema: z.object({
      from: z.string().describe('Departure airport code'),
      to: z.string().describe('Destination airport code'),
      date: z.string().describe('Travel date YYYY-MM-DD'),
      passengers: z.number().default(1)
    }),
    keywords: ['flight', 'travel', 'booking', 'airline']
  }))
  .process(async (exchange) => {
    const { from, to, date, passengers } = exchange.body
    const flights = await searchFlightAPI({
      origin: from,
      destination: to,
      date,
      passengers
    })
    return {
      ...exchange,
      body: {
        flights: flights.slice(0, 5).map(f => ({
          airline: f.airline,
          flightNumber: f.number,
          departure: f.departureTime,
          arrival: f.arrivalTime,
          price: f.price,
          duration: f.duration
        }))
      }
    }
  })
  .to(noop())

// Tool 2: Book restaurant
craft()
  .id('book-restaurant')
  .from(mcp('book-restaurant', {
    description: 'Find and book a restaurant reservation',
    schema: z.object({
      cuisine: z.string(),
      city: z.string(),
      date: z.string(),
      time: z.string(),
      partySize: z.number()
    }),
    keywords: ['restaurant', 'dining', 'reservation', 'food']
  }))
  .process(async (exchange) => {
    const { city, cuisine, date, time, partySize } = exchange.body
    // Search OpenTable or similar
    const restaurants = await searchRestaurants({
      location: city,
      cuisine,
      date,
      time,
      partySize
    })
    
    // Book the best match
    const booking = await bookTable(restaurants[0].id, {
      date,
      time,
      partySize
    })
    
    return {
      ...exchange,
      body: {
        restaurant: restaurants[0].name,
        address: restaurants[0].address,
        time: booking.time,
        confirmationCode: booking.code
      }
    }
  })
  .to(noop())

// Tool 3: Hotel search
craft()
  .id('search-hotels')
  .from(mcp('search-hotels', {
    description: 'Search for hotels in a city',
    schema: z.object({
      city: z.string(),
      checkIn: z.string(),
      checkOut: z.string(),
      guests: z.number().default(1),
      maxPrice: z.number().optional()
    }),
    keywords: ['hotel', 'accommodation', 'booking', 'lodging']
  }))
  .process(async (exchange) => {
    const hotels = await searchHotels(exchange.body)
    return {
      ...exchange,
      body: {
        hotels: hotels.slice(0, 5).map(h => ({
          name: h.name,
          rating: h.rating,
          pricePerNight: h.price,
          amenities: h.amenities,
          availability: h.available
        }))
      }
    }
  })
  .to(noop())