[fsresizer] Port to NamedSuffix template

This commit is contained in:
Adriaan de Groot 2019-01-11 19:49:39 +01:00
parent e28aeebb48
commit ef33aa01f3
3 changed files with 43 additions and 63 deletions

View File

@ -39,46 +39,32 @@
#include "modules/partition/core/PartitionIterator.h" #include "modules/partition/core/PartitionIterator.h"
ResizeFSJob::RelativeSize::RelativeSize()
: m_value( 0 ) static const NamedEnumTable<ResizeFSJob::RelativeUnit>&
, m_unit( None ) unitSuffixes()
{ {
using Unit = ResizeFSJob::RelativeUnit;
static const NamedEnumTable<Unit> names{
{ "%", Unit::Percent },
{ "MiB", Unit::Absolute }
};
return names;
} }
template<int N>
void matchUnitSuffix(
const QString& s,
const char ( &suffix )[N],
ResizeFSJob::RelativeSize::Unit matchedUnit,
int& value,
ResizeFSJob::RelativeSize::Unit& unit
)
{
if ( s.endsWith( suffix ) )
{
value = s.left( s.length() - N + 1 ).toInt();
unit = matchedUnit;
}
}
ResizeFSJob::RelativeSize::RelativeSize( const QString& s ) ResizeFSJob::RelativeSize::RelativeSize( const QString& s )
: m_value( 0 ) : NamedSuffix( unitSuffixes(), s )
, m_unit( None )
{ {
matchUnitSuffix( s, "%", Percent, m_value, m_unit ); if ( ( unit() == RelativeUnit::Percent ) && ( value() > 100 ) )
matchUnitSuffix( s, "MiB", Absolute, m_value, m_unit );
if ( ( unit() == Percent ) && ( value() > 100 ) )
{ {
cDebug() << "Percent value" << value() << "is not valid."; cDebug() << "Percent value" << value() << "is not valid.";
m_value = 0; m_value = 0;
m_unit = None; m_unit = RelativeUnit::None;
} }
if ( !m_value ) if ( !m_value )
m_unit = None; m_unit = RelativeUnit::None;
} }
qint64 qint64
@ -91,11 +77,11 @@ ResizeFSJob::RelativeSize::apply( qint64 totalSectors, qint64 sectorSize )
switch ( m_unit ) switch ( m_unit )
{ {
case None: case unit_t::None:
return -1; return -1;
case Absolute: case unit_t::Absolute:
return CalamaresUtils::MiBtoBytes( value() ) / sectorSize; return CalamaresUtils::MiBtoBytes( value() ) / sectorSize;
case Percent: case unit_t::Percent:
if ( value() == 100 ) if ( value() == 100 )
return totalSectors; // Common-case, avoid futzing around return totalSectors; // Common-case, avoid futzing around
else else

View File

@ -24,7 +24,8 @@
#include <CppJob.h> #include <CppJob.h>
#include <utils/PluginFactory.h> #include "utils/NamedSuffix.h"
#include "utils/PluginFactory.h"
#include <PluginDllMacro.h> #include <PluginDllMacro.h>
@ -37,31 +38,28 @@ class PLUGINDLLEXPORT ResizeFSJob : public Calamares::CppJob
Q_OBJECT Q_OBJECT
public: public:
enum class RelativeUnit
{
None,
Percent,
Absolute
};
/** @brief Size expressions /** @brief Size expressions
* *
* Sizes can be specified in MiB or percent (of the device they * Sizes can be specified in MiB or percent (of the device they
* are on). This class handles parsing of such strings from the * are on). This class handles parsing of such strings from the
* config file. * config file.
*/ */
class RelativeSize class RelativeSize : public NamedSuffix<RelativeUnit, RelativeUnit::None>
{ {
public: public:
RelativeSize(); RelativeSize() : NamedSuffix() { };
RelativeSize( const QString& ); RelativeSize( const QString& );
enum Unit
{
None,
Percent,
Absolute
};
int value() const { return m_value; }
Unit unit() const { return m_unit; }
bool isValid() const bool isValid() const
{ {
return ( unit() != None ) && ( value() > 0 ); return ( unit() != RelativeUnit::None ) && ( value() > 0 );
} }
/** @brief Apply this size to the number of sectors @p totalSectors . /** @brief Apply this size to the number of sectors @p totalSectors .
@ -80,10 +78,6 @@ public:
* Equivalent to apply( d->totalLogical(), d->logicalSize() ) * Equivalent to apply( d->totalLogical(), d->logicalSize() )
*/ */
qint64 apply( Device* d ); qint64 apply( Device* d );
private:
int m_value;
Unit m_unit;
} ; } ;
explicit ResizeFSJob( QObject* parent = nullptr ); explicit ResizeFSJob( QObject* parent = nullptr );

View File

@ -54,13 +54,13 @@ FSResizerTests::initTestCase()
void FSResizerTests::testConfigurationRobust() void FSResizerTests::testConfigurationRobust()
{ {
ResizeFSJob j; ResizeFSJob j;
// Empty config // Empty config
j.setConfigurationMap( QVariantMap() ); j.setConfigurationMap( QVariantMap() );
QVERIFY( j.m_fsname.isEmpty() ); QVERIFY( j.m_fsname.isEmpty() );
QVERIFY( j.m_devicename.isEmpty() ); QVERIFY( j.m_devicename.isEmpty() );
QCOMPARE( j.m_size.unit(), ResizeFSJob::RelativeSize::None ); QCOMPARE( j.m_size.unit(), ResizeFSJob::RelativeUnit::None );
QCOMPARE( j.m_atleast.unit(), ResizeFSJob::RelativeSize::None ); QCOMPARE( j.m_atleast.unit(), ResizeFSJob::RelativeUnit::None );
// Config is missing fs and dev, so it isn't valid // Config is missing fs and dev, so it isn't valid
YAML::Node doc0 = YAML::Load( R"(--- YAML::Node doc0 = YAML::Load( R"(---
@ -70,17 +70,17 @@ atleast: 600MiB
j.setConfigurationMap( CalamaresUtils::yamlMapToVariant( doc0 ).toMap() ); j.setConfigurationMap( CalamaresUtils::yamlMapToVariant( doc0 ).toMap() );
QVERIFY( j.m_fsname.isEmpty() ); QVERIFY( j.m_fsname.isEmpty() );
QVERIFY( j.m_devicename.isEmpty() ); QVERIFY( j.m_devicename.isEmpty() );
QCOMPARE( j.m_size.unit(), ResizeFSJob::RelativeSize::None ); QCOMPARE( j.m_size.unit(), ResizeFSJob::RelativeUnit::None );
QCOMPARE( j.m_atleast.unit(), ResizeFSJob::RelativeSize::None ); QCOMPARE( j.m_atleast.unit(), ResizeFSJob::RelativeUnit::None );
QCOMPARE( j.m_size.value(), 0 ); QCOMPARE( j.m_size.value(), 0 );
QCOMPARE( j.m_atleast.value(), 0 ); QCOMPARE( j.m_atleast.value(), 0 );
} }
void FSResizerTests::testConfigurationValues() void FSResizerTests::testConfigurationValues()
{ {
ResizeFSJob j; ResizeFSJob j;
// Check both // Check both
YAML::Node doc0 = YAML::Load( R"(--- YAML::Node doc0 = YAML::Load( R"(---
fs: / fs: /
size: 100% size: 100%
@ -89,8 +89,8 @@ atleast: 600MiB
j.setConfigurationMap( CalamaresUtils::yamlMapToVariant( doc0 ).toMap() ); j.setConfigurationMap( CalamaresUtils::yamlMapToVariant( doc0 ).toMap() );
QVERIFY( !j.m_fsname.isEmpty() ); QVERIFY( !j.m_fsname.isEmpty() );
QVERIFY( j.m_devicename.isEmpty() ); QVERIFY( j.m_devicename.isEmpty() );
QCOMPARE( j.m_size.unit(), ResizeFSJob::RelativeSize::Percent ); QCOMPARE( j.m_size.unit(), ResizeFSJob::RelativeUnit::Percent );
QCOMPARE( j.m_atleast.unit(), ResizeFSJob::RelativeSize::Absolute ); QCOMPARE( j.m_atleast.unit(), ResizeFSJob::RelativeUnit::Absolute );
QCOMPARE( j.m_size.value(), 100 ); QCOMPARE( j.m_size.value(), 100 );
QCOMPARE( j.m_atleast.value(), 600 ); QCOMPARE( j.m_atleast.value(), 600 );
@ -104,8 +104,8 @@ atleast: 127 %
j.setConfigurationMap( CalamaresUtils::yamlMapToVariant( doc0 ).toMap() ); j.setConfigurationMap( CalamaresUtils::yamlMapToVariant( doc0 ).toMap() );
QVERIFY( !j.m_fsname.isEmpty() ); QVERIFY( !j.m_fsname.isEmpty() );
QVERIFY( !j.m_devicename.isEmpty() ); QVERIFY( !j.m_devicename.isEmpty() );
QCOMPARE( j.m_size.unit(), ResizeFSJob::RelativeSize::Absolute ); QCOMPARE( j.m_size.unit(), ResizeFSJob::RelativeUnit::Absolute );
QCOMPARE( j.m_atleast.unit(), ResizeFSJob::RelativeSize::Percent ); QCOMPARE( j.m_atleast.unit(), ResizeFSJob::RelativeUnit::Percent );
QCOMPARE( j.m_size.value(), 72 ); QCOMPARE( j.m_size.value(), 72 );
QCOMPARE( j.m_atleast.value(), 127 ); QCOMPARE( j.m_atleast.value(), 127 );
@ -119,8 +119,8 @@ size: 71MiB
j.setConfigurationMap( CalamaresUtils::yamlMapToVariant( doc0 ).toMap() ); j.setConfigurationMap( CalamaresUtils::yamlMapToVariant( doc0 ).toMap() );
QVERIFY( !j.m_fsname.isEmpty() ); QVERIFY( !j.m_fsname.isEmpty() );
QVERIFY( j.m_devicename.isEmpty() ); QVERIFY( j.m_devicename.isEmpty() );
QCOMPARE( j.m_size.unit(), ResizeFSJob::RelativeSize::Absolute ); QCOMPARE( j.m_size.unit(), ResizeFSJob::RelativeUnit::Absolute );
QCOMPARE( j.m_atleast.unit(), ResizeFSJob::RelativeSize::None ); QCOMPARE( j.m_atleast.unit(), ResizeFSJob::RelativeUnit::None );
QCOMPARE( j.m_size.value(), 71 ); QCOMPARE( j.m_size.value(), 71 );
QCOMPARE( j.m_atleast.value(), 0 ); QCOMPARE( j.m_atleast.value(), 0 );
} }