Fixes following/follower tab empty when enable hide who to follow

haoict · May 28, 2020 · 56ccdd5 · 56ccdd5
1 parent 07caef9
commit 56ccdd5
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 5 deletions.
diff --git a/Tweak.h b/Tweak.h
@@ -59,4 +59,8 @@
 
 @interface TFNTwitterModuleFooter : NSObject
 @property(nonatomic, readonly) NSURL *url;
-@end
+@end
+
+@interface T1URTTimelineUserItemViewModel : NSObject
+@property(readonly, nonatomic) NSString *scribeComponent;
+@end
diff --git a/Tweak.xm b/Tweak.xm
@@ -53,6 +53,7 @@ static void showDownloadPopup(id twStatus, UIViewController *viewController, voi
       }
 
       NSString *itemClassName = NSStringFromClass([item classForCoder]);
+
       if (hideNewsAndTrending) {
         if ([itemClassName isEqualToString:@"T1Twitter.URTTimelineTrendViewModel"]
             || [itemClassName isEqualToString:@"T1Twitter.URTTimelineEventSummaryViewModel"]
@@ -63,7 +64,12 @@ static void showDownloadPopup(id twStatus, UIViewController *viewController, voi
       }
 
       if (hideWhoToFollow) {
-        if ([itemClassName isEqualToString:@"TFNTwitterUser"] || [itemClassName isEqualToString:@"T1URTTimelineUserItemViewModel"]) {
+        if ([itemClassName isEqualToString:@"TFNTwitterUser"]) {
+          [tbvCell setHidden: YES];
+          return tbvCell;
+        }
+
+        if ([itemClassName isEqualToString:@"T1URTTimelineUserItemViewModel"] && [((T1URTTimelineUserItemViewModel *)item).scribeComponent isEqualToString:@"suggest_who_to_follow"]) {
           [tbvCell setHidden: YES];
           return tbvCell;
         }
@@ -87,8 +93,6 @@ static void showDownloadPopup(id twStatus, UIViewController *viewController, voi
         }
       }
 
-      NSLog(@"hao---%@", itemClassName);
-
       return tbvCell;
     }
 
@@ -100,6 +104,7 @@ static void showDownloadPopup(id twStatus, UIViewController *viewController, voi
       }
 
       NSString *itemClassName = NSStringFromClass([item classForCoder]);
+
       if (hideNewsAndTrending) {
         if ([itemClassName isEqualToString:@"T1Twitter.URTTimelineTrendViewModel"]
             || [itemClassName isEqualToString:@"T1Twitter.URTTimelineEventSummaryViewModel"]
@@ -109,7 +114,11 @@ static void showDownloadPopup(id twStatus, UIViewController *viewController, voi
       }
 
       if (hideWhoToFollow) {
-        if ([itemClassName isEqualToString:@"TFNTwitterUser"] || [itemClassName isEqualToString:@"T1URTTimelineUserItemViewModel"]) {
+        if ([itemClassName isEqualToString:@"TFNTwitterUser"]) {
+          return 0;
+        }
+
+        if ([itemClassName isEqualToString:@"T1URTTimelineUserItemViewModel"] && [((T1URTTimelineUserItemViewModel *)item).scribeComponent isEqualToString:@"suggest_who_to_follow"]) {
           return 0;
         }