source: trunk/utils/edit.c @ 11791

Last change on this file since 11791 was 11791, checked in by jordan, 11 years ago

(trunk utils) #3964 "transmission-edit replace option cuts off strings" -- fixed.

The trailing text after the last substring match wasn't being retained.

  • Property svn:keywords set to Date Rev Author Id
File size: 9.0 KB
Line 
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: edit.c 11791 2011-01-30 16:40:11Z jordan $
11 */
12
13#include <stdio.h>
14
15#include <event2/buffer.h>
16
17#include <libtransmission/transmission.h>
18#include <libtransmission/bencode.h>
19#include <libtransmission/tr-getopt.h>
20#include <libtransmission/utils.h>
21#include <libtransmission/version.h>
22
23#define MY_NAME "transmission-edit"
24
25static int fileCount = 0;
26static tr_bool showVersion = FALSE;
27static const char ** files = NULL;
28static const char * add = NULL;
29static const char * deleteme = NULL;
30static const char * replace[2] = { NULL, NULL };
31
32static tr_option options[] =
33{
34  { 'a', "add", "Add a tracker's announce URL", "a", 1, "<url>" },
35  { 'd', "delete", "Delete a tracker's announce URL", "d", 1, "<url>" },
36  { 'r', "replace", "Search and replace a substring in the announce URLs", "r", 1, "<old> <new>" },
37  { 'V', "version", "Show version number and exit", "V", 0, NULL },
38  { 0, NULL, NULL, NULL, 0, NULL }
39};
40
41static const char *
42getUsage( void )
43{
44    return "Usage: " MY_NAME " [options] torrent-file(s)";
45}
46
47static int
48parseCommandLine( int argc, const char ** argv )
49{
50    int c;
51    const char * optarg;
52
53    while(( c = tr_getopt( getUsage( ), argc, argv, options, &optarg )))
54    {
55        switch( c )
56        {
57            case 'a': add = optarg;
58                      break;
59            case 'd': deleteme = optarg;
60                      break;
61            case 'r': replace[0] = optarg;
62                      c = tr_getopt( getUsage( ), argc, argv, options, &optarg );
63                      if( c != TR_OPT_UNK ) return 1;
64                      replace[1] = optarg;
65                      break;
66            case 'V': showVersion = TRUE;
67                      break;
68            case TR_OPT_UNK: files[fileCount++] = optarg; break;
69            default: return 1;
70        }
71    }
72
73    return 0;
74}
75
76static tr_bool
77removeURL( tr_benc * metainfo, const char * url )
78{
79    const char * str;
80    tr_benc * announce_list;
81    tr_bool changed = FALSE;
82
83    if( tr_bencDictFindStr( metainfo, "announce", &str ) && !strcmp( str, url ) )
84    {
85        printf( "\tRemoved \"%s\" from \"announce\"\n", str );
86        tr_bencDictRemove( metainfo, "announce" );
87        changed = TRUE;
88    }
89
90    if( tr_bencDictFindList( metainfo, "announce-list", &announce_list ) )
91    {
92        tr_benc * tier;
93        int tierIndex = 0;
94        while(( tier = tr_bencListChild( announce_list, tierIndex )))
95        {
96            tr_benc * node;
97            int nodeIndex = 0;
98            while(( node = tr_bencListChild( tier, nodeIndex )))
99            {
100                if( tr_bencGetStr( node, &str ) && !strcmp( str, url ) )
101                {
102                    printf( "\tRemoved \"%s\" from \"announce-list\" tier #%d\n", str, (tierIndex+1) );
103                    tr_bencListRemove( tier, nodeIndex );
104                    changed = TRUE;
105                }
106                else ++nodeIndex;
107            }
108
109            if( tr_bencListSize( tier ) == 0 )
110            {
111                printf( "\tNo URLs left in tier #%d... removing tier\n", (tierIndex+1) );
112                tr_bencListRemove( announce_list, tierIndex );
113            }
114            else ++tierIndex;
115        }
116
117        if( tr_bencListSize( announce_list ) == 0 )
118        {
119            printf( "\tNo tiers left... removing announce-list\n" );
120            tr_bencDictRemove( metainfo, "announce-list" );
121        }
122    }
123
124    /* if we removed the "announce" field and there's still another track left,
125     * use it as the "announce" field */
126    if( changed && !tr_bencDictFindStr( metainfo, "announce", &str ) )
127    {
128        tr_benc * tier;
129        tr_benc * node;
130
131        if(( tier = tr_bencListChild( announce_list, 0 ))) {
132            if(( node = tr_bencListChild( tier, 0 ))) {
133                if( tr_bencGetStr( node, &str ) ) {
134                    tr_bencDictAddStr( metainfo, "announce", str );
135                    printf( "\tAdded \"%s\" to announce\n", str );
136                }
137            }
138        }
139    }
140
141    return changed;
142}
143
144static char*
145replaceSubstr( const char * str, const char * in, const char * out )
146{
147    char * walk;
148    struct evbuffer * buf = evbuffer_new( );
149    const size_t inlen = strlen( in );
150    const size_t outlen = strlen( out );
151
152    while(( walk = strstr( str, in )))
153    {
154        evbuffer_add( buf, str, walk-str );
155        evbuffer_add( buf, out, outlen );
156        str = walk + inlen;
157    }
158
159    evbuffer_add( buf, str, strlen( str ) );
160
161    return evbuffer_free_to_str( buf );
162}
163
164static tr_bool
165replaceURL( tr_benc * metainfo, const char * in, const char * out )
166{
167    const char * str;
168    tr_benc * announce_list;
169    tr_bool changed = FALSE;
170
171    if( tr_bencDictFindStr( metainfo, "announce", &str ) && strstr( str, in ) )
172    {
173        char * newstr = replaceSubstr( str, in, out );
174        printf( "\tReplaced in \"announce\": \"%s\" --> \"%s\"\n", str, newstr );
175        tr_bencDictAddStr( metainfo, "announce", newstr );
176        tr_free( newstr );
177        changed = TRUE;
178    }
179
180    if( tr_bencDictFindList( metainfo, "announce-list", &announce_list ) )
181    {
182        tr_benc * tier;
183        int tierCount = 0;
184        while(( tier = tr_bencListChild( announce_list, tierCount++ )))
185        {
186            tr_benc * node;
187            int nodeCount = 0;
188            while(( node = tr_bencListChild( tier, nodeCount++ )))
189            {
190                if( tr_bencGetStr( node, &str ) && strstr( str, in ) )
191                {
192                    char * newstr = replaceSubstr( str, in, out );
193                    printf( "\tReplaced in \"announce-list\" tier %d: \"%s\" --> \"%s\"\n", tierCount, str, newstr );
194                    tr_bencFree( node );
195                    tr_bencInitStr( node, newstr, -1 );
196                    tr_free( newstr );
197                    changed = TRUE;
198                }
199            }
200        }
201    }
202
203    return changed;
204}
205
206static tr_bool
207addURL( tr_benc * metainfo, const char * url )
208{
209    const char * str;
210    tr_benc * announce_list;
211    tr_bool changed = FALSE;
212    tr_bool match = FALSE;
213
214    /* maybe add it to "announce" */
215    if( !tr_bencDictFindStr( metainfo, "announce", &str ) )
216    {
217        printf( "\tAdded \"%s\" in \"announce\"\n", url );
218        tr_bencDictAddStr( metainfo, "announce", url );
219        changed = TRUE;
220    }
221
222    /* see if it's already in announce-list */
223    if( tr_bencDictFindList( metainfo, "announce-list", &announce_list ) ) {
224        tr_benc * tier;
225        int tierCount = 0;
226        while(( tier = tr_bencListChild( announce_list, tierCount++ ))) {
227            tr_benc * node;
228            int nodeCount = 0;
229            while(( node = tr_bencListChild( tier, nodeCount++ )))
230                if( tr_bencGetStr( node, &str ) && !strcmp( str, url ) )
231                    match = TRUE;
232        }
233    }
234
235    /* if it's not in announce-list, add it now */
236    if( !match )
237    {
238        tr_benc * tier;
239
240        if( !tr_bencDictFindList( metainfo, "announce-list", &announce_list ) )
241            announce_list = tr_bencDictAddList( metainfo, "announce-list", 1 );
242
243        tier = tr_bencListAddList( announce_list, 1 );
244        tr_bencListAddStr( tier, url );
245        printf( "\tAdded \"%s\" to \"announce-list\" tier %zu\n", url, tr_bencListSize( announce_list ) );
246        changed = TRUE;
247    }
248
249    return changed;
250}
251
252int
253main( int argc, char * argv[] )
254{
255    int i;
256    int changedCount = 0;
257
258    files = tr_new0( const char*, argc );
259
260    tr_setMessageLevel( TR_MSG_ERR );
261
262    if( parseCommandLine( argc, (const char**)argv ) )
263        return EXIT_FAILURE;
264
265    if( showVersion )
266    {
267        fprintf( stderr, MY_NAME" "LONG_VERSION_STRING"\n" );
268        return 0;
269    }
270
271    if( fileCount < 1 )
272    {
273        fprintf( stderr, "ERROR: No torrent files specified.\n" );
274        tr_getopt_usage( MY_NAME, getUsage( ), options );
275        fprintf( stderr, "\n" );
276        return EXIT_FAILURE;
277    }
278
279    if( !add && !deleteme && !replace[0] )
280    {
281        fprintf( stderr, "ERROR: Must specify -a, -d or -r\n" );
282        tr_getopt_usage( MY_NAME, getUsage( ), options );
283        fprintf( stderr, "\n" );
284        return EXIT_FAILURE;
285    }
286
287    for( i=0; i<fileCount; ++i )
288    {
289        tr_benc top;
290        tr_bool changed = FALSE;
291        const char * filename = files[i];
292
293        printf( "%s\n", filename );
294
295        if( tr_bencLoadFile( &top, TR_FMT_BENC, filename ) )
296        {
297            printf( "\tError reading file\n" );
298            continue;
299        }
300
301        if( deleteme != NULL )
302            changed |= removeURL( &top, deleteme );
303
304        if( add != NULL )
305            changed = addURL( &top, add );
306
307        if( replace[0] && replace[1] )
308            changed |= replaceURL( &top, replace[0], replace[1] );
309
310        if( changed )
311        {
312            ++changedCount;
313            tr_bencToFile( &top, TR_FMT_BENC, filename );
314        }
315
316        tr_bencFree( &top );
317    }
318
319    printf( "Changed %d files\n", changedCount );
320
321    tr_free( files );
322    return 0;
323}
Note: See TracBrowser for help on using the repository browser.