import { MCPServer } from '@repo/mcp';
import { z } from 'zod';

// Define resource schemas
const userSchema = z.object({
  id: z.string(),
  type: z.literal('user'),
  attributes: z.object({
    name: z.string(),
    email: z.string().email(),
    role: z.enum(['admin', 'user', 'guest'])
  }).optional()
});

// Create server configuration
const serverConfig = {
  name: 'UserService',
  description: 'Provides user context for AI models',
  resources: {
    user: MCPServer.createResourceDefinition(userSchema, {
      description: 'User resource',
      examples: [{
        id: 'user-123',
        type: 'user',
        attributes: {
          name: 'John Doe',
          email: 'john@example.com',
          role: 'admin'
        }
      }]
    })
  }
};

// Initialize server
const server = new MCPServer(serverConfig);

// Register a single resource
server.registerResource({
  id: 'user-123',
  type: 'user',
  attributes: {
    name: 'John Doe',
    email: 'john@example.com',
    role: 'admin'
  }
});

// Register multiple resources
server.registerResources([
  {
    id: 'user-456',
    type: 'user',
    attributes: {
      name: 'Jane Smith',
      email: 'jane@example.com',
      role: 'user'
    }
  },
  {
    id: 'user-789',
    type: 'user',
    attributes: {
      name: 'Bob Johnson',
      email: 'bob@example.com',
      role: 'guest'
    }
  }
]);

// List all resources
const allResources = server.listResources();

// List resources with pagination
const paginatedResources = server.listResources({
  cursor: 'previousCursor',
  limit: 10
});

// List resources of a specific type
const userResources = server.listResources({
  type: 'user'
});

// Get a specific resource
const user = server.getResource('user', 'user-123');

// Update a resource
server.updateResource({
  id: 'user-123',
  type: 'user',
  attributes: {
    name: 'John Doe Updated',
    email: 'john.updated@example.com',
    role: 'admin'
  }
});

// Delete a resource
server.deleteResource('user', 'user-123');

// In your API route (e.g., app/api/mcp/route.ts)
import { MCPServer } from '@repo/mcp';
import { NextRequest, NextResponse } from 'next/server';

// Create server instance (or import from a shared location)
const server = new MCPServer({ /* config */ });

export async function GET(request: NextRequest) {
  const { searchParams } = new URL(request.url);
  const type = searchParams.get('type');
  const id = searchParams.get('id');
  const cursor = searchParams.get('cursor');
  const limit = searchParams.get('limit') ? parseInt(searchParams.get('limit')!) : undefined;

  // Handle resource listing
  if (!id) {
    const resources = server.listResources({
      type: type || undefined,
      cursor: cursor || undefined,
      limit
    });
    return NextResponse.json(resources);
  }

  // Handle resource retrieval
  if (type && id) {
    const resource = server.getResource(type, id);
    if (!resource) {
      return NextResponse.json({ error: 'Resource not found' }, { status: 404 });
    }
    return NextResponse.json(resource);
  }

  return NextResponse.json({ error: 'Invalid request' }, { status: 400 });
}

export async function POST(request: NextRequest) {
  const body = await request.json();
  
  try {
    const resource = server.registerResource(body);
    return NextResponse.json(resource, { status: 201 });
  } catch (error) {
    return NextResponse.json({ error: (error as Error).message }, { status: 400 });
  }
}