org.genx.javadoc.utils.AnnotationUtil Maven / Gradle / Ivy

Go to download
package org.genx.javadoc.utils;

import com.sun.javadoc.AnnotationDesc;
import com.sun.javadoc.AnnotationTypeElementDoc;
import com.sun.javadoc.ProgramElementDoc;
import org.genx.javadoc.vo.AnnotationDocVO;

import java.util.HashMap;
import java.util.Map;

/**
 * Created with IntelliJ IDEA.
 * Description:
 *
 * @author: genx
 * @date: 2019/3/12 10:01
 */
public class AnnotationUtil {

    public static Map readAnnotationMap(ProgramElementDoc doc) {
        return readAnnotationMap(doc.annotations());
    }

    public static Map readAnnotationMap(AnnotationDesc[] annotationDescs) {
        if (annotationDescs == null || annotationDescs.length == 0) {
            return null;
        }
        Map map = new HashMap<>(8);
        for (AnnotationDesc annotation : annotationDescs) {
            AnnotationDocVO annotationVO = new AnnotationDocVO();
            annotationVO.setName(annotation.annotationType().name());
            annotationVO.setText(annotation.toString());
            annotationVO.setClassName(annotation.annotationType().qualifiedTypeName());

            Map data = new HashMap<>(8);
            for (AnnotationDesc.ElementValuePair elementValuePair : annotation.elementValues()) {
                data.put(elementValuePair.element().name(), parseObjectToStringArray(elementValuePair.value().value()));
            }
            //设置默认值
            for (AnnotationTypeElementDoc element : annotation.annotationType().elements()) {
                if (!data.containsKey(element.name()) && element.defaultValue() != null && element.defaultValue().value() != null) {
                    data.put(element.name(), parseObjectToStringArray(element.defaultValue().value()));
                }
            }
            annotationVO.setData(data);
            map.put(annotation.annotationType().qualifiedTypeName(), annotationVO);
        }
        return map;
    }

    private static String[] parseObjectToStringArray(Object value) {
        Object[] array;
        if (value.getClass().isArray()) {
            array = (Object[]) value;
            if (array.length == 0) {
                return null;
            }
        } else {
            if (String.valueOf(value).length() == 0) {
                return null;
            }
            array = new Object[]{value};
        }
        String[] ss = new String[array.length];
        for (int i = 0; i < array.length; i++) {
            ss[i] = String.valueOf(array[i]);
            if (ss[i].startsWith("\"") && ss[i].endsWith("\"")) {
                ss[i] = ss[i].substring(1, ss[i].length() - 1);
            }
        }
        return ss;
    }

}