[libcalamares] Move CommandList into libcalamares

- Move CommandList so it can be used from more modules than
   just ShellProcess
 - Allow a CommandList to run itself. This centralizes
   code for executing one or more commands and simplifies
   the ShellProcess module.

Various small cleanups:
 - mention instance id in log message
 - code formatting / style
This commit is contained in:
Adriaan de Groot 2018-01-15 05:59:44 -05:00
parent 60f4dd7b3b
commit 913690650f
8 changed files with 142 additions and 123 deletions

View File

@ -22,6 +22,7 @@ set( libSources
set( utilsSources set( utilsSources
utils/CalamaresUtils.cpp utils/CalamaresUtils.cpp
utils/CalamaresUtilsSystem.cpp utils/CalamaresUtilsSystem.cpp
utils/CommandList.cpp
utils/Logger.cpp utils/Logger.cpp
utils/PluginFactory.cpp utils/PluginFactory.cpp
utils/Retranslator.cpp utils/Retranslator.cpp

View File

@ -0,0 +1,120 @@
/* === This file is part of Calamares - <https://github.com/calamares> ===
*
* 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 "CommandList.h"
#include "GlobalStorage.h"
#include "JobQueue.h"
#include "utils/CalamaresUtilsSystem.h"
#include "utils/Logger.h"
#include <QVariantList>
static QStringList get_variant_stringlist( const QVariantList& l )
{
QStringList retl;
unsigned int c = 0;
for ( const auto& v : l )
{
if ( v.type() == QVariant::String )
retl.append( v.toString() );
else
cDebug() << "WARNING Bad CommandList element" << c << v.type() << v;
++c;
}
return retl;
}
namespace CalamaresUtils
{
CommandList::CommandList( bool doChroot )
: m_doChroot( doChroot )
{
}
CommandList::CommandList::CommandList( const QVariant& v, bool doChroot )
: CommandList( doChroot )
{
if ( v.type() == QVariant::List )
{
const auto v_list = v.toList();
if ( v_list.count() )
append( get_variant_stringlist( v_list ) );
else
cDebug() << "WARNING: Empty CommandList";
}
else if ( v.type() == QVariant::String )
append( v.toString() );
else
cDebug() << "WARNING: CommandList does not understand variant" << v.type();
}
CommandList::~CommandList()
{
}
Calamares::JobResult CommandList::run( const QObject* parent )
{
System::RunLocation location = m_doChroot ? System::RunLocation::RunInTarget : System::RunLocation::RunInHost;
/* Figure out the replacement for @@ROOT@@ */
QString root = QStringLiteral( "/" );
Calamares::GlobalStorage* gs = Calamares::JobQueue::instance()->globalStorage();
if ( location == System::RunLocation::RunInTarget )
{
if ( !gs || !gs->contains( "rootMountPoint" ) )
{
cDebug() << "ERROR: No rootMountPoint defined.";
return Calamares::JobResult::error( parent->tr( "Could not run command." ),
parent->tr( "No rootMountPoint is defined, so command cannot be run in the target environment." ) );
}
root = gs->value( "rootMountPoint" ).toString();
}
for ( CommandList::const_iterator i = cbegin(); i != cend(); ++i )
{
QString processed_cmd = *i;
processed_cmd.replace( "@@ROOT@@", root ); // FIXME?
bool suppress_result = false;
if ( processed_cmd.startsWith( '-' ) )
{
suppress_result = true;
processed_cmd.remove( 0, 1 ); // Drop the - // FIXME?
}
QStringList shell_cmd { "/bin/sh", "-c" };
shell_cmd << processed_cmd;
ProcessResult r = System::runCommand(
location, shell_cmd, QString(), QString(), 10 );
if ( r.getExitCode() != 0 )
{
if ( suppress_result )
cDebug() << "Error code" << r.getExitCode() << "ignored by ShellProcess configuration.";
else
return r.explainProcess( parent, processed_cmd, 10 );
}
}
return Calamares::JobResult::ok();
}
} // namespace

View File

@ -19,9 +19,14 @@
#ifndef COMMANDLIST_H #ifndef COMMANDLIST_H
#define COMMANDLIST_H #define COMMANDLIST_H
#include "Job.h"
#include <QStringList> #include <QStringList>
#include <QVariant> #include <QVariant>
namespace CalamaresUtils
{
class CommandList : protected QStringList class CommandList : protected QStringList
{ {
public: public:
@ -29,7 +34,12 @@ public:
CommandList( const QVariant& v, bool doChroot = true ); CommandList( const QVariant& v, bool doChroot = true );
~CommandList(); ~CommandList();
bool doChroot() const { return m_doChroot; } bool doChroot() const
{
return m_doChroot;
}
Calamares::JobResult run( const QObject* parent );
using QStringList::isEmpty; using QStringList::isEmpty;
using QStringList::count; using QStringList::count;
@ -41,4 +51,5 @@ private:
bool m_doChroot; bool m_doChroot;
} ; } ;
} // namespace
#endif // COMMANDLIST_H #endif // COMMANDLIST_H

View File

@ -2,7 +2,6 @@ calamares_add_plugin( shellprocess
TYPE job TYPE job
EXPORT_MACRO PLUGINDLLEXPORT_PRO EXPORT_MACRO PLUGINDLLEXPORT_PRO
SOURCES SOURCES
CommandList.cpp
ShellProcessJob.cpp ShellProcessJob.cpp
LINK_PRIVATE_LIBRARIES LINK_PRIVATE_LIBRARIES
calamares calamares
@ -16,7 +15,6 @@ if( ECM_FOUND )
ecm_add_test( ecm_add_test(
Tests.cpp Tests.cpp
CommandList.cpp
TEST_NAME TEST_NAME
shellprocesstest shellprocesstest
LINK_LIBRARIES LINK_LIBRARIES

View File

@ -1,68 +0,0 @@
/* === This file is part of Calamares - <https://github.com/calamares> ===
*
* 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 "CommandList.h"
#include "utils/Logger.h"
#include <QVariantList>
static QStringList get_variant_stringlist(const QVariantList& l)
{
QStringList retl;
unsigned int c = 0;
for ( const auto& v : l )
{
if ( v.type() == QVariant::String )
retl.append( v.toString() );
else
cDebug() << "WARNING Bad CommandList element" << c << v.type() << v;
++c;
}
return retl;
}
CommandList::CommandList( bool doChroot )
: m_doChroot( doChroot )
{
}
CommandList::CommandList::CommandList( const QVariant& v, bool doChroot )
: CommandList( doChroot )
{
if ( v.type() == QVariant::List )
{
const auto v_list = v.toList();
if ( v_list.count() )
{
append( get_variant_stringlist( v_list ) );
}
else
cDebug() << "WARNING: Empty CommandList";
}
else if ( v.type() == QVariant::String )
{
append( v.toString() );
}
else
cDebug() << "WARNING: CommandList does not understand variant" << v.type();
}
CommandList::~CommandList()
{
}

View File

@ -18,8 +18,6 @@
#include "ShellProcessJob.h" #include "ShellProcessJob.h"
#include "CommandList.h"
#include <QProcess> #include <QProcess>
#include <QDateTime> #include <QDateTime>
#include <QThread> #include <QThread>
@ -30,10 +28,9 @@
#include "utils/CalamaresUtils.h" #include "utils/CalamaresUtils.h"
#include "utils/CalamaresUtilsSystem.h" #include "utils/CalamaresUtilsSystem.h"
#include "utils/CommandList.h"
#include "utils/Logger.h" #include "utils/Logger.h"
class CommandList;
ShellProcessJob::ShellProcessJob( QObject* parent ) ShellProcessJob::ShellProcessJob( QObject* parent )
: Calamares::CppJob( parent ) : Calamares::CppJob( parent )
, m_commands( nullptr ) , m_commands( nullptr )
@ -59,56 +56,14 @@ ShellProcessJob::prettyName() const
Calamares::JobResult Calamares::JobResult
ShellProcessJob::exec() ShellProcessJob::exec()
{ {
using CalamaresUtils::System;
System::RunLocation location = m_dontChroot ? System::RunLocation::RunInHost : System::RunLocation::RunInTarget;
if ( ! m_commands || m_commands->isEmpty() ) if ( ! m_commands || m_commands->isEmpty() )
{ {
cDebug() << "WARNING: No commands to execute"; cDebug() << "WARNING: No commands to execute" << moduleInstanceKey();
return Calamares::JobResult::ok(); return Calamares::JobResult::ok();
} }
/* Figure out the replacement for @@ROOT@@ */ return m_commands->run( this );
QString root = QStringLiteral( "/" );
Calamares::GlobalStorage* gs = Calamares::JobQueue::instance()->globalStorage();
if ( location == System::RunLocation::RunInTarget )
{
if ( !gs || !gs->contains( "rootMountPoint" ) )
{
cDebug() << "ERROR: No rootMountPoint defined.";
return Calamares::JobResult::error( tr( "Could not run command." ),
tr( "No rootMountPoint is defined, so command cannot be run in the target environment." ) );
}
root = gs->value( "rootMountPoint" ).toString();
}
for ( CommandList::const_iterator i = m_commands->cbegin(); i != m_commands->cend(); ++i )
{
QString processed_cmd = *i;
processed_cmd.replace( "@@ROOT@@", root ); // FIXME?
bool suppress_result = false;
if ( processed_cmd.startsWith( '-' ) )
{
suppress_result = true;
processed_cmd.remove( 0, 1 ); // Drop the - // FIXME?
}
QStringList shell_cmd { "/bin/sh", "-c" };
shell_cmd << processed_cmd;
CalamaresUtils::ProcessResult r = System::runCommand(
location, shell_cmd, QString(), QString(), 10 );
if ( r.getExitCode() != 0 )
{
if ( suppress_result )
cDebug() << "Error code" << r.getExitCode() << "ignored by ShellProcess configuration.";
else
return r.explainProcess( this, processed_cmd, 10 );
}
}
return Calamares::JobResult::ok();
} }
@ -119,7 +74,7 @@ ShellProcessJob::setConfigurationMap( const QVariantMap& configurationMap )
if ( configurationMap.contains( "script" ) ) if ( configurationMap.contains( "script" ) )
{ {
m_commands = new CommandList( configurationMap.value( "script" ) ); m_commands = new CalamaresUtils::CommandList( configurationMap.value( "script" ), !m_dontChroot );
if ( m_commands->isEmpty() ) if ( m_commands->isEmpty() )
cDebug() << "ShellProcessJob: \"script\" contains no commands for" << moduleInstanceKey(); cDebug() << "ShellProcessJob: \"script\" contains no commands for" << moduleInstanceKey();
} }

View File

@ -24,11 +24,11 @@
#include <CppJob.h> #include <CppJob.h>
#include <utils/CommandList.h>
#include <utils/PluginFactory.h> #include <utils/PluginFactory.h>
#include <PluginDllMacro.h> #include <PluginDllMacro.h>
class CommandList;
class PLUGINDLLEXPORT ShellProcessJob : public Calamares::CppJob class PLUGINDLLEXPORT ShellProcessJob : public Calamares::CppJob
{ {
@ -45,7 +45,7 @@ public:
void setConfigurationMap( const QVariantMap& configurationMap ) override; void setConfigurationMap( const QVariantMap& configurationMap ) override;
private: private:
CommandList *m_commands; CalamaresUtils::CommandList* m_commands;
bool m_dontChroot; bool m_dontChroot;
}; };

View File

@ -17,8 +17,8 @@
*/ */
#include "Tests.h" #include "Tests.h"
#include "CommandList.h"
#include "utils/CommandList.h"
#include "utils/YamlUtils.h" #include "utils/YamlUtils.h"
#include <yaml-cpp/yaml.h> #include <yaml-cpp/yaml.h>
@ -30,6 +30,8 @@
QTEST_GUILESS_MAIN( ShellProcessTests ) QTEST_GUILESS_MAIN( ShellProcessTests )
using CommandList = CalamaresUtils::CommandList;
ShellProcessTests::ShellProcessTests() ShellProcessTests::ShellProcessTests()
{ {
} }