dgx-spark-playbooks/nvidia/txt2kg/assets/frontend/app/api/graph-db/clear/route.ts

import { NextRequest, NextResponse } from 'next/server';
import { getGraphDbService } from '@/lib/graph-db-util';
import { getGraphDbType } from '../../settings/route';
import { ArangoDBService } from '@/lib/arangodb';
import { Neo4jService } from '@/lib/neo4j';

/**
 * POST handler for clearing all data from the graph database
 */
export async function POST(request: NextRequest) {
  try {
    // Get the preferred database type from settings
    const graphDbType = getGraphDbType();
    console.log(`Using graph database for clearing: ${graphDbType}`);
    
    // Get the appropriate service
    const graphDbService = getGraphDbService(graphDbType);
    
    // Clear the database based on type
    if (graphDbType === 'arangodb') {
      const arangoService = graphDbService as ArangoDBService;
      await arangoService.clearDatabase();
    } else if (graphDbType === 'neo4j') {
      // TODO: Implement Neo4j clear functionality when needed
      throw new Error('Clear database functionality not implemented for Neo4j');
    }
    
    // Return success response
    return NextResponse.json({
      success: true,
      message: `Successfully cleared all data from ${graphDbType} database`,
      databaseType: graphDbType
    });
  } catch (error) {
    console.error(`Error in clear database handler:`, error);
    return NextResponse.json(
      { error: `Failed to clear database: ${error instanceof Error ? error.message : String(error)}` },
      { status: 500 }
    );
  }
}