calamares/src/modules/partition/PartitionCoreModule.h

/* === This file is part of Calamares - <http://github.com/calamares> ===
 *
 *   Copyright 2014, Aurélien Gâteau <agateau@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/>.
 */

#ifndef PARTITIONCOREMODULE_H
#define PARTITIONCOREMODULE_H

#include <PartitionModel.h>
#include <Typedefs.h>

// CalaPM
#include <core/partitiontable.h>

// Qt
#include <QList>
#include <QObject>

class CreatePartitionJob;
class Device;
class DeviceModel;
class FileSystem;
class Partition;

class QStandardItemModel;

/**
 * Owns the Qt models and the PM devices
 */
class PartitionCoreModule : public QObject
{
    Q_OBJECT
public:
    enum
    {
        BootLoaderPathRole = Qt::UserRole + 1
    };

    PartitionCoreModule( QObject* parent = nullptr );
    ~PartitionCoreModule();

    DeviceModel* deviceModel() const;

    PartitionModel* partitionModelForDevice( Device* device ) const;

    QAbstractItemModel* bootLoaderModel() const;

    void createPartitionTable( Device* device, PartitionTable::TableType type );

    void createPartition( Device* device, Partition* partition );

    void deletePartition( Device* device, Partition* partition );

    void formatPartition( Device* device, Partition* partition );

    QList< Calamares::job_ptr > jobs() const;

    bool hasRootMountPoint() const
    {
        return m_hasRootMountPoint;
    }

    void refresh( Device* device );

Q_SIGNALS:
    void hasRootMountPointChanged( bool value );

private:
    /**
     * Owns the Device, PartitionModel and the jobs
     */
    struct DeviceInfo
    {
        DeviceInfo( Device* );
        ~DeviceInfo();
        QScopedPointer< Device > device;
        QScopedPointer< PartitionModel > partitionModel;
        QList< Calamares::job_ptr > jobs;

        void forgetChanges();
    };
    QList< DeviceInfo* > m_deviceInfos;

    DeviceModel* m_deviceModel;
    QStandardItemModel* m_bootLoaderModel;
    bool m_hasRootMountPoint = false;

    void listDevices();
    void updateHasRootMountPoint();

    void dumpQueue() const;

    DeviceInfo* infoForDevice( Device* ) const;

    void updateBootLoaderModel();

    Partition* findPartitionByMountPoint( const QString& mountPoint ) const;
};

#endif /* PARTITIONCOREMODULE_H */