/**
 * classpath下的文件。
 */
@Download(source = "classpath:/download/README.txt")
@GetMapping("/classpath")
public void classpath() {
}

/**
 * 指定路径的文件。可以指定文件夹，会自动压缩整个文件夹。
 */
@Download(source = "file:/Users/Shared/README.txt")
@GetMapping("/file")
public void file() {
}

/**
 * 指定下载地址。可以通过配置缓存下载文件，后面不需要重复下载。
 */
@Download(source = "http://127.0.0.1:8080/concept-download/image.jpg")
@GetMapping("/http")
public void http() {
}

/**
 * 返回指定路径的文件。可以指定文件夹，会自动压缩整个文件夹。
 */
@Download
@GetMapping("/file")
public File file() {
    return new File("/Users/Shared/README.txt");
}

/**
 * 支持组合不同类型的文件，会自动压缩所有文件。
 */
@Download(filename = "List.zip")
@GetMapping("/list")
public List<Object> list() {
    List<Object> list = new ArrayList<>();
    list.add(new File("/Users/Shared/README.txt"));
    list.add(new ClassPathResource("/download/README.txt"));
    list.add("http://127.0.0.1:8080/concept-download/image.jpg");
    return list;
}

@Configuration
public class ConceptDownloadConfig {

    @Bean
    public CoroutinesSourceLoader coroutinesSourceLoader() {
        return new CoroutinesSourceLoader();
    }
}

@Download
@GetMapping("/rewrite")
public DownloadOptions.Configurer rewrite() {
    return new DownloadOptions.Configurer() {
        @Override
        public configure(ConfigurableDownloadOptions options) {
            options.setSource(/*被下载的对象*/);
            options.setEventListener(new DownloadEventListener() {
                @Override
                public void onEvent(Object event) {
                    //这个事件监听器并只对该方法生效
                }
            });
        }
    };
}

@Download
@GetMapping("/async")
public DownloadOptions.Configurer async() {
    return new DownloadOptions.Configurer() {
        @Override
        public configure(ConfigurableDownloadOptions options) {
            options.setSource(/*被下载的对象*/);
            options.setAsyncConsumer(new InputStreamConsumer() {
                @Override
                public void consumer(InputStream is) {
                    //获得压缩后的输入流
                }
            });
            options.setAsyncConsumer(new FileConsumer() {
                @Override
                public void consumer(File file) {
                    //获得压缩后的文件，需要启用压缩缓存或源文件缓存才能拿到文件
                }
            });
        }
    };
}

/**
 * 下载处理器。
 */
public interface DownloadHandler {

    /**
     * 执行处理。
     *
     * @param context {@link DownloadContext}
     * @param chain   {@link DownloadHandlerChain}
     */
    Object handle(DownloadContext context, DownloadHandlerChain chain);
}

public class CustomLifecycleListener implements DownloadLifecycleListener {

    /**
     * 下载开始。
     */
    @Override
    public void onStart(DownloadContext context) {
    }

    /**
     * 下载完成。
     */
    @Override
    public void onComplete(DownloadContext context) {
    }
}

/**
 * {@link DownloadContext} 工厂。
 */
public interface DownloadContextFactory {

    /**
     * 创建一个 {@link DownloadContext}。
     *
     * @param options {@link DownloadOptions}
     * @return {@link DownloadContext}
     */
    DownloadContext create(DownloadOptions options);
}

@Download(filename = "压缩包.zip")
@GetMapping("/list")
public List<Object> list() {
    List<Object> list = new ArrayList<>();
    list.add(new File("/Users/Shared/README.txt"));
    list.add(new ClassPathResource("/download/image.jpg"));
    list.add("http://127.0.0.1:8080/concept-download/video.mp4");
    return list;
}

@Data
@SourceModel
@AllArgsConstructor
public static class BusinessModel {

    @SourceName
    private String name;

    @SourceObject
    private String url;
}

@Download
@GetMapping("/business-model")
public List<BusinessModel> businessModel() {
    List<BusinessModel> businessModels = new ArrayList<>();
    businessModels.add(new BusinessModel("BusinessModel.txt", "http://127.0.0.1:8080/concept-download/text.txt"));
    businessModels.add(new BusinessModel("BusinessModel.jpg", "http://127.0.0.1:8080/concept-download/image.jpg"));
    businessModels.add(new BusinessModel("BusinessModel.mp4", "http://127.0.0.1:8080/concept-download/video.mp4"));
    return businessModels;
}

/**
 * 将 {@link String} 转为 {@link Charset} 的 {@link ValueConvertor}。
 */
public class StringToCharsetValueConvertor implements ValueConvertor<String, Charset> {

    @Override
    public Charset convert(String value) {
        return Charset.forName(value);
    }
}

/**
 * 值转换器。
 *
 * @param <Original> 原始类型
 * @param <Target>   目标类型
 */
public interface ValueConvertor<Original, Target> {

    /**
     * 转换。
     *
     * @param value 原始值
     * @return 目标值
     */
    Target convert(Original value);
}

ValueConversion.getInstance().register(ValueConvertor);

/**
 * {@link Source} 工厂。
 */
public interface SourceFactory {

    /**
     * 是否支持需要下载的原始数据对象。
     *
     * @param source  需要下载的原始数据对象
     * @param context {@link DownloadContext}
     * @return 如果支持则返回 true
     */
    boolean support(Object source, DownloadContext context);

    /**
     * 创建。
     *
     * @param source  需要下载的原始数据对象
     * @param context {@link DownloadContext}
     * @return 创建的 {@link Source}
     */
    Source create(Object source, DownloadContext context);
}

/**
 * {@link Source} 加载器。
 */
public interface SourceLoader {

    /**
     * 执行加载。
     *
     * @param source  {@link Source}
     * @param context {@link DownloadContext}
     * @return 加载后的 {@link Source}
     */
    void load(Source source, DownloadContext context);
}

@Download(filename = "压缩包.zip")
@SourceCache(group = "source")
@GetMapping("/source-cache")
public String[] sourceCache() {
    return new String[]{
          "http://127.0.0.1:8080/concept-download/text.txt",
          "http://127.0.0.1:8080/concept-download/image.jpg",
          "http://127.0.0.1:8080/concept-download/video.mp4"
    };
}

/**
 * {@link Source} 压缩器。
 *
 * @see ZipSourceCompressor
 */
public interface SourceCompressor extends OrderProvider {

    /**
     * 获得压缩格式。
     *
     * @return 压缩格式
     */
    String[] getFormats();

    /**
     * 判断是否支持对应的压缩格式。
     *
     * @param format  压缩格式
     * @param context {@link DownloadContext}
     * @return 如果支持则返回 true
     */
    boolean support(String format, DownloadContext context);

    /**
     * 如果支持对应的格式就会调用该方法执行压缩。
     *
     * @param source  {@link Source}
     * @param writer  {@link DownloadWriter}
     * @param context {@link DownloadContext}
     * @return {@link Compression}
     */
    Compression compress(Source source, String format, DownloadWriter writer, DownloadContext context);
}

@Download(filename = "压缩包.zip")
@CompressCache(group = "compress", delete = true)
@GetMapping("/compress-cache")
public String[] compressCache() {
    return new String[]{
          "http://127.0.0.1:8080/concept-download/text.txt",
          "http://127.0.0.1:8080/concept-download/image.jpg",
          "http://127.0.0.1:8080/concept-download/video.mp4"
    };
}

/**
 * 具体操作 {@link InputStream} 和 {@link OutputStream} 的写入器。
 */
public interface DownloadWriter extends OrderProvider {

    /**
     * 该写入器是否支持写入。
     *
     * @param resource {@link Resource}
     * @param range    {@link Range}
     * @param context  {@link DownloadContext}
     * @return 如果支持则返回 true
     */
    boolean support(Resource resource, Range range, DownloadContext context);

    /**
     * 执行写入。
     *
     * @param is      {@link InputStream}
     * @param os      {@link OutputStream}
     * @param range   {@link Range}
     * @param charset {@link Charset}
     * @param length  总大小，可能为 null
     */
    default void write(InputStream is, OutputStream os, Range range, Charset charset, Long length) {
        write(is, os, range, charset, length, null);
    }

    /**
     * 执行写入。
     *
     * @param is       {@link InputStream}
     * @param os       {@link OutputStream}
     * @param range    {@link Range}
     * @param charset  {@link Charset}
     * @param length   总大小，可能为 null
     * @param callback 回调当前进度和增长的大小
     */
    void write(InputStream is, OutputStream os, Range range, Charset charset, Long length, Callback callback);

    /**
     * 进度回调。
     */
    interface Callback {

        /**
         * 回调进度。
         *
         * @param current  当前值
         * @param increase 增长值
         */
        void onWrite(long current, long increase);
    }
}

/**
 * {@link DownloadEvent} 发布器。
 *
 * @see SimpleDownloadEventPublisher
 * @see ApplicationDownloadEventPublisher
 */
public interface DownloadEventPublisher {

    /**
     * 发布事件。
     *
     * @param event 事件
     */
    void publish(Object event);
}

public interface DownloadLogger {

    void info(String message);

    void error(String message, Throwable e);
}

@Bean
public DownloadExecutor downloadExecutor() {
    return new SimpleDownloadExecutor(Executors.newCachedThreadPool());
}

@Service
public class DownloadMockService {

    @Autowired
    private DownloadMock downloadMock;

    @SneakyThrows
    public void mock() {
        //指定输出文件
        File out = new File("/Users/concept/download/mock.zip");
        //模拟下载流程
        //找到注解了@Download的file方法生成下载配置并输出到指定的OutputStream
        downloadMock.download(this, Files.newOutputStream(out.toPath()));
    }

    /**
     * 在模拟下载的方法上添加@Download。
     */
    @Download
    public File file() {
        return new File("/Users/download");
    }
}

@Service
public class DownloadMockService {

    @Autowired
    private DownloadMock downloadMock;

    @SneakyThrows
    public void mock() {
        File out = new File("/Users/tanghanzheng/concept/download/mock.zip");
        //获取返回值，并转成 Mono
        Mono<?> download = (Mono<?>)downloadMock.download(this, Files.newOutputStream(out.toPath()));
        //触发下载
        download.subscribe();
    }

    @Download
    public File file() {
        return new File("/Users/tanghanzheng/IdeaProjects/Github/x/concept/sample/src/main/resources/download");
    }
}