package com.iailab.framework.ai.core.model.wenduoduo.api;
|
|
import com.fasterxml.jackson.annotation.JsonFormat;
|
import com.fasterxml.jackson.annotation.JsonInclude;
|
import com.iailab.framework.common.util.json.JsonUtils;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.core.ParameterizedTypeReference;
|
import org.springframework.http.HttpRequest;
|
import org.springframework.http.HttpStatusCode;
|
import org.springframework.http.MediaType;
|
import org.springframework.util.Assert;
|
import org.springframework.util.LinkedMultiValueMap;
|
import org.springframework.util.MultiValueMap;
|
import org.springframework.web.multipart.MultipartFile;
|
import org.springframework.web.reactive.function.BodyInserters;
|
import org.springframework.web.reactive.function.client.ClientResponse;
|
import org.springframework.web.reactive.function.client.WebClient;
|
import reactor.core.publisher.Flux;
|
import reactor.core.publisher.Mono;
|
|
import java.time.LocalDateTime;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.Objects;
|
import java.util.function.Function;
|
import java.util.function.Predicate;
|
|
/**
|
* 文多多 API
|
*
|
* @author xiaoxin
|
* @see <a href="https://docmee.cn/open-platform/api">PPT 生成 API</a>
|
*/
|
@Slf4j
|
public class WenDuoDuoPptApi {
|
|
public static final String BASE_URL = "https://docmee.cn";
|
public static final String TOKEN_NAME = "token";
|
|
private final WebClient webClient;
|
|
private final Predicate<HttpStatusCode> STATUS_PREDICATE = status -> !status.is2xxSuccessful();
|
|
private final Function<Object, Function<ClientResponse, Mono<? extends Throwable>>> EXCEPTION_FUNCTION =
|
reqParam -> response -> response.bodyToMono(String.class).handle((responseBody, sink) -> {
|
HttpRequest request = response.request();
|
log.error("[WenDuoDuoPptApi] 调用失败！请求方式:[{}]，请求地址:[{}]，请求参数:[{}]，响应数据: [{}]",
|
request.getMethod(), request.getURI(), reqParam, responseBody);
|
sink.error(new IllegalStateException("[WenDuoDuoPptApi] 调用失败！"));
|
});
|
|
public WenDuoDuoPptApi(String token) {
|
Assert.hasText(token, "token 不能为空");
|
this.webClient = WebClient.builder()
|
.baseUrl(BASE_URL)
|
.defaultHeaders((headers) -> {
|
headers.setContentType(MediaType.APPLICATION_JSON);
|
headers.add(TOKEN_NAME, token);
|
})
|
.build();
|
}
|
|
/**
|
* 创建 token
|
*
|
* @param request 请求信息
|
* @return token
|
*/
|
public String createApiToken(CreateTokenRequest request) {
|
return this.webClient.post()
|
.uri("/api/user/createApiToken")
|
.header("Api-Key", request.apiKey)
|
.body(Mono.just(request), CreateTokenRequest.class)
|
.retrieve()
|
.onStatus(STATUS_PREDICATE, EXCEPTION_FUNCTION.apply(request))
|
.bodyToMono(ApiResponse.class)
|
.<String>handle((response, sink) -> {
|
if (response.code != 0) {
|
sink.error(new IllegalStateException("创建 token 异常，" + response.message));
|
return;
|
}
|
sink.next(response.data.get("token").toString());
|
})
|
.block();
|
}
|
|
/**
|
* 创建任务
|
*
|
* @param type 类型
|
* @param content 内容
|
* @param files 文件列表
|
* @return 任务 ID
|
* @see <a href="https://docmee.cn/open-platform/api#%E5%88%9B%E5%BB%BA%E4%BB%BB%E5%8A%A1">创建任务</a>
|
*/
|
public ApiResponse createTask(Integer type, String content, List<MultipartFile> files) {
|
MultiValueMap<String, Object> formData = new LinkedMultiValueMap<>();
|
formData.add("type", type);
|
if (content != null) {
|
formData.add("content", content);
|
}
|
if (files != null) {
|
for (MultipartFile file : files) {
|
formData.add("file", file.getResource());
|
}
|
}
|
return this.webClient.post()
|
.uri("/api/ppt/v2/createTask")
|
.contentType(MediaType.MULTIPART_FORM_DATA)
|
.body(BodyInserters.fromMultipartData(formData))
|
.retrieve()
|
.onStatus(STATUS_PREDICATE, EXCEPTION_FUNCTION.apply(formData))
|
.bodyToMono(ApiResponse.class)
|
.block();
|
}
|
|
/**
|
* 获取生成选项
|
*
|
* @param lang 语种
|
* @return 生成选项
|
*/
|
public Map<String, Object> getOptions(String lang) {
|
String uri = "/api/ppt/v2/options";
|
if (lang != null) {
|
uri += "?lang=" + lang;
|
}
|
return this.webClient.get()
|
.uri(uri)
|
.retrieve()
|
.onStatus(STATUS_PREDICATE, EXCEPTION_FUNCTION.apply(lang))
|
.bodyToMono(new ParameterizedTypeReference<ApiResponse>() {
|
})
|
.<Map<String, Object>>handle((response, sink) -> {
|
if (response.code != 0) {
|
sink.error(new IllegalStateException("获取生成选项异常，" + response.message));
|
return;
|
}
|
sink.next(response.data);
|
})
|
.block();
|
}
|
|
/**
|
* 分页查询 PPT 模板
|
*
|
* @param token 令牌
|
* @param request 请求体
|
* @return 模板列表
|
*/
|
public PagePptTemplateInfo getTemplatePage(TemplateQueryRequest request) {
|
return this.webClient.post()
|
.uri("/api/ppt/templates")
|
.bodyValue(request)
|
.retrieve()
|
.onStatus(STATUS_PREDICATE, EXCEPTION_FUNCTION.apply(request))
|
.bodyToMono(new ParameterizedTypeReference<PagePptTemplateInfo>() {
|
})
|
.block();
|
}
|
|
/**
|
* 生成大纲内容
|
*
|
* @return 大纲内容流
|
*/
|
public Flux<Map<String, Object>> createOutline(CreateOutlineRequest request) {
|
return this.webClient.post()
|
.uri("/api/ppt/v2/generateContent")
|
.body(Mono.just(request), CreateOutlineRequest.class)
|
.retrieve()
|
.onStatus(STATUS_PREDICATE, EXCEPTION_FUNCTION.apply(request))
|
.bodyToFlux(new ParameterizedTypeReference<>() {
|
});
|
}
|
|
/**
|
* 修改大纲内容
|
*
|
* @param request 请求体
|
* @return 大纲内容流
|
*/
|
public Flux<Map<String, Object>> updateOutline(UpdateOutlineRequest request) {
|
return this.webClient.post()
|
.uri("/api/ppt/v2/updateContent")
|
.body(Mono.just(request), UpdateOutlineRequest.class)
|
.retrieve()
|
.onStatus(STATUS_PREDICATE, EXCEPTION_FUNCTION.apply(request))
|
.bodyToFlux(new ParameterizedTypeReference<>() {
|
});
|
}
|
|
/**
|
* 生成 PPT
|
*
|
* @return PPT信息
|
*/
|
public PptInfo create(PptCreateRequest request) {
|
return this.webClient.post()
|
.uri("/api/ppt/v2/generatePptx")
|
.body(Mono.just(request), PptCreateRequest.class)
|
.retrieve()
|
.onStatus(STATUS_PREDICATE, EXCEPTION_FUNCTION.apply(request))
|
.bodyToMono(ApiResponse.class)
|
.<PptInfo>handle((response, sink) -> {
|
if (response.code != 0) {
|
sink.error(new IllegalStateException("生成 PPT 异常，" + response.message));
|
return;
|
}
|
sink.next(Objects.requireNonNull(JsonUtils.parseObject(JsonUtils.toJsonString(response.data.get("pptInfo")), PptInfo.class)));
|
})
|
.block();
|
}
|
|
/**
|
* 创建 Token 请求参数
|
*/
|
@JsonInclude(value = JsonInclude.Include.NON_NULL)
|
public record CreateTokenRequest(
|
String apiKey,
|
String uid,
|
Integer limit
|
) {
|
|
public CreateTokenRequest(String apiKey) {
|
this(apiKey, null, null);
|
}
|
|
}
|
|
/**
|
* API 通用响应
|
*/
|
@JsonInclude(value = JsonInclude.Include.NON_NULL)
|
public record ApiResponse(
|
Integer code,
|
String message,
|
Map<String, Object> data
|
) {
|
}
|
|
/**
|
* 创建任务
|
*/
|
@JsonInclude(value = JsonInclude.Include.NON_NULL)
|
public record CreateTaskRequest(
|
Integer type,
|
String content,
|
List<MultipartFile> files
|
) {
|
}
|
|
/**
|
* 生成大纲内容请求
|
*/
|
@JsonInclude(value = JsonInclude.Include.NON_NULL)
|
public record CreateOutlineRequest(
|
String id,
|
String length,
|
String scene,
|
String audience,
|
String lang,
|
String prompt
|
) {
|
}
|
|
/**
|
* 修改大纲内容请求
|
*/
|
@JsonInclude(value = JsonInclude.Include.NON_NULL)
|
public record UpdateOutlineRequest(
|
String id,
|
String markdown,
|
String question
|
) {
|
}
|
|
/**
|
* 生成 PPT 请求参数
|
*/
|
@JsonInclude(value = JsonInclude.Include.NON_NULL)
|
public record PptCreateRequest(
|
String id,
|
String templateId,
|
String markdown
|
) {
|
}
|
|
/**
|
* PPT 信息
|
*/
|
@JsonInclude(value = JsonInclude.Include.NON_NULL)
|
public record PptInfo(
|
String id,
|
String name,
|
String subject,
|
String coverUrl,
|
String fileUrl,
|
String templateId,
|
String pptxProperty,
|
String userId,
|
String userName,
|
int companyId,
|
@JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss")
|
LocalDateTime updateTime,
|
@JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss")
|
LocalDateTime createTime,
|
String createUser,
|
String updateUser
|
) {
|
}
|
|
/**
|
* 模板查询请求参数
|
*/
|
@JsonInclude(value = JsonInclude.Include.NON_NULL)
|
public record TemplateQueryRequest(
|
int page,
|
int size,
|
Filter filters
|
) {
|
|
/**
|
* 模板查询过滤条件
|
*/
|
@JsonInclude(value = JsonInclude.Include.NON_NULL)
|
public record Filter(
|
int type,
|
String category,
|
String style,
|
String themeColor
|
) {
|
}
|
|
}
|
|
/**
|
* PPT模板分页信息
|
*/
|
@JsonInclude(value = JsonInclude.Include.NON_NULL)
|
public record PagePptTemplateInfo(
|
List<PptTemplateInfo> data,
|
String total
|
) {
|
}
|
|
/**
|
* PPT模板信息
|
*/
|
@JsonInclude(value = JsonInclude.Include.NON_NULL)
|
public record PptTemplateInfo(
|
String id,
|
int type,
|
Integer subType,
|
String layout,
|
String category,
|
String style,
|
String themeColor,
|
String lang,
|
boolean animation,
|
String subject,
|
String coverUrl,
|
String fileUrl,
|
List<String> pageCoverUrls,
|
String pptxProperty,
|
int sort,
|
int num,
|
Integer imgNum,
|
int isDeleted,
|
String userId,
|
int companyId,
|
@JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss")
|
LocalDateTime updateTime,
|
@JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss")
|
LocalDateTime createTime,
|
String createUser,
|
String updateUser
|
) {
|
}
|
|
}
|
