2 files changed, 20 insertions, 15 deletions
diff --git a/Swift/QtUI/QtChatWindow.cpp b/Swift/QtUI/QtChatWindow.cpp
index 574a0a2..3491d3c 100644
--- a/Swift/QtUI/QtChatWindow.cpp
+++ b/Swift/QtUI/QtChatWindow.cpp
@@ -201,6 +201,13 @@ void QtChatWindow::handleFontResized(int fontSizeSteps) {
 void QtChatWindow::handleAlertButtonClicked() {
-	onAlertButtonClicked();
+	const QObject* alertWidget = QObject::sender()->parent();
+	std::map<AlertID, QWidget*>::const_iterator i = alertWidgets_.begin();
+	for ( ; i != alertWidgets_.end(); ++i) {
+		if (i->second == alertWidget) {
+			removeAlert(i->first);
+			break;
+		}
+	}
 }
 
-QtChatWindow::AlertID QtChatWindow::addAlert(const std::string& alertText, const std::string& buttonText) {
+QtChatWindow::AlertID QtChatWindow::addAlert(const std::string& alertText) {
 	QWidget* alertWidget = new QWidget(this);
@@ -209,6 +216,13 @@ QtChatWindow::AlertID QtChatWindow::addAlert(const std::string& alertText, const
 	QLabel* alertLabel = new QLabel(this);
+	alertLabel->setText(alertText.c_str());
 	alertLayout->addWidget(alertLabel);
-	alertButton_ = new QPushButton(this);
-	connect (alertButton_, SIGNAL(clicked()), this, SLOT(handleAlertButtonClicked()));
-	alertLayout->addWidget(alertButton_);
+
+	QToolButton* closeButton = new QToolButton(alertWidget);
+	closeButton->setIcon(style()->standardIcon(QStyle::SP_TitleBarCloseButton));
+	closeButton->setIconSize(QSize(16,16));
+	closeButton->setCursor(Qt::ArrowCursor);
+	closeButton->setStyleSheet("QToolButton { border: none; padding: 0px; }");
+	connect (closeButton, SIGNAL(clicked()), this, SLOT(handleAlertButtonClicked()));
+
+	alertLayout->addWidget(closeButton);
 	QPalette palette = alertWidget->palette();
@@ -219,10 +233,2 @@ QtChatWindow::AlertID QtChatWindow::addAlert(const std::string& alertText, const
 
-	alertLabel->setText(alertText.c_str());
-	if (buttonText.empty()) {
-		alertButton_->hide();
-	} else {
-		alertButton_->setText(buttonText.c_str());
-		alertButton_->show();
-	}
-
 	AlertID id = nextAlertId_++;
diff --git a/Swift/QtUI/QtChatWindow.h b/Swift/QtUI/QtChatWindow.h
index 45eaa6f..bf37557 100644
--- a/Swift/QtUI/QtChatWindow.h
+++ b/Swift/QtUI/QtChatWindow.h
@@ -136,3 +136,3 @@ namespace Swift {
 			void handleFontResized(int fontSizeSteps);
-			AlertID addAlert(const std::string& alertText, const std::string& buttonText = "");
+			AlertID addAlert(const std::string& alertText);
 			void removeAlert(const AlertID id);
@@ -199,3 +199,2 @@ namespace Swift {
 			AlertID nextAlertId_;
-			QPushButton* alertButton_;
 			TabComplete* completer_;