diff options
Diffstat (limited to 'tdecore/kmacroexpander.cpp')
| -rw-r--r-- | tdecore/kmacroexpander.cpp | 529 |
1 files changed, 529 insertions, 0 deletions
diff --git a/tdecore/kmacroexpander.cpp b/tdecore/kmacroexpander.cpp new file mode 100644 index 000000000..c7717029c --- /dev/null +++ b/tdecore/kmacroexpander.cpp @@ -0,0 +1,529 @@ +/* + This file is part of the KDE libraries + + Copyright (c) 2002-2003 Oswald Buddenhagen <ossi@kde.org> + Copyright (c) 2003 Waldo Bastian <bastian@kde.org> + + This library is free software; you can redistribute it and/or + modify it under the terms of the GNU Library General Public + License as published by the Free Software Foundation; either + version 2 of the License, or (at your option) any later version. + + This library is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + Library General Public License for more details. + + You should have received a copy of the GNU Library General Public License + along with this library; see the file COPYING.LIB. If not, write to + the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, + Boston, MA 02110-1301, USA. +*/ + +#include <kmacroexpander.h> + +#include <tqvaluestack.h> +#include <tqregexp.h> + +KMacroExpanderBase::KMacroExpanderBase( TQChar c ) +{ + escapechar = c; +} + +KMacroExpanderBase::~KMacroExpanderBase() +{ +} + +void +KMacroExpanderBase::setEscapeChar( TQChar c ) +{ + escapechar = c; +} + +TQChar +KMacroExpanderBase::escapeChar() const +{ + return escapechar; +} + +void KMacroExpanderBase::expandMacros( TQString &str ) +{ + uint pos; + int len; + TQChar ec( escapechar ); + TQStringList rst; + TQString rsts; + + for (pos = 0; pos < str.length(); ) { + if (ec != (QChar)0) { + if (str.tqunicode()[pos] != ec) + goto nohit; + if (!(len = expandEscapedMacro( str, pos, rst ))) + goto nohit; + } else { + if (!(len = expandPlainMacro( str, pos, rst ))) + goto nohit; + } + if (len < 0) { + pos -= len; + continue; + } + rsts = rst.join( " " ); + rst.clear(); + str.replace( pos, len, rsts ); + pos += rsts.length(); + continue; + nohit: + pos++; + } +} + + +namespace KMacroExpander { + + /** @intern Quoting state of the expander code. Not available publicly. */ + enum Quoting { noquote, singlequote, doublequote, dollarquote, + paren, subst, group, math }; + typedef struct { + Quoting current; + bool dquote; + } State; + typedef struct { + TQString str; + uint pos; + } Save; + +} + +using namespace KMacroExpander; + +bool KMacroExpanderBase::expandMacrosShellQuote( TQString &str, uint &pos ) +{ + int len; + uint pos2; + TQChar ec( escapechar ); + State state = { noquote, false }; + TQValueStack<State> sstack; + TQValueStack<Save> ostack; + TQStringList rst; + TQString rsts; + + while (pos < str.length()) { + TQChar cc( str.tqunicode()[pos] ); + if (ec != (QChar)0) { + if (cc != ec) + goto nohit; + if (!(len = expandEscapedMacro( str, pos, rst ))) + goto nohit; + } else { + if (!(len = expandPlainMacro( str, pos, rst ))) + goto nohit; + } + if (len < 0) { + pos -= len; + continue; + } + if (state.dquote) { + rsts = rst.join( " " ); + rsts.replace( TQRegExp("([$`\"\\\\])"), "\\\\1" ); + } else if (state.current == dollarquote) { + rsts = rst.join( " " ); + rsts.replace( TQRegExp("(['\\\\])"), "\\\\1" ); + } else if (state.current == singlequote) { + rsts = rst.join( " " ); + rsts.replace( '\'', "'\\''"); + } else { + if (rst.isEmpty()) { + str.remove( pos, len ); + continue; + } else { + rsts = "'"; +#if 0 // this could pay off if join() would be cleverer and the strings were long + for (TQStringList::Iterator it = rst.begin(); it != rst.end(); ++it) + (*it).replace( '\'', "'\\''" ); + rsts += rst.join( "' '" ); +#else + for (TQStringList::ConstIterator it = rst.begin(); it != rst.end(); ++it) { + if (it != rst.begin()) + rsts += "' '"; + TQString trsts( *it ); + trsts.replace( '\'', "'\\''" ); + rsts += trsts; + } +#endif + rsts += "'"; + } + } + rst.clear(); + str.replace( pos, len, rsts ); + pos += rsts.length(); + continue; + nohit: + if (state.current == singlequote) { + if (cc == (QChar)'\'') + state = sstack.pop(); + } else if (cc == (QChar)'\\') { + // always swallow the char -> prevent anomalies due to expansion + pos += 2; + continue; + } else if (state.current == dollarquote) { + if (cc == (QChar)'\'') + state = sstack.pop(); + } else if (cc == (QChar)'$') { + cc = str[++pos]; + if (cc == (QChar)'(') { + sstack.push( state ); + if (str[pos + 1] == (QChar)'(') { + Save sav = { str, pos + 2 }; + ostack.push( sav ); + state.current = math; + pos += 2; + continue; + } else { + state.current = paren; + state.dquote = false; + } + } else if (cc == (QChar)'{') { + sstack.push( state ); + state.current = subst; + } else if (!state.dquote) { + if (cc == (QChar)'\'') { + sstack.push( state ); + state.current = dollarquote; + } else if (cc == (QChar)'"') { + sstack.push( state ); + state.current = doublequote; + state.dquote = true; + } + } + // always swallow the char -> prevent anomalies due to expansion + } else if (cc == (QChar)'`') { + str.replace( pos, 1, "$( " ); // add space -> avoid creating $(( + pos2 = pos += 3; + for (;;) { + if (pos2 >= str.length()) { + pos = pos2; + return false; + } + cc = str.tqunicode()[pos2]; + if (cc == (QChar)'`') + break; + if (cc == (QChar)'\\') { + cc = str[++pos2]; + if (cc == (QChar)'$' || cc == (QChar)'`' || cc == (QChar)'\\' || + (cc == (QChar)'"' && state.dquote)) + { + str.remove( pos2 - 1, 1 ); + continue; + } + } + pos2++; + } + str[pos2] = ')'; + sstack.push( state ); + state.current = paren; + state.dquote = false; + continue; + } else if (state.current == doublequote) { + if (cc == (QChar)'"') + state = sstack.pop(); + } else if (cc == (QChar)'\'') { + if (!state.dquote) { + sstack.push( state ); + state.current = singlequote; + } + } else if (cc == (QChar)'"') { + if (!state.dquote) { + sstack.push( state ); + state.current = doublequote; + state.dquote = true; + } + } else if (state.current == subst) { + if (cc == (QChar)'}') + state = sstack.pop(); + } else if (cc == (QChar)')') { + if (state.current == math) { + if (str[pos + 1] == (QChar)')') { + state = sstack.pop(); + pos += 2; + } else { + // false hit: the $(( was a $( ( in fact + // ash does not care, but bash does + pos = ostack.top().pos; + str = ostack.top().str; + ostack.pop(); + state.current = paren; + state.dquote = false; + sstack.push( state ); + } + continue; + } else if (state.current == paren) + state = sstack.pop(); + else + break; + } else if (cc == (QChar)'}') { + if (state.current == KMacroExpander::group) + state = sstack.pop(); + else + break; + } else if (cc == (QChar)'(') { + sstack.push( state ); + state.current = paren; + } else if (cc == (QChar)'{') { + sstack.push( state ); + state.current = KMacroExpander::group; + } + pos++; + } + return sstack.empty(); +} + +bool KMacroExpanderBase::expandMacrosShellQuote( TQString &str ) +{ + uint pos = 0; + return expandMacrosShellQuote( str, pos ) && pos == str.length(); +} + +int KMacroExpanderBase::expandPlainMacro( const TQString &, uint, TQStringList & ) +{ qFatal( "KMacroExpanderBase::expandPlainMacro called!" ); return 0; } + +int KMacroExpanderBase::expandEscapedMacro( const TQString &, uint, TQStringList & ) +{ qFatal( "KMacroExpanderBase::expandEscapedMacro called!" ); return 0; } + + +////////////////////////////////////////////////// + +template<class KT,class VT> +class KMacroMapExpander : public KMacroExpanderBase { + +public: + KMacroMapExpander( const TQMap<KT,VT> &map, TQChar c = '%' ) : + KMacroExpanderBase( c ), macromap( map ) {} + +protected: + virtual int expandPlainMacro( const TQString &str, uint pos, TQStringList &ret ); + virtual int expandEscapedMacro( const TQString &str, uint pos, TQStringList &ret ); + +private: + TQMap<KT,VT> macromap; +}; + +static TQStringList &operator+=( TQStringList &s, const TQString &n) { s << n; return s; } + +//////// + +static bool +isIdentifier( uint c ) +{ + return c == '_' || (c >= 'A' && c <= 'Z') || (c >= 'a' && c <= 'z') || (c >= '0' && c <= '9'); +} + +//////// + +template<class VT> +class KMacroMapExpander<TQChar,VT> : public KMacroExpanderBase { + +public: + KMacroMapExpander( const TQMap<TQChar,VT> &map, TQChar c = '%' ) : + KMacroExpanderBase( c ), macromap( map ) {} + +protected: + virtual int expandPlainMacro( const TQString &str, uint pos, TQStringList &ret ); + virtual int expandEscapedMacro( const TQString &str, uint pos, TQStringList &ret ); + +private: + TQMap<TQChar,VT> macromap; +}; + +template<class VT> +int +KMacroMapExpander<TQChar,VT>::expandPlainMacro( const TQString &str, uint pos, TQStringList &ret ) +{ + TQMapConstIterator<TQChar,VT> it = macromap.find(str[pos]); + if (it != macromap.end()) { + ret += it.data(); + return 1; + } + return 0; +} + +template<class VT> +int +KMacroMapExpander<TQChar,VT>::expandEscapedMacro( const TQString &str, uint pos, TQStringList &ret ) +{ + if (str[pos + 1] == escapeChar()) { + ret += TQString( escapeChar() ); + return 2; + } + TQMapConstIterator<TQChar,VT> it = macromap.find(str[pos+1]); + if (it != macromap.end()) { + ret += it.data(); + return 2; + } + + return 0; +} + +template<class VT> +class KMacroMapExpander<TQString,VT> : public KMacroExpanderBase { + +public: + KMacroMapExpander( const TQMap<TQString,VT> &map, TQChar c = '%' ) : + KMacroExpanderBase( c ), macromap( map ) {} + +protected: + virtual int expandPlainMacro( const TQString &str, uint pos, TQStringList &ret ); + virtual int expandEscapedMacro( const TQString &str, uint pos, TQStringList &ret ); + +private: + TQMap<TQString,VT> macromap; +}; + +template<class VT> +int +KMacroMapExpander<TQString,VT>::expandPlainMacro( const TQString &str, uint pos, TQStringList &ret ) +{ + if (isIdentifier( str[pos - 1].tqunicode() )) + return 0; + uint sl; + for (sl = 0; isIdentifier( str[pos + sl].tqunicode() ); sl++); + if (!sl) + return 0; + TQMapConstIterator<TQString,VT> it = + macromap.find( TQConstString( str.tqunicode() + pos, sl ).string() ); + if (it != macromap.end()) { + ret += it.data(); + return sl; + } + return 0; +} + +template<class VT> +int +KMacroMapExpander<TQString,VT>::expandEscapedMacro( const TQString &str, uint pos, TQStringList &ret ) +{ + if (str[pos + 1] == escapeChar()) { + ret += TQString( escapeChar() ); + return 2; + } + uint sl, rsl, rpos; + if (str[pos + 1] == (QChar)'{') { + rpos = pos + 2; + for (sl = 0; str[rpos + sl] != (QChar)'}'; sl++) + if (rpos + sl >= str.length()) + return 0; + rsl = sl + 3; + } else { + rpos = pos + 1; + for (sl = 0; isIdentifier( str[rpos + sl].tqunicode() ); sl++); + rsl = sl + 1; + } + if (!sl) + return 0; + TQMapConstIterator<TQString,VT> it = + macromap.find( TQConstString( str.tqunicode() + rpos, sl ).string() ); + if (it != macromap.end()) { + ret += it.data(); + return rsl; + } + return 0; +} + +//////////// + +int +KCharMacroExpander::expandPlainMacro( const TQString &str, uint pos, TQStringList &ret ) +{ + if (expandMacro( str[pos], ret )) + return 1; + return 0; +} + +int +KCharMacroExpander::expandEscapedMacro( const TQString &str, uint pos, TQStringList &ret ) +{ + if (str[pos + 1] == escapeChar()) { + ret += TQString( escapeChar() ); + return 2; + } + if (expandMacro( str[pos+1], ret )) + return 2; + return 0; +} + +int +KWordMacroExpander::expandPlainMacro( const TQString &str, uint pos, TQStringList &ret ) +{ + if (isIdentifier( str[pos - 1].tqunicode() )) + return 0; + uint sl; + for (sl = 0; isIdentifier( str[pos + sl].tqunicode() ); sl++); + if (!sl) + return 0; + if (expandMacro( TQConstString( str.tqunicode() + pos, sl ).string(), ret )) + return sl; + return 0; +} + +int +KWordMacroExpander::expandEscapedMacro( const TQString &str, uint pos, TQStringList &ret ) +{ + if (str[pos + 1] == escapeChar()) { + ret += TQString( escapeChar() ); + return 2; + } + uint sl, rsl, rpos; + if (str[pos + 1] == (QChar)'{') { + rpos = pos + 2; + for (sl = 0; str[rpos + sl] != (QChar)'}'; sl++) + if (rpos + sl >= str.length()) + return 0; + rsl = sl + 3; + } else { + rpos = pos + 1; + for (sl = 0; isIdentifier( str[rpos + sl].tqunicode() ); sl++); + rsl = sl + 1; + } + if (!sl) + return 0; + if (expandMacro( TQConstString( str.tqunicode() + rpos, sl ).string(), ret )) + return rsl; + return 0; +} + +//////////// + +template<class KT,class VT> +inline QString +TexpandMacros( const TQString &ostr, const TQMap<KT,VT> &map, TQChar c ) +{ + TQString str( ostr ); + KMacroMapExpander<KT,VT> kmx( map, c ); + kmx.expandMacros( str ); + return str; +} + +template<class KT,class VT> +inline QString +TexpandMacrosShellQuote( const TQString &ostr, const TQMap<KT,VT> &map, TQChar c ) +{ + TQString str( ostr ); + KMacroMapExpander<KT,VT> kmx( map, c ); + if (!kmx.expandMacrosShellQuote( str )) + return TQString(); + return str; +} + +// public API +namespace KMacroExpander { + + TQString expandMacros( const TQString &ostr, const TQMap<TQChar,TQString> &map, TQChar c ) { return TexpandMacros( ostr, map, c ); } + TQString expandMacrosShellQuote( const TQString &ostr, const TQMap<TQChar,TQString> &map, TQChar c ) { return TexpandMacrosShellQuote( ostr, map, c ); } + TQString expandMacros( const TQString &ostr, const TQMap<TQString,TQString> &map, TQChar c ) { return TexpandMacros( ostr, map, c ); } + TQString expandMacrosShellQuote( const TQString &ostr, const TQMap<TQString,TQString> &map, TQChar c ) { return TexpandMacrosShellQuote( ostr, map, c ); } + TQString expandMacros( const TQString &ostr, const TQMap<TQChar,TQStringList> &map, TQChar c ) { return TexpandMacros( ostr, map, c ); } + TQString expandMacrosShellQuote( const TQString &ostr, const TQMap<TQChar,TQStringList> &map, TQChar c ) { return TexpandMacrosShellQuote( ostr, map, c ); } + TQString expandMacros( const TQString &ostr, const TQMap<TQString,TQStringList> &map, TQChar c ) { return TexpandMacros( ostr, map, c ); } + TQString expandMacrosShellQuote( const TQString &ostr, const TQMap<TQString,TQStringList> &map, TQChar c ) { return TexpandMacrosShellQuote( ostr, map, c ); } + +} // namespace |