1/******************************************************************************
2 * $Id: FileListNode.m 13434 2012-08-13 00:52:04Z livings124 $
3 *
4 * Copyright (c) 2008-2012 Transmission authors and contributors
5 *
6 * Permission is hereby granted, free of charge, to any person obtaining a
7 * copy of this software and associated documentation files (the "Software"),
8 * to deal in the Software without restriction, including without limitation
9 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
10 * and/or sell copies of the Software, and to permit persons to whom the
11 * Software is furnished to do so, subject to the following conditions:
12 *
13 * The above copyright notice and this permission notice shall be included in
14 * all copies or substantial portions of the Software.
15 *
16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
21 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
22 * DEALINGS IN THE SOFTWARE.
23 *****************************************************************************/
24
25#import "FileListNode.h"
26
27@interface FileListNode (Private)
28
29- (id) initWithFolder: (BOOL) isFolder name: (NSString *) name path: (NSString *) path torrent: (Torrent *) torrent;
30
31@end
32
33@implementation FileListNode
34
35#warning remove ivars in header when 64-bit only (or it compiles in 32-bit mode)
36@synthesize name = fName;
37@synthesize path = fPath;
38@synthesize torrent = fTorrent;
39@synthesize size = fSize;
40@synthesize icon = fIcon;
41@synthesize isFolder = fIsFolder;
42
43- (id) initWithFolderName: (NSString *) name path: (NSString *) path torrent: (Torrent *) torrent
44{
45    if ((self = [self initWithFolder: YES name: name path: path torrent: torrent]))
46    {
47        fChildren = [[NSMutableArray alloc] init];
48        fSize = 0;
49    }
50    
51    return self;
52}
53
54- (id) initWithFileName: (NSString *) name path: (NSString *) path size: (uint64_t) size index: (NSUInteger) index torrent: (Torrent *) torrent
55{
56    if ((self = [self initWithFolder: NO name: name path: path torrent: torrent]))
57    {
58        fSize = size;
59        [fIndexes addIndex: index];
60    }
61    
62    return self;
63}
64
65- (void) insertChild: (FileListNode *) child
66{
67    NSAssert(fIsFolder, @"method can only be invoked on folders");
68    
69    [fChildren addObject: child];
70}
71
72- (void) insertIndex: (NSUInteger) index withSize: (uint64_t) size
73{
74    NSAssert(fIsFolder, @"method can only be invoked on folders");
75    
76    [fIndexes addIndex: index];
77    fSize += size;
78}
79
80- (id) copyWithZone: (NSZone *) zone
81{
82    //this object is essentially immutable after initial setup
83    return [self retain];
84}
85
86- (void) dealloc
87{
88    [fName release];
89    [fPath release];
90    [fIndexes release];
91    
92    [fIcon release];
93    
94    [fChildren release];
95    
96    [super dealloc];
97}
98
99- (NSString *) description
100{
101    if (!fIsFolder)
102        return [NSString stringWithFormat: @"%@ (%ld)", fName, [fIndexes firstIndex]];
103    else
104        return [NSString stringWithFormat: @"%@ (folder: %@)", fName, fIndexes];
105}
106
107- (NSIndexSet *) indexes
108{
109    return fIndexes;
110}
111
112- (NSImage *) icon
113{
114    if (!fIcon)
115        fIcon = [[[NSWorkspace sharedWorkspace] iconForFileType: fIsFolder ? NSFileTypeForHFSTypeCode(kGenericFolderIcon)
116                                                                            : [fName pathExtension]] retain];
117    return fIcon;
118}
119
120- (NSMutableArray *) children
121{
122    NSAssert(fIsFolder, @"method can only be invoked on folders");
123    
124    return fChildren;
125}
126
127@end
128
129@implementation FileListNode (Private)
130
131- (id) initWithFolder: (BOOL) isFolder name: (NSString *) name path: (NSString *) path torrent: (Torrent *) torrent
132{
133    if ((self = [super init]))
134    {
135        fIsFolder = isFolder;
136        fName = [name retain];
137        fPath = [path retain];
138        
139        fIndexes = [[NSMutableIndexSet alloc] init];
140        
141        fTorrent = torrent;
142    }
143    
144    return self;
145}
146
147@end
148