@chat-js/cli 0.4.0 → 0.6.0

package/templates/chat-app/lib/artifacts/code/client.tsx

@@ -6,7 +6,7 @@ import {
   type ConsoleOutput,
   type ConsoleOutputContent,
 } from "@/components/console";
-import { Artifact } from "@/components/create-artifact";
+import { Artifact, type ArtifactMetadata } from "@/components/create-artifact";
 import { config } from "@/lib/config";
 import { generateUUID, getLanguageFromFileName } from "@/lib/utils";
 
@@ -56,12 +56,36 @@ function detectRequiredHandlers(code: string): string[] {
   return handlers;
 }
 
-interface Metadata {
+export interface CodeArtifactMetadata {
   language: string;
   outputs: ConsoleOutput[];
 }
 
-export const codeArtifact = new Artifact<"code", Metadata>({
+export function isCodeArtifactMetadata(
+  metadata: ArtifactMetadata
+): metadata is CodeArtifactMetadata {
+  return (
+    metadata !== null &&
+    typeof metadata === "object" &&
+    "language" in metadata &&
+    typeof metadata.language === "string" &&
+    "outputs" in metadata &&
+    Array.isArray(metadata.outputs)
+  );
+}
+
+export function getCodeArtifactMetadata(
+  metadata: ArtifactMetadata
+): CodeArtifactMetadata {
+  return isCodeArtifactMetadata(metadata)
+    ? metadata
+    : {
+        language: "python",
+        outputs: [],
+      };
+}
+
+export const codeArtifact = new Artifact<"code", CodeArtifactMetadata>({
   kind: "code",
   description:
     "Useful for code generation; Code execution is only available for Python code.",
@@ -184,14 +208,20 @@ export const codeArtifact = new Artifact<"code", Metadata>({
               },
             ],
           }));
-        } catch (error: any) {
+        } catch (error: unknown) {
           setMetadata((metadata) => ({
             ...metadata,
             outputs: [
               ...metadata.outputs.filter((output) => output.id !== runId),
               {
                 id: runId,
-                contents: [{ type: "text", value: error.message }],
+                contents: [
+                  {
+                    type: "text",
+                    value:
+                      error instanceof Error ? error.message : String(error),
+                  },
+                ],
                 status: "failed",
               },
             ],

package/templates/chat-app/lib/artifacts/sheet/client.tsx

@@ -1,13 +1,21 @@
 import { Copy, LineChart, Redo2, Sparkles, Undo2 } from "lucide-react";
 import { parse, unparse } from "papaparse";
 import { toast } from "sonner";
-import { Artifact } from "@/components/create-artifact";
+import { Artifact, type ArtifactMetadata } from "@/components/create-artifact";
 import { SpreadsheetEditor } from "@/components/sheet-editor";
 import { config } from "@/lib/config";
 
-type Metadata = any;
+export type SheetArtifactMetadata = Record<string, unknown>;
 
-export const sheetArtifact = new Artifact<"sheet", Metadata>({
+export function getSheetArtifactMetadata(
+  metadata: ArtifactMetadata
+): SheetArtifactMetadata {
+  return metadata && typeof metadata === "object"
+    ? Object.fromEntries(Object.entries(metadata))
+    : {};
+}
+
+export const sheetArtifact = new Artifact<"sheet", SheetArtifactMetadata>({
   kind: "sheet",
   description: "Useful for working with spreadsheets",
   initialize: async () => {

package/templates/chat-app/lib/auth-client.ts

@@ -1,10 +1,32 @@
+import { electronProxyClient } from "@better-auth/electron/proxy";
 import { nextCookies } from "better-auth/next-js";
 import { createAuthClient } from "better-auth/react";
+import { config } from "@/lib/config";
+import {
+  ELECTRON_APP_SCHEME,
+  ELECTRON_AUTH_CALLBACK_PATH,
+  ELECTRON_AUTH_CLIENT_ID,
+  ELECTRON_AUTH_COOKIE_PREFIX,
+} from "@/lib/electron-auth";
 
 // Better Auth auto-detects the base URL from window.location.origin on client
 // and uses relative URLs for SSR, so we don't need to specify baseURL
 const authClient = createAuthClient({
-  plugins: [nextCookies()],
+  plugins: [
+    nextCookies(),
+    ...(config.desktopApp.enabled
+      ? [
+          electronProxyClient({
+            callbackPath: ELECTRON_AUTH_CALLBACK_PATH,
+            clientID: ELECTRON_AUTH_CLIENT_ID,
+            cookiePrefix: ELECTRON_AUTH_COOKIE_PREFIX,
+            protocol: {
+              scheme: ELECTRON_APP_SCHEME,
+            },
+          }),
+        ]
+      : []),
+  ],
 });
 
 export default authClient;

package/templates/chat-app/lib/auth.ts

@@ -1,3 +1,4 @@
+import { electron } from "@better-auth/electron";
 import { betterAuth } from "better-auth";
 import { drizzleAdapter } from "better-auth/adapters/drizzle";
 import { nextCookies } from "better-auth/next-js";
@@ -5,6 +6,11 @@ import { env } from "@/lib/env";
 import { config } from "./config";
 import { db } from "./db/client";
 import { schema } from "./db/schema";
+import {
+  ELECTRON_AUTH_CLIENT_ID,
+  ELECTRON_AUTH_COOKIE_PREFIX,
+  ELECTRON_TRUSTED_ORIGINS,
+} from "./electron-auth";
 
 export const auth = betterAuth({
   database: drizzleAdapter(db, {
@@ -16,6 +22,7 @@ export const auth = betterAuth({
     // Vercel URL for preview branches
     ...(env.VERCEL_URL ? [`https://${env.VERCEL_URL}`] : []),
     config.appUrl,
+    ...(config.desktopApp.enabled ? ELECTRON_TRUSTED_ORIGINS : []),
   ],
   secret: env.AUTH_SECRET,
 
@@ -60,7 +67,17 @@ export const auth = betterAuth({
 
     return { google, github, vercel } as const;
   })(),
-  plugins: [nextCookies()],
+  plugins: [
+    nextCookies(),
+    ...(config.desktopApp.enabled
+      ? [
+          electron({
+            clientID: ELECTRON_AUTH_CLIENT_ID,
+            cookiePrefix: ELECTRON_AUTH_COOKIE_PREFIX,
+          }),
+        ]
+      : []),
+  ],
 });
 
 // Infer session type from the auth instance for type safety

package/templates/chat-app/lib/blob.ts

@@ -67,7 +67,7 @@ export function extractFilenameFromUrl(url: string): string | null {
 
     // Remove the prefix if it exists in the URL
     if (filename.startsWith(BLOB_FILE_PREFIX)) {
-      return filename.substring(BLOB_FILE_PREFIX.length);
+      return filename.slice(BLOB_FILE_PREFIX.length);
     }
 
     return filename;

package/templates/chat-app/lib/clone-messages.ts

@@ -334,7 +334,7 @@ export function cloneMessagesWithDocuments<
     messageId: string;
     userId: string;
     title: string;
-    kind: any;
+    kind: unknown;
     content: string | null;
     createdAt: Date;
   },

package/templates/chat-app/lib/config-schema.ts

@@ -254,6 +254,16 @@ export const authenticationConfigSchema = z
     vercel: false,
   });
 
+export const desktopAppConfigSchema = z
+  .object({
+    enabled: z
+      .boolean()
+      .describe("Enable Electron desktop auth/runtime integration"),
+  })
+  .default({
+    enabled: false,
+  });
+
 export const configSchema = z.object({
   appPrefix: z.string().default("chatjs"),
   appName: z.string().default("My AI Chat"),
@@ -326,6 +336,8 @@ export const configSchema = z.object({
 
   authentication: authenticationConfigSchema,
 
+  desktopApp: desktopAppConfigSchema,
+
   ai: aiConfigSchema,
 
   anonymous: anonymousConfigSchema,
@@ -341,6 +353,7 @@ export type AnonymousConfig = z.infer<typeof anonymousConfigSchema>;
 export type AttachmentsConfig = z.infer<typeof attachmentsConfigSchema>;
 export type FeaturesConfig = z.infer<typeof featuresConfigSchema>;
 export type AuthenticationConfig = z.infer<typeof authenticationConfigSchema>;
+export type DesktopAppConfig = z.infer<typeof desktopAppConfigSchema>;
 
 // Gateway-aware input types: model IDs narrowed per gateway for autocomplete
 type ZodConfigInput = z.input<typeof configSchema>;
@@ -389,7 +402,6 @@ interface AiToolsInputFor<G extends GatewayType> {
 }
 
 // Only gateway is required; everything else is an override on top of GATEWAY_MODEL_DEFAULTS
-// biome-ignore lint/style/useConsistentTypeDefinitions: type is used intentionally here
 type AiInputFor<G extends GatewayType> = {
   gateway: G;
   providerOrder?: AiShape["providerOrder"];

package/templates/chat-app/lib/constants.ts

@@ -1,11 +1,10 @@
 import { config } from "@/lib/config";
+import { isPlaywrightTestEnvironment as getIsPlaywrightTestEnvironment } from "@/lib/playwright-test-environment";
 
 const _isProductionEnvironment = process.env.NODE_ENV === "production";
 
-const _isTestEnvironment = Boolean(
-  process.env.PLAYWRIGHT_TEST_BASE_URL ||
-    process.env.PLAYWRIGHT ||
-    process.env.CI_PLAYWRIGHT
+export const isPlaywrightTestEnvironment = getIsPlaywrightTestEnvironment(
+  process.env
 );
 
 export const BLOB_FILE_PREFIX = `${config.appPrefix}/files/`;

package/templates/chat-app/lib/db/migrations/meta/0044_snapshot.json

@@ -94,12 +94,8 @@
           "name": "account_user_id_user_id_fk",
           "tableFrom": "account",
           "tableTo": "user",
-          "columnsFrom": [
-            "user_id"
-          ],
-          "columnsTo": [
-            "id"
-          ],
+          "columnsFrom": ["user_id"],
+          "columnsTo": ["id"],
           "onDelete": "cascade",
           "onUpdate": "no action"
         }
@@ -173,12 +169,8 @@
           "name": "Chat_userId_user_id_fk",
           "tableFrom": "Chat",
           "tableTo": "user",
-          "columnsFrom": [
-            "userId"
-          ],
-          "columnsTo": [
-            "id"
-          ],
+          "columnsFrom": ["userId"],
+          "columnsTo": ["id"],
           "onDelete": "no action",
           "onUpdate": "no action"
         },
@@ -186,12 +178,8 @@
           "name": "Chat_projectId_Project_id_fk",
           "tableFrom": "Chat",
           "tableTo": "Project",
-          "columnsFrom": [
-            "projectId"
-          ],
-          "columnsTo": [
-            "id"
-          ],
+          "columnsFrom": ["projectId"],
+          "columnsTo": ["id"],
           "onDelete": "set null",
           "onUpdate": "no action"
         }
@@ -273,12 +261,8 @@
           "name": "Document_userId_user_id_fk",
           "tableFrom": "Document",
           "tableTo": "user",
-          "columnsFrom": [
-            "userId"
-          ],
-          "columnsTo": [
-            "id"
-          ],
+          "columnsFrom": ["userId"],
+          "columnsTo": ["id"],
           "onDelete": "no action",
           "onUpdate": "no action"
         },
@@ -286,12 +270,8 @@
           "name": "Document_messageId_Message_id_fk",
           "tableFrom": "Document",
           "tableTo": "Message",
-          "columnsFrom": [
-            "messageId"
-          ],
-          "columnsTo": [
-            "id"
-          ],
+          "columnsFrom": ["messageId"],
+          "columnsTo": ["id"],
           "onDelete": "cascade",
           "onUpdate": "no action"
         }
@@ -299,10 +279,7 @@
       "compositePrimaryKeys": {
         "Document_id_createdAt_pk": {
           "name": "Document_id_createdAt_pk",
-          "columns": [
-            "id",
-            "createdAt"
-          ]
+          "columns": ["id", "createdAt"]
         }
       },
       "uniqueConstraints": {},
@@ -450,12 +427,8 @@
           "name": "McpConnector_userId_user_id_fk",
           "tableFrom": "McpConnector",
           "tableTo": "user",
-          "columnsFrom": [
-            "userId"
-          ],
-          "columnsTo": [
-            "id"
-          ],
+          "columnsFrom": ["userId"],
+          "columnsTo": ["id"],
           "onDelete": "cascade",
           "onUpdate": "no action"
         }
@@ -565,12 +538,8 @@
           "name": "McpOAuthSession_mcpConnectorId_McpConnector_id_fk",
           "tableFrom": "McpOAuthSession",
           "tableTo": "McpConnector",
-          "columnsFrom": [
-            "mcpConnectorId"
-          ],
-          "columnsTo": [
-            "id"
-          ],
+          "columnsFrom": ["mcpConnectorId"],
+          "columnsTo": ["id"],
           "onDelete": "cascade",
           "onUpdate": "no action"
         }
@@ -580,9 +549,7 @@
         "McpOAuthSession_state_unique": {
           "name": "McpOAuthSession_state_unique",
           "nullsNotDistinct": false,
-          "columns": [
-            "state"
-          ]
+          "columns": ["state"]
         }
       },
       "policies": {},
@@ -692,12 +659,8 @@
           "name": "Message_chatId_Chat_id_fk",
           "tableFrom": "Message",
           "tableTo": "Chat",
-          "columnsFrom": [
-            "chatId"
-          ],
-          "columnsTo": [
-            "id"
-          ],
+          "columnsFrom": ["chatId"],
+          "columnsTo": ["id"],
           "onDelete": "cascade",
           "onUpdate": "no action"
         }
@@ -915,12 +878,8 @@
           "name": "Part_messageId_Message_id_fk",
           "tableFrom": "Part",
           "tableTo": "Message",
-          "columnsFrom": [
-            "messageId"
-          ],
-          "columnsTo": [
-            "id"
-          ],
+          "columnsFrom": ["messageId"],
+          "columnsTo": ["id"],
           "onDelete": "cascade",
           "onUpdate": "no action"
         }
@@ -1041,12 +1000,8 @@
           "name": "Project_userId_user_id_fk",
           "tableFrom": "Project",
           "tableTo": "user",
-          "columnsFrom": [
-            "userId"
-          ],
-          "columnsTo": [
-            "id"
-          ],
+          "columnsFrom": ["userId"],
+          "columnsTo": ["id"],
           "onDelete": "cascade",
           "onUpdate": "no action"
         }
@@ -1117,12 +1072,8 @@
           "name": "session_user_id_user_id_fk",
           "tableFrom": "session",
           "tableTo": "user",
-          "columnsFrom": [
-            "user_id"
-          ],
-          "columnsTo": [
-            "id"
-          ],
+          "columnsFrom": ["user_id"],
+          "columnsTo": ["id"],
           "onDelete": "cascade",
           "onUpdate": "no action"
         }
@@ -1132,9 +1083,7 @@
         "session_token_unique": {
           "name": "session_token_unique",
           "nullsNotDistinct": false,
-          "columns": [
-            "token"
-          ]
+          "columns": ["token"]
         }
       },
       "policies": {},
@@ -1208,12 +1157,8 @@
           "name": "Suggestion_userId_user_id_fk",
           "tableFrom": "Suggestion",
           "tableTo": "user",
-          "columnsFrom": [
-            "userId"
-          ],
-          "columnsTo": [
-            "id"
-          ],
+          "columnsFrom": ["userId"],
+          "columnsTo": ["id"],
           "onDelete": "no action",
           "onUpdate": "no action"
         },
@@ -1221,14 +1166,8 @@
           "name": "Suggestion_documentId_documentCreatedAt_Document_id_createdAt_fk",
           "tableFrom": "Suggestion",
           "tableTo": "Document",
-          "columnsFrom": [
-            "documentId",
-            "documentCreatedAt"
-          ],
-          "columnsTo": [
-            "id",
-            "createdAt"
-          ],
+          "columnsFrom": ["documentId", "documentCreatedAt"],
+          "columnsTo": ["id", "createdAt"],
           "onDelete": "no action",
           "onUpdate": "no action"
         }
@@ -1236,9 +1175,7 @@
       "compositePrimaryKeys": {
         "Suggestion_id_pk": {
           "name": "Suggestion_id_pk",
-          "columns": [
-            "id"
-          ]
+          "columns": ["id"]
         }
       },
       "uniqueConstraints": {},
@@ -1303,9 +1240,7 @@
         "user_email_unique": {
           "name": "user_email_unique",
           "nullsNotDistinct": false,
-          "columns": [
-            "email"
-          ]
+          "columns": ["email"]
         }
       },
       "policies": {},
@@ -1336,12 +1271,8 @@
           "name": "UserCredit_userId_user_id_fk",
           "tableFrom": "UserCredit",
           "tableTo": "user",
-          "columnsFrom": [
-            "userId"
-          ],
-          "columnsTo": [
-            "id"
-          ],
+          "columnsFrom": ["userId"],
+          "columnsTo": ["id"],
           "onDelete": "cascade",
           "onUpdate": "no action"
         }
@@ -1411,12 +1342,8 @@
           "name": "UserModelPreference_userId_user_id_fk",
           "tableFrom": "UserModelPreference",
           "tableTo": "user",
-          "columnsFrom": [
-            "userId"
-          ],
-          "columnsTo": [
-            "id"
-          ],
+          "columnsFrom": ["userId"],
+          "columnsTo": ["id"],
           "onDelete": "cascade",
           "onUpdate": "no action"
         }
@@ -1424,10 +1351,7 @@
       "compositePrimaryKeys": {
         "UserModelPreference_userId_modelId_pk": {
           "name": "UserModelPreference_userId_modelId_pk",
-          "columns": [
-            "userId",
-            "modelId"
-          ]
+          "columns": ["userId", "modelId"]
         }
       },
       "uniqueConstraints": {},
@@ -1515,12 +1439,8 @@
           "name": "Vote_chatId_Chat_id_fk",
           "tableFrom": "Vote",
           "tableTo": "Chat",
-          "columnsFrom": [
-            "chatId"
-          ],
-          "columnsTo": [
-            "id"
-          ],
+          "columnsFrom": ["chatId"],
+          "columnsTo": ["id"],
           "onDelete": "cascade",
           "onUpdate": "no action"
         },
@@ -1528,12 +1448,8 @@
           "name": "Vote_messageId_Message_id_fk",
           "tableFrom": "Vote",
           "tableTo": "Message",
-          "columnsFrom": [
-            "messageId"
-          ],
-          "columnsTo": [
-            "id"
-          ],
+          "columnsFrom": ["messageId"],
+          "columnsTo": ["id"],
           "onDelete": "cascade",
           "onUpdate": "no action"
         }
@@ -1541,10 +1457,7 @@
       "compositePrimaryKeys": {
         "Vote_chatId_messageId_pk": {
           "name": "Vote_chatId_messageId_pk",
-          "columns": [
-            "chatId",
-            "messageId"
-          ]
+          "columns": ["chatId", "messageId"]
         }
       },
       "uniqueConstraints": {},
@@ -1564,4 +1477,4 @@
     "schemas": {},
     "tables": {}
   }
-}
+}

package/templates/chat-app/lib/db/migrations/meta/_journal.json

@@ -318,4 +318,4 @@
       "breakpoints": true
     }
   ]
-}
+}

package/templates/chat-app/lib/editor/config.ts

@@ -8,8 +8,8 @@ import {
   type HeadingTagType,
   QuoteNode,
 } from "@lexical/rich-text";
-import type { LexicalEditor } from "lexical";
-import { $getSelection, $insertNodes } from "lexical";
+import type { EditorState, LexicalEditor } from "lexical";
+import { $getSelection, $insertNodes, type TextNode } from "lexical";
 
 // Create initial editor configuration
 export function createEditorConfig() {
@@ -37,7 +37,7 @@ function _createHeadingTransform(level: number) {
     export: null,
     importDOM: null,
     regExp: new RegExp(`^(#{1,${level}})\\s$`),
-    replace: (_textNode: any) => {
+    replace: (_textNode: TextNode) => {
       const selection = $getSelection();
       if (selection) {
         const headingTag = `h${level}` as HeadingTagType;
@@ -56,7 +56,7 @@ export const handleEditorChange = ({
   editor,
   onSaveContent,
 }: {
-  editorState: any;
+  editorState: EditorState;
   editor: LexicalEditor;
   onSaveContent: (updatedContent: string, debounce: boolean) => void;
 }) => {