/*
 * Copyright 2015 - 2019 Anton Tananaev (anton@traccar.org)
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.traccar.protocol;

import io.netty.channel.Channel;
import org.traccar.BasePipelineFactory;
import org.traccar.Context;
import org.traccar.StringProtocolEncoder;
import org.traccar.model.Command;
import org.traccar.Protocol;

public class SuntechProtocolEncoder extends StringProtocolEncoder {
	
    public SuntechProtocolEncoder(Protocol protocol) {
        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;
    }
    
    private boolean getM2M(Channel channel, long deviceId) {
    	boolean m2m = false;
    	if (channel != null) {
    		SuntechProtocolDecoder protocolDecoder =
            		BasePipelineFactory.getHandler(channel.pipeline(), SuntechProtocolDecoder.class);
    		if (protocolDecoder != null) {
    			m2m = protocolDecoder.isM2M(deviceId);
    		}
    	}
    	return m2m;
    }

    @Override
    protected Object encodeCommand(Channel channel, Command command) {
        boolean m2m = getM2M(channel, command.getDeviceId());
        
        if (m2m) {
        	String prefix = "CMD";
        	switch (command.getType()) {
	            case Command.TYPE_REBOOT_DEVICE:
	                return formatCommand(command, prefix + ";%s;03;03\r", Command.KEY_UNIQUE_ID);
	            case Command.TYPE_POSITION_SINGLE:
	                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, prefix + ";%s;02;Enable%s\r",
	                                Command.KEY_UNIQUE_ID, Command.KEY_INDEX);
	                    } else {
	                        return formatCommand(command, prefix + ";%s;02;Disable%s\r",
	                                Command.KEY_UNIQUE_ID, Command.KEY_INDEX);
	                    }
	                }
	            case Command.TYPE_ENGINE_STOP:
	                return formatCommand(command, prefix + ";%s;04;01\r", Command.KEY_UNIQUE_ID);
	            case Command.TYPE_ENGINE_RESUME:
	                return formatCommand(command, prefix + ";%s;04;02\r", Command.KEY_UNIQUE_ID);
	            case Command.TYPE_ALARM_ARM:
	                return formatCommand(command, prefix + ";%s;04;03\r", Command.KEY_UNIQUE_ID);
	            case Command.TYPE_ALARM_DISARM:
	                return formatCommand(command, prefix + ";%s;04;04\r", Command.KEY_UNIQUE_ID);
	            default:
	                return null;
	        }
        } else {
        	String prefix = getPrefix(channel);
	        switch (command.getType()) {
	            case Command.TYPE_REBOOT_DEVICE:
	                return formatCommand(command, prefix + ";%s;02;Reboot\r", Command.KEY_UNIQUE_ID);
	            case Command.TYPE_POSITION_SINGLE:
	                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, prefix + ";%s;02;Enable%s\r",
	                                Command.KEY_UNIQUE_ID, Command.KEY_INDEX);
	                    } else {
	                        return formatCommand(command, prefix + ";%s;02;Disable%s\r",
	                                Command.KEY_UNIQUE_ID, Command.KEY_INDEX);
	                    }
	                }
	            case Command.TYPE_ENGINE_STOP:
	                return formatCommand(command, prefix + ";%s;02;Enable1\r", Command.KEY_UNIQUE_ID);
	            case Command.TYPE_ENGINE_RESUME:
	                return formatCommand(command, prefix + ";%s;02;Disable1\r", Command.KEY_UNIQUE_ID);
	            case Command.TYPE_ALARM_ARM:
	                return formatCommand(command, prefix + ";%s;02;Enable2\r", Command.KEY_UNIQUE_ID);
	            case Command.TYPE_ALARM_DISARM:
	                return formatCommand(command, prefix + ";%s;02;Disable2\r", Command.KEY_UNIQUE_ID);
	            default:
	                return null;
	        }
        }
    }

}