refactor: move HarmonyOS project files from harmony/ to branch root

2026-10-15 03:52:36 +00:00 · 2026-04-28 16:52:05 +08:00
parent 4c32dbb949
commit 11ee329943
120 changed files with 0 additions and 0 deletions
--- a/trulymem-core/src/main/ets/components/RconClient.ets
+++ b/trulymem-core/src/main/ets/components/RconClient.ets
@ -0,0 +1,208 @@
+import { socket } from '@kit.NetworkKit';
+import { BusinessError } from '@kit.BasicServicesKit';
+import {RConResult,parseRCONBuffer,buildPacket,GetAuth} from '../components/tools';
+
+type RConCallback = (result: RConResult) => void;
+
+export class RConClient
+{
+   public isActive: boolean; //是否可用标志位，随tcp状态更改
+   private tcp: socket.TCPSocket; //socket连接
+   private ipAddress: socket.NetAddress; //ip地址
+   private passWord: string; //密码
+   private isConnecting: boolean = false; // 标记连接是否正在进行中，防止并发连接
+   private closeHandler: (() => void) | null = null; // 保存 close 事件处理器引用以便清理
+   private authResolve: (() => void) | null = null; // 认证完成信号
+   private nextPacketId: number = 1; // 递增的包序号，避免重复
+
+   constructor(ip:string,LPort:number,PassWord:string) {
+     this.isActive = false;
+     this.tcp = socket.constructTCPSocketInstance();
+     this.ipAddress = {
+       address: ip,
+       port: LPort
+     }
+     this.passWord = PassWord;//初始化tcp连接相关配置
+   }
+  //连接服务器
+   async setConnect(timeOut:number,callBack:RConCallback)//异步获取服务器连接
+   {
+     if (this.isConnecting) {
+       // Already connecting, return an error via callback
+       callBack(RConResult.error("ALREADY_CONNECTING", "A connection attempt is already in progress"));
+       return;
+     }
+     this.isConnecting = true;
+     let finished = false; // To ensure we set isConnecting to false only once
+     const setFinished = () => {
+       if (!finished) {
+         finished = true;
+         this.isConnecting = false;
+       }
+     };
+
+     // 清理旧的事件监听器
+     this.cleanupListeners();
+
+     let connectOptions: socket.TCPConnectOptions = {
+       address: this.ipAddress,
+       timeout: timeOut
+     };
+
+     await this.tcp.connect(connectOptions).then(() => {
+       console.info('Connection established');
+       let result:RConResult = RConResult.result(undefined,"tcp_ok")
+       // 注册 close 事件并保存引用
+       this.closeHandler = () => {
+         this.isActive = false;
+       };
+       this.tcp.on('close', this.closeHandler);
+       //发送认证信息
+       GetAuth(this.tcp,this.passWord,3,(result:RConResult)=>{
+         setFinished();
+         if(!result.success) {
+           callBack(result);
+           // 通知等待的 sendAfterAuth 认证失败
+           this.notifyAuthComplete();
+         }
+         else
+         {
+           this.isActive = true;
+           callBack(result);
+           // 通知等待的 sendAfterAuth 认证成功
+           this.notifyAuthComplete();
+         }
+
+       })
+     }).catch((err: BusinessError) => {
+       setFinished();
+       let result = RConResult.error(JSON.stringify({ code: err.code, message: err.message }), "tcp_fail");
+       callBack(result);
+       this.notifyAuthComplete();
+     });
+   }
+
+   /**
+    * 通知认证流程完成（供 sendAfterAuth 等待机制使用）
+    */
+   private notifyAuthComplete(): void {
+     if (this.authResolve) {
+       this.authResolve();
+       this.authResolve = null;
+     }
+   }
+
+   /**
+    * 获取下一个递增包序号（跳过 -1，因为 -1 在 RCON 协议中表示认证失败）
+    */
+   private getNextPacketId(): number {
+     const id = this.nextPacketId;
+     this.nextPacketId++;
+     if (this.nextPacketId === -1 || this.nextPacketId > 2147483646) {
+       this.nextPacketId = 1;
+     }
+     return id;
+   }
+
+   /**
+    * 清理所有事件监听器
+    */
+   private cleanupListeners(): void {
+     this.tcp.off("close");
+     this.tcp.off("message");
+     this.closeHandler = null;
+   }
+
+   async sendMessage(message:string,type:number,callBack:RConCallback)
+   {//发送一条指令
+     if(!this.isActive)//错误重连机制
+     {
+       await this.setConnect(1000,(result:RConResult)=>{
+         if (!result.success) {
+           // Connection failed, call the outer callback with the error
+           callBack(result);
+           return;
+         }
+         // 重连后需要重新认证，不能直接发送命令
+         // 等待认证完成后再发送消息
+         this.sendAfterAuth(message, type, callBack);
+       })
+     }
+     else {
+       this.sendWithResponse(message, type, callBack);
+     }
+   }
+
+   /**
+    * 发送消息并等待响应（正常路径）
+    */
+    private sendWithResponse(message: string, type: number, callBack: RConCallback): void {
+      const packetId = this.getNextPacketId();
+      const messageHandler = (info: socket.SocketMessageInfo) => {
+        const buffer: ArrayBuffer = info.message;
+        const result = parseRCONBuffer(buffer);
+        callBack(result);
+        this.tcp.off("message", messageHandler);
+      };
+      this.tcp.on("message", messageHandler);
+
+      let sendOption: socket.TCPSendOptions = { data: buildPacket(packetId, type, message) };
+     this.tcp.send(sendOption).catch((err: BusinessError) => {
+       let result = RConResult.error(JSON.stringify({ code: err.code, message: err.message }), "send_auth_fail");
+       callBack(result);
+       this.tcp.off("message", messageHandler);
+       this.setDisconnect();
+     });
+   }
+
+   /**
+    * 重连后等待认证完成再发送消息
+    * 使用 Promise 等待机制替代轮询
+    */
+   private sendAfterAuth(message: string, type: number, callBack: RConCallback): void {
+     // 创建认证等待 Promise（最多 5 秒）
+     const authPromise = new Promise<boolean>((resolve) => {
+       this.authResolve = () => resolve(this.isActive);
+       // 超时兜底
+       setTimeout(() => {
+         if (this.authResolve) {
+           this.authResolve = null;
+           resolve(false);
+         }
+       }, 5000);
+     });
+
+     authPromise.then((authSuccess: boolean) => {
+       if (authSuccess) {
+         this.sendWithResponse(message, type, callBack);
+       } else {
+         let result = RConResult.error("AUTH_TIMEOUT", "Authentication timed out after reconnect");
+         callBack(result);
+       }
+     });
+   }
+
+  async resetServer(ip:string,password:string,port:number,timeout:number,callback:RConCallback)
+  {//重新设置服务器配置
+    this.ipAddress = {
+      address:ip,
+      port:port
+    }
+    this.passWord = password;
+    this.setDisconnect();//断开放弃使用的tcp连接
+    await this.setConnect(timeout,callback);
+  }
+  setDisconnect()
+  {
+    // 清理所有事件监听器
+    this.cleanupListeners();
+    // 断链
+    try {
+      this.tcp.close();
+    } catch {
+      // Socket already closed
+    }
+    this.isActive = false;
+  }
+
+}