packages/taro-platform-harmony-cpp/src/runtime/apis/network/webSocket.ts - main - githubmirror/taro

packages/taro-platform-harmony-cpp/src/runtime/apis/network/webSocket.ts
126 строк4 KB
ZakaryCode
fix(harmony): emit ets api
08 апр 2025, 06:45
08 апр 2025, 06:453e573ae
// OpenHarmony 不支持全局操作 WebSocket
// HarmonyOS 文档链接：https://developer.harmonyos.com/cn/docs/documentation/doc-references/js-apis-net-websocket-0000001168304641
// WX 文档链接：https://developers.weixin.qq.com/miniprogram/dev/api/network/websocket/wx.sendSocketMessage.html
// ✅ wx.connectSocket
// ✅ SocketTask
// ✅ SocketTask.close
// ✅ SocketTask.onClose
// ✅ SocketTask.onError
// ✅ SocketTask.onMessage
// ✅ SocketTask.onOpen
// ✅ SocketTask.send

// ❌ wx.sendSocketMessage
// ❌ wx.onSocketOpen
// ❌ wx.onSocketMessage
// ❌ wx.onSocketError
// ❌ wx.onSocketClose
// ❌ wx.closeSocket

import webSocket from '@ohos.net.webSocket'

import { callAsyncFail, callAsyncSuccess, validateParams } from '../utils'

import type Taro from '@tarojs/taro/types'

const connectSocketSchema = {
  url: 'String'
}

// const closetSocketSchema = {
//   code: 'Number',
//   reason: 'String'
// }

const sendSocketSchema = {
  data: 'String'
}

export const connectSocket: typeof Taro.connectSocket = function (options) {
  let ws
  const SocketTaskWX: any = new Promise((resolve, reject) => {
    ws = webSocket.createWebSocket()

    const { url, header } = options

    try {
      validateParams('uploadFile', options, connectSocketSchema)
    } catch (error) {
      const res = { errMsg: error.message }
      return callAsyncFail(reject, res, options)
    }

    const result = {
      errMsg: 'websocket connect success'
    }
    ws.connect(url, { header }).then((value) => {
      callAsyncSuccess(resolve, value, options)
    }).catch((err) => {
      result.errMsg = 'websocket connect fail'
      callAsyncFail(reject, err, options)
    })
  })

  SocketTaskWX.close = function (closeOptions) {
    return new Promise((resolve, reject) => {
      // TODO: 检验非必须参数
      // try {
      //   validateParams('close', options, closeSocketSchema)
      // } catch (error) {
      //   const res = { errMsg: error.message }
      //   return callAsyncFail(reject, res, options)
      // }

      ws.close(closeOptions).then(value => {
        callAsyncSuccess(resolve, value, closeOptions)
      }, error => {
        callAsyncFail(reject, error, closeOptions)
      })
    })
  }
  SocketTaskWX.onClose = function (onCloseCallback) {
    validateParams('onClose', [onCloseCallback], ['Function'])
    ws.on('close', (err, value) => {
      onCloseCallback(!err ? value : err)
    })
  }
  SocketTaskWX.onError = function (onErrorCallback) {
    validateParams('onError', [onErrorCallback], ['Function'])
    ws.on('error', (err) => {
      onErrorCallback(err)
    })
  }
  SocketTaskWX.onMessage = function (onMessageCallback?: any) {
    validateParams('onMessage', [onMessageCallback], ['Function'])
    ws.on('onMessage', (err, value) => {
      onMessageCallback(!err ? value : err)
    })
  }
  SocketTaskWX.onOpen = function (onOpenCallback) {
    validateParams('onOpen', [onOpenCallback], ['Function'])
    ws.on('open', (err, value) => {
      // TODO：返回数据字段完全不一样，无法兼容，暂不处理
      // wx:{header, profile}, ohos:{err, value:{status, message}}
      onOpenCallback(!err ? value : err)
    })
  }
  SocketTaskWX.send = function (sendOptions: Taro.SocketTask.SendOption) {
    return new Promise((resolve, reject) => {
      const { data } = sendOptions
      try {
        validateParams('send', sendOptions, sendSocketSchema)
      } catch (error) {
        const res = { errMsg: error.message }
        return callAsyncFail(reject, res, options)
      }

      ws.send(data).then(value => {
        callAsyncSuccess(resolve, value, sendOptions)
      }, error => {
        callAsyncFail(reject, error, sendOptions)
      })
    })
  }

  return SocketTaskWX
}