Source/ESBlockingPlugin.m

   1 /*
   2  * Adium is the legal property of its developers, whose names are listed in the copyright file included
   3  * with this source distribution.
   4  *
   5  * This program is free software; you can redistribute it and/or modify it under the terms of the GNU
   6  * General Public License as published by the Free Software Foundation; either version 2 of the License,
   7  * or (at your option) any later version.
   8  *
   9  * This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even
  10  * the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General
  11  * Public License for more details.
  12  *
  13  * You should have received a copy of the GNU General Public License along with this program; if not,
  14  * write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  15  */
  16
  17 #import "AIAccountController.h"
  18 #import "AIContactController.h"
  19 #import "AIMenuController.h"
  20 #import "ESBlockingPlugin.h"
  21 #import <AIUtilities/AIMenuAdditions.h>
  22 #import <Adium/AIAccount.h>
  23 #import <Adium/AIListContact.h>
  24 #import <Adium/AIMetaContact.h>
  25
  26 #define BLOCK_CONTACT   AILocalizedString(@"Block","Block Contact menu item")
  27 #define UNBLOCK_CONTACT AILocalizedString(@"Unblock","Unblock Contact menu item")
  28
  29 @interface ESBlockingPlugin(PRIVATE)
  30 - (void)_blockContact:(AIListContact *)contact unblock:(BOOL)unblock;
  31 - (BOOL)_searchPrivacyListsForListContact:(AIListContact *)contact withDesiredResult:(BOOL)desiredResult;
  32 @end
  33
  34 @implementation ESBlockingPlugin
  35
  36 - (void)installPlugin
  37 {
  38         //Install the Block menu items
  39         blockContactMenuItem = [[NSMenuItem alloc] initWithTitle:BLOCK_CONTACT
  40                                                                                                                  target:self
  41                                                                                                                  action:@selector(blockContact:)
  42                                                                                                   keyEquivalent:@""];
  43         [[adium menuController] addMenuItem:blockContactMenuItem toLocation:LOC_Contact_NegativeAction];
  44
  45     //Add our get info contextual menu items
  46     blockContactContextualMenuItem = [[NSMenuItem alloc] initWithTitle:BLOCK_CONTACT
  47                                                                                                                                 target:self
  48                                                                                                                                 action:@selector(blockContact:)
  49                                                                                                                  keyEquivalent:@""];
  50     [[adium menuController] addContextualMenuItem:blockContactContextualMenuItem toLocation:Context_Contact_NegativeAction];
  51 }
  52
  53 - (void)uninstallPlugin
  54 {
  55         [blockContactMenuItem release];
  56         [blockContactContextualMenuItem release];
  57 }
  58
  59
  60 - (IBAction)blockContact:(id)sender
  61 {
  62         AIListObject *object;
  63         BOOL unblock;
  64
  65         if(sender == blockContactMenuItem){
  66                 object = [[adium contactController] selectedListObject];
  67         }else{
  68                 object = [[adium menuController] currentContextMenuObject];
  69         }
  70
  71         unblock = [[sender title] isEqualToString:UNBLOCK_CONTACT];
  72
  73         //Don't do groups
  74         if([object isKindOfClass:[AIListContact class]]){
  75                 AIListContact *contact = (AIListContact *)object;
  76                 NSString *format = unblock
  77                                                  ? AILocalizedString(@"Are you sure you want to unblock %@?",nil)
  78                                                  : AILocalizedString(@"Are you sure you want to block %@?",nil);
  79
  80                 if(NSRunAlertPanel([NSString stringWithFormat:format, [contact displayName]],
  81                                    @"",
  82                                    AILocalizedString(@"OK",nil),
  83                                    AILocalizedString(@"Cancel",nil),
  84                                    nil)
  85                                 == NSAlertDefaultReturn){
  86
  87                         //Handle metas
  88                         if([object isKindOfClass:[AIMetaContact class]]){
  89                                 AIMetaContact *meta = (AIMetaContact *)object;
  90
  91                                 //Enumerate over the various list contacts contained
  92                                 NSEnumerator *enumerator = [[meta listContacts] objectEnumerator];
  93                                 AIListContact *containedContact = nil;
  94
  95                                 while((containedContact = [enumerator nextObject])){
  96                                         AIAccount <AIAccount_Privacy> *acct = [containedContact account];
  97                                         if([acct conformsToProtocol:@protocol(AIAccount_Privacy)]) {
  98                                                 [self _blockContact:contact unblock:unblock];
  99                                         } else {
 100                                                 NSLog(@"Account %@ does not support blocking (contact %@ not blocked on this account)", acct, containedContact);
 101                                         }
 102                                 }
 103                         }else{
 104                                 AIListContact *contact = (AIListContact *)object;
 105                                 AIAccount <AIAccount_Privacy> *acct = [contact account];
 106                                 if([acct conformsToProtocol:@protocol(AIAccount_Privacy)]) {
 107                                         [self _blockContact:contact unblock:unblock];
 108                                 } else {
 109                                         NSLog(@"Account %@ does not support blocking (contact %@ not blocked on this account)", acct, contact);
 110                                 }
 111                         }
 112                 }
 113         }
 114 }
 115
 116 - (BOOL)validateMenuItem:(id <NSMenuItem>)menuItem
 117 {
 118         AIListObject *object;
 119         BOOL unblock = [[menuItem title] isEqualToString:UNBLOCK_CONTACT];
 120         BOOL anyAccount = NO;
 121
 122         if(menuItem == blockContactMenuItem){
 123                 object = [[adium contactController] selectedListObject];
 124         }else{
 125                 object = [[adium menuController] currentContextMenuObject];
 126         }
 127
 128         //Don't do groups
 129         if([object isKindOfClass:[AIListContact class]]){
 130                 //Handle metas
 131                 if([object isKindOfClass:[AIMetaContact class]]){
 132                         AIMetaContact *meta = (AIMetaContact *)object;
 133
 134                         //Enumerate over the various list contacts contained
 135                         NSEnumerator *enumerator = [[meta listContacts] objectEnumerator];
 136                         AIListContact *contact = nil;
 137
 138                         while((contact = [enumerator nextObject])){
 139                                 AIAccount <AIAccount_Privacy> *acct = [contact account];
 140                                 if([acct conformsToProtocol:@protocol(AIAccount_Privacy)]){
 141                                         anyAccount = YES;
 142                                         if([[acct listObjectIDsOnPrivacyList:PRIVACY_DENY] containsObject:[contact UID]]){
 143                                                 //title: "Unblock"; enabled
 144                                                 if(!unblock){
 145                                                         [menuItem setTitle:UNBLOCK_CONTACT];
 146                                                 }
 147                                                 return YES;
 148                                         }
 149                                 }
 150                         }
 151                         if(anyAccount){
 152                                 //title: "Block"; enabled
 153                                 if(unblock){
 154                                         [menuItem setTitle:BLOCK_CONTACT];
 155                                 }
 156                                 return YES;
 157                         }else{
 158                                 //title: "Block"; disabled
 159                                 [menuItem setTitle:BLOCK_CONTACT];
 160                                 return NO;
 161                         }
 162                 }else{
 163                         AIListContact *contact = (AIListContact *)object;
 164                         AIAccount <AIAccount_Privacy> *acct = [contact account];
 165                         if([acct conformsToProtocol:@protocol(AIAccount_Privacy)]){
 166                                 if([[acct listObjectIDsOnPrivacyList:PRIVACY_DENY] containsObject:[contact UID]]){
 167                                         //title: "Unblock"; enabled
 168                                         if(!unblock){
 169                                                 [menuItem setTitle:UNBLOCK_CONTACT];
 170                                         }
 171                                         return YES;
 172                                 }else{
 173                                         //title: "Block"; enabled
 174                                         if(unblock){
 175                                                 [menuItem setTitle:BLOCK_CONTACT];
 176                                         }
 177                                         return YES;
 178                                 }
 179                         }else{
 180                                 //title: "Block"; disabled
 181                                 [menuItem setTitle:BLOCK_CONTACT];
 182                                 return NO;
 183                         }
 184                 }
 185         }
 186         return NO;
 187 }
 188
 189 #pragma mark Private
 190 //Private --------------------------------------------------------------------------------------------------------------
 191
 192 - (void)_blockContact:(AIListContact *)contact unblock:(BOOL)unblock
 193 {
 194         //We want to block on all accounts with the same service class. If you want someone gone, you want 'em GONE.
 195         NSEnumerator *enumerator = [[[adium accountController] accountsWithServiceClassOfService:[contact service]] objectEnumerator];
 196         AIAccount *account = nil;
 197
 198         while((account = [enumerator nextObject])){
 199                 if([account conformsToProtocol:@protocol(AIAccount_Privacy)]){
 200                         AIAccount <AIAccount_Privacy> *privacyAccount = (AIAccount <AIAccount_Privacy> *)account;
 201                         if(unblock){
 202                                 if([[privacyAccount listObjectIDsOnPrivacyList:PRIVACY_DENY] containsObject:[contact UID]]){
 203                                         [privacyAccount removeListObject:contact fromPrivacyList:PRIVACY_DENY];
 204                                 }
 205                         }else{
 206                                 if(![[privacyAccount listObjectIDsOnPrivacyList:PRIVACY_DENY] containsObject:[contact UID]]){
 207                                         [privacyAccount addListObject:contact toPrivacyList:PRIVACY_DENY];
 208                                 }
 209                         }
 210                 }
 211         }
 212 }
 213
 214 - (BOOL)_searchPrivacyListsForListContact:(AIListContact *)contact withDesiredResult:(BOOL)desiredResult
 215 {
 216         AIAccount *account = nil;
 217         NSEnumerator *enumerator;
 218
 219         enumerator = [[[adium accountController] accountsWithServiceClassOfService:[contact service]] objectEnumerator];
 220
 221         while((account = [enumerator nextObject])){
 222                 if([account conformsToProtocol:@protocol(AIAccount_Privacy)]){
 223                         AIAccount <AIAccount_Privacy> *privacyAccount = (AIAccount <AIAccount_Privacy> *)account;
 224                         if([[privacyAccount listObjectIDsOnPrivacyList:PRIVACY_DENY] containsObject:[contact UID]] == desiredResult){
 225                                 return YES;
 226                         }
 227                 }
 228         }
 229         return NO;
 230 }
 231
 232 @end