// Create a few sessions, two in the project directory and one in /tmp
var session1 = client.newSession(new NewSessionRequest(cwd, List.of()));
var session2 = client.newSession(new NewSessionRequest(cwd, List.of()));
var session3 = client.newSession(new NewSessionRequest("/tmp", List.of()));

// List all sessions (null = no cwd filter)
var allSessions = client.listSessions(new ListSessionsRequest(null));
for (var info : allSessions.sessions()) {
    System.out.println(info.sessionId() + " | cwd: " + info.cwd()
        + " | " + info.title() + " | updated: " + info.updatedAt());
}

// List only the sessions in one working directory
var tmpSessions = client.listSessions(new ListSessionsRequest("/tmp"));

// Resume: reconnect without replaying history
var resumed = client.resumeSession(new ResumeSessionRequest(
    session1.sessionId(), cwd, List.of()));

// The session state is still there, so the conversation continues
client.prompt(new PromptRequest(session1.sessionId(),
    List.of(new TextContent("Message after resuming!"))));

// Close: the agent frees the session's resources
client.closeSession(new CloseSessionRequest(session3.sessionId()));

AcpSyncAgent agent = AcpAgent.sync(transport)
    .initializeHandler(req -> InitializeResponse.ok())

    .newSessionHandler(req -> {
        String sessionId = "sess-" + UUID.randomUUID().toString().substring(0, 8);
        sessions.put(sessionId, new SessionState(
            req.cwd(), null, Instant.now(), new ArrayList<>()));
        return new NewSessionResponse(sessionId, null, null);
    })

    // Return SessionInfo metadata, honoring the optional cwd filter
    .listSessionsHandler(req -> {
        List<SessionInfo> infos = new ArrayList<>();
        for (var entry : sessions.entrySet()) {
            SessionState state = entry.getValue();
            if (req.cwd() != null && !req.cwd().equals(state.cwd())) {
                continue;
            }
            infos.add(new SessionInfo(
                entry.getKey(),
                state.cwd(),
                "Session with " + state.messages().size() + " messages",
                state.createdAt().toString(),
                null,
                null));
        }
        return new ListSessionsResponse(infos);
    })

    // Unlike loadSession, resumeSession sends no session/update notifications
    .resumeSessionHandler(req -> {
        sessions.putIfAbsent(req.sessionId(), new SessionState(
            req.cwd(), null, Instant.now(), new ArrayList<>()));
        return new ResumeSessionResponse(null, null);
    })

    .closeSessionHandler(req -> {
        sessions.remove(req.sessionId());
        return new CloseSessionResponse();
    })

    .promptHandler((req, context) -> {
        SessionState state = sessions.get(req.sessionId());
        if (state != null) {
            state.messages().add(req.text());
        }
        context.sendMessage("Active sessions: " + sessions.size() + "\n"
            + "You said: " + req.text() + "\n");
        return PromptResponse.endTurn();
    })
    .build();

./mvnw package -pl module-20-session-management -q
./mvnw exec:java -pl module-20-session-management