RetroShare/retroshare-gui/src/gui/RetroShareLink.cpp

/***************************************************************************
 *   Copyright (C) 2009                                                    *
 *                                                                         *
 *   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 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 <iostream>
#include <QStringList>
#include <QRegExp>
#include <QApplication>
#include <QMimeData>
#include <QClipboard>
#include "RetroShareLink.h"

#define DEBUG_RSLINK 1

const QString RetroShareLink::HEADER_NAME("retroshare://file");

RetroShareLink::RetroShareLink(const QUrl& url)
{
	// parse
#ifdef DEBUG_RSLINK
	std::cerr << "got new RS link \"" << url.toString().toStdString() << "\"" << std::endl ;
#endif
	QStringList list = url.toString().split ("|");

	if(list.size() < 4)
		goto bad_link ;

	bool ok ;

	_name = list[1] ;
	_size = list[2].toULongLong(&ok) ;
	_hash = list[3].left(40) ;	// normally not necessary, but it's a security.

	if(!ok)
		goto bad_link ;
#ifdef DEBUG_RSLINK
	std::cerr << "New RetroShareLink forged:" << std::endl ;
	std::cerr << "  name = \"" << _name.toStdString() << "\"" << std::endl ;
	std::cerr << "  hash = \"" << _hash.toStdString() << "\"" << std::endl ;
	std::cerr << "  size = " << _size << std::endl ;
#endif

	 check() ;

	 return ;
bad_link:
#ifdef DEBUG_RSLINK
	 std::cerr << "Wrongly formed RS link. Can't process." << std::endl ;
#endif
	 _hash = "" ;
	 _size = 0 ;
	 _name = "" ;
}

RetroShareLink::RetroShareLink(const QString & name, uint64_t size, const QString & hash)
	: _name(name),_size(size),_hash(hash)
{
	check() ;
}

void RetroShareLink::check()
{
	bool valid = true ;

	if(_size > (((uint64_t)1)<<40))	// 1TB. Who has such large files?
		valid = false ;

	if(!checkName(_name))
		valid = false ;

	if(!checkHash(_hash))
		valid = false ;

	if(!valid) // we should throw an exception instead of this crap, but drbob doesn't like exceptions. Why ???
	{
		_hash = "" ;
		_name = "" ;
		_size = 0 ;
	}
}

QString RetroShareLink::toString() const
{
	return HEADER_NAME + "|" + _name + "|" + QString::number(_size) + "|" + _hash ;
}
QString RetroShareLink::toHtml() const
{
	return QString("<a href='") + toString() + "'>" + name() + "</a>" ;
}
QString RetroShareLink::toHtmlFull() const
{
	return QString("<a href='") + toString() + "'>" + toString() + "</a>" ;
}

bool RetroShareLink::checkName(const QString& name)
{
	if(name == "") 
		return false ;

	for(int i=0;i<name.length();++i)
	{
		QChar::Category cat( name[i].category() ) ;

		if(	cat == QChar::Separator_Line
			|| cat == QChar::Other_NotAssigned
			)
		{
#ifdef DEBUG_RSLINK
			std::cerr <<"Unwanted category " << cat << " at place " << i << " in string \"" << name.toStdString() << "\"" << std::endl ;
#endif
			return false ;
		}
	}

	return true ;
}

QUrl RetroShareLink::toUrl() const
{
	return QUrl(toString()) ;
}

bool RetroShareLink::checkHash(const QString& hash)
{
	if(hash.length() != 40)
		return false ;

	QByteArray qb(hash.toAscii()) ;

	for(int i=0;i<qb.length();++i)
	{
		unsigned char b(qb[i]) ;

		if(!((b>47 && b<58) || (b>96 && b<103)))
			return false ;
	}

	return true ;
}


void RSLinkClipboard::copyLinks(const std::vector<RetroShareLink>& links) 
{
	QString res ;
	for(uint32_t i=0;i<links.size();++i)
		res += links[i].toString() + "\n" ;

	QApplication::clipboard()->setText(res) ;
}

std::vector<RetroShareLink> RSLinkClipboard::pasteLinks() 
{
	return parseClipboard() ;
}

std::vector<RetroShareLink> RSLinkClipboard::parseClipboard()
{
	// parse clipboard for links.
	//
	std::vector<RetroShareLink> links ;
	QString text = QApplication::clipboard()->text() ;

	std::cerr << "Parsing clipboard:" << text.toStdString() << std::endl ;

	QRegExp rx("retroshare://file") ;
	rx.setPatternSyntax(QRegExp::Wildcard) ;

	int pos = 0;

	while((pos = rx.indexIn(text, pos)) != -1) 
	{
		QString txt = text.right(text.length()-pos) ;
		QStringList lst = txt.split('|') ;

		if(lst.size() < 4)
			break ;

		bool ok = false ;
		RetroShareLink link(lst[1],lst[2].toULongLong(&ok),lst[3].left(40)) ;

		if(link.valid())
		{
			// check that the link is not already in the list:
			bool already = false ;
			for(uint32_t i=0;i<links.size();++i)
				if(links[i] == link)
				{
					already = true ;
					break ;
				}

			if(!already)
			{
				links.push_back(link) ;
				std::cerr << "captured link: " << link.toString().toStdString() << std::endl ;
			}
		}
		else
			std::cerr << "invalid link" << std::endl ;

		pos += rx.matchedLength();
	}

	return links ;
}

QString RSLinkClipboard::toString()
{
	std::vector<RetroShareLink> links(parseClipboard()) ;

	QString res ;
	for(uint32_t i=0;i<links.size();++i)
		res += links[i].toString() + "\n" ;

	return res ;
}

QString RSLinkClipboard::toHtml()
{
	std::vector<RetroShareLink> links(parseClipboard()) ;

	QString res ;
	for(uint32_t i=0;i<links.size();++i)
		res += links[i].toHtml() + "<br/>" ;

	return res ;
}
QString RSLinkClipboard::toHtmlFull()
{
	std::vector<RetroShareLink> links(parseClipboard()) ;

	QString res ;
	for(uint32_t i=0;i<links.size();++i)
		res += links[i].toHtmlFull() + "<br/>" ;

	return res ;
}
bool RSLinkClipboard::empty()
{
	return  parseClipboard().empty() ;
}