source: trunk/gtk/makemeta-ui.c @ 10970

Last change on this file since 10970 was 10970, checked in by charles, 12 years ago

(trunk T) #3404 "crash when creating a .torrent containing empty content" -- fixed

  • Property svn:keywords set to Date Rev Author Id
File size: 16.2 KB
Line 
1/*
2 * This file Copyright (C) 2007-2010 Mnemosyne LLC
3 *
4 * This file is licensed by the GPL version 2.  Works owned by the
5 * Transmission project are granted a special exemption to clause 2(b)
6 * so that the bulk of its code can remain under the MIT license.
7 * This exemption does not extend to derived works not owned by
8 * the Transmission project.
9 *
10 * $Id: makemeta-ui.c 10970 2010-07-07 22:50:18Z charles $
11 */
12
13#include <string.h>
14
15#include <glib/gi18n.h>
16#include <gtk/gtk.h>
17
18#include <libtransmission/transmission.h>
19#include <libtransmission/makemeta.h>
20#include <libtransmission/utils.h> /* tr_formatter_mem_B() */
21
22#include "conf.h"
23#include "hig.h"
24#include "makemeta-ui.h"
25#include "tr-core.h"
26#include "tr-prefs.h"
27#include "util.h"
28
29#define FILE_CHOSEN_KEY "file-is-chosen"
30
31typedef struct
32{
33    char * target;
34    guint progress_tag;
35    GtkWidget * file_chooser;
36    GtkWidget * folder_chooser;
37    GtkWidget * pieces_lb;
38    GtkWidget * destination_chooser;
39    GtkWidget * comment_check;
40    GtkWidget * comment_entry;
41    GtkWidget * private_check;
42    GtkWidget * progress_label;
43    GtkWidget * progress_bar;
44    GtkWidget * progress_dialog;
45    GtkWidget * dialog;
46    GtkTextBuffer * announce_text_buffer;
47    TrCore * core;
48    tr_metainfo_builder *  builder;
49}
50MakeMetaUI;
51
52static void
53freeMetaUI( gpointer p )
54{
55    MakeMetaUI * ui = p;
56
57    tr_metaInfoBuilderFree( ui->builder );
58    g_free( ui->target );
59    memset( ui, ~0, sizeof( MakeMetaUI ) );
60    g_free( ui );
61}
62
63static gboolean
64onProgressDialogRefresh( gpointer data )
65{
66    char * str = NULL;
67    MakeMetaUI * ui = data;
68    const tr_metainfo_builder * b = ui->builder;
69    GtkDialog * d = GTK_DIALOG( ui->progress_dialog );
70    GtkProgressBar * p = GTK_PROGRESS_BAR( ui->progress_bar );
71    const double fraction = b->pieceCount ? ((double)b->pieceIndex / b->pieceCount) : 0;
72    char * base = g_path_get_basename( b->top );
73
74    /* progress label */
75    if( !b->isDone )
76        str = g_strdup_printf( _( "Creating \"%s\"" ), base );
77    else if( b->result == TR_MAKEMETA_OK )
78        str = g_strdup_printf( _( "Created \"%s\"!" ), base );
79    else if( b->result == TR_MAKEMETA_URL )
80        str = g_strdup_printf( _( "Error: invalid announce URL \"%s\"" ), b->errfile );
81    else if( b->result == TR_MAKEMETA_CANCELLED )
82        str = g_strdup_printf( _( "Cancelled" ) );
83    else if( b->result == TR_MAKEMETA_IO_READ )
84        str = g_strdup_printf( _( "Error reading \"%s\": %s" ), b->errfile, g_strerror( b->my_errno ) );
85    else if( b->result == TR_MAKEMETA_IO_WRITE )
86        str = g_strdup_printf( _( "Error writing \"%s\": %s" ), b->errfile, g_strerror( b->my_errno ) );
87    else
88        g_assert_not_reached( );
89
90    if( str != NULL ) {
91        gtk_label_set_text( GTK_LABEL( ui->progress_label ), str );
92        g_free( str );
93    }
94
95    /* progress bar */
96    if( !b->pieceIndex )
97        str = g_strdup( "" );
98    else {
99        char sizebuf[128];
100        tr_strlsize( sizebuf, (uint64_t)b->pieceIndex *
101                              (uint64_t)b->pieceSize, sizeof( sizebuf ) );
102        /* how much data we've scanned through to generate checksums */
103        str = g_strdup_printf( _( "Scanned %s" ), sizebuf );
104    }
105    gtk_progress_bar_set_fraction( p, fraction );
106    gtk_progress_bar_set_text( p, str );
107    g_free( str );
108
109    /* buttons */
110    gtk_dialog_set_response_sensitive( d, GTK_RESPONSE_CANCEL, !b->isDone );
111    gtk_dialog_set_response_sensitive( d, GTK_RESPONSE_CLOSE, b->isDone );
112    gtk_dialog_set_response_sensitive( d, GTK_RESPONSE_ACCEPT, b->isDone && !b->result );
113
114    g_free( base );
115    return TRUE;
116}
117
118static void
119onProgressDialogDestroyed( gpointer data, GObject * dead UNUSED )
120{
121    MakeMetaUI * ui = data;
122    g_source_remove( ui->progress_tag );
123}
124
125static void
126addTorrent( MakeMetaUI * ui )
127{
128    char * path;
129    const tr_metainfo_builder * b = ui->builder;
130    tr_ctor * ctor = tr_ctorNew( tr_core_session( ui->core ) );
131
132    tr_ctorSetMetainfoFromFile( ctor, ui->target );
133
134    path = g_path_get_dirname( b->top );
135    tr_ctorSetDownloadDir( ctor, TR_FORCE, path );
136    g_free( path );
137
138    tr_core_add_ctor( ui->core, ctor );
139}
140
141static void
142onProgressDialogResponse( GtkDialog * d, int response, gpointer data )
143{
144    MakeMetaUI * ui = data;
145
146    switch( response )
147    {
148        case GTK_RESPONSE_CANCEL:
149            ui->builder->abortFlag = TRUE;
150            gtk_widget_destroy( GTK_WIDGET( d ) );
151            break;
152        case GTK_RESPONSE_ACCEPT:
153            addTorrent( ui );
154            /* fall-through */
155        case GTK_RESPONSE_CLOSE:
156            gtk_widget_destroy( ui->builder->result ? GTK_WIDGET( d ) : ui->dialog );
157            break;
158        default:
159            g_assert( 0 && "unhandled response" );
160    }
161}
162
163static void
164makeProgressDialog( GtkWidget * parent, MakeMetaUI * ui )
165{
166    GtkWidget *d, *l, *w, *v, *fr;
167
168    d = gtk_dialog_new_with_buttons( _( "New Torrent" ),
169            GTK_WINDOW( parent ),
170            GTK_DIALOG_MODAL|GTK_DIALOG_DESTROY_WITH_PARENT,
171            GTK_STOCK_CANCEL, GTK_RESPONSE_CANCEL,
172            GTK_STOCK_CLOSE, GTK_RESPONSE_CLOSE,
173            GTK_STOCK_ADD, GTK_RESPONSE_ACCEPT,
174            NULL );
175    ui->progress_dialog = d;
176    g_signal_connect( d, "response", G_CALLBACK( onProgressDialogResponse ), ui );
177
178    fr = gtk_frame_new( NULL );
179    gtk_container_set_border_width( GTK_CONTAINER( fr ), GUI_PAD_BIG );
180    gtk_frame_set_shadow_type( GTK_FRAME( fr ), GTK_SHADOW_NONE );
181    v = gtk_vbox_new( TRUE, GUI_PAD );
182    gtk_container_add( GTK_CONTAINER( fr ), v );
183
184    l = gtk_label_new( _( "Creating torrent..." ) );
185    gtk_misc_set_alignment( GTK_MISC( l ), 0.0, 0.5 );
186    gtk_label_set_justify( GTK_LABEL( l ), GTK_JUSTIFY_LEFT );
187    ui->progress_label = l;
188    gtk_box_pack_start( GTK_BOX( v ), l, FALSE, FALSE, 0 );
189
190    w = gtk_progress_bar_new( );
191    ui->progress_bar = w;
192    gtk_box_pack_start( GTK_BOX( v ), w, FALSE, FALSE, 0 );
193
194    ui->progress_tag = gtr_timeout_add_seconds( SECONDARY_WINDOW_REFRESH_INTERVAL_SECONDS, onProgressDialogRefresh, ui );
195    g_object_weak_ref( G_OBJECT( d ), onProgressDialogDestroyed, ui );
196    onProgressDialogRefresh( ui );
197
198    gtk_box_pack_start( GTK_BOX( GTK_DIALOG( d )->vbox ), fr, TRUE, TRUE, 0 );
199    gtk_widget_show_all( d );
200}
201
202static void
203onResponse( GtkDialog* d, int response, gpointer user_data )
204{
205    MakeMetaUI * ui = user_data;
206
207    if( response == GTK_RESPONSE_ACCEPT )
208    {
209        if( ui->builder != NULL )
210        {
211            int i;
212            int n;
213            int tier;
214            GtkTextIter start, end;
215            char * dir;
216            char * base;
217            char * tracker_text;
218            char ** tracker_strings;
219            GtkEntry * c_entry = GTK_ENTRY( ui->comment_entry );
220            GtkToggleButton * p_check = GTK_TOGGLE_BUTTON( ui->private_check );
221            GtkToggleButton * c_check = GTK_TOGGLE_BUTTON( ui->comment_check );
222            const char * comment = gtk_entry_get_text( c_entry );
223            const gboolean isPrivate = gtk_toggle_button_get_active( p_check );
224            const gboolean useComment = gtk_toggle_button_get_active( c_check );
225            tr_tracker_info * trackers;
226
227            /* destination file */
228            dir = gtk_file_chooser_get_filename(
229                      GTK_FILE_CHOOSER( ui->destination_chooser ) );
230            base = g_path_get_basename( ui->builder->top );
231            g_free( ui->target );
232            ui->target = g_strdup_printf( "%s/%s.torrent", dir, base );
233
234            /* build the array of trackers */
235            gtk_text_buffer_get_bounds( ui->announce_text_buffer, &start, &end );
236            tracker_text = gtk_text_buffer_get_text( ui->announce_text_buffer,
237                                                     &start, &end, FALSE );
238            tracker_strings = g_strsplit( tracker_text, "\n", 0 );
239            for( i=0; tracker_strings[i]; )
240                ++i;
241            trackers = g_new0( tr_tracker_info, i );
242            for( i=n=tier=0; tracker_strings[i]; ++i ) {
243                const char * str = tracker_strings[i];
244                if( !*str )
245                    ++tier;
246                else {
247                    trackers[n].tier = tier;
248                    trackers[n].announce = tracker_strings[i];
249                    ++n;
250                }
251            }
252
253            /* build the .torrent */
254            makeProgressDialog( GTK_WIDGET( d ), ui );
255            tr_makeMetaInfo( ui->builder, ui->target, trackers, n,
256                             useComment ? comment : NULL, isPrivate );
257
258            /* cleanup */
259            g_free( trackers );
260            g_strfreev( tracker_strings );
261            g_free( tracker_text );
262            g_free( base );
263            g_free( dir );
264        }
265    }
266    else if( response == GTK_RESPONSE_CLOSE )
267    {
268        gtk_widget_destroy( GTK_WIDGET( d ) );
269    }
270}
271
272/***
273****
274***/
275
276static void
277onSourceToggled( GtkToggleButton * tb, gpointer user_data )
278{
279    gtk_widget_set_sensitive( GTK_WIDGET( user_data ),
280                              gtk_toggle_button_get_active( tb ) );
281}
282
283static void
284updatePiecesLabel( MakeMetaUI * ui )
285{
286    const tr_metainfo_builder * builder = ui->builder;
287    const char * filename = builder ? builder->top : NULL;
288    GString * gstr = g_string_new( NULL );
289
290    g_string_append( gstr, "<i>" );
291    if( !filename )
292    {
293        g_string_append( gstr, _( "No source selected" ) );
294    }
295    else
296    {
297        char buf[128];
298        tr_strlsize( buf, builder->totalSize, sizeof( buf ) );
299        g_string_append_printf( gstr, ngettext( "%1$s; %2$'d File",
300                                                "%1$s; %2$'d Files",
301                                                builder->fileCount ),
302                                buf, builder->fileCount );
303        g_string_append( gstr, "; " );
304 
305        tr_formatter_mem_B( buf, builder->pieceSize, sizeof( buf ) );
306        g_string_append_printf( gstr, ngettext( "%1$'d Piece @ %2$s",
307                                                "%1$'d Pieces @ %2$s",
308                                                builder->pieceCount ),
309                                      builder->pieceCount, buf );
310    }
311    g_string_append( gstr, "</i>" );
312    gtk_label_set_markup ( GTK_LABEL( ui->pieces_lb ), gstr->str );
313    g_string_free( gstr, TRUE );
314}
315
316static void
317setFilename( MakeMetaUI * ui, const char * filename )
318{
319    if( ui->builder ) {
320        tr_metaInfoBuilderFree( ui->builder );
321        ui->builder = NULL;
322    }
323
324    if( filename )
325        ui->builder = tr_metaInfoBuilderCreate( filename );
326
327    updatePiecesLabel( ui );
328}
329
330static void
331onChooserChosen( GtkFileChooser * chooser, gpointer user_data )
332{
333    char * filename;
334    MakeMetaUI * ui = user_data;
335
336    g_object_set_data( G_OBJECT( chooser ), FILE_CHOSEN_KEY,
337                       GINT_TO_POINTER( TRUE ) );
338
339    filename = gtk_file_chooser_get_filename( chooser );
340    setFilename( ui, filename );
341    g_free( filename );
342}
343
344static void
345onSourceToggled2( GtkToggleButton * tb, GtkWidget * chooser, MakeMetaUI * ui )
346{
347    if( gtk_toggle_button_get_active( tb ) )
348    {
349        if( g_object_get_data( G_OBJECT( chooser ), FILE_CHOSEN_KEY ) != NULL )
350            onChooserChosen( GTK_FILE_CHOOSER( chooser ), ui );
351        else
352            setFilename( ui, NULL );
353    }
354}
355static void
356onFolderToggled( GtkToggleButton * tb, gpointer data )
357{
358    MakeMetaUI * ui = data;
359    onSourceToggled2( tb, ui->folder_chooser, ui );
360}
361static void
362onFileToggled( GtkToggleButton * tb, gpointer data )
363{
364    MakeMetaUI * ui = data;
365    onSourceToggled2( tb, ui->file_chooser, ui );
366}
367
368static const char *
369getDefaultSavePath( void )
370{
371    const char * path;
372#if GLIB_CHECK_VERSION( 2,14,0 )
373    path = g_get_user_special_dir( G_USER_DIRECTORY_DESKTOP );
374#else
375    path = g_get_home_dir( );
376#endif
377    return path;
378}
379
380GtkWidget*
381make_meta_ui( GtkWindow  * parent, TrCore * core )
382{
383    int row = 0;
384    const char * str;
385    GtkWidget * d, *t, *w, *l, *fr, *sw, *v;
386    GSList * slist;
387    MakeMetaUI * ui = g_new0 ( MakeMetaUI, 1 );
388
389    ui->core = core;
390
391    d = gtk_dialog_new_with_buttons( _( "New Torrent" ),
392                                     parent,
393                                     GTK_DIALOG_DESTROY_WITH_PARENT |
394                                     GTK_DIALOG_NO_SEPARATOR,
395                                     GTK_STOCK_CLOSE, GTK_RESPONSE_CLOSE,
396                                     GTK_STOCK_NEW, GTK_RESPONSE_ACCEPT,
397                                     NULL );
398    ui->dialog = d;
399    g_signal_connect( d, "response", G_CALLBACK( onResponse ), ui );
400    g_object_set_data_full( G_OBJECT( d ), "ui", ui, freeMetaUI );
401
402    t = hig_workarea_create ( );
403
404    hig_workarea_add_section_title ( t, &row, _( "Files" ) );
405
406        str = _( "Sa_ve to:" );
407        w = gtk_file_chooser_button_new( NULL, GTK_FILE_CHOOSER_ACTION_SELECT_FOLDER );
408        gtk_file_chooser_set_current_folder( GTK_FILE_CHOOSER( w ), getDefaultSavePath( ) );
409        ui->destination_chooser = w;
410        hig_workarea_add_row( t, &row, str, w, NULL );
411
412        l = gtk_radio_button_new_with_mnemonic( NULL, _( "Source F_older:" ) );
413        gtk_toggle_button_set_active( GTK_TOGGLE_BUTTON( l ), FALSE );
414        w = gtk_file_chooser_button_new( NULL, GTK_FILE_CHOOSER_ACTION_SELECT_FOLDER );
415        g_signal_connect( l, "toggled", G_CALLBACK( onFolderToggled ), ui );
416        g_signal_connect( l, "toggled", G_CALLBACK( onSourceToggled ), w );
417        g_signal_connect( w, "selection-changed", G_CALLBACK( onChooserChosen ), ui );
418        ui->folder_chooser = w;
419        gtk_widget_set_sensitive( GTK_WIDGET( w ), FALSE );
420        hig_workarea_add_row_w( t, &row, l, w, NULL );
421
422        slist = gtk_radio_button_get_group( GTK_RADIO_BUTTON( l ) ),
423        l = gtk_radio_button_new_with_mnemonic( slist, _( "Source _File:" ) );
424        gtk_toggle_button_set_active( GTK_TOGGLE_BUTTON( l ), TRUE );
425        w = gtk_file_chooser_button_new( NULL, GTK_FILE_CHOOSER_ACTION_OPEN );
426        g_signal_connect( l, "toggled", G_CALLBACK( onFileToggled ), ui );
427        g_signal_connect( l, "toggled", G_CALLBACK( onSourceToggled ), w );
428        g_signal_connect( w, "selection-changed", G_CALLBACK( onChooserChosen ), ui );
429        ui->file_chooser = w;
430        hig_workarea_add_row_w( t, &row, l, w, NULL );
431
432        w = gtk_label_new( NULL );
433        ui->pieces_lb = w;
434        gtk_label_set_markup( GTK_LABEL( w ), _( "<i>No source selected</i>" ) );
435        hig_workarea_add_row( t, &row, NULL, w, NULL );
436
437    hig_workarea_add_section_divider( t, &row );
438    hig_workarea_add_section_title ( t, &row, _( "Properties" ) );
439
440        str = _( "_Trackers:" );
441        v = gtk_vbox_new( FALSE, GUI_PAD_SMALL );
442        ui->announce_text_buffer = gtk_text_buffer_new( NULL );
443        w = gtk_text_view_new_with_buffer( ui->announce_text_buffer );
444        gtk_widget_set_size_request( w, -1, 80 );
445        sw = gtk_scrolled_window_new( NULL, NULL );
446        gtk_scrolled_window_set_policy( GTK_SCROLLED_WINDOW( sw ),
447                                        GTK_POLICY_AUTOMATIC,
448                                        GTK_POLICY_AUTOMATIC );
449        gtk_container_add( GTK_CONTAINER( sw ), w );
450        fr = gtk_frame_new( NULL );
451        gtk_frame_set_shadow_type( GTK_FRAME( fr ), GTK_SHADOW_IN );
452        gtk_container_add( GTK_CONTAINER( fr ), sw );
453        gtk_box_pack_start( GTK_BOX( v ), fr, TRUE, TRUE, 0 );
454        l = gtk_label_new( NULL );
455        gtk_label_set_markup( GTK_LABEL( l ), _( "To add a backup URL, add it on the line after the primary URL.\n"
456                                                 "To add another primary URL, add it after a blank line." ) );
457        gtk_label_set_justify( GTK_LABEL( l ), GTK_JUSTIFY_LEFT );
458        gtk_misc_set_alignment( GTK_MISC( l ), 0.0, 0.5 );
459        gtk_box_pack_start( GTK_BOX( v ), l, FALSE, FALSE, 0 );
460        hig_workarea_add_tall_row( t, &row, str, v, NULL );
461
462        l = gtk_check_button_new_with_mnemonic( _( "Co_mment:" ) );
463        ui->comment_check = l;
464        gtk_toggle_button_set_active( GTK_TOGGLE_BUTTON( l ), FALSE );
465        w = gtk_entry_new( );
466        ui->comment_entry = w;
467        gtk_widget_set_sensitive( GTK_WIDGET( w ), FALSE );
468        g_signal_connect( l, "toggled", G_CALLBACK( onSourceToggled ), w );
469        hig_workarea_add_row_w( t, &row, l, w, NULL );
470
471        w = hig_workarea_add_wide_checkbutton( t, &row, _( "_Private torrent" ), FALSE );
472        ui->private_check = w;
473
474    hig_workarea_finish( t, &row );
475    gtk_box_pack_start( GTK_BOX( GTK_DIALOG( d )->vbox ), t, TRUE, TRUE, 0 );
476
477    return d;
478}
Note: See TracBrowser for help on using the repository browser.