/***************************************************************************
 *   Copyright (C) 2005 by Tarek Saidi                                     *
 *   tarek@linux                                                           *
 *                                                                         *
 *   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 "dialogs/PasswordGenDlg.h"
#include "dialogs/CollectEntropyDlg.h"
#include "crypto/yarrow.h"

bool CGenPwDialog::EntropyCollected=false;

CGenPwDialog::CGenPwDialog(QWidget* parent, bool StandAloneMode,Qt::WFlags fl)
: QDialog(parent,fl)
{
	setupUi(this);
	setMinimumSize(size());
	setMaximumSize(size());
	createBanner(&BannerPixmap,getPixmap("dice"),tr("Password Generator"),width());
	connect(ButtonGenerate,SIGNAL(clicked()),this,SLOT(OnGeneratePw()));
	connect(Radio_1,SIGNAL(toggled(bool)),this,SLOT(OnRadio1StateChanged(bool)));
	connect(Radio_2,SIGNAL(toggled(bool)),this,SLOT(OnRadio2StateChanged(bool)));
	connect(DialogButtons,SIGNAL(rejected()),this,SLOT(OnCancel()));
	connect(DialogButtons,SIGNAL(accepted()),this,SLOT(OnAccept()));
	connect(checkBox1,SIGNAL(clicked()),this,SLOT(estimateQuality()));
	connect(checkBox2,SIGNAL(clicked()),this,SLOT(estimateQuality()));
	connect(checkBox3,SIGNAL(clicked()),this,SLOT(estimateQuality()));
	connect(checkBox4,SIGNAL(clicked()),this,SLOT(estimateQuality()));
	connect(checkBox5,SIGNAL(clicked()),this,SLOT(estimateQuality()));
	connect(checkBox6,SIGNAL(clicked()),this,SLOT(estimateQuality()));
	connect(checkBox7,SIGNAL(clicked()),this,SLOT(estimateQuality()));
	connect(Spin_Num,SIGNAL(valueChanged(int)),this,SLOT(estimateQuality()));
	connect(Check_CollectEntropy,SIGNAL(stateChanged(int)),this,SLOT(OnCollectEntropyChanged(int)));
	connect(Edit_chars,SIGNAL(textChanged(const QString&)),this,SLOT(estimateQuality()));
	connect(Edit_chars,SIGNAL(textEdited(const QString&)),this,SLOT(OnCharsChanged(const QString&)));

	if(!StandAloneMode){
		AcceptButton=DialogButtons->addButton(QDialogButtonBox::Ok);
		AcceptButton->setDisabled(true);
		DialogButtons->addButton(QDialogButtonBox::Cancel);
	}
	else{
		DialogButtons->addButton(QDialogButtonBox::Close);
		AcceptButton=NULL;
	}

	QBitArray pwGenOptions=config->pwGenOptions();
	Radio_1->setChecked(pwGenOptions.at(0));
	Radio_2->setChecked(!pwGenOptions.at(0));
	checkBox1->setChecked(pwGenOptions.at(1));
	checkBox2->setChecked(pwGenOptions.at(2));
	checkBox3->setChecked(pwGenOptions.at(3));
	checkBox4->setChecked(pwGenOptions.at(4));
	checkBox5->setChecked(pwGenOptions.at(5));
	checkBox6->setChecked(pwGenOptions.at(6));
	checkBox7->setChecked(pwGenOptions.at(7));
	Check_CollectEntropy->setChecked(pwGenOptions.at(8));
	Check_CollectOncePerSession->setChecked(pwGenOptions.at(9));
	OnRadio1StateChanged(pwGenOptions.at(0));
	OnRadio2StateChanged(!pwGenOptions.at(0));
	Spin_Num->setValue(config->pwGenLength());
}

CGenPwDialog::~CGenPwDialog(){
	QBitArray pwGenOptions(10);
	pwGenOptions.setBit(0,Radio_1->isChecked());
	pwGenOptions.setBit(1,checkBox1->isChecked());
	pwGenOptions.setBit(2,checkBox2->isChecked());
	pwGenOptions.setBit(3,checkBox3->isChecked());
	pwGenOptions.setBit(4,checkBox4->isChecked());
	pwGenOptions.setBit(5,checkBox5->isChecked());
	pwGenOptions.setBit(6,checkBox6->isChecked());
	pwGenOptions.setBit(7,checkBox7->isChecked());
	pwGenOptions.setBit(8,Check_CollectEntropy->isChecked());
	pwGenOptions.setBit(9,Check_CollectOncePerSession->isChecked());
	config->setPwGenOptions(pwGenOptions);
	config->setPwGenLength(Spin_Num->value());
}

void CGenPwDialog::paintEvent(QPaintEvent *event){
	QDialog::paintEvent(event);
	QPainter painter(this);
	painter.setClipRegion(event->region());
	painter.drawPixmap(QPoint(0,0),BannerPixmap);
}

void CGenPwDialog::OnRadio1StateChanged(bool state)
{
	if(state){
		checkBox1->setEnabled(true);
		checkBox2->setEnabled(true);
		checkBox3->setEnabled(true);
		checkBox4->setEnabled(true);
		checkBox5->setEnabled(true);
		checkBox6->setEnabled(true);
		checkBox7->setEnabled(true);
	}else{
		checkBox1->setDisabled(true);
		checkBox2->setDisabled(true);
		checkBox3->setDisabled(true);
		checkBox4->setDisabled(true);
		checkBox5->setDisabled(true);
		checkBox6->setDisabled(true);
		checkBox7->setDisabled(true);
	}
	estimateQuality();
}

void CGenPwDialog::OnRadio2StateChanged(bool state){
	if(state)
		Edit_chars->setEnabled(true);
	else
		Edit_chars->setDisabled(true);

	estimateQuality();
}

void CGenPwDialog::OnGeneratePw()
{
	/*-------------------------------------------------------
	     ASCII
	  -------------------------------------------------------
	  "A...Z" 65...90
	  "a...z" 97...122
	  "0...9" 48...57
	  Special Characters 33...47; 58...64; 91...96; 123...126
	  "-" 45
	  "_" 95
	  -------------------------------------------------------
	*/

	int num=0;
	char assoctable[255];

	if(Radio_1->isChecked()){
		if(checkBox1->isChecked())
			num+=AddToAssoctable(assoctable,65,90,num);
		if(checkBox2->isChecked())
			num+=AddToAssoctable(assoctable,97,122,num);
		if(checkBox3->isChecked())
			num+=AddToAssoctable(assoctable,48,57,num);
		if(checkBox4->isChecked()){
			num+=AddToAssoctable(assoctable,33,47,num);
			num+=AddToAssoctable(assoctable,58,64,num);
			num+=AddToAssoctable(assoctable,91,96,num);
			num+=AddToAssoctable(assoctable,123,126,num);}
		if(checkBox5->isChecked())
			num+=AddToAssoctable(assoctable,32,32,num);
		if(checkBox6->isChecked() && !checkBox4->isChecked())
			num+=AddToAssoctable(assoctable,45,45,num);
		if(checkBox7->isChecked() && !checkBox4->isChecked())
			num+=AddToAssoctable(assoctable,95,95,num);
	}
	else{
		QString str=Edit_chars->text();
		for(int i=0;i<str.length();i++){
			assoctable[i]=str[i].toAscii();
			num++;
		}
	}
	if(num==0){
		if(Radio_2->isChecked())
			QMessageBox::information(this,tr("Notice"),tr("You need to enter at least one character"),tr("OK"));
		else
			QMessageBox::information(this,tr("Notice"),QString::fromUtf8("You need to select at least one character group."),"OK");
		return;
	}
	int length=Spin_Num->value();
	char* buffer=new char[length+1];
	buffer[length]=0;

	if(Check_CollectEntropy->isChecked()){
		if((Check_CollectOncePerSession->isChecked() && !EntropyCollected) || !Check_CollectOncePerSession->isChecked()){
			CollectEntropyDlg dlg(this);
			dlg.exec();
			EntropyCollected=true;
		}
	}

	unsigned char tmp;
	for(int i=0;i<length;i++){

		do randomize(&tmp,1);
		while(!trim(tmp,num));

		buffer[i]=assoctable[tmp];
	}

	Edit_dest->setText(buffer);
	delete [] buffer;
	if(AcceptButton)AcceptButton->setEnabled(true);
}

int CGenPwDialog::AddToAssoctable(char* table,int start,int end,int pos){
	for(int i=start;i<=end;i++){
	table[pos]=i;
	pos++;
	}
	return (end-start)+1;
}


bool CGenPwDialog::trim(unsigned char &x, int r){
	if(x<r)
		return true;
	if(256%r!=0)
		return false;
	x=x-(x/r)*r;
	return true;
}

void CGenPwDialog::estimateQuality(){
	int num=0;
	if(Radio_1->isChecked()){
		if(checkBox1->isChecked())
			num+=26;
		if(checkBox2->isChecked())
			num+=26;
		if(checkBox3->isChecked())
			num+=10;
		if(checkBox4->isChecked())
			num+=32;
		if(checkBox5->isChecked())
			num++;
		if(checkBox6->isChecked() && !checkBox4->isChecked())
			num++;
		if(checkBox7->isChecked() && !checkBox4->isChecked())
			num++;
	}
	else
		num=Edit_chars->text().length();

	float bits=0;
	if(num)bits=log(num)/log(2);
	bits=bits*((float)Spin_Num->value());
	Progress_Quali->setFormat(tr("%1 Bits").arg((int)bits));
	Progress_Quali->update();
	if(bits>128)bits=128;
	Progress_Quali->setValue((int)bits);
}

void CGenPwDialog::OnCharsChanged(const QString& str){
	bool multiple=false;
	for(int i=0;i<str.size();i++){
		int count=0;
		for(int j=0;j<str.size();j++){
			if(str[i]==str[j]){
				if(count){multiple=true; break;}
				else {count++;}
			}
		}
		if(multiple)break;
	}
	if(!multiple)return;

	QString newstr;
	for(int i=0;i<str.size();i++){
		if(!newstr.count(str[i])){
			newstr+=str[i];
		}
	}
	Edit_chars->setText(newstr);

}

void CGenPwDialog::OnAccept()
{
	done(1);
}

void CGenPwDialog::OnCancel()
{
	done(0);
}

void CGenPwDialog::OnCollectEntropyChanged(int state){
	if(state==Qt::Checked)
		Check_CollectOncePerSession->setDisabled(false);
	else
		Check_CollectOncePerSession->setDisabled(true);

}