Skip to content

Commit 8126f49

Browse files
2 parents 8930ec7 + 578ad37 commit 8126f49

File tree

4 files changed

+14
-9
lines changed

4 files changed

+14
-9
lines changed

src/modules/project.ts

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,14 +1,14 @@
11
import cbws from '../core/websocket';
2-
import { GetProjectPathResponse } from '../types/socketMessageTypes';
2+
import { GetProjectPathResponse, GetProjectSettingsResponse } from '../types/socketMessageTypes';
33
/**
44
* A module for interacting with project settings and paths.
55
*/
66
const cbproject = {
77
/**
88
* Retrieves the project settings from the server.
9-
* @returns {Promise<any>} A promise that resolves with the project settings response.
9+
* @returns {Promise<GetProjectSettingsResponse>} A promise that resolves with the project settings response.
1010
*/
11-
getProjectSettings: (): Promise<any> => {
11+
getProjectSettings: (): Promise<GetProjectSettingsResponse> => {
1212
return cbws.messageManager.sendAndWaitForResponse(
1313
{
1414
"type": "settingEvent",

src/modules/state.ts

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
import cbws from '../core/websocket';
2-
import { AddToAgentStateResponse, GetAgentStateResponse } from '../types/socketMessageTypes';
2+
import { AddToAgentStateResponse, GetAgentStateResponse, GetProjectStateResponse, UpdateProjectStateResponse } from '../types/socketMessageTypes';
33
import { ApplicationState } from '../types/commonTypes';
44

55
const cbstate = {
@@ -57,7 +57,7 @@ const cbstate = {
5757
* Retrieves the current project state from the server via WebSocket.
5858
* @returns {Promise<GetProjectStateResponse>} A promise that resolves with the project's state.
5959
*/
60-
getProjectState: async (): Promise<any> => {
60+
getProjectState: async (): Promise<GetProjectStateResponse> => {
6161
return cbws.messageManager.sendAndWaitForResponse(
6262
{
6363
"type": "projectStateEvent",
@@ -69,9 +69,11 @@ const cbstate = {
6969

7070
/**
7171
* Updates the project state on the server via WebSocket.
72+
* @param {string} key - The key to update in the project state.
73+
* @param {any} value - The value to set for the key.
7274
* @returns {Promise<UpdateProjectStateResponse>} A promise that resolves with the response to the update request.
7375
*/
74-
updateProjectState: async (key:string,value:any): Promise<any> => {
76+
updateProjectState: async (key:string,value:any): Promise<UpdateProjectStateResponse> => {
7577
return cbws.messageManager.sendAndWaitForResponse(
7678
{
7779
"type": "projectStateEvent",

src/modules/task.ts

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ const taskplaner = {
99
* @param {string} task - The task to be added.
1010
* @returns {Promise<AddTaskResponse>} A promise that resolves with the response from the add task event.
1111
*/
12-
addTask: async (task: string): Promise<any> => {
12+
addTask: async (task: string): Promise<AddTaskResponse> => {
1313
return cbws.messageManager.sendAndWaitForResponse(
1414
{
1515
"type": "taskEvent",
@@ -26,7 +26,7 @@ const taskplaner = {
2626
* Retrieves all tasks using a WebSocket message.
2727
* @returns {Promise<GetTasksResponse>} A promise that resolves with the response from the get tasks event.
2828
*/
29-
getTasks: async (): Promise<any> => {
29+
getTasks: async (): Promise<GetTasksResponse> => {
3030
return cbws.messageManager.sendAndWaitForResponse(
3131
{
3232
"type":"taskEvent",
@@ -41,7 +41,7 @@ const taskplaner = {
4141
* @param {string} task - The updated task information.
4242
* @returns {Promise<UpdateTasksResponse>} A promise that resolves with the response from the update task event.
4343
*/
44-
updateTask: async ( task: string): Promise<any> => {
44+
updateTask: async (task: string): Promise<UpdateTasksResponse> => {
4545
return cbws.messageManager.sendAndWaitForResponse(
4646
{
4747
"type": "taskEvent",

src/types/socketMessageTypes.ts

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -572,6 +572,7 @@ export interface MemoryClearResponse extends BaseWebSocketResponse {
572572
export interface AddTaskResponse extends BaseWebSocketResponse {
573573
type: 'addTaskResponse';
574574
task?: import('./commonTypes').Task;
575+
data?: any;
575576
}
576577

577578
export interface GetTasksResponse extends BaseWebSocketResponse {
@@ -817,6 +818,7 @@ export interface GetProjectPathResponse extends BaseWebSocketResponse {
817818
export interface GetProjectSettingsResponse extends BaseWebSocketResponse {
818819
type: 'getProjectSettingsResponse';
819820
projectSettings?: Record<string, any>;
821+
data?: Record<string, any>;
820822
}
821823

822824
export interface GetRepoMapResponse extends BaseWebSocketResponse {
@@ -827,6 +829,7 @@ export interface GetRepoMapResponse extends BaseWebSocketResponse {
827829
export interface GetProjectStateResponse extends BaseWebSocketResponse {
828830
type: 'getProjectStateResponse';
829831
projectState?: Record<string, any>;
832+
data?: Record<string, any>;
830833
}
831834

832835
// ================================

0 commit comments

Comments
 (0)