release/src/router/transmission/qt/add-data.cc

   1 /*
   2  * This file Copyright (C) Mnemosyne LLC
   3  *
   4  * This program is free software; you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License version 2
   6  * as published by the Free Software Foundation.
   7  *
   8  * http://www.gnu.org/licenses/old-licenses/gpl-2.0.html
   9  *
  10  * $Id: add-data.cc 14150 2013-07-27 21:58:14Z jordan $
  11  */
  12
  13 #include <QFile>
  14
  15 #include <libtransmission/transmission.h>
  16 #include <libtransmission/utils.h> // tr_base64_encode()
  17
  18 #include "add-data.h"
  19 #include "utils.h"
  20
  21 int
  22 AddData :: set( const QString& key )
  23 {
  24     if( Utils::isMagnetLink( key ) )
  25     {
  26         magnet = key;
  27         type = MAGNET;
  28     }
  29     else if ( Utils::isUriWithSupportedScheme( key ) )
  30     {
  31         url = key;
  32         type = URL;
  33     }
  34     else if( QFile(key).exists( ) )
  35     {
  36         filename = key;
  37         type = FILENAME;
  38
  39         QFile file( key );
  40         file.open( QIODevice::ReadOnly );
  41         metainfo = file.readAll( );
  42         file.close( );
  43     }
  44     else if( Utils::isHexHashcode( key ) )
  45     {
  46         magnet = QString::fromUtf8("magnet:?xt=urn:btih:") + key;
  47         type = MAGNET;
  48     }
  49     else
  50     {
  51         int len;
  52         char * raw = tr_base64_decode( key.toUtf8().constData(), key.toUtf8().size(), &len );
  53         if( raw ) {
  54             metainfo.append( raw, len );
  55             tr_free( raw );
  56             type = METAINFO;
  57         }
  58         else type = NONE;
  59     }
  60
  61     return type;
  62 }
  63
  64 QByteArray
  65 AddData :: toBase64( ) const
  66 {
  67     QByteArray ret;
  68
  69     if( !metainfo.isEmpty( ) )
  70     {
  71         int len = 0;
  72         char * b64 = tr_base64_encode( metainfo.constData(), metainfo.size(), &len );
  73         ret = QByteArray( b64, len );
  74         tr_free( b64 );
  75     }
  76
  77     return ret;
  78 }
  79
  80 QString
  81 AddData :: readableName( ) const
  82 {
  83     QString ret;
  84
  85     switch( type )
  86     {
  87         case FILENAME: ret = filename; break;
  88
  89         case MAGNET: ret = magnet; break;
  90
  91         case URL: ret = url.toString(); break;
  92
  93         case METAINFO: {
  94             tr_info inf;
  95             tr_ctor * ctor = tr_ctorNew( NULL );
  96             tr_ctorSetMetainfo( ctor, (const uint8_t*)metainfo.constData(), metainfo.size() );
  97             if( tr_torrentParse( ctor, &inf ) == TR_PARSE_OK  ) {
  98                 ret = QString::fromUtf8( inf.name ); // metainfo is required to be UTF-8
  99                 tr_metainfoFree( &inf );
 100             }
 101             tr_ctorFree( ctor );
 102             break;
 103         }
 104     }
 105
 106    return ret;
 107 }