123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158 |
- /**
- * Tencent is pleased to support the open source community by making Tars available.
- *
- * Copyright (C) 2016THL A29 Limited, a Tencent company. All rights reserved.
- *
- * Licensed under the BSD 3-Clause License (the "License"); you may not use this file except
- * in compliance with the License. You may obtain a copy of the License at
- *
- * https://opensource.org/licenses/BSD-3-Clause
- *
- * Unless required by applicable law or agreed to in writing, software distributed
- * under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
- * CONDITIONS OF ANY KIND, either express or implied. See the License for the
- * specific language governing permissions and limitations under the License.
- */
- #include "code_generator.h"
- string CodeGenerator::generateDTSServer(const NamespacePtr &nPtr, const InterfacePtr &pPtr)
- {
- vector<OperationPtr> &vOperation = pPtr->getAllOperationPtr();
- ostringstream str;
- //class
- INC_TAB;
- str << TAB << "class " << pPtr->getId() << "Imp {" << endl;
- INC_TAB;
- str << TAB << "initialize(): PromiseLike<any> | void;" << endl;
- str << TAB << "protected onDispatch(current: " << IDL_NAMESPACE_STR << "Rpc." << IDL_TYPE << "Current, funcName: string, binBuffer: " << IDL_NAMESPACE_STR << "Stream.BinBuffer): number" << endl;
- for (size_t i = 0; i < vOperation.size(); i++)
- {
- OperationPtr &oPtr = vOperation[i];
- str << TAB << oPtr->getId() << "(current: " << pPtr->getId() << "Imp." << oPtr->getId() << "Current";
- vector<ParamDeclPtr> &vParamDecl = oPtr->getAllParamDeclPtr();
- for (size_t j = 0; j < vParamDecl.size(); j++)
- {
- str << ", " << vParamDecl[j]->getTypeIdPtr()->getId() << ": " << getTsType(vParamDecl[j]->getTypeIdPtr()->getTypePtr());
- }
- str << "): any;" << endl;
- }
- DEL_TAB;
- str << TAB << "}" << endl;
- // Additional namespace
- str << TAB << "namespace " << pPtr->getId() << "Imp {" << endl;
- INC_TAB;
- for (size_t i = 0; i < vOperation.size(); i++)
- {
- OperationPtr &oPtr = vOperation[i];
- str << TAB << "interface " << oPtr->getId() << "Current extends " << IDL_NAMESPACE_STR << "Rpc." << IDL_TYPE << "Current {" <<endl;
- INC_TAB;
- str << TAB;
- if (oPtr->getReturnPtr()->getTypePtr())
- {
- str << "sendResponse(ret: " << getTsType(oPtr->getReturnPtr()->getTypePtr());
- vector<ParamDeclPtr> &vParamDecl = oPtr->getAllParamDeclPtr();
- for (size_t j = 0; j < vParamDecl.size(); j++)
- {
- if(!vParamDecl[j]->isOut()) {
- continue;
- }
- str << ", " << vParamDecl[j]->getTypeIdPtr()->getId() << ": " << getTsType(vParamDecl[j]->getTypeIdPtr()->getTypePtr()) ;
- }
- str << "): void;" << endl;
- }
- else
- {
- str << "sendResponse(): void;" << endl;
- }
- DEL_TAB;
- str << TAB << "}" << endl;
- }
- DEL_TAB;
- str << TAB << "}" << endl;
- DEL_TAB;
- return str.str();
- }
- string CodeGenerator::generateDTSServer(const NamespacePtr &pPtr, bool &bNeedStream, bool &bNeedRpc)
- {
- vector<InterfacePtr> &is = pPtr->getAllInterfacePtr();
- ostringstream str;
- for (size_t i = 0; i < is.size(); i++)
- {
- str << generateDTSServer(pPtr, is[i]) << endl;
- }
- if (is.size() != 0)
- {
- bNeedRpc = true;
- bNeedStream = true;
- }
- return str.str();
- }
- void CodeGenerator::generateDTSServer(const ContextPtr &pPtr)
- {
- vector<NamespacePtr> namespaces = pPtr->getNamespaces();
- // generate server classes with encoders and decoders
- ostringstream estr;
- bool bNeedStream = false;
- bool bNeedRpc = false;
- for(size_t i = 0; i < namespaces.size(); i++)
- {
- ostringstream kstr;
- kstr << generateDTS(namespaces[i], bNeedStream);
- kstr << generateDTSServer(namespaces[i], bNeedStream, bNeedRpc);
- estr << generateDTS(namespaces[i], kstr.str());
- }
- if(estr.str().empty())
- {
- return;
- }
- // generate module imports
- ostringstream ostr;
- if (bNeedStream)
- {
- ostr << "import * as " << IDL_NAMESPACE_STR << "Stream from \"" << _sStreamPath << "\";" << endl;
- }
- if (bNeedRpc)
- {
- ostr << "import * as " << IDL_NAMESPACE_STR << "Rpc from \"" << _sRpcPath << "\";" << endl;
- }
- for (map<string, ImportFile>::iterator it = _mapFiles.begin(); it != _mapFiles.end(); it++)
- {
- if (it->second.sModule.empty()) continue;
- if (estr.str().find(it->second.sModule + ".") == string::npos) continue;
- ostr << "import * as " << it->second.sModule << " from \"" << TC_File::excludeFileExt(it->second.sFile) << "\";" << endl;
- }
- ostringstream str;
- str << printHeaderRemark("Server");
- str << DISABLE_TSLINT << endl;
- str << DISABLE_ESLINT << endl;
- str << endl;
- str << ostr.str() << endl;
- str << estr.str() << endl;
- string sFileName = TC_File::excludeFileExt(_sToPath + TC_File::extractFileName(pPtr->getFileName())) + ".d.ts";
- TC_File::makeDirRecursive(_sToPath);
- makeUTF8File(sFileName, str.str());
- }
|