Skip to content

Commit 2db758b

Browse files
dengziidengzii
committed
💥 Support export and import template to file.
1 parent 1173edc commit 2db758b

File tree

5 files changed

+114
-21
lines changed

5 files changed

+114
-21
lines changed

‎src/com/dengzii/plugin/template/Config.kt‎

Lines changed: 2 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
11
package com.dengzii.plugin.template
22

3-
import com.dengzii.plugin.template.model.FileTreeNode
43
import com.dengzii.plugin.template.model.Module
54
import com.dengzii.plugin.template.utils.Logger
65
import com.google.gson.GsonBuilder
@@ -22,7 +21,7 @@ object Config {
2221
private const val KEY_TEMPLATES = "KEY_TEMPLATES"
2322
private const val KEY_INIT = "KEY_INIT"
2423

25-
private val GSON by lazy { GsonBuilder().setLenient().create() }
24+
val GSON by lazy { GsonBuilder().setLenient().create() }
2625
private val STORE by lazy { PropertiesComponent.getInstance() }
2726

2827
private fun clear() = STORE.unsetValue(KEY_TEMPLATES)
@@ -45,7 +44,7 @@ object Config {
4544
arr.filter { !it.isNullOrBlank() }.forEach {
4645
try {
4746
val module = GSON.fromJson(it, Module::class.java)
48-
setParent(module.template)
47+
module.initTemplate()
4948
result.add(module)
5049
} catch (e: Exception) {
5150
clear()
@@ -70,13 +69,4 @@ object Config {
7069
STORE.setValue(KEY_INIT, false, false)
7170
STORE.setValues(KEY_TEMPLATES, t.toTypedArray())
7271
}
73-
74-
private fun setParent(node: FileTreeNode) {
75-
node.children.forEach {
76-
it.parent = node
77-
if (it.isDir) {
78-
setParent(it)
79-
}
80-
}
81-
}
8272
}

‎src/com/dengzii/plugin/template/CreateModuleAction.kt‎

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@ import com.dengzii.plugin.template.utils.Logger
55
import com.dengzii.plugin.template.utils.PluginKit
66
import com.intellij.openapi.actionSystem.AnAction
77
import com.intellij.openapi.actionSystem.AnActionEvent
8+
import com.intellij.openapi.project.Project
89

910
/**
1011
* <pre>
@@ -16,12 +17,18 @@ import com.intellij.openapi.actionSystem.AnActionEvent
1617
</pre> *
1718
*/
1819
class CreateModuleAction : AnAction() {
20+
21+
companion object {
22+
var project: Project? = null
23+
}
24+
1925
override fun actionPerformed(e: AnActionEvent) {
2026
val kit = PluginKit.get(e)
2127
if (!kit.isProjectValid()) {
2228
Logger.d(CreateModuleAction::class.java.simpleName, "Project is not valid.")
2329
return
2430
}
31+
project = kit.project
2532
CreateModuleDialog.createAndShow(kit.project) {
2633
FileWriteCommand.startAction(kit, it)
2734
}

‎src/com/dengzii/plugin/template/model/Module.kt‎

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,6 @@ package com.dengzii.plugin.template.model
33
import com.dengzii.plugin.template.template.AucTemplate
44
import com.dengzii.plugin.template.template.Template
55
import com.dengzii.plugin.template.utils.Logger
6-
import com.intellij.packaging.impl.elements.TestModuleOutputPackagingElement
76

87
class Module(
98
var template: FileTreeNode,
@@ -53,6 +52,15 @@ class Module(
5352
JAVA;
5453
}
5554

55+
fun initTemplate(node: FileTreeNode = template) {
56+
node.children.forEach {
57+
it.parent = node
58+
if (it.isDir) {
59+
initTemplate(it)
60+
}
61+
}
62+
}
63+
5664
fun clone(): Module {
5765
return create(template.clone(), templateName)
5866
}

‎src/com/dengzii/plugin/template/ui/ConfigurePanel.java‎

Lines changed: 85 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,13 @@
11
package com.dengzii.plugin.template.ui;
22

33
import com.dengzii.plugin.template.Config;
4+
import com.dengzii.plugin.template.CreateModuleAction;
45
import com.dengzii.plugin.template.model.Module;
56
import com.dengzii.plugin.template.utils.PopMenuUtils;
7+
import com.intellij.openapi.fileChooser.FileChooser;
8+
import com.intellij.openapi.fileChooser.FileChooserDescriptor;
9+
import com.intellij.openapi.fileChooser.FileChooserDescriptorFactory;
10+
import com.intellij.openapi.vfs.VirtualFile;
611
import com.intellij.ui.DocumentAdapter;
712
import com.intellij.ui.components.JBList;
813
import com.intellij.ui.components.JBTabbedPane;
@@ -13,6 +18,7 @@
1318
import java.awt.*;
1419
import java.awt.event.MouseAdapter;
1520
import java.awt.event.MouseEvent;
21+
import java.io.*;
1622
import java.util.HashMap;
1723
import java.util.List;
1824
import java.util.Map;
@@ -96,6 +102,10 @@ public void mouseClicked(MouseEvent e) {
96102
onCopyConfig();
97103
return null;
98104
});
105+
actionbar.onExport(() -> {
106+
onExportTemplate();
107+
return null;
108+
});
99109

100110
cbPlaceholder.addChangeListener(e -> {
101111
panelPreview.setReplacePlaceholder(cbPlaceholder.isSelected());
@@ -105,7 +115,15 @@ public void mouseClicked(MouseEvent e) {
105115
listTemplate.addListSelectionListener(e -> {
106116
if (noSelectedConfig()) return;
107117
onConfigSelect(getSelectedConfigIndex());
108-
118+
});
119+
listTemplate.addMouseListener(new MouseAdapter() {
120+
@Override
121+
public void mouseClicked(MouseEvent e) {
122+
super.mouseClicked(e);
123+
if (e.getButton() != MouseEvent.BUTTON3) {
124+
return;
125+
}
126+
}
109127
});
110128
//noinspection unchecked
111129
listTemplate.setModel(templateListModel);
@@ -148,11 +166,13 @@ private void onAddConfig(MouseEvent e) {
148166
items.put("Auc Pkg", () -> addModuleTemplate(Module.Companion.getAucPkg()));
149167
items.put("Auc Export", () -> addModuleTemplate(Module.Companion.getAucExport()));
150168
items.put("Android Mvp", () -> addModuleTemplate(Module.Companion.getAndroidMvp()));
169+
items.put("=> Import From File", this::onImportTemplate);
151170
PopMenuUtils.INSTANCE.create(items).show(actionbar, e.getX(), e.getY());
152171
}
153172

154173
private void addModuleTemplate(Module module) {
155174
configs.add(module);
175+
currentConfig = module;
156176
templateListModel.addElement(module.getTemplateName());
157177
listTemplate.doLayout();
158178
listTemplate.setSelectedIndex(configs.indexOf(module));
@@ -188,6 +208,70 @@ private void loadConfig() {
188208
}
189209
}
190210

211+
private void onExportTemplate() {
212+
213+
FileChooserDescriptor descriptor = FileChooserDescriptorFactory.createSingleFolderDescriptor();
214+
descriptor.setTitle("Save Template to File");
215+
VirtualFile vf = FileChooser.chooseFile(descriptor,
216+
CreateModuleAction.Companion.getProject(),
217+
null);
218+
if (vf != null && vf.isWritable()) {
219+
String config = Config.INSTANCE.getGSON().toJson(currentConfig);
220+
File file = new File(vf.getPath(), currentConfig.getTemplateName() + ".json");
221+
OutputStreamWriter outputStream = null;
222+
try {
223+
outputStream = new OutputStreamWriter(new FileOutputStream(file));
224+
outputStream.write(config);
225+
} catch (IOException e) {
226+
e.printStackTrace();
227+
} finally {
228+
if (outputStream != null) {
229+
try {
230+
outputStream.flush();
231+
outputStream.close();
232+
} catch (IOException e) {
233+
e.printStackTrace();
234+
}
235+
}
236+
}
237+
}
238+
}
239+
240+
private void onImportTemplate() {
241+
242+
FileChooserDescriptor descriptor = FileChooserDescriptorFactory.createSingleFileDescriptor("json");
243+
descriptor.setTitle("Import Template From File");
244+
VirtualFile vf = FileChooser.chooseFile(descriptor,
245+
CreateModuleAction.Companion.getProject(),
246+
null);
247+
if (vf != null && vf.exists()) {
248+
File file = new File(vf.getPath());
249+
BufferedInputStream inputStream = null;
250+
try {
251+
inputStream = new BufferedInputStream(new FileInputStream(file));
252+
byte[] bytes = new byte[1024];
253+
int len = 0;
254+
StringBuilder stringBuilder = new StringBuilder();
255+
while ((len = inputStream.read(bytes)) > 0) {
256+
stringBuilder.append(new String(bytes, 0, len));
257+
}
258+
Module template = Config.INSTANCE.getGSON().fromJson(stringBuilder.toString(), Module.class);
259+
template.initTemplate(template.getTemplate());
260+
addModuleTemplate(template);
261+
} catch (IOException e) {
262+
e.printStackTrace();
263+
} finally {
264+
if (inputStream != null) {
265+
try {
266+
inputStream.close();
267+
} catch (IOException e) {
268+
e.printStackTrace();
269+
}
270+
}
271+
}
272+
}
273+
}
274+
191275
private void onConfigSelect(int index) {
192276
if (currentConfig == configs.get(index)) {
193277
return;

‎src/com/dengzii/plugin/template/ui/EditToolbar.kt‎

Lines changed: 11 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,16 +1,12 @@
11
package com.dengzii.plugin.template.ui
22

33
import com.intellij.icons.AllIcons
4-
import com.intellij.util.ui.JBEmptyBorder
54
import java.awt.Dimension
65
import java.awt.FlowLayout
7-
import java.awt.event.ActionEvent
8-
import java.awt.event.ActionListener
96
import java.awt.event.MouseAdapter
107
import javax.swing.Icon
118
import javax.swing.JButton
129
import javax.swing.JPanel
13-
import javax.swing.border.Border
1410

1511
/**
1612
* <pre>
@@ -26,6 +22,7 @@ class EditToolbar : JPanel() {
2622
private val btAdd = JButton("Add")
2723
private val btRemove = JButton("Remove")
2824
private val btCopy = JButton("Copy")
25+
private val btShare = JButton("Export")
2926

3027
init {
3128
initLayout()
@@ -36,20 +33,26 @@ class EditToolbar : JPanel() {
3633
btAdd.addMouseListener(listener)
3734
}
3835

39-
fun onAdd(listener: ()->Unit) {
36+
fun onAdd(listener: () -> Unit) {
4037
btAdd.addActionListener {
4138
listener()
4239
}
4340
}
4441

4542
fun onRemove(listener: () -> Unit) {
46-
btRemove.addActionListener{
43+
btRemove.addActionListener {
4744
listener()
4845
}
4946
}
5047

5148
fun onCopy(listener: () -> Unit) {
52-
btCopy.addActionListener{
49+
btCopy.addActionListener {
50+
listener()
51+
}
52+
}
53+
54+
fun onExport(listener: () -> Unit) {
55+
btShare.addActionListener {
5356
listener()
5457
}
5558
}
@@ -67,6 +70,7 @@ class EditToolbar : JPanel() {
6770
setIconButton(btAdd, AllIcons.General.Add)
6871
setIconButton(btRemove, AllIcons.General.Remove)
6972
setIconButton(btCopy, AllIcons.General.CopyHovered)
73+
setIconButton(btShare, AllIcons.Actions.Download)
7074
}
7175

7276
private fun setIconButton(button: JButton, icon: Icon) {

0 commit comments

Comments
 (0)