Add HuaSheng output control

1 files changed, 18 insertions, 4 deletions

diff --git a/src/main/java/org/traccar/protocol/HuaShengProtocolEncoder.java b/src/main/java/org/traccar/protocol/HuaShengProtocolEncoder.java
index 636196ec4..dc34f7b4e 100644
--- a/src/main/java/org/traccar/protocol/HuaShengProtocolEncoder.java
+++ b/src/main/java/org/traccar/protocol/HuaShengProtocolEncoder.java
@@ -27,13 +27,13 @@ public class HuaShengProtocolEncoder extends BaseProtocolEncoder {
         super(protocol);
     }
 
-    private ByteBuf encodeContent(ByteBuf content) {
+    private ByteBuf encodeContent(int type, ByteBuf content) {
 
         ByteBuf buf = Unpooled.buffer();
         buf.writeByte(0xC0);
         buf.writeShort(0x0000); // flag and version
         buf.writeShort(12 + content.readableBytes());
-        buf.writeShort(HuaShengProtocolDecoder.MSG_SET_REQ);
+        buf.writeShort(type);
         buf.writeShort(0); // checksum
         buf.writeInt(1); // index
         buf.writeBytes(content);
@@ -52,17 +52,31 @@ public class HuaShengProtocolEncoder extends BaseProtocolEncoder {
                 content.writeShort(0x0002);
                 content.writeShort(6); // length
                 content.writeShort(command.getInteger(Command.KEY_FREQUENCY));
-                return encodeContent(content);
+                return encodeContent(HuaShengProtocolDecoder.MSG_SET_REQ, content);
+            case Command.TYPE_OUTPUT_CONTROL:
+                /*
+0x01: Lock the relay1; //relay on
+0x02: Unlock the relay1; //relay off
+0x03: Lock the relay2; //relay2 on
+0x04: Unlock the relay2; //relay2 off
+0x05: Lock the relay3; //relay3 on
+0x06: Unlock the relay3; //realy3 off
+                 */
+                content.writeByte(
+                        (command.getInteger(Command.KEY_INDEX) - 1) * 2
+                        + (2 - command.getInteger(Command.KEY_DATA)));
+                return encodeContent(HuaShengProtocolDecoder.MSG_CTRL_REQ, content);
             case Command.TYPE_ALARM_ARM:
             case Command.TYPE_ALARM_DISARM:
                 content.writeShort(0x0001);
                 content.writeShort(5); // length
                 content.writeByte(command.getType().equals(Command.TYPE_ALARM_ARM) ? 1 : 0);
+                return encodeContent(HuaShengProtocolDecoder.MSG_SET_REQ, content);
             case Command.TYPE_SET_SPEED_LIMIT:
                 content.writeShort(0x0004);
                 content.writeShort(6); // length
                 content.writeShort(command.getInteger(Command.KEY_DATA));
-                return encodeContent(content);
+                return encodeContent(HuaShengProtocolDecoder.MSG_SET_REQ, content);
             default:
                 return null;
         }