@rethinkingstudio/clawpilot 2.0.0-beta.0 → 2.0.0

package/src/media/oss-uploader.ts

@@ -1,306 +0,0 @@
-import { createHmac } from "crypto";
-import { randomUUID } from "crypto";
-import { extname } from "path";
-
-// ---------------------------------------------------------------------------
-// Types
-// ---------------------------------------------------------------------------
-
-export interface StsCredentials {
-  bucket: string;
-  region: string;
-  endpoint: string;
-  baseUrl: string;
-  dirPrefix: string;
-  expiresAt: number;
-  credentials: {
-    accessKeyId: string;
-    accessKeySecret: string;
-    securityToken: string;
-  };
-}
-
-export interface UploadResult {
-  url: string;          // public CDN URL
-  thumbnailUrl?: string; // OSS snapshot URL (video only)
-  mimeType: string;
-  size: number;
-}
-
-// ---------------------------------------------------------------------------
-// OSS HMAC-SHA1 request signing (with STS security token)
-// ---------------------------------------------------------------------------
-
-function ossSign(
-  method: string,
-  contentMd5: string,
-  contentType: string,
-  date: string,
-  canonicalizedHeaders: string,
-  canonicalizedResource: string,
-  secretKey: string
-): string {
-  const stringToSign = [
-    method,
-    contentMd5,
-    contentType,
-    date,
-    canonicalizedHeaders + canonicalizedResource,
-  ].join("\n");
-  return createHmac("sha1", secretKey).update(stringToSign).digest("base64");
-}
-
-function buildBucketEndpoint(endpoint: string, bucket: string): string {
-  const url = new URL(endpoint);
-  if (!url.hostname.startsWith(`${bucket}.`)) {
-    url.hostname = `${bucket}.${url.hostname}`;
-  }
-  return url.toString().replace(/\/$/, "");
-}
-
-// ---------------------------------------------------------------------------
-// Simple PutObject upload (images / small files)
-// ---------------------------------------------------------------------------
-
-export async function putObject(
-  sts: StsCredentials,
-  data: Buffer,
-  ext: string,
-  mimeType: string
-): Promise<UploadResult> {
-  const { bucket, endpoint, baseUrl, dirPrefix, credentials } = sts;
-  const objectKey = `${dirPrefix}/${randomUUID()}${ext}`;
-  const date = new Date().toUTCString();
-  const canonicalizedResource = `/${bucket}/${objectKey}`;
-  const canonicalizedHeaders = `x-oss-security-token:${credentials.securityToken}\n`;
-
-  const sig = ossSign(
-    "PUT",
-    "",
-    mimeType,
-    date,
-    canonicalizedHeaders,
-    canonicalizedResource,
-    credentials.accessKeySecret
-  );
-
-  const url = `${buildBucketEndpoint(endpoint, bucket)}/${objectKey}`;
-  const res = await fetch(url, {
-    method: "PUT",
-    headers: {
-      "Content-Type": mimeType,
-      "Date": date,
-      "x-oss-security-token": credentials.securityToken,
-      "Authorization": `OSS ${credentials.accessKeyId}:${sig}`,
-    },
-    body: data as unknown as BodyInit,
-  });
-
-  if (!res.ok) {
-    const text = await res.text().catch(() => "");
-    throw new Error(`OSS PutObject failed: ${res.status} ${text}`);
-  }
-
-  const publicUrl = `${baseUrl.replace(/\/$/, "")}/${objectKey}`;
-  return { url: publicUrl, mimeType, size: data.length };
-}
-
-// ---------------------------------------------------------------------------
-// Multipart upload (videos > 10 MB)
-// ---------------------------------------------------------------------------
-
-const PART_SIZE = 5 * 1024 * 1024; // 5 MB per part
-
-export async function multipartUpload(
-  sts: StsCredentials,
-  data: Buffer,
-  ext: string,
-  mimeType: string
-): Promise<UploadResult> {
-  const { bucket, endpoint, baseUrl, dirPrefix, credentials } = sts;
-  const objectKey = `${dirPrefix}/${randomUUID()}${ext}`;
-  const baseOssUrl = `${buildBucketEndpoint(endpoint, bucket)}/${objectKey}`;
-
-  function makeHeaders(
-    method: string,
-    contentType: string,
-    extraHeaders: Record<string, string>,
-    resource: string
-  ): Headers {
-    const date = new Date().toUTCString();
-    const canonicalizedHeaders = [
-      "x-oss-security-token:" + credentials.securityToken,
-      ...Object.entries(extraHeaders)
-        .filter(([k]) => k.startsWith("x-oss-"))
-        .sort()
-        .map(([k, v]) => `${k}:${v}`),
-    ]
-      .sort()
-      .join("\n") + "\n";
-
-    const sig = ossSign(
-      method,
-      "",
-      contentType,
-      date,
-      canonicalizedHeaders,
-      `/${bucket}/${objectKey}`,
-      credentials.accessKeySecret
-    );
-
-    const headers = new Headers({
-      "Date": date,
-      "x-oss-security-token": credentials.securityToken,
-      "Authorization": `OSS ${credentials.accessKeyId}:${sig}`,
-      ...extraHeaders,
-    });
-    if (contentType) headers.set("Content-Type", contentType);
-    return headers;
-  }
-
-  // 1. Initiate
-  const initRes = await fetch(`${baseOssUrl}?uploads`, {
-    method: "POST",
-    headers: makeHeaders("POST", mimeType, {}, `/${bucket}/${objectKey}`),
-  });
-  if (!initRes.ok) throw new Error(`OSS InitiateMultipartUpload failed: ${initRes.status}`);
-  const initText = await initRes.text();
-  const uploadIdMatch = initText.match(/<UploadId>([^<]+)<\/UploadId>/);
-  if (!uploadIdMatch) throw new Error("OSS: could not parse UploadId");
-  const uploadId = uploadIdMatch[1];
-
-  // 2. Upload parts
-  const parts: Array<{ partNumber: number; etag: string }> = [];
-  let partNumber = 1;
-  let offset = 0;
-
-  try {
-    while (offset < data.length) {
-      const chunk = data.slice(offset, offset + PART_SIZE);
-      const partDate = new Date().toUTCString();
-      const canonicalizedHeaders = `x-oss-security-token:${credentials.securityToken}\n`;
-      const sig = ossSign(
-        "PUT", "", "application/octet-stream", partDate,
-        canonicalizedHeaders, `/${bucket}/${objectKey}`,
-        credentials.accessKeySecret
-      );
-      const partRes = await fetch(
-        `${baseOssUrl}?partNumber=${partNumber}&uploadId=${uploadId}`,
-        {
-          method: "PUT",
-          headers: {
-            "Content-Type": "application/octet-stream",
-            "Date": partDate,
-            "x-oss-security-token": credentials.securityToken,
-            "Authorization": `OSS ${credentials.accessKeyId}:${sig}`,
-          },
-          body: chunk as unknown as BodyInit,
-        }
-      );
-      if (!partRes.ok) throw new Error(`OSS UploadPart ${partNumber} failed: ${partRes.status}`);
-      const etag = partRes.headers.get("etag") ?? "";
-      parts.push({ partNumber, etag });
-      partNumber++;
-      offset += PART_SIZE;
-    }
-
-    // 3. Complete
-    const completeBody = [
-      "<CompleteMultipartUpload>",
-      ...parts.map(
-        (p) => `<Part><PartNumber>${p.partNumber}</PartNumber><ETag>${p.etag}</ETag></Part>`
-      ),
-      "</CompleteMultipartUpload>",
-    ].join("");
-
-    const completeDate = new Date().toUTCString();
-    const cHeaders = `x-oss-security-token:${credentials.securityToken}\n`;
-    const cSig = ossSign(
-      "POST", "", "application/xml", completeDate,
-      cHeaders, `/${bucket}/${objectKey}`, credentials.accessKeySecret
-    );
-    const completeRes = await fetch(`${baseOssUrl}?uploadId=${uploadId}`, {
-      method: "POST",
-      headers: {
-        "Content-Type": "application/xml",
-        "Date": completeDate,
-        "x-oss-security-token": credentials.securityToken,
-        "Authorization": `OSS ${credentials.accessKeyId}:${cSig}`,
-      },
-      body: completeBody,
-    });
-    if (!completeRes.ok) throw new Error(`OSS CompleteMultipartUpload failed: ${completeRes.status}`);
-
-  } catch (err) {
-    // Best-effort abort to avoid billing for incomplete upload
-    try {
-      const abortDate = new Date().toUTCString();
-      const aHeaders = `x-oss-security-token:${credentials.securityToken}\n`;
-      const aSig = ossSign(
-        "DELETE", "", "", abortDate,
-        aHeaders, `/${bucket}/${objectKey}`, credentials.accessKeySecret
-      );
-      await fetch(`${baseOssUrl}?uploadId=${uploadId}`, {
-        method: "DELETE",
-        headers: {
-          "Date": abortDate,
-          "x-oss-security-token": credentials.securityToken,
-          "Authorization": `OSS ${credentials.accessKeyId}:${aSig}`,
-        },
-      });
-    } catch { /* ignore abort errors */ }
-    throw err;
-  }
-
-  const publicUrl = `${baseUrl.replace(/\/$/, "")}/${objectKey}`;
-  // OSS video snapshot thumbnail URL
-  const thumbnailUrl = `${publicUrl}?x-oss-process=video/snapshot,t_0,f_jpg,w_0,h_0,m_fast`;
-  return { url: publicUrl, thumbnailUrl, mimeType, size: data.length };
-}
-
-// ---------------------------------------------------------------------------
-// Convenience: auto-select PutObject vs multipart based on size
-// ---------------------------------------------------------------------------
-
-const MULTIPART_THRESHOLD = 10 * 1024 * 1024; // 10 MB
-
-export async function uploadMedia(
-  sts: StsCredentials,
-  data: Buffer,
-  mimeType: string,
-  fileName?: string
-): Promise<UploadResult> {
-  const ext = extname(fileName ?? "") || mimeTypeToExt(mimeType);
-  if (data.length >= MULTIPART_THRESHOLD) {
-    return multipartUpload(sts, data, ext, mimeType);
-  }
-  return putObject(sts, data, ext, mimeType);
-}
-
-function mimeTypeToExt(mimeType: string): string {
-  const map: Record<string, string> = {
-    "image/jpeg": ".jpg",
-    "image/png": ".png",
-    "image/gif": ".gif",
-    "image/webp": ".webp",
-    "video/mp4": ".mp4",
-    "video/quicktime": ".mov",
-    "audio/mpeg": ".mp3",
-    "audio/mp4": ".m4a",
-    "audio/wav": ".wav",
-    "audio/x-wav": ".wav",
-    "audio/aac": ".aac",
-    "application/pdf": ".pdf",
-    "text/plain": ".txt",
-    "text/markdown": ".md",
-    "application/json": ".json",
-    "application/vnd.ms-excel": ".xls",
-    "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet": ".xlsx",
-    "application/msword": ".doc",
-    "application/vnd.openxmlformats-officedocument.wordprocessingml.document": ".docx",
-    "application/vnd.ms-powerpoint": ".ppt",
-    "application/vnd.openxmlformats-officedocument.presentationml.presentation": ".pptx",
-    "text/csv": ".csv",
-  };
-  return map[mimeType] ?? ".bin";
-}