summaryrefslogtreecommitdiffstats
path: root/smb4k/core/smb4kbookmarkhandler.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'smb4k/core/smb4kbookmarkhandler.cpp')
-rw-r--r--smb4k/core/smb4kbookmarkhandler.cpp342
1 files changed, 342 insertions, 0 deletions
diff --git a/smb4k/core/smb4kbookmarkhandler.cpp b/smb4k/core/smb4kbookmarkhandler.cpp
new file mode 100644
index 0000000..113d601
--- /dev/null
+++ b/smb4k/core/smb4kbookmarkhandler.cpp
@@ -0,0 +1,342 @@
+/***************************************************************************
+ smb4kbookmarkhandler - This class handles the bookmarks.
+ -------------------
+ begin : Fr Jan 9 2004
+ copyright : (C) 2004 by Alexander Reinholdt
+ email : dustpuppy@mail.berlios.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; 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., 51 Franklin Street, Fifth Floor, Boston, *
+ * MA 02110-1301 USA *
+ ***************************************************************************/
+
+// Qt includes
+#include <qdir.h>
+#include <qfile.h>
+
+// KDE includes
+#include <kstandarddirs.h>
+#include <kdebug.h>
+#include <kapplication.h>
+
+// system specific includes
+#include <stdlib.h>
+#include <math.h>
+
+// application specific includes
+#include "smb4kbookmarkhandler.h"
+#include "smb4kdefs.h"
+#include "smb4kerror.h"
+#include "smb4kglobal.h"
+#include "smb4knetworkitems.h"
+#include "smb4kbookmark.h"
+
+using namespace Smb4KGlobal;
+
+
+Smb4KBookmarkHandler::Smb4KBookmarkHandler( QValueList<Smb4KHostItem *> *hosts,
+QObject *parent, const char *name )
+: QObject( parent, name ), m_hosts( hosts )
+{
+ // First we need the directory.
+ KStandardDirs *stddir = new KStandardDirs();
+ QString dir = locateLocal( "data", "smb4k", KGlobal::instance() );
+
+ if ( !stddir->exists( dir ) )
+ {
+ stddir->makeDir( dir );
+ }
+
+ delete stddir;
+
+ loadBookmarks();
+}
+
+
+Smb4KBookmarkHandler::~Smb4KBookmarkHandler()
+{
+ for ( QValueList<Smb4KBookmark *>::Iterator it = m_bookmarks.begin(); it != m_bookmarks.end(); ++it )
+ {
+ delete *it;
+ }
+
+ m_bookmarks.clear();
+
+ // Do not delete m_hosts here, because it is either NULL
+ // or it is handled outside of this class.
+}
+
+
+
+void Smb4KBookmarkHandler::addBookmark( Smb4KBookmark *bookmark )
+{
+ if ( !bookmark )
+ {
+ return;
+ }
+
+ if ( QString::compare( bookmark->type(), "Printer" ) == 0 )
+ {
+ Smb4KError::error( ERROR_BOOKMARK_PRINTER );
+ return;
+ }
+
+ if ( QString::compare( bookmark->share(), "homes" ) == 0 )
+ {
+ QString share = specifyUser( bookmark->host(), kapp->mainWidget() ? kapp->mainWidget() : 0, "SpecifyUser" );
+
+ bookmark->setShareName( share );
+ }
+
+ // Search for the bookmark:
+ Smb4KBookmark *result = findBookmarkByName( bookmark->bookmark() );
+
+ if ( result )
+ {
+ if ( QString::compare( result->workgroup().upper(), bookmark->workgroup().upper() ) == 0 )
+ {
+ // It's the same bookmark. We'll update it in a very
+ // brutal but efficient way:
+ m_bookmarks.remove( result );
+
+ if ( result )
+ {
+ delete result;
+ }
+ }
+
+ m_bookmarks.append( bookmark );
+ }
+ else
+ {
+ // The bookmark is new. Append it to the list:
+ m_bookmarks.append( bookmark );
+ }
+
+ writeBookmarkList( m_bookmarks );
+}
+
+
+void Smb4KBookmarkHandler::writeBookmarkList( const QValueList<Smb4KBookmark *> &list )
+{
+ if ( list != m_bookmarks )
+ {
+ for ( QValueListIterator<Smb4KBookmark *> it = m_bookmarks.begin(); it != m_bookmarks.end(); ++it )
+ {
+ delete *it;
+ }
+
+ m_bookmarks.clear();
+
+ m_bookmarks = list;
+ }
+
+ QFile file( locateLocal( "data", "smb4k/bookmarks", KGlobal::instance() ) );
+
+ if ( file.open( IO_WriteOnly ) )
+ {
+ QTextStream ts( &file );
+ ts.setEncoding( QTextStream::Locale );
+
+ int serial_number = 0;
+
+ for ( QValueListConstIterator<Smb4KBookmark *> it = m_bookmarks.begin(); it != m_bookmarks.end(); ++it )
+ {
+ if ( !(*it)->label().isEmpty() )
+ {
+ Smb4KBookmark *result = findBookmarkByLabel( (*it)->label() );
+
+ if ( result &&
+ (QString::compare( result->bookmark().upper(), (*it)->bookmark().upper() ) != 0 ||
+ QString::compare( result->workgroup().upper(), (*it)->workgroup().upper() ) != 0) )
+ {
+ Smb4KError::information( INFO_BOOKMARK_LABEL_IN_USE, (*it)->label(), (*it)->bookmark() );
+
+ (*it)->setLabel( QString( "%1 (%2)" ).arg( (*it)->label() ).arg( serial_number++ ) );
+ }
+ }
+
+ ts << (*it)->host() << ","
+ << (*it)->share() << ","
+ << (*it)->workgroup() << ","
+ << (*it)->ip() << ","
+ << (*it)->label() << endl;
+ }
+
+ file.close();
+ }
+ else
+ {
+ Smb4KError::error( ERROR_WRITING_FILE, QDir::currentDirPath()+"/"+file.name() );
+ return;
+ }
+
+ emit bookmarksUpdated();
+}
+
+
+void Smb4KBookmarkHandler::loadBookmarks()
+{
+ QFile file( locateLocal( "data", "smb4k/bookmarks", KGlobal::instance() ) );
+
+ QStringList contents;
+
+ if ( file.open( IO_ReadOnly ) )
+ {
+ QTextStream ts( &file );
+ ts.setEncoding( QTextStream::Locale );
+
+ contents = QStringList::split( '\n', ts.read(), false );
+
+ file.close();
+
+ for ( QStringList::ConstIterator it = contents.begin(); it != contents.end(); ++it )
+ {
+ if ( (*it).startsWith( "#" ) || (*it).startsWith( "[" ) ||
+ QString::compare( (*it).stripWhiteSpace(), QString::null ) == 0 )
+ {
+ continue;
+ }
+ else
+ {
+ // Load old bookmark entries (prior to version 0.7.0)
+ if ( ((*it).stripWhiteSpace())[0].isDigit() )
+ {
+ QString bookmark = (*it).section( "=", 1, -1 ).stripWhiteSpace();
+ m_bookmarks.append( new Smb4KBookmark( bookmark.section( "/", 2, 2 ).stripWhiteSpace(), bookmark.section( "/", 3, 3 ).stripWhiteSpace(), QString::null, QString::null, "Disk" ) );
+ }
+ else
+ {
+ QString host = (*it).section( ",", 0, 0 ).stripWhiteSpace();
+ QString share = (*it).section( ",", 1, 1 ).stripWhiteSpace();
+ QString workgroup = (*it).section( ",", 2, 2 ).stripWhiteSpace();
+ QString ip = (*it).section( ",", 3, 3 ).stripWhiteSpace();
+ QString label = (*it).section( ",", 4, 4 ).stripWhiteSpace();
+
+ m_bookmarks.append( new Smb4KBookmark( host, share, workgroup, ip, "Disk", label ) );
+ }
+ }
+ }
+
+ emit bookmarksUpdated();
+ }
+ else
+ {
+ if ( file.exists() )
+ {
+ Smb4KError::error( ERROR_READING_FILE, file.name() );
+ }
+ else
+ {
+ // Do nothing if the file does not exist.
+ }
+ }
+}
+
+
+Smb4KBookmark *Smb4KBookmarkHandler::findBookmarkByName( const QString &bookmark )
+{
+ // Update the bookmarks:
+ update();
+
+ // Find the bookmark:
+ QValueListConstIterator<Smb4KBookmark *> it;
+
+ for ( it = m_bookmarks.begin(); it != m_bookmarks.end(); ++it )
+ {
+ if ( QString::compare( (*it)->bookmark().upper(), bookmark.upper() ) == 0 )
+ {
+ break;
+ }
+ }
+
+ return it != m_bookmarks.end() ? *it : NULL;
+}
+
+
+Smb4KBookmark *Smb4KBookmarkHandler::findBookmarkByLabel( const QString &label )
+{
+ // Update the bookmarks:
+ update();
+
+ // Find the bookmark:
+ QValueListConstIterator<Smb4KBookmark *> it;
+
+ for ( it = m_bookmarks.begin(); it != m_bookmarks.end(); ++it )
+ {
+ if ( QString::compare( (*it)->label().upper(), label.upper() ) == 0 )
+ {
+ break;
+ }
+ }
+
+ return it != m_bookmarks.end() ? *it : NULL;
+}
+
+
+const QValueList<Smb4KBookmark *> &Smb4KBookmarkHandler::getBookmarks()
+{
+ // Update the bookmarks:
+ update();
+
+ // Return the list of bookmarks:
+ return m_bookmarks;
+}
+
+
+void Smb4KBookmarkHandler::update()
+{
+ // If the user didn't pass the global Smb4KHostItem list, we are not able
+ // to update the bookmarks:
+ if ( !m_hosts )
+ {
+ return;
+ }
+
+ // Search the list of hosts for new IP addresses:
+ for ( QValueList<Smb4KBookmark *>::Iterator it = m_bookmarks.begin(); it != m_bookmarks.end(); ++it )
+ {
+ for ( QValueList<Smb4KHostItem *>::ConstIterator i = m_hosts->begin(); i != m_hosts->end(); ++i )
+ {
+ if ( QString::compare( (*it)->workgroup().lower(), (*i)->workgroup().lower() ) != 0 )
+ {
+ // Continue, if the workgroup is not the same:
+ continue;
+ }
+ else
+ {
+ if ( QString::compare( (*it)->host().lower(), (*i)->name().lower() ) != 0 )
+ {
+ // Continue if the host name is not the same:
+ continue;
+ }
+ else
+ {
+ // Set the IP address if it changed:
+ if ( !(*i)->ip().stripWhiteSpace().isEmpty() &&
+ QString::compare( (*it)->ip(), (*i)->ip() ) != 0 )
+ {
+ (*it)->setIP( (*i)->ip() );
+ }
+
+ break;
+ }
+ }
+ }
+ }
+}
+
+#include "smb4kbookmarkhandler.moc"
generated by cgit v1.2.3 (git 2.25.1) at 2024-06-24 19:54:14 +0000