From 721c2ae7ced2310af3550c21629845396dc8024d Mon Sep 17 00:00:00 2001
From: Adam Saudagar <adsau59@gmail.com>
Date: Tue, 3 Nov 2020 00:34:50 +0530
Subject: [PATCH] show full auto only if has beta access

---
 fishy/gui/gui.py  |  6 ++++--
 fishy/web/urls.py |  1 +
 fishy/web/web.py  | 12 ++++++++++++
 3 files changed, 17 insertions(+), 2 deletions(-)

diff --git a/fishy/gui/gui.py b/fishy/gui/gui.py
index 89f9337..e0ed133 100644
--- a/fishy/gui/gui.py
+++ b/fishy/gui/gui.py
@@ -3,6 +3,7 @@ from tkinter import OptionMenu, Button, IntVar
 from typing import List, Callable, Optional
 import threading
 
+from fishy.web import web
 from ttkthemes import ThemedTk
 
 from fishy.engine.common.event_handler import EngineEventHandler
@@ -52,10 +53,11 @@ class GUI:
     @property
     def engines(self):
         engines = {
-            "Semi Fisher": [lambda: config_top.start_semifisher_config(self), self.engine.toggle_semifisher],
+            "Semi Fisher": [lambda: config_top.start_semifisher_config(self),  # start config function
+                            self.engine.toggle_semifisher],  # start engine function
         }
 
-        if config.get('debug', False):
+        if web.has_beta():
             engines["Full-Auto Fisher"] = [lambda: config_top.start_fullfisher_config(self),
                                            self.engine.toggle_fullfisher]
         return engines
diff --git a/fishy/web/urls.py b/fishy/web/urls.py
index 7779ccc..21ea4a9 100644
--- a/fishy/web/urls.py
+++ b/fishy/web/urls.py
@@ -14,6 +14,7 @@ hole_depleted = domain + "/api/hole_depleted"
 session = domain + "/api/session"
 terms = domain + "/terms.html"
 discord = domain + "/api/discord"
+beta = domain + "/api/beta"
 
 
 def get_notification_page(uid):
diff --git a/fishy/web/web.py b/fishy/web/web.py
index 4f25b5d..a59f62d 100644
--- a/fishy/web/web.py
+++ b/fishy/web/web.py
@@ -127,3 +127,15 @@ def get_session(lazy=True):
 
     _session_id = response.json()["session_id"]
     return _session_id
+
+
+@fallback(False)
+def has_beta():
+    body = {'uid': config.get("uid")}
+    response = requests.get(urls.beta, params=body)
+    result = response.json()
+
+    if not result["success"]:
+        return False
+
+    return response.json()["beta"]