rustfs-console/plugins/03.s3.ts

import { S3Client } from '@aws-sdk/client-s3'
import type { SiteConfig } from '~/types/config'
import type {
  DeserializeHandler,
  DeserializeHandlerArguments,
  DeserializeHandlerOutput
} from '@aws-sdk/types'

interface S3Response {
  response?: {
    body?: string
  }
  [key: string]: any
}

export default defineNuxtPlugin({
  name: 's3-client',
  setup(nuxtApp) {
    const { credentials, isAuthenticated } = useAuth()
    const siteConfig = nuxtApp.$siteConfig as SiteConfig

    if (!isAuthenticated || !credentials.value) {
      return
    }

    const client = new S3Client({
      endpoint: siteConfig.s3.endpoint,
      region: siteConfig.s3.region || 'us-east-1',
      forcePathStyle: true,
      // https://github.com/aws/aws-sdk-js-v3/issues/6834#issuecomment-2611346849
      requestChecksumCalculation: 'WHEN_REQUIRED',
      credentials: {
        accessKeyId: credentials.value?.AccessKeyId || '',
        secretAccessKey: credentials.value?.SecretAccessKey || '',
        sessionToken: credentials.value?.SessionToken || ''
      }
    })

    // 添加中间件处理XML响应和401错误
    client.middlewareStack.add(
      (next: DeserializeHandler<any, any>) =>
        async (
          args: DeserializeHandlerArguments<any>
        ): Promise<DeserializeHandlerOutput<any>> => {
          try {
            const response = (await next(args)) as S3Response

            // 检查响应是否为XML格式
            if (
              response.response?.body &&
              typeof response.response.body === 'string'
            ) {
              const body = response.response.body.trim()

              // 检查是否是空的XML响应
              if (body.match(/^<\?xml[^>]*\?><[^>]*><\/[^>]*>$/)) {
                // 从XML标签名提取属性名
                const tagName = body.match(/<([^>]*)><\/\1>/)?.[1]
                if (tagName) {
                  // 将XML标签名转换为驼峰命名
                  const propertyName = tagName.replace(
                    /(?:^|_)([a-z])/g,
                    (_, letter) => letter.toUpperCase()
                  )
                  // 返回null，保持response结构
                  return {
                    response: response.response,
                    [propertyName]: null
                  } as unknown as DeserializeHandlerOutput<any>
                }
              }
            }

            return response as DeserializeHandlerOutput<any>
          } catch (error: any) {
            if (error?.$metadata?.httpStatusCode === 401) {
              await useAuth().logoutAndRedirect()
            }

            // 处理 S3 客户端错误，优先抛出 error.message 作为新的 Error
            if (error?.Code) {
              throw new Error(error.Code)
            }
            throw error
          }
        },
      {
        step: 'deserialize',
        name: 'handleXmlResponse'
      }
    )

    return {
      provide: {
        s3Client: client
      }
    }
  }
})