Edit onElectedToBeLeader

src/main/java/com/distributed/search/cluster/OnElectionAction.java

 package com.distributed.search.cluster;
 
+import com.distributed.search.grpc.SearchClient;
 import com.distributed.search.grpc.SearchServiceImpl;
+import com.distributed.search.web.CoordinatorWebServer; // New Import
 import io.grpc.Server;
 import io.grpc.ServerBuilder;
+import org.apache.zookeeper.KeeperException;
 
 import java.io.IOException;
+import java.net.InetAddress;
+import java.net.UnknownHostException;
 
 /**
  * Defines the actions taken when a node's role changes in the cluster.
@@ -12,12 +17,20 @@ import java.io.IOException;
 public class OnElectionAction implements OnElectionCallback {
 
     private final ServiceRegistry serviceRegistry;
+    private final LeaderHttpRegistry leaderHttpRegistry; // For discovery
+    private final SearchClient searchClient; // To perform search
     private final int port;
     private Server grpcServer; // The gRPC server instance for Workers
+    private CoordinatorWebServer leaderInternalWebServer; // The Internal API
     private final String sharedDirectoryPath = "storage"; // The central storage folder
 
-    public OnElectionAction(ServiceRegistry serviceRegistry, int port) {
+    public OnElectionAction(ServiceRegistry serviceRegistry,
+                            LeaderHttpRegistry leaderHttpRegistry,
+                            SearchClient searchClient,
+                            int port) {
         this.serviceRegistry = serviceRegistry;
+        this.leaderHttpRegistry = leaderHttpRegistry;
+        this.searchClient = searchClient;
         this.port = port;
     }
 
@@ -26,16 +39,25 @@ public class OnElectionAction implements OnElectionCallback {
      */
     @Override
     public void onElectedToBeLeader() {
-        // If it was a worker, unregister itself from the worker list
         serviceRegistry.unregisterFromCluster();
-
-        // Start watching for other active workers in the cluster
         serviceRegistry.registerForUpdates();
 
-        System.out.println("Node elected as Leader. Monitoring the cluster...");
+        // 1. Start the Internal HTTP API (Running on port + 1000)
+        int internalApiPort = port + 1000;
+        leaderInternalWebServer = new CoordinatorWebServer(internalApiPort, searchClient, sharedDirectoryPath);
+        leaderInternalWebServer.startServer();
+
+        // 2. Register the Internal API address in Zookeeper for the Frontend to see
+        try {
+            String leaderApiAddress = String.format("http://localhost:%d/search", internalApiPort);
+            leaderHttpRegistry.registerLeader(leaderApiAddress);
+        } catch (KeeperException | InterruptedException e) {
+            e.printStackTrace();
+        }
+
+        System.out.println("Node promoted to Leader. Internal API is live.");
 
-        // Note: The Coordinator doesn't need to run a gRPC server for searching.
-        // It acts as a client that sends requests to workers.
+        // Stop gRPC server if it was acting as a worker
         if (grpcServer != null) {
             grpcServer.shutdown();
         }