/***************************************************************************
* Copyright ( C ) 2005 - 2006 by Tarek Saidi *
* tarek . saidi @ arcor . de *
* *
* 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 ; version 2 of the License . *
* *
* 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 , write to the *
* Free Software Foundation , Inc . , *
* 59 Temple Place - Suite 330 , Boston , MA 02111 - 1307 , USA . *
* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
# include <QFileDialog>
# include "SelectIconDlg.h"
# include "PasswordGenDlg.h"
# include "EditEntryDlg.h"
# include "CalendarDlg.h"
# ifdef GLOBAL_AUTOTYPE
# include "TargetWindowDlg.h"
# endif
CEditEntryDlg : : CEditEntryDlg ( IDatabase * _db , IEntryHandle * _entry , QWidget * parent , bool modal , bool newEntry )
: QDialog ( parent )
{
Q_ASSERT ( _db ) ;
Q_ASSERT ( _entry ) ;
entry = _entry ;
db = _db ;
pNewEntry = newEntry ;
setupUi ( this ) ;
ModFlag = false ;
QMenu * ExpirePresetsMenu = new QMenu ( ) ;
setGeometry ( config - > dialogGeometry ( this ) ) ;
connect ( Edit_Title , SIGNAL ( textChanged ( const QString & ) ) , this , SLOT ( OnTitleTextChanged ( const QString & ) ) ) ;
connect ( Edit_Password_w , SIGNAL ( editingFinished ( ) ) , this , SLOT ( OnPasswordwLostFocus ( ) ) ) ;
connect ( Edit_Password_w , SIGNAL ( textChanged ( const QString & ) ) , this , SLOT ( OnPasswordwTextChanged ( ) ) ) ;
connect ( Edit_Password , SIGNAL ( textChanged ( const QString & ) ) , this , SLOT ( OnPasswordTextChanged ( ) ) ) ;
connect ( ButtonEchoMode , SIGNAL ( clicked ( ) ) , this , SLOT ( ChangeEchoMode ( ) ) ) ;
connect ( buttonBox - > button ( QDialogButtonBox : : Cancel ) , SIGNAL ( clicked ( ) ) , this , SLOT ( OnButtonCancel ( ) ) ) ;
connect ( ButtonOpenAttachment , SIGNAL ( clicked ( ) ) , this , SLOT ( OnNewAttachment ( ) ) ) ;
connect ( ButtonDeleteAttachment , SIGNAL ( clicked ( ) ) , this , SLOT ( OnDeleteAttachment ( ) ) ) ;
connect ( ButtonSaveAttachment , SIGNAL ( clicked ( ) ) , this , SLOT ( OnSaveAttachment ( ) ) ) ;
connect ( ButtonGenPw , SIGNAL ( clicked ( ) ) , this , SLOT ( OnButtonGenPw ( ) ) ) ;
connect ( buttonBox - > button ( QDialogButtonBox : : Ok ) , SIGNAL ( clicked ( ) ) , this , SLOT ( OnButtonOK ( ) ) ) ;
connect ( CheckBox_ExpiresNever , SIGNAL ( stateChanged ( int ) ) , this , SLOT ( OnCheckBoxExpiresNeverChanged ( int ) ) ) ;
connect ( Button_Icons , SIGNAL ( clicked ( ) ) , this , SLOT ( OnButtonIcons ( ) ) ) ;
connect ( ExpirePresetsMenu , SIGNAL ( triggered ( QAction * ) ) , this , SLOT ( OnExpirePreset ( QAction * ) ) ) ;
connect ( ButtonExpirePresets , SIGNAL ( triggered ( QAction * ) ) , this , SLOT ( OnCalendar ( QAction * ) ) ) ;
connect ( this , SIGNAL ( finished ( int ) ) , this , SLOT ( OnClose ( ) ) ) ;
// QAction::data() contains the time until expiration in days.
ExpirePresetsMenu - > addAction ( tr ( " Today " ) ) - > setData ( 0 ) ;
ExpirePresetsMenu - > addSeparator ( ) ;
ExpirePresetsMenu - > addAction ( tr ( " 1 Week " ) ) - > setData ( 7 ) ;
ExpirePresetsMenu - > addAction ( tr ( " 2 Weeks " ) ) - > setData ( 14 ) ;
ExpirePresetsMenu - > addAction ( tr ( " 3 Weeks " ) ) - > setData ( 21 ) ;
ExpirePresetsMenu - > addSeparator ( ) ;
ExpirePresetsMenu - > addAction ( tr ( " 1 Month " ) ) - > setData ( 30 ) ;
ExpirePresetsMenu - > addAction ( tr ( " 3 Months " ) ) - > setData ( 90 ) ;
ExpirePresetsMenu - > addAction ( tr ( " 6 Months " ) ) - > setData ( 180 ) ;
ExpirePresetsMenu - > addSeparator ( ) ;
ExpirePresetsMenu - > addAction ( tr ( " 1 Year " ) ) - > setData ( 365 ) ;
ButtonExpirePresets - > setMenu ( ExpirePresetsMenu ) ;
QAction * actionCalendar = new QAction ( tr ( " Calendar... " ) , ButtonExpirePresets ) ;
actionCalendar - > setData ( - 1 ) ;
ButtonExpirePresets - > setDefaultAction ( actionCalendar ) ;
IconIndex = entry - > image ( ) ;
Button_Icons - > setIcon ( db - > icon ( IconIndex ) ) ;
ButtonOpenAttachment - > setIcon ( getIcon ( " fileopen " ) ) ;
ButtonDeleteAttachment - > setIcon ( getIcon ( " filedelete " ) ) ;
ButtonSaveAttachment - > setIcon ( getIcon ( " filesave " ) ) ;
ButtonExpirePresets - > setIcon ( getIcon ( " clock " ) ) ;
OnTitleTextChanged ( entry - > title ( ) ) ;
setWindowIcon ( db - > icon ( IconIndex ) ) ;
Edit_Title - > setText ( entry - > title ( ) ) ;
Edit_UserName - > setText ( entry - > username ( ) ) ;
Edit_URL - > setText ( entry - > url ( ) ) ;
SecString Password = entry - > password ( ) ;
Password . unlock ( ) ;
Edit_Password - > setText ( Password . string ( ) ) ;
Edit_Password_w - > setText ( Password . string ( ) ) ;
Password . lock ( ) ;
if ( ! config - > showPasswords ( ) )
ChangeEchoMode ( ) ;
else
ButtonEchoMode - > setIcon ( getIcon ( " pwd_show " ) ) ;
// MX-COMMENT: This call is not needed. Both Passwords fields will always have the same value
OnPasswordwLostFocus ( ) ;
int bits = ( Password . length ( ) * 8 ) ;
Label_Bits - > setText ( tr ( " %1 Bit " ) . arg ( QString : : number ( bits ) ) ) ;
if ( bits > 128 )
bits = 128 ;
Progress_Quali - > setValue ( 100 * bits / 128 ) ;
Edit_Attachment - > setText ( entry - > binaryDesc ( ) ) ;
Edit_Comment - > setPlainText ( entry - > comment ( ) ) ;
InitGroupComboBox ( ) ;
if ( ! entry - > binarySize ( ) ) {
ButtonSaveAttachment - > setDisabled ( true ) ;
ButtonDeleteAttachment - > setDisabled ( true ) ;
Label_AttachmentSize - > setText ( " " ) ;
}
else
Label_AttachmentSize - > setText ( entry - > friendlySize ( ) ) ;
if ( entry - > expire ( ) = = Date_Never ) {
DateTime_Expire - > setDisabled ( true ) ;
CheckBox_ExpiresNever - > setChecked ( true ) ;
}
else {
DateTime_Expire - > setDateTime ( entry - > expire ( ) ) ;
}
# ifdef AUTOTYPE
QToolButton * buttonTools = new QToolButton ( buttonBox ) ;
buttonTools - > setText ( " Tools " ) ;
buttonTools - > setPopupMode ( QToolButton : : InstantPopup ) ;
QMenu * menuTools = new QMenu ( buttonTools ) ;
connect ( menuTools - > addAction ( " Auto-Type: Customize Sequence " ) , SIGNAL ( triggered ( bool ) ) , SLOT ( OnCustomizeSequence ( ) ) ) ;
# ifdef GLOBAL_AUTOTYPE
connect ( menuTools - > addAction ( " Auto-Type: Select target window " ) , SIGNAL ( triggered ( bool ) ) , SLOT ( OnSelectTarget ( ) ) ) ;
# endif
buttonTools - > setMenu ( menuTools ) ;
buttonBox - > addButton ( buttonTools , QDialogButtonBox : : ResetRole ) ; // ResetRole: workaround to display button on the left
# endif
}
CEditEntryDlg : : ~ CEditEntryDlg ( ) {
}
void CEditEntryDlg : : resizeEvent ( QResizeEvent * event ) {
createBanner ( & BannerPixmap , getPixmap ( " keepassx_large " ) , pNewEntry ? tr ( " New Entry " ) : tr ( " Edit Entry " ) , width ( ) ) ;
QDialog : : resizeEvent ( event ) ;
}
void CEditEntryDlg : : paintEvent ( QPaintEvent * event ) {
QDialog : : paintEvent ( event ) ;
QPainter painter ( this ) ;
painter . setClipRegion ( event - > region ( ) ) ;
painter . drawPixmap ( QPoint ( 0 , 0 ) , BannerPixmap ) ;
}
void CEditEntryDlg : : InitGroupComboBox ( ) {
QString Space ;
groups = db - > sortedGroups ( ) ;
for ( int i = 0 ; i < groups . size ( ) ; i + + ) {
Space . fill ( ' ' , 2 * ( groups [ i ] - > level ( ) ) ) ;
Combo_Group - > insertItem ( i , db - > icon ( groups [ i ] - > image ( ) ) , Space + groups [ i ] - > title ( ) ) ;
if ( groups [ i ] = = entry - > group ( ) ) {
Combo_Group - > setCurrentIndex ( i ) ;
GroupIndex = i ;
}
}
}
void CEditEntryDlg : : OnButtonOK ( )
{
bool EntryMoved = false ;
if ( QString : : compare ( Edit_Password - > text ( ) , Edit_Password_w - > text ( ) ) ! = 0 ) {
QMessageBox : : warning ( NULL , tr ( " Warning " ) , tr ( " Password and password repetition are not equal. \n Please check your input. " ) , tr ( " OK " ) ) ;
return ;
}
if ( CheckBox_ExpiresNever - > checkState ( ) = = Qt : : Checked ) {
DateTime_Expire - > setDateTime ( Date_Never ) ;
}
if ( DateTime_Expire - > dateTime ( ) ! = entry - > expire ( ) )
ModFlag = true ;
if ( entry - > title ( ) ! = Edit_Title - > text ( ) )
ModFlag = true ;
if ( entry - > username ( ) ! = Edit_UserName - > text ( ) )
ModFlag = true ;
if ( entry - > url ( ) ! = Edit_URL - > text ( ) )
ModFlag = true ;
if ( entry - > comment ( ) ! = Edit_Comment - > toPlainText ( ) )
ModFlag = true ;
SecString pw = entry - > password ( ) ;
pw . unlock ( ) ;
QString password = pw . string ( ) ;
if ( password ! = Edit_Password - > text ( ) )
ModFlag = true ;
pw . lock ( ) ;
password . fill ( ' X ' ) ;
if ( entry - > image ( ) ! = IconIndex )
ModFlag = true ;
if ( ModFlag ) {
QDateTime now = QDateTime : : currentDateTime ( ) ;
entry - > setExpire ( DateTime_Expire - > dateTime ( ) ) ;
entry - > setLastAccess ( now ) ;
entry - > setLastMod ( now ) ;
entry - > setTitle ( Edit_Title - > text ( ) ) ;
entry - > setUsername ( Edit_UserName - > text ( ) ) ;
entry - > setUrl ( Edit_URL - > text ( ) ) ;
SecString pw ;
QString password = Edit_Password - > text ( ) ;
pw . setString ( password , true ) ;
entry - > setPassword ( pw ) ;
entry - > setComment ( Edit_Comment - > toPlainText ( ) ) ;
entry - > setImage ( IconIndex ) ;
}
if ( Combo_Group - > currentIndex ( ) ! = GroupIndex ) {
db - > moveEntry ( entry , groups [ Combo_Group - > currentIndex ( ) ] ) ;
EntryMoved = true ;
}
if ( EntryMoved )
done ( ModFlag ? 2 : 3 ) ;
else if ( ModFlag )
done ( 1 ) ;
else
done ( 0 ) ;
}
void CEditEntryDlg : : OnButtonCancel ( )
{
entry - > setLastAccess ( QDateTime : : currentDateTime ( ) ) ;
done ( 0 ) ;
}
void CEditEntryDlg : : ChangeEchoMode ( )
{
if ( Edit_Password - > echoMode ( ) = = QLineEdit : : Normal ) {
Edit_Password - > setEchoMode ( QLineEdit : : Password ) ;
Edit_Password_w - > setEchoMode ( QLineEdit : : Password ) ;
ButtonEchoMode - > setIcon ( getIcon ( " pwd_hide " ) ) ;
}
else
{
Edit_Password - > setEchoMode ( QLineEdit : : Normal ) ;
Edit_Password_w - > setEchoMode ( QLineEdit : : Normal ) ;
ButtonEchoMode - > setIcon ( getIcon ( " pwd_show " ) ) ;
}
}
void CEditEntryDlg : : OnTitleTextChanged ( const QString & txt )
{
setWindowTitle ( ( txt = = " " ) ? tr ( " [Untitled Entry] " ) : txt ) ;
}
void CEditEntryDlg : : OnPasswordTextChanged ( )
{
Edit_Password_w - > setText ( QString ( ) ) ;
int bits = ( Edit_Password - > text ( ) . length ( ) * 8 ) ;
Label_Bits - > setText ( QString : : number ( bits ) + " Bit " ) ;
if ( bits > 128 ) bits = 128 ;
Progress_Quali - > setValue ( 100 * bits / 128 ) ;
}
void CEditEntryDlg : : OnPasswordwTextChanged ( )
{
if ( QString : : compare ( Edit_Password_w - > text ( ) , Edit_Password - > text ( ) . mid ( 0 , ( Edit_Password_w - > text ( ) . length ( ) ) ) ) ! = 0 ) {
QPalette palette ;
palette . setColor ( Edit_Password_w - > backgroundRole ( ) , QColor ( 255 , 125 , 125 ) ) ;
Edit_Password_w - > setPalette ( palette ) ;
} else
{
Edit_Password_w - > setPalette ( QApplication : : palette ( ) ) ;
}
}
void CEditEntryDlg : : OnPasswordwLostFocus ( )
{
if ( QString : : compare ( Edit_Password_w - > text ( ) , Edit_Password - > text ( ) ) ! = 0 ) {
QPalette palette ;
palette . setColor ( Edit_Password_w - > backgroundRole ( ) , QColor ( 255 , 125 , 125 ) ) ;
Edit_Password_w - > setPalette ( palette ) ;
}
else
{
Edit_Password_w - > setPalette ( QApplication : : palette ( ) ) ;
}
}
void CEditEntryDlg : : OnNewAttachment ( )
{
QString filename = QFileDialog : : getOpenFileName ( this , tr ( " Add Attachment... " ) , QDir : : homePath ( ) ) ;
if ( filename = = " " ) return ;
QFile file ( filename ) ;
if ( file . open ( QIODevice : : ReadOnly ) = = false ) {
file . close ( ) ;
QMessageBox : : warning ( NULL , tr ( " Error " ) , tr ( " Could not open file. " ) , tr ( " OK " ) ) ;
return ;
}
ModFlag = true ;
entry - > setBinary ( file . readAll ( ) ) ;
file . close ( ) ;
QFileInfo info ( filename ) ;
entry - > setBinaryDesc ( info . fileName ( ) ) ;
Edit_Attachment - > setText ( entry - > binaryDesc ( ) ) ;
Label_AttachmentSize - > setText ( entry - > friendlySize ( ) ) ;
ButtonOpenAttachment - > setEnabled ( true ) ;
ButtonSaveAttachment - > setEnabled ( true ) ;
ButtonDeleteAttachment - > setEnabled ( true ) ;
}
void CEditEntryDlg : : OnSaveAttachment ( ) {
saveAttachment ( entry , this ) ;
}
void CEditEntryDlg : : saveAttachment ( IEntryHandle * pEntry , QWidget * ParentWidget )
{
if ( ! pEntry - > binarySize ( ) ) {
QMessageBox : : information ( NULL , tr ( " Error " ) , tr ( " The chosen entry has no attachment or it is empty. " ) , tr ( " OK " ) ) ;
return ;
}
QFileDialog FileDlg ( ParentWidget , tr ( " Save Attachment... " ) , QDir : : homePath ( ) ) ;
FileDlg . selectFile ( pEntry - > binaryDesc ( ) ) ;
FileDlg . setAcceptMode ( QFileDialog : : AcceptSave ) ;
if ( ! FileDlg . exec ( ) ) return ;
QString filename = FileDlg . selectedFiles ( ) [ 0 ] ;
QFile file ( filename ) ;
if ( ! file . open ( QIODevice : : WriteOnly | QIODevice : : Truncate ) ) {
QMessageBox : : critical ( NULL , tr ( " Error " ) , tr ( " Could not open file. " ) , tr ( " OK " ) ) ;
return ;
}
int r = file . write ( pEntry - > binary ( ) ) ;
if ( r = = - 1 ) {
file . close ( ) ;
QMessageBox : : critical ( NULL , tr ( " Error " ) , tr ( " Error while writing the file. " ) , tr ( " OK " ) ) ;
return ;
}
if ( r ! = pEntry - > binarySize ( ) ) {
file . close ( ) ;
QMessageBox : : critical ( NULL , tr ( " Error " ) , tr ( " Error while writing the file. " ) , tr ( " OK " ) ) ;
return ;
}
file . close ( ) ;
}
void CEditEntryDlg : : OnDeleteAttachment ( )
{
int r = QMessageBox : : warning ( this , tr ( " Delete Attachment? " ) , tr ( " You are about to delete the attachment of this entry. \n Are you sure? " ) , tr ( " Yes " ) , tr ( " No, Cancel " ) , NULL , 1 , 1 ) ;
if ( r = = 0 ) {
ModFlag = true ;
entry - > setBinary ( QByteArray ( ) ) ;
entry - > setBinaryDesc ( " " ) ;
Edit_Attachment - > setText ( " " ) ;
Label_AttachmentSize - > setText ( " " ) ;
ButtonOpenAttachment - > setEnabled ( true ) ;
ButtonSaveAttachment - > setDisabled ( true ) ;
ButtonDeleteAttachment - > setDisabled ( true ) ;
}
}
void CEditEntryDlg : : OnButtonGenPw ( )
{
CGenPwDialog dlg ( this , false ) ;
if ( dlg . exec ( ) ) {
Edit_Password - > setText ( dlg . Edit_dest - > text ( ) ) ;
Edit_Password_w - > setText ( dlg . Edit_dest - > text ( ) ) ;
ModFlag = true ;
}
}
void CEditEntryDlg : : OnCheckBoxExpiresNeverChanged ( int state ) {
if ( state = = Qt : : Unchecked )
DateTime_Expire - > setDisabled ( false ) ;
else
DateTime_Expire - > setDisabled ( true ) ;
}
void CEditEntryDlg : : OnButtonIcons ( ) {
// CSelectIconDlg dlg(db,Combo_IconPicker->currentIndex(),this);
CSelectIconDlg dlg ( db , IconIndex , this ) ;
int r = dlg . exec ( ) ;
if ( r ! = - 1 )
{
IconIndex = r ;
Button_Icons - > setIcon ( db - > icon ( IconIndex ) ) ;
setWindowIcon ( db - > icon ( IconIndex ) ) ;
}
}
void CEditEntryDlg : : OnExpirePreset ( QAction * action ) {
CheckBox_ExpiresNever - > setChecked ( false ) ;
int days = action - > data ( ) . toInt ( ) ;
switch ( days ) {
case 30 :
case 90 :
case 180 :
DateTime_Expire - > setDate ( QDate : : currentDate ( ) . addMonths ( days / 30 ) ) ;
break ;
case 365 :
DateTime_Expire - > setDate ( QDate : : currentDate ( ) . addYears ( 1 ) ) ;
break ;
default :
DateTime_Expire - > setDate ( QDate : : currentDate ( ) . addDays ( days ) ) ;
}
DateTime_Expire - > setTime ( QTime ( 0 , 0 , 0 ) ) ;
}
void CEditEntryDlg : : OnCalendar ( QAction * action ) {
if ( action - > data ( ) . toInt ( ) ! = - 1 )
return ;
CalendarDialog dlg ( this ) ;
if ( dlg . exec ( ) = = QDialog : : Accepted ) {
CheckBox_ExpiresNever - > setChecked ( false ) ;
DateTime_Expire - > setDate ( dlg . calendarWidget - > selectedDate ( ) ) ;
DateTime_Expire - > setTime ( QTime ( 0 , 0 , 0 ) ) ;
}
}
void CEditEntryDlg : : OnClose ( ) {
config - > setDialogGeometry ( this ) ;
}
void CEditEntryDlg : : OnCustomizeSequence ( ) {
# ifdef AUTOTYPE
QString text = Edit_Comment - > toPlainText ( ) ;
if ( ! text . isEmpty ( ) )
text . append ( " \n " ) ;
Edit_Comment - > setPlainText ( text . append ( " Auto-Type: {USERNAME}{TAB}{PASSWORD}{ENTER} " ) ) ;
# endif
}
void CEditEntryDlg : : OnSelectTarget ( ) {
# ifdef GLOBAL_AUTOTYPE
TargetWindowDlg dlg ( this ) ;
if ( dlg . exec ( ) & & ! dlg . windowTitle ( ) . isEmpty ( ) ) {
QString text = Edit_Comment - > toPlainText ( ) ;
if ( ! text . isEmpty ( ) )
text . append ( " \n " ) ;
Edit_Comment - > setPlainText ( text . append ( " Auto-Type-Window: " + dlg . windowTitle ( ) ) ) ;
}
# endif
}