src/key-viewer-dialog.cpp - NFD-Control-Center - Gitiles

 /* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
 /**
  * Copyright (c) 2013-2017, Regents of the University of California,
  *
  * This file is part of NFD Control Center.  See AUTHORS.md for complete list of NFD
  * authors and contributors.
  *
  * NFD Control Center 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 3 of the License, or (at your option) any later version.
  *
  * NFD Control Center 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 NFD
  * Control Center, e.g., in COPYING.md file.  If not, see <http://www.gnu.org/licenses/>.
  */

 #include "key-viewer-dialog.hpp"
 #include "key-viewer-dialog.moc"
 #include "ui_key-viewer-dialog.h"
 #include "key-tree-item.hpp"

 #include <iostream>

 #include <Qt>
 #include <QMenu>

 namespace ndn {
 namespace ncc {

 static int N_ROWS = 30;

 KeyViewerDialog::KeyViewerDialog(QWidget *parent)
   : QDialog(parent)
   , ui(new Ui::KeyViewerDialog)
   , m_model(new KeyTreeModel)
   , m_keyChain(new KeyChain)
   , m_timer(parent)
 {
   ui->setupUi(this);

   // present();

   connect(ui->treeView, SIGNAL(clicked(const QModelIndex&)), this, SLOT(displayCert(const QModelIndex&)));

   m_timer.setInterval(600000);
   connect(&m_timer, SIGNAL(timeout()), this, SLOT(display()));
   m_timer.start();

   this->setContextMenuPolicy(Qt::CustomContextMenu);

   connect(ui->treeView, SIGNAL(pressed(const QModelIndex&)),
           this, SLOT(getSelection(const QModelIndex&)));
   connect(this, SIGNAL(customContextMenuRequested(const QPoint&)),
           this, SLOT(showContextMenu(const QPoint&)));
 }

 KeyViewerDialog::~KeyViewerDialog()
 {
 }

 void
 KeyViewerDialog::updateModel()
 {
 }

 void
 KeyViewerDialog::display()
 {
   m_keyChain = make_shared<KeyChain>();
   auto newModel = make_shared<KeyTreeModel>();

   std::vector<Name> defaultIdentities;
   m_keyChain->getAllIdentities(defaultIdentities, true);
   for (const auto& identity : defaultIdentities) {
     auto idItem = createIdentityNode(identity);
     idItem->setDefault(true);
     newModel->addChild(idItem);
   }

   std::vector<Name> otherIdentities;
   m_keyChain->getAllIdentities(otherIdentities, false);
   for (const auto& identity : otherIdentities) {
     newModel->addChild(createIdentityNode(identity));
   }

   ui->treeView->setModel(newModel.get());
   ui->treeView->show();
   m_model = newModel;

   auto newTableModel = make_shared<CertTreeModel>(N_ROWS, 2);
   newTableModel->setHeaderData(0, Qt::Horizontal,"Field");
   newTableModel->setHeaderData(1, Qt::Horizontal,"Value");
   ui->certView->setModel(newTableModel.get());
   ui->certView->show();
   m_tableModel = newTableModel;
 }

 void
 KeyViewerDialog::present()
 {
   display();

   show();
   raise();
 }

 void
 KeyViewerDialog::getSelection(const QModelIndex& index)
 {
   m_selectedIndex = index;
 }

 void
 KeyViewerDialog::showContextMenu(const QPoint& point)
 {
   QMenu contextMenu(tr("Key Management"), this);
   QAction action1("Set Default", this);
   connect(&action1, SIGNAL(triggered()), this, SLOT(setDefault()));
   contextMenu.addAction(&action1);
   contextMenu.exec(ui->treeView->mapToGlobal(point));
 }

 void
 KeyViewerDialog::setDefault()
 {
   std::cerr << "default" << std::endl;
   KeyTreeItem* item = static_cast<KeyTreeItem*>(m_selectedIndex.internalPointer());
   if (item == nullptr)
     return;

   Name name(item->name().toString().toStdString());
   std::cerr << name << std::endl;

   switch (item->type()) {
   case KeyTreeItem::Type::ID:
     {
       m_keyChain->setDefaultIdentity(name);
       break;
     }
   case KeyTreeItem::Type::KEY:
     {
       m_keyChain->setDefaultKeyNameForIdentity(name);
       break;
     }
   case KeyTreeItem::Type::CERT:
     {
       m_keyChain->setDefaultCertificateNameForKey(name);
       break;
     }
   default:
     return;
   }
   present();
 }

 void
 KeyViewerDialog::displayCert(const QModelIndex& index)
 {
   std::cerr << "clicked" << std::endl;

   QModelIndex parent = index;
   KeyTreeItem* item = static_cast<KeyTreeItem*>(parent.internalPointer());
   if (item->type() == KeyTreeItem::Type::ID) {
     std::cerr << item->name().toString().toStdString() << std::endl;
     parent = m_model->index(0, 0, parent);
   }

   item = static_cast<KeyTreeItem*>(parent.internalPointer());
   if (item->type() == KeyTreeItem::Type::KEY) {
     std::cerr << item->name().toString().toStdString() << std::endl;
     parent = m_model->index(0, 0, parent);
   }

   shared_ptr<CertTreeModel> tableModel = make_shared<CertTreeModel>(N_ROWS, 2);
   tableModel->setHeaderData(0, Qt::Horizontal,"Field");
   tableModel->setHeaderData(1, Qt::Horizontal,"Value");

   item = static_cast<KeyTreeItem*>(parent.internalPointer());
   if (item->type() == KeyTreeItem::Type::CERT) {
     std::cerr << item->name().toString().toStdString() << std::endl;

     shared_ptr<IdentityCertificate> cert =
       m_keyChain->getCertificate(Name(item->name().toString().toStdString()));

     tableModel->setItem(0, 0, new QStandardItem(QString("Name")));
     tableModel->setItem(0, 1, new QStandardItem(item->name().toString()));
     // tableModel->setData(tableModel->index(0, 0), Qt::red, Qt::BackgroundRole);

     tableModel->setItem(1, 0, new QStandardItem(QString("NotBefore")));
     tableModel->setItem(1, 1, new QStandardItem(QString::fromStdString(time::toIsoString(cert->getNotBefore()))));

     tableModel->setItem(2, 0, new QStandardItem(QString("NotAfter")));
     tableModel->setItem(2, 1, new QStandardItem(QString::fromStdString(time::toIsoString(cert->getNotAfter()))));

     tableModel->setItem(3, 0, new QStandardItem(QString("FingerPrint")));
     tableModel->setItem(3, 1, new QStandardItem(QString("FFFF")));

     Name signerName = cert->getSignature().getKeyLocator().getName();
     tableModel->setItem(4, 0, new QStandardItem(QString("KeyLocator")));
     tableModel->setItem(4, 1, new QStandardItem(QString::fromStdString(signerName.toUri())));
   }
   else {
     tableModel->setItem(0, 0, new QStandardItem(QString("")));
     tableModel->setItem(0, 1, new QStandardItem(QString("")));
   }
   ui->certView->setModel(tableModel.get());
   ui->certView->show();
   m_tableModel = tableModel;
 }

 KeyTreeItem*
 KeyViewerDialog::createIdentityNode(const Name& identity)
 {
   KeyTreeItem* idItem = new KeyTreeItem(QVariant(QString::fromStdString(identity.toUri())),
                                         QVariant(QString::fromStdString(identity.toUri())),
                                         KeyTreeItem::Type::ID);

   std::vector<Name> defaultKeys;
   m_keyChain->getAllKeyNamesOfIdentity(identity, defaultKeys, true);
   for (const auto& keyName : defaultKeys) {
     auto keyItem = createKeyNode(keyName, idItem);
     keyItem->setDefault(true);
     idItem->appendChild(keyItem);
   }

   std::vector<ndn::Name> otherKeys;
   m_keyChain->getAllKeyNamesOfIdentity(identity, otherKeys, false);
   for (const auto& keyName : otherKeys) {
     idItem->appendChild(createKeyNode(keyName, idItem));
   }

   return idItem;
 }

 KeyTreeItem*
 KeyViewerDialog::createKeyNode(const Name& keyName, KeyTreeItem* idItem)
 {
   KeyTreeItem* keyItem = new KeyTreeItem(QVariant(QString::fromStdString(keyName.toUri())),
                                          QVariant(QString::fromStdString(keyName[-1].toUri())),
                                          KeyTreeItem::Type::KEY,
                                          idItem);

   std::vector<Name> defaultCertificates;
   m_keyChain->getAllCertificateNamesOfKey(keyName, defaultCertificates, true);
   for (const auto& certName : defaultCertificates) {
     auto certItem = new KeyTreeItem(QVariant(QString::fromStdString(certName.toUri())),
                                     QVariant(QString::fromStdString(certName[-1].toUri())),
                                     KeyTreeItem::Type::CERT,
                                     keyItem);
     certItem->setDefault(true);
     keyItem->appendChild(certItem);
   }

   std::vector<ndn::Name> otherCertificates;
   m_keyChain->getAllCertificateNamesOfKey(keyName, otherCertificates, false);
   for (const auto& certName : otherCertificates) {
     keyItem->appendChild(new KeyTreeItem(QVariant(QString::fromStdString(certName.toUri())),
                                          QVariant(QString::fromStdString(certName[-1].toUri())),
                                          KeyTreeItem::Type::CERT,
                                          keyItem));
   }

   return keyItem;
 }

 } // namespace ncc
 } // namespace ndn
	/* -- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -- */
	/**
	* Copyright (c) 2013-2017, Regents of the University of California,
	*
	* This file is part of NFD Control Center. See AUTHORS.md for complete list of NFD
	* authors and contributors.
	*
	* NFD Control Center 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 3 of the License, or (at your option) any later version.
	*
	* NFD Control Center 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 NFD
	* Control Center, e.g., in COPYING.md file. If not, see <http://www.gnu.org/licenses/>.
	*/

	#include "key-viewer-dialog.hpp"
	#include "key-viewer-dialog.moc"
	#include "ui_key-viewer-dialog.h"
	#include "key-tree-item.hpp"

	#include <iostream>

	#include <Qt>
	#include <QMenu>

	namespace ndn {
	namespace ncc {

	static int N_ROWS = 30;

	KeyViewerDialog::KeyViewerDialog(QWidget *parent)
	: QDialog(parent)
	, ui(new Ui::KeyViewerDialog)
	, m_model(new KeyTreeModel)
	, m_keyChain(new KeyChain)
	, m_timer(parent)
	{
	ui->setupUi(this);

	// present();

	connect(ui->treeView, SIGNAL(clicked(const QModelIndex&)), this, SLOT(displayCert(const QModelIndex&)));

	m_timer.setInterval(600000);
	connect(&m_timer, SIGNAL(timeout()), this, SLOT(display()));
	m_timer.start();

	this->setContextMenuPolicy(Qt::CustomContextMenu);

	connect(ui->treeView, SIGNAL(pressed(const QModelIndex&)),
	this, SLOT(getSelection(const QModelIndex&)));
	connect(this, SIGNAL(customContextMenuRequested(const QPoint&)),
	this, SLOT(showContextMenu(const QPoint&)));
	}

	KeyViewerDialog::~KeyViewerDialog()
	{
	}

	void
	KeyViewerDialog::updateModel()
	{
	}

	void
	KeyViewerDialog::display()
	{
	m_keyChain = make_shared<KeyChain>();
	auto newModel = make_shared<KeyTreeModel>();

	std::vector<Name> defaultIdentities;
	m_keyChain->getAllIdentities(defaultIdentities, true);
	for (const auto& identity : defaultIdentities) {
	auto idItem = createIdentityNode(identity);
	idItem->setDefault(true);
	newModel->addChild(idItem);
	}

	std::vector<Name> otherIdentities;
	m_keyChain->getAllIdentities(otherIdentities, false);
	for (const auto& identity : otherIdentities) {
	newModel->addChild(createIdentityNode(identity));
	}

	ui->treeView->setModel(newModel.get());
	ui->treeView->show();
	m_model = newModel;

	auto newTableModel = make_shared<CertTreeModel>(N_ROWS, 2);
	newTableModel->setHeaderData(0, Qt::Horizontal,"Field");
	newTableModel->setHeaderData(1, Qt::Horizontal,"Value");
	ui->certView->setModel(newTableModel.get());
	ui->certView->show();
	m_tableModel = newTableModel;
	}

	void
	KeyViewerDialog::present()
	{
	display();

	show();
	raise();
	}

	void
	KeyViewerDialog::getSelection(const QModelIndex& index)
	{
	m_selectedIndex = index;
	}

	void
	KeyViewerDialog::showContextMenu(const QPoint& point)
	{
	QMenu contextMenu(tr("Key Management"), this);
	QAction action1("Set Default", this);
	connect(&action1, SIGNAL(triggered()), this, SLOT(setDefault()));
	contextMenu.addAction(&action1);
	contextMenu.exec(ui->treeView->mapToGlobal(point));
	}

	void
	KeyViewerDialog::setDefault()
	{
	std::cerr << "default" << std::endl;
	KeyTreeItem* item = static_cast<KeyTreeItem*>(m_selectedIndex.internalPointer());
	if (item == nullptr)
	return;

	Name name(item->name().toString().toStdString());
	std::cerr << name << std::endl;

	switch (item->type()) {
	case KeyTreeItem::Type::ID:
	{
	m_keyChain->setDefaultIdentity(name);
	break;
	}
	case KeyTreeItem::Type::KEY:
	{
	m_keyChain->setDefaultKeyNameForIdentity(name);
	break;
	}
	case KeyTreeItem::Type::CERT:
	{
	m_keyChain->setDefaultCertificateNameForKey(name);
	break;
	}
	default:
	return;
	}
	present();
	}

	void
	KeyViewerDialog::displayCert(const QModelIndex& index)
	{
	std::cerr << "clicked" << std::endl;

	QModelIndex parent = index;
	KeyTreeItem* item = static_cast<KeyTreeItem*>(parent.internalPointer());
	if (item->type() == KeyTreeItem::Type::ID) {
	std::cerr << item->name().toString().toStdString() << std::endl;
	parent = m_model->index(0, 0, parent);
	}

	item = static_cast<KeyTreeItem*>(parent.internalPointer());
	if (item->type() == KeyTreeItem::Type::KEY) {
	std::cerr << item->name().toString().toStdString() << std::endl;
	parent = m_model->index(0, 0, parent);
	}

	shared_ptr<CertTreeModel> tableModel = make_shared<CertTreeModel>(N_ROWS, 2);
	tableModel->setHeaderData(0, Qt::Horizontal,"Field");
	tableModel->setHeaderData(1, Qt::Horizontal,"Value");

	item = static_cast<KeyTreeItem*>(parent.internalPointer());
	if (item->type() == KeyTreeItem::Type::CERT) {
	std::cerr << item->name().toString().toStdString() << std::endl;

	shared_ptr<IdentityCertificate> cert =
	m_keyChain->getCertificate(Name(item->name().toString().toStdString()));

	tableModel->setItem(0, 0, new QStandardItem(QString("Name")));
	tableModel->setItem(0, 1, new QStandardItem(item->name().toString()));
	// tableModel->setData(tableModel->index(0, 0), Qt::red, Qt::BackgroundRole);

	tableModel->setItem(1, 0, new QStandardItem(QString("NotBefore")));
	tableModel->setItem(1, 1, new QStandardItem(QString::fromStdString(time::toIsoString(cert->getNotBefore()))));

	tableModel->setItem(2, 0, new QStandardItem(QString("NotAfter")));
	tableModel->setItem(2, 1, new QStandardItem(QString::fromStdString(time::toIsoString(cert->getNotAfter()))));

	tableModel->setItem(3, 0, new QStandardItem(QString("FingerPrint")));
	tableModel->setItem(3, 1, new QStandardItem(QString("FFFF")));

	Name signerName = cert->getSignature().getKeyLocator().getName();
	tableModel->setItem(4, 0, new QStandardItem(QString("KeyLocator")));
	tableModel->setItem(4, 1, new QStandardItem(QString::fromStdString(signerName.toUri())));
	}
	else {
	tableModel->setItem(0, 0, new QStandardItem(QString("")));
	tableModel->setItem(0, 1, new QStandardItem(QString("")));
	}
	ui->certView->setModel(tableModel.get());
	ui->certView->show();
	m_tableModel = tableModel;
	}

	KeyTreeItem*
	KeyViewerDialog::createIdentityNode(const Name& identity)
	{
	KeyTreeItem* idItem = new KeyTreeItem(QVariant(QString::fromStdString(identity.toUri())),
	QVariant(QString::fromStdString(identity.toUri())),
	KeyTreeItem::Type::ID);

	std::vector<Name> defaultKeys;
	m_keyChain->getAllKeyNamesOfIdentity(identity, defaultKeys, true);
	for (const auto& keyName : defaultKeys) {
	auto keyItem = createKeyNode(keyName, idItem);
	keyItem->setDefault(true);
	idItem->appendChild(keyItem);
	}

	std::vector<ndn::Name> otherKeys;
	m_keyChain->getAllKeyNamesOfIdentity(identity, otherKeys, false);
	for (const auto& keyName : otherKeys) {
	idItem->appendChild(createKeyNode(keyName, idItem));
	}

	return idItem;
	}

	KeyTreeItem*
	KeyViewerDialog::createKeyNode(const Name& keyName, KeyTreeItem* idItem)
	{
	KeyTreeItem* keyItem = new KeyTreeItem(QVariant(QString::fromStdString(keyName.toUri())),
	QVariant(QString::fromStdString(keyName[-1].toUri())),
	KeyTreeItem::Type::KEY,
	idItem);

	std::vector<Name> defaultCertificates;
	m_keyChain->getAllCertificateNamesOfKey(keyName, defaultCertificates, true);
	for (const auto& certName : defaultCertificates) {
	auto certItem = new KeyTreeItem(QVariant(QString::fromStdString(certName.toUri())),
	QVariant(QString::fromStdString(certName[-1].toUri())),
	KeyTreeItem::Type::CERT,
	keyItem);
	certItem->setDefault(true);
	keyItem->appendChild(certItem);
	}

	std::vector<ndn::Name> otherCertificates;
	m_keyChain->getAllCertificateNamesOfKey(keyName, otherCertificates, false);
	for (const auto& certName : otherCertificates) {
	keyItem->appendChild(new KeyTreeItem(QVariant(QString::fromStdString(certName.toUri())),
	QVariant(QString::fromStdString(certName[-1].toUri())),
	KeyTreeItem::Type::CERT,
	keyItem));
	}

	return keyItem;
	}

	} // namespace ncc
	} // namespace ndn