Add temporary static homepage

platformatic · May 3, 2024 · dc18fe0 · dc18fe0
1 parent 89cedaf
commit dc18fe0
Show file tree

Hide file tree

Showing 6 changed files with 107 additions and 4 deletions.
diff --git a/config.d.ts b/config.d.ts
@@ -223,6 +223,7 @@ export interface AiWarpConfig {
     }[];
   };
   module?: string;
+  showAiWarpHomepage?: boolean;
   aiProvider:
     | {
         openai: {

diff --git a/index.ts b/index.ts
@@ -1,6 +1,8 @@
+import { join } from 'node:path'
 import { platformaticService, Stackable } from '@platformatic/service'
 import fastifyUser from 'fastify-user'
 import fastifyPlugin from 'fastify-plugin'
+import fastifyStatic from '@fastify/static'
 import { schema } from './lib/schema.js'
 import { Generator } from './lib/generator.js'
 import { AiWarpConfig } from './config.js'
@@ -19,6 +21,12 @@ const stackable: Stackable<AiWarpConfig> = async function (fastify, opts) {
   await fastify.register(rateLimitPlugin, opts)
   await fastify.register(apiPlugin, opts)
 
+  if (config.showAiWarpHomepage !== undefined && config.showAiWarpHomepage) {
+    await fastify.register(fastifyStatic, {
+      root: join(import.meta.dirname, 'static')
+    })
+  }
+
   await fastify.register(platformaticService, opts)
 }
 

diff --git a/lib/schema.ts b/lib/schema.ts
@@ -7,6 +7,10 @@ const aiWarpSchema = {
   properties: {
     ...schema.schema.properties,
     module: { type: 'string' },
+    showAiWarpHomepage: {
+      type: 'boolean',
+      default: true
+    },
     aiProvider: {
       type: 'object',
       oneOf: [

diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -8,7 +8,7 @@
     "start-ai-warp": "./dist/cli/start.js"
   },
   "scripts": {
-    "build": "tsc --build",
+    "build": "tsc --build && cp -r ./static ./dist/",
     "build:config": "node ./dist/lib/schema.js --dump-schema | json2ts > config.d.ts",
     "clean": "rm -fr ./dist",
     "lint": "ts-standard | snazzy",
@@ -38,6 +38,7 @@
     "@azure/openai": "^1.0.0-beta.12",
     "@fastify/error": "^3.4.1",
     "@fastify/rate-limit": "^9.1.0",
+    "@fastify/static": "^7.0.3",
     "@fastify/type-provider-typebox": "^4.0.0",
     "@platformatic/config": "^1.24.0",
     "@platformatic/generators": "^1.24.0",

diff --git a/static/index.html b/static/index.html
@@ -0,0 +1,88 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>AI Warp</title>
+</head>
+<body>
+    <div>
+        <label for="prompt-text">Prompt: </label>
+        <textarea name="prompt text" id="prompt-text" cols="60" rows="1"></textarea>
+        <button id="prompt-button">Prompt</button>
+    </div>
+
+    <p id="response"></p>
+
+    <script>
+        const promptText = document.getElementById('prompt-text')
+        const responseText = document.getElementById('response')
+
+        const promptButton = document.getElementById('prompt-button')
+        promptButton.onclick = async () => {
+            const res = await fetch('/api/v1/stream', {
+                method: 'POST',
+                headers: {
+                    'content-type': 'application/json'
+                },
+                body: JSON.stringify({
+                    prompt: promptText.value
+                })
+            })
+
+            if (res.status !== 200) {
+                const { message, code } = await res.json()
+                responseText.innerHTML = `Error: ${message} (${code})`
+                return
+            }
+
+            responseText.innerHTML = ''
+
+            const reader = res.body.getReader()
+            const decoder = new TextDecoder()
+            while (true) {
+                const { done, value } = await reader.read()
+                if (done) {
+                    break
+                }
+
+                const decodedValue = decoder.decode(value)
+                const lines = decodedValue.split('\n')
+
+                let i = 0
+                while (i < lines.length) {
+                    const line = lines[i]
+                    if (line.length === 0) {
+                        i++
+                        continue
+                    }
+
+                    if (!line.startsWith('event: ')) {
+                        responseText.innerHTML = `Unexpected event type line: ${line}`
+                        return
+                    }
+
+                    const dataLine = lines[i + 1]
+                    if (!dataLine.startsWith('data: ')) {
+                        responseText.innerHTML = `Unexpected data line: ${dataLine}`
+                        return
+                    }
+
+                    const eventType = line.substring('event: '.length)
+                    const data = dataLine.substring('data: '.length)
+                    const json = JSON.parse(data)
+                    if (eventType === 'content') {
+                        const { response } = json
+                        responseText.innerHTML += response
+                    } else if (eventType === 'error') {
+                        const { message, code } = data
+                        responseText.innerHTML = `Error: ${message} (${code})`
+                    }
+
+                    i += 2
+                }
+            }
+        }
+    </script>
+</body>
+</html>