opencode/js/example/ink.tsx
2025-05-28 15:36:35 -05:00

122 lines
3.3 KiB
TypeScript

import React, { useEffect, useState } from "react";
import type { Server } from "../src/server/server";
import type { Session } from "../src/session/session";
import { hc } from "hono/client";
import { createInterface, Interface } from "readline";
const client = hc<Server.App>(`http://localhost:16713`);
const session = await client.session_create.$post().then((res) => res.json());
const initial: {
session: {
info: {
[sessionID: string]: Session.Info;
};
message: {
[sessionID: string]: {
[messageID: string]: Session.Message;
};
};
};
} = {
session: {
info: {
[session.id]: session
},
message: {
[session.id]: {}
},
},
};
import { render, Text, Newline, useStdout, Box } from "ink";
import TextInput from "ink-text-input"
function App() {
const [state, setState] = useState(initial)
const [input, setInput] = useState("")
useEffect(() => {
fetch("http://localhost:16713/event")
.then(stream => {
const decoder = new TextDecoder();
stream.body!.pipeTo(
new WritableStream({
write(chunk) {
const data = decoder.decode(chunk);
if (data.startsWith("data: ")) {
try {
const event = JSON.parse(data.substring(6));
switch (event.type) {
case "storage.write":
const splits: string[] = event.properties.key.split("/");
let item = state as any;
for (let i = 0; i < splits.length; i++) {
const part = splits[i];
if (i === splits.length - 1) {
item[part] = event.properties.body;
continue;
}
if (!item[part]) item[part] = {};
item = item[part];
}
}
setState({ ...state })
} catch {
}
}
},
}),
)
});
}, [])
return (
<>
<Text>{session.title}</Text>
{
Object.values(state.session.message[session.id])
.filter(message => message.role !== "system")
.map(message => {
return Object.values(message.parts)
.map((part, index) => {
if (part.type === "text") {
return <Text key={`${message.id}-${index}`}>{message.role}: {part.text}</Text>
}
if (part.type === "tool-invocation") {
return <Text key={`${message.id}-${index}`}>{message.role}: {part.toolInvocation.toolName} {JSON.stringify(part.toolInvocation.args)}</Text>
}
})
})
}
<Box gap={1} >
<Text>Input:</Text>
<TextInput
value={input}
onChange={setInput}
onSubmit={() => {
setInput("")
client.session_chat.$post({
json: {
sessionID: session.id,
parts: [
{
type: "text",
text: input,
},
],
}
})
}}
/>
</Box>
</>
);
};
console.clear();
render(<App />);