package com.github.houbb.spring.tool.utils;

import com.github.houbb.heaven.util.lang.StringUtil;
import com.github.houbb.spring.tool.exception.SpringToolException;
import org.springframework.core.io.Resource;
import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
import org.springframework.core.io.support.ResourcePatternResolver;
import org.springframework.core.type.classreading.CachingMetadataReaderFactory;
import org.springframework.core.type.classreading.MetadataReader;
import org.springframework.core.type.classreading.MetadataReaderFactory;
import org.springframework.util.ClassUtils;

import java.util.ArrayList;
import java.util.List;
import java.util.function.Function;
import java.util.stream.Collectors;

/**
 * spring 包工具类
 *
 * @since 1.0.0
 */
public class SpringPackageUtil {

    /**
     * 扫描
     * @param basePackages 基础包信息
     * @param resourcePattern 资源正则
     */
    public static List scanClassName(String basePackages, String resourcePattern) {
        try {
            List classNameList = new ArrayList<>();

            if(StringUtil.isEmpty(basePackages)
                || StringUtil.isEmpty(resourcePattern)) {
                return classNameList;
            }

            final ResourcePatternResolver resourcePatternResolver = new PathMatchingResourcePatternResolver();

            // 根据classname生成class对应的资源路径,需要扫描的包路径
            //ResourcePatternResolver.CLASSPATH_ALL_URL_PREFIX +
            String pattern = ClassUtils.convertClassNameToResourcePath(basePackages) + resourcePattern;
            // 获取classname的IO流资源
            Resource[] resources = resourcePatternResolver.getResources(pattern);
            // MetadataReaderFactory接口 ，MetadataReader的工厂接口。允许缓存每个MetadataReader的元数据集
            MetadataReaderFactory readerFactory = new CachingMetadataReaderFactory(resourcePatternResolver);

            for (Resource resource : resources) {
                if (resource.isReadable()) {
                    // 通过class资源（resource）生成MetadataReader
                    MetadataReader metadataReader = readerFactory.getMetadataReader(resource);
                    // 获取class名
                    String className = metadataReader.getClassMetadata().getClassName();
                    classNameList.add(className);
                }
            }

            return classNameList;
        } catch (Exception e) {
            throw new SpringToolException(e);
        }
    }

    /**
     * 扫描
     * @param basePackages 基础包信息
     * @param resourcePattern 资源正则
     */
    public static List scanClass(String basePackages, String resourcePattern) {
        List nameList = scanClassName(basePackages, resourcePattern);

        return nameList.stream().map((Function) s -> {
            try {
                return Class.forName(s);
            } catch (ClassNotFoundException e) {
                throw new RuntimeException(e);
            }
        }).collect(Collectors.toList());
    }

}