diff --git a/client/src/main/java/ctbrec/ui/sites/dreamcam/DreamcamConfigUI.java b/client/src/main/java/ctbrec/ui/sites/dreamcam/DreamcamConfigUI.java index 44fcf407..2132ddd8 100644 --- a/client/src/main/java/ctbrec/ui/sites/dreamcam/DreamcamConfigUI.java +++ b/client/src/main/java/ctbrec/ui/sites/dreamcam/DreamcamConfigUI.java @@ -1,15 +1,18 @@ package ctbrec.ui.sites.dreamcam; +import java.util.ArrayList; +import java.util.List; + import ctbrec.Config; import ctbrec.sites.dreamcam.Dreamcam; import ctbrec.ui.settings.SettingsTab; import ctbrec.ui.sites.AbstractConfigUI; import javafx.geometry.Insets; import javafx.scene.Parent; -import javafx.scene.control.ToggleGroup; +// import javafx.scene.control.ToggleGroup; import javafx.scene.control.Button; import javafx.scene.control.CheckBox; -import javafx.scene.control.RadioButton; +// import javafx.scene.control.RadioButton; import javafx.scene.control.Label; import javafx.scene.layout.GridPane; import javafx.scene.layout.HBox; @@ -54,6 +57,36 @@ public class DreamcamConfigUI extends AbstractConfigUI { GridPane.setMargin(vr, new Insets(0, 0, 0, SettingsTab.CHECKBOX_MARGIN)); layout.add(vr, 1, row++); + Label tabsLabel = new Label("Tabs"); + layout.add(tabsLabel, 0, row); + List tabs = getEnabledTabs(); + + HBox checkboxContainer = new HBox(10); + + CheckBox female = new CheckBox("Girls"); + female.setSelected(tabs.contains("girls")); + female.setOnAction(e -> toggleTab("girls", female.isSelected())); + HBox.setMargin(female, new Insets(0, 0, 0, 7)); + checkboxContainer.getChildren().add(female); + + CheckBox male = new CheckBox("Boys"); + male.setSelected(tabs.contains("men")); + male.setOnAction(e -> toggleTab("men", male.isSelected())); + checkboxContainer.getChildren().add(male); + + CheckBox couples = new CheckBox("Couples"); + couples.setSelected(tabs.contains("couples")); + couples.setOnAction(e -> toggleTab("couples", couples.isSelected())); + checkboxContainer.getChildren().add(couples); + + CheckBox trans = new CheckBox("Trans"); + trans.setSelected(tabs.contains("trans")); + trans.setOnAction(e -> toggleTab("trans", trans.isSelected())); + checkboxContainer.getChildren().add(trans); + + layout.add(checkboxContainer, 1, row, 4, 1); + row++; + var deleteCookies = new Button("Delete Cookies"); deleteCookies.setOnAction(e -> site.getHttpClient().clearCookies()); layout.add(deleteCookies, 1, row); @@ -63,4 +96,18 @@ public class DreamcamConfigUI extends AbstractConfigUI { return layout; } + private void toggleTab(String tab, boolean selected) { + var settings = Config.getInstance().getSettings(); + if (selected) { + settings.dreamcamTabs.add(tab); + } else { + settings.dreamcamTabs.remove(tab); + } + save(); + } + + private List getEnabledTabs() { + return new ArrayList<>(Config.getInstance().getSettings().dreamcamTabs); + } + } diff --git a/client/src/main/java/ctbrec/ui/sites/dreamcam/DreamcamTabProvider.java b/client/src/main/java/ctbrec/ui/sites/dreamcam/DreamcamTabProvider.java index 17226cc0..0a79d606 100644 --- a/client/src/main/java/ctbrec/ui/sites/dreamcam/DreamcamTabProvider.java +++ b/client/src/main/java/ctbrec/ui/sites/dreamcam/DreamcamTabProvider.java @@ -1,11 +1,16 @@ package ctbrec.ui.sites.dreamcam; +import ctbrec.Config; import ctbrec.sites.dreamcam.Dreamcam; -import ctbrec.sites.dreamcam.DreamcamModel; +// import ctbrec.sites.dreamcam.DreamcamModel; import ctbrec.ui.sites.AbstractTabProvider; import ctbrec.ui.tabs.ThumbOverviewTab; import java.util.ArrayList; +import java.util.HashMap; import java.util.List; +import java.util.Map; +// import java.util.Objects; + import javafx.scene.Scene; import javafx.scene.control.Tab; @@ -19,10 +24,18 @@ public class DreamcamTabProvider extends AbstractTabProvider { @Override protected List getSiteTabs(Scene scene) { List tabs = new ArrayList<>(); - tabs.add(createTab("Girls", API_URL + "&tag-categories=girls")); - tabs.add(createTab("Boys", API_URL + "&tag-categories=men")); - tabs.add(createTab("Couples", API_URL + "&tag-categories=couples")); - tabs.add(createTab("Trans", API_URL + "&tag-categories=trans")); + + Map tabMap = new HashMap<>(); + tabMap.put("girls", "Girls"); + tabMap.put("men", "Boys"); + tabMap.put("couples", "Couples"); + tabMap.put("trans", "Trans"); + List enabledTabs = getEnabledTabs(); + for (String tab : enabledTabs) { + String title = tabMap.getOrDefault(tab, tab); + tabs.add(createTab(title, API_URL + "&tag-categories=" + tab)); + } + return tabs; } @@ -34,4 +47,8 @@ public class DreamcamTabProvider extends AbstractTabProvider { return tab; } + private List getEnabledTabs() { + return new ArrayList<>(Config.getInstance().getSettings().dreamcamTabs); + } + }