@hasna/connectors 0.0.1

package/connectors/connect-stabilityai/src/index.ts

@@ -0,0 +1,37 @@
+// Stability AI API
+// TypeScript client for Stability AI's image generation API
+
+export { StabilityAI } from './api';
+export * from './types';
+
+// Re-export individual API classes for advanced usage
+export {
+  StabilityClient,
+  UserApi,
+  EnginesApi,
+  TextToImageApi,
+  ImageToImageApi,
+  InpaintApi,
+  OutpaintApi,
+  UpscaleApi,
+  EditApi,
+  ThreeDApi,
+} from './api';
+
+// Export config utilities
+export {
+  getApiKey,
+  setApiKey,
+  getCurrentProfile,
+  setCurrentProfile,
+  listProfiles,
+  createProfile,
+  deleteProfile,
+  loadProfile,
+  saveProfile,
+  clearConfig,
+  getExportsDir,
+  getImportsDir,
+  ensureExportsDir,
+  ensureImportsDir,
+} from './utils/config';

package/connectors/connect-stabilityai/src/types/index.ts

@@ -0,0 +1,315 @@
+// Stability AI API Types
+// AI-powered image generation, editing, and 3D model creation
+
+// ============================================
+// Configuration
+// ============================================
+
+export interface StabilityConfig {
+  apiKey: string;
+  baseUrl?: string;
+}
+
+// ============================================
+// Common Types
+// ============================================
+
+export type OutputFormat = 'json' | 'pretty';
+
+// ============================================
+// User/Account Types
+// ============================================
+
+export interface UserAccount {
+  id: string;
+  email: string;
+  profile_picture?: string;
+}
+
+export interface UserBalance {
+  credits: number;
+}
+
+// ============================================
+// Engine Types
+// ============================================
+
+export interface Engine {
+  id: string;
+  name: string;
+  description: string;
+  type: EngineType;
+}
+
+export type EngineType = 'AUDIO' | 'CLASSIFICATION' | 'PICTURE' | 'STORAGE' | 'TEXT' | 'VIDEO';
+
+// ============================================
+// Text-to-Image Types
+// ============================================
+
+export type StylePreset =
+  | '3d-model'
+  | 'analog-film'
+  | 'anime'
+  | 'cinematic'
+  | 'comic-book'
+  | 'digital-art'
+  | 'enhance'
+  | 'fantasy-art'
+  | 'isometric'
+  | 'line-art'
+  | 'low-poly'
+  | 'modeling-compound'
+  | 'neon-punk'
+  | 'origami'
+  | 'photographic'
+  | 'pixel-art'
+  | 'tile-texture';
+
+export type AspectRatio =
+  | '16:9'
+  | '1:1'
+  | '21:9'
+  | '2:3'
+  | '3:2'
+  | '4:5'
+  | '5:4'
+  | '9:16'
+  | '9:21';
+
+export type OutputFormatImage = 'jpeg' | 'png' | 'webp';
+
+export interface TextToImageOptions {
+  prompt: string;
+  model?: string;
+  negativePrompt?: string;
+  aspectRatio?: AspectRatio;
+  seed?: number;
+  outputFormat?: OutputFormatImage;
+  stylePreset?: StylePreset;
+  // v1 specific
+  cfgScale?: number;
+  height?: number;
+  width?: number;
+  samples?: number;
+  steps?: number;
+}
+
+export interface GenerationArtifact {
+  base64: string;
+  seed: number;
+  finishReason: 'SUCCESS' | 'CONTENT_FILTERED' | 'ERROR';
+  contentFiltered?: boolean;
+}
+
+export interface TextToImageResponse {
+  artifacts?: GenerationArtifact[];
+  // v2 response format
+  image?: string;
+  seed?: number;
+  finish_reason?: string;
+}
+
+// ============================================
+// Image-to-Image Types
+// ============================================
+
+export interface ImageToImageOptions {
+  image: string | Buffer; // File path or buffer
+  prompt: string;
+  model?: string;
+  negativePrompt?: string;
+  strength?: number; // 0-1, how much to transform
+  seed?: number;
+  outputFormat?: OutputFormatImage;
+  stylePreset?: StylePreset;
+  // v1 specific
+  cfgScale?: number;
+  samples?: number;
+  steps?: number;
+  mode?: 'image-to-image' | 'image-to-image-masking';
+}
+
+export interface ImageToImageResponse {
+  artifacts?: GenerationArtifact[];
+  image?: string;
+  seed?: number;
+  finish_reason?: string;
+}
+
+// ============================================
+// Inpainting Types
+// ============================================
+
+export interface InpaintOptions {
+  image: string | Buffer;
+  mask: string | Buffer;
+  prompt: string;
+  model?: string;
+  negativePrompt?: string;
+  seed?: number;
+  outputFormat?: OutputFormatImage;
+  growMask?: number;
+  // v1 specific
+  cfgScale?: number;
+  samples?: number;
+  steps?: number;
+}
+
+export interface InpaintResponse {
+  artifacts?: GenerationArtifact[];
+  image?: string;
+  seed?: number;
+  finish_reason?: string;
+}
+
+// ============================================
+// Outpainting Types
+// ============================================
+
+export type OutpaintDirection = 'left' | 'right' | 'up' | 'down';
+
+export interface OutpaintOptions {
+  image: string | Buffer;
+  prompt?: string;
+  negativePrompt?: string;
+  left?: number;
+  right?: number;
+  up?: number;
+  down?: number;
+  creativity?: number; // 0-1
+  seed?: number;
+  outputFormat?: OutputFormatImage;
+}
+
+export interface OutpaintResponse {
+  image: string;
+  seed: number;
+  finish_reason: string;
+}
+
+// ============================================
+// Upscale Types
+// ============================================
+
+export type UpscaleMode = 'conservative' | 'creative' | 'fast';
+
+export interface UpscaleOptions {
+  image: string | Buffer;
+  prompt?: string;
+  negativePrompt?: string;
+  outputFormat?: OutputFormatImage;
+  seed?: number;
+  creativity?: number;
+  // Creative upscale specific
+  style?: StylePreset;
+  // Fast upscale - no additional options needed
+}
+
+export interface UpscaleResponse {
+  image: string;
+  seed?: number;
+  finish_reason: string;
+}
+
+// ============================================
+// Edit Types
+// ============================================
+
+export interface EraseOptions {
+  image: string | Buffer;
+  mask?: string | Buffer;
+  growMask?: number;
+  seed?: number;
+  outputFormat?: OutputFormatImage;
+}
+
+export interface SearchAndReplaceOptions {
+  image: string | Buffer;
+  prompt: string;
+  searchPrompt: string;
+  negativePrompt?: string;
+  seed?: number;
+  outputFormat?: OutputFormatImage;
+  growMask?: number;
+}
+
+export interface RemoveBackgroundOptions {
+  image: string | Buffer;
+  outputFormat?: OutputFormatImage;
+}
+
+export interface ReplaceBackgroundOptions {
+  image: string | Buffer;
+  prompt: string;
+  negativePrompt?: string;
+  seed?: number;
+  outputFormat?: OutputFormatImage;
+}
+
+export interface EditResponse {
+  image: string;
+  seed?: number;
+  finish_reason: string;
+}
+
+// ============================================
+// 3D Types
+// ============================================
+
+export interface StableFast3DOptions {
+  image: string | Buffer;
+  textureResolution?: 512 | 1024 | 2048;
+  foregroundRatio?: number; // 0.1-1.0
+  remesh?: 'none' | 'triangle' | 'quad';
+  vertexCount?: number; // For triangle/quad remesh
+}
+
+export interface StableFast3DResponse {
+  glb: string; // Base64 encoded GLB file
+}
+
+export interface StableVideo3DOptions {
+  image: string | Buffer;
+  textureResolution?: 512 | 1024 | 2048;
+  foregroundRatio?: number;
+  remesh?: 'none' | 'triangle' | 'quad';
+  vertexCount?: number;
+}
+
+export interface StableVideo3DResponse {
+  glb: string;
+}
+
+// ============================================
+// Result Types (for async operations)
+// ============================================
+
+export interface GenerationResult {
+  id: string;
+  status: 'pending' | 'processing' | 'succeeded' | 'failed';
+  result?: string; // Base64 image
+  error?: string;
+}
+
+// ============================================
+// API Error Types
+// ============================================
+
+export interface ApiErrorDetail {
+  id: string;
+  name: string;
+  message: string;
+}
+
+export class StabilityApiError extends Error {
+  public readonly statusCode: number;
+  public readonly errors?: ApiErrorDetail[];
+
+  constructor(message: string, statusCode: number, errors?: ApiErrorDetail[]) {
+    super(message);
+    this.name = 'StabilityApiError';
+    this.statusCode = statusCode;
+    this.errors = errors;
+  }
+}

package/connectors/connect-stabilityai/src/utils/config.ts

@@ -0,0 +1,340 @@
+import { existsSync, readFileSync, writeFileSync, mkdirSync, readdirSync, rmSync, renameSync } from 'fs';
+import { homedir } from 'os';
+import { join } from 'path';
+
+const CONNECTOR_NAME = 'connect-stabilityai';
+const DEFAULT_PROFILE = 'default';
+const CURRENT_PROFILE_FILE = 'current_profile';
+const PROFILES_DIR = 'profiles';
+
+export interface ProfileConfig {
+  apiKey?: string;
+}
+
+// Store for --profile flag override (set by CLI before commands run)
+let profileOverride: string | undefined;
+
+// Store config in ~/.connect/connect-stabilityai/ (always in home directory)
+function resolveBaseConfigDir(): string {
+  return join(homedir(), '.connect', CONNECTOR_NAME);
+}
+
+const BASE_CONFIG_DIR = resolveBaseConfigDir();
+
+// ============================================
+// Profile Management
+// ============================================
+
+export function setProfileOverride(profile: string | undefined): void {
+  profileOverride = profile;
+}
+
+export function getProfileOverride(): string | undefined {
+  return profileOverride;
+}
+
+function ensureBaseConfigDir(): void {
+  if (!existsSync(BASE_CONFIG_DIR)) {
+    mkdirSync(BASE_CONFIG_DIR, { recursive: true });
+  }
+}
+
+function getProfilesDir(): string {
+  return join(BASE_CONFIG_DIR, PROFILES_DIR);
+}
+
+function getCurrentProfileFile(): string {
+  return join(BASE_CONFIG_DIR, CURRENT_PROFILE_FILE);
+}
+
+/**
+ * Migrate old config structure (JSON files) to new profile directory structure
+ */
+function migrateToProfileStructure(): void {
+  const profilesDir = getProfilesDir();
+  const defaultProfileDir = join(profilesDir, DEFAULT_PROFILE);
+
+  // If profiles directory already exists, no migration needed
+  if (existsSync(profilesDir)) {
+    return;
+  }
+
+  // Check if old structure exists (JSON files in profiles/)
+  const oldProfilesDir = join(BASE_CONFIG_DIR, 'profiles');
+  const hasOldJsonProfiles = existsSync(oldProfilesDir) &&
+    readdirSync(oldProfilesDir).some(f => f.endsWith('.json'));
+
+  if (!hasOldJsonProfiles) {
+    // Fresh install, just create the profiles directory
+    mkdirSync(profilesDir, { recursive: true });
+    return;
+  }
+
+  // Migrate JSON profiles to directory structure
+  const jsonFiles = readdirSync(oldProfilesDir).filter(f => f.endsWith('.json'));
+
+  for (const jsonFile of jsonFiles) {
+    const profileName = jsonFile.replace('.json', '');
+    const newProfileDir = join(profilesDir, profileName);
+    const oldJsonPath = join(oldProfilesDir, jsonFile);
+
+    mkdirSync(newProfileDir, { recursive: true });
+
+    // Move JSON content to config.json in the new directory
+    if (existsSync(oldJsonPath)) {
+      renameSync(oldJsonPath, join(newProfileDir, 'config.json'));
+    }
+  }
+}
+
+/**
+ * Check if a profile exists
+ */
+export function profileExists(profile: string): boolean {
+  migrateToProfileStructure();
+  const profileDir = join(getProfilesDir(), profile);
+  return existsSync(profileDir);
+}
+
+/**
+ * Get the current active profile name
+ */
+export function getCurrentProfile(): string {
+  // CLI override takes priority
+  if (profileOverride) {
+    return profileOverride;
+  }
+
+  ensureBaseConfigDir();
+  migrateToProfileStructure();
+
+  const currentProfileFile = getCurrentProfileFile();
+  if (existsSync(currentProfileFile)) {
+    try {
+      const profile = readFileSync(currentProfileFile, 'utf-8').trim();
+      if (profile && profileExists(profile)) {
+        return profile;
+      }
+    } catch {
+      // Fall through to default
+    }
+  }
+
+  return DEFAULT_PROFILE;
+}
+
+/**
+ * Set the current active profile
+ */
+export function setCurrentProfile(profile: string): void {
+  ensureBaseConfigDir();
+  migrateToProfileStructure();
+
+  if (!profileExists(profile) && profile !== DEFAULT_PROFILE) {
+    throw new Error(`Profile "${profile}" does not exist`);
+  }
+
+  writeFileSync(getCurrentProfileFile(), profile);
+}
+
+/**
+ * List all available profiles
+ */
+export function listProfiles(): string[] {
+  ensureBaseConfigDir();
+  migrateToProfileStructure();
+
+  const profilesDir = getProfilesDir();
+  if (!existsSync(profilesDir)) {
+    return [];
+  }
+
+  return readdirSync(profilesDir, { withFileTypes: true })
+    .filter(dirent => dirent.isDirectory())
+    .map(dirent => dirent.name)
+    .sort();
+}
+
+/**
+ * Create a new profile
+ */
+export function createProfile(profile: string, config: ProfileConfig = {}): boolean {
+  ensureBaseConfigDir();
+  migrateToProfileStructure();
+
+  if (profileExists(profile)) {
+    return false;
+  }
+
+  // Validate profile name
+  if (!/^[a-zA-Z0-9_-]+$/.test(profile)) {
+    throw new Error('Profile name can only contain letters, numbers, hyphens, and underscores');
+  }
+
+  const profileDir = join(getProfilesDir(), profile);
+  mkdirSync(profileDir, { recursive: true });
+
+  // Create exports and imports subdirectories
+  mkdirSync(join(profileDir, 'exports'), { recursive: true });
+  mkdirSync(join(profileDir, 'imports'), { recursive: true });
+
+  // Save initial config
+  if (Object.keys(config).length > 0) {
+    writeFileSync(join(profileDir, 'config.json'), JSON.stringify(config, null, 2));
+  }
+
+  return true;
+}
+
+/**
+ * Delete a profile
+ */
+export function deleteProfile(profile: string): boolean {
+  if (profile === DEFAULT_PROFILE) {
+    return false;
+  }
+
+  if (!profileExists(profile)) {
+    return false;
+  }
+
+  // Switch to default if deleting current profile
+  if (getCurrentProfile() === profile) {
+    setCurrentProfile(DEFAULT_PROFILE);
+  }
+
+  const profileDir = join(getProfilesDir(), profile);
+  rmSync(profileDir, { recursive: true });
+  return true;
+}
+
+/**
+ * Get the config directory for the current profile
+ */
+function resolveConfigDir(): string {
+  ensureBaseConfigDir();
+  migrateToProfileStructure();
+
+  const profile = getCurrentProfile();
+  const profileDir = join(getProfilesDir(), profile);
+
+  // Ensure profile directory exists
+  if (!existsSync(profileDir)) {
+    mkdirSync(profileDir, { recursive: true });
+  }
+
+  return profileDir;
+}
+
+// Always compute config dir dynamically (no caching - profiles can change)
+function getConfigDirInternal(): string {
+  return resolveConfigDir();
+}
+
+export function getConfigDir(): string {
+  return getConfigDirInternal();
+}
+
+export function getBaseConfigDir(): string {
+  return BASE_CONFIG_DIR;
+}
+
+export function ensureConfigDir(): void {
+  const configDir = getConfigDirInternal();
+  if (!existsSync(configDir)) {
+    mkdirSync(configDir, { recursive: true });
+  }
+}
+
+// ============================================
+// Export/Import Directories
+// ============================================
+
+export function getExportsDir(): string {
+  return join(getConfigDirInternal(), 'exports');
+}
+
+export function getImportsDir(): string {
+  return join(getConfigDirInternal(), 'imports');
+}
+
+export function ensureExportsDir(): string {
+  const dir = getExportsDir();
+  if (!existsSync(dir)) {
+    mkdirSync(dir, { recursive: true });
+  }
+  return dir;
+}
+
+export function ensureImportsDir(): string {
+  const dir = getImportsDir();
+  if (!existsSync(dir)) {
+    mkdirSync(dir, { recursive: true });
+  }
+  return dir;
+}
+
+// ============================================
+// Profile Config Loading/Saving
+// ============================================
+
+/**
+ * Load profile config
+ */
+export function loadProfile(profile?: string): ProfileConfig {
+  ensureConfigDir();
+  const profileName = profile || getCurrentProfile();
+  const profileDir = join(getProfilesDir(), profileName);
+  const configFile = join(profileDir, 'config.json');
+
+  if (!existsSync(configFile)) {
+    return {};
+  }
+
+  try {
+    return JSON.parse(readFileSync(configFile, 'utf-8'));
+  } catch {
+    return {};
+  }
+}
+
+/**
+ * Save profile config
+ */
+export function saveProfile(config: ProfileConfig, profile?: string): void {
+  ensureConfigDir();
+  const profileName = profile || getCurrentProfile();
+  const profileDir = join(getProfilesDir(), profileName);
+
+  if (!existsSync(profileDir)) {
+    mkdirSync(profileDir, { recursive: true });
+  }
+
+  writeFileSync(join(profileDir, 'config.json'), JSON.stringify(config, null, 2));
+}
+
+// ============================================
+// API Key Management
+// ============================================
+
+export function getApiKey(): string | undefined {
+  return process.env.STABILITY_API_KEY || loadProfile().apiKey;
+}
+
+export function setApiKey(apiKey: string): void {
+  const config = loadProfile();
+  config.apiKey = apiKey;
+  saveProfile(config);
+}
+
+// ============================================
+// Utility Functions
+// ============================================
+
+export function clearConfig(): void {
+  saveProfile({});
+}
+
+export function getActiveProfileName(): string {
+  return getCurrentProfile();
+}