@animaapp/anima-sdk 0.1.0 → 0.2.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@animaapp/anima-sdk",
-  "version": "0.1.0",
+  "version": "0.2.0",
   "type": "module",
   "description": "Anima's JavaScript utilities library",
   "author": "Anima App, Inc.",
@@ -16,8 +16,7 @@
     "url": "git+https://github.com/AnimaApp/anima-sdk.git"
   },
   "publishConfig": {
-    "access": "public",
-    "registry": "https://registry.npmjs.org/"
+    "access": "public"
   },
   "scripts": {
     "build": "vite build",
@@ -35,4 +34,4 @@
     "vite-tsconfig-paths": "^5.1.4",
     "vitest": "^3.0.5"
   }
-}
+}

package/src/anima.ts

@@ -1,13 +1,13 @@
+import { convertCodegenFilesToAnimaFiles } from "./codegenToAnimaFiles";
 import { CodegenError } from "./errors";
 import { validateSettings } from "./settings";
 import {
-  AnimaFiles,
   AnimaSDKResult,
   GetCodeHandler,
   GetCodeParams,
   SSECodgenMessage,
 } from "./types";
-import { convertCodegenFilesToAnimaFiles } from "./codegenToAnimaFiles";
+
 export type Auth =
   | { token: string; teamId: string } // for Anima user, it's mandatory to have an associated team
   | { token: string; userId?: string }; // for users from a 3rd-party client (e.g., Bolt) they may have optionally a user id
@@ -77,11 +77,14 @@ export class Anima {
         fileKey: params.fileKey,
         figmaToken: params.figmaToken,
         nodesId: params.nodesId,
+        assetsStorage: params.assetsStorage,
         language: settings.language,
+        model: settings.model,
         framework: settings.framework,
         styling: settings.styling,
         uiLibrary: settings.uiLibrary,
         enableTranslation: settings.enableTranslation,
+        enableUILibraryTheming: settings.enableUILibraryTheming,
       }),
     });
 
@@ -162,6 +165,15 @@ export class Anima {
                 break;
               }
 
+              case "assets_list": {
+                result.assets = data.payload.assets;
+
+                typeof handler === "function"
+                  ? handler(data)
+                  : handler.onAssetsList?.(data.payload);
+                break;
+              }
+
               case "figma_metadata": {
                 result.figmaFileName = data.figmaFileName;
                 result.figmaSelectedFrameName = data.figmaSelectedFrameName;
@@ -169,9 +181,9 @@ export class Anima {
                 typeof handler === "function"
                   ? handler(data)
                   : handler.onFigmaMetadata?.({
-                    figmaFileName: data.figmaFileName,
-                    figmaSelectedFrameName: data.figmaSelectedFrameName,
-                  });
+                      figmaFileName: data.figmaFileName,
+                      figmaSelectedFrameName: data.figmaSelectedFrameName,
+                    });
                 break;
               }
 
@@ -182,7 +194,7 @@ export class Anima {
                 >;
 
                 const files = convertCodegenFilesToAnimaFiles(codegenFiles);
-                
+
                 if (data.payload.status === "success") {
                   result.files = files;
                 }
@@ -190,10 +202,10 @@ export class Anima {
                 typeof handler === "function"
                   ? handler(data)
                   : handler.onGeneratingCode?.({
-                    status: data.payload.status,
-                    progress: data.payload.progress,
-                    files,
-                  });
+                      status: data.payload.status,
+                      progress: data.payload.progress,
+                      files,
+                    });
                 break;
               }
 

package/src/dataStream.ts

@@ -86,13 +86,14 @@ export const createCodegenResponseEventStream = async (
     });
   }
 
+  const encoder = new TextEncoder();
   const seeStream = consumerStream.pipeThrough(
     new TransformStream({
       transform(chunk, controller) {
         const sseString = `event: ${chunk.type}\ndata: ${JSON.stringify(
           chunk
         )}\n\n`;
-        controller.enqueue(sseString);
+        controller.enqueue(encoder.encode(sseString));
       },
     })
   );

package/src/settings.ts

@@ -8,6 +8,7 @@ const CodegenSettingsSchema = z
     z.union([
       z.object({
         framework: z.literal("react"),
+        model: z.string().optional(),
         styling: z.enum([
           "plain_css",
           "css_modules",
@@ -18,6 +19,7 @@ const CodegenSettingsSchema = z
           "inline_styles",
         ]),
         uiLibrary: z.enum(["mui", "antd", "radix", "shadcn"]).optional(),
+        enableUILibraryTheming: z.boolean().optional(),
       }),
       z.object({
         framework: z.literal("html"),
@@ -30,6 +32,7 @@ const CodegenSettingsSchema = z
 // We don't use the z.infer method here because the types returned by zod aren't ergonic
 export type CodegenSettings = {
   language?: "typescript" | "javascript";
+  model?: string,
   framework: "react" | "html";
   styling:
     | "plain_css"
@@ -41,6 +44,7 @@ export type CodegenSettings = {
     | "inline_styles"
   uiLibrary?: "mui" | "antd" | "radix" | "shadcn";
   enableTranslation?: boolean;
+  enableUILibraryTheming?: boolean;
 };
 
 export const validateSettings = (obj: unknown): CodegenSettings => {

package/src/types.ts

@@ -17,16 +17,22 @@ export type BaseResult = {
 
 export type AnimaSDKResult = BaseResult & {
   files: AnimaFiles;
+  assets?: Array<{ name: string; url: string }>;
 };
 
 export type CodegenResult = BaseResult & {
   files: Record<string, { code: string; type: "code" }>;
 };
 
+export type AssetsStorage =
+  | { strategy: "host" }
+  | { strategy: "external"; url: string };
+
 export type GetCodeParams = {
   fileKey: string;
   figmaToken: string;
   nodesId: string[];
+  assetsStorage?: AssetsStorage;
   settings: CodegenSettings;
 };
 
@@ -36,6 +42,11 @@ export type GetCodeHandler =
       onStart?: ({ sessionId }: { sessionId: string }) => void;
       onPreCodegen?: ({ message }: { message: string }) => void;
       onAssetsUploaded?: () => void;
+      onAssetsList?: ({
+        assets,
+      }: {
+        assets: Array<{ name: string; url: string }>;
+      }) => void;
       onFigmaMetadata?: ({
         figmaFileName,
         figmaSelectedFrameName,
@@ -67,6 +78,10 @@ export type SSECodgenMessage =
   | { type: "generating_code"; payload: any }
   | { type: "codegen_completed" }
   | { type: "assets_uploaded" }
+  | {
+      type: "assets_list";
+      payload: { assets: Array<{ name: string; url: string }> };
+    }
   | { type: "aborted" }
   | { type: "error"; payload: SSECodgenMessageErrorPayload }
   | { type: "done" };

package/tsconfig.json

@@ -0,0 +1,5 @@
+{
+  "$schema": "https://json.schemastore.org/tsconfig",
+  "extends": "../tsconfig.json",
+  "exclude": ["../sdk-react"]
+}