123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118 |
- /**
- * 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 "util/tc_option.h"
- #include "util/tc_file.h"
- #include "tars2cs.h"
- void usage()
- {
- cout << "Usage : tars2cs [OPTION] tarsfile" << endl;
- cout << " tars2cs support type: bool byte short int long float double vector map" << endl;
- cout << "supported [OPTION]:" << endl;
- cout << " --help help,print this" << endl;
- cout << " --dir=DIRECTORY generate source file to DIRECTORY(create tars protocol file to DIRECTORY, default is current directory)" << endl;
- cout << " --base-package=NAME package prefix, default 'com.qq.'" << endl;
- cout << " --with-servant also generate servant class" << endl;
- cout << endl;
- exit(0);
- }
- void check(vector<string> &vTars)
- {
- for(size_t i = 0; i < vTars.size(); i++)
- {
- string ext = tars::TC_File::extractFileExt(vTars[i]);
- if(ext == "tars")
- {
- if(!tars::TC_File::isFileExist(vTars[i]))
- {
- cerr << "file '" << vTars[i] << "' not exists" << endl;
- usage();
- exit(0);
- }
- }
- else
- {
- cerr << "only support tars file." << endl;
- exit(0);
- }
- }
- }
- int main(int argc, char* argv[])
- {
- if(argc < 2)
- {
- usage();
- }
- tars::TC_Option option;
- option.decode(argc, argv);
- vector<string> vTars = option.getSingle();
- check(vTars);
- if(option.hasParam("help"))
- {
- usage();
- }
- Tars2Cs t2cs;
- if(option.getValue("dir") != "")
- {
- t2cs.setBaseDir(option.getValue("dir"));
- }
- else
- {
- t2cs.setBaseDir(".");
- }
- //����ǰ
- if(option.hasParam("base-package"))
- {
- t2cs.setBasePackage(option.getValue("base-package"));
- }
- else
- {
- t2cs.setBasePackage("Com.QQ.");
- }
- if(option.hasParam("with-servant"))
- {
- t2cs.setWithServant(true);
- }
- else
- {
- t2cs.setWithServant(false);
- }
- try
- {
- g_parse->setTars(option.hasParam("with-tars"));
- for(size_t i = 0; i < vTars.size(); i++)
- {
- g_parse->parse(vTars[i]);
- t2cs.createFile(vTars[i]);
- }
- }catch(exception& e)
- {
- cerr<<e.what()<<endl;
- }
- return 0;
- }
|