import com.alibaba.fastjson2.JSON;
import lombok.Getter;
import lombok.Setter;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

public class FastJson2Test {

    public static void main(String[] args) {
        SonInDto1 sonInDto1 = new SonInDto1("type.xxxxx1", "xxxxxx1", new SonInDto1.Data("张三"));
        SonInDto2 sonInDto2 = new SonInDto2("type.xxxxx2", "xxxxxx2", new SonInDto2.Data("风格"));
        System.out.println("sonInDto2: " + JSON.toJSONString(sonInDto2.buildWsRequest()));
        System.out.println("sonInDto1: " + JSON.toJSONString(sonInDto1.buildWsRequest()));
    }

    @Getter
    @Setter
    public static abstract class BaseInDto {
        protected String type;
        protected String traceId;

        @lombok.Data
        public static class Data {
            protected String query;
        }

        public abstract BaseInDto.Data getData();

        public OutDto buildWsRequest() {
            String[] tp = this.type.split("\\.");
            String method = tp[0];
            String subMethod = tp[1];
            OutDto inDto = new OutDto();
            inDto.setMethod(method);
            inDto.setSubMethod(subMethod);
            OutDto.Input input = new OutDto.Input();
            input.setType("text");
            input.setData(this.getData().getQuery());
            inDto.setInputs(new ArrayList<>(List.of(input)));
            return inDto;
        }
    }

    @Getter
    @Setter
    public static class SonInDto1 extends BaseInDto {
        protected SonInDto1.Data data;

        public SonInDto1(String type, String traceId, SonInDto1.Data data) {
            this.type = type;
            this.traceId = traceId;
            this.data = data;
        }

        @Getter
        @Setter
        public static class Data extends BaseInDto.Data {
            protected String name;

            public Data(String name) {
                this.name = name;
            }
        }

        @Getter
        @Setter
        public static class Extension extends OutDto.Extension {
            private SonInDto1.Data musicInfo;
        }

        @Override
        public OutDto buildWsRequest() {
            OutDto inDto = super.buildWsRequest();
            SonInDto1.Extension extension = new SonInDto1.Extension();
            this.data.setQuery(null);
            extension.setMusicInfo(this.data);
            inDto.setExtension(extension);
            inDto.setInputs(Collections.EMPTY_LIST);
            return inDto;
        }
    }

    @Getter
    @Setter
    public static class SonInDto2 extends BaseInDto {
        protected SonInDto2.Data data;

        public SonInDto2(String type, String traceId, SonInDto2.Data data) {
            this.type = type;
            this.traceId = traceId;
            this.data = data;
        }

        @Getter
        @Setter
        public static class Data extends BaseInDto.Data {
            protected String style;

            public Data(String style) {
                this.style = style;
            }
        }

        @Override
        public OutDto buildWsRequest() {
            OutDto inDto = super.buildWsRequest();
            OutDto.Extension extension = new OutDto.Extension();
            extension.setStyle(this.data.style);
            inDto.setExtension(extension);
            return inDto;
        }
    }


    @Getter
    @Setter
    public static class OutDto {
        private String method;
        private String subMethod;
        private List<OutDto.Input> inputs;
        private OutDto.Extension extension;

        @Getter
        @Setter
        public static class Input {
            protected String type;
            protected String data;
        }

        @Getter
        @Setter
        public static class Extension {
            protected String style;
            protected List<Object> history;
            protected String[] existingMusicIds;
        }
    }
}