/* Copyright (C) 2004 Bart 'plors' Hakvoort
 *
 *  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 Library General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
 */
 
#include "../include/Utils.h"

#include <sstream>
#include <fstream>
#include <cerrno>
#include <iomanip>
#include <sys/mount.h>

namespace GParted
{
	
Sector Utils::Round( double double_value )
{
	 return static_cast<Sector>( double_value + 0.5 ) ;
}

Gtk::Label * Utils::mk_label( const Glib::ustring & text, bool use_markup, bool align_left, bool wrap, const Glib::ustring & text_color ) 
{
	Gtk::Label * label = manage( new Gtk::Label( text ) ) ;
	
	label ->set_use_markup( use_markup ) ;
	
	if ( align_left )
		label ->set_alignment( Gtk::ALIGN_LEFT ) ;
	
	label ->set_line_wrap( wrap ) ;
	
	if ( text_color != "black" )
	{
		Gdk::Color color( text_color ) ;
		label ->modify_fg( label ->get_state( ), color ) ;
	}
	
	return label ;
}

Glib::ustring Utils::num_to_str( Sector number, bool use_C_locale )
{
	std::stringstream ss ;
	//ss.imbue( std::locale( use_C_locale ? "C" : "" ) ) ; see #157871
	ss << number ;
	return ss .str( ) ;
}

//use http://developer.gnome.org/projects/gup/hig/2.0/design.html#Palette as a starting point..
Glib::ustring Utils::Get_Color( FILESYSTEM filesystem ) 
{ 
	switch( filesystem )
	{
		case FS_UNALLOCATED	: return "#A9A9A9" ;
		case FS_UNKNOWN		: return "#000000" ;
		case FS_UNFORMATTED	: return "#000000" ;
		case FS_EXTENDED	: return "#7DFCFE" ;
		case FS_EXT2		: return "#9DB8D2" ;
		case FS_EXT3		: return "#7590AE" ;
		case FS_LINUX_SWAP	: return "#C1665A" ;
		case FS_FAT16		: return "#00FF00" ;
		case FS_FAT32		: return "#18D918" ;
		case FS_NTFS		: return "#42E5AC" ;
		case FS_REISERFS	: return "#ADA7C8" ;
		case FS_REISER4		: return "#887FA3" ;
		case FS_XFS		: return "#EED680" ;
		case FS_JFS		: return "#E0C39E" ;
		case FS_HFS		: return "#E0B6AF" ;
		case FS_HFSPLUS		: return "#C0A39E" ;
		case FS_UFS		: return "#D1940C" ;
		case FS_USED		: return "#F8F8BA" ;
		case FS_UNUSED		: return "#FFFFFF" ;

		default			: return "#000000" ;
	}
}

Glib::RefPtr<Gdk::Pixbuf> Utils::get_color_as_pixbuf( FILESYSTEM filesystem, int width, int height ) 
{
	Glib::RefPtr<Gdk::Pixbuf> pixbuf = Gdk::Pixbuf::create( Gdk::COLORSPACE_RGB, false, 8, width, height ) ;

	if ( pixbuf )
	{
		std::stringstream hex( Get_Color( filesystem ) .substr( 1 ) + "00" ) ;
		unsigned long dec ;
		hex >> std::hex >> dec ;

		pixbuf ->fill( dec ) ;
	}

	return pixbuf ;
}

Glib::ustring Utils::Get_Filesystem_String( FILESYSTEM filesystem )
{
	switch( filesystem )
	{
		case FS_UNALLOCATED	: return _("unallocated") ; 
		case FS_UNKNOWN		: return _("unknown") ;
		case FS_UNFORMATTED	: return _("unformatted") ;
		case FS_EXTENDED	: return "extended" ;
		case FS_EXT2		: return "ext2" ;
		case FS_EXT3		: return "ext3" ;
		case FS_LINUX_SWAP	: return "linux-swap" ;
		case FS_FAT16		: return "fat16" ;
		case FS_FAT32		: return "fat32" ;
		case FS_NTFS		: return "ntfs" ;
		case FS_REISERFS	: return "reiserfs" ;
		case FS_REISER4		: return "reiser4" ;
		case FS_XFS		: return "xfs" ;
		case FS_JFS		: return "jfs" ;
		case FS_HFS		: return "hfs" ;
		case FS_HFSPLUS		: return "hfs+" ;
		case FS_UFS		: return "ufs" ;
		case FS_USED		: return _("used") ;
		case FS_UNUSED		: return _("unused") ;
					  
		default			: return "" ;
	}
}

bool Utils::mount( const Glib::ustring & node,
		   const Glib::ustring & mountpoint,
		   const Glib::ustring & filesystem,
		   Glib::ustring & error, 
		   unsigned long flags,
		   const Glib::ustring & data ) 
{
	if ( ! ::mount( node .c_str(), mountpoint .c_str(), filesystem .c_str(), flags, data .c_str() ) )
	{
		std::ifstream proc_mounts( "/proc/mounts" ) ;

		if ( proc_mounts )
		{
			bool hit = false ;
			char c_node[255] ;
			std::string line ;

			//search for relevant line in /proc/mounts
			while ( getline( proc_mounts, line ) )
			{
				if ( line .length() > 0 && line[ 0 ] == '/' &&
		     		     sscanf( line .c_str(),"%255s", c_node ) == 1 &&
		     		     c_node == node )
				{
					hit = true ;
					break ;
				}
			}
			
			proc_mounts .close() ;

			//append 'line' to /etc/mtab
			if ( hit )
			{
				std::ofstream mtab( "/etc/mtab", std::ios::app ) ;

				if ( mtab )
				{
					mtab << line << '\n' ;
					mtab .close() ;
	
					return true ;
				}
			}

			//something went wrong while adding the line to mtab
			umount( mountpoint .c_str() ) ;
		}
	}
	else
		error = Glib::strerror( errno ) ;

	
	return false ;
}

bool Utils::unmount( const Glib::ustring & node, const Glib::ustring & mountpoint, Glib::ustring & error ) 
{
	if ( ! umount( mountpoint .c_str() ) )
	{
		//search in /etc/mtab voor node and mountpoint and delete that line
		Glib::ustring mtab_minus_mount ;
		bool hit = false ;
		std::ifstream mtab_in( "/etc/mtab" ) ;

		if ( mtab_in )
		{
			char c_node[255], c_mountpoint[255] ;
			std::string line ;

			while ( getline( mtab_in, line ) )
			{
				if ( line .length() > 0 && line[ 0 ] == '/' &&
		     		     sscanf( line .c_str(),"%s %s", c_node, c_mountpoint ) == 2 &&
		     		     c_node == node && c_mountpoint == mountpoint
				   )
					hit = true ;
				else
					mtab_minus_mount += line + '\n';
			}

			mtab_in .close() ;
		}

		if ( hit )
		{
			std::ofstream mtab_out( "/etc/mtab" ) ;

			if ( mtab_out )
			{
				mtab_out << mtab_minus_mount ;
				mtab_out .close() ;

				return true ;
			}
		}
	}
	else
		error = Glib::strerror( errno ) ;

	
	return false ;
}

Glib::ustring Utils::format_size( Sector size ) 
{
	std::stringstream ss ;	
	//ss .imbue( std::locale( "" ) ) ;  see #157871
	ss << std::setiosflags( std::ios::fixed ) << std::setprecision( 2 ) ;

	if ( size < KIBIBYTE )
	{
		ss << sector_to_unit( size, UNIT_BYTE ) ;
		return String::ucompose( _("%1 B"), ss .str() ) ;
	}
	else if ( size < MEBIBYTE ) 
	{
		ss << sector_to_unit( size, UNIT_KIB ) ;
		return String::ucompose( _("%1 KiB"), ss .str() ) ;
	}
	else if ( size < GIBIBYTE ) 
	{
		ss << sector_to_unit( size, UNIT_MIB ) ;
		return String::ucompose( _("%1 MiB"), ss .str() ) ;
	}
	else if ( size < TEBIBYTE ) 
	{
		ss << sector_to_unit( size, UNIT_GIB ) ;
		return String::ucompose( _("%1 GiB"), ss .str() ) ;
	}
	else
	{
		ss << sector_to_unit( size, UNIT_TIB ) ;
		return String::ucompose( _("%1 TiB"), ss .str() ) ;
	}
}

double Utils::sector_to_unit( Sector sectors, SIZE_UNIT size_unit ) 
{
	/* NOTE: this could have been done more efficient by using static numbers.
	 * However, the performancegain would be unnoticable and this way its easier to read/debug
	 */
	switch ( size_unit )
	{
		case UNIT_BYTE	:
			return sectors * 512 ;
		
		case UNIT_KIB	:
			return sector_to_unit( sectors, UNIT_BYTE ) / 1024 ;
		case UNIT_MIB	:
			return sector_to_unit( sectors, UNIT_KIB ) / 1024 ;
		case UNIT_GIB	:
			return sector_to_unit( sectors, UNIT_MIB ) / 1024 ;
		case UNIT_TIB	:
			return sector_to_unit( sectors, UNIT_GIB ) / 1024 ;
		
		default:
			return sectors ;
	}
}

} //GParted..