kingkont 0.20.29 → 0.20.30

package/lib/providers.js

@@ -1007,9 +1007,82 @@ async function audioViaChatium(s, body) {
 }
 
 // =============================================================================
-// PUBLIC API: file upload (Chatium-storage с fallback на KIE).
+// PUBLIC API: file upload (Chatium-storage с fallback на R2 и KIE).
 // =============================================================================
 
+const r2Upload = require('./r2-upload');
+
+// Подъём chatium-uploads до 5 секунд, дальше race-win на R2.
+// Юзер: «текущий вариант не нужно убивать. просто если видим что 5 секунд
+// не грузится референс в fs.chatium.ru — отправляем через cloudflare».
+const CHATIUM_UPLOAD_TIMEOUT_MS = 5000;
+
+async function _chatiumUploadOnly({ buffer, filename, mime, s }) {
+  const urlReq = chatiumBase(s) + CHATIUM_PATHS.uploadUrl;
+  const ru = await fetch(urlReq, { headers: { 'Authorization': `Bearer ${s.chatium.token}` } });
+  if (!ru.ok) throw new Error('KingKont upload-url: ' + (await ru.text()).slice(0, 200));
+  const { uploadUrl } = await ru.json();
+  if (!uploadUrl) throw new Error('KingKont не вернул uploadUrl');
+  logCall('POST', 'Chatium', uploadUrl, `file=${filename} size=${buffer.length}`);
+  const fd = new FormData();
+  fd.append('Filedata', new Blob([buffer], { type: mime }), filename);
+  const ru2 = await fetch(uploadUrl, { method: 'POST', body: fd });
+  const hash = (await ru2.text()).trim();
+  if (!ru2.ok || !hash) throw new Error(`Chatium upload HTTP ${ru2.status}: ${hash.slice(0, 200)}`);
+  return { url: `${CHATIUM_CDN}/${hash}`, fileName: filename, size: buffer.length, hash, provider: 'kingkont' };
+}
+
+// Race: chatium upload против 5s таймера. Если chatium успел — берём его
+// (закэшируется, экономия для следующих вызовов). Если тайм-аут — стартуем
+// R2 upload и возвращаем его URL. Chatium upload в фоне продолжается —
+// его результат всё равно сохранится в chatium-storage, просто мы для
+// ЭТОЙ генерации использовали R2.
+async function _uploadViaChatiumWithR2Fallback({ buffer, filename, mime, s }) {
+  const chatiumP = _chatiumUploadOnly({ buffer, filename, mime, s })
+    .catch(e => { console.warn('[uploadFile] chatium failed:', e?.message || e); throw e; });
+
+  const TIMEOUT_SENTINEL = Symbol('timeout');
+  const timerP = new Promise(resolve => setTimeout(resolve, CHATIUM_UPLOAD_TIMEOUT_MS, TIMEOUT_SENTINEL));
+
+  // Race: побеждает либо chatium-result, либо TIMEOUT_SENTINEL.
+  // Catch внутри chatiumP подавляет, но Promise.race увидит resolved/rejected
+  // тогда — обернём отдельный wrapper, чтобы reject не «выиграл» race.
+  const chatiumSafe = chatiumP.then(
+    r => ({ ok: true,  result: r }),
+    e => ({ ok: false, error: e }),
+  );
+  const winner = await Promise.race([chatiumSafe, timerP]);
+
+  if (winner !== TIMEOUT_SENTINEL) {
+    if (winner.ok) {
+      console.log('[uploadFile] chatium won race in <5s');
+      return winner.result;
+    }
+    // Chatium быстро упал. Падаем на R2 сразу.
+    console.warn('[uploadFile] chatium fast-failed → R2');
+  } else {
+    console.warn(`[uploadFile] chatium медленно (>${CHATIUM_UPLOAD_TIMEOUT_MS}ms) → R2`);
+  }
+
+  // Fallback на R2.
+  if (!r2Upload.isConfigured()) {
+    // R2 не настроен — ждём chatium докрутится (последний шанс).
+    console.warn('[uploadFile] R2 not configured, waiting chatium...');
+    const r = await chatiumSafe;
+    if (r.ok) return r.result;
+    throw r.error;
+  }
+  const r2 = await r2Upload.uploadToR2(buffer, mime, _extFromFilename(filename));
+  return {
+    url: r2.url, fileName: filename, size: buffer.length, hash: r2.key, provider: 'cloudflare-r2',
+  };
+}
+
+function _extFromFilename(name) {
+  const m = String(name || '').match(/\.([a-zA-Z0-9]+)$/);
+  return m ? m[1].toLowerCase() : null;
+}
+
 /**
  * @returns {Promise<{ url, fileName, size, hash?: string, provider }>}
  */
@@ -1020,18 +1093,7 @@ async function uploadFile({ buffer, filename = 'upload.bin', mime = 'application
   }
 
   if (s.useChatium && s.chatium?.token && s.chatium?.base) {
-    const urlReq = chatiumBase(s) + CHATIUM_PATHS.uploadUrl;
-    const ru = await fetch(urlReq, { headers: { 'Authorization': `Bearer ${s.chatium.token}` } });
-    if (!ru.ok) throw new Error('KingKont upload-url: ' + (await ru.text()).slice(0, 200));
-    const { uploadUrl } = await ru.json();
-    if (!uploadUrl) throw new Error('KingKont не вернул uploadUrl');
-    logCall('POST', 'Chatium', uploadUrl, `file=${filename} size=${buffer.length}`);
-    const fd = new FormData();
-    fd.append('Filedata', new Blob([buffer], { type: mime }), filename);
-    const ru2 = await fetch(uploadUrl, { method: 'POST', body: fd });
-    const hash = (await ru2.text()).trim();
-    if (!ru2.ok || !hash) throw new Error(`Chatium upload HTTP ${ru2.status}: ${hash.slice(0, 200)}`);
-    return { url: `${CHATIUM_CDN}/${hash}`, fileName: filename, size: buffer.length, hash, provider: 'kingkont' };
+    return await _uploadViaChatiumWithR2Fallback({ buffer, filename, mime, s });
   }
 
   if (!s.useKie) throw new Error('Войдите в KingKont или KIE для загрузки файлов.');

package/lib/r2-upload.js

@@ -0,0 +1,93 @@
+// lib/r2-upload.js — загрузка файла в Cloudflare R2 через Cloudflare API (Bearer).
+//
+// Используется как fallback когда chatium-storage медленно / недоступно
+// (см. uploadFile в providers.js → race с 5s timeout).
+//
+// Credentials хардкодим (юзер: «дам наш ключ, не надо чтобы каждый
+// настраивал»). Если в будущем нужно будет дать юзерам кастомизировать —
+// перенесём в settings.json.
+//
+// Способ auth: Cloudflare Account API Token (cfat_...) → Bearer header.
+// Это проще чем S3-protocol SigV4 (нет хитрой подписи), и token-based
+// auth тривиальнее ротировать в случае утечки.
+//
+// API endpoint:
+//   PUT https://api.cloudflare.com/client/v4/accounts/<account>/r2/buckets/<bucket>/objects/<key>
+//   Authorization: Bearer <token>
+//   Content-Type: <mime>
+//   body: <bytes>
+// → 200 OK { success:true, result:{key, size, etag} }
+//
+// Public read через managed-domain (pub-<bucket-id>.r2.dev), включён через
+// API (PUT /r2/buckets/<b>/domains/managed { enabled:true }).
+
+const crypto = require('crypto');
+
+const R2 = {
+  accountId:  'dd50e50ba5d89ce42553cfb2f5d6ee40',
+  apiToken:   'cfat_8HndHS0KMqecVBsjpM5Se2VbLQIvSw2ipCDnhtDm1502679f',
+  bucket:     'kingkont',
+  publicBase: 'https://pub-cd4114af9f7d44c9bf8c9442bc7dddc2.r2.dev',
+};
+// Env override (для dev / тестов / ротации без релиза).
+if (process.env.R2_ACCOUNT_ID)  R2.accountId  = process.env.R2_ACCOUNT_ID;
+if (process.env.R2_API_TOKEN)   R2.apiToken   = process.env.R2_API_TOKEN;
+if (process.env.R2_BUCKET)      R2.bucket     = process.env.R2_BUCKET;
+if (process.env.R2_PUBLIC_BASE) R2.publicBase = process.env.R2_PUBLIC_BASE;
+
+function isConfigured() {
+  return !!(R2.accountId && R2.apiToken && R2.bucket && R2.publicBase);
+}
+
+/**
+ * Залить файл в R2 и вернуть публичную URL'ку.
+ * @param {Buffer} buffer
+ * @param {string} mime
+ * @param {string} [ext]
+ * @returns {Promise<{ url: string, key: string, size: number }>}
+ */
+async function uploadToR2(buffer, mime = 'application/octet-stream', ext) {
+  if (!isConfigured()) throw new Error('R2 не настроен');
+  if (!buffer || !buffer.length) throw new Error('пустой файл');
+
+  const finalExt = (ext || _extFromMime(mime) || 'bin').replace(/[^a-z0-9]/gi, '').slice(0, 8) || 'bin';
+  // UUID-based — юзер просил без content-hash дедупа.
+  const key = `${crypto.randomUUID()}.${finalExt}`;
+
+  const url = `https://api.cloudflare.com/client/v4/accounts/${R2.accountId}/r2/buckets/${R2.bucket}/objects/${encodeURIComponent(key)}`;
+
+  const t0 = Date.now();
+  const r = await fetch(url, {
+    method: 'PUT',
+    headers: {
+      'Authorization': `Bearer ${R2.apiToken}`,
+      'Content-Type': mime,
+    },
+    body: buffer,
+  });
+  if (!r.ok) {
+    const text = await r.text().catch(() => '');
+    throw new Error(`R2 PUT ${r.status}: ${text.slice(0, 300)}`);
+  }
+  // CF API возвращает {success, result}. Проверяем флаг для надёжности.
+  const data = await r.json().catch(() => ({}));
+  if (data && data.success === false) {
+    throw new Error(`R2 PUT не success: ${JSON.stringify(data.errors || data).slice(0, 300)}`);
+  }
+  const publicUrl = `${R2.publicBase.replace(/\/$/, '')}/${key}`;
+  console.log(`[r2-upload] OK ${key} (${buffer.length}B, ${mime}) in ${Date.now() - t0}ms → ${publicUrl}`);
+  return { url: publicUrl, key, size: buffer.length };
+}
+
+function _extFromMime(mime) {
+  return ({
+    'image/jpeg': 'jpg', 'image/jpg': 'jpg',
+    'image/png': 'png', 'image/webp': 'webp', 'image/gif': 'gif', 'image/avif': 'avif',
+    'video/mp4': 'mp4', 'video/quicktime': 'mov', 'video/webm': 'webm',
+    'audio/mpeg': 'mp3', 'audio/wav': 'wav', 'audio/mp4': 'm4a', 'audio/x-m4a': 'm4a',
+    'text/plain': 'txt', 'text/markdown': 'md',
+    'application/octet-stream': 'bin',
+  })[mime] || null;
+}
+
+module.exports = { uploadToR2, isConfigured };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "kingkont",
-  "version": "0.20.29",
+  "version": "0.20.30",
   "description": "KingKont \u00b7 Chatium \u2014 \u043d\u043e\u0434-\u0440\u0435\u0434\u0430\u043a\u0442\u043e\u0440 \u0441\u0446\u0435\u043d \u0441 AI-\u0433\u0435\u043d\u0435\u0440\u0430\u0446\u0438\u0435\u0439 (\u043a\u0430\u0440\u0442\u0438\u043d\u043a\u0438/\u0432\u0438\u0434\u0435\u043e/\u0433\u043e\u043b\u043e\u0441/SFX/\u043c\u0443\u0437\u044b\u043a\u0430/\u0442\u0435\u043a\u0441\u0442)",
   "main": "main.js",
   "bin": {