Skip to content

Commit 2d139d3

Browse files
brandon-lau0brandon-lau0
committed
feat: allow shortscience and googlescholar search to be configurable
1 parent c2eb282 commit 2d139d3

File tree

11 files changed

+211
-60
lines changed

11 files changed

+211
-60
lines changed

‎jabgui/src/main/java/org/jabref/gui/maintable/MainTable.java

Lines changed: 14 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -163,7 +163,8 @@ public MainTable(MainTableDataModel model,
163163
taskExecutor,
164164
Injector.instantiateModelOrService(JournalAbbreviationRepository.class),
165165
entryTypesManager,
166-
importHandler))
166+
importHandler,
167+
preferences.getImporterPreferences()))
167168
.withPseudoClass(MATCHING_SEARCH_AND_GROUPS, entry -> entry.matchCategory().isEqualTo(MatchCategory.MATCHING_SEARCH_AND_GROUPS))
168169
.withPseudoClass(MATCHING_SEARCH_NOT_GROUPS, entry -> entry.matchCategory().isEqualTo(MatchCategory.MATCHING_SEARCH_NOT_GROUPS))
169170
.withPseudoClass(MATCHING_GROUPS_NOT_SEARCH, entry -> entry.matchCategory().isEqualTo(MatchCategory.MATCHING_GROUPS_NOT_SEARCH))
@@ -180,10 +181,10 @@ public MainTable(MainTableDataModel model,
180181
// force match category column to be the first sort order, (match_category column is always the first column)
181182
this.getSortOrder().addFirst(getColumns().getFirst());
182183
this.getSortOrder().addListener((ListChangeListener<TableColumn<BibEntryTableViewModel, ?>>) change -> {
183-
if (!this.getSortOrder().getFirst().equals(getColumns().getFirst())) {
184-
this.getSortOrder().addFirst(getColumns().getFirst());
185-
}
186-
});
184+
if (!this.getSortOrder().getFirst().equals(getColumns().getFirst())) {
185+
this.getSortOrder().addFirst(getColumns().getFirst());
186+
}
187+
});
187188

188189
mainTablePreferences.getColumnPreferences().getColumnSortOrder().forEach(columnModel ->
189190
this.getColumns().stream()
@@ -414,7 +415,7 @@ private void setupKeyBindings(KeyBindingRepository keyBindings) {
414415
event.consume();
415416
break;
416417
case SCROLL_TO_PREVIOUS_MATCH_CATEGORY:
417-
scrollToPreviousMatchCategory();
418+
scrollToPreviousMatchCategory();
418419
event.consume();
419420
break;
420421
case OPEN_URL_OR_DOI:
@@ -576,13 +577,13 @@ public void setCitationMergeMode(boolean citationMerge) {
576577
}
577578

578579
private void updatePlaceholder(VBox placeholderBox) {
579-
if (database.getDatabase().getEntries().isEmpty()) {
580-
this.setPlaceholder(placeholderBox);
581-
// [impl->req~maintable.focus~1]
582-
requestFocus();
583-
} else {
584-
this.setPlaceholder(null);
585-
}
580+
if (database.getDatabase().getEntries().isEmpty()) {
581+
this.setPlaceholder(placeholderBox);
582+
// [impl->req~maintable.focus~1]
583+
requestFocus();
584+
} else {
585+
this.setPlaceholder(null);
586+
}
586587
}
587588

588589
private BibEntry addExampleEntry() {

‎jabgui/src/main/java/org/jabref/gui/maintable/RightClickMenu.java

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,7 @@
3636
import org.jabref.gui.specialfields.SpecialFieldMenuItemFactory;
3737
import org.jabref.logic.citationstyle.CitationStyleOutputFormat;
3838
import org.jabref.logic.citationstyle.CitationStylePreviewLayout;
39+
import org.jabref.logic.importer.ImporterPreferences;
3940
import org.jabref.logic.importer.WebFetchers;
4041
import org.jabref.logic.journals.JournalAbbreviationRepository;
4142
import org.jabref.logic.l10n.Localization;
@@ -62,7 +63,8 @@ public static ContextMenu create(BibEntryTableViewModel entry,
6263
TaskExecutor taskExecutor,
6364
JournalAbbreviationRepository abbreviationRepository,
6465
BibEntryTypesManager entryTypesManager,
65-
ImportHandler importHandler) {
66+
ImportHandler importHandler,
67+
ImporterPreferences importerPreferences) {
6668
ActionFactory factory = new ActionFactory();
6769
ContextMenu contextMenu = new ContextMenu();
6870

@@ -101,7 +103,7 @@ public static ContextMenu create(BibEntryTableViewModel entry,
101103
extractFileReferencesOffline,
102104

103105
factory.createMenuItem(StandardActions.OPEN_URL, new OpenUrlAction(dialogService, stateManager, preferences)),
104-
createSearchSubMenu(factory, dialogService, stateManager, preferences),
106+
createSearchSubMenu(factory, dialogService, stateManager, preferences, importerPreferences),
105107
new SeparatorMenuItem(),
106108

107109
new ChangeEntryTypeMenu(libraryTab.getSelectedEntries(), libraryTab.getBibDatabaseContext(), undoManager, entryTypesManager).asSubMenu(),
@@ -222,11 +224,12 @@ private static Menu createSendSubMenu(ActionFactory factory,
222224
private static Menu createSearchSubMenu(ActionFactory factory,
223225
DialogService dialogService,
224226
StateManager stateManager,
225-
GuiPreferences preferences) {
227+
GuiPreferences preferences,
228+
ImporterPreferences importerPreferences) {
226229
Menu searchMenu = factory.createMenu(StandardActions.SEARCH);
227230
searchMenu.getItems().addAll(
228-
factory.createMenuItem(StandardActions.SEARCH_SHORTSCIENCE, new SearchShortScienceAction(dialogService, stateManager, preferences)),
229-
factory.createMenuItem(StandardActions.SEARCH_GOOGLE_SCHOLAR, new SearchGoogleScholarAction(dialogService, stateManager, preferences))
231+
factory.createMenuItem(StandardActions.SEARCH_SHORTSCIENCE, new SearchShortScienceAction(dialogService, stateManager, preferences, importerPreferences)),
232+
factory.createMenuItem(StandardActions.SEARCH_GOOGLE_SCHOLAR, new SearchGoogleScholarAction(dialogService, stateManager, preferences, importerPreferences))
230233
);
231234
return searchMenu;
232235
}

‎jabgui/src/main/java/org/jabref/gui/maintable/SearchGoogleScholarAction.java

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010
import org.jabref.gui.actions.SimpleCommand;
1111
import org.jabref.gui.desktop.os.NativeDesktop;
1212
import org.jabref.gui.preferences.GuiPreferences;
13+
import org.jabref.logic.importer.ImporterPreferences;
1314
import org.jabref.logic.l10n.Localization;
1415
import org.jabref.logic.util.ExternalLinkCreator;
1516
import org.jabref.model.entry.BibEntry;
@@ -21,11 +22,13 @@ public class SearchGoogleScholarAction extends SimpleCommand {
2122
private final DialogService dialogService;
2223
private final StateManager stateManager;
2324
private final GuiPreferences preferences;
25+
private final ExternalLinkCreator externalLinkCreator;
2426

25-
public SearchGoogleScholarAction(DialogService dialogService, StateManager stateManager, GuiPreferences preferences) {
27+
public SearchGoogleScholarAction(DialogService dialogService, StateManager stateManager, GuiPreferences preferences, ImporterPreferences importerPreferences) {
2628
this.dialogService = dialogService;
2729
this.stateManager = stateManager;
2830
this.preferences = preferences;
31+
this.externalLinkCreator = new ExternalLinkCreator(importerPreferences);
2932

3033
BooleanExpression fieldIsSet = isFieldSetForSelectedEntry(StandardField.TITLE, stateManager);
3134
this.executable.bind(needsEntriesSelected(1, stateManager).and(fieldIsSet));
@@ -40,7 +43,7 @@ public void execute() {
4043
dialogService.notify(Localization.lang("This operation requires exactly one item to be selected."));
4144
return;
4245
}
43-
ExternalLinkCreator.getGoogleScholarSearchURL(bibEntries.getFirst()).ifPresent(url -> {
46+
externalLinkCreator.getGoogleScholarSearchURL(bibEntries.getFirst()).ifPresent(url -> {
4447
try {
4548
NativeDesktop.openExternalViewer(databaseContext, preferences, url, StandardField.URL, dialogService, bibEntries.getFirst());
4649
} catch (IOException ex) {

‎jabgui/src/main/java/org/jabref/gui/maintable/SearchShortScienceAction.java

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010
import org.jabref.gui.actions.SimpleCommand;
1111
import org.jabref.gui.desktop.os.NativeDesktop;
1212
import org.jabref.gui.preferences.GuiPreferences;
13+
import org.jabref.logic.importer.ImporterPreferences;
1314
import org.jabref.logic.l10n.Localization;
1415
import org.jabref.logic.util.ExternalLinkCreator;
1516
import org.jabref.model.entry.BibEntry;
@@ -22,11 +23,13 @@ public class SearchShortScienceAction extends SimpleCommand {
2223
private final DialogService dialogService;
2324
private final StateManager stateManager;
2425
private final GuiPreferences preferences;
26+
private final ExternalLinkCreator externalLinkCreator;
2527

26-
public SearchShortScienceAction(DialogService dialogService, StateManager stateManager, GuiPreferences preferences) {
28+
public SearchShortScienceAction(DialogService dialogService, StateManager stateManager, GuiPreferences preferences, ImporterPreferences importerPreferences) {
2729
this.dialogService = dialogService;
2830
this.stateManager = stateManager;
2931
this.preferences = preferences;
32+
this.externalLinkCreator = new ExternalLinkCreator(importerPreferences);
3033

3134
BooleanExpression fieldIsSet = isFieldSetForSelectedEntry(StandardField.TITLE, stateManager);
3235
this.executable.bind(needsEntriesSelected(1, stateManager).and(fieldIsSet));
@@ -41,7 +44,7 @@ public void execute() {
4144
dialogService.notify(Localization.lang("This operation requires exactly one item to be selected."));
4245
return;
4346
}
44-
ExternalLinkCreator.getShortScienceSearchURL(bibEntries.getFirst()).ifPresent(url -> {
47+
externalLinkCreator.getShortScienceSearchURL(bibEntries.getFirst()).ifPresent(url -> {
4548
try {
4649
NativeDesktop.openExternalViewer(databaseContext, preferences, url, StandardField.URL, dialogService, bibEntries.getFirst());
4750
} catch (IOException ex) {

‎jabgui/src/main/java/org/jabref/gui/preferences/websearch/SearchEngineItem.java

Lines changed: 34 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,34 @@
1+
package org.jabref.gui.preferences.websearch;
2+
3+
import javafx.beans.property.SimpleStringProperty;
4+
import javafx.beans.property.StringProperty;
5+
6+
public class SearchEngineItem {
7+
private final StringProperty name;
8+
private final StringProperty urlTemplate;
9+
10+
public SearchEngineItem(String name, String urlTemplate) {
11+
this.name = new SimpleStringProperty(name);
12+
this.urlTemplate = new SimpleStringProperty(urlTemplate);
13+
}
14+
15+
public StringProperty nameProperty() {
16+
return name;
17+
}
18+
19+
public StringProperty urlTemplateProperty() {
20+
return urlTemplate;
21+
}
22+
23+
public String getName() {
24+
return name.get();
25+
}
26+
27+
public String getUrlTemplate() {
28+
return urlTemplate.get();
29+
}
30+
31+
public void setUrlTemplate(String urlTemplate) {
32+
this.urlTemplate.set(urlTemplate);
33+
}
34+
}

‎jabgui/src/main/java/org/jabref/gui/preferences/websearch/WebSearchTab.java

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,10 @@ public class WebSearchTab extends AbstractPreferenceTabView<WebSearchTabViewMode
4343
@FXML private CheckBox grobidEnabled;
4444
@FXML private TextField grobidURL;
4545

46+
@FXML private TableView<SearchEngineItem> searchEngineTable;
47+
@FXML private TableColumn<SearchEngineItem, String> searchEngineName;
48+
@FXML private TableColumn<SearchEngineItem, String> searchEngineUrlTemplate;
49+
4650
@FXML private TableView<FetcherApiKey> apiKeySelectorTable;
4751
@FXML private TableColumn<FetcherApiKey, String> apiKeyName;
4852
@FXML private TableColumn<FetcherApiKey, String> customApiKey;
@@ -73,6 +77,16 @@ public String getTabName() {
7377
public void initialize() {
7478
this.viewModel = new WebSearchTabViewModel(preferences, dialogService, refAiEnabled);
7579

80+
searchEngineName.setCellValueFactory(param -> param.getValue().nameProperty());
81+
searchEngineName.setCellFactory(TextFieldTableCell.forTableColumn());
82+
searchEngineName.setEditable(false);
83+
84+
searchEngineUrlTemplate.setCellValueFactory(param -> param.getValue().urlTemplateProperty());
85+
searchEngineUrlTemplate.setCellFactory(TextFieldTableCell.forTableColumn());
86+
searchEngineUrlTemplate.setEditable(true);
87+
88+
searchEngineTable.setItems(viewModel.getSearchEngines());
89+
7690
enableWebSearch.selectedProperty().bindBidirectional(viewModel.enableWebSearchProperty());
7791
warnAboutDuplicatesOnImport.selectedProperty().bindBidirectional(viewModel.warnAboutDuplicatesOnImportProperty());
7892
downloadLinkedOnlineFiles.selectedProperty().bindBidirectional(viewModel.shouldDownloadLinkedOnlineFiles());

‎jabgui/src/main/java/org/jabref/gui/preferences/websearch/WebSearchTabViewModel.java

Lines changed: 31 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22

33
import java.io.IOException;
44
import java.net.HttpURLConnection;
5+
import java.util.Map;
56
import java.util.Optional;
67
import java.util.stream.Collectors;
78

@@ -64,6 +65,8 @@ public class WebSearchTabViewModel implements PreferenceTabViewModel {
6465
private final BooleanProperty apikeyPersistProperty = new SimpleBooleanProperty();
6566
private final BooleanProperty apikeyPersistAvailableProperty = new SimpleBooleanProperty();
6667

68+
private final ObservableList<SearchEngineItem> searchEngines = FXCollections.observableArrayList();
69+
6770
private final DialogService dialogService;
6871
private final CliPreferences preferences;
6972
private final DOIPreferences doiPreferences;
@@ -88,6 +91,7 @@ public WebSearchTabViewModel(CliPreferences preferences, DialogService dialogSer
8891
this.refAiEnabled = refAiEnabled;
8992

9093
setupPlainCitationParsers(preferences);
94+
setupSearchEngines();
9195
}
9296

9397
private void setupPlainCitationParsers(CliPreferences preferences) {
@@ -128,6 +132,14 @@ private void setupPlainCitationParsers(CliPreferences preferences) {
128132
});
129133
}
130134

135+
private void setupSearchEngines() {
136+
// Add default search engines
137+
searchEngines.addAll(
138+
new SearchEngineItem("Google Scholar", "https://scholar.google.com/scholar?q={title}"),
139+
new SearchEngineItem("Short Science", "https://www.shortscience.org/internalsearch?q={title}")
140+
);
141+
}
142+
131143
@Override
132144
public void setValues() {
133145
enableWebSearchProperty.setValue(importerPreferences.areImporterEnabled());
@@ -159,6 +171,13 @@ public void setValues() {
159171
return new StudyCatalogItem(name, enabled);
160172
})
161173
.toList());
174+
175+
// Load custom URL templates from preferences if they exist
176+
Map<String, String> savedTemplates = preferences.getImporterPreferences().getSearchEngineUrlTemplates();
177+
if (!savedTemplates.isEmpty()) {
178+
searchEngines.clear();
179+
savedTemplates.forEach((name, url) -> searchEngines.add(new SearchEngineItem(name, url)));
180+
}
162181
}
163182

164183
@Override
@@ -190,6 +209,14 @@ public void storeSettings() {
190209
if (apikeyPersistAvailableProperty.get()) {
191210
preferences.getImporterPreferences().getApiKeys().addAll(apiKeys);
192211
}
212+
213+
// Save custom URL templates to preferences
214+
Map<String, String> templates = searchEngines.stream()
215+
.collect(Collectors.toMap(
216+
SearchEngineItem::getName,
217+
SearchEngineItem::getUrlTemplate
218+
));
219+
preferences.getImporterPreferences().setSearchEngineUrlTemplates(templates);
193220
}
194221

195222
public BooleanProperty enableWebSearchProperty() {
@@ -260,6 +287,10 @@ public BooleanProperty getApikeyPersistProperty() {
260287
return apikeyPersistProperty;
261288
}
262289

290+
public ObservableList<SearchEngineItem> getSearchEngines() {
291+
return searchEngines;
292+
}
293+
263294
public void checkCustomApiKey() {
264295
final String apiKeyName = selectedApiKeyProperty.get().getName();
265296

‎jabgui/src/main/resources/org/jabref/gui/preferences/websearch/WebSearchTab.fxml

Lines changed: 30 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
<?import javafx.scene.layout.HBox?>
1212
<?import javafx.scene.layout.VBox?>
1313
<?import javafx.scene.control.ComboBox?>
14+
<?import org.jabref.gui.preferences.websearch.SearchEngineItem?>
1415
<fx:root spacing="10.0" type="VBox"
1516
xmlns="http://javafx.com/javafx" xmlns:fx="http://javafx.com/fxml"
1617
fx:controller="org.jabref.gui.preferences.websearch.WebSearchTab">
@@ -42,9 +43,9 @@
4243
editable="true">
4344
<columns>
4445
<TableColumn minWidth="90" prefWidth="70"
45-
fx:id="catalogEnabledColumn"
46-
text="%Enabled"
47-
/>
46+
fx:id="catalogEnabledColumn"
47+
text="%Enabled"
48+
/>
4849
<TableColumn
4950
fx:id="catalogColumn"
5051
text="%Catalog"/>
@@ -62,6 +63,28 @@
6263
<TextField fx:id="grobidURL" HBox.hgrow="ALWAYS"/>
6364
</HBox>
6465

66+
<Label styleClass="sectionHeader" text="%Search Engine URL Templates"/>
67+
<Label text="%( Note: Press return to commit changes in the table! )"/>
68+
<TableView
69+
fx:id="searchEngineTable"
70+
VBox.vgrow="ALWAYS"
71+
editable="true">
72+
<columns>
73+
<TableColumn minWidth="120"
74+
fx:id="searchEngineName"
75+
text="%Search Engine"
76+
editable="false"/>
77+
<TableColumn minWidth="300"
78+
fx:id="searchEngineUrlTemplate"
79+
text="%URL Template"
80+
editable="true"/>
81+
</columns>
82+
<columnResizePolicy>
83+
<TableView
84+
fx:constant="CONSTRAINED_RESIZE_POLICY"/>
85+
</columnResizePolicy>
86+
</TableView>
87+
6588
<Label styleClass="sectionHeader" text="%Custom API key"/>
6689
<Label text="%( Note: Press return to commit changes in the table! )"/>
6790
<TableView
@@ -74,12 +97,12 @@
7497
text="%Catalog"
7598
/>
7699
<TableColumn minWidth="120"
77-
fx:id="customApiKey"
78-
text="Key"/>
100+
fx:id="customApiKey"
101+
text="Key"/>
79102

80103
<TableColumn minWidth="120" prefWidth="120"
81-
fx:id="useCustomApiKey"
82-
text="Use Custom Key"/>
104+
fx:id="useCustomApiKey"
105+
text="Use Custom Key"/>
83106
</columns>
84107
<columnResizePolicy>
85108
<TableView

0 commit comments

Comments
 (0)