0f5e061c4a
- Some providers don't provide a single flat JSON object
(e.g. "{time_zone: foo}") but a nested structure
(e.g. "{location: {time_zone: foo}}"), so allow dots
in the selector to do multi-level selection.
77 lines
2.2 KiB
C++
77 lines
2.2 KiB
C++
/* === This file is part of Calamares - <http://github.com/calamares> ===
|
|
*
|
|
* Copyright 2014-2016, Teo Mrnjavac <teo@kde.org>
|
|
* Copyright 2018, Adriaan de Groot <groot@kde.org>
|
|
*
|
|
* Calamares is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* Calamares 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 General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with Calamares. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include "GeoIPJSON.h"
|
|
|
|
#include "utils/CalamaresUtils.h"
|
|
#include "utils/Logger.h"
|
|
#include "utils/YamlUtils.h"
|
|
|
|
#include <QByteArray>
|
|
|
|
#include <yaml-cpp/yaml.h>
|
|
|
|
GeoIPJSON::GeoIPJSON(const QString& attribute)
|
|
: GeoIP( attribute.isEmpty() ? QLatin1String( "time_zone" ) : attribute )
|
|
{
|
|
}
|
|
|
|
static QString
|
|
selectMap( const QVariantMap& m, const QStringList& l, int index)
|
|
{
|
|
if ( index >= l.count() )
|
|
return QString();
|
|
|
|
QString attributeName = l[index];
|
|
if ( index == l.count() - 1 )
|
|
return CalamaresUtils::getString( m, attributeName );
|
|
else
|
|
{
|
|
bool success = false; // bogus
|
|
if ( m.contains( attributeName ) )
|
|
return selectMap( CalamaresUtils::getSubMap( m, attributeName, success ), l, index+1 );
|
|
return QString();
|
|
}
|
|
}
|
|
|
|
GeoIP::RegionZonePair
|
|
GeoIPJSON::processReply( const QByteArray& data )
|
|
{
|
|
try
|
|
{
|
|
YAML::Node doc = YAML::Load( data );
|
|
|
|
QVariant var = CalamaresUtils::yamlToVariant( doc );
|
|
if ( !var.isNull() &&
|
|
var.isValid() &&
|
|
var.type() == QVariant::Map )
|
|
{
|
|
return splitTZString( selectMap( var.toMap(), m_element.split('.'), 0 ) );
|
|
}
|
|
else
|
|
cWarning() << "Invalid YAML data for GeoIPJSON";
|
|
}
|
|
catch ( YAML::Exception& e )
|
|
{
|
|
CalamaresUtils::explainYamlException( e, data, "GeoIP data");
|
|
}
|
|
|
|
return qMakePair( QString(), QString() );
|
|
}
|