From 2fae126d9f3e5f25fd318b7143c1ade8ddc81112 Mon Sep 17 00:00:00 2001
From: G3G4X5X6 <87740076+G3G4X5X6@users.noreply.github.com>
Date: Fri, 22 Nov 2024 18:49:28 +0800
Subject: [PATCH] Update TemplatesPanel.java

#36
---
 .../com/g3g4x5x6/nuclei/panel/tab/TemplatesPanel.java    | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/nuclei/src/main/java/com/g3g4x5x6/nuclei/panel/tab/TemplatesPanel.java b/nuclei/src/main/java/com/g3g4x5x6/nuclei/panel/tab/TemplatesPanel.java
index d04fc3a..51ad9b9 100644
--- a/nuclei/src/main/java/com/g3g4x5x6/nuclei/panel/tab/TemplatesPanel.java
+++ b/nuclei/src/main/java/com/g3g4x5x6/nuclei/panel/tab/TemplatesPanel.java
@@ -457,6 +457,15 @@ public void menuCanceled(MenuEvent e) {
     private void refreshDataForTable() {
         // 搜索功能
         sorter = new TableRowSorter<>(tableModel);
+        sorter.setComparator(0, (o1, o2) -> {
+            try {
+                int num1 = Integer.parseInt(o1.toString());
+                int num2 = Integer.parseInt(o2.toString());
+                return Integer.compare(num1, num2);
+            } catch (NumberFormatException ex) {
+                return o1.toString().compareTo(o2.toString());
+            }
+        });
         templatesTable.setRowSorter(sorter);
         new Thread(() -> {
             NucleiFrame.setStatusProgressBar(L.M("bar.tool.status.progressbar.text", "正在刷新模板列表，请稍后..."));