123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243 |
- /**
- * 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::generateDTS(const EnumPtr &pPtr, const string &sNamespace)
- {
- ostringstream s;
- INC_TAB;
- s << TAB << (_bEnumReverseMappings ? "enum " : "const enum ") << pPtr->getId() << " {" << endl;
- INC_TAB;
- int nenum = -1;
- bool bDependent = false;
- vector<TypeIdPtr>& member = pPtr->getAllMemberPtr();
- for (size_t i = 0; i < member.size(); i++)
- {
- bDependent |= isDependent(sNamespace, member[i]->getId());
- if (member[i]->hasDefault())
- {
- nenum = TC_Common::strto<int>(member[i]->def());
- }
- else
- {
- nenum++;
- }
- s << TAB << member[i]->getId() << " = " << TC_Common::tostr(nenum) << ((i < member.size() - 1) ? "," : "") << endl;
- }
- DEL_TAB;
- s << TAB << "}" << endl;
- DEL_TAB;
- if (!_bMinimalMembers || _bEntry || bDependent || isDependent(sNamespace, pPtr->getId())) {
- return s.str();
- } else {
- return "";
- }
- }
- string CodeGenerator::generateDTS(const ConstPtr &pPtr, const string &sNamespace, bool &bNeedStream)
- {
- if (_bMinimalMembers && !_bEntry && !isDependent(sNamespace, pPtr->getTypeIdPtr()->getId()))
- {
- return "";
- }
- ostringstream s;
- if (_bStringBinaryEncoding && GET_CONST_GRAMMAR_PTR(pPtr)->t == CONST_GRAMMAR(STRING))
- {
- bNeedStream = true;
- }
- INC_TAB;
- s << TAB << "const " << pPtr->getTypeIdPtr()->getId() << ":"
- << getTsType(pPtr->getTypeIdPtr()->getTypePtr()) << ";"
- << endl;
- DEL_TAB;
- return s.str();
- }
- string CodeGenerator::generateDTS(const StructPtr &pPtr, const string &sNamespace)
- {
- if (_bMinimalMembers && !_bEntry && !isDependent(sNamespace, pPtr->getId()))
- {
- return "";
- }
- vector<TypeIdPtr> &member = pPtr->getAllMemberPtr();
- INC_TAB;
- ostringstream s;
- s << TAB << "class " << pPtr->getId() << " {" << endl;
- INC_TAB;
- for (size_t i = 0; i < member.size(); i++)
- {
- s << TAB << (member[i]->getId()) << ": " << getTsType(member[i]->getTypePtr()) << ";" << endl;
- }
- if (member.size() > 0)
- {
- s << endl;
- }
- /*
- * Size Optimize:
- * Remove <mutil_map> support.
- * Remove toBinBuffer, readFromObject, toObject, new, create members.
- */
- if (_iOptimizeLevel != Os)
- {
- s << TAB << "toObject(): " << pPtr->getId() << ".Object;" << endl;
- s << TAB << "readFromObject(json: " << pPtr->getId() << ".Object): " << pPtr->getId() << ";" << endl;
- s << TAB << "toBinBuffer(): " << IDL_NAMESPACE_STR << "Stream.BinBuffer;" << endl;
- s << TAB << "static new(): " << pPtr->getId() << ";" << endl;
- s << TAB << "static create(is: " << IDL_NAMESPACE_STR << "Stream." << IDL_TYPE << "InputStream): " << pPtr->getId() << ";" << endl;
- }
- DEL_TAB;
- s << TAB << "}" << endl << endl;
- s << TAB << "namespace " << pPtr->getId() << " {" << endl;
- INC_TAB;
- s << TAB << "interface Object {" << endl;
- INC_TAB;
- for (size_t i = 0; i < member.size(); i++)
- {
- const string &sType = getTsType(member[i]->getTypePtr(), false);
- s << TAB << (member[i]->getId()) << (member[i]->isRequire() ? ": " : "?: ") << (!sType.empty() ? sType : "never") << ";" << endl;
- }
- DEL_TAB;
- s << TAB << "}" << endl;
- DEL_TAB;
- s << TAB << "}" << endl;
- DEL_TAB;
- return s.str();
- }
- string CodeGenerator::generateDTS(const NamespacePtr &pPtr, bool &bNeedStream)
- {
- // struct
- ostringstream sstr;
- vector<StructPtr> ss(pPtr->getAllStructPtr());
- for (size_t last = 0; last != ss.size() && ss.size() != 0;)
- {
- last = ss.size();
- for (vector<StructPtr>::iterator iter=ss.begin(); iter!=ss.end();)
- {
- string str = generateDTS(*iter, pPtr->getId());
- if (!str.empty())
- {
- sstr << str << endl;
- iter = ss.erase(iter);
- }
- else
- {
- iter++;
- }
- }
- }
- // const
- ostringstream cstr;
- vector<ConstPtr> &cs = pPtr->getAllConstPtr();
- for (size_t i = 0; i < cs.size(); i++)
- {
- cstr << generateDTS(cs[i], pPtr->getId(), bNeedStream);
- }
- // enum
- ostringstream estr;
- vector<EnumPtr> &es = pPtr->getAllEnumPtr();
- for (size_t i = 0; i < es.size(); i++)
- {
- estr << generateDTS(es[i], pPtr->getId());
- }
- ostringstream kstr;
- if (!estr.str().empty()) kstr << estr.str() << endl;
- if (!cstr.str().empty()) kstr << cstr.str() << endl;
- if (!sstr.str().empty())
- {
- bNeedStream = true;
- kstr << sstr.str();
- }
- return kstr.str();
- }
- string CodeGenerator::generateDTS(const NamespacePtr &pPtr, const string &sContent)
- {
- ostringstream str;
- if (!sContent.empty())
- {
- str << "export namespace " << pPtr->getId() << " {" << endl;
- str << sContent;
- str << "}" << endl << endl;
- }
- return str.str();
- }
- void CodeGenerator::generateDTS(const ContextPtr &pPtr)
- {
- vector<NamespacePtr> namespaces = pPtr->getNamespaces();
- // generate encoders and decoders
- ostringstream estr;
- bool bNeedStream = false;
- for (size_t i = 0; i < namespaces.size(); i++)
- {
- estr << generateDTS(namespaces[i], generateDTS(namespaces[i], bNeedStream));
- }
- if (estr.str().empty())
- {
- return;
- }
- // generate module imports
- ostringstream ostr;
- if (bNeedStream)
- {
- ostr << "import * as " << IDL_NAMESPACE_STR << "Stream from \"" << _sStreamPath << "\";" << 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;
- }
- // concat generated code
- ostringstream sstr;
- sstr << printHeaderRemark("Structure");
- sstr << DISABLE_TSLINT << endl;
- sstr << DISABLE_ESLINT << endl;
- sstr << endl;
- sstr << ostr.str() << endl;
- sstr << estr.str() << endl;
- string sFileName = TC_File::excludeFileExt(_sToPath + TC_File::extractFileName(pPtr->getFileName())) + IDL_TYPE + ".d.ts";
- TC_File::makeDirRecursive(_sToPath);
- makeUTF8File(sFileName, sstr.str());
- }
|