Use correct prefix for commands

1 files changed, 28 insertions, 9 deletions

diff --git a/src/main/java/org/traccar/protocol/SuntechProtocolEncoder.java b/src/main/java/org/traccar/protocol/SuntechProtocolEncoder.java
index 55f2c73dd..3b4995110 100644
--- a/src/main/java/org/traccar/protocol/SuntechProtocolEncoder.java
+++ b/src/main/java/org/traccar/protocol/SuntechProtocolEncoder.java
@@ -15,6 +15,8 @@
  */
 package org.traccar.protocol;
 
+import io.netty.channel.Channel;
+import org.traccar.BasePipelineFactory;
 import org.traccar.StringProtocolEncoder;
 import org.traccar.model.Command;
 import org.traccar.Protocol;
@@ -25,32 +27,49 @@ public class SuntechProtocolEncoder extends StringProtocolEncoder {
         super(protocol);
     }
 
+    private String getPrefix(Channel channel) {
+        String prefix = "SA200CMD";
+        if (channel != null) {
+            SuntechProtocolDecoder protocolDecoder =
+                    BasePipelineFactory.getHandler(channel.pipeline(), SuntechProtocolDecoder.class);
+            if (protocolDecoder != null) {
+                String decoderPrefix = protocolDecoder.getPrefix();
+                if (decoderPrefix != null && decoderPrefix.length() > 5) {
+                    prefix = decoderPrefix.substring(0, decoderPrefix.length() - 3) + "CMD";
+                }
+            }
+        }
+        return prefix;
+    }
+
     @Override
-    protected Object encodeCommand(Command command) {
+    protected Object encodeCommand(Channel channel, Command command) {
+
+        String prefix = getPrefix(channel);
 
         switch (command.getType()) {
             case Command.TYPE_REBOOT_DEVICE:
-                return formatCommand(command, "SA200CMD;%s;02;Reboot\r", Command.KEY_UNIQUE_ID);
+                return formatCommand(command, prefix + ";%s;02;Reboot\r", Command.KEY_UNIQUE_ID);
             case Command.TYPE_POSITION_SINGLE:
-                return formatCommand(command, "SA200GTR;%s;02;\r", Command.KEY_UNIQUE_ID);
+                return formatCommand(command, prefix + ";%s;02;\r", Command.KEY_UNIQUE_ID);
             case Command.TYPE_OUTPUT_CONTROL:
                 if (command.getAttributes().containsKey(Command.KEY_DATA)) {
                     if (command.getAttributes().get(Command.KEY_DATA).equals("1")) {
-                        return formatCommand(command, "SA200CMD;%s;02;Enable%s\r",
+                        return formatCommand(command, prefix + ";%s;02;Enable%s\r",
                                 Command.KEY_UNIQUE_ID, Command.KEY_INDEX);
                     } else {
-                        return formatCommand(command, "SA200CMD;%s;02;Disable%s\r",
+                        return formatCommand(command, prefix + ";%s;02;Disable%s\r",
                                 Command.KEY_UNIQUE_ID, Command.KEY_INDEX);
                     }
                 }
             case Command.TYPE_ENGINE_STOP:
-                return formatCommand(command, "SA200CMD;%s;02;Enable1\r", Command.KEY_UNIQUE_ID);
+                return formatCommand(command, prefix + ";%s;02;Enable1\r", Command.KEY_UNIQUE_ID);
             case Command.TYPE_ENGINE_RESUME:
-                return formatCommand(command, "SA200CMD;%s;02;Disable1\r", Command.KEY_UNIQUE_ID);
+                return formatCommand(command, prefix + ";%s;02;Disable1\r", Command.KEY_UNIQUE_ID);
             case Command.TYPE_ALARM_ARM:
-                return formatCommand(command, "SA200CMD;%s;02;Enable2\r", Command.KEY_UNIQUE_ID);
+                return formatCommand(command, prefix + ";%s;02;Enable2\r", Command.KEY_UNIQUE_ID);
             case Command.TYPE_ALARM_DISARM:
-                return formatCommand(command, "SA200CMD;%s;02;Disable2\r", Command.KEY_UNIQUE_ID);
+                return formatCommand(command, prefix + ";%s;02;Disable2\r", Command.KEY_UNIQUE_ID);
             default:
                 return null;
         }