249 lines
9.0 KiB
Java
249 lines
9.0 KiB
Java
package com.dsideal.Ai.Util.Liblib.Kit;
|
||
|
||
import java.security.InvalidKeyException;
|
||
import java.security.NoSuchAlgorithmException;
|
||
import java.util.ArrayList;
|
||
import java.util.List;
|
||
import java.util.concurrent.TimeUnit;
|
||
|
||
import javax.crypto.Mac;
|
||
import javax.crypto.spec.SecretKeySpec;
|
||
|
||
import com.alibaba.fastjson.JSON;
|
||
import com.alibaba.fastjson.JSONObject;
|
||
import lombok.Getter;
|
||
import okhttp3.*;
|
||
import org.apache.commons.codec.binary.Base64;
|
||
import org.apache.commons.lang3.RandomStringUtils;
|
||
import org.slf4j.Logger;
|
||
import org.slf4j.LoggerFactory;
|
||
|
||
import java.io.IOException;
|
||
|
||
public class LibLibCommon {
|
||
// API基础URL
|
||
protected static final String API_BASE_URL = "https://openapi.liblibai.cloud";
|
||
// API访问凭证
|
||
protected static final String accessKey="sOCtVLVTNOZkRMajlhzCmg"; // Access Key
|
||
protected static final String secretKey="PUe8QTRG9i0G9EbpedHmIpLQ0FyxoYY9"; // Secret Key
|
||
// 查询任务状态API路径
|
||
protected static final String QUERY_STATUS_PATH = "/api/generate/webui/status";
|
||
// 日志
|
||
private static final Logger log = LoggerFactory.getLogger(LibLibCommon.class);
|
||
|
||
/**
|
||
* 签名信息类，包含签名、时间戳和随机字符串
|
||
*/
|
||
@Getter
|
||
public static class SignatureInfo {
|
||
private final String signature;
|
||
private final long timestamp;
|
||
private final String signatureNonce;
|
||
|
||
public SignatureInfo(String signature, long timestamp, String signatureNonce) {
|
||
this.signature = signature;
|
||
this.timestamp = timestamp;
|
||
this.signatureNonce = signatureNonce;
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 创建OkHttpClient实例
|
||
* @return OkHttpClient实例
|
||
*/
|
||
protected static OkHttpClient createHttpClient() {
|
||
return new OkHttpClient.Builder()
|
||
.connectTimeout(30, TimeUnit.SECONDS)
|
||
.readTimeout(60, TimeUnit.SECONDS)
|
||
.writeTimeout(60, TimeUnit.SECONDS)
|
||
.build();
|
||
}
|
||
|
||
/**
|
||
* 生成完整的签名信息，包括签名、时间戳和随机字符串
|
||
* @param uri API接口的uri地址
|
||
* @return 签名信息对象
|
||
*/
|
||
public static SignatureInfo sign(String uri) {
|
||
// 当前毫秒时间戳
|
||
long timestamp = System.currentTimeMillis();
|
||
// 随机字符串
|
||
String signatureNonce = RandomStringUtils.randomAlphanumeric(10);
|
||
// 拼接请求数据
|
||
String content = uri + "&" + timestamp + "&" + signatureNonce;
|
||
try {
|
||
// 生成签名
|
||
SecretKeySpec secret = new SecretKeySpec(secretKey.getBytes(), "HmacSHA1");
|
||
Mac mac = Mac.getInstance("HmacSHA1");
|
||
mac.init(secret);
|
||
String signature = Base64.encodeBase64URLSafeString(mac.doFinal(content.getBytes()));
|
||
return new SignatureInfo(signature, timestamp, signatureNonce);
|
||
} catch (NoSuchAlgorithmException e) {
|
||
throw new RuntimeException("no such algorithm");
|
||
} catch (InvalidKeyException e) {
|
||
throw new RuntimeException(e);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 清理URL，移除可能的反引号和多余空格
|
||
* @param url 原始URL
|
||
* @return 清理后的URL
|
||
*/
|
||
protected static String cleanUrl(String url) {
|
||
if (url == null || url.isEmpty()) {
|
||
return url;
|
||
}
|
||
return url.trim().replace("`", "");
|
||
}
|
||
|
||
/**
|
||
* 构建带签名的URL
|
||
* @param uri API路径
|
||
* @param signInfo 签名信息
|
||
* @return 构建好的HttpUrl.Builder
|
||
*/
|
||
protected static HttpUrl.Builder buildSignedUrl(String uri, SignatureInfo signInfo) {
|
||
return HttpUrl.parse(API_BASE_URL + uri).newBuilder()
|
||
.addQueryParameter("AccessKey", accessKey)
|
||
.addQueryParameter("Signature", signInfo.getSignature())
|
||
.addQueryParameter("Timestamp", String.valueOf(signInfo.getTimestamp()))
|
||
.addQueryParameter("SignatureNonce", signInfo.getSignatureNonce());
|
||
}
|
||
|
||
/**
|
||
* 发送HTTP请求并处理响应
|
||
* @param request HTTP请求
|
||
* @param logPrefix 日志前缀
|
||
* @return 响应的JSON对象
|
||
* @throws IOException 异常信息
|
||
*/
|
||
protected static JSONObject sendRequest(Request request, String logPrefix) throws IOException {
|
||
OkHttpClient client = createHttpClient();
|
||
|
||
// 执行请求
|
||
log.info("{}: {}", logPrefix, request.url());
|
||
Response response = client.newCall(request).execute();
|
||
|
||
// 处理响应
|
||
if (!response.isSuccessful()) {
|
||
String errorMsg = logPrefix + "失败，状态码: " + response.code();
|
||
log.error(errorMsg);
|
||
throw new IOException(errorMsg);
|
||
}
|
||
|
||
// 解析响应
|
||
String responseBody = response.body().string();
|
||
log.info("{}响应: {}", logPrefix, responseBody);
|
||
|
||
JSONObject responseJson = JSON.parseObject(responseBody);
|
||
int code = responseJson.getIntValue("code");
|
||
|
||
if (code != 0) {
|
||
String errorMsg = logPrefix + "失败，错误码: " + code + ", 错误信息: " + responseJson.getString("msg");
|
||
log.error(errorMsg);
|
||
throw new IOException(errorMsg);
|
||
}
|
||
|
||
return responseJson;
|
||
}
|
||
|
||
/**
|
||
* 查询生图任务结果
|
||
*
|
||
* @param generateUuid 生图任务UUID
|
||
* @return 任务结果信息
|
||
* @throws IOException 异常信息
|
||
*/
|
||
public static JSONObject queryTaskResult(String generateUuid) throws IOException {
|
||
// 构建请求体
|
||
JSONObject requestBody = new JSONObject();
|
||
requestBody.put("generateUuid", generateUuid);
|
||
|
||
// 获取API路径
|
||
String uri = QUERY_STATUS_PATH;
|
||
|
||
// 生成签名信息
|
||
SignatureInfo signInfo = sign(uri);
|
||
|
||
// 构建带签名的URL
|
||
HttpUrl.Builder urlBuilder = buildSignedUrl(uri, signInfo);
|
||
|
||
// 创建请求
|
||
MediaType mediaType = MediaType.parse("application/json");
|
||
RequestBody body = RequestBody.create(mediaType, requestBody.toJSONString());
|
||
Request request = new Request.Builder()
|
||
.url(urlBuilder.build())
|
||
.method("POST", body)
|
||
.addHeader("Content-Type", "application/json")
|
||
.build();
|
||
|
||
// 发送请求并获取响应
|
||
JSONObject responseJson = sendRequest(request, "查询生图任务结果");
|
||
|
||
return responseJson.getJSONObject("data");
|
||
}
|
||
|
||
/**
|
||
* 获取生成图片的URL列表
|
||
*
|
||
* @param generateUuid 生图任务UUID
|
||
* @return 图片URL列表
|
||
* @throws IOException 异常信息
|
||
*/
|
||
public static List<String> getGeneratedImageUrls(String generateUuid) throws IOException {
|
||
JSONObject resultData = queryTaskResult(generateUuid);
|
||
List<String> imageUrls = new ArrayList<>();
|
||
|
||
// 检查生成状态
|
||
int generateStatus = resultData.getIntValue("generateStatus");
|
||
if (generateStatus == 5) { // 5表示生成成功
|
||
if (resultData.containsKey("images")) {
|
||
for (Object imageObj : resultData.getJSONArray("images")) {
|
||
JSONObject imageJson = (JSONObject) imageObj;
|
||
String imageUrl = imageJson.getString("imageUrl");
|
||
if (imageUrl != null && !imageUrl.isEmpty()) {
|
||
// 清理URL
|
||
imageUrl = cleanUrl(imageUrl);
|
||
imageUrls.add(imageUrl);
|
||
}
|
||
}
|
||
}
|
||
} else {
|
||
log.info("生图任务尚未完成，当前状态: {}, 完成百分比: {}%",
|
||
generateStatus, resultData.getIntValue("percentCompleted"));
|
||
}
|
||
|
||
return imageUrls;
|
||
}
|
||
|
||
/**
|
||
* 创建通用的HTTP请求
|
||
* @param uri API路径
|
||
* @param requestBody 请求体JSON对象
|
||
* @return 构建好的Request对象
|
||
*/
|
||
protected static Request createRequest(String uri, JSONObject requestBody) {
|
||
// 生成签名信息
|
||
SignatureInfo signInfo = sign(uri);
|
||
|
||
// 构建带签名的URL
|
||
HttpUrl.Builder urlBuilder = buildSignedUrl(uri, signInfo);
|
||
|
||
// 创建请求
|
||
MediaType mediaType = MediaType.parse("application/json");
|
||
RequestBody body = RequestBody.create(mediaType, requestBody.toJSONString());
|
||
return new Request.Builder()
|
||
.url(urlBuilder.build())
|
||
.method("POST", body)
|
||
.addHeader("Content-Type", "application/json")
|
||
.build();
|
||
}
|
||
|
||
public static void main(String[] args) {
|
||
String uri = "https://openapi.liblibai.cloud/api/model/version/get";
|
||
SignatureInfo signInfo = sign(uri);
|
||
|
||
System.out.println(signInfo.toString());
|
||
}
|
||
} |