■ ■ ■ ■ ■ ■
jadx-gui/src/main/java/jadx/gui/ui/dialog/FileDialog.java
1 | | - | package jadx.gui.ui.dialog; |
2 | | - | |
3 | | - | import java.awt.Component; |
4 | | - | import java.awt.HeadlessException; |
5 | | - | import java.awt.event.WindowAdapter; |
6 | | - | import java.awt.event.WindowEvent; |
7 | | - | import java.io.File; |
8 | | - | import java.nio.file.Path; |
9 | | - | import java.util.ArrayList; |
10 | | - | import java.util.Arrays; |
11 | | - | import java.util.Collections; |
12 | | - | import java.util.List; |
13 | | - | |
14 | | - | import javax.swing.JDialog; |
15 | | - | import javax.swing.JFileChooser; |
16 | | - | import javax.swing.JOptionPane; |
17 | | - | import javax.swing.filechooser.FileNameExtensionFilter; |
18 | | - | |
19 | | - | import org.jetbrains.annotations.Nullable; |
20 | | - | |
21 | | - | import jadx.api.plugins.utils.CommonFileUtils; |
22 | | - | import jadx.core.utils.Utils; |
23 | | - | import jadx.core.utils.files.FileUtils; |
24 | | - | import jadx.gui.settings.JadxProject; |
25 | | - | import jadx.gui.ui.MainWindow; |
26 | | - | import jadx.gui.utils.NLS; |
27 | | - | |
28 | | - | public class FileDialog { |
29 | | - | |
30 | | - | public enum OpenMode { |
31 | | - | OPEN, |
32 | | - | OPEN_PROJECT, |
33 | | - | ADD, |
34 | | - | SAVE_PROJECT, |
35 | | - | EXPORT, |
36 | | - | CUSTOM_SAVE, |
37 | | - | CUSTOM_OPEN |
38 | | - | } |
39 | | - | |
40 | | - | private final MainWindow mainWindow; |
41 | | - | |
42 | | - | private boolean isOpen; |
43 | | - | private String title; |
44 | | - | private List<String> fileExtList; |
45 | | - | private int selectionMode = JFileChooser.FILES_AND_DIRECTORIES; |
46 | | - | private @Nullable Path currentDir; |
47 | | - | private @Nullable Path selectedFile; |
48 | | - | |
49 | | - | public FileDialog(MainWindow mainWindow, OpenMode mode) { |
50 | | - | this.mainWindow = mainWindow; |
51 | | - | initForMode(mode); |
52 | | - | } |
53 | | - | |
54 | | - | public void setTitle(String title) { |
55 | | - | this.title = title; |
56 | | - | } |
57 | | - | |
58 | | - | public void setFileExtList(List<String> fileExtList) { |
59 | | - | this.fileExtList = fileExtList; |
60 | | - | } |
61 | | - | |
62 | | - | public void setSelectionMode(int selectionMode) { |
63 | | - | this.selectionMode = selectionMode; |
64 | | - | } |
65 | | - | |
66 | | - | public void setSelectedFile(Path path) { |
67 | | - | this.selectedFile = path; |
68 | | - | } |
69 | | - | |
70 | | - | public void setCurrentDir(Path currentDir) { |
71 | | - | this.currentDir = currentDir; |
72 | | - | } |
73 | | - | |
74 | | - | public List<Path> show() { |
75 | | - | FileChooser fileChooser = buildFileChooser(); |
76 | | - | int ret = isOpen ? fileChooser.showOpenDialog(mainWindow) : fileChooser.showSaveDialog(mainWindow); |
77 | | - | if (ret != JFileChooser.APPROVE_OPTION) { |
78 | | - | return Collections.emptyList(); |
79 | | - | } |
80 | | - | currentDir = fileChooser.getCurrentDirectory().toPath(); |
81 | | - | File[] selectedFiles = fileChooser.getSelectedFiles(); |
82 | | - | if (selectedFiles.length != 0) { |
83 | | - | return FileUtils.toPaths(selectedFiles); |
84 | | - | } |
85 | | - | File chosenFile = fileChooser.getSelectedFile(); |
86 | | - | if (chosenFile != null) { |
87 | | - | return Collections.singletonList(chosenFile.toPath()); |
88 | | - | } |
89 | | - | return Collections.emptyList(); |
90 | | - | } |
91 | | - | |
92 | | - | public Path getCurrentDir() { |
93 | | - | return currentDir; |
94 | | - | } |
95 | | - | |
96 | | - | private void initForMode(OpenMode mode) { |
97 | | - | switch (mode) { |
98 | | - | case OPEN: |
99 | | - | case OPEN_PROJECT: |
100 | | - | case ADD: |
101 | | - | if (mode == OpenMode.OPEN_PROJECT) { |
102 | | - | fileExtList = Collections.singletonList(JadxProject.PROJECT_EXTENSION); |
103 | | - | title = NLS.str("file.open_title"); |
104 | | - | } else { |
105 | | - | fileExtList = new ArrayList<>(Arrays.asList("apk", "dex", "jar", "class", "smali", "zip", "xapk", "aar", "arsc")); |
106 | | - | if (mode == OpenMode.OPEN) { |
107 | | - | fileExtList.addAll(Arrays.asList(JadxProject.PROJECT_EXTENSION, "aab")); |
108 | | - | title = NLS.str("file.open_title"); |
109 | | - | } else { |
110 | | - | title = NLS.str("file.add_files_action"); |
111 | | - | } |
112 | | - | } |
113 | | - | selectionMode = JFileChooser.FILES_AND_DIRECTORIES; |
114 | | - | currentDir = mainWindow.getSettings().getLastOpenFilePath(); |
115 | | - | isOpen = true; |
116 | | - | break; |
117 | | - | |
118 | | - | case SAVE_PROJECT: |
119 | | - | title = NLS.str("file.save_project"); |
120 | | - | fileExtList = Collections.singletonList(JadxProject.PROJECT_EXTENSION); |
121 | | - | selectionMode = JFileChooser.FILES_ONLY; |
122 | | - | currentDir = mainWindow.getSettings().getLastSaveFilePath(); |
123 | | - | isOpen = false; |
124 | | - | break; |
125 | | - | |
126 | | - | case EXPORT: |
127 | | - | title = NLS.str("file.save_all_msg"); |
128 | | - | fileExtList = Collections.emptyList(); |
129 | | - | selectionMode = JFileChooser.DIRECTORIES_ONLY; |
130 | | - | currentDir = mainWindow.getSettings().getLastSaveFilePath(); |
131 | | - | isOpen = false; |
132 | | - | break; |
133 | | - | |
134 | | - | case CUSTOM_SAVE: |
135 | | - | isOpen = false; |
136 | | - | break; |
137 | | - | |
138 | | - | case CUSTOM_OPEN: |
139 | | - | isOpen = true; |
140 | | - | break; |
141 | | - | } |
142 | | - | } |
143 | | - | |
144 | | - | private FileChooser buildFileChooser() { |
145 | | - | FileChooser fileChooser = new FileChooser(currentDir); |
146 | | - | fileChooser.setToolTipText(title); |
147 | | - | fileChooser.setFileSelectionMode(selectionMode); |
148 | | - | fileChooser.setMultiSelectionEnabled(isOpen); |
149 | | - | fileChooser.setAcceptAllFileFilterUsed(true); |
150 | | - | if (Utils.notEmpty(fileExtList)) { |
151 | | - | String description = NLS.str("file_dialog.supported_files") + ": (" + Utils.listToString(fileExtList) + ')'; |
152 | | - | fileChooser.setFileFilter(new FileNameExtensionFilter(description, fileExtList.toArray(new String[0]))); |
153 | | - | } |
154 | | - | if (selectedFile != null) { |
155 | | - | fileChooser.setSelectedFile(selectedFile.toFile()); |
156 | | - | } |
157 | | - | return fileChooser; |
158 | | - | } |
159 | | - | |
160 | | - | private class FileChooser extends JFileChooser { |
161 | | - | |
162 | | - | public FileChooser(@Nullable Path currentDirectory) { |
163 | | - | super(currentDirectory == null ? CommonFileUtils.CWD : currentDirectory.toFile()); |
164 | | - | } |
165 | | - | |
166 | | - | @Override |
167 | | - | protected JDialog createDialog(Component parent) throws HeadlessException { |
168 | | - | JDialog dialog = super.createDialog(parent); |
169 | | - | dialog.setTitle(title); |
170 | | - | dialog.setLocationRelativeTo(null); |
171 | | - | mainWindow.getSettings().loadWindowPos(dialog); |
172 | | - | dialog.addWindowListener(new WindowAdapter() { |
173 | | - | @Override |
174 | | - | public void windowClosed(WindowEvent e) { |
175 | | - | mainWindow.getSettings().saveWindowPos(dialog); |
176 | | - | super.windowClosed(e); |
177 | | - | } |
178 | | - | }); |
179 | | - | return dialog; |
180 | | - | } |
181 | | - | |
182 | | - | @Override |
183 | | - | public void approveSelection() { |
184 | | - | if (selectionMode == FILES_AND_DIRECTORIES) { |
185 | | - | File currentFile = getSelectedFile(); |
186 | | - | if (currentFile.isDirectory()) { |
187 | | - | int option = JOptionPane.showConfirmDialog( |
188 | | - | mainWindow, |
189 | | - | NLS.str("file_dialog.load_dir_confirm") + "\n " + currentFile, |
190 | | - | NLS.str("file_dialog.load_dir_title"), |
191 | | - | JOptionPane.YES_NO_OPTION); |
192 | | - | if (option != JOptionPane.YES_OPTION) { |
193 | | - | this.setCurrentDirectory(currentFile); |
194 | | - | this.updateUI(); |
195 | | - | return; |
196 | | - | } |
197 | | - | } |
198 | | - | } |
199 | | - | super.approveSelection(); |
200 | | - | } |
201 | | - | } |
202 | | - | } |
203 | | - | |