cos-mcp
Version:
基于MCP协议的腾讯云COS MCP Server，无需编码即可让大模型快速接入腾讯云存储(COS)和数据万象(CI)能力
183 lines • 6.65 kB
JavaScript
// import fs from 'fs';
import { generateOutPutFileId } from '../utils.js';
export class CIAIService {
bucket;
region;
cos;
constructor(bucket, region, cos) {
this.bucket = bucket;
this.region = region;
this.cos = cos;
}
// 图片处理-图片超分-云上处理
async aiSuperResolution(objectKey) {
try {
const result = await new Promise((resolve, reject) => {
const outPutFileid = generateOutPutFileId(objectKey);
this.cos.request({
Bucket: this.bucket, // 存储桶，必须字段
Region: this.region, // 存储桶所在地域，必须字段 如 ap-beijing
Key: objectKey, // 对象文件名，例如：folder/document.jpg。
Method: 'POST', // 固定值
Action: 'image_process', // 固定值
Headers: {
'Pic-Operations': JSON.stringify({
rules: [
{
fileid: `${outPutFileid}`,
rule: 'ci-process=AISuperResolution',
},
],
}),
},
}, function (error, data) {
if (error) {
// 处理请求失败
reject(error);
}
else {
// 处理请求成功
resolve(data);
}
});
});
return {
isSuccess: true,
message: '图片超分成功',
data: result,
};
}
catch (error) {
return {
isSuccess: false,
message: '图片超分失败',
data: error,
};
}
}
// 图片处理-图片质量评分
async assessQuality(objectKey) {
try {
const result = await this.cos.request({
Bucket: this.bucket,
Region: this.region,
Method: 'GET',
Key: objectKey,
Query: {
'ci-process': 'AssessQuality',
},
});
return {
isSuccess: true,
message: '图片处理成功',
data: result,
};
}
catch (error) {
return {
isSuccess: false,
message: '图片处理失败',
data: error,
};
}
}
// 图片处理-抠图-云上处理
async aiPicMatting(objectKey, width, height) {
try {
const result = await new Promise((resolve, reject) => {
const outPutFileid = generateOutPutFileId(objectKey);
this.cos.request({
Bucket: this.bucket, // 存储桶，必须字段
Region: this.region, // 存储桶所在地域，必须字段 如 ap-beijing
Key: objectKey, // 对象文件名，例如：folder/document.jpg。
Method: 'POST', // 固定值
Action: 'image_process', // 固定值
Headers: {
'Pic-Operations': JSON.stringify({
rules: [
{
fileid: `${outPutFileid}`,
rule: 'ci-process=AIImageCrop&width=' +
width +
'&height=' +
height,
},
],
}),
},
}, function (error, data) {
if (error) {
// 处理请求失败
reject(error);
}
else {
// 处理请求成功
resolve(data);
}
});
});
// const localPath = "结果.png"; // 填写要写入的本地文件路径
// if (result.Body) {
// fs.writeFileSync(localPath, result.Body); // 将图片内容保存本地路径
// } else {
// throw new Error("Result body is undefined");
// }
// if (!result.Body) {
// throw new Error("Result body is undefined");
// }
// const base64Image = `data:image/jpeg;base64,${typeof result.Body === "string" ? Buffer.from(result.Body).toString("base64") : result.Body.toString("base64")}`;
return {
isSuccess: true,
message: '图片处理成功',
data: result,
};
}
catch (error) {
return {
isSuccess: false,
message: '图片处理失败',
data: error,
};
}
}
// 二维码识别-下载时识别
async aiQrcode(objectKey) {
try {
const result = await new Promise((resolve, reject) => {
this.cos.request({
Bucket: this.bucket, // 存储桶，必须字段
Region: this.region, // 存储桶所在地域，必须字段 如 ap-beijing
Method: 'GET',
Key: objectKey,
// Url: url,
Query: {
'ci-process': 'QRcode', // 数据万象处理能力，二维码识别固定为 QRcode,
cover: 0,
},
}, function (error, data) {
if (error) {
// 处理请求失败
reject(error);
}
else {
// 处理请求成功
resolve(data);
}
});
});
return {
isSuccess: true,
message: '二维码识别成功',
data: result,
};
}
catch (error) {
return {
isSuccess: false,
message: '二维码识别失败',
data: error,
};
}
}
}
//# sourceMappingURL=ai.service.js.map