mirror of https://github.com/axmolengine/axmol.git
500 lines
16 KiB
C++
500 lines
16 KiB
C++
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
#include "tolua_fix.h"
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
|
|
#include <map>
|
|
#include <string>
|
|
#include "Lua_web_socket.h"
|
|
#include "cocos2d.h"
|
|
#include "WebSocket.h"
|
|
#include "CCLuaStack.h"
|
|
#include "CCLuaValue.h"
|
|
#include "CCLuaEngine.h"
|
|
|
|
using namespace cocos2d;
|
|
using namespace cocos2d::extension;
|
|
|
|
|
|
static int SendBinaryMessageToLua(int nHandler,const unsigned char* pTable,int nLength)
|
|
{
|
|
if (NULL == pTable || nHandler <= 0) {
|
|
return 0;
|
|
}
|
|
|
|
if (NULL == CCScriptEngineManager::sharedManager()->getScriptEngine()) {
|
|
return 0;
|
|
}
|
|
|
|
CCLuaStack *pStack = CCLuaEngine::defaultEngine()->getLuaStack();
|
|
if (NULL == pStack) {
|
|
return 0;
|
|
}
|
|
|
|
lua_State *tolua_s = pStack->getLuaState();
|
|
if (NULL == tolua_s) {
|
|
return 0;
|
|
}
|
|
|
|
int nRet = 0;
|
|
CCLuaValueArray array;
|
|
for (int i = 0 ; i < nLength; i++) {
|
|
CCLuaValue value = CCLuaValue::intValue(pTable[i]);
|
|
array.push_back(value);
|
|
}
|
|
|
|
pStack->pushCCLuaValueArray(array);
|
|
nRet = pStack->executeFunctionByHandler(nHandler, 1);
|
|
pStack->clean();
|
|
return nRet;
|
|
}
|
|
|
|
class CLuaWebSocket: public WebSocket,public WebSocket::Delegate
|
|
{
|
|
public:
|
|
/*
|
|
* @brief delegate event enum,for lua register handler
|
|
*/
|
|
enum webSocketScriptHandlerType
|
|
{
|
|
kwebSocketScriptHandlerOpen = 0,
|
|
kwebSocketScriptHandlerMessage,
|
|
kwebSocketScriptHandlerClose,
|
|
kwebSocketScriptHandlerError,
|
|
};
|
|
/**
|
|
* @brief Add Handler of DelegateEvent
|
|
*/
|
|
void registerScriptHandler(int nFunID,webSocketScriptHandlerType scriptHandlerType)
|
|
{
|
|
m_mapScriptHandler[scriptHandlerType] = nFunID;
|
|
}
|
|
/**
|
|
* @brief Remove Handler of DelegateEvent
|
|
*/
|
|
void unregisterScriptHandler(webSocketScriptHandlerType scriptHandlerType)
|
|
{
|
|
std::map<int,int>::iterator Iter = m_mapScriptHandler.find(scriptHandlerType);
|
|
|
|
if (m_mapScriptHandler.end() != Iter)
|
|
{
|
|
m_mapScriptHandler.erase(Iter);
|
|
}
|
|
}
|
|
/**
|
|
* @brief Get Handler By DelegateEvent Type
|
|
*/
|
|
int getScriptHandler(webSocketScriptHandlerType scriptHandlerType)
|
|
{
|
|
std::map<int,int>::iterator Iter = m_mapScriptHandler.find(scriptHandlerType);
|
|
|
|
if (m_mapScriptHandler.end() != Iter)
|
|
return Iter->second;
|
|
|
|
return -1;
|
|
}
|
|
|
|
void InitScriptHandleMap()
|
|
{
|
|
m_mapScriptHandler.clear();
|
|
}
|
|
|
|
virtual void onOpen(WebSocket* ws)
|
|
{
|
|
CLuaWebSocket* luaWs = dynamic_cast<CLuaWebSocket*>(ws);
|
|
if (NULL != luaWs) {
|
|
int nHandler = luaWs->getScriptHandler(CLuaWebSocket::kwebSocketScriptHandlerOpen);
|
|
if (-1 != nHandler) {
|
|
CCScriptEngineManager::sharedManager()->getScriptEngine()->executeEvent(nHandler,"");
|
|
}
|
|
}
|
|
}
|
|
|
|
virtual void onMessage(WebSocket* ws, const WebSocket::Data& data)
|
|
{
|
|
CLuaWebSocket* luaWs = dynamic_cast<CLuaWebSocket*>(ws);
|
|
if (NULL != luaWs) {
|
|
if (data.isBinary) {
|
|
int nHandler = luaWs->getScriptHandler(CLuaWebSocket::kwebSocketScriptHandlerMessage);
|
|
if (-1 != nHandler) {
|
|
SendBinaryMessageToLua(nHandler, (const unsigned char*)data.bytes, data.len);
|
|
}
|
|
}
|
|
else{
|
|
|
|
int nHandler = luaWs->getScriptHandler(CLuaWebSocket::kwebSocketScriptHandlerMessage);
|
|
if (-1 != nHandler) {
|
|
CCScriptEngineManager::sharedManager()->getScriptEngine()->executeEvent(nHandler,data.bytes);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
virtual void onClose(WebSocket* ws)
|
|
{
|
|
CLuaWebSocket* luaWs = dynamic_cast<CLuaWebSocket*>(ws);
|
|
if (NULL != luaWs) {
|
|
int nHandler = luaWs->getScriptHandler(CLuaWebSocket::kwebSocketScriptHandlerClose);
|
|
if (-1 != nHandler) {
|
|
CCScriptEngineManager::sharedManager()->getScriptEngine()->executeEvent(nHandler,"");
|
|
}
|
|
}
|
|
}
|
|
|
|
virtual void onError(WebSocket* ws, const WebSocket::ErrorCode& error)
|
|
{
|
|
CLuaWebSocket* luaWs = dynamic_cast<CLuaWebSocket*>(ws);
|
|
if (NULL != luaWs) {
|
|
int nHandler = luaWs->getScriptHandler(CLuaWebSocket::kwebSocketScriptHandlerError);
|
|
if (-1 != nHandler) {
|
|
CCScriptEngineManager::sharedManager()->getScriptEngine()->executeEvent(nHandler,"");
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
private:
|
|
std::map<int,int> m_mapScriptHandler;
|
|
};
|
|
|
|
#ifdef __cplusplus
|
|
static int tolua_collect_WebSocket (lua_State* tolua_S)
|
|
{
|
|
CLuaWebSocket* self = (CLuaWebSocket*) tolua_tousertype(tolua_S,1,0);
|
|
Mtolua_delete(self);
|
|
return 0;
|
|
}
|
|
#endif
|
|
/* function to release collected object via destructor */
|
|
static void tolua_reg_Web_Socket_type(lua_State* tolua_S)
|
|
{
|
|
tolua_usertype(tolua_S, "WebSocket");
|
|
}
|
|
|
|
/* method: create of class WebSocket */
|
|
#ifndef TOLUA_DISABLE_tolua_Cocos2d_WebSocket_create00
|
|
static int tolua_Cocos2d_WebSocket_create00(lua_State* tolua_S)
|
|
{
|
|
#ifndef TOLUA_RELEASE
|
|
tolua_Error tolua_err;
|
|
if (
|
|
!tolua_isusertable(tolua_S,1,"WebSocket",0,&tolua_err) ||
|
|
!tolua_isstring(tolua_S,2,0,&tolua_err) ||
|
|
!tolua_isnoobj(tolua_S,3,&tolua_err)
|
|
)
|
|
goto tolua_lerror;
|
|
else
|
|
#endif
|
|
{
|
|
const char* urlName = ((const char*) tolua_tostring(tolua_S,2,0));
|
|
CLuaWebSocket *wSocket = new CLuaWebSocket();
|
|
wSocket->init(*wSocket, urlName);
|
|
tolua_pushusertype(tolua_S,(void*)wSocket,"WebSocket");
|
|
}
|
|
return 1;
|
|
#ifndef TOLUA_RELEASE
|
|
tolua_lerror:
|
|
tolua_error(tolua_S,"#ferror in function 'create'.",&tolua_err);
|
|
return 0;
|
|
#endif
|
|
}
|
|
#endif //#ifndef TOLUA_DISABLE
|
|
|
|
/* method: createByAProtocol of class WebSocket */
|
|
#ifndef TOLUA_DISABLE_tolua_Cocos2d_WebSocket_createByAProtocol00
|
|
static int tolua_Cocos2d_WebSocket_createByAProtocol00(lua_State* tolua_S)
|
|
{
|
|
#ifndef TOLUA_RELEASE
|
|
tolua_Error tolua_err;
|
|
if (
|
|
!tolua_isusertable(tolua_S,1,"WebSocket",0,&tolua_err) ||
|
|
!tolua_isstring(tolua_S,2,0,&tolua_err) ||
|
|
!tolua_isstring(tolua_S,3,0,&tolua_err) ||
|
|
!tolua_isnoobj(tolua_S,4,&tolua_err)
|
|
)
|
|
goto tolua_lerror;
|
|
else
|
|
#endif
|
|
{
|
|
const char *urlName = ((const char*) tolua_tostring(tolua_S,2,0));
|
|
const char *protocol = ((const char*) tolua_tostring(tolua_S,3,0));
|
|
std::vector<std::string> protocols;
|
|
protocols.push_back(protocol);
|
|
CLuaWebSocket *wSocket = new CLuaWebSocket();
|
|
wSocket->init(*wSocket, urlName,&protocols);
|
|
tolua_pushusertype(tolua_S,(void*)wSocket,"WebSocket");
|
|
}
|
|
return 1;
|
|
#ifndef TOLUA_RELEASE
|
|
tolua_lerror:
|
|
tolua_error(tolua_S,"#ferror in function 'createByAProtocol'.",&tolua_err);
|
|
return 0;
|
|
#endif
|
|
}
|
|
#endif //#ifndef TOLUA_DISABLE
|
|
|
|
/* method: createByAProtocol of class WebSocket */
|
|
#ifndef TOLUA_DISABLE_tolua_Cocos2d_WebSocket_createByProtocolArray00
|
|
static int tolua_Cocos2d_WebSocket_createByProtocolArray00(lua_State* tolua_S)
|
|
{
|
|
#ifndef TOLUA_RELEASE
|
|
tolua_Error tolua_err;
|
|
if (
|
|
!tolua_isusertable(tolua_S,1,"WebSocket",0,&tolua_err) ||
|
|
!tolua_isstring(tolua_S,2,0,&tolua_err) ||
|
|
!tolua_isusertable(tolua_S,3,"CCArray",0,&tolua_err) ||
|
|
!tolua_isnoobj(tolua_S,4,&tolua_err)
|
|
)
|
|
goto tolua_lerror;
|
|
else
|
|
#endif
|
|
{
|
|
const char *urlName = ((const char*) tolua_tostring(tolua_S,2,0));
|
|
CCArray* protocolArray = ((CCArray*) tolua_tousertype(tolua_S,3,0));
|
|
std::vector<std::string> protocols;
|
|
if (NULL != protocolArray) {
|
|
CCObject* pObj = NULL;
|
|
CCARRAY_FOREACH(protocolArray, pObj)
|
|
{
|
|
CCString* pStr = (CCString*)pObj;
|
|
if (NULL != pStr) {
|
|
protocols.push_back(pStr->getCString());
|
|
}
|
|
}
|
|
}
|
|
CLuaWebSocket *wSocket = new CLuaWebSocket();
|
|
wSocket->init(*wSocket, urlName,&protocols);
|
|
tolua_pushusertype(tolua_S,(void*)wSocket,"WebSocket");
|
|
}
|
|
return 1;
|
|
#ifndef TOLUA_RELEASE
|
|
tolua_lerror:
|
|
tolua_error(tolua_S,"#ferror in function 'createByProtocolArray'.",&tolua_err);
|
|
return 0;
|
|
#endif
|
|
}
|
|
#endif //#ifndef TOLUA_DISABLE
|
|
|
|
/* method: getReadyState of class WebSocket */
|
|
#ifndef TOLUA_DISABLE_tolua_Cocos2d_WebSocket_getReadyState00
|
|
static int tolua_Cocos2d_WebSocket_getReadyState00(lua_State* tolua_S)
|
|
{
|
|
#ifndef TOLUA_RELEASE
|
|
tolua_Error tolua_err;
|
|
if (
|
|
!tolua_isusertype(tolua_S,1,"WebSocket",0,&tolua_err) ||
|
|
!tolua_isnoobj(tolua_S,2,&tolua_err)
|
|
)
|
|
goto tolua_lerror;
|
|
else
|
|
#endif
|
|
{
|
|
CLuaWebSocket *self = (CLuaWebSocket*)tolua_tousertype(tolua_S,1,0);
|
|
int tolua_ret = -1;
|
|
if (NULL != self) {
|
|
tolua_ret = self->getReadyState();
|
|
}
|
|
tolua_pushnumber(tolua_S,(lua_Number)tolua_ret);
|
|
}
|
|
return 1;
|
|
#ifndef TOLUA_RELEASE
|
|
tolua_lerror:
|
|
tolua_error(tolua_S,"#ferror in function 'getReadyState'.",&tolua_err);
|
|
return 0;
|
|
#endif
|
|
}
|
|
#endif //#ifndef TOLUA_DISABLE
|
|
|
|
/* method: sendTextMsg of class WebSocket */
|
|
#ifndef TOLUA_DISABLE_tolua_Cocos2d_WebSocket_sendTextMsg00
|
|
static int tolua_Cocos2d_WebSocket_sendTextMsg00(lua_State* tolua_S)
|
|
{
|
|
#ifndef TOLUA_RELEASE
|
|
tolua_Error tolua_err;
|
|
if (
|
|
!tolua_isusertype(tolua_S,1,"WebSocket",0,&tolua_err) ||
|
|
!tolua_isstring(tolua_S,2,0,&tolua_err) ||
|
|
!tolua_isnoobj(tolua_S,3,&tolua_err)
|
|
)
|
|
goto tolua_lerror;
|
|
else
|
|
#endif
|
|
{
|
|
CLuaWebSocket* self = (CLuaWebSocket*) tolua_tousertype(tolua_S,1,0);
|
|
const char *pData = ((const char*) tolua_tostring(tolua_S,2,0));
|
|
if (NULL != self && NULL != pData && strlen(pData) > 0) {
|
|
std::string strData = pData;
|
|
self->send(strData);
|
|
}
|
|
}
|
|
return 0;
|
|
#ifndef TOLUA_RELEASE
|
|
tolua_lerror:
|
|
tolua_error(tolua_S,"#ferror in function 'sendStrMsg'.",&tolua_err);
|
|
return 0;
|
|
#endif
|
|
}
|
|
#endif //#ifndef TOLUA_DISABLE
|
|
|
|
/* method: close of class WebSocket */
|
|
#ifndef TOLUA_DISABLE_tolua_Cocos2d_WebSocket_close00
|
|
static int tolua_Cocos2d_WebSocket_close00(lua_State* tolua_S)
|
|
{
|
|
#ifndef TOLUA_RELEASE
|
|
tolua_Error tolua_err;
|
|
if (
|
|
!tolua_isusertype(tolua_S,1,"WebSocket",0,&tolua_err) ||
|
|
!tolua_isnoobj(tolua_S,2,&tolua_err)
|
|
)
|
|
goto tolua_lerror;
|
|
else
|
|
#endif
|
|
{
|
|
CLuaWebSocket* self = (CLuaWebSocket*) tolua_tousertype(tolua_S,1,0);
|
|
if (NULL != self ) {
|
|
self->close();
|
|
}
|
|
}
|
|
return 0;
|
|
#ifndef TOLUA_RELEASE
|
|
tolua_lerror:
|
|
tolua_error(tolua_S,"#ferror in function 'getReadyState'.",&tolua_err);
|
|
return 0;
|
|
#endif
|
|
}
|
|
#endif //#ifndef TOLUA_DISABLE
|
|
|
|
/* method: addHandlerOfDelegateEvent of class WebSocket */
|
|
#ifndef TOLUA_DISABLE_tolua_Cocos2d_WebSocket_registerScriptHandler00
|
|
static int tolua_Cocos2d_WebSocket_registerScriptHandler00(lua_State* tolua_S)
|
|
{
|
|
#ifndef TOLUA_RELEASE
|
|
tolua_Error tolua_err;
|
|
if (
|
|
!tolua_isusertype(tolua_S,1,"WebSocket",0,&tolua_err) ||
|
|
!toluafix_isfunction(tolua_S,2,"LUA_FUNCTION",0,&tolua_err) ||
|
|
!tolua_isnumber(tolua_S,3,0,&tolua_err) ||
|
|
!tolua_isnoobj(tolua_S,4,&tolua_err)
|
|
)
|
|
goto tolua_lerror;
|
|
else
|
|
#endif
|
|
{
|
|
CLuaWebSocket* self = (CLuaWebSocket*) tolua_tousertype(tolua_S,1,0);
|
|
if (NULL != self ) {
|
|
int nFunID = ( toluafix_ref_function(tolua_S,2,0));
|
|
CLuaWebSocket::webSocketScriptHandlerType handlerType = ((CLuaWebSocket::webSocketScriptHandlerType) (int) tolua_tonumber(tolua_S,3,0));
|
|
self->registerScriptHandler(nFunID, handlerType);
|
|
}
|
|
}
|
|
return 0;
|
|
#ifndef TOLUA_RELEASE
|
|
tolua_lerror:
|
|
tolua_error(tolua_S,"#ferror in function 'registerScriptHandler'.",&tolua_err);
|
|
return 0;
|
|
#endif
|
|
}
|
|
#endif //#ifndef TOLUA_DISABLE
|
|
|
|
/* method: removeHandlerOfDelegateEvent of class WebSocket */
|
|
#ifndef TOLUA_DISABLE_tolua_Cocos2d_WebSocket_unregisterScriptHandler00
|
|
static int tolua_Cocos2d_WebSocket_unregisterScriptHandler00(lua_State* tolua_S)
|
|
{
|
|
#ifndef TOLUA_RELEASE
|
|
tolua_Error tolua_err;
|
|
if (
|
|
!tolua_isusertype(tolua_S,1,"WebSocket",0,&tolua_err) ||
|
|
!tolua_isnumber(tolua_S,2,0,&tolua_err) ||
|
|
!tolua_isnoobj(tolua_S,3,&tolua_err)
|
|
)
|
|
goto tolua_lerror;
|
|
else
|
|
#endif
|
|
{
|
|
CLuaWebSocket* self = (CLuaWebSocket*) tolua_tousertype(tolua_S,1,0);
|
|
if (NULL != self ) {
|
|
CLuaWebSocket::webSocketScriptHandlerType handlerType = ((CLuaWebSocket::webSocketScriptHandlerType) (int) tolua_tonumber(tolua_S,2,0));
|
|
self->unregisterScriptHandler(handlerType);
|
|
}
|
|
}
|
|
return 0;
|
|
#ifndef TOLUA_RELEASE
|
|
tolua_lerror:
|
|
tolua_error(tolua_S,"#ferror in function 'unregisterScriptHandler'.",&tolua_err);
|
|
return 0;
|
|
#endif
|
|
}
|
|
#endif //#ifndef TOLUA_DISABLE
|
|
|
|
/* method: sendBinaryMsg of class WebSocket */
|
|
#ifndef TOLUA_DISABLE_tolua_Cocos2d_WebSocket_sendBinaryMsg00
|
|
static int tolua_Cocos2d_WebSocket_sendBinaryMsg00(lua_State* tolua_S)
|
|
{
|
|
#ifndef TOLUA_RELEASE
|
|
tolua_Error tolua_err;
|
|
if (
|
|
!tolua_isusertype(tolua_S,1,"WebSocket",0,&tolua_err) ||
|
|
!tolua_istable(tolua_S, 2, 0, &tolua_err) ||
|
|
!tolua_isnumber(tolua_S,3, 0,&tolua_err) ||
|
|
!tolua_isnoobj(tolua_S, 4,&tolua_err)
|
|
)
|
|
goto tolua_lerror;
|
|
else
|
|
#endif
|
|
{
|
|
CLuaWebSocket* self = (CLuaWebSocket*) tolua_tousertype(tolua_S,1,0);
|
|
int nLength = lua_tonumber(tolua_S, 3);
|
|
|
|
if (NULL != self && nLength > 0) {
|
|
unsigned char* binaryArray = new unsigned char[nLength];
|
|
for (int i = 0; i < nLength; i++) {
|
|
binaryArray[i] = (unsigned char)tolua_tofieldnumber(tolua_S, 2, i+1, 0);
|
|
}
|
|
self->send(binaryArray, nLength);
|
|
}
|
|
}
|
|
return 0;
|
|
#ifndef TOLUA_RELEASE
|
|
tolua_lerror:
|
|
tolua_error(tolua_S,"#ferror in function 'sendBinaryMsg'.",&tolua_err);
|
|
return 0;
|
|
#endif
|
|
}
|
|
#endif //#ifndef TOLUA_DISABLE
|
|
|
|
TOLUA_API int tolua_web_socket_open(lua_State* tolua_S){
|
|
tolua_open(tolua_S);
|
|
tolua_reg_Web_Socket_type(tolua_S);
|
|
tolua_module(tolua_S,NULL,0);
|
|
tolua_beginmodule(tolua_S,NULL);
|
|
tolua_constant(tolua_S,"kStateConnecting",WebSocket::kStateConnecting);
|
|
tolua_constant(tolua_S,"kStateOpen",WebSocket::kStateOpen);
|
|
tolua_constant(tolua_S,"kStateClosing",WebSocket::kStateClosing);
|
|
tolua_constant(tolua_S,"kStateClosed",WebSocket::kStateClosed);
|
|
tolua_constant(tolua_S,"kwebSocketScriptHandlerOpen",CLuaWebSocket::kwebSocketScriptHandlerOpen);
|
|
tolua_constant(tolua_S,"kwebSocketScriptHandlerMessage",CLuaWebSocket::kwebSocketScriptHandlerMessage);
|
|
tolua_constant(tolua_S,"kwebSocketScriptHandlerClose",CLuaWebSocket::kwebSocketScriptHandlerClose);
|
|
tolua_constant(tolua_S,"kwebSocketScriptHandlerError",CLuaWebSocket::kwebSocketScriptHandlerError);
|
|
#ifdef __cplusplus
|
|
tolua_cclass(tolua_S,"WebSocket","WebSocket","",tolua_collect_WebSocket);
|
|
#else
|
|
tolua_cclass(tolua_S,"WebSocket","WebSocket","",NULL);
|
|
#endif
|
|
tolua_beginmodule(tolua_S,"WebSocket");
|
|
tolua_function(tolua_S, "create", tolua_Cocos2d_WebSocket_create00);
|
|
tolua_function(tolua_S, "createByAProtocol", tolua_Cocos2d_WebSocket_createByAProtocol00);
|
|
tolua_function(tolua_S, "createByProtocolArray", tolua_Cocos2d_WebSocket_createByProtocolArray00);
|
|
tolua_function(tolua_S, "getReadyState", tolua_Cocos2d_WebSocket_getReadyState00);
|
|
tolua_function(tolua_S, "sendTextMsg", tolua_Cocos2d_WebSocket_sendTextMsg00);
|
|
tolua_function(tolua_S, "close", tolua_Cocos2d_WebSocket_close00);
|
|
tolua_function(tolua_S, "registerScriptHandler", tolua_Cocos2d_WebSocket_registerScriptHandler00);
|
|
tolua_function(tolua_S, "unregisterScriptHandler", tolua_Cocos2d_WebSocket_unregisterScriptHandler00);
|
|
tolua_function(tolua_S, "sendBinaryMsg", tolua_Cocos2d_WebSocket_sendBinaryMsg00);
|
|
tolua_endmodule(tolua_S);
|
|
tolua_endmodule(tolua_S);
|
|
return 1;
|
|
}
|