blob: 1b3529bda10bc5d7300f5144fb995de2d9138d24 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
|
/* This file is part of the HandBrake source code.
Homepage: <http://handbrake.fr/>.
It may be used under the terms of the GNU General Public License. */
#import "HBRedirect.h"
@interface HBOutputRedirectListenerEntry : NSObject
{
@public
id<HBOutputRedirectListening> listener;
dispatch_queue_t queue;
}
- (instancetype)initWithListener:(id<HBOutputRedirectListening>)listener queue:(dispatch_queue_t)queue;
@end
@implementation HBOutputRedirectListenerEntry
- (instancetype)initWithListener:(id<HBOutputRedirectListening>)entryListener queue:(dispatch_queue_t)entryQueue
{
self = [super init];
if (self)
{
self->listener = entryListener;
self->queue = entryQueue;
}
return self;
}
@end
@interface HBRedirect ()
/// Set that contains all registered listeners for this output.
@property (nonatomic, readonly) NSMutableSet<HBOutputRedirectListenerEntry *> *listenerEntries;
/// Selector that is called on listeners to forward the output.
@property (nonatomic, readonly) HBRedirectType type;
@end
@implementation HBRedirect
/**
* Returns HBOutputRedirect object used to redirect stdout.
*/
+ (instancetype)stdoutRedirect
{
return [[HBRedirect alloc] initWithType:HBRedirectTypeOutput];
}
/**
* Returns HBOutputRedirect object used to redirect stderr.
*/
+ (instancetype)stderrRedirect
{
return [[HBRedirect alloc] initWithType:HBRedirectTypeError];
}
- (HBOutputRedirectListenerEntry *)entryForListener:(id <HBOutputRedirectListening>)listener
{
for (HBOutputRedirectListenerEntry *entry in _listenerEntries)
{
if (entry->listener == listener)
{
return entry;
}
}
return nil;
}
- (BOOL)containsListener:(id <HBOutputRedirectListening>)listener
{
HBOutputRedirectListenerEntry *entry = [self entryForListener:listener];
return entry && [_listenerEntries containsObject:listener];
}
/**
* Adds specified object as listener for this output. Method @c stdoutRedirect:
* or @c stderrRedirect: of the listener is called to redirect the output.
*/
- (void)addListener:(id <HBOutputRedirectListening>)listener queue:(dispatch_queue_t)queue
{
NSAssert2([listener respondsToSelector:@selector(redirect:type:)], @"Object %@ doesn't respond to selector \"%@\"", listener, NSStringFromSelector(@selector(redirect:type:)));
if (![self containsListener:listener])
{
HBOutputRedirectListenerEntry *entry = [[HBOutputRedirectListenerEntry alloc] initWithListener:listener queue:queue];
[_listenerEntries addObject:entry];
}
if (_listenerEntries.count > 0)
{
[self startRedirect];
}
}
/**
* Stops forwarding for this output to the specified listener object.
*/
- (void)removeListener:(id <HBOutputRedirectListening>)listener
{
HBOutputRedirectListenerEntry *entry = [self entryForListener:listener];
if (entry)
{
[_listenerEntries removeObject:entry];
}
// If last listener is removed, stop redirecting output and autorelease
// self. Remember to set proper global pointer to NULL so the object is
// recreated again when needed.
if (_listenerEntries.count == 0)
{
[self stopRedirect];
}
}
- (instancetype)initWithType:(HBRedirectType)type
{
if (self = [super init])
{
_listenerEntries = [[NSMutableSet alloc] init];
_type = type;
}
return self;
}
- (void)startRedirect {}
- (void)stopRedirect {}
/**
* Called from @c stdoutwrite() and @c stderrwrite() to forward the output to
* listeners.
*/
- (void)forwardOutput:(NSString *)string
{
for (HBOutputRedirectListenerEntry *entry in _listenerEntries)
{
dispatch_async(entry->queue, ^{
[entry->listener redirect:string type:self->_type];
});
}
}
@end
|
