import { HeimdallClient } from 'hmdl';

// Initialize (uses environment variables by default)
const client = new HeimdallClient();

// Or with explicit configuration
const client = new HeimdallClient({
  endpoint: 'http://localhost:4318',
  orgId: 'your-org-id',           // From Settings page
  projectId: 'your-project-id',   // From Settings page
  serviceName: 'my-mcp-server',
  environment: 'development',
});

import { traceMCPTool } from 'hmdl';

const searchDocuments = traceMCPTool(
  async (query: string, limit: number = 10) => {
    // Your implementation here
    return results;
  },
  { name: 'search-documents', paramNames: ['query', 'limit'] }
);

const anotherTool = traceMCPTool(
  async (data: Record<string, unknown>) => {
    return { processed: true, ...data };
  },
  { name: 'another-tool', paramNames: ['data'] }
);

import { HeimdallClient, MCPTool } from 'hmdl';

// Initialize client first
new HeimdallClient();

class MyMCPServer {
  @MCPTool()
  async searchDocuments(query: string, limit: number = 10) {
    // Your implementation
    return results;
  }

  @MCPTool({ name: 'custom-tool-name' })
  async anotherTool(data: unknown) {
    return processedData;
  }
}

// Ensure spans are flushed before exit
process.on('beforeExit', async () => {
  await client.flush();
});

// Or for graceful shutdown
process.on('SIGTERM', async () => {
  await client.shutdown();
  process.exit(0);
});

import { traceMCPTool } from 'hmdl';

app.post('/mcp', async (req, res) => {
  const searchTool = traceMCPTool(async (query: string) => {
    return results;
  }, {
    name: 'search',
    headers: req.headers  // Automatically extracts session/user
  });

  const result = await searchTool('test');
  res.json(result);
});

const myTool = traceMCPTool(
  (ctx: { sessionId?: string; userId?: string }, query: string) => {
    return `Query: ${query}`;
  },
  {
    name: 'my-tool',
    // Context is the first argument (args[0])
    sessionExtractor: (args) => args[0]?.sessionId,
    userExtractor: (args) => args[0]?.userId,
  }
);

import { HeimdallClient } from 'hmdl';

const client = new HeimdallClient();

await client.startSpan('my-operation', async (span) => {
  span.setAttribute('custom.attribute', 'value');
  // Your code here
  return result;
});

const myTool = traceMCPTool(fn, {
  name: 'custom-name',      // Custom span name
  paramNames: ['query', 'limit'],  // Parameter names for better input display
  captureInput: true,       // Capture function arguments (default: true)
  captureOutput: false,     // Don't capture return value
});