Skip to content

Commit 06635f2

Browse files
committed
Refactor hook handlers and improve code formatting for readability
1 parent ef5b176 commit 06635f2

7 files changed

Lines changed: 21 additions & 28 deletions

File tree

src/main/java/com/github/copilot/sdk/CopilotSession.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -549,8 +549,8 @@ CompletableFuture<Object> handleHooksInvoke(String hookType, JsonNode input) {
549549
break;
550550
case "userPromptSubmitted" :
551551
if (hooks.getOnUserPromptSubmitted() != null) {
552-
UserPromptSubmittedHookInput promptInput =
553-
MAPPER.treeToValue(input, UserPromptSubmittedHookInput.class);
552+
UserPromptSubmittedHookInput promptInput = MAPPER.treeToValue(input,
553+
UserPromptSubmittedHookInput.class);
554554
return hooks.getOnUserPromptSubmitted().handle(promptInput, invocation)
555555
.thenApply(output -> (Object) output);
556556
}

src/main/java/com/github/copilot/sdk/json/SessionEndHandler.java

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -16,11 +16,9 @@
1616
*
1717
* <pre>{@code
1818
* SessionEndHandler handler = (input, invocation) -> {
19-
* System.out.println("Session ended: " + input.getReason());
20-
* return CompletableFuture.completedFuture(
21-
* new SessionEndHookOutput()
22-
* .setSessionSummary("Session completed successfully")
23-
* );
19+
* System.out.println("Session ended: " + input.getReason());
20+
* return CompletableFuture
21+
* .completedFuture(new SessionEndHookOutput().setSessionSummary("Session completed successfully"));
2422
* };
2523
* }</pre>
2624
*

src/main/java/com/github/copilot/sdk/json/SessionEndHookInput.java

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -78,8 +78,7 @@ public SessionEndHookInput setCwd(String cwd) {
7878
/**
7979
* Gets the reason for session end.
8080
*
81-
* @return the reason: "complete", "error", "abort", "timeout", or
82-
* "user_exit"
81+
* @return the reason: "complete", "error", "abort", "timeout", or "user_exit"
8382
*/
8483
public String getReason() {
8584
return reason;

src/main/java/com/github/copilot/sdk/json/SessionHooks.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -160,7 +160,7 @@ public SessionHooks setOnSessionEnd(SessionEndHandler onSessionEnd) {
160160
* @return {@code true} if at least one hook handler is set
161161
*/
162162
public boolean hasHooks() {
163-
return onPreToolUse != null || onPostToolUse != null || onUserPromptSubmitted != null
164-
|| onSessionStart != null || onSessionEnd != null;
163+
return onPreToolUse != null || onPostToolUse != null || onUserPromptSubmitted != null || onSessionStart != null
164+
|| onSessionEnd != null;
165165
}
166166
}

src/main/java/com/github/copilot/sdk/json/SessionStartHandler.java

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -16,11 +16,9 @@
1616
*
1717
* <pre>{@code
1818
* SessionStartHandler handler = (input, invocation) -> {
19-
* System.out.println("Session started from: " + input.getSource());
20-
* return CompletableFuture.completedFuture(
21-
* new SessionStartHookOutput()
22-
* .setAdditionalContext("Custom initialization context")
23-
* );
19+
* System.out.println("Session started from: " + input.getSource());
20+
* return CompletableFuture
21+
* .completedFuture(new SessionStartHookOutput().setAdditionalContext("Custom initialization context"));
2422
* };
2523
* }</pre>
2624
*

src/main/java/com/github/copilot/sdk/json/UserPromptSubmittedHandler.java

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -16,12 +16,10 @@
1616
*
1717
* <pre>{@code
1818
* UserPromptSubmittedHandler handler = (input, invocation) -> {
19-
* System.out.println("User submitted: " + input.getPrompt());
20-
* // Optionally modify the prompt
21-
* return CompletableFuture.completedFuture(
22-
* new UserPromptSubmittedHookOutput()
23-
* .setModifiedPrompt(input.getPrompt() + " (enhanced)")
24-
* );
19+
* System.out.println("User submitted: " + input.getPrompt());
20+
* // Optionally modify the prompt
21+
* return CompletableFuture.completedFuture(
22+
* new UserPromptSubmittedHookOutput().setModifiedPrompt(input.getPrompt() + " (enhanced)"));
2523
* };
2624
* }</pre>
2725
*

src/test/java/com/github/copilot/sdk/HooksTest.java

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -30,8 +30,8 @@
3030
import com.github.copilot.sdk.json.UserPromptSubmittedHookInput;
3131

3232
/**
33-
* Tests for hooks functionality (pre-tool-use, post-tool-use, user-prompt-submitted,
34-
* session-start, and session-end hooks).
33+
* Tests for hooks functionality (pre-tool-use, post-tool-use,
34+
* user-prompt-submitted, session-start, and session-end hooks).
3535
*
3636
* <p>
3737
* These tests use the shared CapiProxy infrastructure for deterministic API
@@ -205,8 +205,8 @@ void testUserPromptSubmittedHookInvokedWhenUserSendsMessage() throws Exception {
205205
List<UserPromptSubmittedHookInput> promptInputs = new ArrayList<>();
206206
final String[] sessionIdHolder = new String[1];
207207

208-
SessionConfig config =
209-
new SessionConfig().setHooks(new SessionHooks().setOnUserPromptSubmitted((input, invocation) -> {
208+
SessionConfig config = new SessionConfig()
209+
.setHooks(new SessionHooks().setOnUserPromptSubmitted((input, invocation) -> {
210210
promptInputs.add(input);
211211
assertEquals(sessionIdHolder[0], invocation.getSessionId());
212212
return CompletableFuture.completedFuture(null);
@@ -233,8 +233,8 @@ void testSessionStartHookInvokedWhenSessionCreated() throws Exception {
233233

234234
List<SessionStartHookInput> startInputs = new ArrayList<>();
235235

236-
SessionConfig config =
237-
new SessionConfig().setHooks(new SessionHooks().setOnSessionStart((input, invocation) -> {
236+
SessionConfig config = new SessionConfig()
237+
.setHooks(new SessionHooks().setOnSessionStart((input, invocation) -> {
238238
startInputs.add(input);
239239
return CompletableFuture.completedFuture(null);
240240
}));

0 commit comments

Comments
 (0)