Files
rustfs-console/composables/useObject.ts
T

106 lines
2.8 KiB
TypeScript

import { DeleteObjectCommand, GetObjectCommand, HeadObjectCommand, ListObjectsV2Command, PutObjectCommand,GetObjectTaggingCommand, PutObjectTaggingCommand,DeleteObjectTaggingCommand} from '@aws-sdk/client-s3'
import { getSignedUrl as _getSignedUrl } from '@aws-sdk/s3-request-presigner'
export function useObject({ bucket, region }: { bucket: string, region?: string }) {
const $client = useNuxtApp().$s3Client
const headObject = async (key: string) => {
const params = {
Bucket: bucket,
Key: key
}
return await $client.send(new HeadObjectCommand(params))
}
const getSignedUrl = async (key: string, expiresIn = 3600) => {
const command = new GetObjectCommand({
Bucket: bucket,
Key: key,
})
return await _getSignedUrl($client, command, { expiresIn })
}
const putObject = async (key: string, body: Blob | string) => {
const params = {
Bucket: bucket,
Key: key,
Body: body
}
return await $client.send(new PutObjectCommand(params))
}
const deleteObject = async (key: string) => {
const params = {
Bucket: bucket,
Key: key
}
return await $client.send(new DeleteObjectCommand(params))
}
async function mapAllFiles(bucketName: string, prefix: string, callback: (fileKey: string) => void) {
let isTruncated = true;
let continuationToken: string | undefined = undefined;
while (isTruncated) {
const params: {
Bucket: string;
Prefix: string;
ContinuationToken?: string;
} = {
Bucket: bucketName,
Prefix: prefix,
ContinuationToken: continuationToken,
};
try {
const data = await $client.send(new ListObjectsV2Command(params));
data.Contents?.forEach((item) => {
if (item.Key) {
callback(item.Key);
}
});
isTruncated = data.IsTruncated || false;
continuationToken = data.NextContinuationToken;
} catch (error) {
console.error("Error listing files: ", error);
throw error;
}
}
}
async function getObjectTagging(key: string) {
const params = {
Bucket: bucket,
Key: key,
}
return await $client.send(new GetObjectTaggingCommand(params))
}
async function putObjectTagging(key: string, tagging: any) {
const params = {
Bucket: bucket,
Key: key,
Tagging: tagging
}
return await $client.send(new PutObjectTaggingCommand(params))
}
async function deleteObjectTagging(key: string) {
const params = {
Bucket: bucket,
Key: key,
}
return await $client.send(new DeleteObjectTaggingCommand(params))
}
return { headObject, putObject, deleteObject, getSignedUrl, mapAllFiles,getObjectTagging ,putObjectTagging,deleteObjectTagging}
}