-
Notifications
You must be signed in to change notification settings - Fork 23
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat: ability to configure and use image thumbnails
- Loading branch information
Showing
8 changed files
with
228 additions
and
51 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,128 @@ | ||
| package run.halo.s3os; | ||
|
|
||
| import static run.halo.s3os.S3OsPlugin.POLICY_SETTING_NAME; | ||
|
|
||
| import java.net.URI; | ||
| import java.net.URL; | ||
| import java.util.List; | ||
| import java.util.Map; | ||
| import com.google.common.cache.Cache; | ||
| import com.google.common.cache.CacheBuilder; | ||
| import lombok.Builder; | ||
| import lombok.RequiredArgsConstructor; | ||
| import org.apache.commons.lang3.StringUtils; | ||
| import org.springframework.data.domain.Sort; | ||
| import org.springframework.stereotype.Component; | ||
| import org.springframework.util.Assert; | ||
| import org.springframework.web.util.UriComponentsBuilder; | ||
| import reactor.core.publisher.Flux; | ||
| import reactor.core.publisher.Mono; | ||
| import run.halo.app.core.attachment.ThumbnailProvider; | ||
| import run.halo.app.core.attachment.ThumbnailSize; | ||
| import run.halo.app.core.extension.attachment.Policy; | ||
| import run.halo.app.core.extension.attachment.PolicyTemplate; | ||
| import run.halo.app.extension.ConfigMap; | ||
| import run.halo.app.extension.ListOptions; | ||
| import run.halo.app.extension.ReactiveExtensionClient; | ||
|
|
||
| @Component | ||
| @RequiredArgsConstructor | ||
| public class S3ThumbnailProvider implements ThumbnailProvider { | ||
| static final String WIDTH_PLACEHOLDER = "{width}"; | ||
| private final Cache<String, S3PropsCacheValue> s3PropsCache = CacheBuilder.newBuilder() | ||
| .maximumSize(50) | ||
| .build(); | ||
|
|
||
| private final ReactiveExtensionClient client; | ||
|
|
||
| @Override | ||
| public Mono<URI> generate(ThumbnailContext thumbnailContext) { | ||
| var url = thumbnailContext.getImageUrl().toString(); | ||
| var size = thumbnailContext.getSize(); | ||
| return getCacheValue(url) | ||
| .mapNotNull(cacheValue -> placedPattern(cacheValue.pattern(), size)) | ||
| .map(param -> { | ||
| if (param.startsWith("?")) { | ||
| UriComponentsBuilder.fromHttpUrl(url) | ||
| .queryParam(param.substring(1)); | ||
| } | ||
| return url + param; | ||
| }) | ||
| .map(URI::create); | ||
| } | ||
|
|
||
| private static String placedPattern(String pattern, ThumbnailSize size) { | ||
| return StringUtils.replace(pattern, WIDTH_PLACEHOLDER, String.valueOf(size.getWidth())); | ||
| } | ||
|
|
||
| @Override | ||
| public Mono<Void> delete(URL url) { | ||
| // do nothing for s3 | ||
| return Mono.empty(); | ||
| } | ||
|
|
||
| @Override | ||
| public Mono<Boolean> supports(ThumbnailContext thumbnailContext) { | ||
| var url = thumbnailContext.getImageUrl().toString(); | ||
| return getCacheValue(url).hasElement(); | ||
| } | ||
|
|
||
| private Mono<S3PropsCacheValue> getCacheValue(String imageUrl) { | ||
| return Flux.fromIterable(s3PropsCache.asMap().entrySet()) | ||
| .filter(entry -> imageUrl.startsWith(entry.getKey())) | ||
| .next() | ||
| .map(Map.Entry::getValue) | ||
| .switchIfEmpty(Mono.defer(() -> listAllS3ObjectDomain() | ||
| .filter(entry -> imageUrl.startsWith(entry.getKey())) | ||
| .map(Map.Entry::getValue) | ||
| .next() | ||
| )); | ||
| } | ||
|
|
||
| @Builder | ||
| record S3PropsCacheValue(String pattern, String configMapName) { | ||
| } | ||
|
|
||
| private Flux<Map.Entry<String, S3PropsCacheValue>> listAllS3ObjectDomain() { | ||
| return listS3PolicyTemplateNames() | ||
| .collectList() | ||
| .flatMapMany(this::listAllS3Policy) | ||
| .flatMap(s3Policy -> { | ||
| var s3ConfigMapName = s3Policy.getSpec().getConfigMapName(); | ||
| return fetchS3PropsByConfigMapName(s3ConfigMapName) | ||
| .mapNotNull(properties -> { | ||
| var thumbnailParam = properties.getThumbnailParam(); | ||
| if (thumbnailParam == null || !thumbnailParam.hasPattern()) { | ||
| return null; | ||
| } | ||
| var objectDomain = properties.toObjectURL(""); | ||
| var cacheValue = S3PropsCacheValue.builder() | ||
| .pattern(thumbnailParam.pattern()) | ||
| .configMapName(s3ConfigMapName) | ||
| .build(); | ||
| return Map.entry(objectDomain, cacheValue); | ||
| }); | ||
| }) | ||
| .doOnNext(cache -> s3PropsCache.put(cache.getKey(), cache.getValue())); | ||
| } | ||
|
|
||
| private Flux<Policy> listAllS3Policy(List<String> s3PolicyTemplateNames) { | ||
| Assert.notNull(s3PolicyTemplateNames, "The s3PolicyTemplateNames must not be null."); | ||
| return client.listAll(Policy.class, new ListOptions(), Sort.unsorted()) | ||
| .filter(policy -> { | ||
| var templateName = policy.getSpec().getTemplateName(); | ||
| return s3PolicyTemplateNames.contains(templateName); | ||
| }); | ||
| } | ||
|
|
||
| private Mono<S3OsProperties> fetchS3PropsByConfigMapName(String name) { | ||
| return client.fetch(ConfigMap.class, name) | ||
| .map(S3OsProperties::convertFrom); | ||
| } | ||
|
|
||
| private Flux<String> listS3PolicyTemplateNames() { | ||
| return client.listAll(PolicyTemplate.class, new ListOptions(), Sort.unsorted()) | ||
| .filter(policyTemplate -> POLICY_SETTING_NAME.equals(policyTemplate.getSpec().getSettingName())) | ||
| .map(template -> template.getMetadata().getName()); | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,9 @@ | ||
| apiVersion: plugin.halo.run/v1alpha1 | ||
| kind: ExtensionDefinition | ||
| metadata: | ||
| name: s3os-thumbnail-provider | ||
| spec: | ||
| className: run.halo.s3os.S3ThumbnailProvider | ||
| extensionPointName: thumbnail-provider | ||
| displayName: "S3 协议 OSS 缩略图生成" | ||
| description: "为上传到支持 S3 协议的 OSS 的图片生成缩略图" |
Oops, something went wrong.