76 lines
2.0 KiB
TypeScript
76 lines
2.0 KiB
TypeScript
|
import { Client } from "minio";
|
||
|
|
import { z } from "zod";
|
||
|
|
|
||
|
|
if (!process.env.MINIO_ACCESS_KEY || !process.env.MINIO_SECRET_KEY) {
|
||
|
|
throw new Error(
|
||
|
|
"MINIO_ACCESS_KEY or MINIO_SECRET_KEY not found in environment variables"
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
// Initialize MinIO client
|
||
|
|
const minioClient = new Client({
|
||
|
|
endPoint: "s3.raj.how",
|
||
|
|
port: 443,
|
||
|
|
useSSL: true,
|
||
|
|
accessKey: process.env.MINIO_ACCESS_KEY,
|
||
|
|
secretKey: process.env.MINIO_SECRET_KEY,
|
||
|
|
});
|
||
|
|
|
||
|
|
// Define schema for uploading file
|
||
|
|
export const UploadFileParams = z.object({
|
||
|
|
bucketName: z.string().default("public").optional(),
|
||
|
|
fileName: z.string().describe("make sure this is unique"),
|
||
|
|
filePath: z
|
||
|
|
.string()
|
||
|
|
.describe(
|
||
|
|
"put all files inside 'anya' directory by default unless user specifies otherwise"
|
||
|
|
),
|
||
|
|
});
|
||
|
|
export type UploadFileParams = z.infer<typeof UploadFileParams>;
|
||
|
|
|
||
|
|
// Define schema for getting file list
|
||
|
|
export const GetFileListParams = z.object({
|
||
|
|
bucketName: z.string().default("public").optional(),
|
||
|
|
});
|
||
|
|
export type GetFileListParams = z.infer<typeof GetFileListParams>;
|
||
|
|
|
||
|
|
// Upload file to MinIO bucket and return public URL
|
||
|
|
export async function upload_file({
|
||
|
|
bucketName = "public",
|
||
|
|
fileName,
|
||
|
|
filePath,
|
||
|
|
}: UploadFileParams) {
|
||
|
|
try {
|
||
|
|
await minioClient.fPutObject(bucketName, fileName, filePath);
|
||
|
|
const publicUrl = `https://s3.raj.how/${bucketName}/${fileName}`;
|
||
|
|
return {
|
||
|
|
publicUrl,
|
||
|
|
};
|
||
|
|
} catch (error) {
|
||
|
|
return {
|
||
|
|
error: JSON.stringify(error),
|
||
|
|
};
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Get list of all files in the bucket and return their public URLs
|
||
|
|
export async function get_file_list({
|
||
|
|
bucketName = "public",
|
||
|
|
}: GetFileListParams) {
|
||
|
|
try {
|
||
|
|
const fileUrls: string[] = [];
|
||
|
|
const stream = await minioClient.listObjects(bucketName, "", true);
|
||
|
|
|
||
|
|
for await (const obj of stream) {
|
||
|
|
const fileUrl = `https://s3.raj.how/${bucketName}/${obj.name}`;
|
||
|
|
fileUrls.push(fileUrl);
|
||
|
|
}
|
||
|
|
|
||
|
|
return fileUrls;
|
||
|
|
} catch (error) {
|
||
|
|
return {
|
||
|
|
error: JSON.stringify(error),
|
||
|
|
};
|
||
|
|
}
|
||
|
|
}
|