From f0a07ce380ae4ff0927c3ffdd7548092448e4ba0 Mon Sep 17 00:00:00 2001
From: Donnie Adams <donnie@acorn.io>
Date: Wed, 28 Aug 2024 20:59:52 -0400
Subject: [PATCH] feat: add ability to save-as assistant

Signed-off-by: Donnie Adams <donnie@acorn.io>
---
 actions/me/scripts.tsx             |   2 +-
 actions/threads.tsx                |  18 ++++
 components/scripts/script-save.tsx | 152 +++++++++++++++++++++++++----
 3 files changed, 154 insertions(+), 18 deletions(-)

diff --git a/actions/me/scripts.tsx b/actions/me/scripts.tsx
index 37d13049..6fa25791 100644
--- a/actions/me/scripts.tsx
+++ b/actions/me/scripts.tsx
@@ -101,7 +101,7 @@ export async function getScript(id: string): Promise<ParsedScript | undefined> {
   }
 }
 
-export async function createScript(script: Script) {
+export async function createScript(script: Script): Promise<Script> {
   return await create(script, `scripts`);
 }
 
diff --git a/actions/threads.tsx b/actions/threads.tsx
index 28b51488..7dda06da 100644
--- a/actions/threads.tsx
+++ b/actions/threads.tsx
@@ -146,6 +146,24 @@ export async function renameThread(id: string, name: string) {
   );
 }
 
+export async function updateThreadScript(
+  threadId: string,
+  scriptId: string,
+  script: string
+) {
+  const threadsDir = THREADS_DIR();
+  const threadPath = path.join(threadsDir, threadId);
+  const meta = await fs.readFile(path.join(threadPath, META_FILE), 'utf-8');
+  const threadMeta = JSON.parse(meta) as ThreadMeta;
+  threadMeta.scriptId = scriptId;
+  threadMeta.script = script;
+  threadMeta.updated = new Date();
+  await fs.writeFile(
+    path.join(threadPath, META_FILE),
+    JSON.stringify(threadMeta)
+  );
+}
+
 export async function updateThreadWorkspace(id: string, workspace: string) {
   const threadsDir = THREADS_DIR();
   const threadPath = path.join(threadsDir, id);
diff --git a/components/scripts/script-save.tsx b/components/scripts/script-save.tsx
index 071a108f..37df7a1a 100644
--- a/components/scripts/script-save.tsx
+++ b/components/scripts/script-save.tsx
@@ -4,18 +4,30 @@ import {
   Dropdown,
   DropdownTrigger,
   DropdownItem,
+  Modal,
+  ModalContent,
+  ModalBody,
+  ModalFooter,
+  ModalHeader,
+  Checkbox,
+  Spinner,
 } from '@nextui-org/react';
-import React, { useContext } from 'react';
+import React, { useContext, useState } from 'react';
 import { ChatContext } from '@/contexts/chat';
 import { PiFloppyDiskThin, PiUser } from 'react-icons/pi';
-import { updateScript } from '@/actions/me/scripts';
+import { createScript, updateScript } from '@/actions/me/scripts';
 import { stringify } from '@/actions/gptscript';
 import { gatewayTool } from '@/actions/knowledge/util';
 import { MessageType } from '@/components/chat/messages';
+import { Input } from '@nextui-org/input';
+import { updateThreadScript } from '@/actions/threads';
 
 const SaveScriptDropdown = () => {
   const {
+    selectedThreadId,
     scriptId,
+    setScriptId,
+    setScript,
     tools,
     socket,
     scriptContent,
@@ -23,6 +35,13 @@ const SaveScriptDropdown = () => {
     setMessages,
   } = useContext(ChatContext);
 
+  const [isOpen, setIsOpen] = useState(false);
+  const [newScriptName, setNewScriptName] = useState('');
+  const [newScriptPrivate, setNewScriptPrivate] = useState(true);
+  const [saving, setSaving] = useState(false);
+  const [successMessage, setSuccessMessage] = useState('');
+  const [errorMessage, setErrorMessage] = useState('');
+
   const knowledgeTool = gatewayTool();
 
   function saveScript() {
@@ -59,23 +78,122 @@ const SaveScriptDropdown = () => {
     });
   }
 
+  async function saveScriptAs(newName: string) {
+    // The knowledge tool is dynamic and not controlled by the user. Don't add it to the saved tool.
+    const addedTools = tools.filter((t) => t !== knowledgeTool);
+
+    // find the root tool and then add the new tool
+    for (const block of scriptContent!) {
+      if (block.type === 'tool') {
+        block.tools = (block.tools || [])
+          .filter((t) => !addedTools.includes(t))
+          .concat(...addedTools);
+        block.name = newName;
+        break;
+      }
+    }
+
+    try {
+      const content = await stringify(scriptContent);
+      const slug =
+        newName.toLowerCase().replaceAll(' ', '-') +
+        '-' +
+        Math.random().toString(36).substring(2, 7);
+
+      const newScript = await createScript({
+        displayName: newName,
+        slug: slug,
+        content: content,
+        visibility: newScriptPrivate ? 'private' : 'public',
+      });
+
+      setScriptContent([...scriptContent]);
+      socket?.emit('saveScript');
+
+      await updateThreadScript(
+        selectedThreadId!,
+        '' + newScript.id,
+        newScript.publicURL || ''
+      );
+
+      setScriptId('' + newScript.id);
+      setScript(newScript.publicURL!);
+
+      setSuccessMessage(`New Assistant Saved As ${newScript.displayName}`);
+      await new Promise((resolve) => setTimeout(resolve, 3000));
+
+      setIsOpen(false);
+    } catch (e: any) {
+      console.error(e);
+      setErrorMessage(e.toString());
+    }
+
+    setSaving(false);
+  }
+
   return (
-    <Dropdown placement="bottom-start">
-      <DropdownTrigger>
-        <Button variant="light" isIconOnly>
-          <PiFloppyDiskThin className="size-5" />
-        </Button>
-      </DropdownTrigger>
-      <DropdownMenu>
-        {scriptId ? (
-          <DropdownItem key="save" onPress={saveScript}>
-            Save Assistant
+    <>
+      <Dropdown placement="bottom-start">
+        <DropdownTrigger>
+          <Button variant="light" isIconOnly>
+            <PiFloppyDiskThin className="size-5" />
+          </Button>
+        </DropdownTrigger>
+        <DropdownMenu>
+          {scriptId ? (
+            <DropdownItem key="save" onPress={saveScript}>
+              Save Assistant
+            </DropdownItem>
+          ) : (
+            (null as any)
+          )}
+          <DropdownItem key="save-as" onPress={() => setIsOpen(true)}>
+            Save Assistant As
           </DropdownItem>
-        ) : (
-          (null as any)
-        )}
-      </DropdownMenu>
-    </Dropdown>
+        </DropdownMenu>
+      </Dropdown>
+
+      <Modal backdrop="opaque" isOpen={isOpen} onOpenChange={setIsOpen}>
+        <ModalContent>
+          <ModalHeader className="flex flex-col gap-1">
+            Name Your New Assistant
+          </ModalHeader>
+          <ModalBody>
+            <Input
+              aria-label="new name"
+              label="New Name"
+              value={newScriptName}
+              onChange={(e) => setNewScriptName(e.target.value)}
+            />
+            <Checkbox
+              defaultSelected={newScriptPrivate}
+              onChange={(e) => setNewScriptPrivate(e.target.checked)}
+            >
+              Private
+            </Checkbox>
+            <p className="text-red-500">{errorMessage}</p>
+            <p className="text-green-500">{successMessage}</p>
+          </ModalBody>
+          <ModalFooter>
+            <Button
+              disabled={saving}
+              aria-label="save-as"
+              color="primary"
+              onPress={() => {
+                setSaving(true);
+                saveScriptAs(newScriptName);
+              }}
+            >
+              {saving ? (
+                <Spinner size="sm" className="text-center" color="white" />
+              ) : (
+                'Save As'
+              )}
+            </Button>
+          </ModalFooter>
+        </ModalContent>
+      </Modal>
+    </>
   );
 };