1/******************************************************************************
2 * Copyright (c) 2006-2012 Transmission authors and contributors
3 *
4 * Permission is hereby granted, free of charge, to any person obtaining a
5 * copy of this software and associated documentation files (the "Software"),
6 * to deal in the Software without restriction, including without limitation
7 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
8 * and/or sell copies of the Software, and to permit persons to whom the
9 * Software is furnished to do so, subject to the following conditions:
10 *
11 * The above copyright notice and this permission notice shall be included in
12 * all copies or substantial portions of the Software.
13 *
14 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
15 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
16 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
17 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
18 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
19 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
20 * DEALINGS IN THE SOFTWARE.
21 *****************************************************************************/
22
23#import "Badger.h"
24#import "BadgeView.h"
25#import "NSStringAdditions.h"
26#import "Torrent.h"
27
28@implementation Badger
29
30- (id) initWithLib: (tr_session *) lib
31{
32    if ((self = [super init]))
33    {
34        fLib = lib;
35
36        BadgeView * view = [[BadgeView alloc] initWithLib: lib];
37        [[NSApp dockTile] setContentView: view];
38
39        fHashes = [[NSMutableSet alloc] init];
40    }
41
42    return self;
43}
44
45
46- (void) updateBadgeWithDownload: (CGFloat) downloadRate upload: (CGFloat) uploadRate
47{
48    const CGFloat displayDlRate = [[NSUserDefaults standardUserDefaults] boolForKey: @"BadgeDownloadRate"]
49                                    ? downloadRate : 0.0;
50    const CGFloat displayUlRate = [[NSUserDefaults standardUserDefaults] boolForKey: @"BadgeUploadRate"]
51                                    ? uploadRate : 0.0;
52
53    //only update if the badged values change
54    if ([(BadgeView *)[[NSApp dockTile] contentView] setRatesWithDownload: displayDlRate upload: displayUlRate])
55        [[NSApp dockTile] display];
56}
57
58- (void) addCompletedTorrent: (Torrent *) torrent
59{
60    NSParameterAssert(torrent != nil);
61
62    [fHashes addObject: [torrent hashString]];
63    [[NSApp dockTile] setBadgeLabel: [NSString formattedUInteger: [fHashes count]]];
64}
65
66- (void) removeTorrent: (Torrent *) torrent
67{
68    if ([fHashes member: [torrent hashString]])
69    {
70        [fHashes removeObject: [torrent hashString]];
71        if ([fHashes count] > 0)
72            [[NSApp dockTile] setBadgeLabel: [NSString formattedUInteger: [fHashes count]]];
73        else
74            [[NSApp dockTile] setBadgeLabel: @""];
75    }
76}
77
78- (void) clearCompleted
79{
80    if ([fHashes count] > 0)
81    {
82        [fHashes removeAllObjects];
83        [[NSApp dockTile] setBadgeLabel: @""];
84    }
85}
86
87- (void) setQuitting
88{
89    [self clearCompleted];
90    [(BadgeView *)[[NSApp dockTile] contentView] setQuitting];
91    [[NSApp dockTile] display];
92}
93
94@end
95