From 25cc4720f77474bc5e64e72034272a406483dcc0 Mon Sep 17 00:00:00 2001
From: gdlbo <41114720+gdlbo@users.noreply.github.com>
Date: Tue, 20 Feb 2024 00:56:36 +0300
Subject: [PATCH] added vote dialog

---
 .../com/vk/newsfeed/holders/AdMarkHolder.java |  2 +-
 .../java/ru/vtosters/hooks/PostsMenuHook.java | 26 ++++++++++++++++---
 .../com/vk/discover/DiscoverFragment.smali    |  2 +-
 .../presenters/EntriesListPresenter.smali     |  2 +-
 .../com/vk/wall/post/PostViewPresenter.smali  |  2 +-
 5 files changed, 26 insertions(+), 8 deletions(-)

diff --git a/app/src/main/java/com/vk/newsfeed/holders/AdMarkHolder.java b/app/src/main/java/com/vk/newsfeed/holders/AdMarkHolder.java
index acdc35fc7..9535e211d 100644
--- a/app/src/main/java/com/vk/newsfeed/holders/AdMarkHolder.java
+++ b/app/src/main/java/com/vk/newsfeed/holders/AdMarkHolder.java
@@ -71,7 +71,7 @@ public final class AdMarkHolder extends BaseNewsEntryHolder<NewsEntry> {
         this.F.setText(disclamer);
     }
 
-    private static void voteDialog(int ownerId, int postId, Context ctx) {
+    public static void voteDialog(int ownerId, int postId, Context ctx) {
         new VkAlertDialog.Builder(ctx)
                 .setTitle("SponsorPost")
                 .setMessage("Этот пост является рекламным?")
diff --git a/app/src/main/java/ru/vtosters/hooks/PostsMenuHook.java b/app/src/main/java/ru/vtosters/hooks/PostsMenuHook.java
index 4b7f8f8b4..2afecd588 100644
--- a/app/src/main/java/ru/vtosters/hooks/PostsMenuHook.java
+++ b/app/src/main/java/ru/vtosters/hooks/PostsMenuHook.java
@@ -1,18 +1,36 @@
 package ru.vtosters.hooks;
 
+import android.content.Context;
+import android.util.Log;
 import com.vk.dto.newsfeed.entries.NewsEntry;
+import com.vk.dto.newsfeed.entries.Post;
 import com.vk.newsfeed.NewsEntryActionsAdapter;
-import ru.vtosters.lite.utils.AndroidUtils;
+import com.vk.newsfeed.holders.AdMarkHolder;
+import ru.vtosters.lite.utils.AccountManagerUtils;
 
 public class PostsMenuHook {
     private static final int id = 200;
 
     public static void injectList(NewsEntryActionsAdapter actionsAdapter, NewsEntry newsEntry) {
-        actionsAdapter.a(id, "SponsorPost");
+        if (newsEntry instanceof Post) {
+            int ownerId = ((Post) newsEntry).b();
+
+            if (ownerId != AccountManagerUtils.getUserId()) {
+                actionsAdapter.a(id, "SponsorPost");
+            }
+        }
     }
 
-    public static void injectButtons(int i, NewsEntry newsEntry) {
-        AndroidUtils.sendToast("SponsorPost");
+    public static void injectButtons(int i, NewsEntry newsEntry, Context context) {
+        if (newsEntry instanceof Post) {
+            Post post = (Post) newsEntry;
+            int ownerId = post.b();
+            int postId = post.P1();
+
+            AdMarkHolder.voteDialog(ownerId, postId, context);
+        } else {
+            Log.d("PostsMenuHook", "unsupported instance");
+        }
     }
 
     public static boolean isCustomButton(int i) {
diff --git a/smali/smali_classes2/com/vk/discover/DiscoverFragment.smali b/smali/smali_classes2/com/vk/discover/DiscoverFragment.smali
index e57a8cd0a..d695ef65d 100644
--- a/smali/smali_classes2/com/vk/discover/DiscoverFragment.smali
+++ b/smali/smali_classes2/com/vk/discover/DiscoverFragment.smali
@@ -897,7 +897,7 @@
 
     if-eqz v2, :cond_99
 
-    invoke-static {p1, p2}, Lru/vtosters/hooks/PostsMenuHook;->injectButtons(ILcom/vk/dto/newsfeed/entries/NewsEntry;)V
+    invoke-static {p1, p2, v0}, Lru/vtosters/hooks/PostsMenuHook;->injectButtons(ILcom/vk/dto/newsfeed/entries/NewsEntry;Landroid/content/Context;)V
 
     return-void
 
diff --git a/smali/smali_classes3/com/vk/newsfeed/presenters/EntriesListPresenter.smali b/smali/smali_classes3/com/vk/newsfeed/presenters/EntriesListPresenter.smali
index e93773735..c5002acd6 100644
--- a/smali/smali_classes3/com/vk/newsfeed/presenters/EntriesListPresenter.smali
+++ b/smali/smali_classes3/com/vk/newsfeed/presenters/EntriesListPresenter.smali
@@ -2866,7 +2866,7 @@
 
     if-eqz v3, :cond_20
 
-    invoke-static {p2, p3}, Lru/vtosters/hooks/PostsMenuHook;->injectButtons(ILcom/vk/dto/newsfeed/entries/NewsEntry;)V
+    invoke-static {p2, p3, v2}, Lru/vtosters/hooks/PostsMenuHook;->injectButtons(ILcom/vk/dto/newsfeed/entries/NewsEntry;Landroid/content/Context;)V
 
     goto/16 :goto_1
 
diff --git a/smali/smali_classes4/com/vk/wall/post/PostViewPresenter.smali b/smali/smali_classes4/com/vk/wall/post/PostViewPresenter.smali
index 7d543530c..50043d48c 100644
--- a/smali/smali_classes4/com/vk/wall/post/PostViewPresenter.smali
+++ b/smali/smali_classes4/com/vk/wall/post/PostViewPresenter.smali
@@ -3803,7 +3803,7 @@
 
     if-eqz v3, :cond_20
 
-    invoke-static {p1, v4}, Lru/vtosters/hooks/PostsMenuHook;->injectButtons(ILcom/vk/dto/newsfeed/entries/NewsEntry;)V
+    invoke-static {p1, v4, v0}, Lru/vtosters/hooks/PostsMenuHook;->injectButtons(ILcom/vk/dto/newsfeed/entries/NewsEntry;Landroid/content/Context;)V
 
     goto/16 :goto_5