solver-sdk 1.6.8 → 1.7.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -37,6 +37,24 @@ class CodeSolverWebSocketClient {
37
37
  this.activeProjectId = null;
38
38
  /** Обработчики событий мышления */
39
39
  this.thinkingEventHandlers = new Map();
40
+ /** Таймеры для ping/pong */
41
+ this.pingIntervals = new Map();
42
+ /** Статистика ping/pong */
43
+ this.pingStats = new Map();
44
+ /** Количество последовательных таймаутов */
45
+ this.pingTimeouts = new Map();
46
+ /** Задержка по умолчанию между ping-сообщениями (30 секунд) */
47
+ this.defaultPingInterval = 30000;
48
+ /** Порог таймаута (количество пропущенных pong) */
49
+ this.timeoutThreshold = 3;
50
+ /** Хранилище обработчиков ping/pong */
51
+ this.pingPongEventHandlers = new Map();
52
+ /** Токены сессий для разных пространств имен */
53
+ this.sessionTokens = new Map();
54
+ /** Состояние подключения для разных пространств имен */
55
+ this.connectionState = new Map();
56
+ /** Таймер для проверки здоровья соединений */
57
+ this.healthCheckTimer = null;
40
58
  this.baseURL = baseURL.replace(/^http/, 'ws');
41
59
  this.options = {
42
60
  ...options,
@@ -179,54 +197,6 @@ class CodeSolverWebSocketClient {
179
197
  this.clients.set(namespace, client);
180
198
  return client;
181
199
  }
182
- /**
183
- * Подключается к пространству имен рассуждений
184
- * @param reasoningId ID рассуждения (опционально)
185
- * @returns Promise с результатом подключения
186
- */
187
- async connectToReasoning(reasoningId) {
188
- try {
189
- this.logger('info', 'Подключение к пространству имен рассуждений', { reasoningId });
190
- // Если указан ID рассуждения, сохраняем его
191
- if (reasoningId) {
192
- this.activeReasoningId = reasoningId;
193
- }
194
- // Подключаемся к пространству имен рассуждений
195
- const client = await this.connect(WebSocketNamespace.REASONING);
196
- // Аутентифицируемся с увеличенным таймаутом
197
- try {
198
- const authResult = await client.emitWithAck(websocket_events_constants_js_1.WebSocketEvents.AUTHENTICATE, {
199
- token: this.options.apiKey,
200
- reasoningId: this.activeReasoningId
201
- }, 10000);
202
- this.logger('debug', 'Результат аутентификации в namespace рассуждений', authResult);
203
- }
204
- catch (error) {
205
- this.logger('error', 'Ошибка аутентификации в namespace рассуждений', error);
206
- return false;
207
- }
208
- // Если у нас есть ID рассуждения, присоединяемся к нему
209
- if (this.activeReasoningId) {
210
- try {
211
- const joinResult = await client.emitWithAck(websocket_events_constants_js_1.WebSocketEvents.JOIN_REASONING, {
212
- reasoningId: this.activeReasoningId,
213
- token: this.options.apiKey
214
- }, 10000);
215
- this.logger('debug', 'Результат присоединения к рассуждению', joinResult);
216
- return true;
217
- }
218
- catch (error) {
219
- this.logger('error', 'Ошибка присоединения к рассуждению', error);
220
- return false;
221
- }
222
- }
223
- return true;
224
- }
225
- catch (error) {
226
- this.logger('error', 'Ошибка подключения к пространству имен рассуждений', error);
227
- return false;
228
- }
229
- }
230
200
  /**
231
201
  * Подключается к пространству имен индексации
232
202
  * @param projectId ID проекта (опционально)
@@ -357,11 +327,17 @@ class CodeSolverWebSocketClient {
357
327
  }
358
328
  /**
359
329
  * Отключается от всех пространств имен
330
+ * Отключает автоматический механизм ping/pong
360
331
  */
361
332
  disconnectAll() {
333
+ // Отключаем ping/pong для всех соединений
334
+ this.disablePingPong();
335
+ // Отключаемся от всех namespace
362
336
  for (const [namespace, client] of this.clients.entries()) {
363
- client.close();
364
- this.clients.delete(namespace);
337
+ if (client) {
338
+ client.close();
339
+ this.clients.delete(namespace);
340
+ }
365
341
  }
366
342
  // Сбрасываем активные сессии
367
343
  this.activeReasoningId = null;
@@ -698,6 +674,960 @@ class CodeSolverWebSocketClient {
698
674
  }
699
675
  }
700
676
  }
677
+ /**
678
+ * Включить автоматическую отправку ping-сообщений и сбор статистики
679
+ * @param {number} interval - Интервал между ping-сообщениями в миллисекундах
680
+ * @param {number} timeoutThreshold - Количество пропущенных pong-сообщений, после которого соединение считается потерянным
681
+ * @returns {boolean} - Успешность включения ping/pong
682
+ */
683
+ enablePingPong(interval = this.defaultPingInterval, timeoutThreshold = 3) {
684
+ // Сохраняем порог таймаута
685
+ this.timeoutThreshold = timeoutThreshold;
686
+ // Для каждого активного соединения
687
+ for (const [namespace, client] of this.clients.entries()) {
688
+ try {
689
+ // Проверяем, активно ли соединение
690
+ if (!client || !this.isConnected(namespace)) {
691
+ this.logger('warn', `Невозможно включить ping/pong для неактивного соединения в ${namespace}`);
692
+ continue;
693
+ }
694
+ // Останавливаем существующий таймер, если есть
695
+ this.disablePingPong(namespace);
696
+ // Инициализируем статистику, если не была создана
697
+ if (!this.pingStats.has(namespace)) {
698
+ this.pingStats.set(namespace, {
699
+ namespace,
700
+ socketId: client.getSocketId(),
701
+ pingSent: 0,
702
+ pongReceived: 0,
703
+ averageRtt: 0,
704
+ minRtt: Number.MAX_SAFE_INTEGER,
705
+ maxRtt: 0,
706
+ lastRtt: 0,
707
+ lastPongTimestamp: Date.now(),
708
+ isConnected: true
709
+ });
710
+ }
711
+ // Сбрасываем счетчик таймаутов
712
+ this.pingTimeouts.set(namespace, 0);
713
+ // Устанавливаем обработчик для события connection_pong
714
+ client.on(websocket_events_constants_js_1.WebSocketEvents.CONNECTION_PONG, (data) => {
715
+ // Обновляем статистику
716
+ const stats = this.pingStats.get(namespace);
717
+ if (stats) {
718
+ stats.pongReceived++;
719
+ stats.lastPongTimestamp = Date.now();
720
+ stats.isConnected = true;
721
+ // Рассчитываем RTT, если есть метка времени эхо
722
+ if (data && data.echo) {
723
+ const rtt = Date.now() - data.echo;
724
+ stats.lastRtt = rtt;
725
+ // Обновляем min и max
726
+ stats.minRtt = Math.min(stats.minRtt, rtt);
727
+ stats.maxRtt = Math.max(stats.maxRtt, rtt);
728
+ // Обновляем среднее значение
729
+ stats.averageRtt = (stats.averageRtt * (stats.pongReceived - 1) + rtt) / stats.pongReceived;
730
+ }
731
+ // Сбрасываем счетчик таймаутов
732
+ this.pingTimeouts.set(namespace, 0);
733
+ }
734
+ // Логируем получение pong
735
+ this.logger('debug', `Получен pong для ${namespace}`, {
736
+ rtt: stats?.lastRtt,
737
+ socketId: client.getSocketId()
738
+ });
739
+ });
740
+ // Устанавливаем интервал отправки ping
741
+ const pingInterval = setInterval(() => {
742
+ if (this.isConnected(namespace)) {
743
+ // Формируем данные ping
744
+ const pingData = { timestamp: Date.now() };
745
+ // Отправляем ping
746
+ const sent = this.send(namespace, websocket_events_constants_js_1.WebSocketEvents.CONNECTION_PING, pingData);
747
+ // Если успешно отправлено, обновляем статистику
748
+ if (sent) {
749
+ const stats = this.pingStats.get(namespace);
750
+ if (stats) {
751
+ stats.pingSent++;
752
+ }
753
+ this.logger('debug', `Отправлен ping для ${namespace}`, pingData);
754
+ }
755
+ else {
756
+ this.logger('warn', `Не удалось отправить ping для ${namespace}`);
757
+ }
758
+ // Проверяем таймаут
759
+ const timeouts = this.pingTimeouts.get(namespace) || 0;
760
+ const stats = this.pingStats.get(namespace);
761
+ // Если разница между отправленными и полученными превышает порог,
762
+ // или последний pong был получен слишком давно
763
+ if ((stats && stats.pingSent - stats.pongReceived > this.timeoutThreshold) ||
764
+ (stats && Date.now() - stats.lastPongTimestamp > interval * this.timeoutThreshold)) {
765
+ // Увеличиваем счетчик таймаутов
766
+ this.pingTimeouts.set(namespace, timeouts + 1);
767
+ if (timeouts + 1 >= this.timeoutThreshold) {
768
+ // Соединение потеряно
769
+ this.logger('error', `Соединение потеряно (таймаут ping/pong) для ${namespace}`);
770
+ // Установка флага неактивного соединения
771
+ if (stats) {
772
+ stats.isConnected = false;
773
+ }
774
+ // На прямую отправку события через socket
775
+ this.send(namespace, 'connection_timeout', {
776
+ namespace,
777
+ socketId: client.getSocketId(),
778
+ timeouts: timeouts + 1,
779
+ threshold: this.timeoutThreshold
780
+ });
781
+ // Также вызываем обработчики событий
782
+ const timeoutHandlers = this.pingPongEventHandlers.get('connection_timeout') || [];
783
+ timeoutHandlers.forEach(handler => {
784
+ try {
785
+ handler({
786
+ namespace,
787
+ socketId: client.getSocketId(),
788
+ timeouts: timeouts + 1,
789
+ threshold: this.timeoutThreshold
790
+ });
791
+ }
792
+ catch (error) {
793
+ this.logger('error', `Ошибка при обработке события connection_timeout`, error);
794
+ }
795
+ });
796
+ }
797
+ }
798
+ }
799
+ }, interval);
800
+ // Сохраняем интервал
801
+ this.pingIntervals.set(namespace, pingInterval);
802
+ this.logger('info', `Включен механизм ping/pong для ${namespace} с интервалом ${interval}ms`);
803
+ }
804
+ catch (error) {
805
+ this.logger('error', `Ошибка при включении ping/pong для ${namespace}`, error);
806
+ return false;
807
+ }
808
+ }
809
+ return true;
810
+ }
811
+ /**
812
+ * Отключить автоматическую отправку ping-сообщений
813
+ * @param {WebSocketNamespace} [namespace] - Пространство имен для отключения (если не указано - отключается везде)
814
+ */
815
+ disablePingPong(namespace) {
816
+ if (namespace) {
817
+ // Отключаем для указанного namespace
818
+ const interval = this.pingIntervals.get(namespace);
819
+ if (interval) {
820
+ clearInterval(interval);
821
+ this.pingIntervals.delete(namespace);
822
+ this.logger('info', `Отключен механизм ping/pong для ${namespace}`);
823
+ }
824
+ }
825
+ else {
826
+ // Отключаем для всех namespace
827
+ for (const [ns, interval] of this.pingIntervals.entries()) {
828
+ clearInterval(interval);
829
+ this.pingIntervals.delete(ns);
830
+ this.logger('info', `Отключен механизм ping/pong для ${ns}`);
831
+ }
832
+ }
833
+ }
834
+ /**
835
+ * Получить статистику ping/pong
836
+ * @param {WebSocketNamespace} [namespace] - Пространство имен для получения статистики
837
+ * @returns {PingPongStats | PingPongStats[] | null} - Статистика ping/pong
838
+ */
839
+ getPingStats(namespace) {
840
+ if (namespace) {
841
+ // Возвращаем статистику для указанного namespace
842
+ return this.pingStats.get(namespace) || null;
843
+ }
844
+ else {
845
+ // Возвращаем статистику для всех namespace
846
+ return Array.from(this.pingStats.values());
847
+ }
848
+ }
849
+ /**
850
+ * Добавляет обработчик для событий ping/pong
851
+ * @param {string} eventType - Тип события (connection_timeout)
852
+ * @param {(data: any) => void} handler - Обработчик события
853
+ */
854
+ onPingPongEvent(eventType, handler) {
855
+ if (!this.pingPongEventHandlers.has(eventType)) {
856
+ this.pingPongEventHandlers.set(eventType, []);
857
+ }
858
+ const handlers = this.pingPongEventHandlers.get(eventType);
859
+ if (handlers) {
860
+ handlers.push(handler);
861
+ }
862
+ }
863
+ /**
864
+ * Удаляет обработчик для событий ping/pong
865
+ * @param {string} eventType - Тип события
866
+ * @param {(data: any) => void} [handler] - Обработчик события (если не указан, удаляются все обработчики)
867
+ */
868
+ offPingPongEvent(eventType, handler) {
869
+ if (!handler) {
870
+ // Если обработчик не указан, удаляем все обработчики для этого типа события
871
+ this.pingPongEventHandlers.delete(eventType);
872
+ }
873
+ else {
874
+ // Если обработчик указан, удаляем только его
875
+ const handlers = this.pingPongEventHandlers.get(eventType);
876
+ if (handlers) {
877
+ const index = handlers.findIndex(h => h === handler);
878
+ if (index !== -1) {
879
+ handlers.splice(index, 1);
880
+ }
881
+ }
882
+ }
883
+ }
884
+ /**
885
+ * Возвращает функцию-обработчик для pong-ответов, которая рассчитывает RTT
886
+ * @returns {(data: any) => void} Функция-обработчик
887
+ */
888
+ getPongHandler() {
889
+ return (data) => {
890
+ if (data && data.echo) {
891
+ const rtt = Date.now() - data.echo;
892
+ console.log(`[PONG] RTT: ${rtt}ms, namespace: ${data.namespace || 'unknown'}`);
893
+ return rtt;
894
+ }
895
+ return -1;
896
+ };
897
+ }
898
+ /**
899
+ * Выполняет диагностику соединения и возвращает подробный отчет
900
+ * @param {WebSocketNamespace} namespace Пространство имен
901
+ * @returns {ConnectionDiagnostics} Объект с диагностической информацией
902
+ */
903
+ diagnoseConnection(namespace) {
904
+ const client = this.clients.get(namespace);
905
+ const stats = this.pingStats.get(namespace);
906
+ const connectionState = this.getConnectionState(namespace);
907
+ const sessionToken = this.getSessionToken(namespace);
908
+ return {
909
+ namespace,
910
+ isConnected: client?.isConnected() || false,
911
+ socketId: client?.getSocketId() || null,
912
+ lastActivity: stats?.lastPongTimestamp || 0,
913
+ rtt: {
914
+ current: stats?.lastRtt || -1,
915
+ min: stats?.minRtt === Number.MAX_SAFE_INTEGER ? -1 : (stats?.minRtt || -1),
916
+ max: stats?.maxRtt || -1,
917
+ avg: stats?.averageRtt || -1
918
+ },
919
+ pingSent: stats?.pingSent || 0,
920
+ pongReceived: stats?.pongReceived || 0,
921
+ missedPongs: (stats?.pingSent || 0) - (stats?.pongReceived || 0),
922
+ timeoutCount: this.pingTimeouts.get(namespace) || 0,
923
+ reconnectAttempts: connectionState.reconnectAttempts,
924
+ lastConnectTime: connectionState.lastConnectTime,
925
+ sessionRecovery: {
926
+ hasSessionToken: !!sessionToken,
927
+ tokenLength: sessionToken?.length || 0,
928
+ wasRecovered: !!sessionToken && (stats?.pongReceived || 0) > 0
929
+ }
930
+ };
931
+ }
932
+ /**
933
+ * Выполняет диагностику всех активных соединений
934
+ * @returns {Record<string, ConnectionDiagnostics>} Объект с диагностической информацией по всем соединениям
935
+ */
936
+ diagnoseAllConnections() {
937
+ const result = {};
938
+ // Проверяем каждое возможное пространство имен
939
+ for (const namespace of Object.values(WebSocketNamespace)) {
940
+ if (this.clients.has(namespace)) {
941
+ result[String(namespace)] = this.diagnoseConnection(namespace);
942
+ }
943
+ }
944
+ return result;
945
+ }
946
+ /**
947
+ * Рассчитывает задержку для переподключения на основе количества попыток и стратегии
948
+ * @param {WebSocketNamespace} namespace Пространство имен
949
+ * @returns {number} Задержка в миллисекундах
950
+ */
951
+ calculateReconnectDelay(namespace) {
952
+ const state = this.getConnectionState(namespace);
953
+ const attempts = state.reconnectAttempts;
954
+ const strategy = this.options.reconnectStrategy || 'exponential';
955
+ const baseDelay = this.options.retryDelay || 1000;
956
+ const maxDelay = this.options.maxRetryDelay || 30000;
957
+ if (strategy === 'exponential') {
958
+ // Экспоненциальный рост с фактором 1.5
959
+ const calculatedDelay = Math.min(baseDelay * Math.pow(1.5, attempts), maxDelay);
960
+ // Добавляем случайный фактор (jitter) для предотвращения штормов переподключений
961
+ return calculatedDelay * (0.8 + Math.random() * 0.4);
962
+ }
963
+ else {
964
+ // Линейный рост
965
+ return Math.min(baseDelay * (attempts + 1), maxDelay);
966
+ }
967
+ }
968
+ /**
969
+ * Принудительно переподключает соединение для указанного пространства имен
970
+ * @param {WebSocketNamespace} namespace Пространство имен
971
+ * @param {boolean} immediate Выполнить переподключение немедленно, без задержки
972
+ * @returns {Promise<boolean>} Успешность операции
973
+ */
974
+ async reconnectNamespace(namespace, immediate = false) {
975
+ const client = this.clients.get(namespace);
976
+ try {
977
+ // Если клиент уже существует, закрываем его
978
+ if (client) {
979
+ this.logger('info', `Принудительное переподключение для ${namespace}`);
980
+ try {
981
+ // Отключаем ping/pong для этого namespace
982
+ this.disablePingPong(namespace);
983
+ // Закрываем соединение
984
+ client.close();
985
+ }
986
+ catch (e) {
987
+ this.logger('warn', `Ошибка при закрытии соединения с ${namespace}: ${e instanceof Error ? e.message : String(e)}`);
988
+ }
989
+ // Удаляем клиент из кэша
990
+ this.clients.delete(namespace);
991
+ }
992
+ else {
993
+ this.logger('info', `Инициируем новое подключение для ${namespace}`);
994
+ }
995
+ // Устанавливаем состояние переподключения
996
+ this.setConnectionState(namespace, false, true);
997
+ // Инкрементируем счетчик попыток
998
+ this.incrementReconnectAttempts(namespace);
999
+ // Рассчитываем задержку, если не требуется немедленное переподключение
1000
+ if (!immediate) {
1001
+ const delay = this.calculateReconnectDelay(namespace);
1002
+ this.logger('info', `Переподключение для ${namespace} через ${delay}ms (попытка ${this.getConnectionState(namespace).reconnectAttempts})`);
1003
+ // Ждем рассчитанное время
1004
+ await new Promise(resolve => setTimeout(resolve, delay));
1005
+ }
1006
+ // Выполняем подключение с соответствующими параметрами
1007
+ let params = {};
1008
+ // Проверяем тип пространства имен и добавляем соответствующие параметры
1009
+ if (namespace === WebSocketNamespace.REASONING && this.activeReasoningId) {
1010
+ params.reasoningId = this.activeReasoningId;
1011
+ }
1012
+ else if ((namespace === WebSocketNamespace.INDEXING || namespace === WebSocketNamespace.DEPENDENCIES) && this.activeProjectId) {
1013
+ params.projectId = this.activeProjectId;
1014
+ }
1015
+ // Подключаемся
1016
+ await this.connect(namespace, params);
1017
+ // Если это пространство имен рассуждений и есть активное рассуждение,
1018
+ // пытаемся присоединиться к нему
1019
+ if (namespace === WebSocketNamespace.REASONING && this.activeReasoningId) {
1020
+ await this.connectToReasoning(this.activeReasoningId);
1021
+ }
1022
+ else if (namespace === WebSocketNamespace.INDEXING && this.activeProjectId) {
1023
+ await this.connectToIndexing(this.activeProjectId);
1024
+ }
1025
+ else if (namespace === WebSocketNamespace.DEPENDENCIES && this.activeProjectId) {
1026
+ await this.connectToDependencies(this.activeProjectId);
1027
+ }
1028
+ return true;
1029
+ }
1030
+ catch (error) {
1031
+ this.logger('error', `Ошибка при переподключении к ${namespace}: ${error instanceof Error ? error.message : String(error)}`);
1032
+ return false;
1033
+ }
1034
+ }
1035
+ /**
1036
+ * Настраивает периодическую проверку здоровья соединения
1037
+ * @param {number} [interval=30000] Интервал проверки в миллисекундах
1038
+ */
1039
+ setupConnectionHealthCheck(interval = 30000) {
1040
+ // Останавливаем существующую проверку, если она есть
1041
+ if (this.healthCheckTimer) {
1042
+ clearInterval(this.healthCheckTimer);
1043
+ }
1044
+ this.healthCheckTimer = setInterval(() => {
1045
+ for (const namespace of Object.values(WebSocketNamespace)) {
1046
+ const typedNamespace = namespace;
1047
+ const client = this.clients.get(typedNamespace);
1048
+ if (!client)
1049
+ continue;
1050
+ // Проверяем соединение через WebSocket клиент
1051
+ if (!client.isConnected()) {
1052
+ this.logger('warn', `Соединение с ${namespace} не активно, инициируем переподключение`);
1053
+ this.reconnectNamespace(typedNamespace, false).catch(() => { });
1054
+ continue;
1055
+ }
1056
+ // Проверяем статистику ping/pong
1057
+ const stats = this.pingStats.get(typedNamespace);
1058
+ if (stats) {
1059
+ const now = Date.now();
1060
+ // Если последний pong был получен слишком давно
1061
+ if (now - stats.lastPongTimestamp > interval * 2) {
1062
+ this.logger('warn', `Долгое отсутствие активности для ${namespace}, проверка соединения...`);
1063
+ // Отправляем проверочный ping
1064
+ this.send(typedNamespace, 'connection_health_check', { timestamp: now, echo: now });
1065
+ // Устанавливаем таймаут для проверки ответа
1066
+ setTimeout(() => {
1067
+ const currentStats = this.pingStats.get(typedNamespace);
1068
+ if (currentStats && now - currentStats.lastPongTimestamp > interval * 2) {
1069
+ this.logger('error', `Соединение не отвечает для ${namespace}, инициируем переподключение`);
1070
+ this.reconnectNamespace(typedNamespace, false).catch(() => { });
1071
+ }
1072
+ }, 5000); // Ждем 5 секунд на ответ
1073
+ }
1074
+ }
1075
+ }
1076
+ }, interval);
1077
+ this.logger('info', 'Настроена периодическая проверка здоровья соединения с интервалом ' + interval + 'ms');
1078
+ }
1079
+ /**
1080
+ * Сохраняет токен сессии для пространства имен
1081
+ * @param {WebSocketNamespace} namespace Пространство имен
1082
+ * @param {string} token Токен сессии
1083
+ */
1084
+ saveSessionToken(namespace, token) {
1085
+ if (this.options.enableSessionPersistence !== false) {
1086
+ this.sessionTokens.set(namespace, token);
1087
+ this.logger('info', `Сохранен токен сессии для ${namespace}`, { tokenLength: token.length });
1088
+ }
1089
+ }
1090
+ /**
1091
+ * Получает сохраненный токен сессии для пространства имен
1092
+ * @param {WebSocketNamespace} namespace Пространство имен
1093
+ * @returns {string | null} Токен сессии или null, если не найден
1094
+ */
1095
+ getSessionToken(namespace) {
1096
+ if (this.options.enableSessionPersistence === false) {
1097
+ return null;
1098
+ }
1099
+ return this.sessionTokens.get(namespace) || null;
1100
+ }
1101
+ /**
1102
+ * Удаляет сохраненный токен сессии для пространства имен
1103
+ * @param {WebSocketNamespace} namespace Пространство имен
1104
+ */
1105
+ clearSessionToken(namespace) {
1106
+ this.sessionTokens.delete(namespace);
1107
+ this.logger('info', `Удален токен сессии для ${namespace}`);
1108
+ }
1109
+ /**
1110
+ * Устанавливает состояние подключения для пространства имен
1111
+ * @param {WebSocketNamespace} namespace Пространство имен
1112
+ * @param {boolean} connected Состояние подключения
1113
+ * @param {boolean} reconnecting Состояние переподключения
1114
+ */
1115
+ setConnectionState(namespace, connected, reconnecting = false) {
1116
+ const state = this.getConnectionState(namespace);
1117
+ state.connected = connected;
1118
+ state.reconnecting = reconnecting;
1119
+ if (connected) {
1120
+ state.lastConnectTime = Date.now();
1121
+ state.reconnectAttempts = 0;
1122
+ }
1123
+ this.connectionState.set(namespace, state);
1124
+ }
1125
+ /**
1126
+ * Увеличивает счетчик попыток переподключения для пространства имен
1127
+ * @param {WebSocketNamespace} namespace Пространство имен
1128
+ * @returns {number} Новое количество попыток
1129
+ */
1130
+ incrementReconnectAttempts(namespace) {
1131
+ const state = this.getConnectionState(namespace);
1132
+ state.reconnectAttempts++;
1133
+ state.reconnecting = true;
1134
+ this.connectionState.set(namespace, state);
1135
+ return state.reconnectAttempts;
1136
+ }
1137
+ /**
1138
+ * Получает состояние подключения для пространства имен
1139
+ * @param {WebSocketNamespace} namespace Пространство имен
1140
+ * @returns {object} Состояние подключения
1141
+ */
1142
+ getConnectionState(namespace) {
1143
+ return this.connectionState.get(namespace) || {
1144
+ lastConnectTime: 0,
1145
+ reconnectAttempts: 0,
1146
+ connected: false,
1147
+ reconnecting: false
1148
+ };
1149
+ }
1150
+ /**
1151
+ * Устанавливает ID активной сессии рассуждения
1152
+ * @param {string} reasoningId ID сессии рассуждения
1153
+ * @returns {boolean} Успешность установки
1154
+ */
1155
+ setActiveReasoningId(reasoningId) {
1156
+ if (!reasoningId) {
1157
+ this.logger('error', 'Попытка установить пустой reasoningId');
1158
+ return false;
1159
+ }
1160
+ this.activeReasoningId = reasoningId;
1161
+ this.logger('info', `Установлен активный reasoningId: ${reasoningId}`);
1162
+ // Если мы уже подключены к пространству имен рассуждений, проверяем, нужно ли присоединиться
1163
+ const client = this.clients.get(WebSocketNamespace.REASONING);
1164
+ if (client && client.isConnected()) {
1165
+ // Пробуем присоединиться к рассуждению асинхронно, но не ждем результата
1166
+ this.logger('debug', `Автоматическое присоединение к рассуждению: ${reasoningId}`);
1167
+ client.emitWithAck(websocket_events_constants_js_1.WebSocketEvents.JOIN_REASONING, {
1168
+ reasoningId,
1169
+ token: this.options.apiKey
1170
+ }, 10000).catch(err => {
1171
+ this.logger('warn', `Не удалось автоматически присоединиться к рассуждению: ${err.message}`);
1172
+ });
1173
+ }
1174
+ return true;
1175
+ }
1176
+ /**
1177
+ * Устанавливает ID активного проекта
1178
+ * @param {string} projectId ID проекта
1179
+ * @returns {boolean} Успешность установки
1180
+ */
1181
+ setActiveProjectId(projectId) {
1182
+ if (!projectId) {
1183
+ this.logger('error', 'Попытка установить пустой projectId');
1184
+ return false;
1185
+ }
1186
+ this.activeProjectId = projectId;
1187
+ this.logger('info', `Установлен активный projectId: ${projectId}`);
1188
+ return true;
1189
+ }
1190
+ /**
1191
+ * Устанавливает ID активной сессии рассуждения с расширенными возможностями
1192
+ * @param {string} reasoningId ID сессии рассуждения
1193
+ * @param {boolean} waitForJoin Дождаться результата присоединения
1194
+ * @param {boolean} createIfNotExists Создать новое рассуждение, если ID не существует
1195
+ * @returns {Promise<boolean>} Результат операции
1196
+ */
1197
+ async setActiveReasoningIdAsync(reasoningId, waitForJoin = false, createIfNotExists = false) {
1198
+ if (!reasoningId && !createIfNotExists) {
1199
+ this.logger('error', 'Попытка установить пустой reasoningId');
1200
+ return false;
1201
+ }
1202
+ // Если указан createIfNotExists и нет reasoningId, создаем новое рассуждение
1203
+ if (createIfNotExists && !reasoningId) {
1204
+ try {
1205
+ reasoningId = await this.createNewReasoning();
1206
+ this.logger('info', `Создано новое рассуждение с ID: ${reasoningId}`);
1207
+ }
1208
+ catch (error) {
1209
+ this.logger('error', 'Не удалось создать новое рассуждение', error);
1210
+ return false;
1211
+ }
1212
+ }
1213
+ // Отключаемся от предыдущего рассуждения, если оно отличается
1214
+ if (this.activeReasoningId && this.activeReasoningId !== reasoningId) {
1215
+ this.logger('debug', `Отписываемся от предыдущего рассуждения: ${this.activeReasoningId}`);
1216
+ this.unsubscribeFromThinking(this.activeReasoningId);
1217
+ }
1218
+ this.activeReasoningId = reasoningId;
1219
+ this.logger('info', `Установлен активный reasoningId: ${reasoningId}`);
1220
+ // Если соединение уже установлено, присоединяемся к рассуждению
1221
+ const client = this.clients.get(WebSocketNamespace.REASONING);
1222
+ if (client && client.isConnected()) {
1223
+ try {
1224
+ if (waitForJoin) {
1225
+ this.logger('debug', `Ожидание присоединения к рассуждению: ${reasoningId}`);
1226
+ const joinResult = await client.emitWithAck(websocket_events_constants_js_1.WebSocketEvents.JOIN_REASONING, {
1227
+ reasoningId,
1228
+ token: this.options.apiKey
1229
+ }, 10000);
1230
+ if (joinResult.success === true) {
1231
+ this.logger('info', `Успешно присоединились к рассуждению: ${reasoningId}`);
1232
+ return true;
1233
+ }
1234
+ else {
1235
+ this.logger('warn', `Не удалось присоединиться к рассуждению: ${reasoningId}`, joinResult.error);
1236
+ return false;
1237
+ }
1238
+ }
1239
+ else {
1240
+ // Асинхронное присоединение
1241
+ this.logger('debug', `Асинхронное присоединение к рассуждению: ${reasoningId}`);
1242
+ client.emitWithAck(websocket_events_constants_js_1.WebSocketEvents.JOIN_REASONING, {
1243
+ reasoningId,
1244
+ token: this.options.apiKey
1245
+ }, 10000).then(result => {
1246
+ if (result.success === true) {
1247
+ this.logger('info', `Успешно присоединились к рассуждению: ${reasoningId}`);
1248
+ }
1249
+ else {
1250
+ this.logger('warn', `Не удалось присоединиться к рассуждению: ${reasoningId}`, result.error);
1251
+ }
1252
+ }).catch(err => {
1253
+ this.logger('warn', `Ошибка при присоединении к рассуждению: ${reasoningId}`, err);
1254
+ });
1255
+ }
1256
+ }
1257
+ catch (error) {
1258
+ this.logger('error', `Ошибка при присоединении к рассуждению: ${reasoningId}`, error);
1259
+ if (waitForJoin)
1260
+ return false;
1261
+ }
1262
+ }
1263
+ else if (waitForJoin) {
1264
+ // Если ждем присоединения, но соединение отсутствует
1265
+ this.logger('warn', `Нет активного соединения с сервером для присоединения к рассуждению: ${reasoningId}`);
1266
+ return false;
1267
+ }
1268
+ return true;
1269
+ }
1270
+ /**
1271
+ * Создает новое рассуждение на сервере
1272
+ * @private
1273
+ * @returns {Promise<string>} ID нового рассуждения
1274
+ */
1275
+ async createNewReasoning() {
1276
+ // Подключаемся, если еще не подключены
1277
+ if (!this.isConnectedToReasoning()) {
1278
+ this.logger('debug', 'Подключение к пространству имен рассуждений для создания нового рассуждения');
1279
+ const connected = await this.connectToReasoning();
1280
+ if (!connected) {
1281
+ throw new Error('Не удалось подключиться к пространству имен рассуждений');
1282
+ }
1283
+ }
1284
+ const client = this.clients.get(WebSocketNamespace.REASONING);
1285
+ if (!client) {
1286
+ throw new Error('Не удалось получить клиент для пространства имен рассуждений');
1287
+ }
1288
+ this.logger('debug', 'Отправка запроса на создание нового рассуждения');
1289
+ const result = await client.emitWithAck(websocket_events_constants_js_1.WebSocketEvents.CREATE_REASONING, {
1290
+ token: this.options.apiKey
1291
+ }, 10000);
1292
+ if (!result.reasoningId) {
1293
+ throw new Error(`Сервер не вернул ID рассуждения: ${JSON.stringify(result)}`);
1294
+ }
1295
+ return result.reasoningId;
1296
+ }
1297
+ /**
1298
+ * Проверяет существование рассуждения на сервере
1299
+ * @param {string} reasoningId ID рассуждения для проверки
1300
+ * @returns {Promise<boolean>} Существует ли рассуждение
1301
+ */
1302
+ async checkReasoningExists(reasoningId) {
1303
+ if (!reasoningId) {
1304
+ return false;
1305
+ }
1306
+ // Подключаемся, если еще не подключены
1307
+ if (!this.isConnectedToReasoning()) {
1308
+ const connected = await this.connectToReasoning();
1309
+ if (!connected) {
1310
+ this.logger('warn', 'Не удалось подключиться к пространству имен рассуждений для проверки существования');
1311
+ return false;
1312
+ }
1313
+ }
1314
+ const client = this.clients.get(WebSocketNamespace.REASONING);
1315
+ if (!client) {
1316
+ this.logger('warn', 'Не удалось получить клиент для пространства имен рассуждений');
1317
+ return false;
1318
+ }
1319
+ try {
1320
+ this.logger('debug', `Проверка существования рассуждения: ${reasoningId}`);
1321
+ // Используем более правильный подход для проверки существования сессии
1322
+ // Пытаемся присоединиться, и если успешно - значит сессия существует
1323
+ const result = await client.emitWithAck(websocket_events_constants_js_1.WebSocketEvents.JOIN_REASONING, {
1324
+ reasoningId,
1325
+ token: this.options.apiKey
1326
+ }, 10000);
1327
+ return result.success === true;
1328
+ }
1329
+ catch (error) {
1330
+ this.logger('warn', `Ошибка при проверке существования рассуждения: ${reasoningId}`, error);
1331
+ return false;
1332
+ }
1333
+ }
1334
+ /**
1335
+ * Подключается к пространству имен рассуждений
1336
+ * @param reasoningId ID рассуждения (опционально)
1337
+ * @param options Дополнительные настройки подключения
1338
+ * @returns Promise с результатом подключения
1339
+ */
1340
+ async connectToReasoning(reasoningId, options = {}) {
1341
+ try {
1342
+ // Значения опций по умолчанию
1343
+ const { autoJoin = true, createIfNotExists = false, checkExistence = false, saveSession = this.options.enableSessionPersistence !== false } = options;
1344
+ this.logger('info', 'Подключение к пространству имен рассуждений', {
1345
+ reasoningId,
1346
+ options: { autoJoin, createIfNotExists, checkExistence, saveSession }
1347
+ });
1348
+ // Если указан ID рассуждения, обрабатываем его
1349
+ if (reasoningId) {
1350
+ // Если нужно проверить существование
1351
+ if (checkExistence) {
1352
+ const client = this.clients.get(WebSocketNamespace.REASONING);
1353
+ // Если уже подключены, проверяем существование
1354
+ if (client && client.isConnected()) {
1355
+ const exists = await this.checkReasoningExists(reasoningId);
1356
+ if (!exists) {
1357
+ if (createIfNotExists) {
1358
+ this.logger('info', `Рассуждение ${reasoningId} не существует, создаем новое`);
1359
+ reasoningId = await this.createNewReasoning();
1360
+ }
1361
+ else {
1362
+ this.logger('warn', `Рассуждение ${reasoningId} не существует`);
1363
+ return false;
1364
+ }
1365
+ }
1366
+ }
1367
+ }
1368
+ // Устанавливаем ID активного рассуждения
1369
+ this.activeReasoningId = reasoningId;
1370
+ }
1371
+ else if (createIfNotExists) {
1372
+ // Если нет ID, но нужно создать, создаем новое рассуждение
1373
+ this.logger('info', 'Создание нового рассуждения');
1374
+ try {
1375
+ // Сначала подключаемся к пространству имен
1376
+ const client = await this.connect(WebSocketNamespace.REASONING);
1377
+ // Затем создаем новое рассуждение
1378
+ const newReasoningId = await this.createNewReasoning();
1379
+ this.activeReasoningId = newReasoningId;
1380
+ this.logger('info', `Создано новое рассуждение: ${newReasoningId}`);
1381
+ }
1382
+ catch (error) {
1383
+ this.logger('error', 'Ошибка при создании нового рассуждения', error);
1384
+ return false;
1385
+ }
1386
+ }
1387
+ // Подключаемся к пространству имен рассуждений
1388
+ const client = await this.connect(WebSocketNamespace.REASONING);
1389
+ // Аутентифицируемся с увеличенным таймаутом
1390
+ try {
1391
+ const authResult = await client.emitWithAck(websocket_events_constants_js_1.WebSocketEvents.AUTHENTICATE, {
1392
+ token: this.options.apiKey,
1393
+ reasoningId: this.activeReasoningId
1394
+ }, 10000);
1395
+ this.logger('debug', 'Результат аутентификации в namespace рассуждений', authResult);
1396
+ // Если сервер вернул токен сессии и нужно его сохранить
1397
+ if (saveSession && authResult.sessionToken) {
1398
+ this.saveSessionToken(WebSocketNamespace.REASONING, authResult.sessionToken);
1399
+ }
1400
+ }
1401
+ catch (error) {
1402
+ this.logger('error', 'Ошибка аутентификации в namespace рассуждений', error);
1403
+ return false;
1404
+ }
1405
+ // Если у нас есть ID рассуждения и включено автоматическое присоединение
1406
+ if (this.activeReasoningId && autoJoin) {
1407
+ try {
1408
+ const joinResult = await client.emitWithAck(websocket_events_constants_js_1.WebSocketEvents.JOIN_REASONING, {
1409
+ reasoningId: this.activeReasoningId,
1410
+ token: this.options.apiKey
1411
+ }, 10000);
1412
+ this.logger('debug', 'Результат присоединения к рассуждению', joinResult);
1413
+ if (joinResult.success !== true) {
1414
+ this.logger('warn', `Не удалось присоединиться к рассуждению: ${this.activeReasoningId}`, joinResult.error);
1415
+ return false;
1416
+ }
1417
+ }
1418
+ catch (error) {
1419
+ this.logger('error', 'Ошибка присоединения к рассуждению', error);
1420
+ return false;
1421
+ }
1422
+ }
1423
+ // Включаем автоматический ping/pong, если настроено
1424
+ if (this.options.enableAutoPing !== false) {
1425
+ this.enablePingPong(this.options.pingInterval, this.options.pingTimeoutThreshold);
1426
+ }
1427
+ return true;
1428
+ }
1429
+ catch (error) {
1430
+ this.logger('error', 'Ошибка подключения к пространству имен рассуждений', error);
1431
+ return false;
1432
+ }
1433
+ }
1434
+ /**
1435
+ * Получает текущий статус сессии рассуждения
1436
+ * @param {string} reasoningId ID сессии рассуждения (опционально, по умолчанию активная)
1437
+ * @returns {Promise<{exists: boolean, isActive: boolean, metadata?: any}>} Статус сессии
1438
+ */
1439
+ async getReasoningStatus(reasoningId) {
1440
+ const targetId = reasoningId || this.activeReasoningId;
1441
+ if (!targetId) {
1442
+ return { exists: false, isActive: false };
1443
+ }
1444
+ if (!this.isConnectedToReasoning()) {
1445
+ const connected = await this.connectToReasoning(undefined, { autoJoin: false });
1446
+ if (!connected) {
1447
+ this.logger('warn', 'Не удалось подключиться к пространству имен рассуждений');
1448
+ return { exists: false, isActive: false };
1449
+ }
1450
+ }
1451
+ const client = this.clients.get(WebSocketNamespace.REASONING);
1452
+ if (!client) {
1453
+ return { exists: false, isActive: false };
1454
+ }
1455
+ try {
1456
+ // Используем событие GET_REASONING_STATUS для получения статуса
1457
+ // Если сервер не поддерживает это событие, используем более простую проверку
1458
+ try {
1459
+ const status = await client.emitWithAck('get_reasoning_status', {
1460
+ reasoningId: targetId,
1461
+ token: this.options.apiKey
1462
+ }, 10000);
1463
+ return {
1464
+ exists: status.exists === true,
1465
+ isActive: status.isActive === true,
1466
+ metadata: status.metadata || {}
1467
+ };
1468
+ }
1469
+ catch (e) {
1470
+ // Если произошла ошибка (скорее всего, сервер не поддерживает этот метод),
1471
+ // используем проверку через JOIN_REASONING
1472
+ const result = await client.emitWithAck(websocket_events_constants_js_1.WebSocketEvents.JOIN_REASONING, {
1473
+ reasoningId: targetId,
1474
+ token: this.options.apiKey
1475
+ }, 10000);
1476
+ return {
1477
+ exists: result.success === true,
1478
+ isActive: result.success === true,
1479
+ metadata: result.data
1480
+ };
1481
+ }
1482
+ }
1483
+ catch (error) {
1484
+ this.logger('warn', `Ошибка при получении статуса рассуждения: ${targetId}`, error);
1485
+ return { exists: false, isActive: false };
1486
+ }
1487
+ }
1488
+ /**
1489
+ * Присоединяется к сессии рассуждения
1490
+ * @param {string} reasoningId ID сессии рассуждения
1491
+ * @param {boolean} setActive Установить как активное рассуждение
1492
+ * @returns {Promise<boolean>} Результат операции
1493
+ */
1494
+ async joinReasoning(reasoningId, setActive = true) {
1495
+ if (!reasoningId) {
1496
+ this.logger('error', 'Попытка присоединиться к пустому reasoningId');
1497
+ return false;
1498
+ }
1499
+ // Если нужно установить как активное, делаем это
1500
+ if (setActive) {
1501
+ this.activeReasoningId = reasoningId;
1502
+ }
1503
+ // Если не подключены, подключаемся сначала
1504
+ if (!this.isConnectedToReasoning()) {
1505
+ const connected = await this.connectToReasoning(reasoningId);
1506
+ if (!connected) {
1507
+ this.logger('warn', `Не удалось подключиться к пространству имен рассуждений для ${reasoningId}`);
1508
+ return false;
1509
+ }
1510
+ return true; // connectToReasoning уже выполнил join
1511
+ }
1512
+ // Если уже подключены, отправляем join запрос
1513
+ const client = this.clients.get(WebSocketNamespace.REASONING);
1514
+ if (!client) {
1515
+ this.logger('warn', 'Не удалось получить клиент для пространства имен рассуждений');
1516
+ return false;
1517
+ }
1518
+ try {
1519
+ const joinResult = await client.emitWithAck(websocket_events_constants_js_1.WebSocketEvents.JOIN_REASONING, {
1520
+ reasoningId,
1521
+ token: this.options.apiKey
1522
+ }, 10000);
1523
+ this.logger('debug', `Результат присоединения к рассуждению ${reasoningId}`, joinResult);
1524
+ return joinResult.success === true;
1525
+ }
1526
+ catch (error) {
1527
+ this.logger('error', `Ошибка при присоединении к рассуждению ${reasoningId}`, error);
1528
+ return false;
1529
+ }
1530
+ }
1531
+ /**
1532
+ * Подключается к пространству имен рассуждений с расширенными опциями
1533
+ * @param reasoningId ID рассуждения (опционально)
1534
+ * @param options Дополнительные настройки подключения
1535
+ * @returns Promise с результатом подключения
1536
+ */
1537
+ async connectToReasoningEx(reasoningId, options = {}) {
1538
+ try {
1539
+ // Значения опций по умолчанию
1540
+ const { autoJoin = true, createIfNotExists = false, checkExistence = false, saveSession = this.options.enableSessionPersistence !== false } = options;
1541
+ this.logger('info', 'Подключение к пространству имен рассуждений', {
1542
+ reasoningId,
1543
+ options: { autoJoin, createIfNotExists, checkExistence, saveSession }
1544
+ });
1545
+ // Если указан ID рассуждения, обрабатываем его
1546
+ if (reasoningId) {
1547
+ // Если нужно проверить существование
1548
+ if (checkExistence) {
1549
+ const client = this.clients.get(WebSocketNamespace.REASONING);
1550
+ // Если уже подключены, проверяем существование
1551
+ if (client && client.isConnected()) {
1552
+ const exists = await this.checkReasoningExists(reasoningId);
1553
+ if (!exists) {
1554
+ if (createIfNotExists) {
1555
+ this.logger('info', `Рассуждение ${reasoningId} не существует, создаем новое`);
1556
+ reasoningId = await this.createNewReasoning();
1557
+ }
1558
+ else {
1559
+ this.logger('warn', `Рассуждение ${reasoningId} не существует`);
1560
+ return false;
1561
+ }
1562
+ }
1563
+ }
1564
+ }
1565
+ // Устанавливаем ID активного рассуждения
1566
+ this.activeReasoningId = reasoningId;
1567
+ }
1568
+ else if (createIfNotExists) {
1569
+ // Если нет ID, но нужно создать, создаем новое рассуждение
1570
+ this.logger('info', 'Создание нового рассуждения');
1571
+ try {
1572
+ // Сначала подключаемся к пространству имен
1573
+ const client = await this.connect(WebSocketNamespace.REASONING);
1574
+ // Затем создаем новое рассуждение
1575
+ const newReasoningId = await this.createNewReasoning();
1576
+ this.activeReasoningId = newReasoningId;
1577
+ this.logger('info', `Создано новое рассуждение: ${newReasoningId}`);
1578
+ }
1579
+ catch (error) {
1580
+ this.logger('error', 'Ошибка при создании нового рассуждения', error);
1581
+ return false;
1582
+ }
1583
+ }
1584
+ // Подключаемся к пространству имен рассуждений
1585
+ const client = await this.connect(WebSocketNamespace.REASONING);
1586
+ // Аутентифицируемся с увеличенным таймаутом
1587
+ try {
1588
+ const authResult = await client.emitWithAck(websocket_events_constants_js_1.WebSocketEvents.AUTHENTICATE, {
1589
+ token: this.options.apiKey,
1590
+ reasoningId: this.activeReasoningId
1591
+ }, 10000);
1592
+ this.logger('debug', 'Результат аутентификации в namespace рассуждений', authResult);
1593
+ // Если сервер вернул токен сессии и нужно его сохранить
1594
+ if (saveSession && authResult.sessionToken) {
1595
+ this.saveSessionToken(WebSocketNamespace.REASONING, authResult.sessionToken);
1596
+ }
1597
+ }
1598
+ catch (error) {
1599
+ this.logger('error', 'Ошибка аутентификации в namespace рассуждений', error);
1600
+ return false;
1601
+ }
1602
+ // Если у нас есть ID рассуждения и включено автоматическое присоединение
1603
+ if (this.activeReasoningId && autoJoin) {
1604
+ try {
1605
+ const joinResult = await client.emitWithAck(websocket_events_constants_js_1.WebSocketEvents.JOIN_REASONING, {
1606
+ reasoningId: this.activeReasoningId,
1607
+ token: this.options.apiKey
1608
+ }, 10000);
1609
+ this.logger('debug', 'Результат присоединения к рассуждению', joinResult);
1610
+ if (joinResult.success !== true) {
1611
+ this.logger('warn', `Не удалось присоединиться к рассуждению: ${this.activeReasoningId}`, joinResult.error);
1612
+ return false;
1613
+ }
1614
+ }
1615
+ catch (error) {
1616
+ this.logger('error', 'Ошибка присоединения к рассуждению', error);
1617
+ return false;
1618
+ }
1619
+ }
1620
+ // Включаем автоматический ping/pong, если настроено
1621
+ if (this.options.enableAutoPing !== false) {
1622
+ this.enablePingPong(this.options.pingInterval, this.options.pingTimeoutThreshold);
1623
+ }
1624
+ return true;
1625
+ }
1626
+ catch (error) {
1627
+ this.logger('error', 'Ошибка подключения к пространству имен рассуждений', error);
1628
+ return false;
1629
+ }
1630
+ }
701
1631
  }
702
1632
  exports.CodeSolverWebSocketClient = CodeSolverWebSocketClient;
703
1633
  //# sourceMappingURL=code-solver-websocket-client.js.map