package com.wecom.robot.dto;

import com.alibaba.fastjson.annotation.JSONField;
import lombok.Data;

@Data
public class WxSendMessageRequest {

    private String touser;
    
    @JSONField(name = "open_kfid")
    private String openKfid;
    
    private String msgtype;
    private TextContent text;
    private ImageContent image;
    private LinkContent link;

    @Data
    public static class TextContent {
        private String content;
    }

    @Data
    public static class ImageContent {
        private String mediaId;
    }

    @Data
    public static class LinkContent {
        private String title;
        private String desc;
        private String url;
        private String thumbMediaId;
    }

    public static WxSendMessageRequest text(String touser, String openKfid, String content) {
        WxSendMessageRequest request = new WxSendMessageRequest();
        request.setTouser(touser);
        request.setOpenKfid(openKfid);
        request.setMsgtype("text");
        TextContent textContent = new TextContent();
        textContent.setContent(content);
        request.setText(textContent);
        return request;
    }

    public static WxSendMessageRequest image(String touser, String openKfid, String mediaId) {
        WxSendMessageRequest request = new WxSendMessageRequest();
        request.setTouser(touser);
        request.setOpenKfid(openKfid);
        request.setMsgtype("image");
        ImageContent imageContent = new ImageContent();
        imageContent.setMediaId(mediaId);
        request.setImage(imageContent);
        return request;
    }
}