TWAdBlockSettingsViewController.xm

#import "TWAdBlockSettingsViewController.h"

extern "C" NSBundle *tweakBundle;
extern "C" NSUserDefaults *tweakDefaults;

#define LOC(x, d) [tweakBundle localizedStringForKey:x value:d table:nil]

%hook _TtC6Twitch27SettingsSwitchTableViewCell
%new
- (id)delegate {
  return MSHookIvar<id>(self, "delegate");
}
%new
- (void)setDelegate:(id)delegate {
  MSHookIvar<id>(self, "delegate") = delegate;
}
%new
- (BOOL)isOn {
  return MSHookIvar<UISwitch *>(self, "$__lazy_storage_$_switchView").isOn;
}
%new
- (void)configureWithTitle:(NSString *)title
                   subtitle:(NSString *)subtitle
                  isEnabled:(BOOL)isEnabled
                       isOn:(BOOL)isOn
    accessibilityIdentifier:(NSString *)accessibilityIdentifier {
  self.textLabel.text = title;
  self.detailTextLabel.text = subtitle;
  UISwitch *switchView = MSHookIvar<UISwitch *>(self, "$__lazy_storage_$_switchView");
  switchView.enabled = isEnabled;
  switchView.on = isOn;
  self.accessibilityIdentifier = accessibilityIdentifier;
}
- (void)settingsSwitchToggled {
  id<SettingsSwitchTableViewCellDelegate> delegate =
      MSHookIvar<id<SettingsSwitchTableViewCellDelegate>>(self, "delegate");
  if (![delegate respondsToSelector:@selector(settingsCellSwitchToggled:)]) return %orig;
  [delegate settingsCellSwitchToggled:self];
}
%end

%subclass TWAdBlockSettingsViewController : TWBaseTableViewController
%property(nonatomic, assign) BOOL adblockEnabled;
%property(nonatomic, assign) BOOL proxyEnabled;
%property(nonatomic, assign) BOOL customProxyEnabled;
- (instancetype)initWithTableViewStyle:(NSInteger)tableViewStyle themeManager:(id)themeManager {
  if ((self = %orig)) {
    self.adblockEnabled = [tweakDefaults boolForKey:@"TWAdBlockEnabled"];
    self.proxyEnabled = [tweakDefaults boolForKey:@"TWAdBlockProxyEnabled"];
    self.customProxyEnabled = [tweakDefaults boolForKey:@"TWAdBlockCustomProxyEnabled"];
  }
  return self;
}
- (void)viewDidLoad {
  %orig;
  self.title = @"TwitchAdBlock";
}
%new
- (NSInteger)numberOfSectionsInTableView:(UITableView *)tableView {
  return self.adblockEnabled ? 2 : 1;
}
- (NSInteger)tableView:(UITableView *)tableView numberOfRowsInSection:(NSInteger)section {
  switch (section) {
    case 0:
      return 1;
    case 1:
      return self.proxyEnabled ? self.customProxyEnabled ? 3 : 2 : 1;
    default:
      return 0;
  }
}
- (UITableViewCell *)tableView:(UITableView *)tableView
         cellForRowAtIndexPath:(NSIndexPath *)indexPath {
  UITableViewCell *cell;
  switch (indexPath.section) {
    case 0:
      cell = [[objc_getClass("_TtC6Twitch27SettingsSwitchTableViewCell") alloc]
            initWithStyle:UITableViewCellStyleDefault
          reuseIdentifier:@"AdBlockSwitchCell"];
      [(_TtC6Twitch27SettingsSwitchTableViewCell *)cell
               configureWithTitle:LOC(@"settings.adblock.title", @"Ad Block")
                         subtitle:nil
                        isEnabled:YES
                             isOn:[tweakDefaults boolForKey:@"TWAdBlockEnabled"]
          accessibilityIdentifier:@"AdBlockSwitchCell"];
      [(_TtC6Twitch27SettingsSwitchTableViewCell *)cell setDelegate:self];
      return cell;
    case 1:
      switch (indexPath.row) {
        case 0:
          cell = [[objc_getClass("_TtC6Twitch27SettingsSwitchTableViewCell") alloc]
                initWithStyle:UITableViewCellStyleDefault
              reuseIdentifier:@"AdBlockProxySwitchCell"];
          [(_TtC6Twitch27SettingsSwitchTableViewCell *)cell
                   configureWithTitle:LOC(@"settings.proxy.title", @"Ad Block Proxy")
                             subtitle:nil
                            isEnabled:YES
                                 isOn:[tweakDefaults boolForKey:@"TWAdBlockProxyEnabled"]
              accessibilityIdentifier:@"AdBlockProxySwitchCell"];
          [(_TtC6Twitch27SettingsSwitchTableViewCell *)cell setDelegate:self];
          return cell;
        case 1:
          cell = [[objc_getClass("_TtC6Twitch27SettingsSwitchTableViewCell") alloc]
                initWithStyle:UITableViewCellStyleDefault
              reuseIdentifier:@"AdBlockCustomProxySwitchCell"];
          [(_TtC6Twitch27SettingsSwitchTableViewCell *)cell
                   configureWithTitle:LOC(@"settings.custom_proxy.title", @"Custom Proxy")
                             subtitle:nil
                            isEnabled:YES
                                 isOn:[tweakDefaults boolForKey:@"TWAdBlockCustomProxyEnabled"]
              accessibilityIdentifier:@"AdBlockCustomProxySwitchCell"];
          [(_TtC6Twitch27SettingsSwitchTableViewCell *)cell setDelegate:self];
          return cell;
        case 2:
          cell = [[objc_getClass("TWAdBlockSettingsTextFieldTableViewCell") alloc]
                initWithStyle:UITableViewCellStyleDefault
              reuseIdentifier:@"TWAdBlockProxy"];
          TWAdBlockSettingsTextField *textField =
              ((TWAdBlockSettingsTextFieldTableViewCell *)cell).textField;
          textField.textField.placeholder = PROXY_URL;
          textField.textField.text = [tweakDefaults stringForKey:@"TWAdBlockProxy"];
          textField.delegate = self;
          return cell;
      }
    default:
      return nil;
  }
}
%new
- (NSString *)tableView:(UITableView *)tableView titleForFooterInSection:(NSInteger)section {
  switch (section) {
    case 0:
      return LOC(@"settings.adblock.footer", @"Choose whether or not you want to block ads");
    case 1:
      return LOC(@"settings.proxy.footer",
                 @"Proxy specific requests through a proxy server based in an ad-free country");
    default:
      return nil;
  }
}
%new
- (void)settingsCellSwitchToggled:(UISwitch *)sender {
  if ([sender.accessibilityIdentifier isEqualToString:@"AdBlockSwitchCell"]) {
    [tweakDefaults setBool:sender.isOn forKey:@"TWAdBlockEnabled"];
    self.adblockEnabled = sender.isOn;

    NSIndexSet *sections = [NSIndexSet indexSetWithIndex:1];
    if (sender.isOn)
      [self.tableView insertSections:sections withRowAnimation:UITableViewRowAnimationFade];
    else
      [self.tableView deleteSections:sections withRowAnimation:UITableViewRowAnimationFade];
  } else if ([sender.accessibilityIdentifier isEqualToString:@"AdBlockProxySwitchCell"]) {
    [tweakDefaults setBool:sender.isOn forKey:@"TWAdBlockProxyEnabled"];
    self.proxyEnabled = sender.isOn;

    NSMutableArray *indexPaths = [NSMutableArray array];
    [indexPaths addObject:[NSIndexPath indexPathForRow:1 inSection:1]];
    if (self.customProxyEnabled) [indexPaths addObject:[NSIndexPath indexPathForRow:2 inSection:1]];
    if (sender.isOn)
      [self.tableView insertRowsAtIndexPaths:indexPaths
                            withRowAnimation:UITableViewRowAnimationFade];
    else
      [self.tableView deleteRowsAtIndexPaths:indexPaths
                            withRowAnimation:UITableViewRowAnimationFade];
  } else if ([sender.accessibilityIdentifier isEqualToString:@"AdBlockCustomProxySwitchCell"]) {
    [tweakDefaults setBool:sender.isOn forKey:@"TWAdBlockCustomProxyEnabled"];
    self.customProxyEnabled = sender.isOn;

    NSArray *indexPaths = @[ [NSIndexPath indexPathForRow:2 inSection:1] ];
    if (sender.isOn)
      [self.tableView insertRowsAtIndexPaths:indexPaths
                            withRowAnimation:UITableViewRowAnimationFade];
    else
      [self.tableView deleteRowsAtIndexPaths:indexPaths
                            withRowAnimation:UITableViewRowAnimationFade];
  }

  [tweakDefaults synchronize];
  notify_post("com.level3tjg.twitchadblock/updatePrefs");
}
%new
- (void)textFieldDidEndEditing:(UITextField *)textField {
  [tweakDefaults setValue:textField.text forKey:@"TWAdBlockProxy"];
  [tweakDefaults synchronize];
  notify_post("com.level3tjg.twitchadblock/updatePrefs");
}
%end

%ctor {
  if (![NSProcessInfo.processInfo.processName isEqualToString:@"mediaserverd"]) %init;
}