/* Copyright (C) 2004 Bart
 * Copyright (C) 2008, 2009, 2010, 2011, 2012 Curtis Gedak
 *
 *  This program 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 2 of the License, or
 *  (at your option) any later version.
 *
 *  This program 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 this program; if not, see <http://www.gnu.org/licenses/>.
 */

#ifndef GPARTED_GPARTED_CORE_H
#define GPARTED_GPARTED_CORE_H

#include "BlockSpecial.h"
#include "Device.h"
#include "FileSystem.h"
#include "Operation.h"
#include "Partition.h"
#include "PartitionLUKS.h"
#include "PartitionVector.h"
#include "SupportedFileSystems.h"
#include "Utils.h"

#include <parted/parted.h>
#include <vector>
#include <fstream>
#include <glibmm/thread.h>
#include <glibmm/ustring.h>
#include <memory>


namespace GParted
{


class GParted_Core
{
friend class EraseFileSystemSignaturesTest;  // To allow unit testing to call private
                                             // method.

public:
	static Glib::Thread *mainthread;
	GParted_Core() ;

	static Glib::ustring get_version_and_config_string();
	static void find_supported_core();
	void find_supported_filesystems() ;
	void set_user_devices( const std::vector<Glib::ustring> & user_devices ) ;
	void set_devices( std::vector<Device> & devices ) ;
	void set_devices_thread( std::vector<Device> * pdevices );

	bool valid_partition(const Device& device, Partition& partition, Glib::ustring& error);
	bool apply_operation_to_disk( Operation * operation );

	bool set_disklabel( const Device & device, const Glib::ustring & disklabel );
	bool new_disklabel( const Glib::ustring & device_path, const Glib::ustring & disklabel,
	                    bool recreate_dmraid_devs = true );

	bool toggle_flag( const Partition & partition, const Glib::ustring & flag, bool state ) ;
	
	const std::vector<FS> & get_filesystems() const ;
	const FS& get_fs(FSType fstype) const;
	static std::vector<Glib::ustring> get_disklabeltypes() ;
	std::map<Glib::ustring, bool> get_available_flags( const Partition & partition ) ;
	Glib::ustring get_thread_status_message() ;

	static FileSystem * get_filesystem_object( FSType fstype );
	static bool supported_filesystem( FSType fstype );
	static FS_Limits get_filesystem_limits( FSType fstype, const Partition & partition );
	static bool filesystem_resize_disallowed( const Partition & partition ) ;
	static void insert_unallocated( const Glib::ustring & device_path,
	                                PartitionVector & partitions,
	                                Sector start,
	                                Sector end,
	                                Byte_Value sector_size,
	                                bool inside_extended );
	static Glib::ustring check_logical_esp_warning(PartitionType ptntype, bool esp_flag);
	static void compose_partition_flags(Partition& partition, const Glib::ustring& disktype);

private:
	//detectionstuff..
	void set_thread_status_message( Glib::ustring msg ) ;
	static Glib::ustring get_partition_path(const PedPartition *lp_partition);
	void set_device_from_disk( Device & device, const Glib::ustring & device_path );
	void set_device_serial_number( Device & device );
	void set_device_partitions( Device & device, PedDevice* lp_device, PedDisk* lp_disk ) ;
	void set_device_one_partition( Device & device, PedDevice * lp_device, FSType fstype,
	                               std::vector<Glib::ustring> & messages );
	void set_luks_partition( PartitionLUKS & partition );
	void set_partition_label_and_uuid( Partition & partition );
	static FSType detect_filesystem_in_encryption_mapping(const Glib::ustring& path,
	                                                      std::vector<Glib::ustring>& messages);
	static FSType detect_filesystem_internal(const Glib::ustring& path, Byte_Value sector_size);
	static FSType detect_filesystem(const PedDevice *lp_device, const PedPartition *lp_partition,
	                                std::vector<Glib::ustring> &messages);
	void read_label( Partition & partition ) ;
	void read_uuid( Partition & partition ) ;
	void set_mountpoints( Partition & partition );
	bool set_mountpoints_helper( Partition & partition, const Glib::ustring & path );
	bool is_busy(const Glib::ustring& device_path, FSType fstype, const Glib::ustring& partition_path);
	void set_used_sectors( Partition & partition, PedDisk* lp_disk );
	void mounted_fs_set_used_sectors(Partition& partition);
	void LP_set_used_sectors( Partition & partition, PedDisk* lp_disk ) ;
	void set_flags( Partition & partition, PedPartition* lp_partition ) ;
	
	//operationstuff...
	bool create( Partition & new_partition, OperationDetail & operationdetail );
	bool create_partition( Partition & new_partition, OperationDetail & operationdetail, Sector min_size = 0 ) ;
	bool create_filesystem( const Partition & partition, OperationDetail & operationdetail ) ;

	bool format( const Partition & partition, OperationDetail & operationdetail ) ;

	bool delete_partition( const Partition & partition, OperationDetail & operationdetail );

	bool remove_filesystem( const Partition & partition, OperationDetail & operationdetail ) ;

	bool label_filesystem( const Partition & partition, OperationDetail & operationdetail );

	bool name_partition( const Partition & partition, OperationDetail & operationdetail );

	bool change_filesystem_uuid( const Partition & partition, OperationDetail & operation_detail );

	bool resize_move( const Partition & partition_old,
	                  Partition & partition_new,
	                  OperationDetail & operationdetail );
	bool move( const Partition & partition_old,
	           const Partition & partition_new,
	           OperationDetail & operationdetail );
	bool move_filesystem( const Partition & partition_old,
			      const Partition & partition_new,
			      OperationDetail & operationdetail ) ;
	bool resize_move_filesystem_using_libparted( const Partition & partition_old,
				      		     const Partition & partition_new,
					      	     OperationDetail & operationdetail ) ;
	void thread_lp_ped_file_system_resize( PedFileSystem * fs,
	                                       PedGeometry * lp_geom,
	                                       bool * return_value );
	bool resize( const Partition & partition_old,
		     const Partition & partition_new,
		     OperationDetail & operationdetail ) ;
	bool resize_encryption( const Partition & partition_old,
	                        const Partition & partition_new,
	                        OperationDetail & operationdetail );
	bool resize_plain( const Partition & partition_old,
	                   const Partition & partition_new,
	                   OperationDetail & operationdetail );
	bool resize_move_partition( const Partition & partition_old,
	                            const Partition & partition_new,
	                            OperationDetail & operationdetail,
	                            bool rollback_on_fail );
	bool resize_move_partition_implement( const Partition & partition_old,
	                                      const Partition & partition_new,
	                                      Sector & new_start,
	                                      Sector & new_end );
	bool shrink_encryption( const Partition & partition_old,
	                        const Partition & partition_new,
	                        OperationDetail & operationdetail );
	bool maximize_encryption( const Partition & partition,
	                          OperationDetail & operationdetail );
	bool shrink_filesystem( const Partition & partition_old,
	                        const Partition & partition_new,
	                        OperationDetail & operationdetail );
	bool maximize_filesystem( const Partition & partition, OperationDetail & operationdetail ) ;
	bool recreate_linux_swap_filesystem( const Partition & partition,
	                                     OperationDetail & operationdetail );
	bool resize_filesystem_implement( const Partition & partition_old,
	                                  const Partition & partition_new,
	                                  OperationDetail & operationdetail );

	bool copy( const Partition & partition_src,
	           Partition & partition_dst,
	           OperationDetail & operationdetail );
	bool copy_filesystem( const Partition & partition_src,
	                      Partition & partition_dst,
	                      OperationDetail & operationdetail );
	bool copy_filesystem_internal( const Partition & partition_src,
	                               const Partition & partition_dst,
	                               OperationDetail & operationdetail,
	                               bool cancel_safe );
	bool copy_filesystem_internal( const Partition & partition_src,
	                               const Partition & partition_dst,
	                               OperationDetail & operationdetail,
	                               Byte_Value & total_done,
	                               bool cancel_safe );
	bool copy_blocks( const Glib::ustring & src_device,
	                  const Glib::ustring & dst_device,
	                  Sector src_start,
	                  Sector dst_start,
	                  Byte_Value src_sector_size,
	                  Byte_Value dst_sector_size,
	                  Byte_Value src_length,
	                  OperationDetail & operationdetail,
	                  Byte_Value & total_done,
	                  bool cancel_safe );
	void rollback_move_filesystem( const Partition & partition_src,
	                               const Partition & partition_dst,
	                               OperationDetail & operationdetail,
	                               Byte_Value total_done );

	bool check_repair_filesystem( const Partition & partition, OperationDetail & operationdetail ) ;
	bool check_repair_maximize( const Partition & partition,
	                            OperationDetail & operationdetail );

	bool set_partition_type(const Partition& partition, OperationDetail& operationdetail);
	static bool set_partition_type_using_flag(PedPartition* lp_partition,
					          PedPartitionFlag lp_flag,
					          OperationDetail& operationdetail);
	static bool set_partition_flag(PedPartition* lp_partition,
	                               PedPartitionFlag lp_flag,
	                               OperationDetail& operationdetail);
	bool set_partition_type_using_fstype(PedPartition* lp_partition,
	                                     FSType fstype,
	                                     OperationDetail& operationdetail);

	bool calibrate_partition( Partition & partition, OperationDetail & operationdetail ) ;
	bool calculate_exact_geom( const Partition & partition_old,
			           Partition & partition_new,
				   OperationDetail & operationdetail ) ;
	bool update_dmraid_entry( const Partition & partition_new, OperationDetail & operationdetail );
	bool erase_filesystem_signatures( const Partition & partition, OperationDetail & operationdetail ) ;
	bool update_bootsector( const Partition & partition, OperationDetail & operationdetail ) ;

	//general..	
	void capture_libparted_messages( OperationDetail & operationdetail, bool success );

	static bool get_device( const Glib::ustring & device_path, PedDevice *& lp_device, bool flush = false );
	static bool get_disk(PedDevice *lp_device, PedDisk*& lp_disk);
	static bool get_device_and_disk(const Glib::ustring& device_path, PedDevice*& lp_device, PedDisk*& lp_disk);
	static void destroy_device_and_disk( PedDevice*& lp_device, PedDisk*& lp_disk );
	static bool commit( PedDisk* lp_disk );
	static bool commit_to_os( PedDisk* lp_disk, std::time_t timeout );
	static void settle_device( std::time_t timeout );
	static bool useable_device(const PedDevice* lp_device);
	static PedPartition* get_lp_partition( const PedDisk* lp_disk, const Partition & partition );

	static PedExceptionOption ped_exception_handler( PedException * e ) ;

	std::vector<PedPartitionFlag> m_all_libparted_flags;
	std::vector<Glib::ustring> device_paths ;
	bool probe_devices ;
	Glib::ustring thread_status_message;  //Used to pass data to show_pulsebar method
	static std::unique_ptr<SupportedFileSystems> supported_filesystems;
};


}  // namespace GParted


#endif /* GPARTED_GPARTED_CORE_H */
