Branch data     Line data    Source code

       1                 :            : /***************************************************************************
       2                 :            :                                   qgshstoreutils.h
       3                 :            :                               ---------------------
       4                 :            :     begin                : Sept 2018
       5                 :            :     copyright            : (C) 2018 by Mathieu Pellerin
       6                 :            :     email                : nirvn dot asia at gmail dot com
       7                 :            :  ***************************************************************************
       8                 :            :  *                                                                         *
       9                 :            :  *   This program is free software; you can redistribute it and/or modify  *
      10                 :            :  *   it under the terms of the GNU General Public License as published by  *
      11                 :            :  *   the Free Software Foundation; either version 2 of the License, or     *
      12                 :            :  *   (at your option) any later version.                                   *
      13                 :            :  *                                                                         *
      14                 :            :  ***************************************************************************/
      15                 :            : 
      16                 :            : #include "qgshstoreutils.h"
      17                 :            : 
      18                 :            : #include <QRegularExpression>
      19                 :            : 
      20                 :          0 : QVariantMap QgsHstoreUtils::parse( const QString &string )
      21                 :            : {
      22                 :          0 :   QVariantMap map;
      23                 :          0 :   QList<QString> bits;
      24                 :          0 :   static const QList<QString > sSeps{ "=>", "," };
      25                 :            : 
      26                 :          0 :   int i = 0;
      27                 :          0 :   while ( i < string.length() )
      28                 :            :   {
      29                 :          0 :     while ( i < string.length() && string.at( i ).isSpace() )
      30                 :          0 :       ++i;
      31                 :          0 :     QString current = string.mid( i );
      32                 :          0 :     QString sep = sSeps.at( bits.length() );
      33                 :          0 :     if ( current.startsWith( '"' ) )
      34                 :            :     {
      35                 :          0 :       QRegularExpression re( QStringLiteral( "^\"((?:\\\\.|[^\"\\\\])*)\".*" ) );
      36                 :          0 :       QRegularExpressionMatch match = re.match( current );
      37                 :          0 :       bits << QString();
      38                 :          0 :       if ( match.hasMatch() )
      39                 :            :       {
      40                 :          0 :         bits[bits.length() - 1] = match.captured( 1 ).replace( QLatin1String( "\\\"" ), QLatin1String( "\"" ) ).replace( QLatin1String( "\\\\" ), QLatin1String( "\\" ) );
      41                 :          0 :         i += match.captured( 1 ).length() + 2;
      42                 :          0 :         while ( i < string.length() && string.at( i ).isSpace() )
      43                 :          0 :           ++i;
      44                 :            : 
      45                 :          0 :         if ( string.midRef( i ).startsWith( sep ) )
      46                 :            :         {
      47                 :          0 :           i += sep.length();
      48                 :          0 :         }
      49                 :          0 :         else if ( i < string.length() )
      50                 :            :         {
      51                 :            :           // hstore string format broken, end construction
      52                 :          0 :           i += current.length();
      53                 :          0 :         }
      54                 :          0 :       }
      55                 :            :       else
      56                 :            :       {
      57                 :            :         // hstore string format broken, end construction
      58                 :          0 :         i += current.length();
      59                 :          0 :         bits[bits.length() - 1] = current.trimmed();
      60                 :            :       }
      61                 :          0 :     }
      62                 :            :     else
      63                 :            :     {
      64                 :          0 :       int sepPos = current.indexOf( sep );
      65                 :          0 :       if ( sepPos < 0 )
      66                 :            :       {
      67                 :          0 :         i += current.length();
      68                 :          0 :         bits << current.trimmed();
      69                 :          0 :       }
      70                 :            :       else
      71                 :            :       {
      72                 :          0 :         i += sepPos + sep.length();
      73                 :          0 :         bits << current.left( sepPos ).trimmed();
      74                 :            :       }
      75                 :            :     }
      76                 :            : 
      77                 :          0 :     if ( bits.length() == 2 )
      78                 :            :     {
      79                 :          0 :       if ( !bits.at( 0 ).isEmpty() && !bits.at( 1 ).isEmpty() )
      80                 :          0 :         map[ bits.at( 0 ) ] = bits.at( 1 );
      81                 :          0 :       bits.clear();
      82                 :          0 :     }
      83                 :          0 :   }
      84                 :            : 
      85                 :          0 :   return map;
      86                 :          0 : }
      87                 :            : 
      88                 :          0 : QString QgsHstoreUtils::build( const QVariantMap &map )
      89                 :            : {
      90                 :          0 :   QStringList list;
      91                 :          0 :   for ( QVariantMap::const_iterator it = map.constBegin(); it != map.constEnd(); ++it )
      92                 :            :   {
      93                 :          0 :     QString key = it.key();
      94                 :          0 :     QString value = it.value().toString();
      95                 :          0 :     list << QString( "\"%1\"=>\"%2\"" ).arg( key.replace( "\\", "\\\\" ).replace( "\"", "\\\"" ),
      96                 :          0 :          value.replace( "\\", "\\\\" ).replace( "\"", "\\\"" ) );
      97                 :          0 :   }
      98                 :          0 :   return list.join( ',' );
      99                 :          0 : }