diff --git a/client/src/main/java/ctbrec/ui/sites/chaturbate/ChaturbateTabProvider.java b/client/src/main/java/ctbrec/ui/sites/chaturbate/ChaturbateTabProvider.java index 56931572..707b5a32 100644 --- a/client/src/main/java/ctbrec/ui/sites/chaturbate/ChaturbateTabProvider.java +++ b/client/src/main/java/ctbrec/ui/sites/chaturbate/ChaturbateTabProvider.java @@ -9,7 +9,9 @@ import javafx.scene.Scene; import javafx.scene.control.Tab; import java.util.ArrayList; +import java.util.HashMap; import java.util.List; +import java.util.Map; import java.util.regex.Pattern; public class ChaturbateTabProvider extends AbstractTabProvider { @@ -26,38 +28,53 @@ public class ChaturbateTabProvider extends AbstractTabProvider { @Override protected List getSiteTabs(Scene scene) { List tabs = new ArrayList<>(); - + + Map tabMap = new HashMap<>(); + tabMap.put("f", "Female"); + tabMap.put("nf", "New Female"); + tabMap.put("m", "Male"); + tabMap.put("nm", "New Male"); + tabMap.put("c", "Couples"); + tabMap.put("t", "Trans"); + tabMap.put("xp", "Private"); + tabMap.put("xh", "Hidden"); + tabMap.put("xg", "Gaming"); + tabMap.put("ar", "Top Rated"); + tabMap.put("at", "Trending"); + for (String tab : getEnabledTabs()) { + String title = tabMap.getOrDefault(tab, tab); switch (tab) { case "ft": - tabs.add(createTab("Featured", apiUrl + "/roomlist/room-list/?enable_recommendations=false")); + tabs.add(createTab(title, apiUrl + "/roomlist/room-list/?enable_recommendations=false")); break; case "f": case "m": case "c": case "t": - tabs.add(createTab("Female", apiUrl + "/roomlist/room-list/?enable_recommendations=false&genders=" + tab)); + // String title = tabMap.getOrDefault(tab, tab); + tabs.add(createTab(title, apiUrl + "/roomlist/room-list/?enable_recommendations=false&genders=" + tab)); break; case "nf": - addNewCamsTab(tabs, "New Female", "f"); + addNewCamsTab(tabs, title, "f"); break; case "nm": - addNewCamsTab(tabs, "New Male", "m"); + addNewCamsTab(tabs, title, "m"); break; case "xp": - tabs.add(createTab("Private", apiUrl + "/roomlist/room-list/?enable_recommendations=false&private=true")); + tabs.add(createTab(title, apiUrl + "/roomlist/room-list/?enable_recommendations=false&private=true")); break; case "xh": - tabs.add(createTab("Hidden", apiUrl + "/roomlist/room-list/?enable_recommendations=false&hidden=true")); + tabs.add(createTab(title, apiUrl + "/roomlist/room-list/?enable_recommendations=false&hidden=true")); break; case "xg": - tabs.add(createTab("Gaming", apiUrl + "/roomlist/room-list/?enable_recommendations=false&gaming=true")); + tabs.add(createTab(title, apiUrl + "/roomlist/room-list/?enable_recommendations=false&gaming=true")); break; case "ar": - tabs.add(createApiTab("Top Rated", apiUrl + "/discover/carousels/top-rated/")); + tabs.add(createApiTab(title, apiUrl + "/discover/carousels/top-rated/")); break; case "at": - tabs.add(createApiTab("Trending", apiUrl + "/discover/carousels/trending/")); + tabs.add(createApiTab(title, apiUrl + "/discover/carousels/trending/")); break; default: if (Pattern.matches("^[A-Z].+", tab)) { @@ -66,7 +83,7 @@ public class ChaturbateTabProvider extends AbstractTabProvider { break; } } - + tabs.add(followedTab); return tabs; }