main.cpp
#include <iostream>
#include <getopt.h>
#include "parsingargs.h"
#include <string.h>
using namespace std;
int main(int argc, char * argv[])
{
//string tmpPara = "-p \"4567\" --out 1.log "; //ok
//string tmpPara = "xxxx -p \"4567\" --out 1.log ";//ok
//string tmpPara = "-p \"4567\" --out 1.log 2.log"; //ok
string tmpPara = "";
for(int i=1;i <argc; i++)
{
cout << i << "=" << argv[i] <<"---"<< endl;
if(strlen(argv[i]) == 0) //處理空字符串
{
cout << "find NULL" << endl;
tmpPara += char(31);
}
else
{
tmpPara += argv[i];
}
tmpPara += " ";
}
std::map<std::string, std::vector<std::string> > result;
ParsingArgs pa;
pa.AddArgType('l',"getlist", ParsingArgs::NO_VALUE);
pa.AddArgType('p',"getuser", ParsingArgs::MAYBE_VALUE);
pa.AddArgType('o',"outFile", ParsingArgs::MUST_VALUE);
bool bExit = false;
do
{
result.clear();
cout << "input is:" << tmpPara << "---size = " << tmpPara.size()<< endl;
std::string errPos;
int iRet = pa.Parse(tmpPara,result, errPos);
if(0>iRet)
{
cout << "參數錯誤" << iRet << errPos << endl;
}
else
{
map<std::string, std::vector<std::string> >::iterator it = result.begin();
for(; it != result.end(); ++it)
{
cout << "key=" << it->first<<endl;
for(int i=0; i<it->second.size(); ++i)
{
cout << " value =" << it->second[i] << "------" << endl;
}
}
}
string str;
cout << ">>> ";
getline(cin, tmpPara);
if(0 == tmpPara.compare("exit"))
{
bExit = true;
}
}while(!bExit);
return 0;
}
parsingargs.h
#ifndef PARSINGARGS_H
#define PARSINGARGS_H
/* purpose @ 解析輸入的參數,需先通過AddArgType將必須參數和可允許的參數key加入到判定列表中
* 通過Parse中的result將結果返回,其中結果的key為合法的key,vecotr為參數列表
* 參數列表支持去掉參數前后的引號和\對引號和\的轉義
*
* 特殊合法字段:
* 格式 實際存儲值
* \\value\" \value"
* "\\\value\"" \value"
*
* 注意事項:
* 1、輸入參數列表中參數分隔以空格區分
* 2、- 后跟單字符關鍵字,--后跟長字符串關鍵字
* 3、關鍵字不能重復出現,長短關鍵字不能同時出現在參數列表,否則會Parse函數會提示參數錯誤
*
* 用法:
* ParsingArgs pa;
* pa.AddArgType('l',"getlist", ParsingArgs::NO_VALUE); //NO_VALUE關鍵字后不能有參數
* pa.AddArgType('p',"getuser", ParsingArgs::MAYBE_VALUE); //MAYBE_VALUE 關鍵字后可能有關鍵字
* pa.AddArgType('o',"outFile", ParsingArgs::MUST_VALUE); // MUST_VALUE 關鍵字后必須有參數
* std::map<std::string, std::vector<std::string> > result;
* int iRet = pa.Parse(tmpPara,result); //result以輸入關鍵字為key存儲相關的值序列
*
* date @ 2014.02.19
* author @ haibin.wang
*
*/
#include <map>
#include <vector>
#include <string>
class ParsingArgs
{
public:
ParsingArgs();
~ParsingArgs();
enum KeyFlag{ INVALID_KEY=-1, NO_VALUE, MAYBE_VALUE, MUST_VALUE};
/* pur @ 添加解釋參數,一個參數可以是長參數,也可以是縮寫的段參數,短參數只能為單字符,longName和shortName至少要有一個
* para @ shortName 短參數名,0為不要短參數
* para @ longName 長參數名 ,NULL為不要長參數
* para @ flag 是否需要參數,0不需要,1必須要,2可要可不要
* return @ true 添加成功,false添加失敗
*/
bool AddArgType(const char shortName, const char * longName = NULL, KeyFlag flag=NO_VALUE);
/* pur @ 根據參數類型解釋傳入的字符串
* para @ paras 需要解釋的字符串
* para @ result 返回解析后的結果
* para @ errPos 當錯誤的時候返回出錯的大概位置
* return @ 0 解釋成功,負數 解釋失敗
* -1 未知參數錯誤
-2 不能有參數的選項有參數錯誤
* -3 必有參數選項后沒有跟參數
* -4 關鍵字沒有加入到AddArgType中
* -5 關鍵字重復
*/
int Parse(const std::string & paras, std::map<std::string, std::vector<std::string> > & result, std::string &errPos);
private:
/* pur @ 判定傳入的參數是否是已經添加的參數類型,如果是則去掉-或--,并返回
* para @ key 要判定的參數
* return @ -1 不是合法參數類型 否則返回Option中的flag
*/
KeyFlag GetKeyFlag(std::string &key);
/* pur @ 刪除關鍵字前的-或--
*/
void RemoveKeyFlag(std::string & paras);
/* pur @ 從Paras中獲取一個單詞,自動過濾掉單詞前后引號,并實現\對空格和引號的轉義
* para @ Paras 返回第一個單詞后的所有內容
* para @ word 返回第一單詞
* return @ 成功返回true,false失敗
*/
bool GetWord(std::string & Paras, std::string & word);
/* pur @ 檢查關鍵字是否重復
* para @ key 被檢查的關鍵字
* para @ result已存儲的關鍵字序列
* return @ true 是重復的,false不重復
*/
bool IsDuplicateKey(const std::string &key, const std::map<std::string, std::vector<std::string> > & result);
struct Option
{
std::string m_longName;
char m_shortName;
KeyFlag m_flag;
};
std::vector<Option> m_args; //參數信息列表
};
#endif
parsingargs.cpp
#include "parsingargs.h"
#include <list>
ParsingArgs::ParsingArgs()
{
}
ParsingArgs::~ParsingArgs()
{
}
bool ParsingArgs::AddArgType(char shortName, const char * longName, KeyFlag flag)
{
if(NULL == longName && 0 == shortName)
{
return false;
}
Option tmp;
tmp.m_longName = longName;
tmp.m_shortName = shortName;
tmp.m_flag = flag;
m_args.push_back(tmp);
return true;
}
ParsingArgs::KeyFlag ParsingArgs::GetKeyFlag(std::string &key) //返回flag,
{
for(int i=0; i<m_args.size(); ++i)
{
std::string shortName = "-";
std::string longName = "--";
shortName += m_args[i].m_shortName;
longName += m_args[i].m_longName;
if( 0 == key.compare(shortName) ||
(0==key.compare(longName))
)
{
RemoveKeyFlag(key);
return m_args[i].m_flag;
}
}
return INVALID_KEY;
}
void ParsingArgs::RemoveKeyFlag(std::string & word)
{
if(word.size()>=2)
{
if(word[1] == '-')
{
word.erase(1,1);
}
if(word[0] == '-')
{
word.erase(0,1);
}
}
}
/* pur @ 從Paras中獲取一個單詞,自動過濾掉單詞前后引號,并實現\對空格和引號的轉義
* para @ Paras 返回第一個單詞后的所有內容
* para @ word 返回第一單詞
* return @ 成功返回true,false失敗
*/
bool ParsingArgs::GetWord(std::string & Paras, std::string & word)
{
size_t iNotSpacePos = Paras.find_first_not_of(' ',0);//查找第一個非空格字符位置
if(iNotSpacePos == std::string::npos)
{
Paras.clear();
word.clear();
return true;
}
int length = Paras.size();
std::list<char> specialChar;
int islashPos = -1;
for(int i=iNotSpacePos; i<length; i++)
{
char cur=Paras[i];
bool bOk = false;
switch(cur)
{
case ' ':
if(specialChar.empty())
{
if(i!=(length-1))
{
Paras = std::string(Paras, i+1, length-i-1);
}
else
{//最后一個是空格
Paras.clear();
}
bOk = true;
}
else
{
if(specialChar.back() == '\\')
{
specialChar.pop_back();
}
word.append(1,cur);
}
break;
case '"':
if(specialChar.empty())
{
specialChar.push_back(cur);
}
else if(specialChar.back() == cur)
{ //找到匹配的括號
specialChar.pop_back();
}
else if(specialChar.back() == '\\')
{
word.append(1,cur);
specialChar.pop_back();
}
else
{
word.clear();
return false;
}
break;
case '\\':
if(specialChar.empty())
{
specialChar.push_back(cur);
islashPos = i;
}
else if(specialChar.back() == '"')
{
if(i<(length-1))
{
if('"'==Paras[i+1] || '\\'==Paras[i+1])
{
specialChar.push_back(cur);
}
else
{
word.append(1,cur);
}
}
else
{
word.clear();
return false;
}
}
else if('\\' == specialChar.back())
{
word.append(1,cur);
specialChar.pop_back();
}
else
{
word.clear();
return false;
}
break;
default:
word.append(1,Paras[i]);
if(i==(length-1))
{
bOk = true;
Paras.clear();
}
break;
}
if(bOk)
{
return true;
}
}//for end
if(specialChar.empty())
{
Paras.clear();
return true;
}
else
{
return false;
}
}
bool ParsingArgs::IsDuplicateKey(const std::string &key, const std::map<std::string, std::vector<std::string> > & result)
{
if(result.find(key) != result.end())
{
return true; //關鍵字重復
}
for(int i=0; i<m_args.size(); ++i)
{
if( (key.compare(m_args[i].m_longName) == 0 && result.find(std::string(1, m_args[i].m_shortName)) != result.end())
|| (key.compare(std::string(1, m_args[i].m_shortName)) == 0 && result.find(m_args[i].m_longName) != result.end())
)
{
return true;
}
}
return false;
}
int ParsingArgs::Parse(const std::string & Paras, std::map<std::string, std::vector<std::string> > & result, std::string &errPos)
{
std::string tmpString = Paras;
KeyFlag keyFlag = INVALID_KEY; //參數標識
std::string sKey = ""; //key關鍵字
bool bFindValue = false; //是否已經有value
while(!tmpString.empty())
{
std::string word = "";
bool bRet = GetWord(tmpString, word);
if(bRet == false)
{
errPos = tmpString;
return -4;//參數解析錯誤
}
else
{
KeyFlag tmpFlag = GetKeyFlag(word);
if(IsDuplicateKey(word, result))
{
errPos = tmpString;
return -5;
}
if(tmpFlag != INVALID_KEY)
{
if(tmpFlag == MUST_VALUE && keyFlag == MUST_VALUE && !bFindValue)
{
errPos = tmpString;
return -3;
}
keyFlag = tmpFlag;
std::vector<std::string> tmp;
result[word] = tmp;
sKey = word;
bFindValue = false;
}
else
{
switch(keyFlag)
{
case MAYBE_VALUE:
case MUST_VALUE:
{
std::map<std::string, std::vector<std::string> >::iterator it = result.find(sKey);
if(it != result.end())
{
it->second.push_back(word);
bFindValue = true;
}
else
{
errPos = tmpString;
return -1;// 沒有發現相關的key
}
}
break;
case NO_VALUE:
errPos = tmpString;
return -2; //不能有參數的選項后有參數
default:
errPos = tmpString;
return -1;//參數錯誤
}//switch end
}
}
}//while end
return 0;
}