From b4338be5488d6e3f939612dd877857f6aeb969a6 Mon Sep 17 00:00:00 2001
From: BarsTiger <kit.barsic@gmail.com>
Date: Fri, 28 Jan 2022 14:29:43 +0200
Subject: [PATCH] GUI development

---
 horsy.py         |  8 ++++++++
 horsygui.py      | 39 +++++++++++++++++++++++++++++++++------
 modules/liker.py | 14 +++++++++++++-
 3 files changed, 54 insertions(+), 7 deletions(-)

diff --git a/horsy.py b/horsy.py
index 34f7b0d..fcc6d4e 100644
--- a/horsy.py
+++ b/horsy.py
@@ -94,5 +94,13 @@ if option in ['list', 'l']:
     from modules.manager import apps_list
     apps_list()
 
+if option in ['like']:
+    from modules.liker import like
+    like(app)
+
+if option in ['dislike']:
+    from modules.liker import dislike
+    dislike(app)
+
 if isNoArgs:
     input('[EXIT] Press enter to exit horsy...')
diff --git a/horsygui.py b/horsygui.py
index 28c85f6..446125e 100644
--- a/horsygui.py
+++ b/horsygui.py
@@ -9,20 +9,20 @@ if __name__ == "__main__":
     app = QtWidgets.QApplication(sys.argv)
 
 
-MainWindow = QtWidgets.QMainWindow()
-ui = gui.Ui_MainWindow()
-ui.setupUi(MainWindow)
-
 UiMainWindow = QtWidgets.QMainWindow()
+ui = gui.Ui_MainWindow()
+ui.setupUi(UiMainWindow)
+
+UiLoginWindow = QtWidgets.QMainWindow()
 login_ui = gui.Ui_LoginWindow()
-login_ui.setupUi(UiMainWindow)
+login_ui.setupUi(UiLoginWindow)
 
 UiDownloadWindow = QtWidgets.QMainWindow()
 download_ui = gui.Ui_DownloadWindow()
 download_ui.setupUi(UiDownloadWindow)
 
 if __name__ == "__main__":
-    MainWindow.show()
+    UiMainWindow.show()
 
 # Functions
 def refresh_gui():
@@ -110,6 +110,31 @@ def info_gui():
     except:
         return
 
+def like_gui():
+    from modules.liker import like
+    try:
+        app_name = ui.search_table.currentItem().text()
+        if app_name == "":
+            return
+        else:
+            like = like(app_name, login_ui, UiLoginWindow)
+            if 'Success' not in like:
+                gui.popup("Like", like)
+    except:
+        return
+
+def dislike_gui():
+    from modules.liker import dislike
+    try:
+        app_name = ui.search_table.currentItem().text()
+        if app_name == "":
+            return
+        else:
+            dislike = dislike(app_name, login_ui, UiLoginWindow)
+            if 'Success' not in dislike:
+                gui.popup("Dislike", dislike)
+    except:
+        return
 
 def upload_gui():
     from modules.uploader import upload
@@ -129,6 +154,8 @@ if __name__ == "__main__":
     ui.source_button.clicked.connect(get_source_gui)
     ui.info_button.clicked.connect(info_gui)
     ui.search_table.itemDoubleClicked.connect(info_gui)
+    ui.like_button.clicked.connect(like_gui)
+    ui.dislike_button.clicked.connect(dislike_gui)
     ui.upload_button.clicked.connect(upload_gui)
 
 
diff --git a/modules/liker.py b/modules/liker.py
index be4280e..0b08f4f 100644
--- a/modules/liker.py
+++ b/modules/liker.py
@@ -9,5 +9,17 @@ def like(package, is_gui=False, login_ui=None, Ui_LoginWindow=None):
         "rate": 1,
         "packageName": package
     }
-    r = requests.post(f"{horsy_vars.protocol}{horsy_vars.server_url}/packages/rate", json=body).text
+    r = requests.post(f"{horsy_vars.protocol}{horsy_vars.server_url}/packages/rate", json=body).json()
+    print(r["message"])
+    return r["message"]
 
+
+def dislike(package, is_gui=False, login_ui=None, Ui_LoginWindow=None):
+    body = {
+        "auth": get_auth(is_gui, login_ui, Ui_LoginWindow),
+        "rate": 0,
+        "packageName": package
+    }
+    r = requests.post(f"{horsy_vars.protocol}{horsy_vars.server_url}/packages/rate", json=body).json()
+    print(r["message"])
+    return r["message"]