@umituz/react-native-ai-pruna-provider 1.0.7 → 1.0.8

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@umituz/react-native-ai-pruna-provider",
-  "version": "1.0.7",
+  "version": "1.0.8",
   "description": "Pruna AI provider for React Native - implements IAIProvider interface for unified AI generation",
   "main": "./src/index.ts",
   "types": "./src/index.ts",

package/src/infrastructure/services/pruna-api-client.ts

@@ -4,7 +4,7 @@
  *
  * Endpoints:
  *   POST /v1/predictions  — submit generation (with Try-Sync header for immediate results)
- *   POST /v1/files        — upload images for p-video (requires file URL, not base64)
+ *   POST /v1/files        — upload files (images, audio) for p-video (requires file URL, not base64)
  *   GET  {poll_url}       — poll async results
  *
  * Authentication: `apikey` header
@@ -14,45 +14,11 @@
 import type { PrunaModelId, PrunaPredictionResponse, PrunaFileUploadResponse } from "../../domain/entities/pruna.types";
 import { PRUNA_BASE_URL, PRUNA_PREDICTIONS_URL, PRUNA_FILES_URL } from "./pruna-provider.constants";
 import { generationLogCollector } from "../utils/log-collector";
+import { detectMimeType } from "../utils/mime-detection.util";
+import { getExtensionForMime } from "../utils/constants/mime.constants";
 
 const TAG = 'pruna-api';
 
-/**
- * Detect MIME type from raw binary bytes.
- * Supports image (png, jpeg, webp) and audio (mp3, wav, flac, m4a/aac) formats.
- */
-function detectMimeType(bytes: Uint8Array): string {
-  // JPEG
-  if (bytes[0] === 0xFF && bytes[1] === 0xD8) return 'image/jpeg';
-  // PNG
-  if (bytes[0] === 0x89 && bytes[1] === 0x50) return 'image/png';
-  // RIFF container — WAV or WebP
-  if (bytes[0] === 0x52 && bytes[1] === 0x49 && bytes[2] === 0x46 && bytes[3] === 0x46) {
-    if (bytes.length > 11 && bytes[8] === 0x57 && bytes[9] === 0x41 && bytes[10] === 0x56 && bytes[11] === 0x45) return 'audio/wav';
-    return 'image/webp';
-  }
-  // MP3 with ID3 tag
-  if (bytes[0] === 0x49 && bytes[1] === 0x44 && bytes[2] === 0x33) return 'audio/mpeg';
-  // MP3 sync word (0xFF followed by 0xE0-0xFF)
-  if (bytes[0] === 0xFF && (bytes[1] & 0xE0) === 0xE0 && bytes[1] !== 0xFF) return 'audio/mpeg';
-  // FLAC
-  if (bytes[0] === 0x66 && bytes[1] === 0x4C && bytes[2] === 0x61 && bytes[3] === 0x43) return 'audio/flac';
-  // M4A/AAC (MP4 container — ftyp box at offset 4)
-  if (bytes.length > 7 && bytes[4] === 0x66 && bytes[5] === 0x74 && bytes[6] === 0x79 && bytes[7] === 0x70) return 'audio/mp4';
-  // Default
-  return 'image/png';
-}
-
-const MIME_EXTENSION_MAP: Record<string, string> = {
-  'image/png': 'png',
-  'image/jpeg': 'jpg',
-  'image/webp': 'webp',
-  'audio/mpeg': 'mp3',
-  'audio/wav': 'wav',
-  'audio/flac': 'flac',
-  'audio/mp4': 'm4a',
-};
-
 /**
  * Upload a base64 file (image or audio) to Pruna's file storage.
  * p-video requires file URLs (not raw base64).
@@ -87,7 +53,7 @@ export async function uploadFileToStorage(
   }
 
   const mime = detectMimeType(bytes);
-  const ext = MIME_EXTENSION_MAP[mime] || mime.split('/')[1];
+  const ext = getExtensionForMime(mime);
   const blob = new Blob([bytes], { type: mime });
   const formData = new FormData();
   formData.append('content', blob, `upload.${ext}`);

package/src/infrastructure/utils/constants/index.ts

@@ -0,0 +1,12 @@
+export {
+  MIME_IMAGE_PNG,
+  MIME_IMAGE_JPEG,
+  MIME_IMAGE_WEBP,
+  MIME_AUDIO_MPEG,
+  MIME_AUDIO_WAV,
+  MIME_AUDIO_FLAC,
+  MIME_AUDIO_MP4,
+  MIME_DEFAULT,
+  MIME_TO_EXTENSION,
+  getExtensionForMime,
+} from "./mime.constants";

package/src/infrastructure/utils/constants/mime.constants.ts

@@ -0,0 +1,37 @@
+/**
+ * MIME Type Constants
+ * Supported media types for Pruna file uploads (images + audio)
+ */
+
+// ── Image MIME types ────────────────────────────────────────
+export const MIME_IMAGE_PNG = 'image/png' as const;
+export const MIME_IMAGE_JPEG = 'image/jpeg' as const;
+export const MIME_IMAGE_WEBP = 'image/webp' as const;
+
+// ── Audio MIME types (p-video audio input: flac, mp3, wav) ──
+export const MIME_AUDIO_MPEG = 'audio/mpeg' as const;
+export const MIME_AUDIO_WAV = 'audio/wav' as const;
+export const MIME_AUDIO_FLAC = 'audio/flac' as const;
+export const MIME_AUDIO_MP4 = 'audio/mp4' as const;
+
+// ── Fallback ────────────────────────────────────────────────
+export const MIME_DEFAULT = MIME_IMAGE_PNG;
+
+/** Maps MIME type → file extension for upload naming */
+export const MIME_TO_EXTENSION: Readonly<Record<string, string>> = {
+  [MIME_IMAGE_PNG]: 'png',
+  [MIME_IMAGE_JPEG]: 'jpg',
+  [MIME_IMAGE_WEBP]: 'webp',
+  [MIME_AUDIO_MPEG]: 'mp3',
+  [MIME_AUDIO_WAV]: 'wav',
+  [MIME_AUDIO_FLAC]: 'flac',
+  [MIME_AUDIO_MP4]: 'm4a',
+};
+
+/**
+ * Get file extension for a MIME type.
+ * Falls back to the subtype (e.g. "png" from "image/png").
+ */
+export function getExtensionForMime(mime: string): string {
+  return MIME_TO_EXTENSION[mime] || mime.split('/')[1] || 'bin';
+}

package/src/infrastructure/utils/index.ts

@@ -28,3 +28,9 @@ export {
 
 export { generationLogCollector } from "./log-collector";
 export type { LogEntry } from "./log-collector";
+
+export { detectMimeType } from "./mime-detection.util";
+export {
+  MIME_TO_EXTENSION,
+  getExtensionForMime,
+} from "./constants/mime.constants";

package/src/infrastructure/utils/mime-detection.util.ts

@@ -0,0 +1,68 @@
+/**
+ * MIME Type Detection Utility
+ * Detects file type from binary content using magic byte signatures.
+ *
+ * Supported formats:
+ *   Image: PNG, JPEG, WebP
+ *   Audio: MP3 (ID3 + sync word), WAV (RIFF/WAVE), FLAC, M4A/AAC (MP4 ftyp)
+ */
+
+import {
+  MIME_IMAGE_PNG,
+  MIME_IMAGE_JPEG,
+  MIME_IMAGE_WEBP,
+  MIME_AUDIO_MPEG,
+  MIME_AUDIO_WAV,
+  MIME_AUDIO_FLAC,
+  MIME_AUDIO_MP4,
+  MIME_DEFAULT,
+} from "./constants/mime.constants";
+
+/**
+ * Detect MIME type from raw binary bytes using magic number signatures.
+ *
+ * Detection order is intentional:
+ *   1. JPEG (0xFF 0xD8) — checked before MP3 sync word to avoid false positives
+ *   2. PNG  (0x89 0x50)
+ *   3. RIFF container → distinguish WAV vs WebP via subformat at offset 8-11
+ *   4. MP3 with ID3 tag (0x49 0x44 0x33)
+ *   5. MP3 sync word (0xFF 0xE_) — after JPEG to prevent overlap
+ *   6. FLAC (fLaC)
+ *   7. M4A/AAC (ftyp box at offset 4)
+ */
+export function detectMimeType(bytes: Uint8Array): string {
+  if (bytes.length < 4) return MIME_DEFAULT;
+
+  // ── Image formats ───────────────────────────────────────
+  // JPEG: FF D8
+  if (bytes[0] === 0xFF && bytes[1] === 0xD8) return MIME_IMAGE_JPEG;
+
+  // PNG: 89 50 4E 47
+  if (bytes[0] === 0x89 && bytes[1] === 0x50) return MIME_IMAGE_PNG;
+
+  // RIFF container — WAV (RIFF....WAVE) or WebP (RIFF....WEBP)
+  if (bytes[0] === 0x52 && bytes[1] === 0x49 && bytes[2] === 0x46 && bytes[3] === 0x46) {
+    if (bytes.length > 11) {
+      // WAVE at offset 8
+      if (bytes[8] === 0x57 && bytes[9] === 0x41 && bytes[10] === 0x56 && bytes[11] === 0x45) return MIME_AUDIO_WAV;
+      // WEBP at offset 8
+      if (bytes[8] === 0x57 && bytes[9] === 0x45 && bytes[10] === 0x42 && bytes[11] === 0x50) return MIME_IMAGE_WEBP;
+    }
+    return MIME_IMAGE_WEBP;
+  }
+
+  // ── Audio formats ───────────────────────────────────────
+  // MP3 with ID3v2 tag: 49 44 33
+  if (bytes[0] === 0x49 && bytes[1] === 0x44 && bytes[2] === 0x33) return MIME_AUDIO_MPEG;
+
+  // MP3 frame sync word: FF Ex/Fx (but not FF FF)
+  if (bytes[0] === 0xFF && (bytes[1] & 0xE0) === 0xE0 && bytes[1] !== 0xFF) return MIME_AUDIO_MPEG;
+
+  // FLAC: 66 4C 61 43 ("fLaC")
+  if (bytes[0] === 0x66 && bytes[1] === 0x4C && bytes[2] === 0x61 && bytes[3] === 0x43) return MIME_AUDIO_FLAC;
+
+  // M4A / AAC in MP4 container: ftyp box at offset 4
+  if (bytes.length > 7 && bytes[4] === 0x66 && bytes[5] === 0x74 && bytes[6] === 0x79 && bytes[7] === 0x70) return MIME_AUDIO_MP4;
+
+  return MIME_DEFAULT;
+}