ServeCommand.java

package io.github.randomcodespace.iq.cli;

import io.github.randomcodespace.iq.config.CodeIqConfig;
import io.github.randomcodespace.iq.graph.GraphStore;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import picocli.CommandLine.Command;
import picocli.CommandLine.Option;
import picocli.CommandLine.Parameters;

import java.nio.file.Path;
import java.text.NumberFormat;
import java.util.Locale;
import java.util.concurrent.Callable;

/**
 * Start the web UI + REST API + MCP server.
 * <p>
 * This command expects Neo4j to already be populated by {@code enrich}.
 * It simply starts the Spring Boot web server and blocks until shutdown.
 * <p>
 * Pipeline: {@code index} → {@code enrich} → {@code serve}
 */
@Component
@Command(name = "serve", mixinStandardHelpOptions = true,
        description = "Start web UI + REST API + MCP server")
public class ServeCommand implements Callable<Integer> {

    private static final Logger log = LoggerFactory.getLogger(ServeCommand.class);

    @Parameters(index = "0", defaultValue = ".", description = "Path to analyzed codebase")
    private Path path;

    @Option(names = {"--port", "-p"}, defaultValue = "8080", description = "Server port")
    private int port;

    @Option(names = {"--host"}, defaultValue = "0.0.0.0", description = "Bind address")
    private String host;

    @Option(names = {"--graph"}, description = "Path to shared graph directory (overrides default)")
    private Path graphPath;

    @Autowired
    private CodeIqConfig config;

    @Autowired(required = false)
    private GraphStore graphStore;

    @Override
    public Integer call() {
        Path root = path.toAbsolutePath().normalize();
        config.setRootPath(root.toString());
        NumberFormat nf = NumberFormat.getIntegerInstance(Locale.US);

        // Report Neo4j graph status
        if (graphStore != null) {
            try {
                long nodeCount = graphStore.count();
                long edgeCount = graphStore.countEdges();
                if (nodeCount > 0) {
                    CliOutput.success("Neo4j graph: " + nf.format(nodeCount) + " nodes, "
                            + nf.format(edgeCount) + " edges");
                } else {
                    CliOutput.warn("Neo4j graph is empty. Run 'code-iq enrich " + root + "' to populate.");
                }
            } catch (Exception e) {
                log.debug("Could not check Neo4j state", e);
                CliOutput.warn("Could not read Neo4j graph. Run 'code-iq enrich' first.");
            }
        }

        CliOutput.step("\uD83D\uDE80", "@|bold,green Server started|@");
        System.out.println();
        CliOutput.info("  URL:       http://" + host + ":" + port);
        CliOutput.info("  REST API:  http://" + host + ":" + port + "/api");
        CliOutput.info("  MCP:       http://" + host + ":" + port + "/mcp");
        CliOutput.info("  Health:    http://" + host + ":" + port + "/actuator/health");
        CliOutput.info("  Codebase:  " + root);
        System.out.println();
        CliOutput.info("Press Ctrl+C to stop.");

        try {
            Thread.currentThread().join();
        } catch (InterruptedException e) {
            Thread.currentThread().interrupt();
        }

        return 0;
    }

    public Path getPath() { return path; }
    public int getPort() { return port; }
    public String getHost() { return host; }
    public Path getGraphPath() { return graphPath; }
}