package chat.module.entity;

import chat.consts.ProtoConstants;
import chat.server.call.ResultCode;
import chat.server.moquette.message.ModifiedMqttPubAckMessage;
import chat.server.moquette.message.MqttFixedHeader;
import chat.server.moquette.message.MqttMessage;
import chat.server.moquette.message.MqttMessageIdVariableHeader;
import chat.server.moquette.message.MqttMessageType;
import chat.server.moquette.message.MqttQoS;
import chat.user.User;
import chat.user.UserStore;
import chat.util.ByteStrToStr;
import cn.wildfirechat.proto.WFCMessage;
import cn.wildfirechat.proto.WFCMessage.Message;

import frame.object.data.Entity;
import frame.object.data.ID;
import frame.persist.NamedSQL;
import frame.persist.SQLRunner;
import io.netty.buffer.ByteBuf;
import io.netty.buffer.Unpooled;

public class MessageRecord {

	private String id;
	private String spaceid;
	private String spacecode;
	private String typecode;
	private int type;
	private String path;
	private String content;
	private int persistflag;
	private String senderid;
	private long messageid;
	private String statuscode;
	private long timestamp;
	private String parentid;
	private int level;
	
	private User sender;
	private WFCMessage.Message message;
	
	public MessageRecord() {
		timestamp = System.currentTimeMillis();
	}
	
	public static MessageRecord getInstance(String spaceCode, String spaceId, Message message) throws Exception {
		MessageRecord result = new MessageRecord();
		
		result.id = ID.newValue();
		
		if (message.getContent().getType() == 1) {
			result.typecode = "text";
		}
		else if (message.getContent().getType() == 2) {
			result.typecode = "voice";
		}
		else if (message.getContent().getType() == 3) {
			result.typecode = "image";
		}
		else if (message.getContent().getType() == 4) {
			result.typecode = "location";
		}
		else if (message.getContent().getType() == 5) {
			result.typecode = "file";
		}
		else if (message.getContent().getType() == 6) {
			result.typecode = "video";
		}
		else if (message.getContent().getType() == 7) {
			result.typecode = "sticker";
		}
		else if (message.getContent().getType() == 8) {
			result.typecode = "richmedia";
		}
		else if (message.getContent().getType() == 104) {
			result.typecode = "groupNotiy";
		}
		else if (message.getContent().getType() == 90) {
			result.typecode = "newFriendNotify";
		}
		
		result.type = message.getContent().getType();
		result.spaceid = spaceId;
		result.spacecode = spaceCode;

		if (message.getContent().getType() == 90) {
			result.content = message.getContent().getContent();
		}
		else if (message.getContent().getSearchableContent() == null || ("").equals(message.getContent().getSearchableContent())) {
			result.content = ByteStrToStr.transformStr(message.getContent().getData());
		}
		else {
			result.content = message.getContent().getSearchableContent();
		}
		
		result.persistflag = message.getContent().getPersistFlag();
		result.senderid = message.getFromUser();
		result.messageid = message.getMessageId();
		result.statuscode = "send";
		result.timestamp = System.currentTimeMillis();
		result.sender = UserStore.getById(message.getFromUser());
		
		return result;
	}
	
	public static MessageRecord getTextInstance(String content, User sender) throws Exception {
		MessageRecord result = new MessageRecord();
		
		result.id = ID.newValue();
		result.messageid = newMessageId();
		result.sender = sender;
		result.senderid = sender.getId();
		result.content = content;
		
		return result;
	}

	public static MessageRecord getImageInstance(String content) {
		// TODO Auto-generated method stub
		return null;
	}

	public void load(Entity entity) throws Exception {
		// TODO Auto-generated method stub
		id = entity.getString("id");
		spaceid = entity.getString("spaceid");
		spacecode = entity.getString("spacecode");
		typecode = entity.getString("typecode");
		type = entity.getInteger("type");
		content = entity.getString("content");
		path = entity.getString("path");
		persistflag = entity.getInteger("persistflag");
		senderid = entity.getString("senderid");
		messageid = Long.parseLong(entity.getString("messageid"));
		statuscode = entity.getString("statuscode");
		timestamp = Long.parseLong(entity.getString("timestamp"));
		parentid = entity.getString("parentid");
		level = entity.getInteger("level");
	}
	
	public void pushTo(Entity entity) throws Exception {
		entity.set("id", id);
		entity.set("spaceid", spaceid);
		entity.set("spacecode", spacecode);
		entity.set("content", content);
		entity.set("typecode", typecode);
		entity.set("type", type);
		entity.set("content", content);
		entity.set("path", path);
		entity.set("persistflag", persistflag);
		entity.set("senderid", senderid);
		entity.set("content", content);
		entity.set("messageid", messageid);
		entity.set("statuscode", statuscode);
		entity.set("timestamp", timestamp);
		entity.set("parentid", parentid);
		entity.set("level", level);
	}
	
	public MqttMessage toMqttMessage(String targetId, WFCMessage.MessageContent content) {
		WFCMessage.Message wfcMessage = toWFCMessage(targetId, content);
		
		byte[] bytes = wfcMessage.toByteArray();
		
        ByteBuf buf = Unpooled.buffer();
        buf.ensureWritable(1).writeByte(ResultCode.Success.getValue());
        buf.ensureWritable(bytes.length).writeBytes(bytes);

        MqttMessageIdVariableHeader header = MqttMessageIdVariableHeader.from(Integer.parseInt(String.valueOf(messageid)));
        MqttFixedHeader fixedHeader = new MqttFixedHeader(MqttMessageType.PUBACK, false, MqttQoS.AT_MOST_ONCE, false, 0);
        ModifiedMqttPubAckMessage mqttMessage = new ModifiedMqttPubAckMessage(fixedHeader, header, buf);

		return mqttMessage;
	}
	
	public Message toWFCMessage(String targetId, WFCMessage.MessageContent contentMessage) {
		WFCMessage.Message message = null;
		
		try {
			WFCMessage.Message.Builder builder = WFCMessage.Message.newBuilder();
			if (contentMessage != null) {
				long timestamp = System.currentTimeMillis();
				builder = WFCMessage.Message.newBuilder().setContent(contentMessage).setServerTimestamp(timestamp);
	            builder.setConversation(builder.getConversationBuilder().setType(ProtoConstants.ConversationType.ConversationType_Group).setTarget(targetId).setLine(0));
	            builder.setFromUser("user000");
                long messageId = newMessageId();
                builder.setMessageId(messageId);
			} else {
				builder = WFCMessage.Message.newBuilder();
				builder.setMessageId(messageid);
				builder.setFromUser(senderid);
		        WFCMessage.Conversation.Builder cb = WFCMessage.Conversation.newBuilder();
		        cb.setType(0);
		        cb.setTarget(targetId);
		        cb.setLine(0);
		        builder.setConversation(cb.build());				
			}
			
	        WFCMessage.MessageContent.Builder contentBuilder = null;
	        if (contentMessage == null) {
		        contentBuilder = WFCMessage.MessageContent.newBuilder();
		        contentBuilder.setType(1);
		        contentBuilder.setSearchableContent(content);
		        contentBuilder.setMediaType(0);
		        contentBuilder.setPersistFlag(0);
		        contentBuilder.setExpireDuration(0);
		        contentBuilder.setMentionedType(0);	  
		        WFCMessage.MessageContent messageContent = contentBuilder.build();
		        
		        builder.setContent(messageContent);
		        builder.setServerTimestamp(System.currentTimeMillis());
	        }
			
			message = builder.build();
		}
		catch (Exception e) {
			e.printStackTrace();
		}		
			
		return message;
	}

	public String getSenderId() {
		return senderid;
	}

	public String getContent() {
		return content;
	}

	public void setSender(User sender) {
		if (sender == null) {
			this.sender = null;
			this.senderid = null;
			return;
		}
		
		this.sender = sender;
		this.senderid = sender.getId();
	}
	
	public WFCMessage.Message getWFCMessage() {
		return message;
	}

	public static int newMessageId() throws Exception {
		NamedSQL namedSQL = NamedSQL.getInstance("nextval");
		namedSQL.setParam("name", "messageId");
		
		int result = SQLRunner.getInteger(namedSQL);
		return result;
	}

	public String getId() {
		return id;
	}

	public boolean isFreshThan(long timestamp) {
		return this.timestamp > timestamp;
	}
	
	public long getMessageid() {
		return messageid;
	}

	public void setMessageid(long messageid) {
		this.messageid = messageid;
	}

	public WFCMessage.Message getMessage() {
		return message;
	}

	public void setMessage(WFCMessage.Message message) {
		this.message = message;
	}

	public String getTypecode() {
		return typecode;
	}

	public void setTypecode(String typecode) {
		this.typecode = typecode;
	}

	public int getPersistflag() {
		return persistflag;
	}

	public void setPersistflag(int persistflag) {
		this.persistflag = persistflag;
	}

	public int getType() {
		return type;
	}

	public void setType(int type) {
		this.type = type;
	}

	public long getTimestamp() {
		return timestamp;
	}

	public void setTimestamp(long timestamp) {
		this.timestamp = timestamp;
	}

	public String getSpacecode() {
		return spacecode;
	}

	public void setSpacecode(String spacecode) {
		this.spacecode = spacecode;
	}

	public String getSpaceid() {
		return spaceid;
	}

	public void setSpaceid(String spaceid) {
		this.spaceid = spaceid;
	}

}