@tscircuit/fake-snippets 0.0.66 → 0.0.68

package/bun-tests/fake-snippets-api/routes/snippets/star-count.test.ts

@@ -1,51 +0,0 @@
-import { getTestServer } from "bun-tests/fake-snippets-api/fixtures/get-test-server"
-import { expect, test } from "bun:test"
-
-test("star count is updated correctly", async () => {
-  const { axios, db } = await getTestServer()
-
-  // Create a snippet using the API
-  const snippet = {
-    unscoped_name: "TestSnippet",
-    owner_name: "testuser",
-    code: "Test Content",
-    created_at: "2023-01-01T00:00:00Z",
-    updated_at: "2023-01-01T00:00:00Z",
-    name: "testuser/TestSnippet",
-    snippet_type: "package",
-    description: "Test Description",
-  }
-  const createResponse = await axios.post("/api/snippets/create", snippet)
-  expect(createResponse.status).toBe(200)
-  const createdSnippet = createResponse.data.snippet
-
-  db.addStar("user1", createdSnippet.snippet_id)
-  db.addStar("user2", createdSnippet.snippet_id)
-  db.addStar("user3", createdSnippet.snippet_id)
-
-  // Test star count in list endpoint
-  const listResponse = await axios.get("/api/snippets/list")
-  expect(listResponse.status).toBe(200)
-  expect(listResponse.data.snippets[0].star_count).toBe(3)
-
-  // Test star count in get endpoint
-  const getResponse = await axios.get("/api/snippets/get", {
-    params: { snippet_id: createdSnippet.snippet_id },
-  })
-  expect(getResponse.status).toBe(200)
-  expect(getResponse.data.snippet.star_count).toBe(3)
-
-  await axios.post("/api/snippets/add_star", {
-    snippet_id: createdSnippet.snippet_id,
-  })
-
-  // Remove a star using the API
-  await axios.post("/api/snippets/remove_star", {
-    snippet_id: createdSnippet.snippet_id,
-  })
-
-  // Verify updated star count
-  const updatedListResponse = await axios.get("/api/snippets/list")
-  expect(updatedListResponse.status).toBe(200)
-  expect(updatedListResponse.data.snippets[0].star_count).toBe(3)
-})

package/bun-tests/fake-snippets-api/routes/snippets/update.test.ts

@@ -1,175 +0,0 @@
-import { getTestServer } from "bun-tests/fake-snippets-api/fixtures/get-test-server"
-import { test, expect } from "bun:test"
-
-test("update snippet", async () => {
-  const { axios, db } = await getTestServer()
-
-  // Add a test snippet
-  const snippet = {
-    unscoped_name: "TestSnippet",
-    owner_name: "testuser",
-    code: "Original Content",
-    created_at: "2023-01-01T00:00:00Z",
-    updated_at: "2023-01-01T00:00:00Z",
-    name: "testuser/TestSnippet",
-    snippet_type: "package",
-    description: "Original Description",
-    compiled_js: null,
-  }
-  db.addSnippet(snippet as any)
-
-  const addedPackage = db.packages[0]
-
-  // Update the snippet
-  const updatedCode = "Updated Content"
-  const updatedCompiledJs = "console.log('Updated Content')"
-  const response = await axios.post("/api/snippets/update", {
-    snippet_id: addedPackage.package_id,
-    code: updatedCode,
-    compiled_js: updatedCompiledJs,
-  })
-
-  expect(response.status).toBe(200)
-  expect(response.data.snippet.code).toBe(updatedCode)
-  expect(response.data.snippet.compiled_js).toBe(updatedCompiledJs)
-  expect(response.data.snippet.updated_at).not.toBe(addedPackage.created_at)
-
-  // Verify the snippet was updated in the database
-  const updatedPackageFiles = db.packageFiles.filter(
-    (p) => p.package_release_id === addedPackage.latest_package_release_id,
-  )
-  expect(updatedPackageFiles.length).toBe(3)
-  expect(updatedPackageFiles[0].content_text).toBe(updatedCode)
-  expect(updatedPackageFiles[1].content_text).toBe("") // dts
-  expect(updatedPackageFiles[2].content_text).toBe(updatedCompiledJs)
-})
-
-test("update non-existent snippet", async () => {
-  const { axios } = await getTestServer()
-
-  try {
-    await axios.post("/api/snippets/update", {
-      snippet_id: "non-existent-id",
-      code: "Updated Content",
-      compiled_js: "console.log('Updated Content')",
-    })
-    // If the request doesn't throw an error, fail the test
-    expect(true).toBe(false)
-  } catch (error: any) {
-    expect(error.status).toBe(404)
-    expect(error.data.error.message).toBe("Snippet not found")
-  }
-})
-
-test("update snippet with null compiled_js", async () => {
-  const { axios, db } = await getTestServer()
-
-  // Add a test snippet with compiled_js
-  const snippet = {
-    unscoped_name: "TestSnippet",
-    owner_name: "testuser",
-    code: "Original Content",
-    created_at: "2023-01-01T00:00:00Z",
-    updated_at: "2023-01-01T00:00:00Z",
-    name: "testuser/TestSnippet",
-    snippet_type: "package",
-    description: "Original Description",
-    compiled_js: "console.log('Original Content')",
-  }
-
-  db.addSnippet(snippet as any)
-
-  const addedPackage = db.packages[0]
-
-  // Update the snippet with null compiled_js
-  const response = await axios.post("/api/snippets/update", {
-    snippet_id: addedPackage.package_id,
-    compiled_js: "",
-  })
-
-  expect(response.status).toBe(200)
-  expect(response.data.snippet.compiled_js).toBeEmpty()
-
-  // Verify the snippet was updated in the database
-  const updatedPackageFiles = db.packageFiles.filter(
-    (p) => p.package_release_id === addedPackage.latest_package_release_id,
-  )
-  expect(updatedPackageFiles.length).toBe(3)
-  expect(updatedPackageFiles[0].content_text).toBe(snippet.code)
-  expect(updatedPackageFiles[1].content_text).toBe("")
-})
-
-test("update snippet after create snippet", async () => {
-  const { axios, db } = await getTestServer()
-
-  const snippet = {
-    unscoped_name: "TestSnippet",
-    code: "Test Content",
-    snippet_type: "package",
-    description: "Test Description",
-  }
-
-  await axios.post("/api/snippets/create", snippet)
-
-  const createdSnippet = db.packages[0]
-
-  const updatedCode = "Updated Content"
-  const response = await axios.post("/api/snippets/update", {
-    snippet_id: createdSnippet.package_id,
-    code: updatedCode,
-  })
-
-  expect(response.status).toBe(200)
-  expect(response.data.snippet.code).toBe(updatedCode)
-  // Don't check updated_at timestamp as it might not change in tests
-})
-
-test("update snippet visibility", async () => {
-  const { axios, db } = await getTestServer()
-
-  const snippet = {
-    unscoped_name: "TestSnippet",
-    code: "Test Content",
-    snippet_type: "package",
-    description: "Test Description",
-  }
-
-  await axios.post("/api/snippets/create", snippet)
-
-  const createdSnippet = db.packages[0]
-
-  const response = await axios.post("/api/snippets/update", {
-    snippet_id: createdSnippet.package_id,
-    is_private: true,
-  })
-
-  expect(response.status).toBe(200)
-  expect(response.data.snippet.is_private).toBe(true)
-})
-
-test("update snippet visibility with unauthenticated user", async () => {
-  const { axios, db } = await getTestServer()
-
-  const snippet = {
-    unscoped_name: "TestSnippet",
-    code: "Test Content",
-    snippet_type: "package",
-    description: "Test Description",
-  }
-
-  db.addSnippet(snippet as any)
-
-  const createdSnippet = db.packages[0]
-
-  try {
-    await axios.post("/api/snippets/update", {
-      snippet_id: createdSnippet.package_id,
-      is_private: true,
-    })
-  } catch (error: any) {
-    expect(error.status).toBe(403)
-    expect(error.data.error.message).toBe(
-      "You don't have permission to update this snippet",
-    )
-  }
-})

package/src/components/AiChatInterface.tsx

@@ -1,229 +0,0 @@
-import { useState, useRef, useEffect } from "react"
-import { Button } from "@/components/ui/button"
-import ChatInput from "./ChatInput"
-import { useAiApi } from "@/hooks/use-ai-api"
-import { createCircuitBoard1Template } from "@tscircuit/prompt-benchmarks"
-import { TextDelta } from "@anthropic-ai/sdk/resources/messages.mjs"
-import { MagicWandIcon } from "@radix-ui/react-icons"
-import { AiChatMessage } from "./AiChatMessage"
-import { useLocation } from "wouter"
-import { useSnippet } from "@/hooks/use-snippet"
-import { Edit2 } from "lucide-react"
-import { SnippetLink } from "./SnippetLink"
-import { useGlobalStore } from "@/hooks/use-global-store"
-import { useSignIn } from "@/hooks/use-sign-in"
-import { extractCodefence } from "extract-codefence"
-import { PrefetchPageLink } from "./PrefetchPageLink"
-
-export default function AIChatInterface({
-  code,
-  hasUnsavedChanges,
-  snippetId,
-  onCodeChange,
-  onStartStreaming,
-  onStopStreaming,
-  errorMessage,
-  disabled,
-}: {
-  code: string
-  disabled?: boolean
-  hasUnsavedChanges: boolean
-  snippetId?: string | null
-  onCodeChange: (code: string) => void
-  onStartStreaming: () => void
-  onStopStreaming: () => void
-  errorMessage: string | null
-}) {
-  const [messages, setMessages] = useState<AiChatMessage[]>([])
-  const [isStreaming, setIsStreaming] = useState(false)
-  const anthropic = useAiApi()
-  const messagesEndRef = useRef<HTMLDivElement>(null)
-  const { data: snippet } = useSnippet(snippetId!)
-  const [currentCodeBlock, setCurrentCodeBlock] = useState<string | null>(null)
-  const [location, navigate] = useLocation()
-  const isStreamingRef = useRef(false)
-  const isLoggedIn = useGlobalStore((s) => Boolean(s.session))
-  const signIn = useSignIn()
-
-  useEffect(() => {
-    messagesEndRef.current?.scrollIntoView({ behavior: "smooth" })
-  }, [messages])
-
-  const addMessage = async (message: string) => {
-    const newMessages = messages.concat([
-      {
-        sender: "user",
-        content: message,
-      },
-      {
-        sender: "bot",
-        content: "",
-        codeVersion: messages.filter((m) => m.sender === "bot").length,
-      },
-    ])
-    setMessages(newMessages)
-    setIsStreaming(true)
-    onStartStreaming()
-
-    try {
-      const stream = await anthropic.messages.stream({
-        model: "claude-3-sonnet-20240229",
-        system: createCircuitBoard1Template({
-          currentCode: code,
-        }),
-        messages: [
-          // TODO: include previous messages
-          {
-            role: "user",
-            content: message,
-          },
-        ],
-        max_tokens: 1000,
-      })
-
-      let accumulatedContent = ""
-      let isInCodeBlock = false
-
-      for await (const chunk of stream) {
-        if (chunk.type === "content_block_delta") {
-          const chunkText = (chunk.delta as TextDelta).text
-          accumulatedContent += chunkText
-
-          if (chunkText.includes("```")) {
-            isInCodeBlock = !isInCodeBlock
-            if (isInCodeBlock) {
-              setCurrentCodeBlock("")
-            } else {
-              const codeContent = extractCodefence(accumulatedContent)
-              if (codeContent) {
-                onCodeChange(codeContent)
-              }
-              setCurrentCodeBlock(null)
-            }
-          } else if (isInCodeBlock) {
-            setCurrentCodeBlock((prev) => {
-              const updatedCode = (prev || "") + chunkText
-              onCodeChange(updatedCode)
-              return updatedCode
-            })
-          }
-
-          setMessages((prevMessages) => {
-            const updatedMessages = [...prevMessages]
-            updatedMessages[updatedMessages.length - 1].content =
-              accumulatedContent
-            return updatedMessages
-          })
-        }
-      }
-    } catch (error) {
-      console.error("Error streaming response:", error)
-      setMessages((prevMessages) => {
-        const updatedMessages = [...prevMessages]
-        updatedMessages[updatedMessages.length - 1].content =
-          "An error occurred while generating the response."
-        return updatedMessages
-      })
-    } finally {
-      setIsStreaming(false)
-      onStopStreaming()
-    }
-  }
-
-  useEffect(() => {
-    const searchParams = new URLSearchParams(
-      window.location.search.split("?")[1],
-    )
-    const initialPrompt = searchParams.get("initial_prompt")
-
-    if (initialPrompt && messages.length === 0 && !isStreamingRef.current) {
-      isStreamingRef.current = true
-      addMessage(initialPrompt)
-    }
-  }, [])
-
-  return (
-    <div className="flex flex-col h-[calc(100vh-60px)] max-w-2xl mx-auto p-4 bg-gray-100">
-      <div className="flex-1 overflow-y-auto space-y-4 mb-4">
-        {snippet && (
-          <div className="flex pl-4 p-2 rounded items-center bg-white border border-gray-200 text-sm mb-4 shadow-sm">
-            <SnippetLink snippet={snippet} />
-            <div className="flex-grow" />
-            <PrefetchPageLink href={`/editor?snippet_id=${snippet.snippet_id}`}>
-              <Button
-                size="sm"
-                className="text-xs"
-                variant="ghost"
-                disabled={hasUnsavedChanges}
-              >
-                Open in Editor
-                <Edit2 className="w-3 h-3 ml-2 opacity-60" />
-              </Button>
-            </PrefetchPageLink>
-          </div>
-        )}
-        {messages.length === 0 && isLoggedIn && (
-          <div className="text-gray-500 text-xl text-center pt-[30vh] flex flex-col items-center">
-            <div>Submit a prompt to {snippet ? "edit!" : "get started!"}</div>
-            <div className="mt-2">
-              This is our legacy AI chat interface. For a better experience,
-              please use{" "}
-              <PrefetchPageLink href="https://chat.tscircuit.com">
-                chat.tscircuit.com
-              </PrefetchPageLink>
-              .
-            </div>
-            <div className="text-6xl mt-4">↓</div>
-          </div>
-        )}
-        {!isLoggedIn && (
-          <div className="text-gray-500 text-xl text-center pt-[30vh] flex flex-col items-center">
-            <div>
-              Sign in use the AI chat or{" "}
-              <PrefetchPageLink
-                className="text-blue-500 underline"
-                href="/quickstart"
-              >
-                use the regular editor
-              </PrefetchPageLink>
-            </div>
-            <div className="mt-4 flex gap-2">
-              <Button onClick={() => signIn()}>Sign In</Button>
-              <Button onClick={() => signIn()} variant="outline">
-                Sign Up
-              </Button>
-            </div>
-          </div>
-        )}
-        {messages.map((message, index) => (
-          <AiChatMessage key={index} message={message} />
-        ))}
-        <div ref={messagesEndRef} />
-      </div>
-      {code && errorMessage && !isStreaming && (
-        <div className="flex justify-end mr-6">
-          <Button
-            onClick={() => {
-              addMessage(`Fix this error: ${errorMessage}`)
-            }}
-            disabled={!isLoggedIn}
-            className="mb-2 bg-green-50 hover:bg-green-100"
-            variant="outline"
-          >
-            <MagicWandIcon className="w-4 h-4 mr-2" />
-            <span className="font-bold">Fix Error with AI</span>
-            <span className="italic font-normal ml-2">
-              "{errorMessage.slice(0, 26)}..."
-            </span>
-          </Button>
-        </div>
-      )}
-      <ChatInput
-        onSubmit={async (message: string) => {
-          addMessage(message)
-        }}
-        disabled={isStreaming || !isLoggedIn}
-      />
-    </div>
-  )
-}