@goodandready/dsh-subscriptions 0.1.1 → 0.2.0

package/lib/images.js

@@ -0,0 +1,111 @@
+// Генерация картинок на подписке.
+//
+// Здесь только протокол: куда идти, что послать и как прочитать ответ. Всё
+// остальное — сохранение файла, вложение в разговор, карточка — дело плагина
+// генерации; этот плагин лишь одалживает свой аккаунт.
+//
+// Токен наружу не отдаётся: плагин объявляет службу внутри процесса, а не
+// маршрут в сети. Харнесс на этой машине доступен без пароля, и ручка,
+// раздающая живой токен подписки, была бы дырой пошире тех, что мы закрывали.
+
+/** Куда уходит запрос у подписки ChatGPT. */
+export const CODEX_URL = 'https://chatgpt.com/backend-api/codex/images/generations'
+/** Модель, которую отдаёт этот адрес. */
+export const CODEX_MODEL = 'gpt-image-2'
+/** Куда уходит запрос у подписки Grok. */
+export const GROK_URL = 'https://api.x.ai/v1/images/generations'
+/** Модель, которую отдаёт этот адрес. */
+export const GROK_MODEL = 'grok-imagine-image-2.0'
+
+/** Размеры, которые понимает ChatGPT. */
+export const SIZES = ['1024x1024', '1024x1536', '1536x1024', 'auto']
+
+/** Grok мыслит не размерами, а соотношением сторон. */
+const GROK_ASPECT = {
+  '1024x1024': '1:1',
+  '1024x1536': '2:3',
+  '1536x1024': '3:2',
+  auto: 'auto',
+}
+
+export function codexBody({ prompt, size, quality }) {
+  const text = String(prompt || '').trim()
+  if (!text) throw new Error('нужен непустой запрос')
+  return {
+    prompt: text,
+    model: CODEX_MODEL,
+    ...(size ? { size } : {}),
+    ...(quality ? { quality } : {}),
+  }
+}
+
+export function grokBody({ prompt, size, quality }) {
+  const text = String(prompt || '').trim()
+  if (!text) throw new Error('нужен непустой запрос')
+  // У Grok качество всего двух ступеней: высокое складывается со средним.
+  const level = quality === 'low' ? 'low'
+    : (quality === 'medium' || quality === 'high') ? 'medium'
+      : undefined
+  return {
+    prompt: text,
+    model: GROK_MODEL,
+    response_format: 'b64_json',
+    ...(size && GROK_ASPECT[size] ? { aspect_ratio: GROK_ASPECT[size] } : {}),
+    ...(level ? { quality: level } : {}),
+  }
+}
+
+/** Разбор ответа: обе стороны отвечают одинаково. */
+export function parseImages(payload) {
+  const body = payload && typeof payload === 'object' ? payload : {}
+  const rows = Array.isArray(body.data) ? body.data : []
+  const images = []
+  for (const row of rows) {
+    if (!row || typeof row !== 'object') continue
+    if (typeof row.b64_json !== 'string' || !row.b64_json) continue
+    images.push({
+      b64_json: row.b64_json,
+      ...(typeof row.revised_prompt === 'string' && row.revised_prompt
+        ? { revisedPrompt: row.revised_prompt }
+        : {}),
+    })
+  }
+  if (!images.length) throw new Error('в ответе нет картинок')
+  return images
+}
+
+/**
+ * Один запрос к нужному адресу с заголовками этого провайдера.
+ *
+ * @param options {{provider, prompt, size, quality, session, fetchImpl, signal}}
+ *   session — уже освежённый блок токенов: accessToken и, для ChatGPT, accountId.
+ */
+export async function generateOnce(options) {
+  const { provider, session, fetchImpl, signal } = options
+  const isCodex = provider === 'codex'
+  const url = isCodex ? CODEX_URL : GROK_URL
+  const body = isCodex ? codexBody(options) : grokBody(options)
+  const headers = isCodex
+    ? {
+      authorization: `Bearer ${session.accessToken}`,
+      // ChatGPT различает аккаунты отдельным заголовком, без него отвечает отказом.
+      'chatgpt-account-id': session.accountId || '',
+      originator: 'codex_cli_rs',
+      'content-type': 'application/json',
+      accept: 'application/json',
+    }
+    : {
+      authorization: `Bearer ${session.accessToken}`,
+      'content-type': 'application/json',
+      accept: 'application/json',
+    }
+
+  const res = await fetchImpl(url, { method: 'POST', headers, body: JSON.stringify(body), signal })
+  const payload = await res.json().catch(() => ({}))
+  if (!res.ok) {
+    const detail = payload && payload.error
+      && (payload.error.message || payload.error.code || payload.error)
+    throw new Error(`${provider} HTTP ${res.status}${detail ? ': ' + String(detail).slice(0, 200) : ''}`)
+  }
+  return parseImages(payload)
+}

package/lib/index.js

@@ -5,6 +5,8 @@ import { parseCallbackInput, requestOrigin, webCallbackUri } from './oauth.js'
 import { writeJson, writeHtml, readBody, isTrustedSettingsRequest, queryOf } from './http.js'
 import { getVendor } from './vendors/index.js'
 import { SubscriptionAdapter } from './adapter.js'
+import { generateOnce, SIZES as IMAGE_SIZES } from './images.js'
+import { parseBlob } from './blob.js'
 import { createAccountStore, normalizeSlots, vendorConfig } from './accounts.js'
 import {
   inspectGoogleAccount,
@@ -77,6 +79,46 @@ export function apply(ctx, config) {
 
   const live = () => Config(structuredClone(getConfig() ?? {})) ?? config
   const store = createAccountStore({ credentials: ctx.credentials, getConfig: live, fetchImpl: fetch })
+
+  // Служба генерации картинок на подписке.
+  //
+  // Наружу отдаётся действие, а не токен: маршрут в сети раздавал бы живой
+  // ключ доступа каждому, кто дотянется до харнесса, а служба живёт внутри
+  // процесса и видна только другим плагинам. Обновлением токена по-прежнему
+  // занимается один хозяин — этот плагин.
+  ctx.effect(() => ctx.provide('subscriptionImages', {
+    /** Провайдеры, у которых есть вход прямо сейчас. */
+    async available() {
+      const logged = await store.loggedInProviders()
+      return ['codex', 'grok'].filter((name) => logged && logged[name])
+    },
+    sizes: IMAGE_SIZES,
+    /**
+     * @param request {{provider, prompt, size, quality, signal}}
+     * @returns [{ b64_json, revisedPrompt? }]
+     */
+    async generate(request) {
+      const provider = request && request.provider
+      if (provider !== 'codex' && provider !== 'grok') {
+        throw new Error(`неизвестный провайдер подписки: ${provider}`)
+      }
+      const accounts = await store.listAccounts(provider)
+      const slot = (accounts || []).find((row) => row && row.ref)
+      if (!slot) throw new Error(`нет входа в ${provider}: войдите в разделе «Подписки»`)
+      const raw = await store.resolveRaw(slot.ref)
+      if (!raw) throw new Error(`нет входа в ${provider}: войдите в разделе «Подписки»`)
+      const session = await store.ensureFresh(provider, parseBlob(raw), slot.ref)
+      return generateOnce({
+        provider,
+        prompt: request.prompt,
+        size: request.size,
+        quality: request.quality,
+        session,
+        fetchImpl: fetch,
+        signal: request.signal,
+      })
+    },
+  }), 'dsh-subscriptions: служба генерации картинок')
   const pending = new Map()
   const adapter = new SubscriptionAdapter({
     listAccounts: (provider) => store.listAccounts(provider),

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@goodandready/dsh-subscriptions",
-  "version": "0.1.1",
+  "version": "0.2.0",
   "description": "Use ChatGPT Codex, Claude, Grok, and Antigravity subscriptions as DeepSeek Harness LLM providers via OAuth.",
   "license": "MIT",
   "type": "module",