From 7e912f9f3de47eba380975f20b63671f287b25e8 Mon Sep 17 00:00:00 2001
From: godotg <jaysunxiao@gmail.com>
Date: Sat, 20 Apr 2024 15:03:45 +0800
Subject: [PATCH] feat[es]: es support

---
 .../java/com/zfoo/protocol/generate/GenerateProtocolNote.java   | 1 +
 .../java/com/zfoo/protocol/generate/GenerateProtocolPath.java   | 2 ++
 .../java/com/zfoo/protocol/serializer/es/GenerateEsUtils.java   | 2 +-
 .../test/java/com/zfoo/protocol/generate/GenerateTesting.java   | 1 +
 4 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/protocol/src/main/java/com/zfoo/protocol/generate/GenerateProtocolNote.java b/protocol/src/main/java/com/zfoo/protocol/generate/GenerateProtocolNote.java
index 11d6e795..7eb0d1d0 100644
--- a/protocol/src/main/java/com/zfoo/protocol/generate/GenerateProtocolNote.java
+++ b/protocol/src/main/java/com/zfoo/protocol/generate/GenerateProtocolNote.java
@@ -93,6 +93,7 @@ public abstract class GenerateProtocolNote {
             case Cpp:
             case Go:
             case JavaScript:
+            case ES:
             case TypeScript:
             case CSharp:
             case Protobuf:
diff --git a/protocol/src/main/java/com/zfoo/protocol/generate/GenerateProtocolPath.java b/protocol/src/main/java/com/zfoo/protocol/generate/GenerateProtocolPath.java
index 2e387c48..c9da7af0 100644
--- a/protocol/src/main/java/com/zfoo/protocol/generate/GenerateProtocolPath.java
+++ b/protocol/src/main/java/com/zfoo/protocol/generate/GenerateProtocolPath.java
@@ -59,6 +59,8 @@ public abstract class GenerateProtocolPath {
                 break;
             case JavaScript:
                 break;
+            case ES:
+                break;
             case TypeScript:
                 break;
             case CSharp:
diff --git a/protocol/src/main/java/com/zfoo/protocol/serializer/es/GenerateEsUtils.java b/protocol/src/main/java/com/zfoo/protocol/serializer/es/GenerateEsUtils.java
index 27ec7b02..0d69e956 100644
--- a/protocol/src/main/java/com/zfoo/protocol/serializer/es/GenerateEsUtils.java
+++ b/protocol/src/main/java/com/zfoo/protocol/serializer/es/GenerateEsUtils.java
@@ -98,7 +98,7 @@ public abstract class GenerateEsUtils {
             var protocolId = protocol.protocolId();
             var protocolName = protocol.protocolConstructor().getDeclaringClass().getSimpleName();
             var path = GenerateProtocolPath.protocolAbsolutePath(protocol.protocolId(), CodeLanguage.ES);
-            importBuilder.append(StringUtils.format("import {} from './{}';", protocolName, path)).append(LS);
+            importBuilder.append(StringUtils.format("import {} from './{}.mjs';", protocolName, path)).append(LS);
             initProtocolBuilder.append(StringUtils.format("protocols.set({}, {});", protocolId, protocolName)).append(LS);
         }
 
diff --git a/protocol/src/test/java/com/zfoo/protocol/generate/GenerateTesting.java b/protocol/src/test/java/com/zfoo/protocol/generate/GenerateTesting.java
index 4129d321..9987c335 100644
--- a/protocol/src/test/java/com/zfoo/protocol/generate/GenerateTesting.java
+++ b/protocol/src/test/java/com/zfoo/protocol/generate/GenerateTesting.java
@@ -38,6 +38,7 @@ public class GenerateTesting {
         generateLanguages.add(CodeLanguage.Cpp);
         generateLanguages.add(CodeLanguage.Go);
         generateLanguages.add(CodeLanguage.JavaScript);
+        generateLanguages.add(CodeLanguage.ES);
         generateLanguages.add(CodeLanguage.TypeScript);
         generateLanguages.add(CodeLanguage.Lua);
         generateLanguages.add(CodeLanguage.CSharp);